32a0b19a38
Change-Id: I7b8c704ebc847169c2d09bb0bea13d0b60a9cff6 Former-commit-id: 2aab48d1032c0b6fd14645437836ad5ca13cb415
117 lines
No EOL
4.5 KiB
Python
117 lines
No EOL
4.5 KiB
Python
import sys
|
|
import xml.etree.ElementTree as ET
|
|
|
|
xsitype = '{http://www.w3.org/2001/XMLSchema-instance}type'
|
|
|
|
def upgradeBundle(bundleFile):
|
|
tree = ET.parse(bundleFile)
|
|
root = tree.getroot()
|
|
iterpath = 'bundles/bundle/displayList/displays'
|
|
if root.tag == 'bundle':
|
|
iterpath = 'displayList/displays'
|
|
for display in root.iterfind(iterpath):
|
|
if display.get(xsitype) == "skewtDisplay":
|
|
plugins = getPlugins(display)
|
|
nsharp = False
|
|
varheight = False
|
|
for plugin in plugins:
|
|
nsharp |= isNsharp(plugin)
|
|
varheight |= isHodoVarHeight(plugin)
|
|
if varheight and nsharp:
|
|
# This will cause the bundle to continue loading old sounding,
|
|
# this is not a big problem until that is deleted
|
|
print 'Cannot convert bundle with both var height hodo and nsharp'
|
|
elif varheight:
|
|
convertDisplayToHodoVarHeight(display)
|
|
elif nsharp:
|
|
convertDisplayToNsharp(display)
|
|
elif display.get(xsitype) == "d2DNSharpDisplay":
|
|
display.set(xsitype,'nsharpSkewTPaneDisplay')
|
|
descriptor = display.find('descriptor')
|
|
descriptor.set(xsitype,'nsharpSkewTPaneDescriptor')
|
|
tree.write(bundleFile.replace('.xml','-converted.xml'))
|
|
|
|
def getPlugins(xmlDisplay):
|
|
plugins = set()
|
|
for resourceData in xmlDisplay.iterfind('descriptor/resource/resourceData'):
|
|
plugin = getPluginName(resourceData)
|
|
if plugin is not None:
|
|
plugins.add(plugin)
|
|
return plugins;
|
|
|
|
def getPluginName(resourceData):
|
|
if resourceData.get(xsitype) == 'gribSoundingSkewTResourceData':
|
|
return 'grib'
|
|
elif resourceData.get(xsitype) == 'skewTResourceData':
|
|
return getConstraintValue(resourceData, 'pluginName')
|
|
return None
|
|
|
|
def getConstraintValue(resourceData, key):
|
|
for mapping in resourceData.iterfind('metadataMap/mapping'):
|
|
if(mapping.get('key') == key):
|
|
return mapping.find('constraint').get('constraintValue')
|
|
return None
|
|
|
|
def isNsharp(plugin):
|
|
return plugin == 'grib' or plugin == 'bufrua' or\
|
|
plugin == 'goes' or plugin == 'poes' or\
|
|
plugin == 'acarssounding' or\
|
|
plugin == 'NAM' or plugin == 'GFS'
|
|
|
|
def isHodoVarHeight(plugin):
|
|
return plugin == 'radar' or plugin == 'profiler'
|
|
|
|
def convertDisplayToNsharp(xmlDisplay):
|
|
# TODO this will be written when all the nsharp types are converted.
|
|
pass
|
|
|
|
def convertDisplayToHodoVarHeight(xmlDisplay):
|
|
xmlDisplay.set(xsitype,'varHeightRenderableDisplay')
|
|
xmlDisplay.set('tabTitle','Var vs height : Log 1050-150')
|
|
descriptor = xmlDisplay.find('descriptor')
|
|
descriptor.set(xsitype,'varHeightHodoDescriptor')
|
|
toRemove = []
|
|
for resource in descriptor.iterfind('resource'):
|
|
resourceData = resource.find('resourceData')
|
|
pluginName = getPluginName(resourceData)
|
|
type = resourceData.get(xsitype)
|
|
if type == 'skewTBkgResourceData':
|
|
toRemove.append(resource)
|
|
elif type == 'skewTResourceData':
|
|
resourceData.set(xsitype,'varHeightResourceData')
|
|
resourceData.set('parameter','Wind')
|
|
resourceData.set('parameterName','Wind')
|
|
if pluginName == 'radar':
|
|
ET.SubElement(resourceData, 'source').text = 'VWP'
|
|
else:
|
|
ET.SubElement(resourceData, 'source').text = pluginName
|
|
else:
|
|
print "Removing unrecognized resource of type: " + type
|
|
descriptor.remove(resource)
|
|
for resource in toRemove:
|
|
descriptor.remove(resource)
|
|
|
|
heightScale = ET.SubElement(descriptor, 'heightScale')
|
|
heightScale.set('unit','MILLIBARS')
|
|
heightScale.set('name','Log 1050-150')
|
|
heightScale.set('minVal','1050.0')
|
|
heightScale.set('maxVal','150.0')
|
|
heightScale.set('parameter','P')
|
|
heightScale.set('parameterUnit','hPa')
|
|
heightScale.set('scale','LOG')
|
|
heightScale.set('heightType','PRESSURE')
|
|
ET.SubElement(heightScale, 'labels').text = '1000,850,700,500,400,300,250,200,150'
|
|
|
|
gridGeometry = descriptor.find('gridGeometry')
|
|
gridGeometry.set('rangeX','0 999')
|
|
gridGeometry.set('rangeY','0 999')
|
|
gridGeometry.set('envelopeMinX','0.0')
|
|
gridGeometry.set('envelopeMaxX','1000.0')
|
|
gridGeometry.set('envelopeMinY','0.0')
|
|
gridGeometry.set('envelopeMaxY','1000.0')
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
for arg in sys.argv[1:]:
|
|
upgradeBundle(arg) |