[otb] add helper script

git-svn-id: http://sextante.googlecode.com/svn/trunk/soft/bindings/qgis-plugin@77 881b9c09-3ef8-f3c2-ec3d-21d735c97f4d
qgis · Apr 13, 2012 · 1a1d9d6 · 1a1d9d6
1 parent 53b99cb
commit 1a1d9d6
Showing 1 changed file with 267 additions and 0 deletions.
diff --git a/src/sextante/otb/helper/generate_application_descriptors.py b/src/sextante/otb/helper/generate_application_descriptors.py
@@ -0,0 +1,267 @@
+#!/usr/bin/python
+
+import os, sys
+import otbApplication
+
+outputpath='/home/jmalik/otb-sextante'
+endl = os.linesep
+
+def convertendl(s):
+    '''Convert a string for compatibility in txt dump'''
+    return s.replace('\n', '\\n')
+
+def get_app_list():
+  blackList = ["TestApplication"]
+  appNames = [app for app in otbApplication.Registry.GetAvailableApplications() if app not in blackList]
+  return appNames
+
+
+def generate_all_app_descriptors( ) :
+  appliIdx = 0
+  for appliname in get_app_list():
+    appliIdx = appliIdx+1
+    generate_app_descriptor( appliname )
+
+def generate_app_descriptor( appliname ) :
+  print appliname
+
+  appInstance = otbApplication.Registry.CreateApplication(appliname)
+  appInstance.UpdateParameters() # TODO need this ?
+
+  out = ""
+
+  # the executable
+  out += "otbcli_" + appliname + endl
+
+  # long name
+  # for the moment, long name == short name
+  out += convertendl(appInstance.GetName()) + endl
+
+  # group
+  out += "OTB" + endl
+
+  for paramKey in appInstance.GetParametersKeys():
+    pdesc =  generate_param_descriptor(appInstance, paramKey) 
+    if len(pdesc) > 0:
+      out += pdesc + endl
+
+  with open( os.path.join(outputpath, appliname + '.txt'), 'w' ) as outfile:
+    outfile.write(out)
+
+
+
+def generate_param_descriptor( appInstance, paramKey ):
+  paramcreationfunction = {
+      otbApplication.ParameterType_Empty : generate_parameter_Empty,
+      otbApplication.ParameterType_Int : generate_parameter_Int,
+      otbApplication.ParameterType_Float : generate_parameter_Float,
+      otbApplication.ParameterType_String : generate_parameter_String,
+      otbApplication.ParameterType_StringList : generate_parameter_NOTHANDLED,
+      otbApplication.ParameterType_InputFilename : generate_parameter_InputFilename,
+      otbApplication.ParameterType_OutputFilename : generate_parameter_OutputFilename,
+      otbApplication.ParameterType_Directory : generate_parameter_Directory,
+      otbApplication.ParameterType_Choice : generate_parameter_Choice,
+      otbApplication.ParameterType_InputImage : generate_parameter_InputImage,
+      otbApplication.ParameterType_InputImageList : generate_parameter_InputImageList,
+      otbApplication.ParameterType_InputVectorData : generate_parameter_InputVectorData,
+      otbApplication.ParameterType_InputVectorDataList : generate_parameter_InputVectorDataList,
+      otbApplication.ParameterType_OutputImage : generate_parameter_OutputImage,
+      otbApplication.ParameterType_OutputVectorData : generate_parameter_OutputVectorData,
+      otbApplication.ParameterType_Radius : generate_parameter_Radius,
+      otbApplication.ParameterType_Group : generate_parameter_NOTHANDLED,
+      otbApplication.ParameterType_ListView : generate_parameter_NOTHANDLED,
+      otbApplication.ParameterType_ComplexInputImage : generate_parameter_ComplexInputImage,
+      otbApplication.ParameterType_ComplexOutputImage : generate_parameter_ComplexOutputImage,
+      otbApplication.ParameterType_RAM : generate_parameter_RAM
+      }
+  return paramcreationfunction[ appInstance.GetParameterType(paramKey) ](appInstance, paramKey)
+
+def generate_parameter_Empty( appInstance, paramKey ):
+  out = "ParameterBoolean"
+  out += "|"
+  out += "-" + paramKey
+  out += "|"
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+  if appInstance.IsParameterEnabled(paramKey):
+    out += "True"
+  return out
+
+
+def generate_parameter_Int( appInstance, paramKey ):
+  out = "ParameterNumber"
+  out += "|"
+
+  out += "-" + paramKey
+  out += "|"
+
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+
+  out += "None"
+  out += "|"
+  out += "None"
+  out += "|"
+
+  defaultVal = "None"
+  try:
+    defaultVal = str(appInstance.GetParameterInt(paramKey))
+  except:
+    pass
+  out += defaultVal
+  return out
+
+
+def generate_parameter_Float( appInstance, paramKey ):
+  out = "ParameterNumber"
+  out += "|"
+
+  out += "-" + paramKey
+  out += "|"
+
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+
+  out += "None"
+  out += "|"
+  out += "None"
+  out += "|"
+
+  defaultVal = "None"
+  try:
+    defaultVal = str(appInstance.GetParameterFloat(paramKey))
+  except:
+    pass
+  out += defaultVal
+  return out
+
+def generate_parameter_String( appInstance, paramKey ):
+  out = "ParameterString"
+  out += "|"
+
+  out += "-" + paramKey
+  out += "|"
+
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+
+  defaultVal = ""
+  try:
+    defaultVal = str(appInstance.GetParameterString(paramKey))
+  except:
+    pass
+  out += defaultVal
+  return out
+
+
+def generate_parameter_InputFilename( appInstance, paramKey ):
+  return generate_parameter_String( appInstance, paramKey )
+
+def generate_parameter_OutputFilename( appInstance, paramKey ):
+  return generate_parameter_String( appInstance, paramKey )
+
+def generate_parameter_Directory( appInstance, paramKey ):
+  return generate_parameter_String( appInstance, paramKey )
+
+def generate_parameter_Choice( appInstance, paramKey ):
+  out = "ParameterSelection"
+  out += "|"
+
+  out += "-" + paramKey
+  out += "|"
+
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+
+  for choice in appInstance.GetChoiceKeys(paramKey):
+    out += choice
+    out += ";"
+  return out
+
+def generate_parameter_InputImage( appInstance, paramKey ):
+  out = "ParameterRaster"
+  out += "|"
+
+  out += "-" + paramKey
+  out += "|"
+
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+
+  out += str(appInstance.IsMandatory(paramKey))
+
+  return out
+
+
+def generate_parameter_InputImageList( appInstance, paramKey ):
+  return ""
+
+def generate_parameter_InputVectorData( appInstance, paramKey ):
+  out = "ParameterVector"
+  out += "|"
+
+  out += "-" + paramKey
+  out += "|"
+
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+
+  out += str(appInstance.IsMandatory(paramKey))
+
+  return out
+
+
+def generate_parameter_InputVectorDataList( appInstance, paramKey ):
+  return ""
+
+
+def generate_parameter_OutputImage( appInstance, paramKey ):
+  out = "OutputRaster"
+  out += "|"
+
+  out += "-" + paramKey
+  out += "|"
+
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+
+  return out
+
+
+def generate_parameter_NOTHANDLED( appInstance, paramKey ):
+  return ""
+
+
+def generate_parameter_OutputVectorData( appInstance, paramKey ):
+  out = "OutputVector"
+  out += "|"
+
+  out += "-" + paramKey
+  out += "|"
+
+  out += convertendl(appInstance.GetParameterDescription(paramKey))
+  out += "|"
+
+  return out
+
+
+def generate_parameter_Radius( appInstance, paramKey ):
+  return generate_parameter_Int(appInstance, paramKey)
+
+
+def generate_parameter_ComplexInputImage( appInstance, paramKey ):
+  return generate_parameter_InputImage(appInstance, paramKey)
+
+
+def generate_parameter_ComplexOutputImage( appInstance, paramKey ):
+  return generate_parameter_OutputImage(appInstance, paramKey)
+
+
+def generate_parameter_RAM( appInstance, paramKey ):
+  return generate_parameter_Int(appInstance, paramKey)
+
+
+
+
+if __name__ == "__main__" :
+  generate_all_app_descriptors()