awips2/cave/build/tools/convCT.py
2017-04-21 18:33:55 -06:00

63 lines
1.8 KiB
Python

#!/usr/bin/env python
##
# This software was developed and / or modified by Raytheon Company,
# pursuant to Contract DG133W-05-CQ-1067 with the US Government.
#
# U.S. EXPORT CONTROLLED TECHNICAL DATA
# This software product contains export-restricted data whose
# export/transfer/disclosure is restricted by U.S. law. Dissemination
# to non-U.S. persons whether in the United States or abroad requires
# an export license or other authorization.
#
# Contractor Name: Raytheon Company
# Contractor Address: 6825 Pine Street, Suite 340
# Mail Stop B8
# Omaha, NE 68106
# 402.291.0100
#
# See the AWIPS II Master Rights File ("Master Rights File.pdf") for
# further licensing information.
##
# Converts netcdf style colormaps to AWIPS II XML colormaps
#
# Usage: ./convCT.py colormap1 colormap2 colormap3
#
# Requires scipy and numpy
#
# Deposits files in /tmp
#
# SOFTWARE HISTORY
# Date Ticket# Engineer Description
# ------------ ---------- ----------- --------------------------
# Jun 23, 2008 chammack Initial creation
#
import pupynere as netcdf
import numpy
import sys
import os
def convert(i):
return str((i & 0xFF) / 255.0)
ct = sys.argv
numct = len(ct)
for k in range(1, numct):
print 'Converting: ' + ct[k]
nc = netcdf.netcdf_file(ct[k], "r")
colors = nc.variables['tableColors'][:][0]
f = open('/tmp/' + os.path.basename(ct[k]).replace('.COLORTABLE', '.cmap'), 'w')
f.write('<colorMap>\n')
aVal = 1.0
for i in range(numpy.shape(colors)[1]):
f.write(" <color ")
f.write('r = "' + convert(colors[0,i]) + '" ')
f.write('g = "' + convert(colors[1,i]) + '" ')
f.write('b = "' + convert(colors[2,i]) + '" ')
f.write('a = "' + str(aVal) + '" ')
f.write('/>\n')
f.write('</colorMap>\n')
f.close()