1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2012 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.
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 *
25 class MEDLoaderBasicsTest(unittest.TestCase):
26 def testExampleReadFieldOnAllEntity1(self):
27 from MEDLoaderDataForTest import MEDLoaderDataForTest
28 #! [PySnippetReadFieldOnAllEntity1_1]
29 fname="PyExamples1.med"
31 fieldName="FieldOnAll"
34 #! [PySnippetReadFieldOnAllEntity1_1]
35 #! [PySnippetWriteFieldOnAllEntity1_2]
36 m=MEDLoaderDataForTest.build2DMesh_3()
39 f=m.getMeasureField(ON_CELLS)
40 f=f.buildNewTimeReprFromThis(ONE_TIME,False)
41 f.setTime(5.5,iteration,order)
43 # MEDCoupling finished, MEDLoader advanced API specific part starting from here
45 mm.setMeshAtLevel(0,m)
46 ff=MEDFileField1TS.New()
47 ff.setFieldNoProfileSBT(f)
50 #! [PySnippetWriteFieldOnAllEntity1_2]
51 #! [PySnippetReadFieldOnAllEntity1_3]
52 medfileField1TS=MEDFileField1TS.New(fname,fieldName,iteration,order)
53 mm=MEDFileMesh.New(fname)
54 fread=medfileField1TS.getFieldOnMeshAtLevel(ON_CELLS,0,mm)
55 fread2=medfileField1TS.getFieldAtLevel(ON_CELLS,0)
56 self.assertTrue(fread.isEqual(f,1e-12,1e-12))
57 self.assertTrue(fread2.isEqual(f,1e-12,1e-12))
58 #! [PySnippetReadFieldOnAllEntity1_3]
59 #! [PySnippetReadFieldOnAllEntity1_4]
60 medfileFieldMTS=MEDFileFieldMultiTS.New(fname,fieldName)
61 mm=MEDFileMesh.New(fname)
62 fread=medfileFieldMTS.getFieldOnMeshAtLevel(ON_CELLS,iteration,order,0,mm)
63 fread2=medfileFieldMTS.getFieldAtLevel(ON_CELLS,iteration,order,0)
64 self.assertTrue(fread.isEqual(f,1e-12,1e-12))
65 self.assertTrue(fread2.isEqual(f,1e-12,1e-12))
66 #! [PySnippetReadFieldOnAllEntity1_4]
67 #! [PySnippetReadFieldOnAllEntity1_5]
68 medfileFieldMTS=MEDFileFieldMultiTS.New(fname,fieldName)
69 for medfileField1TS in medfileFieldMTS:
70 if medfileField1TS.getTime()[:2]==[iteration,order]:
71 fread=medfileField1TS.getFieldOnMeshAtLevel(ON_CELLS,0,mm)
72 fread2=medfileField1TS.getFieldAtLevel(ON_CELLS,0)
73 self.assertTrue(fread.isEqual(f,1e-12,1e-12))
74 self.assertTrue(fread2.isEqual(f,1e-12,1e-12))
77 #! [PySnippetReadFieldOnAllEntity1_5]
80 def testExampleReadFieldPartial1(self):
81 from MEDLoaderDataForTest import MEDLoaderDataForTest
82 #! [PySnippetReadFieldPartial1_1]
83 fname="PyExamples2.med"
85 fieldName="FieldPartial"
88 #! [PySnippetReadFieldPartial1_1]
89 #! [PySnippetWriteFieldPartial1_2]
90 m=MEDLoaderDataForTest.build2DMesh_1()
91 m.sortCellsInMEDFileFrmt()
93 # end of generation of a mesh -> let's create a field on that mesh
94 f=m.getMeasureField(ON_CELLS)
95 f=f.buildNewTimeReprFromThis(ONE_TIME,False)
96 f.setTime(5.5,iteration,order)
98 # The MEDCoupling part is finished -> let's perform advanced API
100 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
101 # Let's building a sub field
102 pfl=DataArrayInt.New([1,3,4,5])
103 pfl.setName("myPfl") # here it is necessary to give a name to be compliant with MED file
104 f=f[pfl] ; f.getMesh().setName(m.getName()) # of course f should be in coherence with pfl -> f[pfl]
106 ff=MEDFileField1TS.New()
107 tmp=f.getMesh() # useless line, only to show that mesh into f is not considered by MEDFileField1TS.setFieldProfile
108 f.setMesh(None) # useless line, only to show that mesh into f is not considered by MEDFileField1TS.setFieldProfile
109 ff.setFieldProfile(f,mm,0,pfl)
110 f.setMesh(tmp) # useless line, only to show that mesh into f is not considered by MEDFileField1TS.setFieldProfile
113 #! [PySnippetWriteFieldPartial1_2]
114 #! [PySnippetReadFieldPartial1_3]
115 mm=MEDFileMesh.New(fname)
116 medfileField1TS=MEDFileField1TS.New(fname,fieldName,iteration,order)
117 fread=medfileField1TS.getFieldOnMeshAtLevel(ON_CELLS,0,mm)
118 fread2=medfileField1TS.getFieldAtLevel(ON_CELLS,0)
119 self.assertTrue(fread.isEqual(f,1e-12,1e-12))
120 self.assertTrue(fread2.isEqual(f,1e-12,1e-12))
121 #! [PySnippetReadFieldPartial1_3]
122 #! [PySnippetReadFieldPartial1_4]
123 medfileField1TS=MEDFileField1TS.New(fname,fieldName,iteration,order)
124 mm=MEDFileMesh.New(fname)
125 valsRead,pflRead=medfileField1TS.getFieldWithProfile(ON_CELLS,0,mm)
126 self.assertEqual(valsRead.getName(),f.getName())
128 self.assertTrue(valsRead.isEqual(f.getArray(),1e-12))
129 pflRead.setName(pfl.getName())
130 self.assertTrue(pflRead.isEqual(pfl))
131 #! [PySnippetReadFieldPartial1_4]
132 #! [PySnippetReadFieldPartial1_5]
133 firstApproachMesh=fread.getMesh()
134 mm=MEDFileMesh.New(fname)
135 wholeMesh=mm.getMeshAtLevel(0)
136 wholeMesh.tryToShareSameCoords(firstApproachMesh,1e-12)
137 isIncluded,pflComputed=wholeMesh.areCellsIncludedIn(firstApproachMesh,2)
138 self.assertTrue(isIncluded)
139 self.assertEqual(pflComputed.getName(),mm.getName())
140 pflComputed.setName(pflRead.getName())
141 self.assertTrue(pflComputed.isEqual(pflRead))
142 #! [PySnippetReadFieldPartial1_5]
143 #! [PySnippetReadFieldPartial1_6]
144 mm=MEDFileMesh.New(fname)
145 wholeMesh=mm.getMeshAtLevel(0)
146 computedMesh=wholeMesh[pflRead] ; computedMesh.setName(mm.getName())
147 self.assertTrue(computedMesh.isEqual(fread.getMesh(),1e-12))
148 fieldFromSecondApproach=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME)
149 fieldFromSecondApproach.setName(medfileField1TS.getName())
150 fieldFromSecondApproach.setMesh(computedMesh)
151 fieldFromSecondApproach.setArray(valsRead)
152 fieldFromSecondApproach.setTime(medfileField1TS.getTime()[2],medfileField1TS.getTime()[0],medfileField1TS.getTime()[1])
153 self.assertTrue(fieldFromSecondApproach.isEqual(fread,1e-12,1e-12))
154 #! [PySnippetReadFieldPartial1_6]
157 def testExampleMeshAdvAPI1(self):
158 da=DataArrayDouble.New([0.,1.1,2.3,3.6])
160 cmesh=MEDCouplingCMesh.New() ; cmesh.setCoords(da,da,da)
161 myMesh=cmesh.buildUnstructured()
162 #! [PySnippetMeshAdvAPI1_1]
163 self.assertTrue(isinstance(myMesh,MEDCouplingUMesh))
164 myMesh.setName(meshName)
165 MEDLoader.WriteUMesh("wFile1.med",myMesh,True)
166 #! [PySnippetMeshAdvAPI1_1]
167 #! [PySnippetMeshAdvAPI1_2]
168 self.assertTrue(isinstance(myMesh,MEDCouplingUMesh))
169 myMesh.setName(meshName)
170 MEDLoader.WriteUMesh("wFile1.med",myMesh,True)
171 #! [PySnippetMeshAdvAPI1_2]
172 f=myMesh.getMeasureField(ON_CELLS)
173 f=f.buildNewTimeReprFromThis(ONE_TIME,False)
175 #! [PySnippetMeshAdvAPI1_3]
176 MEDLoader.WriteUMesh("file3.med",f.getMesh(),True)
178 fileNameMultiTimeStep="file3.med"
179 MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileNameMultiTimeStep,f)
181 f.applyFunc("sqrt(x)");
182 #Writing second time step with iteration==2 and order==0
183 MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file3.med",f);
184 #! [PySnippetMeshAdvAPI1_3]
185 #! [PySnippetMeshAdvAPI1_11]
186 timeStepsIds=MEDLoader.GetCellFieldIterations("file3.med","Example2","myField")
187 self.assertEqual([(1, 0),(2, 0)],timeStepsIds)
188 fs=MEDLoader.ReadFieldsOnSameMesh(ON_CELLS,"file3.med","Example2",0,"myField",timeStepsIds);
189 #! [PySnippetMeshAdvAPI1_11]
191 myMesh0=myMesh[:] ; myMesh0.setName("Example2")
192 myMesh1=myMesh0.buildDescendingConnectivity()[0] ; myMesh1.setName(myMesh0.getName())
193 myMesh2=myMesh1.buildDescendingConnectivity()[0] ; myMesh2.setName(myMesh0.getName())
194 myMesh3=myMesh2.buildDescendingConnectivity()[0] ; myMesh3.setName(myMesh0.getName())
195 mm=MEDFileUMesh.New()
196 mm.setMeshAtLevel(0,myMesh0)
197 mm.setMeshAtLevel(-1,myMesh1)
198 mm.setMeshAtLevel(-2,myMesh2)
199 mm.setMeshAtLevel(-3,myMesh3)
200 mm.write("file2.med",2)
201 F1Cell=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME)
202 F1Cell.setMesh(myMesh0)
203 F1Cell.setArray(myMesh0.getCoords()[:myMesh0.getNumberOfCells()])
204 F1Cell.setTime(1000.,2,3)
205 F1Cell.setName("F1Cell")
206 MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell)
207 F1Cell1=F1Cell.deepCpy()
208 F1Cell1.setMesh(myMesh1)
209 F1Cell1.setArray(myMesh1.getBarycenterAndOwner())
210 MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell1)
211 #! [PySnippetMeshAdvAPI1_12]
212 f1Cell_3D=MEDLoader.ReadFieldCell("file2.med","Example2",0,"F1Cell",2,3)
213 #! [PySnippetMeshAdvAPI1_12]
214 #! [PySnippetMeshAdvAPI1_13]
215 f1Cell_2D=MEDLoader.ReadFieldCell("file2.med","Example2",-1,"F1Cell",2,3)
216 #! [PySnippetMeshAdvAPI1_13]
217 self.assertTrue(F1Cell.isEqual(f1Cell_3D,1e-12,1e-12))
218 #! [PySnippetMeshAdvAPI1_8]
219 self.assertEqual(3,MEDLoader.ReadUMeshDimFromFile("file2.med","Example2"))
220 #! [PySnippetMeshAdvAPI1_8]
221 #! [PySnippetMeshAdvAPI1_7]
222 m2D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",0)
223 #! [PySnippetMeshAdvAPI1_7]
224 #! [PySnippetMeshAdvAPI1_4]
225 m2D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-1)
226 #! [PySnippetMeshAdvAPI1_4]
227 #! [PySnippetMeshAdvAPI1_5]
228 m1D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-2)
229 #! [PySnippetMeshAdvAPI1_5]
230 #! [PySnippetMeshAdvAPI1_6]
231 m0D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-3)
232 #! [PySnippetMeshAdvAPI1_6]
234 mm.removeMeshAtLevel(-i)
236 mm.setMeshAtLevel(0,myMesh1)
237 mm.setMeshAtLevel(-1,myMesh2)
239 mm.write("file1.med",2)
240 #! [PySnippetMeshAdvAPI1_9]
241 m2D=MEDLoader.ReadUMeshFromFile("file1.med","MyMesh",0)
242 #! [PySnippetMeshAdvAPI1_9]
243 #! [PySnippetMeshAdvAPI1_10]
244 m1D=MEDLoader.ReadUMeshFromFile("file1.med","MyMesh",-1)
245 #! [PySnippetMeshAdvAPI1_10]