src/MEDLoader/Swig/MEDLoaderTest.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2015  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, or (at your option) any later version.
   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 import MEDLoader
  23 import unittest
  24 from math import pi,e,sqrt
  25 from MEDLoaderDataForTest import MEDLoaderDataForTest
  26
  27 class MEDLoaderTest(unittest.TestCase):
  28     def testMesh1DRW(self):
  29         mesh=MEDLoaderDataForTest.build1DMesh_1();
  30         mesh.checkCoherency();
  31         MEDLoader.MEDLoader.WriteUMesh("Pyfile1.med",mesh,True);
  32         mesh_rw=MEDLoader.MEDLoader.ReadUMeshFromFile("Pyfile1.med",mesh.getName(),0);
  33         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  34         pass
  35
  36     def testMesh2DCurveRW(self):
  37         mesh=MEDLoaderDataForTest.build2DCurveMesh_1();
  38         mesh.checkCoherency();
  39         MEDLoader.MEDLoader.WriteUMesh("Pyfile2.med",mesh,True);
  40         mesh_rw=MEDLoader.MEDLoader.ReadUMeshFromFile("Pyfile2.med",mesh.getName(),0);
  41         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  42         pass
  43
  44     def testMesh2DRW(self):
  45         mesh=MEDLoaderDataForTest.build2DMesh_1();
  46         mesh.checkCoherency();
  47         MEDLoader.MEDLoader.WriteUMesh("Pyfile3.med",mesh,True);
  48         mesh_rw=MEDLoader.MEDLoader.ReadUMeshFromFile("Pyfile3.med",mesh.getName(),0);
  49         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  50         pass
  51
  52     def testMesh3DSurfRW(self):
  53         mesh=MEDLoaderDataForTest.build3DSurfMesh_1();
  54         mesh.checkCoherency();
  55         MEDLoader.MEDLoader.WriteUMesh("Pyfile4.med",mesh,True);
  56         mesh_rw=MEDLoader.MEDLoader.ReadUMeshFromFile("Pyfile4.med",mesh.getName(),0);
  57         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  58         pass
  59
  60     def testMesh3DRW(self):
  61         mesh=MEDLoaderDataForTest.build3DMesh_1();
  62         mesh.checkCoherency();
  63         MEDLoader.MEDLoader.WriteUMesh("Pyfile5.med",mesh,True);
  64         mesh_rw=MEDLoader.MEDLoader.ReadUMeshFromFile("Pyfile5.med",mesh.getName(),0);
  65         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  66         pass
  67
  68     def testFieldRW1(self):
  69         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
  70         MEDLoader.MEDLoader.WriteField("Pyfile6.med",f1,True);
  71         f2=MEDLoader.MEDLoader.ReadFieldCell("Pyfile6.med",f1.getMesh().getName(),0,f1.getName(),0,1);
  72         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
  73         #
  74         f1=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
  75         MEDLoader.MEDLoader.WriteField("Pyfile7.med",f1,True);
  76         f2=MEDLoader.MEDLoader.ReadFieldNode("Pyfile7.med",f1.getMesh().getName(),0,f1.getName(),2,3);
  77         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
  78         self.assertRaises(Exception,MEDLoader.MEDLoader.ReadFieldCell,"Pyfile7.med",f1.getMesh().getName(),0,f1.getName(),2,3);
  79         pass
  80
  81     def testFieldRW2(self):
  82         fileName="Pyfile8.med";
  83         VAL1=12345.67890314;
  84         VAL2=-1111111111111.;
  85         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
  86         MEDLoader.MEDLoader.WriteField(fileName,f1,True);
  87         f1.setTime(10.,8,9);
  88         f1.getArray().setIJ(0,0,VAL1);
  89         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
  90         f1.setTime(10.14,18,19);
  91         f1.getArray().setIJ(0,0,VAL2);
  92         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
  93         #retrieving time steps...
  94         f2=MEDLoader.MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),0,f1.getName(),8,9);
  95         f1.setTime(10.,8,9);
  96         f1.getArray().setIJ(0,0,VAL1);
  97         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
  98         f2=MEDLoader.MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),0,f1.getName(),0,1);
  99         f3=MEDLoaderDataForTest.buildVecFieldOnCells_1();
 100         self.assertTrue(f3.isEqual(f2,1e-12,1e-12));
 101         f2=MEDLoader.MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),0,f1.getName(),18,19);
 102         f1.setTime(10.14,18,19);
 103         f1.getArray().setIJ(0,0,VAL2);
 104         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 105         #test of throw on invalid (dt,it)
 106         self.assertRaises(Exception,MEDLoader.MEDLoader.ReadFieldCell,fileName,f1.getMesh().getName(),0,f1.getName(),28,19);
 107         #ON NODES
 108         f1=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
 109         fileName2="Pyfile9.med";
 110         MEDLoader.MEDLoader.WriteField(fileName2,f1,True);
 111         f1.setTime(110.,108,109);
 112         tmp=f1.getArray().getPointer();
 113         f1.getArray().setIJ(0,3,VAL1);
 114         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName2,f1);
 115         f1.setTime(210.,208,209);
 116         f1.getArray().setIJ(0,3,VAL2);
 117         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName2,f1);
 118         f2=MEDLoader.MEDLoader.ReadFieldNode(fileName2,f1.getMesh().getName(),0,f1.getName(),108,109);
 119         f1.setTime(110.,108,109);
 120         f1.getArray().setIJ(0,3,VAL1);
 121         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 122         f2=MEDLoader.MEDLoader.ReadFieldNode(fileName2,f1.getMesh().getName(),0,f1.getName(),2,3);
 123         f3=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
 124         self.assertTrue(f3.isEqual(f2,1e-12,1e-12));
 125         f2=MEDLoader.MEDLoader.ReadFieldNode(fileName2,f1.getMesh().getName(),0,f1.getName(),208,209);
 126         f1.setTime(210.,208,209);
 127         f1.getArray().setIJ(0,3,VAL2);
 128         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 129         pass
 130
 131     #
 132     # Multi field in a same file, but this field has several
 133     #
 134     def testFieldRW3(self):
 135         fileName="Pyfile11.med";
 136         VAL1=12345.67890314;
 137         VAL2=-1111111111111.;
 138         name1="AField";
 139         name3="AMesh1";
 140         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
 141         f1.getMesh().setName(name3);
 142         f1.setName(name1);
 143         f1.setTime(10.,8,9);
 144         tmp=f1.getArray().getPointer();
 145         f1.getArray().setIJ(0,0,VAL1);
 146         MEDLoader.MEDLoader.WriteField(fileName,f1,True);
 147         f1.setTime(10.14,18,19);
 148         f1.getArray().setIJ(0,0,VAL2);
 149         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 150         f1.getMesh().setName(name3);
 151         f1.setTime(10.55,28,29);
 152         f1.getArray().setIJ(0,0,3*VAL1);
 153         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 154         vec=MEDLoader.MEDLoader.GetMeshNamesOnField(fileName,name1);
 155         f1.setTime(10.66,38,39);
 156         f1.getArray().setIJ(0,0,3*VAL2);
 157         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 158         f1.setTime(10.77,48,49);
 159         f1.getArray().setIJ(0,0,4*VAL2);
 160         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 161         #ON NODES
 162         f1=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
 163         f1.setName(name1);
 164         f1.getMesh().setName(name3);
 165         f1.setTime(110.,8,9);
 166         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 167         f1.setTime(110.,108,109);
 168         tmp=f1.getArray().getPointer();
 169         f1.getArray().setIJ(0,3,VAL1);
 170         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 171         f1.setTime(210.,208,209);
 172         f1.getArray().setIJ(0,3,VAL2);
 173         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 174         #
 175         it1=MEDLoader.MEDLoader.GetCellFieldIterations(fileName,name3,name1);
 176         self.assertEqual(5,len(it1));
 177         self.assertEqual(8,it1[0][0]); self.assertEqual(9,it1[0][1]);
 178         self.assertEqual(18,it1[1][0]); self.assertEqual(19,it1[1][1]);
 179         self.assertEqual(28,it1[2][0]); self.assertEqual(29,it1[2][1]);
 180         self.assertEqual(38,it1[3][0]); self.assertEqual(39,it1[3][1]);
 181         self.assertEqual(48,it1[4][0]); self.assertEqual(49,it1[4][1]);
 182         it3=MEDLoader.MEDLoader.GetNodeFieldIterations(fileName,name3,name1);
 183         self.assertEqual(3,len(it3));
 184         self.assertEqual(8,it3[0][0]); self.assertEqual(9,it3[0][1]);
 185         self.assertEqual(108,it3[1][0]); self.assertEqual(109,it3[1][1]);
 186         self.assertEqual(208,it3[2][0]); self.assertEqual(209,it3[2][1]);
 187         #
 188         #
 189         f1=MEDLoader.MEDLoader.ReadFieldCell(fileName,name3,0,name1,8,9);
 190         self.assertAlmostEqual(VAL1,f1.getArray().getIJ(0,0),13);
 191         f1=MEDLoader.MEDLoader.ReadFieldCell(fileName,name3,0,name1,18,19);
 192         self.assertAlmostEqual(VAL2,f1.getArray().getIJ(0,0),13);
 193         f1=MEDLoader.MEDLoader.ReadFieldCell(fileName,name3,0,name1,28,29);
 194         self.assertAlmostEqual(3*VAL1,f1.getArray().getIJ(0,0),13);
 195         f1=MEDLoader.MEDLoader.ReadFieldCell(fileName,name3,0,name1,38,39);
 196         self.assertAlmostEqual(3*VAL2,f1.getArray().getIJ(0,0),13);
 197         f1=MEDLoader.MEDLoader.ReadFieldCell(fileName,name3,0,name1,48,49);
 198         self.assertAlmostEqual(4*VAL2,f1.getArray().getIJ(0,0),13);
 199         #
 200         f1=MEDLoader.MEDLoader.ReadFieldNode(fileName,name3,0,name1,8,9);
 201         self.assertAlmostEqual(71.,f1.getArray().getIJ(0,3),13);
 202         f1=MEDLoader.MEDLoader.ReadFieldNode(fileName,name3,0,name1,108,109);
 203         self.assertAlmostEqual(VAL1,f1.getArray().getIJ(0,3),13);
 204         f1=MEDLoader.MEDLoader.ReadFieldNode(fileName,name3,0,name1,208,209);
 205         self.assertAlmostEqual(VAL2,f1.getArray().getIJ(0,3),13);
 206         pass
 207
 208     def testMultiMeshRW1(self):
 209         fileName="Pyfile10.med";
 210         mesh1=MEDLoaderDataForTest.build3DMesh_1();
 211         part1=[1,2,4,13,15]
 212         mesh2=mesh1.buildPartOfMySelf(part1,True);
 213         mesh2.setName("mesh2");
 214         part2=[3,4,13,14]
 215         mesh3=mesh1.buildPartOfMySelf(part2,True);
 216         mesh3.setName("mesh3");
 217         mesh4=MEDLoader.MEDCouplingUMesh.New();
 218         mesh4.setName("mesh4");
 219         mesh4.setMeshDimension(3);
 220         mesh4.allocateCells(1);
 221         conn=[0,11,1,3]
 222         mesh4.insertNextCell(MEDLoader.NORM_TETRA4,4,conn[0:4])
 223         mesh4.finishInsertingCells();
 224         mesh4.setCoords(mesh1.getCoords());
 225         meshes=[mesh1,mesh2,mesh3,mesh4]
 226         mnane="3DToto";
 227         MEDLoader.MEDLoader.WriteUMeshesPartition(fileName,mnane,meshes,True);
 228         #
 229         mesh5=MEDLoader.MEDLoader.ReadUMeshFromFile(fileName,mnane);
 230         mesh1.setName(mnane);
 231         part3=[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17]
 232         mesh6=mesh5.buildPartOfMySelf(part3,True);
 233         mesh6.setName(mnane);
 234         self.assertTrue(mesh6.isEqual(mesh1,1e-12));
 235         grps=MEDLoader.MEDLoader.GetMeshGroupsNames(fileName,mnane);
 236         self.assertEqual(4,len(grps));
 237         grps.index("mesh2");
 238         grps.index("mesh3");
 239         grps.index("mesh4");
 240         grps.index("3DMesh_1");
 241         #
 242         vec=("mesh2",);
 243         mesh2_2=MEDLoader.MEDLoader.ReadUMeshFromGroups(fileName,mnane,0,vec);
 244         self.assertTrue(mesh2_2.isEqual(mesh2,1e-12));
 245         vec=["mesh3"];
 246         mesh3_2=MEDLoader.MEDLoader.ReadUMeshFromGroups(fileName,mnane,0,vec);
 247         self.assertTrue(mesh3_2.isEqual(mesh3,1e-12));
 248         vec=["mesh4"];
 249         mesh4_2=MEDLoader.MEDLoader.ReadUMeshFromGroups(fileName,mnane,0,vec);
 250         self.assertTrue(mesh4_2.isEqual(mesh4,1e-12));
 251         vec="3DMesh_1";
 252         mesh1_2=MEDLoader.MEDLoader.ReadUMeshFromGroups(fileName,mnane,0,vec);
 253         mesh1.setName("3DMesh_1");
 254         self.assertTrue(mesh1_2.isEqual(mesh1,1e-12));
 255         #
 256         vec=["Family_-3","Family_-5"];
 257         mesh2_2=MEDLoader.MEDLoader.ReadUMeshFromFamilies(fileName,mnane,0,vec);
 258         mesh2_2.setName("mesh2");
 259         self.assertTrue(mesh2_2.isEqual(mesh2,1e-12));
 260         #
 261         ret=MEDLoader.MEDLoader.GetMeshFamiliesNamesOnGroup(fileName,"3DToto","3DMesh_1");
 262         self.assertEqual(4,len(ret));
 263         self.assertEqual(ret[0],"Family_-2");
 264         self.assertEqual(ret[1],"Family_-3");
 265         self.assertEqual(ret[2],"Family_-4");
 266         self.assertEqual(ret[3],"Family_-5");
 267         #
 268         ret1=MEDLoader.MEDLoader.GetMeshGroupsNamesOnFamily(fileName,"3DToto","Family_-3");
 269         self.assertEqual(2,len(ret1));
 270         self.assertEqual(ret1[0],"3DMesh_1");
 271         self.assertEqual(ret1[1],"mesh2");
 272         pass
 273
 274     def testFieldProfilRW1(self):
 275         fileName="Pyfile12.med";
 276         mesh1=MEDLoaderDataForTest.build3DMesh_1();
 277         da,b,newNbOfNodes=mesh1.mergeNodes(1e-12);
 278         MEDLoader.MEDLoader.WriteUMesh(fileName,mesh1,True);
 279         part1=[1,2,4,13,15]
 280         mesh2=mesh1.buildPartOfMySelf(part1,True);
 281         mesh2.setName(mesh1.getName());#<- important for the test
 282         #
 283         nbOfCells=mesh2.getNumberOfCells();
 284         self.assertEqual(5,nbOfCells);
 285         f1=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_CELLS,MEDLoader.ONE_TIME);
 286         f1.setName("VectorFieldOnCells");
 287         f1.setMesh(mesh2);
 288         array=MEDLoader.DataArrayDouble.New();
 289         array.alloc(nbOfCells,2);
 290         f1.setArray(array);
 291         arr1=[71.,171.,10.,110.,20.,120.,30.,130.,40.,140.]
 292         array.setValues(arr1,nbOfCells,2);
 293         f1.setTime(3.14,2,7);
 294         f1.checkCoherency();
 295         #
 296         MEDLoader.MEDLoader.WriteField(fileName,f1,False);#<- False important for the test
 297         #
 298         f2=MEDLoader.MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),0,f1.getName(),2,7);
 299         tt=MEDLoader.MEDLoader.GetTypesOfField(fileName,f1.getMesh().getName(),f1.getName());
 300         self.assertEqual(tt,[MEDLoader.ON_CELLS]);
 301         f2.checkCoherency();
 302         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 303         #
 304         pass
 305
 306     def testFieldGaussRW1(self):
 307         fileName="Pyfile13.med";
 308         f1=MEDLoaderDataForTest.buildVecFieldOnGauss_1();
 309         MEDLoader.MEDLoader.WriteField(fileName,f1,True);
 310         f2=MEDLoader.MEDLoader.ReadField(MEDLoader.ON_GAUSS_PT,fileName,f1.getMesh().getName(),0,f1.getName(),1,5);
 311         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 312         pass
 313
 314     def testFieldGaussNERW1(self):
 315         fileName="Pyfile14.med";
 316         f1=MEDLoaderDataForTest.buildVecFieldOnGaussNE_1();
 317         MEDLoader.MEDLoader.WriteField(fileName,f1,True);
 318         self.assertEqual([MEDLoader.ON_GAUSS_NE],MEDLoader.MEDLoader.GetTypesOfField(fileName,'2DMesh_2','MyFieldOnGaussNE')) #Bug 22/5/2014
 319         f2=MEDLoader.MEDLoader.ReadField(MEDLoader.ON_GAUSS_NE,fileName,f1.getMesh().getName(),0,f1.getName(),1,5);
 320         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 321         pass
 322
 323     def testMesh3DSurfShuffleRW(self):
 324         fileName="Pyfile15.med";
 325         mesh=MEDLoaderDataForTest.build3DSurfMesh_1();
 326         renumber1=[2,5,1,0,3,4]
 327         mesh.renumberCells(renumber1,False);
 328         mesh.checkCoherency();
 329         MEDLoader.MEDLoader.WriteUMesh(fileName,mesh,True);
 330         mesh_rw=MEDLoader.MEDLoader.ReadUMeshFromFile(fileName,mesh.getName(),0);
 331         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
 332         pass
 333
 334     def testMultiFieldShuffleRW1(self):
 335         fileName="Pyfile17.med";
 336         m=MEDLoaderDataForTest.build3DMesh_2();
 337         self.assertEqual(20,m.getNumberOfCells());
 338         self.assertEqual(45,m.getNumberOfNodes());
 339         polys=[1,4,6]
 340         m.convertToPolyTypes(polys);
 341         renum=[1,3,2,8,9,12,13,16,19,0,4,7,5,15,14,17,10,18,6,11]
 342         m.renumberCells(renum,False);
 343         m.orientCorrectlyPolyhedrons();
 344         # Writing
 345         MEDLoader.MEDLoader.WriteUMesh(fileName,m,True);
 346         f1Tmp=m.getMeasureField(False);
 347         f1=f1Tmp.buildNewTimeReprFromThis(MEDLoader.ONE_TIME,False);
 348         f1.setTime(0.,1,2);
 349         f_1=f1.cloneWithMesh(True);
 350         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 351         f1.applyFunc("2*x");
 352         f1.setTime(0.01,3,4);
 353         f_2=f1.cloneWithMesh(True);
 354         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 355         f1.applyFunc("2*x/3");
 356         f1.setTime(0.02,5,6);
 357         f_3=f1.cloneWithMesh(True);
 358         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 359         # Reading
 360         its=[(1,2),(3,4),(5,6)];
 361         fs=MEDLoader.MEDLoader.ReadFieldsOnSameMesh(MEDLoader.ON_CELLS,fileName,f_1.getMesh().getName(),0,f_1.getName(),its);
 362         self.assertEqual(3,len(fs));
 363         self.assertTrue(fs[0].isEqual(f_1,1e-12,1e-12));
 364         self.assertTrue(fs[1].isEqual(f_2,1e-12,1e-12));
 365         self.assertTrue(fs[2].isEqual(f_3,1e-12,1e-12));
 366         pass
 367
 368     def testWriteUMeshesRW1(self):
 369         fileName="Pyfile18.med";
 370         m3d=MEDLoaderDataForTest.build3DMesh_2();
 371         pt=[0.,0.,-0.3]
 372         vec=[0.,0.,1.]
 373         nodes=m3d.findNodesOnPlane(pt,vec,1e-12);
 374         m2d=m3d.buildFacePartOfMySelfNode(nodes,True);
 375         renumber=[1,2,0,4,3]
 376         m2d.renumberCells(renumber,False);
 377         m2d.setName("ExampleOfMultiDimW");
 378         meshes=[m2d,m3d]
 379         MEDLoader.MEDLoader.WriteUMeshes(fileName,meshes,True);
 380         m3d_bis=MEDLoader.MEDLoader.ReadUMeshFromFile(fileName,m2d.getName(),0);
 381         self.assertTrue(not m3d_bis.isEqual(m3d,1e-12));
 382         m3d_bis.setName(m3d.getName());
 383         self.assertTrue(m3d_bis.isEqual(m3d,1e-12));
 384         m2d_bis=MEDLoader.MEDLoader.ReadUMeshFromFile(fileName,m2d.getName(),-1);#-1 for faces
 385         self.assertTrue(m2d_bis.isEqual(m2d,1e-12));
 386         # Creation of a field on faces.
 387         f1=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_CELLS,MEDLoader.ONE_TIME);
 388         f1.setName("FieldOnFacesShuffle");
 389         f1.setMesh(m2d);
 390         array=MEDLoader.DataArrayDouble.New();
 391         arr1=[71.,171.,10.,110.,20.,120.,30.,130.,40.,140.]
 392         array.setValues(arr1,m2d.getNumberOfCells(),2);
 393         array.setInfoOnComponent(0,"plkj [mm]");
 394         array.setInfoOnComponent(1,"pqqqss [mm]");
 395         f1.setArray(array);
 396         tmp=array.setValues(arr1,m2d.getNumberOfCells(),2);
 397         f1.setTime(3.14,2,7);
 398         f1.checkCoherency();
 399         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 400         f2=MEDLoader.MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),-1,f1.getName(),2,7);
 401         self.assertTrue(f2.isEqual(f1,1e-12,1e-12));
 402         pass
 403
 404     def testFieldNodeProfilRW1(self):
 405         fileName="Pyfile19.med";
 406         fileName2="Pyfile20.med";
 407         m=MEDLoaderDataForTest.build2DMesh_1();
 408         nbOfNodes=m.getNumberOfNodes();
 409         MEDLoader.MEDLoader.WriteUMesh(fileName,m,True);
 410         f1=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_NODES,MEDLoader.ONE_TIME);
 411         f1.setName("VFieldOnNodes");
 412         f1.setMesh(m);
 413         array=MEDLoader.DataArrayDouble.New();
 414         arr1=[1.,101.,2.,102.,3.,103.,4.,104.,5.,105.,6.,106.,7.,107.,8.,108.,9.,109.,10.,110.,11.,111.,12.,112.]
 415         array.setValues(arr1,nbOfNodes,2);
 416         f1.setArray(array);
 417         array.setInfoOnComponent(0,"tyty [mm]");
 418         array.setInfoOnComponent(1,"uiop [MW]");
 419         f1.setTime(3.14,2,7);
 420         f1.checkCoherency();
 421         arr2=[1,4]
 422         f2=f1.buildSubPart(arr2);
 423         f2.getMesh().setName(f1.getMesh().getName());
 424         MEDLoader.MEDLoader.WriteField(fileName,f2,False);#<- False important for the test
 425         #
 426         f3=MEDLoader.MEDLoader.ReadFieldNode(fileName,f2.getMesh().getName(),0,f2.getName(),2,7);
 427         f3.checkCoherency();
 428         self.assertTrue(f3.isEqual(f2,1e-12,1e-12));
 429         #
 430         arr3=[1,3,0,5,2,4]
 431         f2.renumberNodes(arr3);
 432         MEDLoader.MEDLoader.WriteUMesh(fileName2,m,True);
 433         MEDLoader.MEDLoader.WriteField(fileName2,f2,False);#<- False important for the test
 434         f3=MEDLoader.MEDLoader.ReadFieldNode(fileName2,f2.getMesh().getName(),0,f2.getName(),2,7);
 435         f3.checkCoherency();
 436         self.assertTrue(f3.isEqual(f2,1e-12,1e-12));
 437         #
 438         pass
 439
 440     def testFieldNodeProfilRW2(self):
 441         fileName="Pyfile23.med";
 442         mesh=MEDLoaderDataForTest.build3DSurfMesh_1();
 443         MEDLoader.MEDLoader.WriteUMesh(fileName,mesh,True);
 444         #
 445         f1=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_NODES,MEDLoader.ONE_TIME);
 446         f1.setName("FieldMix");
 447         f1.setMesh(mesh);
 448         arr2=[1071.,1171.,1010.,1110.,1020.,1120.,1030.,1130.,1040.,1140.,1050.,1150.,
 449               1060.,1160.,1070.,1170.,1080.,1180.,1090.,1190.,1091.,1191.,1092.,1192.];
 450         array=MEDLoader.DataArrayDouble.New();
 451         array.setValues(arr2,12,2);
 452         f1.setArray(array);
 453         array.setInfoOnComponent(0,"plkj [mm]");
 454         array.setInfoOnComponent(1,"pqqqss [mm]");
 455         tmp=array.getPointer();
 456         f1.setTime(3.17,2,7);
 457         #
 458         renumArr=[3,7,2,1,5,11,10,0,9,6,8,4]
 459         f1.renumberNodes(renumArr);
 460         f1.checkCoherency();
 461         MEDLoader.MEDLoader.WriteField(fileName,f1,False);#<- False important for the test
 462         f2=MEDLoader.MEDLoader.ReadFieldNode(fileName,f1.getMesh().getName(),0,f1.getName(),2,7);
 463         self.assertTrue(f2.isEqual(f1,1e-12,1e-12));
 464         #
 465         pass
 466
 467     def testMixCellAndNodesFieldRW1(self):
 468         fileName="Pyfile21.med";
 469         mesh=MEDLoaderDataForTest.build3DSurfMesh_1();
 470         f1=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_CELLS,MEDLoader.ONE_TIME);
 471         f1.setName("FieldMix");
 472         f1.setMesh(mesh);
 473         array=MEDLoader.DataArrayDouble.New();
 474         f1.setArray(array);
 475         arr1=[71.,171.,10.,110.,20.,120.,30.,130.,40.,140.,50.,150.]
 476         array.setValues(arr1,6,2);
 477         array.setInfoOnComponent(0,"plkj [mm]");
 478         array.setInfoOnComponent(1,"pqqqss [mm]");
 479         f1.setTime(3.14,2,7);
 480         f1.checkCoherency();
 481         #
 482         f2=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_NODES,MEDLoader.ONE_TIME);
 483         f2.setName("FieldMix");
 484         f2.setMesh(mesh);
 485         array=MEDLoader.DataArrayDouble.New();
 486         f2.setArray(array);
 487         arr2=[1071.,1171.,1010.,1110.,1020.,1120.,1030.,1130.,1040.,1140.,1050.,1150.,
 488               1060.,1160.,1070.,1170.,1080.,1180.,1090.,1190.,1091.,1191.,1092.,1192.]
 489         array.setValues(arr2,12,2)
 490         array.setInfoOnComponent(0,"plkj [mm]");
 491         array.setInfoOnComponent(1,"pqqqss [mm]");
 492         f2.setTime(3.14,2,7);
 493         f2.checkCoherency();
 494         #
 495         MEDLoader.MEDLoader.WriteField(fileName,f1,True);
 496         ts=MEDLoader.MEDLoader.GetTypesOfField(fileName,f1.getMesh().getName(),f1.getName());
 497         self.assertEqual(1,len(ts));
 498         self.assertEqual(MEDLoader.ON_CELLS,ts[0]);
 499         fs=MEDLoader.MEDLoader.GetAllFieldNamesOnMesh(fileName,f1.getMesh().getName());
 500         self.assertEqual(1,len(fs));
 501         self.assertTrue(fs[0]=="FieldMix");
 502         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f2);
 503         fs=MEDLoader.MEDLoader.GetAllFieldNamesOnMesh(fileName,f1.getMesh().getName());
 504         self.assertEqual(1,len(fs));
 505         self.assertTrue(fs[0]=="FieldMix");
 506         #
 507         ts=MEDLoader.MEDLoader.GetTypesOfField(fileName,f1.getMesh().getName(),f1.getName());
 508         self.assertEqual(2,len(ts));
 509         self.assertEqual(MEDLoader.ON_NODES,ts[0]);
 510         self.assertEqual(MEDLoader.ON_CELLS,ts[1]);
 511         #
 512         f3=MEDLoader.MEDLoader.ReadFieldNode(fileName,f1.getMesh().getName(),0,f1.getName(),2,7);
 513         self.assertTrue(f3.isEqual(f2,1e-12,1e-12));
 514         f3=MEDLoader.MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),0,f1.getName(),2,7);
 515         self.assertTrue(f3.isEqual(f1,1e-12,1e-12));
 516         #
 517         pass
 518
 519     def testGetAllFieldNamesRW1(self):
 520         fileName="Pyfile22.med";
 521         mesh=MEDLoaderDataForTest.build2DMesh_2();
 522         f1=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_NODES,MEDLoader.ONE_TIME);
 523         f1.setName("Field1");
 524         f1.setTime(3.44,5,6);
 525         f1.setMesh(mesh);
 526         f1.fillFromAnalytic(2,"x+y");
 527         MEDLoader.MEDLoader.WriteField(fileName,f1,True);
 528         f1.setTime(1002.3,7,8);
 529         f1.fillFromAnalytic(2,"x+77.*y");
 530         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 531         f1.setName("Field2");
 532         MEDLoader.MEDLoader.WriteField(fileName,f1,False);
 533         f1.setName("Field3");
 534         mesh.setName("2DMesh_2Bis");
 535         MEDLoader.MEDLoader.WriteField(fileName,f1,False);
 536         f1=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_CELLS,MEDLoader.ONE_TIME);
 537         f1.setName("Field8");
 538         f1.setTime(8.99,7,9);
 539         f1.setMesh(mesh);
 540         f1.fillFromAnalytic(3,"3*x+y");
 541         MEDLoader.MEDLoader.WriteField(fileName,f1,False);
 542         fs=MEDLoader.MEDLoader.GetAllFieldNames(fileName);
 543         self.assertEqual(4,len(fs));
 544         self.assertTrue(fs[0]=="Field1");
 545         self.assertTrue(fs[1]=="Field2");
 546         self.assertTrue(fs[2]=="Field3");
 547         self.assertTrue(fs[3]=="Field8");
 548         pass
 549
 550     def testBigNbOfCompoNonReg(self):
 551         fileName="Pyfile57.med"
 552         m=MEDLoader.MEDCouplingCMesh() ; m.setCoords(MEDLoader.DataArrayDouble([0,1,2,3]),MEDLoader.DataArrayDouble([0,1]),MEDLoader.DataArrayDouble([0,1]))
 553         m=m.buildUnstructured() ; m.setName("TinyMesh")
 554         f=MEDLoader.MEDCouplingFieldDouble(MEDLoader.ON_CELLS) ; f.setMesh(m)
 555         nbOfCompo=4100
 556         arr=MEDLoader.DataArrayDouble(nbOfCompo*3) ; arr.iota()
 557         arr.rearrange(nbOfCompo)
 558         arr.setInfoOnComponents(["c%i"%(i) for i in xrange(nbOfCompo)])
 559         f.setArray(arr)
 560         f.setName("FieldBigCompo")
 561         MEDLoader.MEDLoader.WriteField(fileName,f,True)
 562         f2=MEDLoader.MEDLoader.ReadFieldCell(fileName,m.getName(),0,f.getName(),-1,-1)
 563         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 564         pass
 565
 566     def testMultiMeshTypeWrite0(self):
 567         fname="Pyfile73.med"
 568         m=MEDLoader.MEDCoupling1SGTUMesh("mesh",MEDLoader.NORM_QUAD4) ; m.allocateCells()
 569         m.insertNextCell([0,2,1,3])
 570         m.setCoords(MEDLoader.DataArrayDouble([0.,0.,1.,1.,1.,0.,0.,1.],4,2))
 571         #
 572         ms=[m.deepCpy() for i in xrange(4)]
 573         for i,elt in enumerate(ms):
 574             elt.translate([float(i)*1.5,0.])
 575             pass
 576         #
 577         m0=MEDLoader.MEDCoupling1SGTUMesh.Merge1SGTUMeshes(ms)
 578         f=m0.getMeasureField(False) ; f.getArray().setInfoOnComponents(["ABC [defg]"])
 579         MEDLoader.MEDLoader.WriteField(fname,f,True)
 580         #
 581         fRead=MEDLoader.MEDLoader.ReadFieldCell(fname,"merge",0,f.getName(),-1,-1)
 582         fRead.setMesh(MEDLoader.MEDCoupling1SGTUMesh(fRead.getMesh()))
 583         self.assertTrue(f.isEqual(fRead,1e-12,1e-12))
 584         #
 585         m0=m0.buildUnstructured() ; m0.convertAllToPoly()
 586         m0=MEDLoader.MEDCoupling1DGTUMesh(m0)
 587         f=m0.getMeasureField(False) ; f.getArray().setInfoOnComponents(["ABC [defg]"])
 588         MEDLoader.MEDLoader.WriteField(fname,f,True)
 589         #
 590         fRead=MEDLoader.MEDLoader.ReadFieldCell(fname,"merge",0,f.getName(),-1,-1)
 591         fRead.setMesh(MEDLoader.MEDCoupling1DGTUMesh(fRead.getMesh()))
 592         self.assertTrue(f.isEqual(fRead,1e-12,1e-12))
 593         #
 594         m0=MEDLoader.MEDCouplingCMesh()
 595         arr=MEDLoader.DataArrayDouble(4) ; arr.iota()
 596         m0.setCoords(arr,arr)
 597         m0.setName("mesh")
 598         f=m0.getMeasureField(False) ; f.getArray().setInfoOnComponents(["ABC [defg]"])
 599         MEDLoader.MEDLoader.WriteField(fname,f,True)
 600         #
 601         fRead=MEDLoader.MEDLoader.ReadFieldCell(fname,"mesh",0,f.getName(),-1,-1)
 602         self.assertTrue(f.isEqual(fRead,1e-12,1e-12))
 603         #
 604         c=m0.buildUnstructured().getCoords()
 605         m0=MEDLoader.MEDCouplingCurveLinearMesh("mesh")
 606         m0.setNodeGridStructure([4,4])
 607         m0.setCoords(c)
 608         f=m0.getMeasureField(False) ; f.getArray().setInfoOnComponents(["ABC [defg]"])
 609         MEDLoader.MEDLoader.WriteField(fname,f,True)
 610         #
 611         fRead=MEDLoader.MEDLoader.ReadFieldCell(fname,"mesh",0,f.getName(),-1,-1)
 612         self.assertTrue(f.isEqual(fRead,1e-12,1e-12))
 613         pass
 614
 615     def testMultiMeshTypeWrite1(self):
 616         fname="Pyfile74.med"
 617         m=MEDLoader.MEDCoupling1SGTUMesh("mesh",MEDLoader.NORM_QUAD4) ; m.allocateCells()
 618         m.insertNextCell([0,2,1,3])
 619         m.setCoords(MEDLoader.DataArrayDouble([0.,0.,1.,1.,1.,0.,0.,1.],4,2))
 620         #
 621         ms=[m.deepCpy() for i in xrange(4)]
 622         for i,elt in enumerate(ms):
 623             elt.translate([float(i)*1.5,0.])
 624             pass
 625         m0=MEDLoader.MEDCoupling1SGTUMesh.Merge1SGTUMeshes(ms)
 626         MEDLoader.MEDLoader.WriteMesh(fname,m0,True)
 627         #
 628         mRead=MEDLoader.MEDLoader.ReadMeshFromFile(fname,"merge",0)
 629         self.assertTrue(isinstance(mRead,MEDLoader.MEDCouplingUMesh))
 630         mRead=MEDLoader.MEDCoupling1SGTUMesh(mRead)
 631         self.assertTrue(m0.isEqual(mRead,1e-12))
 632         #
 633         m0=m0.buildUnstructured() ; m0.convertAllToPoly()
 634         m0=MEDLoader.MEDCoupling1DGTUMesh(m0)
 635         MEDLoader.MEDLoader.WriteMesh(fname,m0,True)
 636         #
 637         mRead=MEDLoader.MEDLoader.ReadMeshFromFile(fname,"merge",0)
 638         mRead=MEDLoader.MEDCoupling1DGTUMesh(mRead)
 639         self.assertTrue(m0.isEqual(mRead,1e-12))
 640         #
 641         m0=MEDLoader.MEDCouplingCMesh()
 642         arr=MEDLoader.DataArrayDouble(4) ; arr.iota()
 643         m0.setCoords(arr,arr)
 644         m0.setName("mesh")
 645         MEDLoader.MEDLoader.WriteMesh(fname,m0,True)
 646         #
 647         mRead=MEDLoader.MEDLoader.ReadMeshFromFile(fname,0)
 648         self.assertTrue(isinstance(mRead,MEDLoader.MEDCouplingCMesh))
 649         self.assertTrue(m0.isEqual(mRead,1e-12))
 650         #
 651         c=m0.buildUnstructured().getCoords()
 652         m0=MEDLoader.MEDCouplingCurveLinearMesh("mesh")
 653         m0.setNodeGridStructure([4,4])
 654         m0.setCoords(c)
 655         MEDLoader.MEDLoader.WriteMesh(fname,m0,True)
 656         #
 657         mRead=MEDLoader.MEDLoader.ReadMeshFromFile(fname,0)
 658         self.assertTrue(isinstance(mRead,MEDLoader.MEDCouplingCurveLinearMesh))
 659         self.assertTrue(m0.isEqual(mRead,1e-12))
 660         pass
 661
 662     def testChangeGroupName(self):
 663         """ This test is a non regression test on MEDFileUMesh.changeGroupName thanks to Alliance.
 664         """
 665         mfd=MEDLoaderDataForTest.buildAMEDFileDataWithGroupOnOneFamilyForSauv()
 666         mesh = mfd.getMeshes().getMeshAtPos(0)
 667         mesh.changeGroupName("grp0_LM1", "xonall1")
 668         self.assertTrue("xonall1" in mesh.getGroupsNames())
 669         pass
 670
 671     def testFieldWithTooLongName(self):
 672         """ This test is a non regression test, to check that in basic API the policies are taken into account.
 673         """
 674         fname="Pyfile75.med"
 675         # Coordinates
 676         coords = [0.,0., 0.,1., 1.,1., 1.,0.]
 677         # lvl 0 connectivity
 678         conn2D   = [1,2,3,4]
 679         # lvl 0 mesh
 680         m=MEDLoader.MEDCouplingUMesh.New("mesh",2)
 681         m.allocateCells(1)
 682         m.insertNextCell(MEDLoader.NORM_QUAD4,4,conn2D)
 683         m.finishInsertingCells()
 684         # assigning coordinates
 685         meshCoords=MEDLoader.DataArrayDouble.New()
 686         meshCoords.setValues(coords, 4, 2)
 687         m.setCoords(meshCoords)
 688         #
 689         f=MEDLoader.MEDCouplingFieldDouble.New(MEDLoader.ON_CELLS,MEDLoader.ONE_TIME)
 690         f.setMesh(m)
 691         d=MEDLoader.DataArrayDouble.New()
 692         d.alloc(1,1)
 693         d.iota(1.)
 694         # seting a long name
 695         d.setInfoOnComponent(0,"CONCENTRATION of I129")
 696         f.setArray(d)
 697         f.setName("field")
 698         #
 699         mm=MEDLoader.MEDFileUMesh()
 700         MEDLoader.MEDLoader.SetTooLongStrPolicy(2)
 701         MEDLoader.MEDLoader.AssignStaticWritePropertiesTo(mm)
 702         self.assertEqual(2,mm.getTooLongStrPolicy())
 703         MEDLoader.MEDLoader.SetTooLongStrPolicy(0)
 704         MEDLoader.MEDLoader.AssignStaticWritePropertiesTo(mm)
 705         self.assertEqual(0,mm.getTooLongStrPolicy())
 706         del mm
 707         #
 708         MEDLoader.MEDLoader.SetTooLongStrPolicy(2)
 709         self.assertRaises(MEDLoader.InterpKernelException,MEDLoader.MEDLoader.WriteField,fname,f,True)# the component name is too long + policy 2 -> throw
 710         f.getArray().setInfoOnComponent(0,'I129')
 711         MEDLoader.MEDLoader.WriteField(fname,f,True)
 712         pass
 713
 714     def testUsingAlreadyWrittenMesh2(self):
 715         """ This test focuses on MEDLoader.WriteFieldUsingAlreadyWrittenMesh with mesh different from UMesh.
 716         """
 717         fname="Pyfile76.med"
 718         mesh=MEDLoader.MEDCouplingCMesh("mesh")
 719         arrX=MEDLoader.DataArrayDouble([0,1,2,3])
 720         arrY=MEDLoader.DataArrayDouble([0,2,3,5,7])
 721         arrZ=MEDLoader.DataArrayDouble([7])
 722         mesh.setCoords(arrX,arrY,arrZ)
 723         #
 724         f1=MEDLoader.MEDCouplingFieldDouble(MEDLoader.ON_NODES) ; f1.setName("f1")
 725         f1.setMesh(mesh)
 726         arr=MEDLoader.DataArrayDouble(20) ; arr.iota()
 727         f1.setArray(arr)
 728         f1.checkCoherency()
 729         #
 730         f2=MEDLoader.MEDCouplingFieldDouble(MEDLoader.ON_NODES) ; f2.setName("f2")
 731         f2.setMesh(mesh)
 732         arr=MEDLoader.DataArrayDouble(20) ; arr.iota() ; arr*=3
 733         f2.setArray(arr)
 734         f2.checkCoherency()
 735         #
 736         f11=f1.deepCpy() ; (f11.getArray())[:]*=4 ; f11.setTime(1.1,5,6)
 737         #
 738         MEDLoader.MEDLoader.WriteMesh(fname,f1.getMesh(),True)
 739         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fname,f1)
 740         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fname,f2)
 741         MEDLoader.MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fname,f11)
 742         ##
 743         f1r=MEDLoader.MEDLoader.ReadFieldNode(fname,"mesh",0,"f1",-1,-1);
 744         self.assertTrue(f1.isEqual(f1r,1e-12,1e-12))
 745         self.assertTrue(f1r.getArray().isEqual(MEDLoader.DataArrayDouble([0.,1.,2.,3.,4.,5.,6.,7.,8.,9.,10.,11.,12.,13.,14.,15.,16.,17.,18.,19.]),1e-12))
 746         f2r=MEDLoader.MEDLoader.ReadFieldNode(fname,"mesh",0,"f2",-1,-1);
 747         self.assertTrue(f2.isEqual(f2r,1e-12,1e-12))
 748         self.assertTrue(f2r.getArray().isEqual(MEDLoader.DataArrayDouble([0.,3.,6.,9.,12.,15.,18.,21.,24.,27.,30.,33.,36.,39.,42.,45.,48.,51.,54.,57.]),1e-12))
 749         f3r=MEDLoader.MEDLoader.ReadFieldNode(fname,"mesh",0,"f1",5,6);
 750         self.assertTrue(f11.isEqual(f3r,1e-12,1e-12))
 751         self.assertTrue(f3r.getArray().isEqual(MEDLoader.DataArrayDouble([0.,4.,8.,12.,16.,20.,24.,28.,32.,36.,40.,44.,48.,52.,56.,60.,64.,68.,72.,76.]),1e-12))
 752         pass
 753     pass
 754
 755 if __name__ == "__main__":
 756   unittest.main()