1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2013 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 os.remove("wFile1.med")
168 #! [PySnippetMeshAdvAPI1_2]
169 self.assertTrue(isinstance(myMesh,MEDCouplingUMesh))
170 myMesh.setName(meshName)
171 MEDLoader.WriteUMesh("wFile1.med",myMesh,False)
172 #! [PySnippetMeshAdvAPI1_2]
173 f=myMesh.getMeasureField(ON_CELLS)
174 f=f.buildNewTimeReprFromThis(ONE_TIME,False)
176 #! [PySnippetMeshAdvAPI1_3]
177 MEDLoader.WriteUMesh("file3.med",f.getMesh(),True)
179 fileNameMultiTimeStep="file3.med"
180 MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileNameMultiTimeStep,f)
182 f.applyFunc("sqrt(x)");
183 #Writing second time step with iteration==2 and order==0
184 MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file3.med",f);
185 #! [PySnippetMeshAdvAPI1_3]
186 #! [PySnippetMeshAdvAPI1_11]
187 timeStepsIds=MEDLoader.GetCellFieldIterations("file3.med","Example2","myField")
188 self.assertEqual([(1, 0),(2, 0)],timeStepsIds)
189 fs=MEDLoader.ReadFieldsOnSameMesh(ON_CELLS,"file3.med","Example2",0,"myField",timeStepsIds);
190 #! [PySnippetMeshAdvAPI1_11]
192 myMesh0=myMesh[:] ; myMesh0.setName("Example2")
193 myMesh1=myMesh0.buildDescendingConnectivity()[0] ; myMesh1.setName(myMesh0.getName())
194 myMesh2=myMesh1.buildDescendingConnectivity()[0] ; myMesh2.setName(myMesh0.getName())
195 myMesh3=myMesh2.buildDescendingConnectivity()[0] ; myMesh3.setName(myMesh0.getName())
196 mm=MEDFileUMesh.New()
197 mm.setMeshAtLevel(0,myMesh0)
198 mm.setMeshAtLevel(-1,myMesh1)
199 mm.setMeshAtLevel(-2,myMesh2)
200 mm.setMeshAtLevel(-3,myMesh3)
201 mm.write("file2.med",2)
202 F1Cell=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME)
203 F1Cell.setMesh(myMesh0)
204 F1Cell.setArray(myMesh0.getCoords()[:myMesh0.getNumberOfCells()])
205 F1Cell.setTime(1000.,2,3)
206 F1Cell.setName("F1Cell")
207 MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell)
208 F1Cell1=F1Cell.deepCpy()
209 F1Cell1.setMesh(myMesh1)
210 F1Cell1.setArray(myMesh1.getBarycenterAndOwner())
211 MEDLoader.WriteFieldUsingAlreadyWrittenMesh("file2.med",F1Cell1)
212 #! [PySnippetMeshAdvAPI1_12]
213 f1Cell_3D=MEDLoader.ReadFieldCell("file2.med","Example2",0,"F1Cell",2,3)
214 #! [PySnippetMeshAdvAPI1_12]
215 #! [PySnippetMeshAdvAPI1_13]
216 f1Cell_2D=MEDLoader.ReadFieldCell("file2.med","Example2",-1,"F1Cell",2,3)
217 #! [PySnippetMeshAdvAPI1_13]
218 self.assertTrue(F1Cell.isEqual(f1Cell_3D,1e-12,1e-12))
219 #! [PySnippetMeshAdvAPI1_8]
220 self.assertEqual(3,MEDLoader.ReadUMeshDimFromFile("file2.med","Example2"))
221 #! [PySnippetMeshAdvAPI1_8]
222 #! [PySnippetMeshAdvAPI1_7]
223 m2D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",0)
224 #! [PySnippetMeshAdvAPI1_7]
225 #! [PySnippetMeshAdvAPI1_4]
226 m2D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-1)
227 #! [PySnippetMeshAdvAPI1_4]
228 #! [PySnippetMeshAdvAPI1_5]
229 m1D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-2)
230 #! [PySnippetMeshAdvAPI1_5]
231 #! [PySnippetMeshAdvAPI1_6]
232 m0D=MEDLoader.ReadUMeshFromFile("file2.med","Example2",-3)
233 #! [PySnippetMeshAdvAPI1_6]
235 mm.removeMeshAtLevel(-i)
237 mm.setMeshAtLevel(0,myMesh1)
238 mm.setMeshAtLevel(-1,myMesh2)
240 mm.write("file1.med",2)
241 #! [PySnippetMeshAdvAPI1_9]
242 m2D=MEDLoader.ReadUMeshFromFile("file1.med","MyMesh",0)
243 #! [PySnippetMeshAdvAPI1_9]
244 #! [PySnippetMeshAdvAPI1_10]
245 m1D=MEDLoader.ReadUMeshFromFile("file1.med","MyMesh",-1)
246 #! [PySnippetMeshAdvAPI1_10]