src/MEDLoader/Swig/MEDLoaderExamplesTest.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2013  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         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)
 175         f.setName("myField")
 176 #! [PySnippetMeshAdvAPI1_3]
 177         MEDLoader.WriteUMesh("file3.med",f.getMesh(),True)
 178         f.setTime(1.2,1,0)
 179         fileNameMultiTimeStep="file3.med"
 180         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileNameMultiTimeStep,f)
 181         f.setTime(1.3,2,0)
 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]
 191         ###
 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]
 234         for i in xrange(4):
 235             mm.removeMeshAtLevel(-i)
 236             pass
 237         mm.setMeshAtLevel(0,myMesh1)
 238         mm.setMeshAtLevel(-1,myMesh2)
 239         mm.setName("MyMesh")
 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]
 247         pass
 248
 249     pass
 250
 251 unittest.main()