src/MEDLoader/Swig/MEDLoaderTest3.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 from math import pi,e,sqrt
  25 from MEDLoaderDataForTest import MEDLoaderDataForTest
  26
  27 class MEDLoaderTest(unittest.TestCase):
  28     def testMEDMesh1(self):
  29         fileName="Pyfile18.med"
  30         mname="ExampleOfMultiDimW"
  31         medmesh=MEDFileMesh.New(fileName,mname)
  32         self.assertRaises(InterpKernelException,MEDFileMesh.New,fileName,"")
  33         self.assertEqual((0,-1),medmesh.getNonEmptyLevels())
  34         m1_0=medmesh.getLevel0Mesh(True)
  35         m1_1=MEDLoader.ReadUMeshFromFile(fileName,mname,0)
  36         self.assertTrue(m1_0.isEqual(m1_1,1e-12));
  37         m2_0=medmesh.getLevelM1Mesh(True)
  38         m2_1=MEDLoader.ReadUMeshFromFile(fileName,mname,-1)
  39         self.assertTrue(m2_0.isEqual(m2_1,1e-12));
  40         pass
  41     def testMEDMesh2(self):
  42         fileName="Pyfile10.med"
  43         mname="3DToto"
  44         outFileName="MEDFileMesh1.med"
  45         medmesh=MEDFileUMesh.New(fileName,mname)
  46         self.assertEqual((0,),medmesh.getNonEmptyLevels())
  47         m1_0=medmesh.getLevel0Mesh(True)
  48         m1_1=MEDLoader.ReadUMeshFromFile(fileName,mname,0)
  49         self.assertTrue(m1_0.isEqual(m1_1,1e-12));
  50         g1_0=medmesh.getGroup(0,"mesh2",True)
  51         g1_1=MEDLoader.ReadUMeshFromGroups(fileName,mname,0,["mesh2"]);
  52         self.assertTrue(g1_0.isEqual(g1_1,1e-12));
  53         g1_0=medmesh.getGroup(0,"mesh3",True)
  54         g1_1=MEDLoader.ReadUMeshFromGroups(fileName,mname,0,["mesh3"]);
  55         self.assertTrue(g1_0.isEqual(g1_1,1e-12));
  56         g1_0=medmesh.getGroups(0,["mesh3","mesh2"])
  57         g1_1=MEDLoader.ReadUMeshFromGroups(fileName,mname,0,["mesh3","mesh2"]);
  58         g1_1.setName(g1_0.getName())
  59         self.assertTrue(g1_0.isEqual(g1_1,1e-12));
  60         g1_0=medmesh.getFamily(0,"Family_2",True)
  61         g1_1=MEDLoader.ReadUMeshFromFamilies(fileName,mname,0,["Family_2"]);
  62         self.assertTrue(g1_0.isEqual(g1_1,1e-12));
  63         g1_0=medmesh.getFamilies(0,["Family_2","Family_4"],True)
  64         g1_1=MEDLoader.ReadUMeshFromFamilies(fileName,mname,0,["Family_2","Family_4"]);
  65         g1_1.setName(g1_0.getName())
  66         self.assertTrue(g1_0.isEqual(g1_1,1e-12));
  67         self.assertTrue(g1_0.isEqual(g1_1,1e-12));
  68         medmesh.write(outFileName,2);
  69         self.assertEqual([2,3,5,14,16],medmesh.getGroupArr(0,"mesh2",True).getValues());
  70         self.assertEqual([2,3,16],medmesh.getFamilyArr(0,"Family_2",True).getValues());
  71         self.assertEqual([2,3,5,14,16],medmesh.getFamiliesArr(0,["Family_4","Family_2"],True).getValues());
  72         self.assertEqual([19,2,3,4,5,14,15,16],medmesh.getGroupsArr(0,["mesh2","mesh4","mesh3"],True).getValues());
  73         famn=medmesh.getFamilyNameGivenId(0)
  74         self.assertRaises(InterpKernelException,medmesh.getNodeFamilyArr,famn,True);
  75         #without renum
  76         self.assertEqual([2,3,5,14,16],medmesh.getGroupArr(0,"mesh2",False).getValues());
  77         self.assertEqual([2,3,16],medmesh.getFamilyArr(0,"Family_2",False).getValues());
  78         self.assertEqual([2,3,5,14,16],medmesh.getFamiliesArr(0,["Family_4","Family_2"],False).getValues());
  79         self.assertEqual([0,2,3,4,5,14,15,16],medmesh.getGroupsArr(0,["mesh2","mesh3","mesh4"],False).getValues());
  80         self.assertRaises(InterpKernelException,medmesh.getNodeFamilyArr,famn,False);
  81         pass
  82
  83     # this tests emulates MEDMEM ( Except that it works ! ) The permutation are NOT taken into account
  84     def testMEDMesh3(self):
  85         outFileName="MEDFileMesh3.med"
  86         c=DataArrayDouble.New()
  87         coords=[-0.3,-0.3, 0.2,-0.3, 0.7,-0.3, -0.3,0.2, 0.2,0.2, 0.7,0.2, -0.3,0.7, 0.2,0.7, 0.7,0.7 ];
  88         targetConn=[0,3,4,1, 1,4,2, 4,5,2, 6,7,4,3, 7,8,5,4]
  89         c.setValues(coords,9,2)
  90         m=MEDCouplingUMesh.New();
  91         m.setMeshDimension(2);
  92         m.allocateCells(5);
  93         m.insertNextCell(NORM_TRI3,3,targetConn[4:7])
  94         m.insertNextCell(NORM_TRI3,3,targetConn[7:10])
  95         m.insertNextCell(NORM_QUAD4,4,targetConn[0:4])
  96         m.insertNextCell(NORM_POLYGON,4,targetConn[10:14])
  97         m.insertNextCell(NORM_POLYGON,4,targetConn[14:18])
  98         m.finishInsertingCells();
  99         m.setCoords(c)
 100         m.checkCoherency()
 101         m1=MEDCouplingUMesh.New();
 102         m1.setMeshDimension(1);
 103         m1.allocateCells(3);
 104         m1.insertNextCell(NORM_SEG2,2,[1,4])
 105         m1.insertNextCell(NORM_SEG2,2,[3,6])
 106         m1.insertNextCell(NORM_SEG3,3,[2,8,5])
 107         m1.finishInsertingCells();
 108         m1.setCoords(c)
 109         m1.checkCoherency()
 110         m2=MEDCouplingUMesh.New();
 111         m2.setMeshDimension(0);
 112         m2.allocateCells(4);
 113         m2.insertNextCell(NORM_POINT1,1,[1])
 114         m2.insertNextCell(NORM_POINT1,1,[3])
 115         m2.insertNextCell(NORM_POINT1,1,[2])
 116         m2.insertNextCell(NORM_POINT1,1,[6])
 117         m2.finishInsertingCells();
 118         m2.setCoords(c)
 119         m2.checkCoherency()
 120         #
 121         mm=MEDFileUMesh.New()
 122         mm.setName("MyFirstMEDCouplingMEDmesh")
 123         mm.setDescription("IHopeToConvinceLastMEDMEMUsers")
 124         mm.setCoords(c)
 125         mm.setMeshAtLevel(-1,m1);
 126         mm.setMeshAtLevel(0,m);
 127         mm.setMeshAtLevel(-2,m2);
 128         # playing with groups
 129         g1_2=DataArrayInt.New()
 130         g1_2.setValues([1,3],2,1)
 131         g1_2.setName("G1")
 132         g2_2=DataArrayInt.New()
 133         g2_2.setValues([1,2,3],3,1)
 134         g2_2.setName("G2")
 135         mm.setGroupsAtLevel(0,[g1_2,g2_2],False)
 136         g1_1=DataArrayInt.New()
 137         g1_1.setValues([0,1,2],3,1)
 138         g1_1.setName("G1")
 139         g2_1=DataArrayInt.New()
 140         g2_1.setValues([0,2],2,1)
 141         g2_1.setName("G2")
 142         mm.setGroupsAtLevel(-1,[g1_1,g2_1],False)
 143         g1_N=DataArrayInt.New()
 144         g1_N.setValues(range(8),8,1)
 145         g1_N.setName("G1")
 146         g2_N=DataArrayInt.New()
 147         g2_N.setValues(range(9),9,1)
 148         g2_N.setName("G2")
 149         mm.setGroupsAtLevel(1,[g1_N,g2_N],False)
 150         mm.createGroupOnAll(0,"GrpOnAllCell")
 151         # check content of mm
 152         t=mm.getGroupArr(0,"G1",False)
 153         self.assertTrue(g1_2.isEqual(t));
 154         t=mm.getGroupArr(0,"G2",False)
 155         self.assertTrue(g2_2.isEqual(t));
 156         t=mm.getGroupArr(-1,"G1",False)
 157         self.assertTrue(g1_1.isEqual(t));
 158         t=mm.getGroupArr(-1,"G2",False)
 159         self.assertTrue(g2_1.isEqual(t));
 160         t=mm.getGroupArr(1,"G1",False)
 161         self.assertTrue(g1_N.isEqual(t));
 162         t=mm.getGroupArr(1,"G2",False)
 163         self.assertTrue(g2_N.isEqual(t));
 164         self.assertTrue(mm.existsGroup("GrpOnAllCell"));
 165         t=mm.getGroupArr(0,"GrpOnAllCell")
 166         self.assertTrue(t.getValues()==range(5))
 167         #
 168         mmCpy=mm.deepCpy()
 169         self.assertTrue(mm.isEqual(mmCpy,1e-12)[0]) ; del mm
 170         mmCpy.write(outFileName,2);
 171         #
 172         mm=MEDFileMesh.New(outFileName)
 173         mbis=mm.getMeshAtLevel(0)
 174         m.setName(mm.getName()) ; m.setDescription(mm.getDescription())
 175         self.assertTrue(m.isEqual(mbis,1e-12));
 176         #
 177         self.assertEqual(([[(3, 2), (4, 1), (5, 8)], [(1, 2), (2, 1)], [(0, 4)]], 2, 2, 9),MEDLoader.GetUMeshGlobalInfo(outFileName,"MyFirstMEDCouplingMEDmesh"))
 178         pass
 179
 180     # this test is the testMEDMesh3 except that permutation is dealed here
 181     def testMEDMesh4(self):
 182         outFileName="MEDFileMesh4.med"
 183         c=DataArrayDouble.New()
 184         coords=[-0.3,-0.3, 0.2,-0.3, 0.7,-0.3, -0.3,0.2, 0.2,0.2, 0.7,0.2, -0.3,0.7, 0.2,0.7, 0.7,0.7 ];
 185         targetConn=[0,3,4,1, 1,4,2, 4,5,2, 6,7,4,3, 7,8,5,4]
 186         c.setValues(coords,9,2)
 187         c.setInfoOnComponent(0,"abcdef [km]")
 188         c.setInfoOnComponent(1,"ghij [MW]")
 189         m=MEDCouplingUMesh.New();
 190         m.setMeshDimension(2);
 191         m.allocateCells(5);
 192         m.insertNextCell(NORM_QUAD4,4,targetConn[0:4])
 193         m.insertNextCell(NORM_TRI3,3,targetConn[4:7])
 194         m.insertNextCell(NORM_TRI3,3,targetConn[7:10])
 195         m.insertNextCell(NORM_QUAD4,4,targetConn[10:14])
 196         m.insertNextCell(NORM_QUAD4,4,targetConn[14:18])
 197         m.finishInsertingCells();
 198         m.setCoords(c)
 199         m.checkCoherency()
 200         m1=MEDCouplingUMesh.New();
 201         m1.setMeshDimension(1);
 202         m1.allocateCells(3);
 203         m1.insertNextCell(NORM_SEG2,2,[1,4])
 204         m1.insertNextCell(NORM_SEG3,3,[2,8,5])
 205         m1.insertNextCell(NORM_SEG2,2,[3,6])
 206         m1.finishInsertingCells();
 207         m1.setCoords(c)
 208         m1.checkCoherency()
 209         m2=MEDCouplingUMesh.New();
 210         m2.setMeshDimension(0);
 211         m2.allocateCells(4);
 212         m2.insertNextCell(NORM_POINT1,1,[1])
 213         m2.insertNextCell(NORM_POINT1,1,[3])
 214         m2.insertNextCell(NORM_POINT1,1,[2])
 215         m2.insertNextCell(NORM_POINT1,1,[6])
 216         m2.finishInsertingCells();
 217         m2.setCoords(c)
 218         m2.checkCoherency()
 219         #
 220         mm=MEDFileUMesh.New()
 221         mm.setName("My2ndMEDCouplingMEDmesh")
 222         mm.setDescription("ThisIsImpossibleToDoWithMEDMEM")
 223         mm.setCoords(c)
 224         renumNode=DataArrayInt.New()
 225         renumNode.setValues([10,11,12,13,14,15,16,17,18],9,1)
 226         mm.setRenumFieldArr(1,renumNode)
 227         mm.setMeshAtLevel(-1,m1,True);
 228         mm.setMeshAtLevel(0,m,True);
 229         mm.setMeshAtLevel(-2,m2,True);
 230         mm.removeMeshAtLevel(-2)
 231         mm.setMeshAtLevel(-2,m2,True);
 232         # playing with groups
 233         g1_2=DataArrayInt.New()
 234         g1_2.setValues([2,3],2,1)
 235         g1_2.setName("G1")
 236         g2_2=DataArrayInt.New()
 237         g2_2.setValues([2,0,3],3,1)
 238         g2_2.setName("G2")
 239         mm.setGroupsAtLevel(0,[g1_2,g2_2],True)
 240         g1_1=DataArrayInt.New()
 241         g1_1.setValues([0,2,1],3,1)
 242         g1_1.setName("G1")
 243         g2_1=DataArrayInt.New()
 244         g2_1.setValues([0,2],2,1)
 245         g2_1.setName("G2")
 246         mm.setGroupsAtLevel(-1,[g1_1,g2_1],True)
 247         g1_N=DataArrayInt.New()
 248         g1_N.setValues([10,11,12,13,14,15,16,17],8,1)
 249         g1_N.setName("G1")
 250         g2_N=DataArrayInt.New()
 251         g2_N.setValues([10,11,12,13,14,15,16,17,18],9,1)
 252         g2_N.setName("G2")
 253         mm.setGroupsAtLevel(1,[g1_N,g2_N],True)
 254         # check content of mm
 255         t=mm.getGroupArr(0,"G1",True)
 256         self.assertTrue(g1_2.isEqual(t));
 257         t=mm.getGroupArr(0,"G2",True)
 258         self.assertTrue(g2_2.isEqual(t));
 259         t=mm.getGroupArr(-1,"G1",True)
 260         self.assertTrue(g1_1.isEqual(t));
 261         t=mm.getGroupArr(-1,"G2",True)
 262         self.assertTrue(g2_1.isEqual(t));
 263         self.assertTrue(not mm.existsGroup("GrpOnAllCell"));
 264         #
 265         mm.write(outFileName,2);
 266         mm2=MEDFileMesh.New(outFileName)
 267         res=mm.isEqual(mm2,1e-12)
 268         self.assertTrue(res[0])
 269         l=list(mm2.getFamiliesOnGroup("G2")) ; l.sort()
 270         self.assertEqual(['Family_10','Family_11','Family_3','Family_4','Family_7'],l)
 271         mm2.keepFamIdsOnlyOnLevs([3],[-1])
 272         for lev in mm.getGrpNonEmptyLevelsExt("G2"):
 273             self.assertEqual(mm.getGroupArr(lev,"G2").getValues(),mm2.getGroupArr(lev,"G2").getValues())
 274             pass
 275         l=list(mm2.getFamiliesOnGroup("G2")) ; l.sort()
 276         self.assertEqual(['Family_10','Family_11','Family_12','Family_3','Family_4','Family_7'],l)
 277         #
 278         self.assertEqual([7,7,6],mm2.getFamilyFieldAtLevel(-1).getValues())
 279         mm2.getFamilyFieldAtLevel(-1).setIJ(1,0,8)
 280         self.assertEqual([7,8,6],mm2.getFamilyFieldAtLevel(-1).getValues())
 281         self.assertTrue(not mm2.existsFamily("Family_8"))
 282         mm2.createGroupOnAll(-1,"GrpOnAllFace")
 283         self.assertTrue(mm2.existsFamily("Family_8"))
 284         self.assertEqual(range(3),mm2.getGroupArr(-1,"GrpOnAllFace").getValues())
 285         pass
 286
 287     #testing persistence of retrieved arrays
 288     def testMEDMesh5(self):
 289         fileName="Pyfile18.med"
 290         mname="ExampleOfMultiDimW"
 291         medmesh=MEDFileUMesh.New(fileName,mname)
 292         m1_0=medmesh.getLevel0Mesh(True)
 293         da1=medmesh.getFamilyFieldAtLevel(0)
 294         del medmesh
 295         self.assertEqual(20,m1_0.getNumberOfCells())
 296         self.assertEqual(20,da1.getNumberOfTuples())
 297         pass
 298
 299     def testMEDMesh6(self):
 300         outFileName="MEDFileMesh5.med"
 301         m=MEDFileCMesh.New()
 302         m.setTime(-1,-1,2.3)
 303         m1=MEDCouplingCMesh.New();
 304         da=DataArrayDouble.New()
 305         da.setValues([0.,1.,2.],3,1)
 306         da.setInfoOnComponent(0,"XX [mm]")
 307         m1.setCoordsAt(0,da)
 308         da=DataArrayDouble.New()
 309         da.setValues([0.,1.2],2,1)
 310         da.setInfoOnComponent(0,"YY [km]")
 311         m1.setCoordsAt(1,da)
 312         da=DataArrayDouble.New()
 313         da.setValues([0.,1.3],2,1)
 314         da.setInfoOnComponent(0,"ZZ [um]")
 315         m1.setCoordsAt(2,da)
 316         m.setMesh(m1)
 317         m.setName("myFirstCartMesh")
 318         m.setDescription("mmmmpppppppp")
 319         m.setTimeValue(2.3)
 320         m.setTimeUnit("ms")
 321         da=DataArrayInt.New()
 322         da.setValues([0,0,1,0,1,2,4,3,0,1,2,2],12,1)
 323         m.setFamilyFieldArr(1,da)
 324         m.setFamilyId("family1",1)
 325         da=m.getFamilyArr(1,"family1")
 326         expected1=[2,4,9]
 327         self.assertEqual(expected1,da.getValues())
 328         m.write(outFileName,2);
 329         mm=MEDFileMesh.New(outFileName)
 330         self.assertTrue(m.isEqual(mm,1e-12)[0])
 331         self.assertEqual(expected1,mm.getFamilyArr(1,"family1").getValues())
 332         m2=mm.getMesh()
 333         tt=m.getTime()
 334         m1.setTime(tt[2],tt[0],tt[1])
 335         m1.setName(m.getName())
 336         m1.setTimeUnit(m.getTimeUnit())
 337         m1.setDescription(m.getDescription())
 338         self.assertTrue(m2.isEqual(m1,1e-12));
 339         pass
 340
 341     def testMEDMesh7(self):
 342         fileName="Pyfile24.med"
 343         m2,m1,m0,f2,f1,f0,p,n2,n1,n0,fns,fids,grpns,famIdsPerGrp=MEDLoaderDataForTest.buildMultiLevelMesh_1()
 344         m=MEDFileUMesh.New()
 345         m.setCoords(m2.getCoords())
 346         m.setMeshAtLevel(0,m2)
 347         m.setMeshAtLevel(-1,m1)
 348         m.setMeshAtLevel(-2,m0)
 349         m.setFamilyFieldArr(0,f2)
 350         m.setFamilyFieldArr(-1,f1)
 351         m.setFamilyFieldArr(-2,f0)
 352         m.setFamilyFieldArr(1,p)
 353         m.setRenumFieldArr(0,n2)
 354         m.setRenumFieldArr(-1,n1)
 355         m.setRenumFieldArr(-2,n0)
 356         nbOfFams=len(fns)
 357         for i in xrange(nbOfFams):
 358             m.addFamily(fns[i],fids[i])
 359             pass
 360         nbOfGrps=len(grpns)
 361         for i in xrange(nbOfGrps):
 362             m.setFamiliesIdsOnGroup(grpns[i],famIdsPerGrp[i])
 363             pass
 364         m.setName(m2.getName())
 365         m.setDescription(m2.getDescription())
 366         #
 367         self.assertEqual((-1,),m.getGrpNonEmptyLevels("A2A4"))
 368         self.assertEqual((),m.getGrpNonEmptyLevels("A1"))
 369         self.assertEqual((-2,),m.getGrpNonEmptyLevels("AP2"))
 370         self.assertEqual((-1,-2),m.getGrpsNonEmptyLevels(["A2A4","AP2"]))
 371         self.assertEqual((-1,),m.getFamNonEmptyLevels('A4A3____________________________'))
 372         self.assertEqual((0,),m.getFamNonEmptyLevels('MESH____DALT3___DALLE___________'))
 373         self.assertEqual((0,-1,),m.getFamsNonEmptyLevels(['MESH____DALT3___DALLE___________','A4A3____________________________']))
 374         self.assertEqual(('A1A2','A2A4','A3A1','A3C5','A4A3','B1C1','B2B4','B3B1','B4C3','C1C4','C2B2','C3C2','C4B3','C5A4'),m.getGroupsOnSpecifiedLev(-1))
 375         self.assertEqual(('DALLE','DALQ1','DALQ2','DALT3','MESH'),m.getGroupsOnSpecifiedLev(0))
 376         #
 377         m.write(fileName,2)
 378         self.assertRaises(InterpKernelException,MEDFileField1TS,fileName)#throw because no field in file fileName
 379         pass
 380
 381     def funcToTestDelItem(self,ff):
 382         del ff[[0.02,(3,4)]]
 383         pass
 384
 385     #emulation of pointe.med file.
 386     def testMEDField1(self):
 387         mm=MEDFileMesh.New("Pyfile17.med")
 388         mm.write("Pyfile17_bis.med",2)
 389         ff=MEDFileFieldMultiTS("Pyfile17.med")
 390         tsExpected=[[1,2],[3,4],[5,6]]
 391         self.assertEqual(3,len(ff))
 392         for pos,f1ts in enumerate(ff):
 393             self.assertEqual(tsExpected[pos],f1ts.getTime()[:2])
 394             self.assertEqual(type(f1ts),MEDFileField1TS)
 395             pass
 396         self.assertEqual("MeasureOfMesh_Extruded",ff.getName())
 397         self.assertEqual([3,4],ff[1].getTime()[:-1])
 398         self.assertEqual([3,4],ff[3,4].getTime()[:-1])
 399         self.assertEqual([3,4],ff[0.01].getTime()[:-1])
 400         ff.write("Pyfile17_bis.med",0)
 401         #
 402         ts=ff.getTimeSteps() ; ts=[elt[:-1] for elt in ts]
 403         self.assertEqual([(1,2),(3,4),(5,6)],ts)
 404         self.funcToTestDelItem(ff)
 405         ts=ff.getTimeSteps() ; ts=[elt[:-1] for elt in ts]
 406         self.assertEqual([(1,2)],ts)
 407         pass
 408
 409     #profiles
 410     def testMEDField2(self):
 411         mm=MEDFileMesh.New("Pyfile19.med")
 412         mm.write("Pyfile19_bis.med",2)
 413         ff=MEDFileFieldMultiTS.New("Pyfile19.med","VFieldOnNodes")
 414         ff.write("Pyfile19_bis.med",0)
 415         self.assertEqual([('tyty','mm'),('uiop','MW')],MEDLoader.GetComponentsNamesOfField("Pyfile19_bis.med","VFieldOnNodes"))
 416         pass
 417
 418     #gauss points
 419     def testMEDField3(self):
 420         mm=MEDFileMesh.New("Pyfile13.med")
 421         mm.write("Pyfile13_bis.med",2)
 422         ff=MEDFileFieldMultiTS.New("Pyfile13.med","MyFirstFieldOnGaussPoint")
 423         ff.write("Pyfile13_bis.med",0)
 424         ff=MEDFileField1TS.New("Pyfile13.med","MyFirstFieldOnGaussPoint",1,5)
 425         f=ff.getFieldAtLevel(ON_GAUSS_PT,0)
 426         f2=MEDLoader.ReadFieldGauss("Pyfile13.med",'2DMesh_2',0,'MyFirstFieldOnGaussPoint',1,5)
 427         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 428         ff3=MEDFileField1TS.New("Pyfile13.med","MyFirstFieldOnGaussPoint")
 429         f3=ff3.getFieldAtLevel(ON_GAUSS_PT,0)
 430         self.assertTrue(f.isEqual(f3,1e-12,1e-12))
 431         ff4=MEDFileField1TS.New("Pyfile13.med")
 432         f4=ff4.getFieldAtLevel(ON_GAUSS_PT,0)
 433         self.assertTrue(f.isEqual(f4,1e-12,1e-12))
 434         pass
 435
 436     #gauss NE
 437     def testMEDField4(self):
 438         mm=MEDFileMesh.New("Pyfile14.med")
 439         mm.write("Pyfile14_bis.med",2)
 440         ff=MEDFileFieldMultiTS.New("Pyfile14.med","MyFieldOnGaussNE")
 441         ff.write("Pyfile14_bis.med",0)
 442         ff=MEDFileField1TS.New("Pyfile14.med","MyFieldOnGaussNE",1,5)
 443         f=ff.getFieldAtLevel(ON_GAUSS_NE,0)
 444         f2=MEDLoader.ReadFieldGaussNE("Pyfile14.med",'2DMesh_2',0,"MyFieldOnGaussNE",1,5)
 445         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 446         pass
 447
 448     # MEDField get/set on pointe.med
 449     def testMEDField5(self):
 450         ff=MEDFileField1TS.New("Pyfile17.med","MeasureOfMesh_Extruded",1,2)
 451         f=ff.getFieldAtLevel(ON_CELLS,0)
 452         f2=MEDLoader.ReadFieldCell("Pyfile17.med","Extruded",0,"MeasureOfMesh_Extruded",1,2)
 453         self.assertTrue(f.getMesh().getCoords().isEqual(f2.getMesh().getCoords(),1e-12))
 454         f.getMesh().tryToShareSameCoords(f2.getMesh(),1e-12)
 455         f.changeUnderlyingMesh(f2.getMesh(),22,1e-12)
 456         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 457         # no with renumbering
 458         f=ff.getFieldAtLevel(ON_CELLS,0,1)
 459         f2=MEDLoader.ReadFieldCell("Pyfile17.med","Extruded",0,"MeasureOfMesh_Extruded",1,2)
 460         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 461         f=ff.getFieldAtLevel(ON_CELLS,0,3)
 462         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 463         f=ff.getFieldAtLevel(ON_CELLS,0,2)
 464         self.assertTrue(not f.isEqual(f2,1e-12,1e-12))
 465         f.changeUnderlyingMesh(f2.getMesh(),12,1e-12)
 466         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 467         pass
 468
 469     # MEDField get/set on profiles nodes
 470     def testMEDField6(self):
 471         ff=MEDFileFieldMultiTS.New("Pyfile7.med","VectorFieldOnNodes")
 472         its=ff.getIterations()
 473         self.assertRaises(InterpKernelException,ff.getFieldAtLevel,ON_CELLS,its[0][0],its[0][1],0)# request on cell and it is not on cells
 474         f=ff.getFieldAtLevel(ON_NODES,its[0][0],its[0][1],0)
 475         f2=MEDLoader.ReadFieldNode("Pyfile7.med",'3DSurfMesh_1',0,"VectorFieldOnNodes",its[0][0],its[0][1])
 476         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 477         ff=MEDFileFieldMultiTS.New("Pyfile19.med","VFieldOnNodes")
 478         its=ff.getIterations()
 479         f=ff.getFieldAtLevel(ON_NODES,its[0][0],its[0][1],0)
 480         f2=MEDLoader.ReadFieldNode("Pyfile19.med",'2DMesh_1',0,"VFieldOnNodes",its[0][0],its[0][1])
 481         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 482         self.assertRaises(InterpKernelException,ff.getFieldAtLevel,ON_CELLS,its[0][0],its[0][1],0)# request on cell and it is not on cells
 483         self.assertRaises(InterpKernelException,ff.getFieldAtLevel,ON_NODES,its[0][0],its[0][1],0,1)#request renumber following mesh : it is on profile !
 484         pass
 485
 486     # MEDField get/set on profiles cells
 487     def testMEDField7(self):
 488         ff=MEDFileFieldMultiTS.New("Pyfile12.med","VectorFieldOnCells")
 489         its=ff.getIterations()
 490         f=ff.getFieldAtLevel(ON_CELLS,its[0][0],its[0][1],0)
 491         f2=MEDLoader.ReadFieldCell("Pyfile12.med",'3DMesh_1',0,"VectorFieldOnCells",its[0][0],its[0][1])
 492         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
 493         pass
 494
 495     #first test of assignation. No profile and types sorted by type.
 496     def testMEDField8(self):
 497         fname="Pyfile25.med"
 498         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
 499         m1=f1.getMesh()
 500         mm1=MEDFileUMesh.New()
 501         mm1.setCoords(m1.getCoords())
 502         mm1.setMeshAtLevel(0,m1)
 503         mm1.setName(m1.getName())
 504         mm1.write(fname,2)
 505         ff1=MEDFileField1TS.New()
 506         ff1.setFieldNoProfileSBT(f1)
 507         ff1.write(fname,0)
 508         f2=MEDLoader.ReadFieldCell(fname,f1.getMesh().getName(),0,f1.getName(),f1.getTime()[1],f1.getTime()[2]);
 509         itt,orr,ti=ff1.getTime()
 510         self.assertEqual(0,itt); self.assertEqual(1,orr); self.assertAlmostEqual(2.,ti,14);
 511         self.assertTrue(f1.isEqual(f2,1e-12,1e-12))
 512         ff1.setTime(3,4,2.3)
 513         itt,orr,ti=ff1.getTime()
 514         self.assertEqual(3,itt); self.assertEqual(4,orr); self.assertAlmostEqual(2.3,ti,14);
 515         da,infos=ff1.getUndergroundDataArrayExt()
 516         f2.getArray().setName(da.getName())#da has the same name than f2
 517         self.assertTrue(da.isEqual(f2.getArray(),1e-12))
 518         self.assertEqual([((3, 0), (0, 2)), ((4, 0), (2, 4)), ((6, 0), (4, 5)), ((5, 0), (5, 6))],infos)
 519         #
 520         fname="Pyfile26.med"
 521         f1=MEDLoaderDataForTest.buildVecFieldOnNodes_1();
 522         m1=f1.getMesh()
 523         mm1=MEDFileUMesh.New()
 524         mm1.setCoords(m1.getCoords())
 525         mm1.setMeshAtLevel(0,m1)
 526         mm1.setName(m1.getName())
 527         mm1.write(fname,2)
 528         ff1=MEDFileField1TS.New()
 529         ff1.setFieldNoProfileSBT(f1)
 530         nv=1456.
 531         da=ff1.getUndergroundDataArray().setIJ(0,0,nv)
 532         ff1.write(fname,0)
 533         f2=MEDLoader.ReadFieldNode(fname,f1.getMesh().getName(),0,f1.getName(),f1.getTime()[1],f1.getTime()[2])
 534         self.assertTrue(not f1.isEqual(f2,1e-12,1e-12))
 535         f1.getArray().setIJ(0,0,nv)
 536         self.assertTrue(f1.isEqual(f2,1e-12,1e-12))
 537         #
 538         fname="Pyfile27.med"
 539         f1=MEDLoaderDataForTest.buildVecFieldOnGaussNE_1();
 540         m1=f1.getMesh()
 541         mm1=MEDFileUMesh.New()
 542         mm1.setCoords(m1.getCoords())
 543         mm1.setMeshAtLevel(0,m1)
 544         mm1.setName(m1.getName())
 545         mm1.write(fname,2)
 546         ff1=MEDFileField1TS.New()
 547         ff1.setFieldNoProfileSBT(f1)
 548         ff1.write(fname,0)
 549         f2=MEDLoader.ReadFieldGaussNE(fname,f1.getMesh().getName(),0,f1.getName(),f1.getTime()[1],f1.getTime()[2])
 550         self.assertTrue(f1.isEqual(f2,1e-12,1e-12))
 551         da,infos=ff1.getUndergroundDataArrayExt()
 552         f2.getArray().setName(da.getName())#da has the same name than f2
 553         self.assertTrue(da.isEqual(f2.getArray(),1e-12))
 554         self.assertEqual([((3, 0), (0, 6)), ((4, 0), (6, 14)), ((6, 0), (14, 20))],infos)
 555         #
 556         fname="Pyfile28.med"
 557         f1=MEDLoaderDataForTest.buildVecFieldOnGauss_2_Simpler();
 558         m1=f1.getMesh()
 559         mm1=MEDFileUMesh.New()
 560         mm1.setCoords(m1.getCoords())
 561         mm1.setMeshAtLevel(0,m1)
 562         mm1.setName(m1.getName())
 563         mm1.write(fname,2)
 564         ff1=MEDFileField1TS.New()
 565         ff1.setFieldNoProfileSBT(f1)
 566         ff1.write(fname,0)
 567         ff2=MEDFileField1TS.New(fname,f1.getName(),f1.getTime()[1],f1.getTime()[2])
 568         f2=ff2.getFieldAtLevel(ON_GAUSS_PT,0)
 569         self.assertTrue(f1.isEqual(f2,1e-12,1e-12))
 570         sbt=ff2.getFieldSplitedByType2()
 571         loc1=ff2.getLocalization("Loc_MyFirstFieldOnGaussPoint_NORM_TRI6_5")
 572         self.assertEqual("Loc_MyFirstFieldOnGaussPoint_NORM_TRI6_5",loc1.getName())
 573         self.assertEqual((-1, 1,-1,-1,1,-1,-1,0,0,-1,0,0),loc1.getRefCoords())
 574         self.assertEqual(6,loc1.getNumberOfPointsInCells())
 575         self.assertEqual(3,loc1.getNumberOfGaussPoints())
 576         self.assertEqual(2,loc1.getDimension())
 577         da,infos=ff2.getUndergroundDataArrayExt()
 578         f2.getArray().setName(da.getName())#da has the same name than f2
 579         self.assertTrue(da.isEqual(f2.getArray(),1e-12))
 580         self.assertEqual(53,da.getNumberOfTuples())
 581         self.assertEqual([((3, 0), (0, 18)), ((3, 1), (18, 30)), ((3, 2), (30, 36)), ((4, 0), (36, 42)), ((4, 1), (42, 44)), ((6, 0), (44, 53))],infos)
 582         #
 583         pass
 584
 585     def testMEDFileData1(self):
 586         fname="Pyfile29.med"
 587         d=MEDFileData.New()
 588         #
 589         m1=MEDLoaderDataForTest.build1DMesh_1()
 590         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ; mm1.setName(m1.getName())
 591         mmm1=MEDFileMeshMultiTS.New() ;
 592         mmm1.setOneTimeStep(mm1)
 593         m2=MEDLoaderDataForTest.build2DCurveMesh_1()
 594         mm2=MEDFileUMesh.New() ; mm2.setCoords(m2.getCoords()) ; mm2.setMeshAtLevel(0,m2) ; mm2.setName(m2.getName())
 595         mmm2=MEDFileMeshMultiTS.New() ; mmm2.setOneTimeStep(mm2)
 596         ms=MEDFileMeshes.New(); ms.setMeshAtPos(0,mm1) ; ms.setMeshAtPos(1,mm2)
 597         d.setMeshes(ms)
 598         for name,mmm in zip(["1DMesh_1","2DCurveMesh_1"],ms):
 599             self.assertEqual(name,mmm.getName())
 600             self.assertEqual(type(mmm),MEDFileUMesh)
 601             pass
 602         self.assertEqual(('1DMesh_1', '2DCurveMesh_1'),d.getMeshes().getMeshesNames())
 603         #
 604         ff1=MEDFileFieldMultiTS.New()
 605         ff21=MEDFileFieldMultiTS.New()
 606         ff22=MEDFileFieldMultiTS.New()
 607         f1=m1.getMeasureField(True) ; f1.setName("f1") ; f1=f1.buildNewTimeReprFromThis(ONE_TIME,False)
 608         f1.getArray().setInfoOnComponent(0,"power [kW]")
 609         ff1.appendFieldNoProfileSBT(f1)
 610         f21=m2.getMeasureField(True) ; f21.setName("f21") ; f21=f21.buildNewTimeReprFromThis(ONE_TIME,False)
 611         f21.getArray().setInfoOnComponent(0,"sta [mm]") ;
 612         ff21.appendFieldNoProfileSBT(f21)
 613         f22=f21.deepCpy() ; f22.setName("f22") ; f22=f22.buildNewTimeReprFromThis(ONE_TIME,False) ;
 614         f22.applyFunc(2,"3*x*IVec+2*x*JVec")
 615         f22.getArray().setInfoOnComponent(0,"distance [km]") ; f22.getArray().setInfoOnComponent(1,"displacement [cm]")
 616         ff22.appendFieldNoProfileSBT(f22)
 617         fs=MEDFileFields.New()
 618         fs.pushField(ff1) ; fs.pushField(ff21) ; fs.pushField(ff22)
 619         for name,fmts in zip(["f1","f21","f22"],fs):
 620             self.assertEqual(name,fmts.getName())
 621             pass
 622         d.setFields(fs)
 623         #
 624         fname2="Pyfile29_2.med"
 625         d.write(fname2,2)
 626         #
 627         d2=MEDFileData.New(fname2)
 628         self.assertEqual(2,d2.getNumberOfMeshes())
 629         self.assertEqual(3,d2.getNumberOfFields())
 630         self.assertTrue(isinstance(d2.getMeshes().getMeshAtPos(0),MEDFileUMesh))
 631         self.assertTrue(isinstance(d2.getMeshes()[0],MEDFileUMesh))
 632         self.assertTrue(isinstance(d2.getMeshes()['2DCurveMesh_1'],MEDFileUMesh))
 633         m1bis=d2.getMeshes().getMeshAtPos(0).getMeshAtLevel(0)
 634         self.assertTrue(m1.isEqual(m1bis,1e-12))
 635         self.assertEqual(('f1', 'f21', 'f22'),d2.getFields().getFieldsNames())
 636         self.assertEqual([(-1,-1,0.0)],d2.getFields().getFieldAtPos(2).getTimeSteps())
 637         self.assertEqual([(-1,-1,0.0)],d2.getFields()[2].getTimeSteps())
 638         self.assertEqual([(-1,-1,0.0)],d2.getFields().getFieldWithName("f21").getTimeSteps())
 639         self.assertEqual([(-1,-1,0.0)],d2.getFields()["f21"].getTimeSteps())
 640         pass
 641
 642     def testMEDField9(self):
 643         # first test field profile WR. Full type but with some type missing
 644         fname="Pyfile30.med"
 645         m1=MEDLoaderDataForTest.build2DMesh_3()
 646         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ;
 647         mm1.write(fname,2)
 648         ff1=MEDFileField1TS.New()
 649         f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f1.setName("F1")
 650         d=DataArrayDouble.New() ; d.alloc(2*9,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 651         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 652         da=DataArrayInt.New(); da.alloc(9,1) ; da.iota(0) ; da.setName("sup1")
 653         #
 654         ff1.setFieldProfile(f1,mm1,0,da)
 655         ff1.changePflsNames([(["sup1_NORM_QUAD4"],"ForV650")])
 656         ff1=ff1.deepCpy()
 657         ff1.write(fname,0)
 658         #
 659         vals,pfl=ff1.getFieldWithProfile(ON_CELLS,0,mm1) ; vals.setName("")
 660         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))# profiles names cannot be contracted in pfl array name
 661         self.assertTrue(vals.isEqual(d,1e-14))
 662         #
 663         ff2=MEDFileField1TS.New(fname,f1.getName(),-1,-1)
 664         ff3=MEDFileField1TS.New(fname,f1.getName(),-1,-1)
 665         ff2.deepCpyGlobs(ff3)
 666         sbt=ff2.getFieldSplitedByType2()
 667         self.assertEqual(3,sbt[0][0])#TRI3
 668         self.assertEqual(0,sbt[0][1][0][0])#CELL For TRI3
 669         self.assertEqual("",sbt[0][1][0][2])#no profile For TRI3
 670         self.assertEqual([7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18],sbt[0][1][0][1].getValues())# values for TRI3
 671         self.assertEqual(4,sbt[1][0])#QUAD4
 672         self.assertEqual(0,sbt[1][1][0][0])#CELL For QUAD4
 673         self.assertEqual("ForV650",sbt[1][1][0][2])# profile For QUAD4
 674         self.assertEqual([19, 20, 21, 22, 23, 24],sbt[1][1][0][1].getValues())# values for QUAD4
 675         self.assertEqual([0],ff2.getTypesOfFieldAvailable())
 676         vals,pfl=ff2.getFieldWithProfile(ON_CELLS,0,mm1) ; vals.setName("")
 677         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 678         self.assertTrue(vals.isEqual(d,1e-14))
 679         pass
 680
 681     def testMEDField10(self):
 682         fname="Pyfile31.med"
 683         m1=MEDLoaderDataForTest.build2DMesh_1()
 684         m1.renumberCells([0,1,4,2,3,5],False)
 685         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ; mm1.setName(m1.getName())
 686         mm1.write(fname,2)
 687         ff1=MEDFileFieldMultiTS.New()
 688         f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f1.setName("F2")
 689         d=DataArrayDouble.New() ; d.alloc(2*4,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 690         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 691         da=DataArrayInt.New(); da.setValues([0,1,2,4],4,1) ; da.setName("sup2")
 692         #
 693         ff1.appendFieldProfile(f1,mm1,0,da)
 694         f1.setTime(1.2,1,2) ; e=d.applyFunc("2*x") ; e.copyStringInfoFrom(d) ; f1.setArray(e) ;
 695         ff1.appendFieldProfile(f1,mm1,0,da)
 696         ff1=ff1.deepCpy()
 697         ff1.write(fname,0)
 698         #
 699         vals,pfl=ff1.getFieldWithProfile(ON_CELLS,1,2,0,mm1) ; vals.setName("")
 700         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 701         self.assertTrue(vals.isEqual(e,1e-14))
 702         vals,pfl=ff1.getFieldWithProfile(ON_CELLS,-1,-1,0,mm1) ; vals.setName("")
 703         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 704         self.assertTrue(vals.isEqual(d,1e-14))
 705         #
 706         ff2=MEDFileFieldMultiTS.New(fname,f1.getName())
 707         self.assertEqual([(-1,-1,0.0), (1,2,1.2)],ff2.getTimeSteps())
 708         vals,pfl=ff2.getFieldWithProfile(ON_CELLS,1,2,0,mm1) ; vals.setName("")
 709         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 710         self.assertTrue(vals.isEqual(e,1e-14))
 711         vals,pfl=ff2.getFieldWithProfile(ON_CELLS,-1,-1,0,mm1) ; vals.setName("")
 712         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 713         self.assertTrue(vals.isEqual(d,1e-14))
 714         pass
 715
 716     # idem testMEDField9 method except that here testing profile on nodes and not on cells.
 717     def testMEDField11(self):
 718         fname="Pyfile32.med"
 719         m1=MEDLoaderDataForTest.build2DMesh_1()
 720         m1.renumberCells([0,1,4,2,3,5],False)
 721         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ;
 722         mm1.write(fname,2)
 723         ff1=MEDFileField1TS.New()
 724         f1=MEDCouplingFieldDouble.New(ON_NODES,ONE_TIME) ; f1.setName("F1Node")
 725         d=DataArrayDouble.New() ; d.alloc(2*6,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 726         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 727         da=DataArrayInt.New(); da.setValues([1,2,4,5,7,8],6,1) ; da.setName("sup1Node")
 728         #
 729         ff1.setFieldProfile(f1,mm1,0,da)
 730         self.assertEqual(ff1.getNonEmptyLevels(),(-1, []))
 731         ff1.write(fname,0)
 732         #
 733         vals,pfl=ff1.getFieldWithProfile(ON_NODES,0,mm1) ; vals.setName("")
 734         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 735         self.assertTrue(vals.isEqual(d,1e-14))
 736         ## #
 737         ff2=MEDFileField1TS.New(fname,f1.getName(),-1,-1)
 738         vals,pfl=ff2.getFieldWithProfile(ON_NODES,0,mm1) ; vals.setName("")
 739         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 740         self.assertTrue(vals.isEqual(d,1e-14))
 741         pass
 742
 743     def testMEDField12(self):
 744         fname="Pyfile33.med"
 745         m1=MEDLoaderDataForTest.build2DMesh_1()
 746         m1.renumberCells([0,1,4,2,3,5],False)
 747         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ;
 748         mm1.write(fname,2)
 749         ff1=MEDFileFieldMultiTS.New()
 750         f1=MEDCouplingFieldDouble.New(ON_NODES,ONE_TIME) ; f1.setName("F1Node")
 751         d=DataArrayDouble.New() ; d.alloc(2*6,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 752         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 753         da=DataArrayInt.New(); da.setValues([1,2,4,5,7,8],6,1) ; da.setName("sup1Node")
 754         #
 755         ff1.appendFieldProfile(f1,mm1,0,da)
 756         f1.setTime(1.2,1,2) ; e=d.applyFunc("2*x") ; e.copyStringInfoFrom(d) ; f1.setArray(e) ;
 757         ff1.appendFieldProfile(f1,mm1,0,da)
 758         ff1.write(fname,0)
 759         #
 760         vals,pfl=ff1.getFieldWithProfile(ON_NODES,1,2,0,mm1) ; vals.setName("")
 761         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 762         self.assertTrue(vals.isEqual(e,1e-14))
 763         vals,pfl=ff1.getFieldWithProfile(ON_NODES,-1,-1,0,mm1) ; vals.setName("")
 764         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 765         self.assertTrue(vals.isEqual(d,1e-14))
 766         #
 767         ff2=MEDFileFieldMultiTS.New(fname,f1.getName())
 768         vals,pfl=ff2.getFieldWithProfile(ON_NODES,1,2,0,mm1) ; vals.setName("")
 769         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 770         self.assertTrue(vals.isEqual(e,1e-14))
 771         vals,pfl=ff2.getFieldWithProfile(ON_NODES,-1,-1,0,mm1) ; vals.setName("")
 772         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 773         self.assertTrue(vals.isEqual(d,1e-14))
 774         pass
 775
 776     def testMEDField13(self):
 777         fname="Pyfile34.med"
 778         m1=MEDLoaderDataForTest.build2DMesh_1()
 779         m1.renumberCells([0,1,4,2,3,5],False)
 780         tmp=m1.getName();
 781         m1=m1.buildPartOfMySelf(range(5),True) ; m1.setName(tmp) # suppression of last cell that is a polygon
 782         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ;
 783         mm1.write(fname,2)
 784         ff1=MEDFileField1TS.New()
 785         f1=MEDCouplingFieldDouble.New(ON_GAUSS_NE,ONE_TIME) ; f1.setName("F3Node")
 786         d=DataArrayDouble.New() ; d.alloc(2*11,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 787         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 788         da=DataArrayInt.New(); da.setValues([0,2,3],3,1) ; da.setName("sup1NodeElt")
 789         #
 790         ff1.setFieldProfile(f1,mm1,0,da)
 791         ff1.write(fname,0)
 792         #
 793         vals,pfl=ff1.getFieldWithProfile(ON_GAUSS_NE,0,mm1) ; vals.setName("")
 794         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 795         self.assertTrue(vals.isEqual(d,1e-14))
 796         #
 797         ff2=MEDFileField1TS.New(fname,f1.getName(),-1,-1)
 798         vals,pfl=ff2.getFieldWithProfile(ON_GAUSS_NE,0,mm1) ; vals.setName("")
 799         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 800         self.assertTrue(vals.isEqual(d,1e-14))
 801         pass
 802
 803     def testMEDField14(self):
 804         fname="Pyfile35.med"
 805         m1=MEDLoaderDataForTest.build2DMesh_1()
 806         m1.renumberCells([0,1,4,2,3,5],False)
 807         tmp=m1.getName();
 808         m1=m1.buildPartOfMySelf(range(5),True) ; m1.setName(tmp) # suppression of last cell that is a polygon
 809         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ;
 810         mm1.write(fname,2)
 811         ff1=MEDFileFieldMultiTS.New()
 812         f1=MEDCouplingFieldDouble.New(ON_GAUSS_NE,ONE_TIME) ; f1.setName("F4Node")
 813         d=DataArrayDouble.New() ; d.alloc(2*11,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 814         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 815         da=DataArrayInt.New(); da.setValues([0,2,3],3,1) ; da.setName("sup1NodeElt")
 816         #
 817         ff1.appendFieldProfile(f1,mm1,0,da)
 818         f1.setTime(1.2,1,2) ; e=d.applyFunc("2*x") ; e.copyStringInfoFrom(d) ; f1.setArray(e) ;
 819         ff1.appendFieldProfile(f1,mm1,0,da)
 820         ff1.write(fname,0)
 821         #
 822         vals,pfl=ff1.getFieldWithProfile(ON_GAUSS_NE,-1,-1,0,mm1) ; vals.setName("")
 823         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 824         self.assertTrue(vals.isEqual(d,1e-14))
 825         vals,pfl=ff1.getFieldWithProfile(ON_GAUSS_NE,1,2,0,mm1) ; vals.setName("")
 826         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 827         self.assertTrue(vals.isEqual(e,1e-14))
 828         self.assertEqual([[3],[3]],ff1.getTypesOfFieldAvailable())
 829         #
 830         ff2=MEDFileFieldMultiTS.New(fname,f1.getName())
 831         vals,pfl=ff1.getFieldWithProfile(ON_GAUSS_NE,-1,-1,0,mm1) ; vals.setName("")
 832         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 833         self.assertTrue(vals.isEqual(d,1e-14))
 834         vals,pfl=ff1.getFieldWithProfile(ON_GAUSS_NE,1,2,0,mm1) ; vals.setName("")
 835         self.assertTrue(pfl.isEqualWithoutConsideringStr(da))
 836         self.assertTrue(vals.isEqual(e,1e-14))
 837         pass
 838     # Tricky test of the case of in a MED file containing a Field on GAUSS_NE is lying on a profile that is reality represents all the geom entities of a level.
 839     # By default when using setFieldProfile method such profile is not created because it is not useful ! So here a trick is used to force MEDLoader to do that
 840     # for the necessity of the test ! The idea is too create artificially a mesh having one more fictious cell per type and to roll back right after !
 841     def testMEDField15(self):
 842         fname="Pyfile36.med"
 843         m0=MEDLoaderDataForTest.build2DMesh_1()
 844         m0.renumberCells([0,1,4,2,3,5],False)
 845         tmp=m0.getName();
 846         m1=m0.buildPartOfMySelf([0,1,1,2,3,3,4,4],True) ; m1.setName(tmp) # suppression of last cell that is a polygon and creation of one more cell per type
 847         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ;
 848         ff1=MEDFileField1TS.New()
 849         f1=MEDCouplingFieldDouble.New(ON_GAUSS_NE,ONE_TIME) ; f1.setName("F4Node")
 850         d=DataArrayDouble.New() ; d.alloc(2*20,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 851         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 852         da=DataArrayInt.New(); da.setValues([0,1,3,4,6],5,1) ; da.setName("sup1NodeElt")
 853         #
 854         ff1.setFieldProfile(f1,mm1,0,da)
 855         m1=m0.buildPartOfMySelf(range(5),True) ; m1.setName(tmp) ; mm1.setMeshAtLevel(0,m1) ;
 856         mm1.write(fname,2)
 857         ff1.write(fname,0)
 858         f1=ff1.getFieldOnMeshAtLevel(ON_GAUSS_NE,m1,0)
 859         f2,p1=ff1.getFieldWithProfile(ON_GAUSS_NE,0,mm1) ; f2.setName("")
 860         self.assertTrue(p1.isIdentity())
 861         self.assertEqual(5,p1.getNumberOfTuples())
 862         self.assertTrue(f1.getArray().isEqual(f2,1e-12))
 863         pass
 864     # Test for getFieldAtTopLevel method
 865     def testMEDField16(self):
 866         fname="Pyfile37.med"
 867         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
 868         m1=f1.getMesh()
 869         mm1=MEDFileUMesh.New()
 870         mm1.setCoords(m1.getCoords())
 871         mm1.setMeshAtLevel(0,m1)
 872         mm1.setName(m1.getName())
 873         ff1=MEDFileField1TS.New()
 874         ff1.setFieldNoProfileSBT(f1)
 875         m2=m1.buildDescendingConnectivity()[0]
 876         m2.sortCellsInMEDFileFrmt()
 877         m2.setName(m1.getName())
 878         mm1.setMeshAtLevel(-1,m2)
 879         mm1.write(fname,2)
 880         f2=m2.getMeasureField(True)
 881         dd=DataArrayDouble.New()
 882         dd.alloc(f2.getArray().getNumberOfTuples(),3)
 883         dd[:,0]=f2.getArray()
 884         dd[:,1]=2*f2.getArray()
 885         dd[:,2]=3*f2.getArray()
 886         f2=f2.buildNewTimeReprFromThis(ONE_TIME,False)
 887         f2.setArray(dd)
 888         f2.copyTinyStringsFrom(f1)
 889         f2.copyTinyAttrFrom(f1)
 890         ff1.setFieldNoProfileSBT(f2)
 891         ff1.write(fname,0)
 892         # Reading Pyfile37.med
 893         ff2=MEDFileField1TS.New(fname,f2.getName(),0,1)
 894         f1bis=ff2.getFieldAtLevel(ON_CELLS,0)
 895         self.assertTrue(f1.isEqual(f1bis,1e-12,1e-12))
 896         f1bis=ff2.getFieldAtLevel(ON_CELLS,-1)
 897         self.assertTrue(f2.isEqual(f1bis,1e-12,1e-12))
 898         f1bis=ff2.getFieldAtTopLevel(ON_CELLS)
 899         self.assertTrue(f1.isEqual(f1bis,1e-12,1e-12))
 900         # More complex
 901         fname="Pyfile38.med"
 902         mm1.write(fname,2)
 903         ff1=MEDFileField1TS.New()
 904         ff1.setFieldNoProfileSBT(f2)
 905         ff1.write(fname,0)
 906         ff2=MEDFileField1TS.New(fname,f2.getName(),0,1)
 907         f1bis=ff2.getFieldAtTopLevel(ON_CELLS)
 908         self.assertTrue(f2.isEqual(f1bis,1e-12,1e-12))
 909         pass
 910
 911     # Non regression test to check that globals are correctly appended on MEDFileFields::setFieldAtPos
 912     def testMEDField17(self):
 913         fname="Pyfile39.med"
 914         m1=MEDLoaderDataForTest.build2DMesh_1()
 915         m1.renumberCells([0,1,4,2,3,5],False)
 916         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ; mm1.setName(m1.getName())
 917         mm1.write(fname,2)
 918         ffs=MEDFileFields.New()
 919         ff1=MEDFileFieldMultiTS.New()
 920         f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f1.setName("F2")
 921         d=DataArrayDouble.New() ; d.alloc(2*4,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 922         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 923         da=DataArrayInt.New(); da.setValues([0,1,2,4],4,1) ; da.setName("sup2")
 924         #
 925         ff1.appendFieldProfile(f1,mm1,0,da)
 926         f1.setTime(1.2,1,2) ; e=d.applyFunc("2*x") ; e.copyStringInfoFrom(d) ; f1.setArray(e) ;
 927         ff1.appendFieldProfile(f1,mm1,0,da)
 928         ffs.resize(1)
 929         ffs.setFieldAtPos(0,ff1)
 930         ffs=ffs.deepCpy()
 931         ffs.write(fname,0)
 932         #
 933         ffsr=MEDFileFields.New(fname)
 934         ff3=ffsr.getFieldAtPos(0)
 935         f4=ff3.getFieldAtTopLevel(ON_CELLS,1,2)
 936         self.assertTrue(f4.getArray().isEqual(f1.getArray(),1e-12))
 937         pass
 938
 939     # Non regression test to check that globals are correctly appended on MEDFileFields::setFieldAtPos
 940     def testMEDField18(self):
 941         fname="Pyfile40.med"
 942         m1=MEDLoaderDataForTest.build2DMesh_1()
 943         m1.renumberCells([0,1,4,2,3,5],False)
 944         mm1=MEDFileUMesh.New() ; mm1.setCoords(m1.getCoords()) ; mm1.setMeshAtLevel(0,m1) ; mm1.setName(m1.getName())
 945         mm1.write(fname,2)
 946         ffs=MEDFileFields.New()
 947         ff1=MEDFileFieldMultiTS.New()
 948         f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f1.setName("F2")
 949         d=DataArrayDouble.New() ; d.alloc(2*4,1) ; d.iota(7.); d.rearrange(2); d.setInfoOnComponent(0,"sigX [MPa]") ; d.setInfoOnComponent(1,"sigY [GPa]")
 950         f1.setArray(d) # note that f1 is NOT defined fully (no mesh !). It is not a bug of test it is too test that MEDFileField1TS.appendFieldProfile is NOT sensible of that.
 951         da=DataArrayInt.New(); da.setValues([0,1,2,4],4,1) ; da.setName("sup2")
 952         #
 953         ff1.appendFieldProfile(f1,mm1,0,da)
 954         f1.setTime(1.2,1,2) ; e=d.applyFunc("2*x") ; e.copyStringInfoFrom(d) ; f1.setArray(e) ;
 955         ff1.appendFieldProfile(f1,mm1,0,da)
 956         ffs.pushField(ff1)
 957         ffs.write(fname,0)
 958         #
 959         ffsr=MEDFileFields.New(fname)
 960         ff3=ffsr.getFieldAtPos(0)
 961         f4=ff3.getFieldAtTopLevel(ON_CELLS,1,2)
 962         self.assertTrue(f4.getArray().isEqual(f1.getArray(),1e-12))
 963         pass
 964
 965     def testMEDFieldBug1(self):
 966         fname="Pyfile13.med"
 967         d=MEDFileData.New(fname)
 968         self.assertEqual(('GP_MyFirstFieldOnGaussPoint0', 'GP_MyFirstFieldOnGaussPoint1', 'GP_MyFirstFieldOnGaussPoint2'),d.getFields().getFieldAtPos(0).getLocs())
 969         pass
 970
 971     def testMEDMesh8(self):
 972         m=MEDLoaderDataForTest.build1DMesh_1()
 973         m.convertQuadraticCellsToLinear()
 974         mm=MEDFileUMesh.New()
 975         mm.setMeshAtLevel(0,m)
 976         g1=DataArrayInt.New() ; g1.setValues([0,2],2,1) ; g1.setName("g1")
 977         g2=DataArrayInt.New() ; g2.setValues([1,3],2,1) ; g2.setName("g2")
 978         g3=DataArrayInt.New() ; g3.setValues([1,2,3],3,1) ; g3.setName("g3")
 979         mm.setGroupsAtLevel(0,[g1,g2],False)
 980         self.assertEqual(('g1','g2'),mm.getGroupsNames())
 981         self.assertEqual(('Family_2','Family_3'),mm.getFamiliesNames())
 982         self.assertEqual(('Family_2',),mm.getFamiliesOnGroup('g1'))
 983         self.assertEqual(('Family_3',),mm.getFamiliesOnGroup('g2'))
 984         mm.assignFamilyNameWithGroupName()
 985         self.assertEqual(('g1','g2'),mm.getGroupsNames())
 986         self.assertEqual(('g1','g2'),mm.getFamiliesNames())
 987         self.assertEqual(('g1',),mm.getFamiliesOnGroup('g1'))
 988         self.assertEqual(('g2',),mm.getFamiliesOnGroup('g2'))
 989         #
 990         mm=MEDFileUMesh.New()
 991         mm.setMeshAtLevel(0,m)
 992         mm.setGroupsAtLevel(0,[g1,g2,g3],False)
 993         self.assertEqual(('g1','g2','g3'),mm.getGroupsNames())
 994         self.assertEqual(('Family_2', 'Family_4', 'Family_5'),mm.getFamiliesNames())
 995         self.assertEqual(('Family_2', 'Family_4'),mm.getFamiliesOnGroup('g1'))
 996         self.assertEqual(('Family_5',),mm.getFamiliesOnGroup('g2'))
 997         self.assertEqual(('Family_4','Family_5',),mm.getFamiliesOnGroup('g3'))
 998         mm.assignFamilyNameWithGroupName() # here it does nothing because no such group-family bijection found
 999         self.assertEqual(('g1','g2','g3'),mm.getGroupsNames())
1000         self.assertEqual(('Family_2', 'Family_4', 'Family_5'),mm.getFamiliesNames())
1001         self.assertEqual(('Family_2', 'Family_4'),mm.getFamiliesOnGroup('g1'))
1002         self.assertEqual(('Family_5',),mm.getFamiliesOnGroup('g2'))
1003         self.assertEqual(('Family_4','Family_5',),mm.getFamiliesOnGroup('g3'))
1004         mm.changeFamilyId(5,6)
1005         g=mm.getGroupArr(0,"g3")
1006         self.assertTrue(g.isEqual(g3));
1007         g=mm.getGroupArr(0,"g2")
1008         self.assertTrue(g.isEqual(g2));
1009         g=mm.getGroupArr(0,"g1")
1010         self.assertTrue(g.isEqual(g1));
1011         pass
1012
1013     # bug detected by gauthier
1014     def testMEDLoaderMEDLoaderNSReadFieldDoubleDataInMedFile(self):
1015         fname="Pyfile41.med"
1016         f1=MEDLoaderDataForTest.buildVecFieldOnCells_1();
1017         m1=f1.getMesh()
1018         mm1=MEDFileUMesh.New()
1019         mm1.setCoords(m1.getCoords())
1020         mm1.setMeshAtLevel(0,m1)
1021         mm1.write(fname,2)
1022         ff1=MEDFileField1TS.New()
1023         ff1.setFieldNoProfileSBT(f1)
1024         ff1.write(fname,0)
1025         # writing mesh1 and field1, now creation of mesh2 and field2
1026         f2=f1.deepCpy()
1027         m2=f2.getMesh()
1028         m2.translate([0.5,0.6,0.7])
1029         m2.setName("3DSurfMesh_2")
1030         f2.getArray()[:]*=2.
1031         f2.setName("VectorFieldOnCells2")
1032         mm2=MEDFileUMesh.New()
1033         mm2.setCoords(m2.getCoords())
1034         mm2.setMeshAtLevel(0,m2)
1035         mm2.write(fname,0)
1036         ff2=MEDFileField1TS.New()
1037         ff2.setFieldNoProfileSBT(f2)
1038         ff2.write(fname,0)
1039         #
1040         f3=MEDLoader.ReadFieldCell(fname,"3DSurfMesh_1",0,"VectorFieldOnCells",0,1)
1041         self.assertTrue(f3.isEqual(f1,1e-12,1e-12))
1042         f4=MEDLoader.ReadFieldCell(fname,"3DSurfMesh_2",0,"VectorFieldOnCells2",0,1)
1043         self.assertTrue(f4.isEqual(f2,1e-12,1e-12))
1044         pass
1045
1046     def testMEDLoaderMultiLevelCellField1(self):
1047         fname="Pyfile42.med"
1048         m2,m1,m0,f2,f1,f0,p,n2,n1,n0,fns,fids,grpns,famIdsPerGrp=MEDLoaderDataForTest.buildMultiLevelMesh_1()
1049         m=MEDFileUMesh.New()
1050         m.setCoords(m2.getCoords())
1051         m.setMeshAtLevel(0,m2)
1052         m.setMeshAtLevel(-1,m1)
1053         m.setMeshAtLevel(-2,m0)
1054         m.write(fname,2)
1055         #
1056         FieldName1="Field1"
1057         compNames1=["comp1","comp2","comp3"]
1058         ff1=MEDFileField1TS.New()
1059         da2=DataArrayDouble.New()
1060         da2.alloc(m2.getNumberOfCells()*len(compNames1),1)
1061         da2.iota(7.)
1062         da2.rearrange(len(compNames1))
1063         da2.setInfoOnComponents(compNames1)
1064         f2=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f2.setName(FieldName1) ; f2.setArray(da2) ; f2.setMesh(m2) ; f2.checkCoherency()
1065         ff1.setFieldNoProfileSBT(f2)
1066         self.assertEqual(ff1.getNonEmptyLevels(),(2, [0]))
1067         da0=DataArrayDouble.New()
1068         da0.alloc(m0.getNumberOfCells()*len(compNames1),1)
1069         da0.iota(190.)
1070         da0.rearrange(len(compNames1))
1071         da0.setInfoOnComponents(compNames1)
1072         f0=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f0.setName(FieldName1) ; f0.setArray(da0) ; f0.setMesh(m0) ; f0.checkCoherency()
1073         ff1.setFieldNoProfileSBT(f0)
1074         self.assertEqual(ff1.getNonEmptyLevels(),(2, [0,-2]))
1075         da1=DataArrayDouble.New()
1076         da1.alloc(m1.getNumberOfCells()*len(compNames1),1)
1077         da1.iota(90.)
1078         da1.rearrange(len(compNames1))
1079         da1.setInfoOnComponents(compNames1)
1080         f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f1.setName(FieldName1) ; f1.setArray(da1) ; f1.setMesh(m1) ; f1.checkCoherency()
1081         ff1.setFieldNoProfileSBT(f1)
1082         self.assertEqual(ff1.getNonEmptyLevels(),(2, [0,-1,-2]))
1083         #
1084         ff1.write(fname,0)
1085         #
1086         FieldName2="Field2"
1087         compNames2=["comp11","comp22"]
1088         ff2=MEDFileField1TS.New()
1089         da0=DataArrayDouble.New()
1090         da0.alloc(m0.getNumberOfCells()*2,1)
1091         da0.iota(-190.)
1092         da0.rearrange(2)
1093         da0.setInfoOnComponents(compNames2)
1094         f0=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f0.setName(FieldName2) ; f0.setArray(da0) ; f0.setMesh(m0) ; f0.checkCoherency()
1095         ff2.setFieldNoProfileSBT(f0)
1096         self.assertEqual(ff2.getNonEmptyLevels(),(0, [0]))
1097         da1=DataArrayDouble.New()
1098         da1.alloc(m1.getNumberOfCells()*len(compNames2),1)
1099         da1.iota(-90.)
1100         da1.rearrange(len(compNames2))
1101         da1.setInfoOnComponents(compNames2)
1102         f1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f1.setName(FieldName2) ; f1.setArray(da1) ; f1.setMesh(m1) ; f1.checkCoherency()
1103         ff2.setFieldNoProfileSBT(f1)
1104         self.assertEqual(ff2.getNonEmptyLevels(),(1, [0,-1]))
1105         #
1106         ff2.write(fname,0)
1107         #
1108         ff1=MEDFileField1TS.New(fname,FieldName1,-1,-1)
1109         self.assertEqual(ff1.getNonEmptyLevels(),(2, [0,-1,-2]))
1110         self.assertEqual(ff1.getFieldSplitedByType(),[(0, [(0, (0, 4), '', '')]), (1, [(0, (4, 84), '', '')]), (3, [(0, (84, 148), '', '')]), (4, [(0, (148, 212), '', '')])])
1111         ff2=MEDFileField1TS.New(fname,FieldName2,-1,-1)
1112         self.assertEqual(ff2.getNonEmptyLevels(),(1, [0,-1]))
1113         self.assertEqual(ff2.getFieldSplitedByType(),[(0, [(0, (0, 4), '', '')]), (1, [(0, (4, 84), '', '')])])
1114         pass
1115
1116     def testFieldOnPflRetrieveOnMdimRelMax1(self):
1117         fname="Pyfile43.med"
1118         m2,m1,m0,f2,f1,f0,p,n2,n1,n0,fns,fids,grpns,famIdsPerGrp=MEDLoaderDataForTest.buildMultiLevelMesh_1()
1119         m=MEDFileUMesh.New()
1120         m.setMeshAtLevel(0,m2)
1121         m.setMeshAtLevel(-1,m1)
1122         m.setMeshAtLevel(-2,m0)
1123         f=MEDFileField1TS.New()
1124         ff=MEDCouplingFieldDouble.New(ON_NODES,ONE_TIME)
1125         ff.setName("NodeFieldPfl")
1126         arr=DataArrayDouble.New() ; arr.setValues([1.,10.,100.,2.,20.,200.],2,3)
1127         ff.setArray(arr)
1128         pfl=DataArrayInt.New() ; pfl.setValues([2,3],2,1) ; pfl.setName("PflNode")
1129         f.setFieldProfile(ff,m,-2,pfl)
1130         tes0=f.getFieldOnMeshAtLevel(ON_NODES,-1,m)
1131         self.assertEqual(ON_NODES,tes0.getTypeOfField())
1132         self.assertEqual(1,tes0.getMesh().getMeshDimension())
1133         self.assertEqual(1,tes0.getMesh().getNumberOfCells())
1134         self.assertEqual(2,tes0.getMesh().getNumberOfNodes())
1135         self.assertEqual([1,0,1],tes0.getMesh().getNodalConnectivity().getValues())
1136         self.assertEqual([0,3],tes0.getMesh().getNodalConnectivityIndex().getValues())
1137         self.assertEqual(2,tes0.getArray().getNumberOfTuples())
1138         self.assertEqual(3,tes0.getArray().getNumberOfComponents())
1139         expected1=[1.,10.,100.,2.,20.,200.]
1140         nodeCoordsWithValue1=[10.,2.5,0.]
1141         nodeCoordsWithValue2=[10.,3.75,0.]
1142         for i in xrange(3):
1143             self.assertAlmostEqual(nodeCoordsWithValue1[i],tes0.getMesh().getCoordinatesOfNode(0)[i],13);
1144             self.assertAlmostEqual(nodeCoordsWithValue2[i],tes0.getMesh().getCoordinatesOfNode(1)[i],13);
1145             pass
1146         for i in xrange(6):
1147             self.assertAlmostEqual(expected1[i],tes0.getArray().getIJ(0,i),13);
1148             pass
1149         del tes0
1150         #
1151         tes1=f.getFieldOnMeshAtLevel(ON_NODES,1,m)
1152         self.assertEqual(ON_CELLS,tes1.getTypeOfField())# it is not a bug even if ON_NODES has been sepecified
1153         self.assertEqual(0,tes1.getMesh().getMeshDimension())
1154         self.assertEqual(2,tes1.getMesh().getNumberOfCells())
1155         self.assertEqual(135,tes1.getMesh().getNumberOfNodes())
1156         self.assertEqual([0,2,0,3],tes1.getMesh().getNodalConnectivity().getValues())
1157         self.assertEqual([0,2,4],tes1.getMesh().getNodalConnectivityIndex().getValues())
1158         self.assertEqual(2,tes1.getArray().getNumberOfTuples())
1159         self.assertEqual(3,tes1.getArray().getNumberOfComponents())
1160         for i in xrange(6):
1161             self.assertAlmostEqual(expected1[i],tes1.getArray().getIJ(0,i),13);
1162             pass
1163         m.write(fname,2)
1164         f.write(fname,0)
1165         #
1166         pfl=DataArrayInt.New() ; pfl.setValues([3,2],2,1) ; pfl.setName("PflNode")
1167         f=MEDFileField1TS.New()
1168         f.setFieldProfile(ff,m,-2,pfl)
1169         tes2=f.getFieldOnMeshAtLevel(ON_NODES,-1,m)
1170         self.assertEqual(ON_NODES,tes2.getTypeOfField())
1171         self.assertEqual(1,tes2.getMesh().getMeshDimension())
1172         self.assertEqual(1,tes2.getMesh().getNumberOfCells())
1173         self.assertEqual(2,tes2.getMesh().getNumberOfNodes())
1174         self.assertEqual([1,0,1],tes2.getMesh().getNodalConnectivity().getValues())
1175         self.assertEqual([0,3],tes2.getMesh().getNodalConnectivityIndex().getValues())
1176         self.assertEqual(2,tes2.getArray().getNumberOfTuples())
1177         self.assertEqual(3,tes2.getArray().getNumberOfComponents())
1178         expected2=[2.,20.,200.,1.,10.,100.]
1179         for i in xrange(3):
1180             self.assertAlmostEqual(nodeCoordsWithValue1[i],tes2.getMesh().getCoordinatesOfNode(0)[i],13);
1181             self.assertAlmostEqual(nodeCoordsWithValue2[i],tes2.getMesh().getCoordinatesOfNode(1)[i],13);
1182             pass
1183         for i in xrange(6):
1184             self.assertAlmostEqual(expected2[i],tes2.getArray().getIJ(0,i),13);#compare tes2 and tes3
1185             pass
1186         #
1187         tes3=f.getFieldOnMeshAtLevel(ON_NODES,1,m)
1188         self.assertEqual(ON_CELLS,tes3.getTypeOfField())# it is not a bug even if ON_NODES has been sepecified
1189         self.assertEqual(0,tes3.getMesh().getMeshDimension())
1190         self.assertEqual(2,tes3.getMesh().getNumberOfCells())
1191         self.assertEqual(135,tes3.getMesh().getNumberOfNodes())
1192         self.assertEqual([0,3,0,2],tes3.getMesh().getNodalConnectivity().getValues())
1193         self.assertEqual([0,2,4],tes3.getMesh().getNodalConnectivityIndex().getValues())
1194         self.assertEqual(2,tes3.getArray().getNumberOfTuples())
1195         self.assertEqual(3,tes3.getArray().getNumberOfComponents())
1196         for i in xrange(6):
1197             self.assertAlmostEqual(expected1[i],tes3.getArray().getIJ(0,i),13);
1198             pass
1199         pass
1200
1201     def testDuplicateNodesOnM1Group1(self):
1202         fname="Pyfile44.med"
1203         m=MEDCouplingCMesh.New()
1204         m.setCoordsAt(0,DataArrayDouble.New([0.,1.1,2.3,3.6,5.,6.5]))
1205         m.setCoordsAt(1,DataArrayDouble.New([0.,1.1,2.3,3.6,5.]))
1206         m=m.buildUnstructured() ; m.setName("AnthonyDuplicate")
1207         m.getCoords().setInfoOnComponents(["X [km]","Z [mm]"])
1208         m2=m.buildDescendingConnectivity()[0][[8,11,14,20,21,22,23,24,25,26,31,32,33,34,35,36,37]]
1209         m2.setName(m.getName())
1210         grp=DataArrayInt.New([4,6,8]) ; grp.setName("Grp")
1211         grp2=DataArrayInt.New([9,16]) ; grp2.setName("Grp2")
1212         mm=MEDFileUMesh.New()
1213         mm.setMeshAtLevel(0,m)
1214         mm.setMeshAtLevel(-1,m2)
1215         mm.setGroupsAtLevel(-1,[grp,grp2])
1216         grpNode=DataArrayInt.New([4,21,23]) ; grpNode.setName("GrpNode")
1217         mm.setGroupsAtLevel(1,[grpNode])
1218         ref0=[4,15,14,20,21,4,16,15,21,22,4,17,16,22,23]
1219         ref1=[4,9,8,14,15,4,10,9,15,16,4,11,10,16,17]
1220         ref2=[4,9,8,14,30,4,10,9,30,31,4,11,10,31,32]
1221         #
1222         self.assertEqual(30,mm.getNumberOfNodes())
1223         self.assertEqual(ref0,mm.getMeshAtLevel(0)[[12,13,14]].getNodalConnectivity().getValues())
1224         self.assertEqual(ref1,mm.getMeshAtLevel(0)[[7,8,9]].getNodalConnectivity().getValues())
1225         #
1226         nodes,cells,cells2=mm.duplicateNodesOnM1Group("Grp")
1227         self.assertEqual([15,16,17],nodes.getValues());
1228         self.assertEqual([7,8,9],cells.getValues());
1229         self.assertEqual([12,13,14],cells2.getValues());
1230         self.assertEqual(33,mm.getNumberOfNodes())
1231         self.assertEqual([4,6,8],mm.getGroupArr(-1,"Grp").getValues())
1232         self.assertEqual([9,16],mm.getGroupArr(-1,"Grp2").getValues())
1233         self.assertEqual([4,21,23],mm.getGroupArr(1,"GrpNode").getValues())
1234         self.assertEqual([17,18,19],mm.getGroupArr(-1,"Grp_dup").getValues())
1235         self.assertEqual(ref0,mm.getMeshAtLevel(0)[[12,13,14]].getNodalConnectivity().getValues())#cells 7,8,9 and 12,13,14 are lying on "Grp" but only 7,8 and 9 are renumbered
1236         self.assertEqual(ref2,mm.getMeshAtLevel(0)[[7,8,9]].getNodalConnectivity().getValues())#
1237         self.assertRaises(InterpKernelException,mm.getGroup(-1,"Grp_dup").checkGeoEquivalWith,mm.getGroup(-1,"Grp"),2,1e-12);# Grp_dup and Grp are not equal considering connectivity only
1238         mm.getGroup(-1,"Grp_dup").checkGeoEquivalWith(mm.getGroup(-1,"Grp"),12,1e-12)# Grp_dup and Grp are equal considering connectivity and coordinates
1239         refValues=DataArrayDouble.New([1.21,1.32,1.43,1.54,1.65,1.32,1.44,1.56,1.68,1.8,1.43,1.56,1.69,1.82,1.95,1.54,1.68,1.82,1.96,2.1])
1240         valsToTest=mm.getMeshAtLevel(0).getMeasureField(True).getArray() ; delta=(valsToTest-refValues) ; delta.abs()
1241         self.assertTrue(delta.getMaxValue()[0]<1e-12)
1242         #
1243         mm.getCoords()[-len(nodes):]+=[0.,-0.3]
1244         self.assertRaises(InterpKernelException,mm.getGroup(-1,"Grp_dup").checkGeoEquivalWith,mm.getGroup(-1,"Grp"),12,1e-12);
1245         refValues2=refValues[:] ; refValues2[7:10]=[1.365,1.26,1.35]
1246         valsToTest=mm.getMeshAtLevel(0).getMeasureField(True).getArray() ; delta=(valsToTest-refValues2) ; delta.abs()
1247         self.assertTrue(delta.getMaxValue()[0]<1e-12)
1248         mm.write(fname,2)
1249         pass
1250
1251     def testDuplicateNodesOnM1Group2(self):
1252         fname="Pyfile45.med"
1253         m=MEDCouplingCMesh.New()
1254         m.setCoordsAt(0,DataArrayDouble.New([0.,1.1,2.3,3.6,5.,6.5]))
1255         m.setCoordsAt(1,DataArrayDouble.New([0.,1.1,2.3,3.6,5.]))
1256         m=m.buildUnstructured() ; m.setName("AnthonyDuplicate")
1257         m.getCoords().setInfoOnComponents(["X [km]","Z [mm]"])
1258         m2=m.buildDescendingConnectivity()[0][[8,11,14,20,21,22,23,24,25,26,31,32,33,34,35,36,37]]
1259         m2.setName(m.getName())
1260         grp=DataArrayInt.New([4,6]) ; grp.setName("Grp")
1261         grp2=DataArrayInt.New([9,16]) ; grp2.setName("Grp2")
1262         mm=MEDFileUMesh.New()
1263         mm.setMeshAtLevel(0,m)
1264         mm.setMeshAtLevel(-1,m2)
1265         mm.setGroupsAtLevel(-1,[grp,grp2])
1266         grpNode=DataArrayInt.New([4,21,23]) ; grpNode.setName("GrpNode")
1267         mm.setGroupsAtLevel(1,[grpNode])
1268         ref0=[4,15,14,20,21,4,16,15,21,22,4,17,16,22,23]
1269         ref1=[4,9,8,14,15,4,10,9,15,16]
1270         ref2=[4,9,8,14,30,4,10,9,30,16]
1271         #
1272         self.assertEqual(30,mm.getNumberOfNodes())
1273         self.assertEqual(ref0,mm.getMeshAtLevel(0)[[12,13,14]].getNodalConnectivity().getValues())
1274         self.assertEqual(ref1,mm.getMeshAtLevel(0)[[7,8]].getNodalConnectivity().getValues())
1275         #
1276         nodes,cells,cells2=mm.duplicateNodesOnM1Group("Grp")
1277         self.assertEqual([15],nodes.getValues());
1278         self.assertEqual([7,8],cells.getValues());
1279         self.assertEqual([12,13],cells2.getValues());
1280         self.assertEqual(31,mm.getNumberOfNodes())
1281         self.assertEqual([4,6],mm.getGroupArr(-1,"Grp").getValues())
1282         self.assertEqual([9,16],mm.getGroupArr(-1,"Grp2").getValues())
1283         self.assertEqual([4,21,23],mm.getGroupArr(1,"GrpNode").getValues())
1284         self.assertEqual([17,18],mm.getGroupArr(-1,"Grp_dup").getValues())
1285         self.assertEqual(ref0,mm.getMeshAtLevel(0)[[12,13,14]].getNodalConnectivity().getValues())#cells 7,8,9 and 12,13,14 are lying on "Grp" but only 7,8 and 9 are renumbered
1286         self.assertEqual(ref2,mm.getMeshAtLevel(0)[[7,8]].getNodalConnectivity().getValues())#
1287         self.assertRaises(InterpKernelException,mm.getGroup(-1,"Grp_dup").checkGeoEquivalWith,mm.getGroup(-1,"Grp"),2,1e-12);# Grp_dup and Grp are not equal considering connectivity only
1288         mm.getGroup(-1,"Grp_dup").checkGeoEquivalWith(mm.getGroup(-1,"Grp"),12,1e-12)# Grp_dup and Grp are equal considering connectivity and coordinates
1289         refValues=DataArrayDouble.New([1.21,1.32,1.43,1.54,1.65,1.32,1.44,1.56,1.68,1.8,1.43,1.56,1.69,1.82,1.95,1.54,1.68,1.82,1.96,2.1])
1290         valsToTest=mm.getMeshAtLevel(0).getMeasureField(True).getArray() ; delta=(valsToTest-refValues) ; delta.abs()
1291         self.assertTrue(delta.getMaxValue()[0]<1e-12)
1292         #
1293         mm.getCoords()[-len(nodes):]+=[0.,-0.3]
1294         self.assertRaises(InterpKernelException,mm.getGroup(-1,"Grp_dup").checkGeoEquivalWith,mm.getGroup(-1,"Grp"),12,1e-12);
1295         refValues2=refValues[:] ; refValues2[7:9]=[1.365,1.47]
1296         valsToTest=mm.getMeshAtLevel(0).getMeasureField(True).getArray() ; delta=(valsToTest-refValues2) ; delta.abs()
1297         self.assertTrue(delta.getMaxValue()[0]<1e-12)
1298         mm.write(fname,2)
1299         pass
1300
1301     def testBasicConstructors(self):
1302         fname="Pyfile18.med"
1303         m=MEDFileMesh(fname)
1304         m=MEDFileMesh(fname,"ExampleOfMultiDimW",-1,-1)
1305         m=MEDFileMesh(fname)
1306         m=MEDFileUMesh(fname,"ExampleOfMultiDimW",-1,-1)
1307         m=MEDFileUMesh(fname)
1308         m=MEDFileUMesh()
1309         self.testMEDMesh6()
1310         m=MEDFileCMesh("MEDFileMesh5.med")
1311         m=MEDFileCMesh("MEDFileMesh5.med","myFirstCartMesh",-1,-1)
1312         m=MEDFileCMesh()
1313         m=MEDFileMeshMultiTS()
1314         m=MEDFileMeshMultiTS(fname)
1315         m=MEDFileMeshMultiTS(fname,"ExampleOfMultiDimW")
1316         m=MEDFileMeshes()
1317         m=MEDFileMeshes(fname)
1318         m=MEDFileField1TS()
1319         m=MEDFileField1TS(fname,"FieldOnFacesShuffle",2,7)
1320         m=MEDFileFieldMultiTS()
1321         m=MEDFileFieldMultiTS(fname,"FieldOnFacesShuffle")
1322         m=MEDFileFields()
1323         m=MEDFileFields(fname)
1324         m=MEDFileData()
1325         m=MEDFileData(fname)
1326         #
1327         m=DataArrayInt() ; m=DataArrayInt(5,2) ; m=DataArrayInt([6,5,4,3,2,1],3,2)
1328         m=DataArrayDouble() ; m=DataArrayDouble(5,2) ; m=DataArrayDouble([6,5,4,3,2,1],3,2)
1329         m=MEDCouplingUMesh("jjj",2) ; m=MEDCouplingUMesh()
1330         m=MEDCouplingCMesh()
1331         m=MEDCouplingFieldDouble(ON_CELLS,ONE_TIME)
1332         m=MEDCouplingFieldTemplate(ON_NODES)
1333         m=MEDCouplingMultiFields([])
1334         m=MEDCouplingFieldOverTime([])
1335         pass
1336
1337     # This is a non regression test. When a field lies partially on a mesh but fully on one of its geometric type.
1338     def testBugSemiPartialField(self):
1339         fname="Pyfile46.med"
1340         m=MEDLoaderDataForTest.build2DMesh_3()
1341         m=m[:10] ; m.setName("mesh")
1342         f=m.getMeasureField(ON_CELLS)
1343         f=f.buildNewTimeReprFromThis(ONE_TIME,False)
1344         f.setTime(5.5,3,4)
1345         f.setName("SemiPartialField")
1346         #
1347         f1=f[:6] ; f1.getMesh().setName(m.getName())
1348         f2=f[6:] ; f2.getMesh().setName(m.getName())
1349         #
1350         mm=MEDFileUMesh.New()
1351         mm.setMeshAtLevel(0,m)
1352         ff=MEDFileField1TS.New()
1353         ff.setFieldProfile(f1,mm,0,DataArrayInt.Range(0,6,1)) # no name on profile -> normally it is an error but in this special case
1354         mm.write(fname,2)
1355         ff.write(fname,0)
1356         #
1357         ff2=MEDFileField1TS.New(fname,f.getName(),f.getTime()[1],f.getTime()[2])
1358         fread=ff2.getFieldOnMeshAtLevel(ON_CELLS,0,mm)
1359         fread2=ff2.getFieldAtLevel(ON_CELLS,0)
1360         #
1361         fread.checkCoherency()
1362         fread2.checkCoherency()
1363         self.assertTrue(fread.isEqual(f1,1e-12,1e-12))
1364         self.assertTrue(fread2.isEqual(f1,1e-12,1e-12))
1365         pass
1366
1367     def testUnPolyze1(self):
1368         fname="Pyfile47.med"
1369         mm=MEDLoaderDataForTest.buildMLMeshUnPolyze(self)
1370         ref=[13,14,14,12,12,12,12,12,12,12,12,13,12,14,14,13,15,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12]
1371         self.assertEqual(ref,mm.getFamilyFieldAtLevel(1).getValues())
1372         self.assertEqual(mm.unPolyze()[:3],(True,[[3,2,0],[4,3,2],[5,4,5],[14,2,9],[16,3,11],[31,2,14]],[[3,3,0],[4,3,3],[5,3,6],[14,3,9],[16,3,12],[18,1,15]]))
1373         mm.write(fname,2)
1374         self.assertEqual(mm.getGroupArr(0,"grp0_L0").getValues(),[0,1,2,6])
1375         self.assertEqual(mm.getGroupArr(0,"grp1_L0").getValues(),[1,3,4,5,6])
1376         self.assertEqual(mm.getGroupArr(-1,"grp0_LM1").getValues(),[1,2,3,4,5])
1377         self.assertEqual(mm.getGroupArr(-1,"grp1_LM1").getValues(),[3,4,5,6])
1378         self.assertEqual(mm.getGroupArr(-1,"grp2_LM1").getValues(),[2,6,7,8])
1379         self.assertEqual(mm.getGroupArr(1,"grp0_Node").getValues(),[0,11,15,16])
1380         self.assertEqual(mm.getGroupArr(1,"grp1_Node").getValues(),[1,2,13,14,16])
1381         self.assertEqual(mm.getFamilyFieldAtLevel(1).getValues(),ref)
1382         # to test
1383         mm.setRenumFieldArr(0,None)
1384         mm.setFamilyFieldArr(-1,None)
1385         pass
1386
1387     def testUnPolyze2(self):
1388         fname="Pyfile48.med"
1389         mfd=MEDFileData.New()
1390         mm=MEDLoaderDataForTest.buildMLMeshUnPolyze(self)
1391         meshes=MEDFileMeshes.New()
1392         meshes.pushMesh(mm)
1393         mfd.setMeshes(meshes)
1394         fields=MEDFileFields.New()
1395         mfd.setFields(fields)
1396         ff=MEDFileFieldMultiTS.New()
1397         fields.pushField(ff)
1398         #
1399         f0_0=MEDCouplingFieldDouble.New(ON_NODES,ONE_TIME) ; f0_0.setName("f0")
1400         f0_0.setTime(9.5,3,4)
1401         da=DataArrayDouble.New(38*2) ; da.iota(6.) ; da.rearrange(2) ; da.setInfoOnComponents(["Power [MW]","Density [kg/m^3]"])
1402         f0_0.setArray(da)
1403         f0_0.setMesh(mm.getMeshAtLevel(0))
1404         ff.appendFieldNoProfileSBT(f0_0)
1405         ff0=ff.getTimeStepAtPos(0)
1406         f0_1=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f0_1.setName("f0")
1407         f0_1.setTime(9.5,3,4)
1408         pfl=DataArrayInt.New([1,4,5,6]) ; pfl.setName("pfltest")
1409         f0_1.setMesh(mm.getMeshAtLevel(0)[pfl])
1410         da=DataArrayDouble.New([1401.,101401.,1602.,101602.,3100.,103100.,3101.,103101.],4,2) ; da.setInfoOnComponents(["Power [MW]","Density [kg/m^3]"])
1411         f0_1.setArray(da)
1412         ff0.setFieldProfile(f0_1,mm,0,pfl)
1413         f0_2=MEDCouplingFieldDouble.New(ON_CELLS,ONE_TIME) ; f0_2.setName("f0")#provoquer error
1414         f0_2.setTime(9.5,3,4)
1415         pfl2=DataArrayInt.New([0,1,2,3,4,5,6,8]) ; pfl2.setName("pfltestM1")
1416         da=DataArrayDouble.New([300.,100300.,301.,100301.,400.,100400.,401.,100401.,402.,100402.,3200.,103200.,3201.,103201.,3203.,103203.],8,2) ; da.setInfoOnComponents(["Power [MW]","Density [kg/m^3]"])#provoquer error
1417         f0_2.setMesh(mm.getMeshAtLevel(-1)[pfl2])
1418         f0_2.setArray(da)
1419         ff0.setFieldProfile(f0_2,mm,-1,pfl2)
1420         mfd.getFields().shallowCpyGlobs(ff0)
1421         #
1422         mfd.unPolyzeMeshes()
1423         #
1424         fmts=mfd.getFields()[0]
1425         self.assertEqual(fmts.getNumberOfTS(),1)
1426         self.assertEqual(fmts.getTimeSteps(),[(3,4,9.5)])
1427         arr,entry=fmts.getUndergroundDataArrayExt(3,4)
1428         self.assertEqual(entry,[((3,0),(38,40)),((4,0),(40,43)),((5,0),(43,46)),((14,0),(46,48)),((16,0),(48,49)),((18,0),(49,50)),((40,0),(0,38))])
1429         self.assertTrue(arr[38:40].isEqualWithoutConsideringStr(DataArrayDouble([300.0,100300.0,301.0,100301.0],2,2),1e-8))
1430         self.assertTrue(arr[40:43].isEqualWithoutConsideringStr(DataArrayDouble([400.0,100400.0,401.0,100401.0,402.0,100402.0],3,2),1e-8))
1431         self.assertTrue(arr[43:46].isEqualWithoutConsideringStr(DataArrayDouble([3200.0,103200.0,3201.0,103201.0,3203.0,103203.0],3,2),1e-8))
1432         self.assertTrue(arr[46:48].isEqualWithoutConsideringStr(DataArrayDouble([1401.0,101401.0,3100.0,103100.0],2,2),1e-8))
1433         self.assertTrue(arr[48:49].isEqualWithoutConsideringStr(DataArrayDouble([1602.0,101602.0],1,2),1e-8))
1434         self.assertTrue(arr[49:50].isEqualWithoutConsideringStr(DataArrayDouble([3101.0,103101.0],1,2),1e-8))
1435         self.assertEqual(('NewPfl_0','NewPfl_1','NewPfl_2'),fmts.getPflsReallyUsed())
1436         self.assertEqual([(3,[(0,(38,40),'NewPfl_0','')]),(4,[(0,(40,43),'','')]),(5,[(0,(43,46),'','')]),(14,[(0,(46,48),'NewPfl_1','')]),(16,[(0,(48,49),'NewPfl_2','')]),(18,[(0,(49,50),'','')]),(40,[(1,(0,38),'','')])],fmts.getFieldSplitedByType(3,4))
1437         self.assertEqual(fmts.getProfile("NewPfl_0").getValues(),[0,1])
1438         self.assertEqual(fmts.getProfile("NewPfl_1").getValues(),[1,2])
1439         self.assertEqual(fmts.getProfile("NewPfl_2").getValues(),[2])
1440         ftest0=fmts.getFieldOnMeshAtLevel(ON_CELLS,3,4,0,mfd.getMeshes()[0])
1441         self.assertTrue(ftest0.getArray().isEqualWithoutConsideringStr(DataArrayDouble([1401.,101401.,3100.,103100.,1602.,101602.,3101.,103101.],4,2),1e-8))
1442         self.assertEqual(ftest0.getMesh().getNodalConnectivity().getValues(),[14,4,5,6,7,14,26,27,28,29,16,20,21,22,23,24,25,18,30,31,32,33,34,35,36,37])
1443         self.assertEqual(ftest0.getMesh().getNodalConnectivityIndex().getValues(),[0,5,10,17,26])
1444         ftest1=fmts.getFieldOnMeshAtLevel(ON_CELLS,3,4,-1,mfd.getMeshes()[0])
1445         self.assertTrue(ftest1.getArray().isEqualWithoutConsideringStr(DataArrayDouble([300.,100300.,301.,100301.,400.,100400.,401.,100401.,402.,100402.,3200.,103200.,3201.,103201.,3203.,103203.]),1e-8))
1446         self.assertEqual(ftest1.getMesh().getNodalConnectivity().getValues(),[3,0,1,2,3,3,4,5,4,6,7,8,9,4,10,11,12,13,4,14,15,16,17,5,18,19,20,21,22,5,23,24,25,26,27,5,31,32,33,34,35,36,37])
1447         self.assertEqual(ftest1.getMesh().getNodalConnectivityIndex().getValues(),[0,4,8,13,18,23,29,35,43])
1448         #
1449         mfd.write(fname,2)
1450         pass
1451
1452     def testGaussWriteOnPfl1(self):
1453         fname="Pyfile49.med"
1454         fname2="Pyfile50.med"
1455         coords=DataArrayDouble([0.,0.,0.,1.,1.,1.,1.,0.,0.,0.5,0.5,1.,1.,0.5,0.5,0.],8,2)
1456         mQ8=MEDCouplingUMesh("",2) ; mQ8.setCoords(coords)
1457         mQ8.allocateCells(1)
1458         mQ8.insertNextCell(NORM_QUAD8,range(8))
1459         mQ8.finishInsertingCells()
1460         mQ4=MEDCouplingUMesh("",2) ; mQ4.setCoords(coords)
1461         mQ4.allocateCells(1)
1462         mQ4.insertNextCell(NORM_QUAD4,range(4))
1463         mQ4.finishInsertingCells()
1464         mT3=MEDCouplingUMesh("",2) ; mT3.setCoords(coords)
1465         mT3.allocateCells(1)
1466         mT3.insertNextCell(NORM_TRI3,range(3))
1467         mT3.finishInsertingCells()
1468
1469         tr=[[0.,4.],[2.,4.],[4.,4.],[6.,4.],[8.,4.],[10.,4.],[12.,4.],[14.,4.],[16.,4.],[18.,4.],[20.,4.],[0.,0.],[2.,0.], [0.,2.],[2.,2.],[4.,2.],[6.,2.],[8.,2.],[10.,2.],[12.,2.]]
1470         ms=11*[mT3]+2*[mQ4]+7*[mQ8]
1471         ms[:]=(elt.deepCpy() for elt in ms)
1472         for m,t in zip(ms,tr):
1473             d=m.getCoords() ; d+= t
1474             pass
1475         m=MEDCouplingUMesh.MergeUMeshes(ms)
1476         m.setName("mesh")
1477         m2=m[:13] ; m2.setName(m.getName())
1478         ### Use case 1 : Pfl on all tri3 and on all quad4. If we were on CELLS or GAUSS_NE no pfl were needed. But here 2 discs in tri3.
1479         ### So here 2 pfls will be created (pfl_TRI3_loc_0 and pfl_TRI3_loc_1)
1480         f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME)
1481         f.setMesh(m2)
1482         f.setTime(4.5,1,2)
1483         da=DataArrayDouble(34) ; da.iota(3.)
1484         f.setArray(da)
1485         f.setName("fieldCellOnPflWithoutPfl")
1486         f.setGaussLocalizationOnCells([0,1,2,3,4,5,6,7,8],[0.,0.,1.,0.,1.,1.],[0.3,0.3,0.7,0.7],[0.8,0.2])
1487         f.setGaussLocalizationOnCells([9,10],[0.,0.,1.,0.,1.,1.],[0.3,0.3,0.7,0.7,0.8,0.8],[0.8,0.07,0.13])
1488         f.setGaussLocalizationOnCells([11,12],[0.,0.,1.,0.,1.,1.,0.,1.],[0.3,0.3,0.7,0.7,0.8,0.8,0.8,0.8,0.8,0.8],[0.8,0.07,0.1,0.01,0.02])
1489         f.checkCoherency()
1490         #
1491         mm=MEDFileUMesh()
1492         mm.setMeshAtLevel(0,m)
1493         mm.write(fname,2)
1494         #
1495         f1ts=MEDFileField1TS.New()
1496         pfl=DataArrayInt(range(13)) ; pfl.setName("pfl")
1497         f1ts.setFieldProfile(f,mm,0,pfl)
1498         f1ts.write(fname,0)
1499         #
1500         self.assertEqual(f1ts.getPfls(),('pfl_NORM_TRI3_loc_0', 'pfl_NORM_TRI3_loc_1'))
1501         self.assertEqual(f1ts.getPflsReallyUsed(),('pfl_NORM_TRI3_loc_0', 'pfl_NORM_TRI3_loc_1'))
1502         da1=DataArrayInt([0,1,2,3,4,5,6,7,8]) ; da1.setName("pfl_NORM_TRI3_loc_0")
1503         self.assertTrue(f1ts.getProfile("pfl_NORM_TRI3_loc_0").isEqual(da1))
1504         da1=DataArrayInt([9,10]) ; da1.setName("pfl_NORM_TRI3_loc_1")
1505         self.assertTrue(f1ts.getProfile("pfl_NORM_TRI3_loc_1").isEqual(da1))
1506         self.assertEqual(f1ts.getLocs(),('Loc_fieldCellOnPflWithoutPfl_NORM_TRI3_0', 'Loc_fieldCellOnPflWithoutPfl_NORM_TRI3_1', 'Loc_fieldCellOnPflWithoutPfl_NORM_QUAD4_2'))
1507         self.assertEqual(f1ts.getLocsReallyUsed(),('Loc_fieldCellOnPflWithoutPfl_NORM_TRI3_0', 'Loc_fieldCellOnPflWithoutPfl_NORM_TRI3_1', 'Loc_fieldCellOnPflWithoutPfl_NORM_QUAD4_2'))
1508         #
1509         dataRead=MEDFileData.New(fname)
1510         mRead=dataRead.getMeshes()[0]
1511         f1tsRead=dataRead.getFields()[0][0]
1512         f1tsRead.getFieldOnMeshAtLevel(ON_GAUSS_PT,0,mRead)
1513         f2=f1tsRead.getFieldOnMeshAtLevel(ON_GAUSS_PT,0,mRead)
1514         self.assertTrue(f.isEqual(f2,1e-12,1e-12))
1515         f2_bis=MEDLoader.ReadFieldGauss(fname,m.getName(),0,f.getName(),f.getTime()[1],f.getTime()[2])
1516         f2_bis.checkCoherency()
1517         self.assertTrue(f.isEqual(f2_bis,1e-12,1e-12))
1518         #
1519         MEDLoader.WriteField(fname2,f,True)
1520         f2_ter=MEDLoader.ReadFieldGauss(fname2,m.getName(),0,f.getName(),f.getTime()[1],f.getTime()[2])
1521         self.assertTrue(f.isEqual(f2_ter,1e-12,1e-12))
1522         ## Use case 2 : Pfl on part tri3 with 2 disc and on part quad8 with 1 disc
1523         f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME)
1524         pfl=DataArrayInt([1,2,5,6,8,9,15,16,17,18]) ; pfl.setName("pfl2")
1525         m2=m[pfl] ; m2.setName(m.getName())
1526         f.setMesh(m2)
1527         f.setTime(4.5,1,2)
1528         da=DataArrayDouble(35) ; da.iota(3.)
1529         f.setArray(da)
1530         f.setName("fieldCellOnPflWithoutPfl2")
1531         f.setGaussLocalizationOnCells([0,1,3],[0.,0.,1.,0.,1.,1.],[0.3,0.3,0.7,0.7],[0.8,0.2])
1532         f.setGaussLocalizationOnCells([2,4,5],[0.,0.,1.,0.,1.,1.],[0.3,0.3,0.7,0.7,0.8,0.8],[0.8,0.07,0.13])
1533         f.setGaussLocalizationOnCells([6,7,8,9],[0.,0.,1.,0.,1.,1.,0.,1.,0.5,0.,1.,0.5,0.5,1.,0.,0.5],[0.3,0.3,0.7,0.7,0.8,0.8,0.8,0.8,0.8,0.8],[0.8,0.07,0.1,0.01,0.02])
1534         f.checkCoherency()
1535         #
1536         mm=MEDFileUMesh()
1537         mm.setMeshAtLevel(0,m)
1538         mm.write(fname,2)
1539         f1ts=MEDFileField1TS.New()
1540         f1ts.setFieldProfile(f,mm,0,pfl)
1541         self.assertEqual(f1ts.getPfls(),('pfl2_NORM_TRI3_loc_0','pfl2_NORM_TRI3_loc_1','pfl2_NORM_QUAD8_loc_2'))
1542         self.assertEqual(f1ts.getProfile("pfl2_NORM_TRI3_loc_0").getValues(),[1,2,6])
1543         self.assertEqual(f1ts.getProfile("pfl2_NORM_TRI3_loc_1").getValues(),[5,8,9])
1544         self.assertEqual(f1ts.getProfile("pfl2_NORM_QUAD8_loc_2").getValues(),[2,3,4,5])
1545         f1ts.write(fname,0)
1546         dataRead=MEDFileData.New(fname)
1547         mRead=dataRead.getMeshes()[0]
1548         f1tsRead=dataRead.getFields()[0][0]
1549         f1tsRead.getFieldOnMeshAtLevel(ON_GAUSS_PT,0,mRead)
1550         f3=f1tsRead.getFieldOnMeshAtLevel(ON_GAUSS_PT,0,mRead)
1551         f3.renumberCells([0,1,3,2,4,5,6,7,8,9])
1552         self.assertTrue(f.isEqual(f3,1e-12,1e-12))
1553         f3_bis=MEDLoader.ReadFieldGauss(fname,m.getName(),0,f.getName(),f.getTime()[1],f.getTime()[2])
1554         f3_bis.renumberCells([0,1,3,2,4,5,6,7,8,9])
1555         self.assertTrue(f.isEqual(f3_bis,1e-12,1e-12))
1556         #
1557         MEDLoader.WriteField(fname2,f,True)
1558         f3_ter=MEDLoader.ReadFieldGauss(fname2,m.getName(),0,f.getName(),f.getTime()[1],f.getTime()[2])
1559         f3_ter.renumberCells([0,1,3,2,4,5,6,7,8,9])
1560         self.assertTrue(f.isEqual(f3_ter,1e-12,1e-12))
1561         ## Use case 3 : no pfl but creation of pfls due to gauss pts
1562         f=MEDCouplingFieldDouble.New(ON_GAUSS_PT,ONE_TIME)
1563         f.setMesh(m)
1564         f.setTime(4.5,1,2)
1565         da=DataArrayDouble(60) ; da.iota(3.)
1566         f.setArray(da)
1567         f.setName("fieldCellWithoutPfl")
1568         f.setGaussLocalizationOnCells([0,1,2,3,4,5,6,7,8],[0.,0.,1.,0.,1.,1.],[0.3,0.3,0.7,0.7],[0.8,0.2])
1569         f.setGaussLocalizationOnCells([9,10],[0.,0.,1.,0.,1.,1.],[0.3,0.3,0.7,0.7,0.8,0.8],[0.8,0.07,0.13])
1570         f.setGaussLocalizationOnCells([11,12],[0.,0.,1.,0.,1.,1.,0.,1.],[0.3,0.3,0.7,0.7,0.8,0.8,0.8,0.8,0.8,0.8],[0.8,0.07,0.1,0.01,0.02])
1571         f.setGaussLocalizationOnCells([13,14,15,17,18],[0.,0.,1.,0.,1.,1.,0.,1.,0.5,0.,1.,0.5,0.5,1.,0.,0.5],[0.3,0.3,0.7,0.7,0.8,0.8,0.8,0.8],[0.8,0.1,0.03,0.07])
1572         f.setGaussLocalizationOnCells([16,19],[0.,0.,1.,0.,1.,1.,0.,1.,0.5,0.,1.,0.5,0.5,1.,0.,0.5],[0.3,0.3,0.7,0.7,0.8,0.8],[0.8,0.1,0.1])
1573         f.checkCoherency()
1574         mm=MEDFileUMesh()
1575         mm.setMeshAtLevel(0,m)
1576         f1ts=MEDFileField1TS.New()
1577         f1ts.setFieldNoProfileSBT(f)
1578         self.assertEqual(f1ts.getPfls(),('Pfl_fieldCellWithoutPfl_NORM_TRI3_0','Pfl_fieldCellWithoutPfl_NORM_TRI3_1','Pfl_fieldCellWithoutPfl_NORM_QUAD8_3','Pfl_fieldCellWithoutPfl_NORM_QUAD8_4'))
1579         self.assertEqual(f1ts.getProfile("Pfl_fieldCellWithoutPfl_NORM_TRI3_0").getValues(),[0,1,2,3,4,5,6,7,8])
1580         self.assertEqual(f1ts.getProfile("Pfl_fieldCellWithoutPfl_NORM_TRI3_1").getValues(),[9,10])
1581         self.assertEqual(f1ts.getProfile("Pfl_fieldCellWithoutPfl_NORM_QUAD8_3").getValues(),[0,1,2,4,5])
1582         self.assertEqual(f1ts.getProfile("Pfl_fieldCellWithoutPfl_NORM_QUAD8_4").getValues(),[3,6])
1583         mm.write(fname,2)
1584         f1ts.write(fname,0)
1585         #
1586         dataRead=MEDFileData.New(fname)
1587         mRead=dataRead.getMeshes()[0]
1588         f1tsRead=dataRead.getFields()[0][0]
1589         f3=f1tsRead.getFieldOnMeshAtLevel(ON_GAUSS_PT,0,mRead)
1590         f3.renumberCells([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,17,18,16,19])
1591         self.assertTrue(f.isEqual(f3,1e-12,1e-12))
1592         f3_bis=MEDLoader.ReadFieldGauss(fname,m.getName(),0,f.getName(),f.getTime()[1],f.getTime()[2])
1593         f3_bis.renumberCells([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,17,18,16,19])
1594         self.assertTrue(f.isEqual(f3_bis,1e-12,1e-12))
1595         #
1596         MEDLoader.WriteField(fname2,f,True)
1597         f3_ter=MEDLoader.ReadFieldGauss(fname2,m.getName(),0,f.getName(),f.getTime()[1],f.getTime()[2])
1598         f3_ter.renumberCells([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,17,18,16,19])
1599         self.assertTrue(f.isEqual(f3_ter,1e-12,1e-12))
1600         pass
1601
1602     # Testing profile on nodes when the profile is identity but not on all nodes.
1603     def testMEDFieldPflOnNode1(self):
1604         fname="Pyfile51.med"
1605         coo=DataArrayDouble([0.,0.,0.5,0.,1.,0.,0.,0.5,0.5,0.5,1.,0.5,0.,1.,0.5,1.,1.,1.],9,2)
1606         m0=MEDCouplingUMesh("Mesh",2)
1607         m0.allocateCells(5)
1608         m0.insertNextCell(NORM_TRI3,[1,4,2])
1609         m0.insertNextCell(NORM_TRI3,[4,5,2])
1610         m0.insertNextCell(NORM_QUAD4,[0,3,4,1])
1611         m0.insertNextCell(NORM_QUAD4,[3,6,7,4])
1612         m0.insertNextCell(NORM_QUAD4,[4,7,8,5])
1613         m0.finishInsertingCells()
1614         m0.setCoords(coo)
1615         m1=MEDCouplingUMesh(m0.getName(),1)
1616         m1.allocateCells(9)
1617         conn1=[0,1,0,3,3,4,4,1,5,4,2,4,1,2,3,6,5,8]
1618         for i in xrange(9):
1619             m1.insertNextCell(NORM_SEG2,conn1[2*i:2*i+2])
1620             pass
1621         m1.finishInsertingCells()
1622         m1.setCoords(coo)
1623         #
1624         m=MEDFileUMesh()
1625         m.setMeshAtLevel(0,m0)
1626         m.setMeshAtLevel(-1,m1)
1627         #
1628         dt=3 ; it=2 ; tim=4.5
1629         fieldNode0=MEDCouplingFieldDouble(ON_NODES,ONE_TIME)
1630         fieldNode0.setName("fieldNode0")
1631         fieldNode0.setTime(tim,dt,it)
1632         pfl0=DataArrayInt([0,1,2,3,4]) ; pfl0.setName("PflIdentity0") # important to keep like that
1633         arr=DataArrayDouble([10,11,12,13,14])
1634         fieldNode0.setArray(arr)
1635         f0=MEDFileField1TS()
1636         f0.setFieldProfile(fieldNode0,m,0,pfl0)
1637         m.write(fname,2) ; f0.write(fname,0)
1638         fieldNode1=MEDCouplingFieldDouble(ON_NODES,ONE_TIME)
1639         fieldNode1.setName("fieldNode1")
1640         fieldNode1.setTime(tim,dt,it)
1641         pfl1=DataArrayInt([0,1,2,3,4,5,6]) ; pfl1.setName("PflIdentity1")
1642         arr1=DataArrayDouble([20,21,22,23,24,25,26])
1643         fieldNode1.setArray(arr1)
1644         f1=MEDFileField1TS()
1645         f1.setFieldProfile(fieldNode1,m,-1,pfl1)
1646         f1.write(fname,0)
1647         del m,f0,m0,m1,f1
1648         ## Reading from file
1649         m=MEDFileMesh.New(fname)
1650         m0=m.getMeshAtLevel(0)
1651         m00=m0.deepCpy() ; m00=m00[[0,2]] ; m00.setName(m.getName()) ; m00.zipCoords()
1652         fieldNode0.setMesh(m00)
1653         f0=MEDFileField1TS.New(fname,fieldNode0.getName(),dt,it)
1654         ff0_1=f0.getFieldOnMeshAtLevel(ON_NODES,m0)
1655         ff0_1.checkCoherency()
1656         self.assertTrue(ff0_1.isEqual(fieldNode0,1e-12,1e-12))
1657         ff0_2=f0.getFieldAtLevel(ON_NODES,0)
1658         ff0_2.checkCoherency()
1659         self.assertTrue(ff0_2.isEqual(fieldNode0,1e-12,1e-12))
1660         ff0_3=f0.getFieldOnMeshAtLevel(ON_NODES,0,m)
1661         ff0_3.checkCoherency()
1662         self.assertTrue(ff0_3.isEqual(fieldNode0,1e-12,1e-12))
1663         ff0_4=MEDLoader.ReadFieldNode(fname,m.getName(),0,fieldNode0.getName(),dt,it)
1664         ff0_4.checkCoherency()
1665         self.assertTrue(ff0_4.isEqual(fieldNode0,1e-12,1e-12))
1666         f1=MEDFileField1TS.New(fname,fieldNode1.getName(),dt,it)
1667         m1=m.getMeshAtLevel(-1)
1668         m10=m1.deepCpy() ; m10=m10[[0,1,2,3,4,5,6,7]] ; m10.setName(m.getName()) ; m10.zipCoords()
1669         fieldNode1.setMesh(m10)
1670         ff1_1=f1.getFieldOnMeshAtLevel(ON_NODES,m1)
1671         ff1_1.checkCoherency()
1672         self.assertTrue(ff1_1.isEqual(fieldNode1,1e-12,1e-12))
1673         ff1_2=f1.getFieldAtLevel(ON_NODES,-1)
1674         ff1_2.checkCoherency()
1675         self.assertTrue(ff1_2.isEqual(fieldNode1,1e-12,1e-12))
1676         ff1_3=f1.getFieldOnMeshAtLevel(ON_NODES,-1,m)
1677         ff1_3.checkCoherency()
1678         self.assertTrue(ff1_3.isEqual(fieldNode1,1e-12,1e-12))
1679         ff1_4=MEDLoader.ReadFieldNode(fname,m.getName(),-1,fieldNode1.getName(),dt,it)
1680         ff1_4.checkCoherency()
1681         self.assertTrue(ff1_4.getMesh().isEqual(m10,1e-12))
1682         self.assertRaises(InterpKernelException,f1.getFieldOnMeshAtLevel,ON_NODES,m0) # error because impossible to build a sub mesh at level 0 lying on nodes [0,1,2,3,4,5,6]
1683         self.assertRaises(InterpKernelException,f1.getFieldAtLevel,ON_NODES,0) # error because impossible to build a sub mesh at level 0 lying on nodes [0,1,2,3,4,5,6]
1684         self.assertRaises(InterpKernelException,f1.getFieldOnMeshAtLevel,ON_NODES,0,m) # error because impossible to build a sub mesh at level 0 lying on nodes [0,1,2,3,4,5,6]
1685         arr_r,pfl1_r=f1.getFieldWithProfile(ON_NODES,-1,m)
1686         arr_r.setName(fieldNode1.getArray().getName())
1687         self.assertTrue(arr_r.isEqual(fieldNode1.getArray(),1e-12))
1688         pfl1_r.setName(pfl1.getName())
1689         self.assertTrue(pfl1_r.isEqual(pfl1))
1690         pass
1691
1692         # Testing profile on nodes when the profile is identity but not on all nodes.
1693     def testMEDFieldPflOnCell1(self):
1694         fname="Pyfile52.med"
1695         coo=DataArrayDouble([0.,0.,0.5,0.,1.,0.,0.,0.5,0.5,0.5,1.,0.5,0.,1.,0.5,1.,1.,1.],9,2)
1696         m0=MEDCouplingUMesh("Mesh",2)
1697         m0.allocateCells(5)
1698         m0.insertNextCell(NORM_TRI3,[1,4,2])
1699         m0.insertNextCell(NORM_TRI3,[4,5,2])
1700         m0.insertNextCell(NORM_QUAD4,[0,3,4,1])
1701         m0.insertNextCell(NORM_QUAD4,[3,6,7,4])
1702         m0.insertNextCell(NORM_QUAD4,[4,7,8,5])
1703         m0.finishInsertingCells()
1704         m0.setCoords(coo)
1705         m1=MEDCouplingUMesh(m0.getName(),1)
1706         m1.allocateCells(9)
1707         conn1=[0,1,0,3,3,4,4,1,5,4,2,4,1,2,3,6,5,8]
1708         for i in xrange(9):
1709             m1.insertNextCell(NORM_SEG2,conn1[2*i:2*i+2])
1710             pass
1711         m1.finishInsertingCells()
1712         m1.setCoords(coo)
1713         #
1714         m=MEDFileUMesh()
1715         m.setMeshAtLevel(0,m0)
1716         m.setMeshAtLevel(-1,m1)
1717         #
1718         dt=3 ; it=2 ; tim=4.5
1719         fieldCell0=MEDCouplingFieldDouble(ON_CELLS,ONE_TIME)
1720         fieldCell0.setName("fieldCell0")
1721         fieldCell0.setTime(tim,dt,it)
1722         pfl0=DataArrayInt([0,1,2]) ; pfl0.setName("PflIdentity0") # important to keep like that
1723         arr=DataArrayDouble([10,11,12])
1724         fieldCell0.setArray(arr)
1725         f0=MEDFileField1TS()
1726         f0.setFieldProfile(fieldCell0,m,0,pfl0)
1727         m.write(fname,2) ; f0.write(fname,0)
1728         fieldCell1=MEDCouplingFieldDouble(ON_CELLS,ONE_TIME)
1729         fieldCell1.setName("fieldCell1")
1730         fieldCell1.setTime(tim,dt,it)
1731         pfl1=DataArrayInt([0,1,2,3,4,5,6]) ; pfl1.setName("PflIdentity1")
1732         arr1=DataArrayDouble([20,21,22,23,24,25,26])
1733         fieldCell1.setArray(arr1)
1734         f1=MEDFileField1TS()
1735         f1.setFieldProfile(fieldCell1,m,-1,pfl1)
1736         f1.write(fname,0)
1737         del m,f0,m0,m1,f1
1738         ## Reading from file
1739         m=MEDFileMesh.New(fname)
1740         m0=m.getMeshAtLevel(0)
1741         m00=m0.deepCpy() ; m00=m00[pfl0] ; m00.setName(m.getName())
1742         fieldCell0.setMesh(m00)
1743         f0=MEDFileField1TS.New(fname,fieldCell0.getName(),dt,it)
1744         ff0_1=f0.getFieldOnMeshAtLevel(ON_CELLS,m0)
1745         ff0_1.checkCoherency()
1746         self.assertTrue(ff0_1.isEqual(fieldCell0,1e-12,1e-12))
1747         ff0_2=f0.getFieldAtLevel(ON_CELLS,0)
1748         ff0_2.checkCoherency()
1749         self.assertTrue(ff0_2.isEqual(fieldCell0,1e-12,1e-12))
1750         ff0_3=f0.getFieldOnMeshAtLevel(ON_CELLS,0,m)
1751         ff0_3.checkCoherency()
1752         self.assertTrue(ff0_3.isEqual(fieldCell0,1e-12,1e-12))
1753         ff0_4=MEDLoader.ReadFieldCell(fname,m.getName(),0,fieldCell0.getName(),dt,it)
1754         ff0_4.checkCoherency()
1755         self.assertTrue(ff0_4.isEqual(fieldCell0,1e-12,1e-12))
1756         f1=MEDFileField1TS.New(fname,fieldCell1.getName(),dt,it)
1757         m1=m.getMeshAtLevel(-1)
1758         m10=m1.deepCpy() ; m10=m10[pfl1] ; m10.setName(m.getName())
1759         fieldCell1.setMesh(m10)
1760         ff1_1=f1.getFieldOnMeshAtLevel(ON_CELLS,m1)
1761         ff1_1.checkCoherency()
1762         self.assertTrue(ff1_1.isEqual(fieldCell1,1e-12,1e-12))
1763         ff1_2=f1.getFieldAtLevel(ON_CELLS,-1)
1764         ff1_2.checkCoherency()
1765         self.assertTrue(ff1_2.isEqual(fieldCell1,1e-12,1e-12))
1766         ff1_3=f1.getFieldOnMeshAtLevel(ON_CELLS,-1,m)
1767         ff1_3.checkCoherency()
1768         self.assertTrue(ff1_3.isEqual(fieldCell1,1e-12,1e-12))
1769         ff1_4=MEDLoader.ReadFieldCell(fname,m.getName(),-1,fieldCell1.getName(),dt,it)
1770         ff1_4.checkCoherency()
1771         self.assertTrue(ff1_4.getMesh().isEqual(m10,1e-12))
1772         self.assertRaises(InterpKernelException,f1.getFieldOnMeshAtLevel,ON_CELLS,m0) # error because impossible to build a sub mesh at level 0 lying on cells [0,1,2,3,4,5,6]
1773         self.assertRaises(InterpKernelException,f1.getFieldAtLevel,ON_CELLS,0) # error because impossible to build a sub mesh at level 0 lying on cells [0,1,2,3,4,5,6]
1774         self.assertRaises(InterpKernelException,f1.getFieldOnMeshAtLevel,ON_CELLS,0,m) # error because impossible to build a sub mesh at level 0 lying on cells [0,1,2,3,4,5,6]
1775         arr_r,pfl1_r=f1.getFieldWithProfile(ON_CELLS,-1,m)
1776         arr_r.setName(fieldCell1.getArray().getName())
1777         self.assertTrue(arr_r.isEqual(fieldCell1.getArray(),1e-12))
1778         pfl1_r.setName(pfl1.getName())
1779         self.assertTrue(pfl1_r.isEqual(pfl1))
1780         pass
1781
1782     def testMEDFileUMeshZipCoords1(self):
1783         m=MEDFileUMesh()
1784         coo=DataArrayDouble(30) ; coo.iota(1.) ; coo.rearrange(3) ; coo.setInfoOnComponents(["aaa [b]","cc [dd]", "e [fff]"])
1785         m0=MEDCouplingUMesh("toto",2) ; m0.allocateCells(0) ; m0.insertNextCell(NORM_TRI3,[1,2,3]) ; m0.insertNextCell(NORM_QUAD4,[2,4,3,4]) ; m0.insertNextCell(NORM_POLYGON,[1,6,6,6,2])
1786         m1=MEDCouplingUMesh("toto",1) ; m1.allocateCells(0) ; m1.insertNextCell(NORM_SEG2,[1,6]) ; m1.insertNextCell(NORM_SEG2,[7,3])
1787         m2=MEDCouplingUMesh("toto",0) ; m2.allocateCells(0) ; m2.insertNextCell(NORM_POINT1,[2]) ; m2.insertNextCell(NORM_POINT1,[6]) ; m2.insertNextCell(NORM_POINT1,[8])
1788         m0.setCoords(coo) ; m.setMeshAtLevel(0,m0)
1789         m1.setCoords(coo) ; m.setMeshAtLevel(-1,m1)
1790         m2.setCoords(coo) ; m.setMeshAtLevel(-2,m2)
1791         numCoo=DataArrayInt(10) ; numCoo.iota(3) ; m.setRenumFieldArr(1,numCoo)
1792         famCoo=DataArrayInt(10) ; famCoo.iota(4) ; m.setFamilyFieldArr(1,famCoo)
1793         da=DataArrayInt([20,30,40]) ; m.setRenumFieldArr(0,da) ; da=DataArrayInt([200,300,400]) ; m.setFamilyFieldArr(0,da)
1794         da=DataArrayInt([50,60]) ; m.setRenumFieldArr(-1,da) ; da=DataArrayInt([500,600]) ; m.setFamilyFieldArr(-1,da)
1795         da=DataArrayInt([70,80,90]) ; m.setRenumFieldArr(-2,da) ; da=DataArrayInt([700,800,900]) ; m.setFamilyFieldArr(-2,da)
1796         o2n=m.zipCoords()
1797         self.assertTrue(o2n.isEqual(DataArrayInt([-1,0,1,2,3,-1,4,5,6,-1])))
1798         self.assertTrue(m.getNumberFieldAtLevel(1).isEqual(DataArrayInt([4,5,6,7,9,10,11])))
1799         self.assertTrue(m.getFamilyFieldAtLevel(1).isEqual(DataArrayInt([5,6,7,8,10,11,12])))
1800         self.assertTrue(m.getMeshAtLevel(0).getNodalConnectivity().isEqual(DataArrayInt([3,0,1,2,4,1,3,2,3,5,0,4,4,4,1])))
1801         self.assertTrue(m.getMeshAtLevel(0).getNodalConnectivityIndex().isEqual(DataArrayInt([0,4,9,15])))
1802         self.assertTrue(m.getMeshAtLevel(-1).getNodalConnectivity().isEqual(DataArrayInt([1,0,4,1,5,2])))
1803         self.assertTrue(m.getMeshAtLevel(-1).getNodalConnectivityIndex().isEqual(DataArrayInt([0,3,6])))
1804         self.assertTrue(m.getMeshAtLevel(-2).getNodalConnectivity().isEqual(DataArrayInt([0,1,0,4,0,6])))
1805         self.assertTrue(m.getMeshAtLevel(-2).getNodalConnectivityIndex().isEqual(DataArrayInt([0,2,4,6])))
1806         pass
1807
1808     def testMEDUMeshAddNodeGroup1(self):
1809         fname="Pyfile53.med"
1810         m=MEDFileUMesh()
1811         coo=DataArrayDouble(39) ; coo.iota(1.) ; coo.rearrange(3) ; coo.setInfoOnComponents(["aaa [b]","cc [dd]", "e [fff]"])
1812         m0=MEDCouplingUMesh("toto",2) ; m0.allocateCells(0) ; m0.insertNextCell(NORM_TRI3,[1,2,3]) ; m0.insertNextCell(NORM_QUAD4,[2,4,3,4]) ; m0.insertNextCell(NORM_POLYGON,[1,6,6,6,2])
1813         m1=MEDCouplingUMesh("toto",1) ; m1.allocateCells(0) ; m1.insertNextCell(NORM_SEG2,[1,6]) ; m1.insertNextCell(NORM_SEG2,[7,3])
1814         m2=MEDCouplingUMesh("toto",0) ; m2.allocateCells(0) ; m2.insertNextCell(NORM_POINT1,[2]) ; m2.insertNextCell(NORM_POINT1,[6]) ; m2.insertNextCell(NORM_POINT1,[8])
1815         m0.setCoords(coo) ; m.setMeshAtLevel(0,m0)
1816         m1.setCoords(coo) ; m.setMeshAtLevel(-1,m1)
1817         m2.setCoords(coo) ; m.setMeshAtLevel(-2,m2)
1818         #
1819         mm=m.deepCpy()
1820         famCoo=DataArrayInt([0,2,0,3,2,0,-1,0,0,0,0,-1,3]) ; mm.setFamilyFieldArr(1,famCoo)
1821         da0=DataArrayInt([0,0,0]) ; mm.setFamilyFieldArr(0,da0)
1822         da1=DataArrayInt([0,3]) ; mm.setFamilyFieldArr(-1,da1)
1823         da2=DataArrayInt([0,0,0]) ; mm.setFamilyFieldArr(-2,da2)
1824         mm.setFamilyId("MyFam",2)
1825         mm.setFamilyId("MyOtherFam",3)
1826         mm.setFamilyId("MyOther-1",-1)
1827         mm.setFamiliesOnGroup("grp0",["MyOtherFam"])
1828         mm.setFamiliesOnGroup("grpA",["MyOther-1"])
1829         #
1830         daTest=DataArrayInt([1,3,4,6,9,10,12]) ; daTest.setName("grp1")
1831         mm.addNodeGroup(daTest)
1832         self.assertTrue(mm.getGroupArr(1,daTest.getName()).isEqual(daTest))
1833         self.assertTrue(mm.getFamilyFieldAtLevel(1).isEqual(DataArrayInt([6,2,6,8,2,6,5,6,6,7,7,4,8])))
1834         for lev,arr in [(0,da0),(-1,da1),(-2,da2)]:
1835             self.assertTrue(mm.getFamilyFieldAtLevel(lev).isEqual(arr))
1836             pass
1837         self.assertEqual(mm.getFamiliesNames(),('Family_4','Family_5','Family_7','Family_8','MyFam','MyOther-1','MyOtherFam'))
1838         self.assertEqual(mm.getGroupsNames(),('grp0','grp1','grpA'))
1839         self.assertEqual(mm.getFamilyNameGivenId(3),'MyOtherFam')
1840         self.assertEqual(mm.getFamilyNameGivenId(2),'MyFam')
1841         for famName,famId in [('Family_4',4),('Family_5',5),('Family_7',7),('Family_8',8)]:
1842             self.assertEqual(mm.getFamilyNameGivenId(famId),famName)
1843             pass
1844         self.assertEqual(mm.getFamiliesOnGroup("grp0"),('MyOtherFam','Family_8'))
1845         da=DataArrayInt([3,12]) ; da.setName("grp0")
1846         self.assertTrue(mm.getGroupArr(1,"grp0").isEqual(da))
1847         da.setValues([1])
1848         self.assertTrue(mm.getGroupArr(-1,"grp0").isEqual(da))
1849         mm.write(fname,2)
1850         mm=MEDFileMesh.New(fname)
1851         self.assertTrue(mm.getGroupArr(1,daTest.getName()).isEqual(daTest))
1852         self.assertTrue(mm.getFamilyFieldAtLevel(1).isEqual(DataArrayInt([6,2,6,8,2,6,5,6,6,7,7,4,8])))
1853         for lev,arr in [(0,da0),(-1,da1),(-2,da2)]:
1854             self.assertTrue(mm.getFamilyFieldAtLevel(lev).isEqual(arr))
1855             pass
1856         self.assertEqual(mm.getFamiliesNames(),('FAMILLE_ZERO','Family_4','Family_5','Family_7','Family_8','MyFam','MyOther-1','MyOtherFam'))
1857         self.assertEqual(mm.getGroupsNames(),('grp0','grp1','grpA'))
1858         self.assertEqual(mm.getFamilyNameGivenId(3),'MyOtherFam')
1859         self.assertEqual(mm.getFamilyNameGivenId(2),'MyFam')
1860         for famName,famId in [('Family_4',4),('Family_5',5),('Family_7',7),('Family_8',8)]:
1861             self.assertEqual(mm.getFamilyNameGivenId(famId),famName)
1862             pass
1863         self.assertEqual(mm.getFamiliesOnGroup("grp0"),('Family_8','MyOtherFam'))
1864         da=DataArrayInt([3,12]) ; da.setName("grp0")
1865         self.assertTrue(mm.getGroupArr(1,"grp0").isEqual(da))
1866         da.setValues([1])
1867         self.assertTrue(mm.getGroupArr(-1,"grp0").isEqual(da))
1868         pass
1869
1870     def testMEDUMeshAddGroup1(self):
1871         fname="Pyfile54.med"
1872         m=MEDFileUMesh()
1873         coo=DataArrayDouble(9) ; coo.iota(1.) ; coo.rearrange(3) ; coo.setInfoOnComponents(["aaa [b]","cc [dd]", "e [fff]"])
1874         m0=MEDCouplingUMesh("toto",2) ; m0.allocateCells(0)
1875         for i in xrange(7):
1876             m0.insertNextCell(NORM_TRI3,[1,2,1])
1877             pass
1878         for i in xrange(4):
1879             m0.insertNextCell(NORM_QUAD4,[1,1,2,0])
1880             pass
1881         for i in xrange(2):
1882             m0.insertNextCell(NORM_POLYGON,[0,0,1,1,2,2])
1883             pass
1884         m1=MEDCouplingUMesh("toto",1) ; m1.allocateCells(0) ; m1.insertNextCell(NORM_SEG2,[1,6]) ; m1.insertNextCell(NORM_SEG2,[7,3])
1885         m2=MEDCouplingUMesh("toto",0) ; m2.allocateCells(0) ; m2.insertNextCell(NORM_POINT1,[2]) ; m2.insertNextCell(NORM_POINT1,[6]) ; m2.insertNextCell(NORM_POINT1,[8])
1886         m0.setCoords(coo) ; m.setMeshAtLevel(0,m0)
1887         m1.setCoords(coo) ; m.setMeshAtLevel(-1,m1)
1888         m2.setCoords(coo) ; m.setMeshAtLevel(-2,m2)
1889         #
1890         mm=m.deepCpy()
1891         famCoo=DataArrayInt([0,2,0,3,2,0,-1,0,0,0,0,-1,3]) ; mm.setFamilyFieldArr(0,famCoo)
1892         da0=DataArrayInt([0,0,0]) ; mm.setFamilyFieldArr(1,da0)
1893         da1=DataArrayInt([0,3]) ; mm.setFamilyFieldArr(-1,da1)
1894         da2=DataArrayInt([0,0,0]) ; mm.setFamilyFieldArr(-2,da2)
1895         mm.setFamilyId("MyFam",2)
1896         mm.setFamilyId("MyOtherFam",3)
1897         mm.setFamilyId("MyOther-1",-1)
1898         mm.setFamiliesOnGroup("grp0",["MyOtherFam"])
1899         mm.setFamiliesOnGroup("grpA",["MyOther-1"])
1900         #
1901         daTest=DataArrayInt([1,3,4,6,9,10,12]) ; daTest.setName("grp1")
1902         mm.addGroup(0,daTest)
1903         self.assertTrue(mm.getGroupArr(0,daTest.getName()).isEqual(daTest))
1904         self.assertTrue(mm.getFamilyFieldAtLevel(0).isEqual(DataArrayInt([6,2,6,8,2,6,5,6,6,7,7,4,8])))
1905         for lev,arr in [(1,da0),(-1,da1),(-2,da2)]:
1906             self.assertTrue(mm.getFamilyFieldAtLevel(lev).isEqual(arr))
1907             pass
1908         self.assertEqual(mm.getFamiliesNames(),('Family_4','Family_5','Family_7','Family_8','MyFam','MyOther-1','MyOtherFam'))
1909         self.assertEqual(mm.getGroupsNames(),('grp0','grp1','grpA'))
1910         self.assertEqual(mm.getFamilyNameGivenId(3),'MyOtherFam')
1911         self.assertEqual(mm.getFamilyNameGivenId(2),'MyFam')
1912         for famName,famId in [('Family_4',4),('Family_5',5),('Family_7',7),('Family_8',8)]:
1913             self.assertEqual(mm.getFamilyNameGivenId(famId),famName)
1914             pass
1915         self.assertEqual(mm.getFamiliesOnGroup("grp0"),('MyOtherFam','Family_8'))
1916         da=DataArrayInt([3,12]) ; da.setName("grp0")
1917         self.assertTrue(mm.getGroupArr(0,"grp0").isEqual(da))
1918         da.setValues([1])
1919         self.assertTrue(mm.getGroupArr(-1,"grp0").isEqual(da))
1920         mm.write(fname,2)
1921         mm=MEDFileMesh.New(fname)
1922         self.assertTrue(mm.getGroupArr(0,daTest.getName()).isEqual(daTest))
1923         self.assertTrue(mm.getFamilyFieldAtLevel(0).isEqual(DataArrayInt([6,2,6,8,2,6,5,6,6,7,7,4,8])))
1924         for lev,arr in [(1,da0),(-1,da1),(-2,da2)]:
1925             self.assertTrue(mm.getFamilyFieldAtLevel(lev).isEqual(arr))
1926             pass
1927         self.assertEqual(mm.getFamiliesNames(),('FAMILLE_ZERO','Family_4','Family_5','Family_7','Family_8','MyFam','MyOther-1','MyOtherFam'))
1928         self.assertEqual(mm.getGroupsNames(),('grp0','grp1','grpA'))
1929         self.assertEqual(mm.getFamilyNameGivenId(3),'MyOtherFam')
1930         self.assertEqual(mm.getFamilyNameGivenId(2),'MyFam')
1931         for famName,famId in [('Family_4',4),('Family_5',5),('Family_7',7),('Family_8',8)]:
1932             self.assertEqual(mm.getFamilyNameGivenId(famId),famName)
1933             pass
1934         self.assertEqual(mm.getFamiliesOnGroup("grp0"),('Family_8','MyOtherFam'))
1935         da=DataArrayInt([3,12]) ; da.setName("grp0")
1936         self.assertTrue(mm.getGroupArr(0,"grp0").isEqual(da))
1937         da.setValues([1])
1938         self.assertTrue(mm.getGroupArr(-1,"grp0").isEqual(da))
1939         pass
1940
1941     def testHeapMem1(self):
1942         import platform
1943         ver=platform.python_version_tuple()
1944         if int(ver[0])!=2 or int(ver[1])<7:
1945             return
1946         m=MEDCouplingCMesh()
1947         arr=DataArrayDouble(10,1) ; arr.iota(0)
1948         m.setCoords(arr,arr)
1949         m=m.buildUnstructured()
1950         m.setName("mm")
1951         f=m.getMeasureField(ON_CELLS)
1952         self.assertIn(m.getHeapMemorySize(),xrange(3552-100,3552+100))
1953         self.assertIn(f.getHeapMemorySize(),xrange(4215-100,4215+100))
1954         #
1955         mm=MEDFileUMesh()
1956         mm.setMeshAtLevel(0,m)
1957         self.assertIn(mm.getHeapMemorySize(),xrange(3889-100,3889+100))
1958         ff=MEDFileField1TS()
1959         ff.setFieldNoProfileSBT(f)
1960         self.assertIn(ff.getHeapMemorySize(),xrange(711-10,711+10))
1961         #
1962         fff=MEDFileFieldMultiTS()
1963         fff.appendFieldNoProfileSBT(f)
1964         self.assertIn(fff.getHeapMemorySize(),xrange(743-10,743+10))
1965         self.assertIn(fff[-1,-1].getHeapMemorySize(),xrange(711-10,711+10))
1966         f.setTime(1.,0,-1)
1967         fff.appendFieldNoProfileSBT(f)
1968         self.assertIn(fff.getHeapMemorySize(),xrange(1462-10,1462+10))
1969         self.assertIn(fff[0,-1].getHeapMemorySize(),xrange(711-10,711+10))
1970         f2=f[:50]
1971         f2.setTime(2.,1,-1)
1972         pfl=DataArrayInt.Range(0,50,1) ; pfl.setName("pfl")
1973         fff.appendFieldProfile(f2,mm,0,pfl)
1974         self.assertIn(fff.getHeapMemorySize(),xrange(2178-100,2178+100))
1975         self.assertIn(fff.getProfile("pfl_NORM_QUAD4").getHeapMemorySize(),xrange(215-10,215+10))
1976         self.assertIn(fff[1,-1].getHeapMemorySize(),xrange(700-10,700+10))
1977         pass
1978
1979     def testCurveLinearMesh1(self):
1980         fname="Pyfile55.med"
1981         mesh=MEDCouplingCurveLinearMesh();
1982         mesh.setTime(2.3,4,5);
1983         mesh.setTimeUnit("us");
1984         mesh.setName("Example of Cuve linear mesh");
1985         mesh.setDescription("buildCLMesh");
1986         a1=DataArrayDouble(3*20,1);
1987         a1.iota(7.) ; a1.rearrange(3);
1988         mesh.setCoords(a1);
1989         mesh.setNodeGridStructure([4,5]);
1990         mesh.checkCoherency();
1991         #
1992         m=MEDFileCurveLinearMesh()
1993         m.setMesh(mesh)
1994         d=DataArrayInt(20) ; d.iota(4)
1995         m.setFamilyFieldArr(1,d)
1996         d3=DataArrayInt(20) ; d3.iota(400)
1997         m.setRenumFieldArr(1,d3)
1998         d2=DataArrayInt(12) ; d2.iota(40)
1999         m.setFamilyFieldArr(0,d2)
2000         d4=DataArrayInt(21) ; d4.iota(4000)
2001         self.assertRaises(InterpKernelException,m.setRenumFieldArr,1,d4)
2002         d4.popBackSilent()
2003         m.setRenumFieldArr(1,d4)
2004         m.write(fname,2)
2005         #
2006         m1=MEDFileCurveLinearMesh(fname)
2007         mm=m1.getMesh()
2008         self.assertTrue(mm.isEqual(mesh,1e-12))
2009         #
2010         m1=MEDFileMesh.New(fname)
2011         self.assertTrue(isinstance(m1,MEDFileCurveLinearMesh))
2012         self.assertTrue(m1.getMesh().isEqual(mesh,1e-12))
2013         pass
2014
2015     def testParameters1(self):
2016         fname="Pyfile56.med"
2017         m=MEDCouplingCMesh() ; arr=DataArrayDouble([0.,1.2,3.5]) ; m.setCoords(arr,arr) ; m.setName("mesh")
2018         mm=MEDFileCMesh() ; mm.setMesh(m)
2019         ms=MEDFileMeshes() ; ms.pushMesh(mm)
2020         data=MEDFileData()
2021         p=MEDFileParameters()
2022         data.setParams(p) ; data.setMeshes(ms)
2023         pts=MEDFileParameterMultiTS()
2024         pts.setName("A") ; pts.setDescription("An example of parameter") ; pts.setTimeUnit("ms")
2025         pts.appendValue(1,2,3.4,567.89)
2026         pts.appendValue(2,3,5.6,999.123)
2027         pts2=pts.deepCpy() ; pts2.setName("B") ; pts2.setDescription("A second example")
2028         p.pushParam(pts) ; p.pushParam(pts2)
2029         data.write(fname,2)
2030         p2=MEDFileParameters(fname)
2031         self.assertTrue(p.isEqual(p2,1e-14)[0])
2032         self.assertAlmostEqual(p[1][1,2].getValue(),567.89,13)
2033         p3=p.deepCpy()
2034         pts4=pts2.deepCpy()
2035         pts3=pts2.deepCpy()
2036         self.assertTrue(pts3.isEqual(pts2,1e-14)[0])
2037         pts2.eraseTimeStepIds([0])
2038         self.assertTrue(not pts3.isEqual(pts2,1e-14)[0])
2039         del pts3[[3.4]]
2040         self.assertTrue(pts3.isEqual(pts2,1e-14)[0])
2041         self.assertRaises(InterpKernelException,p[1].__getitem__,(1,2))
2042         self.assertRaises(InterpKernelException,p["B"].__getitem__,(1,2))
2043         self.assertAlmostEqual(p[0][1,2].getValue(),567.89,13)
2044         self.assertAlmostEqual(p["A"][1,2].getValue(),567.89,13)
2045         p=p3
2046         self.assertTrue(p.isEqual(p2,1e-14)[0])
2047         self.assertTrue(p2["B"].isEqual(pts,1e-14)[0])
2048         self.assertTrue(not p2["B"].isEqual(pts2,1e-14)[0])
2049         self.assertAlmostEqual(p2[0][1,2].getValue(),567.89,13)
2050         self.assertEqual(p.getParamsNames(),('A','B'))
2051         ptsr=MEDFileParameterMultiTS(fname,"B")
2052         self.assertTrue(ptsr.isEqual(pts4,1e-14)[0])
2053         ptsr=MEDFileParameterMultiTS(fname)
2054         self.assertTrue(ptsr.isEqual(pts,1e-14)[0])
2055         p1tsr=MEDFileParameterDouble1TS(fname)
2056         self.assertEqual(p1tsr.getName(),"A")
2057         self.assertAlmostEqual(p1tsr.getValue(),567.89,13)
2058         p1tsr=MEDFileParameterDouble1TS(fname,"B")
2059         self.assertEqual(p1tsr.getName(),"B")
2060         self.assertAlmostEqual(p1tsr.getValue(),567.89,13)
2061         p1tsr=MEDFileParameterDouble1TS(fname,"B",2,3)
2062         self.assertEqual(p1tsr.getName(),"B")
2063         self.assertAlmostEqual(p1tsr.getValue(),999.123,13)
2064         data2=MEDFileData(fname)
2065         self.assertEqual(2,data2.getNumberOfParams())
2066         self.assertAlmostEqual(data2.getParams()["B"][1,2].getValue(),567.89,13)
2067         pass
2068
2069     def testNamesOnCellAndNodesInMeshes1(self):
2070         fname="Pyfile58.med"
2071         fname2="Pyfile59.med"
2072         m=MEDLoaderDataForTest.build3DSurfMesh_1()
2073         m1=m.buildDescendingConnectivity()[0]
2074         m1.sortCellsInMEDFileFrmt()
2075         #
2076         mm=MEDFileUMesh()
2077         mm.setMeshAtLevel(0,m)
2078         mm.setMeshAtLevel(-1,m1)
2079         namesCellL0=DataArrayAsciiChar(6,16)
2080         namesCellL0[:]=["CellL0#%.3d      "%(i) for i in xrange(6)]
2081         mm.setNameFieldAtLevel(0,namesCellL0)
2082         namesCellL1=DataArrayAsciiChar.Aggregate([namesCellL0,namesCellL0,namesCellL0.substr(2)])
2083         namesCellL1[:]=["CellLM1#%.3d     "%(i) for i in xrange(16)]
2084         mm.setNameFieldAtLevel(-1,namesCellL1)
2085         namesNodes=namesCellL1.substr(4,16)
2086         namesNodes[:]=["Node#%.3d        "%(i) for i in xrange(12)]
2087         mm.setNameFieldAtLevel(1,namesNodes)
2088         mm.write(fname,2)
2089         #
2090         mmr=MEDFileMesh.New(fname)
2091         self.assertTrue(mm.getNameFieldAtLevel(0).isEqual(DataArrayAsciiChar(["CellL0#%.3d      "%(i) for i in xrange(6)])))
2092         self.assertTrue(mm.getNameFieldAtLevel(-1).isEqual(DataArrayAsciiChar(["CellLM1#%.3d     "%(i) for i in xrange(16)])))
2093         self.assertTrue(mm.getNameFieldAtLevel(1).isEqual(DataArrayAsciiChar(["Node#%.3d        "%(i) for i in xrange(12)])))
2094         self.assertTrue(mm.isEqual(mmr,1e-12)[0])
2095         mmr.getNameFieldAtLevel(1).setIJ(0,0,'M')
2096         self.assertTrue(not mm.isEqual(mmr,1e-12)[0])
2097         mmr.getNameFieldAtLevel(1).setIJ(0,0,'N')
2098         self.assertTrue(mm.isEqual(mmr,1e-12)[0])
2099         mmCpy=mm.deepCpy()
2100         self.assertTrue(mm.isEqual(mmCpy,1e-12)[0])
2101         # remove names on nodes
2102         mmCpy.setNameFieldAtLevel(1,None)
2103         self.assertTrue(not mm.isEqual(mmCpy,1e-12)[0])
2104         mm.setNameFieldAtLevel(1,None)
2105         self.assertTrue(mm.isEqual(mmCpy,1e-12)[0])
2106         mm.setNameFieldAtLevel(-1,None)
2107         mm.write(fname,2)
2108         mmr=MEDFileMesh.New(fname)
2109         self.assertEqual(mmr.getNameFieldAtLevel(1),None)
2110         self.assertTrue(mmr.getNameFieldAtLevel(0).isEqual(DataArrayAsciiChar(["CellL0#%.3d      "%(i) for i in xrange(6)])))
2111         self.assertEqual(mmr.getNameFieldAtLevel(-1),None)
2112         #
2113         c=MEDCouplingCMesh()
2114         arr=DataArrayDouble([0.,1.1,2.3])
2115         c.setCoords(arr,arr)
2116         c.setName("cmesh")
2117         cc=MEDFileCMesh()
2118         cc.setMesh(c)
2119         cc.setNameFieldAtLevel(0,DataArrayAsciiChar(["Cell#%.3d        "%(i) for i in xrange(4)]))
2120         cc.setNameFieldAtLevel(1,DataArrayAsciiChar(["Node#%.3d        "%(i) for i in xrange(9)]))
2121         cc.write(fname2,2)
2122         ccr=MEDFileMesh.New(fname2)
2123         self.assertTrue(ccr.getNameFieldAtLevel(0).isEqual(DataArrayAsciiChar(["Cell#%.3d        "%(i) for i in xrange(4)])))
2124         self.assertTrue(ccr.getNameFieldAtLevel(1).isEqual(DataArrayAsciiChar(["Node#%.3d        "%(i) for i in xrange(9)])))
2125         self.assertTrue(cc.isEqual(ccr,1e-12)[0])
2126         ccr.getNameFieldAtLevel(1).setIJ(0,0,'M')
2127         self.assertTrue(not cc.isEqual(ccr,1e-12)[0])
2128         ccr.getNameFieldAtLevel(1).setIJ(0,0,'N')
2129         self.assertTrue(cc.isEqual(ccr,1e-12)[0])
2130         ccCpy=cc.deepCpy()
2131         self.assertTrue(cc.isEqual(ccCpy,1e-12)[0])
2132         pass
2133
2134     def testToExportInExamples1(self):
2135         m=MEDCouplingCMesh()
2136         arr=DataArrayDouble([0.,1.,2.,3.,4.])
2137         m.setCoords(arr,arr)
2138         m=m.buildUnstructured() ; m.setName("mesh")
2139         grp1=DataArrayInt([0,1,2,4,5,6,8,9,10,12,13,14]) ; grp1.setName("grp1")
2140         grp2=DataArrayInt([3,7,11,15]) ; grp2.setName("grp2")
2141         m2=m.computeSkin()
2142         mm=MEDFileUMesh()
2143         mm.setMeshAtLevel(0,m)
2144         mm.setMeshAtLevel(-1,m2)
2145         mm.setGroupsAtLevel(0,[grp1,grp2])
2146         mm.write("example.med",2)
2147         #
2148         m0=mm.getMeshAtLevel(0)
2149         m1=mm.getMeshAtLevel(-1)
2150         grp1=mm.getGroupArr(0,"grp1")
2151         grp2=mm.getGroupArr(0,"grp2")
2152         grps=[grp1,grp2]
2153         whichGrp=DataArrayInt(m0.getNumberOfCells())
2154         whichGrp.fillWithValue(-1)
2155         for grpId,grp in enumerate(grps):
2156             whichGrp[grp]=grpId
2157             pass
2158         a,b,bI,c,cI=m0.buildDescendingConnectivity()
2159         e,f=a.areCellsIncludedIn(m1,2)
2160         self.assertTrue(e)
2161         c2,c2I=MEDCouplingUMesh.ExtractFromIndexedArrays(f,c,cI)
2162         self.assertTrue(c2I.deltaShiftIndex().isUniform(1))
2163         c2.transformWithIndArr(whichGrp)
2164         splitOfM1=len(grps)*[None]
2165         for grpId,grp in enumerate(grps):
2166             tmp=c2.getIdsEqual(grpId)
2167             splitOfM1[grpId]=tmp
2168             pass
2169         splitOfM1[0].isEqual(DataArrayInt([0,1,2,3,6,8,10,11,12,13]))
2170         splitOfM1[1].isEqual(DataArrayInt([4,5,7,9,14,15]))
2171         pass
2172     pass
2173
2174 unittest.main()