src/MEDLoader/Swig/MEDLoaderTest2.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
  21 from MEDLoader import *
  22 import unittest
  23 from math import pi,e,sqrt
  24 from MEDLoaderDataForTest import MEDLoaderDataForTest
  25
  26 class MEDLoaderTest(unittest.TestCase):
  27     def testMesh1DRW(self):
  28         mesh=MEDLoaderDataForTest.build1DMesh_1();
  29         mesh.checkCoherency();
  30         MEDLoader.WriteUMeshDep("Pyfile1.med",mesh,False);
  31         mesh_rw=MEDLoader.ReadUMeshFromFile("Pyfile1.med",mesh.getName(),0);
  32         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  33         pass
  34
  35     def testMesh2DCurveRW(self):
  36         mesh=MEDLoaderDataForTest.build2DCurveMesh_1();
  37         mesh.checkCoherency();
  38         MEDLoader.WriteUMeshDep("Pyfile2.med",mesh,False);
  39         mesh_rw=MEDLoader.ReadUMeshFromFile("Pyfile2.med",mesh.getName(),0);
  40         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  41         pass
  42
  43     def testMesh2DRW(self):
  44         mesh=MEDLoaderDataForTest.build2DMesh_1();
  45         mesh.checkCoherency();
  46         MEDLoader.WriteUMeshDep("Pyfile3.med",mesh,False);
  47         mesh_rw=MEDLoader.ReadUMeshFromFile("Pyfile3.med",mesh.getName(),0);
  48         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  49         pass
  50
  51     def testMesh3DSurfRW(self):
  52         mesh=MEDLoaderDataForTest.build3DSurfMesh_1();
  53         mesh.checkCoherency();
  54         MEDLoader.WriteUMeshDep("Pyfile4.med",mesh,False);
  55         mesh_rw=MEDLoader.ReadUMeshFromFile("Pyfile4.med",mesh.getName(),0);
  56         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  57         pass
  58
  59     def testMesh3DRW(self):
  60         mesh=MEDLoaderDataForTest.build3DMesh_1();
  61         mesh.checkCoherency();
  62         MEDLoader.WriteUMeshDep("Pyfile5.med",mesh,False);
  63         mesh_rw=MEDLoader.ReadUMeshFromFile("Pyfile5.med",mesh.getName(),0);
  64         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
  65         pass
  66
  67     def testFieldRW1(self):
  68         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
  69         MEDLoader.WriteFieldDep("Pyfile6.med",f1,False);
  70         f2=MEDLoader.ReadFieldCell("Pyfile6.med",f1.getMesh().getName(),0,f1.getName(),0,1);
  71         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
  72         #
  73         f1=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
  74         MEDLoader.WriteFieldDep("Pyfile7.med",f1,False);
  75         f2=MEDLoader.ReadFieldNode("Pyfile7.med",f1.getMesh().getName(),0,f1.getName(),2,3);
  76         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
  77         pass
  78
  79     def testFieldRW2(self):
  80         fileName="Pyfile8.med";
  81         VAL1=12345.67890314;
  82         VAL2=-1111111111111.;
  83         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
  84         MEDLoader.WriteFieldDep(fileName,f1,False);
  85         f1.setTime(10.,8,9);
  86         f1.getArray().setIJ(0,0,VAL1);
  87         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
  88         f1.setTime(10.14,18,19);
  89         f1.getArray().setIJ(0,0,VAL2);
  90         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
  91         #retrieving time steps...
  92         f2=MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),0,f1.getName(),8,9);
  93         f1.setTime(10.,8,9);
  94         f1.getArray().setIJ(0,0,VAL1);
  95         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
  96         f2=MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),0,f1.getName(),0,1);
  97         f3=MEDLoaderDataForTest.buildVecFieldOnCells_1();
  98         self.assertTrue(f3.isEqual(f2,1e-12,1e-12));
  99         f2=MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),0,f1.getName(),18,19);
 100         f1.setTime(10.14,18,19);
 101         f1.getArray().setIJ(0,0,VAL2);
 102         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 103         #ON NODES
 104         f1=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
 105         fileName2="Pyfile9.med";
 106         MEDLoader.WriteFieldDep(fileName2,f1,False);
 107         f1.setTime(110.,108,109);
 108         tmp=f1.getArray().getPointer();
 109         f1.getArray().setIJ(0,3,VAL1);
 110         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName2,f1);
 111         f1.setTime(210.,208,209);
 112         f1.getArray().setIJ(0,3,VAL2);
 113         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName2,f1);
 114         f2=MEDLoader.ReadFieldNode(fileName2,f1.getMesh().getName(),0,f1.getName(),108,109);
 115         f1.setTime(110.,108,109);
 116         f1.getArray().setIJ(0,3,VAL1);
 117         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 118         f2=MEDLoader.ReadFieldNode(fileName2,f1.getMesh().getName(),0,f1.getName(),2,3);
 119         f3=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
 120         self.assertTrue(f3.isEqual(f2,1e-12,1e-12));
 121         f2=MEDLoader.ReadFieldNode(fileName2,f1.getMesh().getName(),0,f1.getName(),208,209);
 122         f1.setTime(210.,208,209);
 123         f1.getArray().setIJ(0,3,VAL2);
 124         self.assertTrue(f1.isEqual(f2,1e-12,1e-12));
 125         pass
 126
 127     #
 128     # Multi field in a same file, but this field has several
 129     #
 130     def testFieldRW3(self):
 131         fileName="Pyfile11.med";
 132         VAL1=12345.67890314;
 133         VAL2=-1111111111111.;
 134         name1="AField";
 135         name3="AMesh1";
 136         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
 137         f1.getMesh().setName(name3);
 138         f1.setName(name1);
 139         f1.setTime(10.,8,9);
 140         tmp=f1.getArray().getPointer();
 141         f1.getArray().setIJ(0,0,VAL1);
 142         MEDLoader.WriteFieldDep(fileName,f1,False);
 143         f1.setTime(10.14,18,19);
 144         f1.getArray().setIJ(0,0,VAL2);
 145         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 146         f1.getMesh().setName(name3);
 147         f1.setTime(10.55,28,29);
 148         f1.getArray().setIJ(0,0,3*VAL1);
 149         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 150         f1.setTime(10.66,38,39);
 151         f1.getArray().setIJ(0,0,3*VAL2);
 152         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 153         f1.setTime(10.77,48,49);
 154         f1.getArray().setIJ(0,0,4*VAL2);
 155         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 156         #ON NODES
 157         f1=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
 158         f1.setName(name1);
 159         f1.getMesh().setName(name3);
 160         f1.setTime(110.,8,9);
 161         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 162         f1.setTime(110.,108,109);
 163         tmp=f1.getArray().getPointer();
 164         f1.getArray().setIJ(0,3,VAL1);
 165         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 166         f1.setTime(210.,208,209);
 167         f1.getArray().setIJ(0,3,VAL2);
 168         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 169         #
 170         it1=MEDLoader.GetCellFieldIterations(fileName,name3,name1);
 171         self.assertEqual(5,len(it1));
 172         self.assertEqual(8,it1[0][0]); self.assertEqual(9,it1[0][1]);
 173         self.assertEqual(18,it1[1][0]); self.assertEqual(19,it1[1][1]);
 174         self.assertEqual(28,it1[2][0]); self.assertEqual(29,it1[2][1]);
 175         self.assertEqual(38,it1[3][0]); self.assertEqual(39,it1[3][1]);
 176         self.assertEqual(48,it1[4][0]); self.assertEqual(49,it1[4][1]);
 177         it3=MEDLoader.GetNodeFieldIterations(fileName,name3,name1);
 178         self.assertEqual(3,len(it3));
 179         self.assertEqual(8,it3[0][0]); self.assertEqual(9,it3[0][1]);
 180         self.assertEqual(108,it3[1][0]); self.assertEqual(109,it3[1][1]);
 181         self.assertEqual(208,it3[2][0]); self.assertEqual(209,it3[2][1]);
 182         #
 183         #
 184         f1=MEDLoader.ReadFieldCell(fileName,name3,0,name1,8,9);
 185         self.assertAlmostEqual(VAL1,f1.getArray().getIJ(0,0),13);
 186         f1=MEDLoader.ReadFieldCell(fileName,name3,0,name1,18,19);
 187         self.assertAlmostEqual(VAL2,f1.getArray().getIJ(0,0),13);
 188         f1=MEDLoader.ReadFieldCell(fileName,name3,0,name1,28,29);
 189         self.assertAlmostEqual(3*VAL1,f1.getArray().getIJ(0,0),13);
 190         f1=MEDLoader.ReadFieldCell(fileName,name3,0,name1,38,39);
 191         self.assertAlmostEqual(3*VAL2,f1.getArray().getIJ(0,0),13);
 192         f1=MEDLoader.ReadFieldCell(fileName,name3,0,name1,48,49);
 193         self.assertAlmostEqual(4*VAL2,f1.getArray().getIJ(0,0),13);
 194         #
 195         f1=MEDLoader.ReadFieldNode(fileName,name3,0,name1,8,9);
 196         self.assertAlmostEqual(71.,f1.getArray().getIJ(0,3),13);
 197         f1=MEDLoader.ReadFieldNode(fileName,name3,0,name1,108,109);
 198         self.assertAlmostEqual(VAL1,f1.getArray().getIJ(0,3),13);
 199         f1=MEDLoader.ReadFieldNode(fileName,name3,0,name1,208,209);
 200         self.assertAlmostEqual(VAL2,f1.getArray().getIJ(0,3),13);
 201         pass
 202
 203     def testMultiMeshRW1(self):
 204         fileName="Pyfile10.med";
 205         mesh1=MEDLoaderDataForTest.build3DMesh_1();
 206         part1=[1,2,4,13,15]
 207         mesh2=mesh1.buildPartOfMySelf(part1,True);
 208         mesh2.setName("mesh2");
 209         part2=[3,4,13,14]
 210         mesh3=mesh1.buildPartOfMySelf(part2,True);
 211         mesh3.setName("mesh3");
 212         mesh4=MEDCouplingUMesh.New();
 213         mesh4.setName("mesh4");
 214         mesh4.setMeshDimension(3);
 215         mesh4.allocateCells(1);
 216         conn=[0,11,1,3]
 217         mesh4.insertNextCell(NORM_TETRA4,4,conn[0:4])
 218         mesh4.finishInsertingCells();
 219         mesh4.setCoords(mesh1.getCoords());
 220         meshes=[mesh1,mesh2,mesh3,mesh4]
 221         mnane="3DToto";
 222         MEDLoader.WriteUMeshesPartitionDep(fileName,mnane,meshes,False);
 223         #
 224         mesh5=MEDLoader.ReadUMeshFromFile(fileName,mnane);
 225         mesh1.setName(mnane);
 226         part3=[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17]
 227         mesh6=mesh5.buildPartOfMySelf(part3,True);
 228         mesh6.setName(mnane);
 229         self.assertTrue(mesh6.isEqual(mesh1,1e-12));
 230         grps=MEDLoader.GetMeshGroupsNames(fileName,mnane);
 231         self.assertEqual(4,len(grps));
 232         grps.index("mesh2");
 233         grps.index("mesh3");
 234         grps.index("mesh4");
 235         grps.index("3DMesh_1");
 236         #
 237         vec=["mesh2"];
 238         mesh2_2=MEDLoader.ReadUMeshFromGroups(fileName,mnane,0,vec);
 239         self.assertTrue(mesh2_2.isEqual(mesh2,1e-12));
 240         vec=["mesh3"];
 241         mesh3_2=MEDLoader.ReadUMeshFromGroups(fileName,mnane,0,vec);
 242         self.assertTrue(mesh3_2.isEqual(mesh3,1e-12));
 243         vec=["mesh4"];
 244         mesh4_2=MEDLoader.ReadUMeshFromGroups(fileName,mnane,0,vec);
 245         self.assertTrue(mesh4_2.isEqual(mesh4,1e-12));
 246         vec=["3DMesh_1"];
 247         mesh1_2=MEDLoader.ReadUMeshFromGroups(fileName,mnane,0,vec);
 248         mesh1.setName("3DMesh_1");
 249         self.assertTrue(mesh1_2.isEqual(mesh1,1e-12));
 250         #
 251         vec=["Family_4","Family_2"];
 252         mesh2_2=MEDLoader.ReadUMeshFromFamilies(fileName,mnane,0,vec);
 253         mesh2_2.setName("mesh2");
 254         self.assertTrue(mesh2_2.isEqual(mesh2,1e-12));
 255         pass
 256
 257     def testMesh3DSurfShuffleRW(self):
 258         fileName="Pyfile15.med";
 259         mesh=MEDLoaderDataForTest.build3DSurfMesh_1();
 260         renumber1=[2,5,1,0,3,4]
 261         mesh.renumberCells(renumber1,False);
 262         mesh.checkCoherency();
 263         MEDLoader.WriteUMeshDep(fileName,mesh,False);
 264         mesh_rw=MEDLoader.ReadUMeshFromFile(fileName,mesh.getName(),0);
 265         self.assertTrue(mesh.isEqual(mesh_rw,1e-12));
 266         pass
 267
 268     def testMultiFieldShuffleRW1(self):
 269         fileName="Pyfile17.med";
 270         m=MEDLoaderDataForTest.build3DMesh_2();
 271         self.assertEqual(20,m.getNumberOfCells());
 272         self.assertEqual(45,m.getNumberOfNodes());
 273         polys=[1,4,6]
 274         m.convertToPolyTypes(polys);
 275         renum=[1,3,2,8,9,12,13,16,19,0,4,7,5,15,14,17,10,18,6,11]
 276         m.renumberCells(renum,False);
 277         m.orientCorrectlyPolyhedrons();
 278         # Writing
 279         MEDLoader.WriteUMeshDep(fileName,m,False);
 280         f1Tmp=m.getMeasureField(False);
 281         f1=f1Tmp.buildNewTimeReprFromThis(ONE_TIME,False);
 282         f1.setTime(0.,1,2);
 283         f_1=f1.cloneWithMesh(True);
 284         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 285         f1.applyFunc("2*x");
 286         f1.setTime(0.01,3,4);
 287         f_2=f1.cloneWithMesh(True);
 288         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 289         f1.applyFunc("2*x/3");
 290         f1.setTime(0.02,5,6);
 291         f_3=f1.cloneWithMesh(True);
 292         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 293         # Reading
 294         its=[(1,2),(3,4),(5,6)];
 295         fs=MEDLoader.ReadFieldsOnSameMesh(ON_CELLS,fileName,f_1.getMesh().getName(),0,f_1.getName(),its);
 296         self.assertEqual(3,len(fs));
 297         self.assertTrue(fs[0].isEqual(f_1,1e-12,1e-12));
 298         self.assertTrue(fs[1].isEqual(f_2,1e-12,1e-12));
 299         self.assertTrue(fs[2].isEqual(f_3,1e-12,1e-12));
 300         pass
 301
 302     def testWriteUMeshesRW1(self):
 303         fileName="Pyfile18.med";
 304         m3d=MEDLoaderDataForTest.build3DMesh_2();
 305         pt=[0.,0.,-0.3]
 306         vec=[0.,0.,1.]
 307         nodes=m3d.findNodesOnPlane(pt,vec,1e-12);
 308         m2d=m3d.buildFacePartOfMySelfNode(nodes,True);
 309         renumber=[1,2,0,4,3]
 310         m2d.renumberCells(renumber,False);
 311         m2d.setName("ExampleOfMultiDimW");
 312         meshes=[m2d,m3d]
 313         MEDLoader.WriteUMeshes(fileName,meshes,False);
 314         m3d_bis=MEDLoader.ReadUMeshFromFile(fileName,m2d.getName(),0);
 315         self.assertTrue(not m3d_bis.isEqual(m3d,1e-12));
 316         m3d_bis.setName(m3d.getName());
 317         self.assertTrue(m3d_bis.isEqual(m3d,1e-12));
 318         m2d_bis=MEDLoader.ReadUMeshFromFile(fileName,m2d.getName(),-1);#-1 for faces
 319         self.assertTrue(m2d_bis.isEqual(m2d,1e-12));
 320         # Creation of a field on faces.
 321         f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME);
 322         f1.setName("FieldOnFacesShuffle");
 323         f1.setMesh(m2d);
 324         array=DataArrayDouble.New();
 325         arr1=[71.,171.,10.,110.,20.,120.,30.,130.,40.,140.]
 326         array.setValues(arr1,m2d.getNumberOfCells(),2);
 327         array.setInfoOnComponent(0,"plkj [mm]");
 328         array.setInfoOnComponent(1,"pqqqss [mm]");
 329         f1.setArray(array);
 330         tmp=array.setValues(arr1,m2d.getNumberOfCells(),2);
 331         f1.setTime(3.14,2,7);
 332         f1.checkCoherency();
 333         MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fileName,f1);
 334         f2=MEDLoader.ReadFieldCell(fileName,f1.getMesh().getName(),-1,f1.getName(),2,7);
 335         self.assertTrue(f2.isEqual(f1,1e-12,1e-12));
 336         pass
 337     pass
 338
 339 unittest.main()