-"""
-Copyright (C) 2014-20xx CEA/DEN, EDF R&D
-"""
+# Copyright (C) 2014-2022 CEA/DEN, EDF R&D
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+#
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+#
+
+## @package Plugins
+# ExportFeature class definition
-import EventsAPI
import ModelAPI
+import ExchangeAPI
+import EventsAPI
import salome
from salome.geom import geomBuilder
+from salome.shaper import model
-class ExportFeature(ModelAPI.ModelAPI_Feature):
+import os
- "Feature to create a box by drawing a sketch and extruding it"
+## @ingroup Plugins
+# Feature to export all shapes and groups into the GEOM module
+class ExportFeature(ModelAPI.ModelAPI_Feature):
+ ## The constructor.
def __init__(self):
ModelAPI.ModelAPI_Feature.__init__(self)
+ pass
@staticmethod
+ ## Export kind. Static.
def ID():
return "ExportToGEOM"
+ ## Returns the kind of a feature.
def getKind(self):
return ExportFeature.ID()
- # This feature is action: has no property pannel and executes immideately
- # def isAction(self):
- # return True
-
- def isInHistory(self):
- return False
+ ## This feature is action: has no property panel and executes immediately.
+ def isAction(self):
+ return True
+ ## This feature has no attributes, as it is action.
def initAttributes(self):
- # This feature has no attributes, but should perform some actions on initialization
+ pass
+
+ ## Export the results, groups and fields via XAO
+ def exportViaXAO(self):
+ # if part-set is active, iterate also parts
+ for isPart in (True, False):
+ aResultType = model.ModelAPI_ResultBody_group()
+ if isPart:
+ aResultType = model.ModelAPI_ResultPart_group()
+ # iterate all results of Part, export one by one due to issue 2882
+ for aResIndex in range(self.Part.size(aResultType)):
+ anObject = self.Part.object(aResultType, aResIndex)
+ aResult = model.objectToResult(anObject)
+ # do not export picture
+ if aResult.hasTexture() is True:
+ continue
+ if not aResult is None:
+ if (not aResult.shape() or aResult.shape().isNull()) and isPart:
+ aPart = model.modelAPI_ResultPart(aResult)
+ aPartDoc = aPart.partDoc()
+ if not aPartDoc or not aPartDoc.isOpened():
+ EventsAPI.Events_InfoMessage("ExportToGEOM", "For export to GEOM some Part is not activated", self).send()
+ return
+
+ if not aResult.shape() or aResult.shape().isNull():
+ continue
+ tmpXAOFile = model.getTmpFileName("shaper_", ".xao")
+ self.tmpXAOFile = tmpXAOFile
+ #print "Export to %s"%tmpXAOFile
+ exportXAO = ExchangeAPI.exportToXAO(self.Part, tmpXAOFile, model.selection(aResult), "automatic_shaper_export_to_XAO")
+ if not os.path.exists(tmpXAOFile) or os.stat(tmpXAOFile).st_size == 0:
+ exportXAO.feature().setError("Error in exportToXAO. No XAO file has been created.")
+ return
+ imported, shape, subShapes, groups, fields = self.geompy.ImportXAO(tmpXAOFile)
+ self.geompy.addToStudy( shape, shape.GetName() )
+ # add sub-shapes and groups to the object browser
+ for obj in subShapes + groups:
+ name = obj.GetName()
+ self.geompy.addToStudyInFather(shape, obj, name)
+ # add fields to the object browser
+ for field in fields:
+ name = field.GetName()
+ self.geompy.addToStudyInFather(shape, field, name)
+ # add steps to the object browser
+ steps = field.getSteps()
+ for i_step in steps:
+ step = field.getStep(i_step)
+ i_stamp = step.GetStamp()
+ step_name = "Step %i %i"%(i_step, i_stamp)
+ self.geompy.addToStudyInFather( field, step, step_name )
+ # Remove the temporary file
+ os.remove(tmpXAOFile)
+ pass
+ pass
+ pass
+
+ ## Exports all shapes and groups into the GEOM module.
+ def execute(self):
aSession = ModelAPI.ModelAPI_Session.get()
+ ## Get active document
self.Part = aSession.activeDocument()
+ ## List of objects created in the old geom for later use
self.geomObjects = []
- self.geompy = geomBuilder.New(salome.myStudy)
-
- # Export bodies and groups
- self.exportBodies()
- self.exportGroups()
-
- def exportBodies(self):
- # Get all bodies
- kResultBodyType = "Bodies"
- aPartSize = self.Part.size(kResultBodyType)
- if aPartSize == 0:
- EventsAPI.Events_Error_send("No results in the active document")
- return
-
- anObjList = [self.Part.object(kResultBodyType, idx) for idx in xrange(aPartSize)]
- for idx, anObject in enumerate(anObjList):
- aResult = ModelAPI.modelAPI_Result(anObject)
- aBodyResult = ModelAPI.modelAPI_ResultBody(aResult)
- if not aBodyResult:
- continue
- aShape = aBodyResult.shape()
- aDump = aShape.getShapeStream()
- # Load shape to SALOME Geom
- aBrep = self.geompy.RestoreShape(aDump)
- self.geompy.addToStudy(aBrep, "NewGeomShape_{0}".format(idx + 1))
- self.geomObjects.append([aShape, aBrep])
-
- def exportGroups(self):
- # iterate all features to find groups
- aFeaturesNum = self.Part.size("Features")
- groupIndex = 0
- for anIndex in range(0, aFeaturesNum):
- aFeature = self.Part.object("Features", anIndex)
- aSelectionList = aFeature.data().selectionList("group_list")
- # if a group has been found
- if aSelectionList:
- groupIndex = groupIndex + 1
- self.createGroupFromList(aSelectionList, "NewGeomGroup_{0}".format(groupIndex))
-
- def createGroupFromList(self, theSelectionList, theGroupName):
- # iterate on all selected entities of the group
- # and get the corresponding ID
- aSelectionNum = theSelectionList.size()
- Ids = []
- for aSelIndex in range(0, aSelectionNum):
- aSelection = theSelectionList.value(aSelIndex)
- aSelectionContext = aSelection.context()
- anID = aSelection.Id()
- Ids.append(anID)
- if aSelection.value().isVertex():
- groupType = "VERTEX"
- elif aSelection.value().isEdge():
- groupType = "EDGE"
- elif aSelection.value().isFace():
- groupType = "FACE"
- else:
- groupType = "SOLID"
-
- # iterate on exported objects and check if the current
- # group refers to this object
- for obj in self.geomObjects:
- if aSelectionContext.shape().isEqual(obj[0]):
- aGroup = self.geompy.CreateGroup(obj[1], self.geompy.ShapeType[groupType])
- self.geompy.UnionIDs(aGroup,Ids)
- self.geompy.addToStudyInFather(obj[1], aGroup, theGroupName)
-
- def execute(self):
- # Nothing to execute: all logic would be in the initAttributes
+ ## geomBuilder tool
+ salome.salome_init(1)
+ self.geompy = geomBuilder.New()
+
+ self.exportViaXAO()
+
pass
