src/PythonAPI/model/dump/DumpAssistant.py

   1 # Copyright (C) 2014-2019  CEA/DEN, EDF R&D
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License, or (at your option) any later version.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 #
  17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 #
  19
  20 """Package for dumping purposes.
  21 """
  22
  23 import ModelHighAPI
  24
  25 import ModelAPI
  26 import SketchAPI
  27
  28 import sys
  29 import inspect
  30
  31 def singleton(cls):
  32     instance = cls()
  33     instance.__call__ = lambda: instance
  34     return instance
  35
  36
  37 ## @ingroup CPPHighAPI
  38 #  Collect information about features that may be dumped and stores the model as a Python script
  39 @singleton
  40 class DumpAssistant(ModelHighAPI.ModelHighAPI_Dumper):
  41
  42     ## Constructor
  43     def __init__(self):
  44         ModelHighAPI.ModelHighAPI_Dumper.__init__(self)
  45         ModelHighAPI.ModelHighAPI_Dumper.setInstance(self)
  46         self.collectFeatures()
  47         self.myEngine = None
  48
  49     ## Collect feature wrappers, which allow dumping (have method dump)
  50     def collectFeatures(self):
  51         self.myFeatures = {}
  52         self.myWrapperNames = {}
  53         for aModule in sys.modules:
  54             for aName, anObj in inspect.getmembers(sys.modules[aModule], inspect.isclass):
  55                 if issubclass(anObj, ModelHighAPI.ModelHighAPI_Interface) and hasattr(anObj, "ID") and anObj.dump != ModelHighAPI.ModelHighAPI_Interface.dump:
  56                     self.myFeatures[anObj.ID()] = anObj
  57                     self.myWrapperNames[anObj.ID()] = aName
  58
  59     ## Create wrapper for a given feature and dump it
  60     def dumpFeature(self, theFeature, theForce):
  61         aDumper = None
  62         aFeatureKind = theFeature.getKind()
  63         if aFeatureKind in self.myFeatures:
  64             # Dump only feature created by user (in history).
  65             # Also dump Export and RemoveResults features (hard-coded here in order not to change the data model).
  66             # For all other features, just keep their name.
  67             if theForce or theFeature.isInHistory() or aFeatureKind=="Export" or aFeatureKind=="RemoveResults":
  68                 aDumper = self.myFeatures[aFeatureKind](theFeature)
  69                 # Dump comment for the operation before the dumping of the feature to improve the readability of a script.
  70                 self.__print__("\n### Create " + theFeature.getKind())
  71                 self.newline()
  72             else:
  73                 self.name(theFeature)
  74                 self.clearNotDumped()
  75         else:
  76             # Probably the feature is a constraint, try to dump it with SketchAPI_Constraint.
  77             # In case of theFeature is not a constraint, it will not be dumped.
  78             self.name(theFeature, False, True, True)
  79             aDumper = self.myFeatures[SketchAPI.SketchAPI_Constraint.ID()](theFeature)
  80         if aDumper is not None:
  81             aDumper.dump(self)
  82
  83     ## Create wrapper for a folder and dump it
  84     def dumpFolder(self, theFolder):
  85         if theFolder.ID() in self.myFeatures:
  86             self.myFeatures[theFolder.ID()](theFolder).dump(self)
  87
  88     ## Dump all parameters
  89     def dumpParameter(self, theFeature):
  90         aFeatureKind = theFeature.getKind()
  91         if aFeatureKind == "Parameter" and aFeatureKind in self.myFeatures:
  92             self.myFeatures[aFeatureKind](theFeature).dump(self)
  93
  94     ## Return getter for specified attribute
  95     def attributeGetter(self, theFeature, theAttrName):
  96         aFeatureKind = theFeature.getKind()
  97         if aFeatureKind in self.myFeatures:
  98             return self.myFeatures[aFeatureKind](theFeature).attributeGetter(theAttrName)
  99         return std_string()
 100
 101     ## Return name of wrapper feature
 102     def featureWrapper(self, theFeature):
 103         aFeatureKind = theFeature.getKind()
 104         if aFeatureKind in self.myWrapperNames:
 105             return self.myWrapperNames[aFeatureKind]
 106         return std_string()
 107
 108     ## Exports the dumped variables names to the SHAPERSTUDY
 109     def exportVariable(self, theEntry, theVarName):
 110         try:
 111           if self.myEngine == None:
 112             import SHAPERSTUDY_utils
 113             aComponent = SHAPERSTUDY_utils.findOrCreateComponent()
 114             self.myEngine = SHAPERSTUDY_utils.getEngine()
 115           if self.myEngine != 1:
 116             self.myEngine.StoreVariableName(theEntry, theVarName)
 117         except:
 118           self.myEngine = 1 # invalid access
 119         pass
 120
 121 # Instance of dumper
 122 dumper = DumpAssistant