Salome HOME
278ce201702286399406f0f8423ea5dc99896545
[modules/smesh.git] / doc / salome / gui / SMESH / collect_mesh_methods.py
1 #!/usr/bin/env python
2 #################################################################################
3 #
4 # File:   collect_mesh_methods.py
5 # Author: Vadim SANDLER, Open CASCADE S.A.S (vadim.sandler@opencascade.com)
6 #
7 #################################################################################
8 #
9 # Extraction of the meshing algorithm classes
10 # dynamically added by the plug-in to the Mesh
11 # class.
12
13 # This script is intended for internal usage - only
14 # for generatation of the extra developer documentation for
15 # the meshing plug-in(s).
16
17 # Usage:
18 #       collect_mesh_methods.py <plugin_name>
19 # where
20 #   <plugin_name> is a name of the plug-in module
21 #
22 # Notes:
23 # - the script is supposed to be run in correct environment
24 # i.e. PYTHONPATH, SMESH_MeshersList and other important
25 # variables are set properly; otherwise the script will fail.
26 #
27 ################################################################################
28
29 import sys
30
31 def main(plugin, dummymeshhelp = True, output_file = "smesh.py"):
32     plugin_module = plugin + "DC"
33     try:
34         mod = __import__(plugin_module)
35         methods = {}
36         for attr in dir( mod ):
37             if attr.startswith( '_' ): continue
38             algo = getattr( mod, attr )
39             if type( algo ).__name__ == 'classobj' and hasattr( algo, "meshMethod" ):
40                 method = getattr( algo, "meshMethod" )
41                 if method not in methods: methods[ method ] = []
42                 methods[ method ].append( algo )
43                 pass
44             pass
45         if methods:
46             output = []
47             if dummymeshhelp:
48                 # Add dummy Mesh help
49                 # This is supposed to be done when generating documentation for meshing plug-ins
50                 output.append( "## This class allows defining and managing a mesh." )
51                 output.append( "#" )
52                 output.append( "#  @note The documentation below does not provide complete description of class @b %Mesh" )
53                 output.append( "#  from @b %smesh.py package. This documentation provides only information about" )
54                 output.append( "#  the methods dynamically added to the %Mesh class by the " + plugin + " plugin" )
55                 output.append( "#  For more details on the %Mesh class, please refer to the SALOME %Mesh module" )
56                 output.append( "#  documentation." )
57                 pass
58             else:
59                 # Extend documentation for Mesh class with information about dynamically added methods.
60                 # This is supposed to be done only when building documentation for SMESH module
61                 output.append( "## This class allows defining and managing a mesh." )
62                 output.append( "#" )
63                 output.append( "#  @note Some methods are dynamically added to the @b %Mesh class in runtime by meshing " )
64                 output.append( "#  plug-in modules. If you fail to find help on some methods in the documentation of SMESH module, " )
65                 output.append( "#  try to look into the documentation for the meshing plug-ins." )
66                 pass
67             output.append( "class Mesh:" )
68             for method in methods:
69                 docHelper = ""
70                 for algo in methods[ method ]:
71                     if hasattr( algo, "docHelper" ): docHelper = getattr( algo, "docHelper" )
72                     if docHelper: break
73                     pass
74                 if not docHelper: docHelper = "Creates new algorithm."
75                 output.append( " ## %s." % docHelper )
76                 output.append( " #  This method is dynamically added to %Mesh class by the meshing plug-in(s). " )
77                 output.append( " #" )
78                 output.append( " #  If the optional @a geom_shape parameter is not set, this algorithm is global (applied to whole mesh)." )
79                 output.append( " #  Otherwise, this algorithm defines a submesh based on @a geom_shape subshape." )
80                 output.append( " #  @param algo_type type of algorithm to be created; allowed values are specified by classes implemented by plug-in (see below)" )
81                 output.append( " #  @param geom_shape if defined, the subshape to be meshed (GEOM_Object)" )
82                 output.append( " #  @return An instance of Mesh_Algorithm sub-class according to the specified @a algo_type:" )
83                 output.append( " #  %s" % ", ".join( [ "%s.%s" % ( plugin_module, algo.__name__ ) for algo in methods[ method ] ] ) )
84                 output.append( " def %s(algo_type, geom_shape=0):" % method )
85                 output.append( "   pass" )
86                 pass
87             f = open(output_file, "w")
88             for line in output: f.write( line + "\n" )
89             f.close()
90             pass
91         pass
92     except Exception, e:
93         print e
94         pass
95     pass
96     
97 if __name__ == "__main__":
98     import optparse
99     parser = optparse.OptionParser(usage="%prog [options] plugin")
100     h  = "Output file (smesh.py by default)"
101     parser.add_option("-o", "--output", dest="output",
102                       action="store", default=None, metavar="file",
103                       help=h)
104     h  = "If this option is True, dummy help for Mesh class is added. "
105     h += "This option should be False (default) when building documentation for SMESH module "
106     h += "and True when building documentation for meshing plug-ins."
107     parser.add_option("-d", "--dummy-mesh-help", dest="dummymeshhelp",
108                       action="store_true", default=False,
109                       help=h)
110     (options, args) = parser.parse_args()
111
112     if len( args ) < 1: sys.exit("Plugin name is not specified")
113     main( args[0], options.dummymeshhelp, options.output )
114     pass