src/MEDLoader/Swig/MEDLoaderExamplesTest.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2012  CEA/DEN, EDF R&D
   3 #
   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.
   8 #
   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.
  13 #
  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
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 # Author : Anthony Geay (CEA/DEN)
  21
  22 from MEDLoader import *
  23 import unittest
  24
  25 class MEDLoaderBasicsTest(unittest.TestCase):
  26     def testExampleReadFieldOnAllEntity1(self):
  27         from MEDLoaderDataForTest import MEDLoaderDataForTest
  28 #! [PySnippetReadFieldOnAllEntity1_1]
  29         fname="PyExamples1.med"
  30         meshName="mesh"
  31         fieldName="FieldOnAll"
  32         iteration=3
  33         order=4
  34 #! [PySnippetReadFieldOnAllEntity1_1]
  35 #! [PySnippetWriteFieldOnAllEntity1_2]
  36         m=MEDLoaderDataForTest.build2DMesh_3()
  37         m=m[:10]
  38         m.setName(meshName)
  39         f=m.getMeasureField(ON_CELLS)
  40         f=f.buildNewTimeReprFromThis(ONE_TIME,False)
  41         f.setTime(5.5,iteration,order)
  42         f.setName(fieldName)
  43         # MEDCoupling finished, MEDLoader advanced API specific part starting from here
  44         mm=MEDFileUMesh.New()
  45         mm.setMeshAtLevel(0,m)
  46         ff=MEDFileField1TS.New()
  47         ff.setFieldNoProfileSBT(f)
  48         mm.write(fname,2)
  49         ff.write(fname,0)
  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))
  75                 pass
  76             pass
  77 #! [PySnippetReadFieldOnAllEntity1_5]
  78         pass
  79
  80     def testExampleReadFieldPartial1(self):
  81         from MEDLoaderDataForTest import MEDLoaderDataForTest
  82 #! [PySnippetReadFieldPartial1_1]
  83         fname="PyExamples2.med"
  84         meshName="mesh"
  85         fieldName="FieldPartial"
  86         iteration=3
  87         order=4
  88 #! [PySnippetReadFieldPartial1_1]
  89 #! [PySnippetWriteFieldPartial1_2]
  90         m=MEDLoaderDataForTest.build2DMesh_1()
  91         m.sortCellsInMEDFileFrmt()
  92         m.setName(meshName)
  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)
  97         f.setName(fieldName)
  98         # The MEDCoupling part is finished -> let's perform advanced API
  99         mm=MEDFileUMesh.New()
 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]
 105         #
 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
 111         mm.write(fname,2)
 112         ff.write(fname,0)
 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())
 127         valsRead.setName("")
 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]
 155         pass
 156
 157     def testExampleMeshAdvAPI1(self):
 158         da=DataArrayDouble.New([0.,1.1,2.3,3.6])
 159         meshName="Example2"
 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)
 174         f.setName("myField")
 175 #! [PySnippetMeshAdvAPI1_3]
 176         MEDLoader.WriteUMesh("file3.med",f.getMesh(),True)
 177         f.setTime(1.2,1,0)
 178         fileNameMultiTimeStep="file3.med"
 179         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileNameMultiTimeStep,f)
 180         f.setTime(1.3,2,0)
 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]
 190         ###
 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]
 233         for i in xrange(4):
 234             mm.removeMeshAtLevel(-i)
 235             pass
 236         mm.setMeshAtLevel(0,myMesh1)
 237         mm.setMeshAtLevel(-1,myMesh2)
 238         mm.setName("MyMesh")
 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]
 246         pass
 247
 248     pass
 249
 250 unittest.main()