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
  48     ## Collect feature wrappers, which allow dumping (have method dump)
  49     def collectFeatures(self):
  50         self.myFeatures = {}
  51         self.myWrapperNames = {}
  52         for aModule in sys.modules:
  53             for aName, anObj in inspect.getmembers(sys.modules[aModule], inspect.isclass):
  54                 if issubclass(anObj, ModelHighAPI.ModelHighAPI_Interface) and hasattr(anObj, "ID") and anObj.dump != ModelHighAPI.ModelHighAPI_Interface.dump:
  55                     self.myFeatures[anObj.ID()] = anObj
  56                     self.myWrapperNames[anObj.ID()] = aName
  57
  58     ## Create wrapper for a given feature and dump it
  59     def dumpFeature(self, theFeature, theForce):
  60         aFeatureKind = theFeature.getKind()
  61         if aFeatureKind in self.myFeatures:
  62             # Dump only feature created by user (in history).
  63             # Also dump Export and RemoveResults features (hard-coded here in order not to change the data model).
  64             # For all other features, just keep their name.
  65             if theForce or theFeature.isInHistory() or aFeatureKind=="Export" or aFeatureKind=="RemoveResults":
  66                 self.myFeatures[aFeatureKind](theFeature).dump(self)
  67             else:
  68                 self.name(theFeature)
  69                 self.clearNotDumped()
  70         else:
  71             # Probably the feature is a constraint, try to dump it with SketchAPI_Constraint.
  72             # In case of theFeature is not a constraint, it will not be dumped.
  73             self.myFeatures[SketchAPI.SketchAPI_Constraint.ID()](theFeature).dump(self)
  74
  75     ## Create wrapper for a folder and dump it
  76     def dumpFolder(self, theFolder):
  77         if theFolder.ID() in self.myFeatures:
  78             self.myFeatures[theFolder.ID()](theFolder).dump(self)
  79
  80     ## Dump all parameters
  81     def dumpParameter(self, theFeature):
  82         aFeatureKind = theFeature.getKind()
  83         if aFeatureKind == "Parameter" and aFeatureKind in self.myFeatures:
  84             self.myFeatures[aFeatureKind](theFeature).dump(self)
  85
  86     ## Return getter for specified attribute
  87     def attributeGetter(self, theFeature, theAttrName):
  88         aFeatureKind = theFeature.getKind()
  89         if aFeatureKind in self.myFeatures:
  90             return self.myFeatures[aFeatureKind](theFeature).attributeGetter(theAttrName)
  91         return std_string()
  92
  93     ## Return name of wrapper feature
  94     def featureWrapper(self, theFeature):
  95         aFeatureKind = theFeature.getKind()
  96         if aFeatureKind in self.myWrapperNames:
  97             return self.myWrapperNames[aFeatureKind]
  98         return std_string()
  99
 100 # Instance of dumper
 101 dumper = DumpAssistant