1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2015 CEA/DEN, EDF R&D
4 # This library is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU Lesser General Public
6 # License as published by the Free Software Foundation; either
7 # version 2.1 of the License, or (at your option) any later version.
9 # This library is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 # Lesser General Public License for more details.
14 # You should have received a copy of the GNU Lesser General Public
15 # License along with this library; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 # Author : Anthony Geay (CEA/DEN)
22 from MEDLoader import *
26 class MEDLoaderBasicsTest(unittest.TestCase):
27 def testExampleReadFieldOnAllEntity1(self):
28 from MEDLoaderDataForTest import MEDLoaderDataForTest
29 #! [PySnippetReadFieldOnAllEntity1_1]
30 fname="PyExamples1.med"
32 fieldName="FieldOnAll"
35 #! [PySnippetReadFieldOnAllEntity1_1]
36 #! [PySnippetWriteFieldOnAllEntity1_2]
37 m=MEDLoaderDataForTest.build2DMesh_3()
40 f=m.getMeasureField(ON_CELLS)
41 f=f.buildNewTimeReprFromThis(ONE_TIME,False)
42 f.setTime(5.5,iteration,order)
44 # MEDCoupling finished, MEDLoader advanced API specific part starting from here
46 mm.setMeshAtLevel(0,m)
47 ff=MEDFileField1TS.New()
48 ff.setFieldNoProfileSBT(f)
51 #! [PySnippetWriteFieldOnAllEntity1_2]
52 #! [PySnippetReadFieldOnAllEntity1_3]
53 medfileField1TS=MEDFileField1TS.New(fname,fieldName,iteration,order)
54 mm=MEDFileMesh.New(fname)
55 fread=medfileField1TS.getFieldOnMeshAtLevel(ON_CELLS,0,mm)
56 fread2=medfileField1TS.getFieldAtLevel(ON_CELLS,0)
57 self.assertTrue(fread.isEqual(f,1e-12,1e-12))
58 self.assertTrue(fread2.isEqual(f,1e-12,1e-12))
59 #! [PySnippetReadFieldOnAllEntity1_3]
60 #! [PySnippetReadFieldOnAllEntity1_4]
61 medfileFieldMTS=MEDFileFieldMultiTS.New(fname,fieldName)
62 mm=MEDFileMesh.New(fname)
63 fread=medfileFieldMTS.getFieldOnMeshAtLevel(ON_CELLS,iteration,order,0,mm)
64 fread2=medfileFieldMTS.getFieldAtLevel(ON_CELLS,iteration,order,0)
65 self.assertTrue(fread.isEqual(f,1e-12,1e-12))
66 self.assertTrue(fread2.isEqual(f,1e-12,1e-12))
67 #! [PySnippetReadFieldOnAllEntity1_4]
68 #! [PySnippetReadFieldOnAllEntity1_5]
69 medfileFieldMTS=MEDFileFieldMultiTS.New(fname,fieldName)
70 for medfileField1TS in medfileFieldMTS:
71 if medfileField1TS.getTime()[:2]==[iteration,order]:
72 fread=medfileField1TS.getFieldOnMeshAtLevel(ON_CELLS,0,mm)
73 fread2=medfileField1TS.getFieldAtLevel(ON_CELLS,0)
74 self.assertTrue(fread.isEqual(f,1e-12,1e-12))
75 self.assertTrue(fread2.isEqual(f,1e-12,1e-12))
78 #! [PySnippetReadFieldOnAllEntity1_5]
81 def testExampleReadFieldPartial1(self):
82 from MEDLoaderDataForTest import MEDLoaderDataForTest
83 #! [PySnippetReadFieldPartial1_1]
84 fname="PyExamples2.med"
86 fieldName="FieldPartial"
89 #! [PySnippetReadFieldPartial1_1]
90 #! [PySnippetWriteFieldPartial1_2]
91 m=MEDLoaderDataForTest.build2DMesh_1()
92 m.sortCellsInMEDFileFrmt()
94 # end of generation of a mesh -> let's create a field on that mesh
95 f=m.getMeasureField(ON_CELLS)
96 f=f.buildNewTimeReprFromThis(ONE_TIME,False)
97 f.setTime(5.5,iteration,order)
99 # The MEDCoupling part is finished -> let's perform advanced API
100 mm=MEDFileUMesh.New()
101 mm.setMeshAtLevel(0,m) # the MED file data structure is ready for writing. Of course mm could have been more complicated with groups, families, multilevel
102 # Let's building a sub field
103 pfl=DataArrayInt.New([1,3,4,5])
104 pfl.setName("myPfl") # here it is necessary to give a name to be compliant with MED file
105 f=f[pfl] ; f.getMesh().setName(m.getName()) # of course f should be in coherence with pfl -> f[pfl]
107 ff=MEDFileField1TS.New()
108 tmp=f.getMesh() # useless line, only to show that mesh into f is not considered by MEDFileField1TS.setFieldProfile
109 f.setMesh(None) # useless line, only to show that mesh into f is not considered by MEDFileField1TS.setFieldProfile
110 ff.setFieldProfile(f,mm,0,pfl)
111 f.setMesh(tmp) # useless line, only to show that mesh into f is not considered by MEDFileField1TS.setFieldProfile
114 #! [PySnippetWriteFieldPartial1_2]
115 #! [PySnippetReadFieldPartial1_3]
116 mm=MEDFileMesh.New(fname)
117 medfileField1TS=MEDFileField1TS.New(fname,fieldName,iteration,order)
118 fread=medfileField1TS.getFieldOnMeshAtLevel(ON_CELLS,0,mm)
119 fread2=medfileField1TS.getFieldAtLevel(ON_CELLS,0)
120 self.assertTrue(fread.isEqual(f,1e-12,1e-12))
121 self.assertTrue(fread2.isEqual(f,1e-12,1e-12))
122 #! [PySnippetReadFieldPartial1_3]
123 #! [PySnippetReadFieldPartial1_4]
124 medfileField1TS=MEDFileField1TS.New(fname,fieldName,iteration,order)
125 mm=MEDFileMesh.New(fname)
126 valsRead,pflRead=medfileField1TS.getFieldWithProfile(ON_CELLS,0,mm)
127 self.assertEqual(valsRead.getName(),f.getName())
129 self.assertTrue(valsRead.isEqual(f.getArray(),1e-12))
130 pflRead.setName(pfl.getName())
131 self.assertTrue(pflRead.isEqual(pfl))
132 #! [PySnippetReadFieldPartial1_4]
133 #! [PySnippetReadFieldPartial1_5]
134 firstApproachMesh=fread.getMesh()
135 mm=MEDFileMesh.New(fname)
136 wholeMesh=mm.getMeshAtLevel(0)
137 wholeMesh.tryToShareSameCoords(firstApproachMesh,1e-12)
138 isIncluded,pflComputed=wholeMesh.areCellsIncludedIn(firstApproachMesh,2)
139 self.assertTrue(isIncluded)
140 self.assertEqual(pflComputed.getName(),mm.getName())
141 pflComputed.setName(pflRead.getName())
142 self.assertTrue(pflComputed.isEqual(pflRead))
143 #! [PySnippetReadFieldPartial1_5]
144 #! [PySnippetReadFieldPartial1_6]
145 mm=MEDFileMesh.New(fname)
146 wholeMesh=mm.getMeshAtLevel(0)
147 computedMesh=wholeMesh[pflRead] ; computedMesh.setName(mm.getName())
148 self.assertTrue(computedMesh.isEqual(fread.getMesh(),1e-12))
149 fieldFromSecondApproach=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME)
150 fieldFromSecondApproach.setName(medfileField1TS.getName())
151 fieldFromSecondApproach.setMesh(computedMesh)
152 fieldFromSecondApproach.setArray(valsRead)
153 fieldFromSecondApproach.setTime(medfileField1TS.getTime()[2],medfileField1TS.getTime()[0],medfileField1TS.getTime()[1])
154 self.assertTrue(fieldFromSecondApproach.isEqual(fread,1e-12,1e-12))
155 #! [PySnippetReadFieldPartial1_6]
158 def testExampleMeshAdvAPI1(self):
159 da=DataArrayDouble.New([0.,1.1,2.3,3.6])
161 cmesh=MEDCouplingCMesh.New() ; cmesh.setCoords(da,da,da)
162 myMesh=cmesh.buildUnstructured()
163 #! [PySnippetMeshAdvAPI1_1]
164 self.assertTrue(isinstance(myMesh,MEDCouplingUMesh))
165 myMesh.setName(meshName)
166 WriteUMesh("wFile1.med",myMesh,True)
167 #! [PySnippetMeshAdvAPI1_1]
168 os.remove("wFile1.med")
169 #! [PySnippetMeshAdvAPI1_2]
170 self.assertTrue(isinstance(myMesh,MEDCouplingUMesh))
171 myMesh.setName(meshName)
172 WriteUMesh("wFile1.med",myMesh,False)
173 #! [PySnippetMeshAdvAPI1_2]
174 f=myMesh.getMeasureField(ON_CELLS)
175 f=f.buildNewTimeReprFromThis(ONE_TIME,False)
177 #! [PySnippetMeshAdvAPI1_3]
178 WriteUMesh("file3.med",f.getMesh(),True)
180 fileNameMultiTimeStep="file3.med"
181 WriteFieldUsingAlreadyWrittenMesh(fileNameMultiTimeStep,f)
183 f.applyFunc("sqrt(x)");
184 #Writing second time step with iteration==2 and order==0
185 WriteFieldUsingAlreadyWrittenMesh("file3.med",f);
186 #! [PySnippetMeshAdvAPI1_3]
187 #! [PySnippetMeshAdvAPI1_11]
188 timeStepsIds=GetCellFieldIterations("file3.med","Example2","myField")
189 self.assertEqual([(1, 0),(2, 0)],timeStepsIds)
190 fs=ReadFieldsOnSameMesh(ON_CELLS,"file3.med","Example2",0,"myField",timeStepsIds);
191 #! [PySnippetMeshAdvAPI1_11]
193 myMesh0=myMesh[:] ; myMesh0.setName("Example2")
194 myMesh1=myMesh0.buildDescendingConnectivity()[0] ; myMesh1.setName(myMesh0.getName())
195 myMesh2=myMesh1.buildDescendingConnectivity()[0] ; myMesh2.setName(myMesh0.getName())
196 myMesh3=myMesh2.buildDescendingConnectivity()[0] ; myMesh3.setName(myMesh0.getName())
197 mm=MEDFileUMesh.New()
198 mm.setMeshAtLevel(0,myMesh0)
199 mm.setMeshAtLevel(-1,myMesh1)
200 mm.setMeshAtLevel(-2,myMesh2)
201 mm.setMeshAtLevel(-3,myMesh3)
202 mm.write("file2.med",2)
203 F1Cell=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME)
204 F1Cell.setMesh(myMesh0)
205 F1Cell.setArray(myMesh0.getCoords()[:myMesh0.getNumberOfCells()])
206 F1Cell.setTime(1000.,2,3)
207 F1Cell.setName("F1Cell")
208 WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell)
209 F1Cell1=F1Cell.deepCopy()
210 F1Cell1.setMesh(myMesh1)
211 F1Cell1.setArray(myMesh1.computeCellCenterOfMass())
212 WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell1)
213 #! [PySnippetMeshAdvAPI1_12]
214 f1Cell_3D=ReadFieldCell("file2.med","Example2",0,"F1Cell",2,3)
215 #! [PySnippetMeshAdvAPI1_12]
216 #! [PySnippetMeshAdvAPI1_13]
217 f1Cell_2D=ReadFieldCell("file2.med","Example2",-1,"F1Cell",2,3)
218 #! [PySnippetMeshAdvAPI1_13]
219 self.assertTrue(F1Cell.isEqual(f1Cell_3D,1e-12,1e-12))
220 #! [PySnippetMeshAdvAPI1_8]
221 self.assertEqual(3,ReadUMeshDimFromFile("file2.med","Example2"))
222 #! [PySnippetMeshAdvAPI1_8]
223 #! [PySnippetMeshAdvAPI1_7]
224 m2D=ReadUMeshFromFile("file2.med","Example2",0)
225 #! [PySnippetMeshAdvAPI1_7]
226 #! [PySnippetMeshAdvAPI1_4]
227 m2D=ReadUMeshFromFile("file2.med","Example2",-1)
228 #! [PySnippetMeshAdvAPI1_4]
229 #! [PySnippetMeshAdvAPI1_5]
230 m1D=ReadUMeshFromFile("file2.med","Example2",-2)
231 #! [PySnippetMeshAdvAPI1_5]
232 #! [PySnippetMeshAdvAPI1_6]
233 m0D=ReadUMeshFromFile("file2.med","Example2",-3)
234 #! [PySnippetMeshAdvAPI1_6]
236 mm.removeMeshAtLevel(-i)
238 mm.setMeshAtLevel(0,myMesh1)
239 mm.setMeshAtLevel(-1,myMesh2)
241 mm.write("file1.med",2)
242 #! [PySnippetMeshAdvAPI1_9]
243 m2D=ReadUMeshFromFile("file1.med","MyMesh",0)
244 #! [PySnippetMeshAdvAPI1_9]
245 #! [PySnippetMeshAdvAPI1_10]
246 m1D=ReadUMeshFromFile("file1.med","MyMesh",-1)
247 #! [PySnippetMeshAdvAPI1_10]