doc/salome/gui/SMESH/merge_mesh_class.py

   1 import inspect
   2 import sys
   3 from types import FunctionType
   4 import copy
   5
   6 ORIGIN_MODULE_SUFFIX = "_origin"
   7 DYNAMIC_MODULE_SUFFIX = "_dynamic"
   8
   9
  10 def main(module_name, output_file = "smeshBuilder.py"):
  11     oringin_module_name = module_name + ORIGIN_MODULE_SUFFIX
  12     dynamic_module_name = module_name + DYNAMIC_MODULE_SUFFIX
  13     try:
  14         exec( "import %s" % oringin_module_name )
  15         exec( "origin_module = %s" % oringin_module_name )
  16         origin_module_lines = inspect.getsourcelines( origin_module )[0]
  17         origin_meshClass_lines = inspect.getsourcelines(origin_module.Mesh)[0]
  18         origin_module_text = "".join( origin_module_lines )
  19         origin_meshClass_text = "".join( origin_meshClass_lines )
  20
  21         exec( "import %s" % dynamic_module_name )
  22         exec( "dynanmic_module = %s" % dynamic_module_name )
  23         dynanmic_meshClass = dynanmic_module.Mesh
  24
  25         new_meshClass_lines = copy.copy(origin_meshClass_lines)
  26         # remove end of class 'pass'
  27         if new_meshClass_lines[-1].find("pass") > 0:
  28             new_meshClass_lines.pop()
  29
  30         dynanmic_meshClass_methods = [x for x, y in dynanmic_meshClass.__dict__.items() if type(y) == FunctionType]
  31         for method in dynanmic_meshClass_methods:
  32             exec( "method_lines = inspect.getsourcelines(dynanmic_module.Mesh.%s)[0]"  % method)
  33             new_meshClass_lines+=method_lines
  34             pass
  35         new_meshClass_text = "".join( new_meshClass_lines )
  36
  37         f = open( output_file, "w" )
  38
  39         f.write( origin_module_text.replace( origin_meshClass_text, new_meshClass_text) )
  40         f.close()
  41     except Exception, e:
  42         print e
  43         pass
  44     pass
  45
  46
  47 if __name__ == "__main__":
  48     import optparse
  49     parser = optparse.OptionParser(usage="%prog [options] modulename")
  50     h  = "Output file (smeshBuilder.py by default)"
  51     parser.add_option("-o", "--output", dest="output",
  52                       action="store", default="smeshBuilder.py", metavar="file",
  53                       help=h)
  54
  55     (options, args) = parser.parse_args()
  56
  57     if len( args ) < 1: sys.exit("Module name is not specified")
  58     main( args[0], options.output )
  59     pass