]> SALOME platform Git repositories - modules/shaper.git/blob - src/ExchangePlugin/Test/TestExport.py
Salome HOME
Merge branch 'Pre_2.8.0_development'
[modules/shaper.git] / src / ExchangePlugin / Test / TestExport.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 """
22       TestExport.py
23       Unit test of ExchangePlugin_TestExport class
24 """
25 #=========================================================================
26 # Initialization of the test
27 #=========================================================================
28 import os
29 import math
30
31 from ModelAPI import *
32
33 from TestImport import testImport
34
35 from salome.shaper import model
36
37 __updated__ = "2015-05-22"
38
39 aSession = ModelAPI_Session.get()
40 #=========================================================================
41 # Help functions
42 #=========================================================================
43 def removeFile(theFileName):
44     try: os.remove(theFileName)
45     except OSError: pass
46     assert not os.path.exists(theFileName), \
47             "Can not remove file {0}".format(theFileName)
48
49 #=========================================================================
50 # Common test function
51 #=========================================================================
52 def testExport(theType, theFormat, theFile, theVolume, theDelta):
53     # Import a reference part
54     aSession.startOperation("Add part")
55     aPartFeature = aSession.moduleDocument().addFeature("Part")
56     aSession.finishOperation()
57     aPart = aSession.activeDocument()
58
59     aSession.startOperation("Import screw")
60     anImportFeature = aPart.addFeature("Import")
61     anImportFeature.string("file_path").setValue("Data/screw.step")
62     aSession.finishOperation()
63
64     removeFile(theFile)
65     # Export a part
66     aSession.startOperation("Export part")
67     anExportFeature = aPart.addFeature("Export")
68     anExportFeature.string("file_format").setValue(theFormat)
69     print "theFile=",theFile
70     anExportFeature.string("file_path").setValue(theFile)
71     anExportFeature.string("ExportType").setValue("Regular")
72     aSelectionListAttr = anExportFeature.selectionList("selection_list")
73     aSelectionListAttr.setSelectionType("solids")
74     aSelectionListAttr.append(anImportFeature.firstResult(), anImportFeature.firstResult().shape())
75     aSession.finishOperation()
76
77     assert os.path.exists(theFile)
78
79     # Test exported file by importing
80     testImport(theType, theFile, theVolume, theDelta)
81
82 def testExportXAO():
83     # Import a reference part
84     aSession.startOperation("Add part")
85     aPartFeature = aSession.moduleDocument().addFeature("Part")
86     aSession.finishOperation()
87     aPart = aSession.activeDocument()
88
89     aSession.startOperation("Import Box_1")
90     anImportFeature = aPart.addFeature("Import")
91     anImportFeature.string("file_path").setValue("Data/Box_1.brep")
92     aSession.finishOperation()
93
94     # Create groups
95     aSession.startOperation("First group")
96     aGroupFeature = aSession.activeDocument().addFeature("Group")
97     aGroupFeature.data().setName("boite_1")
98     aSelectionListAttr = aGroupFeature.selectionList("group_list")
99     aSelectionListAttr.setSelectionType("solid")
100     aSelectionListAttr.append(anImportFeature.lastResult(), None)
101     aSession.finishOperation()
102
103     aSession.startOperation("Second Group")
104     aGroupFeature = aSession.activeDocument().addFeature("Group")
105     aGroupFeature.data().setName("")
106     aSelectionListAttr = aGroupFeature.selectionList("group_list")
107     aSelectionListAttr.setSelectionType("face")
108     aSelectionListAttr.append("Box_1_1/Shape1")
109     aSelectionListAttr.append("Box_1_1/Shape2")
110     aSession.finishOperation()
111
112     aSession.startOperation("Create a field")
113     aField = aSession.activeDocument().addFeature("Field")
114     aSelectionListAttr = aField.selectionList("selected")
115     aSelectionListAttr.setSelectionType("face")
116     aSelectionListAttr.append("Box_1_1/Shape1")
117     aSelectionListAttr.append("Box_1_1/Shape2")
118     aComponentNames = aField.stringArray("components_names")
119     aComponentNames.setSize(2) # two components
120     aComponentNames.setValue(0, "temperatue")
121     aComponentNames.setValue(1, "porosity")
122     aStamps = aField.intArray("stamps")
123     aStamps.setSize(1) # one step
124     aStamps.setValue(0, 10)
125     aTables = aField.tables("values")
126     aTables.setType(2) # double
127     aTables.setSize(1 + 2, 2, 1) # default row + number of selected, number of compoents, number of steps (tables)
128     aTables.setValue(1., 0, 0, 0) # value, index of selection, index of component, index of step
129     aTables.setValue(2., 1, 0, 0)
130     aTables.setValue(3., 2, 0, 0)
131     aTables.setValue(4., 0, 1, 0)
132     aTables.setValue(5., 1, 1, 0)
133     aTables.setValue(6., 2, 1, 0)
134     aSession.finishOperation()
135
136     # Export
137     aSession.startOperation("Export to XAO")
138     anExportFeature = aPart.addFeature("Export")
139     anExportFeature.string("xao_file_path").setValue("Data/export.xao")
140     anExportFeature.string("file_format").setValue("XAO")
141     anExportFeature.string("ExportType").setValue("XAO")
142     anExportFeature.string("xao_author").setValue("me")
143     anExportFeature.string("xao_geometry_name").setValue("mygeom")
144     aSession.finishOperation()
145
146     # Check exported file
147 #    import filecmp
148 #    assert filecmp.cmp("Data/export.xao", "Data/export_ref.xao")
149
150 if __name__ == '__main__':
151 #=========================================================================
152 # Export a shape into BREP
153 #=========================================================================
154     aRealVolume = 3.78827401738e-06
155     testExport("BREP", "BREP", os.path.join(os.getcwd(), "Data", "screw_export.brep"), aRealVolume, 10 ** -17)
156     testExport("BRP", "BREP", os.path.join(os.getcwd(), "Data", "screw_export.brp"), aRealVolume, 10 ** -17)
157 #=========================================================================
158 # Export a shape into STEP
159 #=========================================================================
160     testExport("STEP", "STEP", os.path.join(os.getcwd(), "Data", "screw_export.step"), 3.78825807533e-06, 10 ** -17)
161     testExport("STP", "STEP", os.path.join(os.getcwd(), "Data", "screw_export.stp"), 3.78825807533e-06, 10 ** -17)
162 #=========================================================================
163 # Export a shape into IGES
164 #=========================================================================
165     testExport("IGES-5.1", "IGES-5.1", os.path.join(os.getcwd(), "Data", "screw_export-5.1.iges"), 3.78829613776e-06, 10 ** -17)
166     testExport("IGS-5.1", "IGES-5.1", os.path.join(os.getcwd(), "Data", "screw_export-5.1.igs"), 3.78829613776e-06, 10 ** -17)
167     testExport("IGES-5.3", "IGES-5.3", os.path.join(os.getcwd(), "Data", "screw_export-5.3.iges"), 3.78827401651e-06, 10 ** -17)
168     testExport("IGS-5.3", "IGES-5.3", os.path.join(os.getcwd(), "Data", "screw_export-5.3.igs"), 3.78827401651e-06, 10 ** -17)
169 #=========================================================================
170 # Export a shape into XAO
171 #=========================================================================
172     testExportXAO()
173 #=========================================================================
174 # End of test
175 #=========================================================================
176
177 assert(model.checkPythonDump())