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