# 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
+# 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_Object(SHAPERSTUDY_ORB__POA.SHAPER_Object):
"""
Construct 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
- 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}
- return types[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")
+
+ 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)
+ 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):
+ """
+ Construct 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):
+ """
+ Returns the type of the selected sub-shapes
+ """
+ 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):
+ """
+ Construct 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
+ 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 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 = SHAPERSTUDY_Group.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 GetShape( self ):
+ return SHAPERSTUDY_Group.getShape()
pass
