# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
#
+import SHAPERSTUDY_ORB
import SHAPERSTUDY_ORB__POA
import GEOM
-from SHAPERSTUDY_utils import getEngine
+from SHAPERSTUDY_utils import getEngine, getStudy
+import salome
import StudyData_Swig
-class SHAPERSTUDY_Object(SHAPERSTUDY_ORB__POA.SHAPER_Object):
+# converter from the integer values to idl shape_type enumerations
+__shape_types__ = {
+ 0:GEOM.COMPOUND, 1:GEOM.COMPSOLID, 2:GEOM.SOLID,
+ 3:GEOM.SHELL, 4:GEOM.FACE, 5:GEOM.WIRE,
+ 6:GEOM.EDGE, 7:GEOM.VERTEX, 8:GEOM.SHAPE, 9:GEOM.FLAT}
+
+class SHAPERSTUDY_GenericObject:
+ """
+ Implement methods of SALOME::GenericObj
+ """
+
+ def Register(self):
+ """
+ Increase the reference count (mark as used by another object).
+ """
+ return
+
+ def UnRegister(self):
+ """
+ Decrease the reference count (release by another object)
+ """
+ return
+
+ def Destroy(self):
+ """
+ Obsolete, left for compatibility reasons only. Use UnRegister() instead
+ """
+ return
+
+ pass
+
+
+class SHAPERSTUDY_Object(SHAPERSTUDY_ORB__POA.SHAPER_Object,
+ SHAPERSTUDY_GenericObject):
"""
- Construct an instance of SHAPERSTUDY Object.
+ Constructs an instance of SHAPERSTUDY Object.
"""
def __init__ ( self, *args):
self.SO = None
self.data = None
+ self.entry = None
+ self.type = 1 # by default it is a shape (Import feature in GEOMImpl_Types.hxx)
pass
def GetShapeType( self ):
"""
if self.data is None:
return GEOM.SHAPE
- return self.data.type();
+ global __shape_types__
+ return __shape_types__[self.data.type()];
def IsMainShape( self ):
"""
def getShape( self ):
"""
- Get the TopoDS_Shape, for colocated case only.
+ Get the TopoDS_Shape, for collocated case only.
Called by GEOM_Client to get TopoDS_Shape pointer
"""
if self.data is None:
Get geometric shape of the object as a byte stream in BRep format
"""
if self.data is None:
- return
- return self.data.shapeStream()
+ return b''
+ return self.data.shapeStream().encode()
+
+ def GetOldShapeStream( self ):
+ """
+ Get geometric shape of the object as a byte stream in BRep format
+ """
+ if self.data is None:
+ return b''
+ return self.data.oldShapeStream().encode()
def SetShapeByStream(self, theStream):
"""
Get internal type of operation created this object.
In SMESH is used to find out if an object is GROUP (type == 37)
"""
- # 1 corresponds to the Import feature (GEOMImpl_Types.hxx )
- return 1
+ return self.type
+
+ def SetType( self, theType ):
+ """
+ Sets internal type of operation created this object.
+ In SMESH is used to find out if an object is GROUP (type == 37, for shape it is default=1)
+ """
+ self.type = theType
def GetTick( self ):
"""
Returns true if the current object has connection to a parametrical model
which can be modified by parameters change.
"""
- return True
+ return not self.IsDead() and self.type == 1 # only break link for shapes are accessible now
+
+ def IsDead(self):
+ """
+ Returns true if the shape is dead - no parametrical link to the SHAPER exists
+ """
+ return self.GetEntry().startswith("dead")
+
+ def MakeDead(self):
+ """
+ Makes the dead-copy of the shape and returns it.
+ """
+ aStudy = getStudy()
+ aBuilder = aStudy.NewBuilder()
+ aRes, aHistSO = self.SO.FindSubObject(2)
+ if not aRes: # create a "history" folder if it does not exist
+ aHistSO = aBuilder.NewObjectToTag(self.SO, 2)
+ aHistSO.SetAttrString("AttributeName", "History")
- def BreakLinks(self):
+ aDeadSO = aBuilder.NewObject(aHistSO)
+ anIndex = aDeadSO.Tag()
+ aDeadSO.SetAttrString("AttributeName", self.SO.GetName() + " (" + str(anIndex) + ")")
+ aDead = SHAPERSTUDY_Object()
+ aDeadEntry = "dead" + str(anIndex) + "_" + self.GetEntry()
+ aDead.SetEntry(aDeadEntry)
+ aDead.SetShapeByStream(self.data.oldShapeStream())
+ aDeadObj = aDead._this()
+ anIOR = salome.orb.object_to_string(aDeadObj)
+ aDeadSO.SetAttrString("AttributeIOR", anIOR)
+ aDead.SetSO(aDeadSO)
+ if self.GetTick() > 2:
+ aDead.data.setTick(self.GetTick() - 1) # set the tick of an old shape
+ # make dead-copy also sub-groups
+ aSOIter = aStudy.NewChildIterator(self.SO)
+ while aSOIter.More():
+ aGroupSO = aSOIter.Value()
+ anIOR = aGroupSO.GetIOR()
+ if len(anIOR):
+ aGroup = salome.orb.string_to_object(anIOR)
+ if isinstance(aGroup, SHAPERSTUDY_ORB._objref_SHAPER_Group):
+ aDeadGroup = SHAPERSTUDY_Group()
+ aDeadGroupEntry = "dead" + str(anIndex) + "_" + aGroup.GetEntry()
+ aDeadGroup.SetEntry(aDeadGroupEntry)
+ aDeadGroup.SetShapeByPointer(aGroup.getShape())
+ aDeadGroup.SetSelectionType(aGroup.GetSelectionType())
+ aDeadGroup.SetSelection(aGroup.GetSelection())
+ aDeadGroupSO = aBuilder.NewObject(aDeadSO)
+ aDeadGroup.SetSO(aDeadGroupSO)
+ # 15.01.20 groups and fields names stays the same
+ #aDeadGroupSO.SetAttrString("AttributeName", aGroupSO.GetName() + " (" + str(anIndex) + ")")
+ aDeadGroupObj = aDeadGroup._this()
+ anIOR = salome.orb.object_to_string(aDeadGroupObj)
+ aDeadGroupSO.SetAttrString("AttributeIOR", anIOR)
+ aSOIter.Next()
+
+ return aDeadObj
+
+ def SetShapeByPointer(self, theShape):
+ """
+ Sets the shape by the pointer to the TopoDS_Shape
+ """
+ if not self.data:
+ self.data = StudyData_Swig.StudyData_Object()
+ self.data.SetShapeByPointer(theShape)
+
+ pass
+
+class SHAPERSTUDY_Group(SHAPERSTUDY_ORB__POA.SHAPER_Group, SHAPERSTUDY_Object):
+ """
+ Constructs an instance of SHAPERSTUDY Group
+ """
+ def __init__ ( self, *args):
+ self.seltype = None
+ self.selection = []
+ self.SO = None
+ self.data = None
+ self.entry = None
+ self.type = 37 # a group type
+ pass
+
+ def SetSelectionType(self, theType):
+ """
+ Sets what is returned in the GEOM_IGroupOperations::GetType
+ """
+ self.seltype = theType
+
+ def GetSelectionType(self):
"""
- Breaks links to parametrical mode for parametrical shape
+ Returns the type of the selected sub-shapes
"""
- print("##### Break parametrical links")
+ return self.seltype
+
+ def SetSelection(self, theSelection):
+ """
+ Sets what is returned in the GEOM_IGroupOperations::GetObjects
+ """
+ self.data = None # nullify the cashed shape when selection is changed
+ self.selection = theSelection
+
+ def GetSelection(self):
+ """
+ Returns the selected sub-shapes indices
+ """
+ return self.selection
+
+ def GetMainShape( self ):
+ """
+ Main shape is groups owner
+ """
+ return self.SO.GetFather().GetObject()
+
+ def getShape( self ):
+ """
+ Redefinition of the getShape method: here it creates a shape by the
+ main shape and the group index.
+ """
+ if not self.data:
+ self.data = StudyData_Swig.StudyData_Object()
+ # convert selection to long list
+ anArg = StudyData_Swig.LongList()
+ for l in self.selection:
+ anArg.append(l)
+ return self.data.groupShape(self.GetMainShape().getShape(), anArg)
+
+ pass
+
+class SHAPERSTUDY_Field(SHAPERSTUDY_ORB__POA.SHAPER_Field, SHAPERSTUDY_Group):
+ """
+ Constructs an instance of SHAPERSTUDY Field (inherits selection from a Group object)
+ """
+ def __init__ ( self, *args):
+ self.seltype = None
+ self.selection = []
+ self.SO = None
+ self.data = None
+ self.entry = None
+ self.type = 52 # a field type
+ self.valtype = None # type of the values
+ self.steps = [] # list of long
+ self.components = [] # string array, names of the components
+ self.name = None # name, string
+ self.fieldsteps = {} # FieldSteps objects identified by step ID
+ pass
+
+ def SetValuesType( self, theType ):
+ """
+ Sets the type of values in the field
+ """
+ self.valtype = theType
+
+ def GetDataType( self ):
+ """
+ Returns the type of values in the field in terms of GEOM enumeration
+ """
+ if self.valtype == 0:
+ return GEOM.FDT_Bool
+ elif self.valtype == 1:
+ return GEOM.FDT_Int
+ elif self.valtype == 2:
+ return GEOM.FDT_Double
+ elif self.valtype == 3:
+ return GEOM.FDT_String
+ return None # unknown case
+
+ def GetShape ( self ):
+ """
+ Returns the shape the field lies on
+ """
+ return super().GetMainShape()
+
+ def SetSteps( self, theSteps ):
+ self.steps = theSteps
+
+ def GetSteps( self ):
+ return self.steps
+
+ def SetComponents( self, theComponents ):
+ self.components = theComponents
+
+ def GetComponents( self ):
+ return self.components
+
+ def GetDimension( self ):
+ aShapeType = super().GetSelectionType()
+ if aShapeType == 8:
+ return -1 # whole part
+ elif aShapeType == 7:
+ return 0 # vertex
+ elif aShapeType == 6:
+ return 1 # edge
+ elif aShapeType == 4:
+ return 2 # face
+ elif aShapeType == 2:
+ return 3 # solid
+ return None # unknown case
+
+ def ClearFieldSteps( self ):
+ self.fieldsteps = {}
+
+ def AddFieldStep( self, theStampID, theStepID, theValues):
+ aFieldStep = None
+ if self.valtype == 0:
+ aFieldStep = SHAPER_BoolFieldStep()
+ elif self.valtype == 1:
+ aFieldStep = SHAPER_IntFieldStep()
+ elif self.valtype == 2:
+ aFieldStep = SHAPER_DoubleFieldStep()
+
+ aFieldStep.SetStep(theStampID, theStepID, theValues)
+ self.fieldsteps[theStepID] = aFieldStep._this()
+
+ def GetStep( self, theStepID ):
+ return self.fieldsteps[theStepID]
+
+ pass
+
+class SHAPER_FieldStep:
+ """
+ Base class for all step-classes
+ """
+ def __init__ ( self, *args):
+ self.stamp = None # long, ID of stamp
+ self.step = None # long, ID of step
+ self.values = None # array of values of the needed type
+
+ """
+ Defines all parameters of the step
+ """
+ def SetStep( self, theStamp, theStep, theValues ):
+ self.stamp = theStamp
+ self.step = theStep
+ self.values = theValues
+
+ """
+ Returns stamp ID
+ """
+ def GetStamp( self ):
+ return self.stamp
+ """
+ Returns step ID
+ """
+ def GetID( self ):
+ return self.step
+ """
+ Returns a name of a sub-shape if the sub-shape is published in the study
+ """
+ def GetSubShape(self, theSubID):
+ # the SHAPER study does not support sub-shapes for now
+ return ""
+
+
+class SHAPER_DoubleFieldStep(SHAPERSTUDY_ORB__POA.SHAPER_DoubleFieldStep, SHAPER_FieldStep):
+ """
+ Constructs an instance of SHAPERSTUDY Field step of type Double
+ """
+ def __init__ ( self, *args):
+ pass
+
+ """
+ Returns values as an array of the needed type
+ """
+ def GetValues( self ):
+ aResult = [] # to make any type of result, create a corba-type
+ for i in self.values:
+ aResult.append(float(i))
+ return aResult
+
+ pass
+
+class SHAPER_IntFieldStep(SHAPERSTUDY_ORB__POA.SHAPER_IntFieldStep, SHAPER_FieldStep):
+ """
+ Constructs an instance of SHAPERSTUDY Field step of type Double
+ """
+ def __init__ ( self, *args):
+ pass
+
+ """
+ Returns values as an array of the needed type
+ """
+ def GetValues( self ):
+ aResult = [] # to make any type of result, create a corba-type
+ for i in self.values:
+ aResult.append(int(i))
+ return aResult
+
+ pass
+
+class SHAPER_BoolFieldStep(SHAPERSTUDY_ORB__POA.SHAPER_BoolFieldStep, SHAPER_FieldStep):
+ """
+ Constructs an instance of SHAPERSTUDY Field step of type Double
+ """
+ def __init__ ( self, *args):
+ pass
+
+ """
+ Returns values as an array of the needed type
+ """
+ def GetValues( self ):
+ aResult = [] # to make any type of result, create a corba-type
+ for i in self.values:
+ aResult.append(int(i))
+ return aResult
pass
