]> SALOME platform Git repositories - modules/shaper.git/blob - src/ConnectorPlugin/ConnectorPlugin_ExportFeature.py
Salome HOME
Merge remote-tracking branch 'remotes/origin/HigherLevelObjectsHistory'
[modules/shaper.git] / src / ConnectorPlugin / ConnectorPlugin_ExportFeature.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 Plugins
21 #  ExportFeature class definition
22
23 import ModelAPI
24 import ExchangeAPI
25
26 import salome
27 from salome.geom import geomBuilder
28
29 from salome.shaper import model
30
31 import os
32
33 def getTmpFileName(ext):
34     import tempfile
35     tempdir = tempfile.gettempdir()
36     tmp_file = tempfile.NamedTemporaryFile(suffix=".%s"%ext, prefix='shaper_', dir=tempdir, delete=False)
37     tmp_filename = tmp_file.name
38     if os.name == "nt":
39         tmp_filename.replace("\\", "/")
40     return tmp_filename
41
42 ## @ingroup Plugins
43 #  Feature to export all shapes and groups into the GEOM module
44 class ExportFeature(ModelAPI.ModelAPI_Feature):
45
46     ## The constructor.
47     def __init__(self):
48         ModelAPI.ModelAPI_Feature.__init__(self)
49         pass
50
51     @staticmethod
52     ## Export kind. Static.
53     def ID():
54         return "ExportToGEOM"
55
56     ## Returns the kind of a feature.
57     def getKind(self):
58         return ExportFeature.ID()
59
60     ## This feature is action: has no property pannel and executes immediately.
61     def isAction(self):
62         return True
63
64     ## This feature has no attributes, as it is action.
65     def initAttributes(self):
66       pass
67
68     ## Export the results, groups and fields via XAO
69     def exportViaXAO(self):
70         tmpXAOFile = getTmpFileName("xao")
71         self.tmpXAOFile = tmpXAOFile
72         #print "Export to %s"%tmpXAOFile
73         exportXAO = ExchangeAPI.exportToXAO(self.Part, tmpXAOFile, "automatic_shaper_export_to_XAO")
74         if not os.path.exists(tmpXAOFile) or os.stat(tmpXAOFile).st_size == 0:
75             exportXAO.feature().setError("Error in exportToXAO. No XAO file has been created.")
76             return
77         imported, shape, subShapes, groups, fields = self.geompy.ImportXAO(tmpXAOFile)
78         self.geompy.addToStudy( shape, shape.GetName() )
79         # add sub-shapes and groups to the object browser
80         for obj in subShapes + groups:
81             name = obj.GetName()
82             self.geompy.addToStudyInFather(shape, obj, name)
83         # add fields to the object browser
84         for field in fields:
85             name = field.GetName()
86             self.geompy.addToStudyInFather(shape, field, name)
87             # add steps to the object browser
88             steps = field.getSteps()
89             for i_step in steps:
90                 step = field.getStep(i_step)
91                 i_stamp = step.GetStamp()
92                 step_name = "Step %i %i"%(i_step, i_stamp)
93                 self.geompy.addToStudyInFather( field, step, step_name )
94         # Remove the temporary file
95         os.remove(tmpXAOFile)
96         pass
97
98     ## Exports all shapes and groups into the GEOM module.
99     def execute(self):
100         aSession = ModelAPI.ModelAPI_Session.get()
101         ## Get active document
102         self.Part = aSession.activeDocument()
103         ## List of objects created in the old geom for later use
104         self.geomObjects = []
105         ## geomBuilder tool
106         salome.salome_init(1)
107         self.geompy = geomBuilder.New()
108
109         self.exportViaXAO()
110
111         pass