X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FMEDLoader%2FSwig%2FMEDLoaderTest3.py;h=0c6efa4523af782e11d673f0f5ca9863cbb60b8a;hb=7f53ba0ad6eebec56c2936b923ac3ae728f41074;hp=a058b967fe66867b54d4d0e9bc2010bc554624d3;hpb=1e36a6710aab710674e20fbd89f6a9a8f238c023;p=tools%2Fmedcoupling.git diff --git a/src/MEDLoader/Swig/MEDLoaderTest3.py b/src/MEDLoader/Swig/MEDLoaderTest3.py index a058b967f..0c6efa452 100644 --- a/src/MEDLoader/Swig/MEDLoaderTest3.py +++ b/src/MEDLoader/Swig/MEDLoaderTest3.py @@ -1,5 +1,5 @@ # -*- coding: iso-8859-1 -*- -# Copyright (C) 2007-2015 CEA/DEN, EDF R&D +# Copyright (C) 2007-2016 CEA/DEN, EDF R&D # # This library is free software; you can redistribute it and/or # modify it under the terms of the GNU Lesser General Public @@ -1373,6 +1373,119 @@ class MEDLoaderTest3(unittest.TestCase): self.assertTrue(delta.getMaxValue()[0]<1e-12) mm.write(fname,2) + def testBuildInnerBoundaryAlongM1Group4(self): + """ Test case where cells touch the M1 group on some nodes only and not on full egdes (triangle mesh for ex) + """ + coo = DataArrayDouble([0.,0., 1.,0., 2.,0., 3.,0., + 0.,1., 1.,1., 2.,1., 3.,1., + 0.,2., 1.,2., 2.,2., 3.,2.], 12, 2) + conn = [3,0,4,1, 3,1,4,5, + 3,5,9,10, 3,5,10,6, + 3,2,6,7, 3,2,7,3, + 3,4,8,9, 3,4,9,5, + 3,1,5,6, 3,1,6,2, + 3,6,10,11,3,6,11,7] + # Only TRI3: + connI = DataArrayInt() + connI.alloc(13, 1); connI.iota(); connI *= 4 + m2 = MEDCouplingUMesh("2D", 2) + m2.setCoords(coo) + m2.setConnectivity(DataArrayInt(conn), connI) + m2.checkConsistency() + m1, _, _, _, _ = m2.buildDescendingConnectivity() + grpIds = DataArrayInt([9,11]); grpIds.setName("group") + grpIds2 = DataArrayInt([0,1]); grpIds2.setName("group2") + mfu = MEDFileUMesh() + mfu.setMeshAtLevel(0, m2) + mfu.setMeshAtLevel(-1, m1) + mfu.setGroupsAtLevel(-1, [grpIds, grpIds2]) + nNod = m2.getNumberOfNodes() + nodesDup, cells1, cells2 = mfu.buildInnerBoundaryAlongM1Group("group") + m2_bis = mfu.getMeshAtLevel(0) + m2_bis.checkConsistency() + m1_bis = mfu.getMeshAtLevel(-1) + m1_bis.checkConsistency() + self.assertEqual(nNod+2, mfu.getNumberOfNodes()) + self.assertEqual(nNod+2, m2_bis.getNumberOfNodes()) + self.assertEqual(nNod+2, m1_bis.getNumberOfNodes()) + self.assertEqual([6,7], nodesDup.getValues()) + self.assertEqual([2.,1., 3.,1.], m2_bis.getCoords()[nNod:].getValues()) + self.assertEqual(set([3,10,11]), set(cells1.getValues())) + self.assertEqual(set([8,9,4,5]), set(cells2.getValues())) + self.assertEqual([9,11],mfu.getGroupArr(-1,"group").getValues()) + self.assertEqual([23,24],mfu.getGroupArr(-1,"group_dup").getValues()) + self.assertEqual([0,1],mfu.getGroupArr(-1,"group2").getValues()) +# mfu.getMeshAtLevel(0).writeVTK("/tmp/mfu_M0.vtu") + ref0 =[3, 5, 10, 12, 3, 12, 10, 11, 3, 12, 11, 13] + ref1 =[3, 2, 6, 7, 3, 2, 7, 3, 3, 1, 5, 6, 3, 1, 6, 2] + self.assertEqual(ref0,mfu.getMeshAtLevel(0)[[3,10,11]].getNodalConnectivity().getValues()) + self.assertEqual(ref1,mfu.getMeshAtLevel(0)[[4,5,8,9]].getNodalConnectivity().getValues()) + self.assertRaises(InterpKernelException,mfu.getGroup(-1,"group_dup").checkGeoEquivalWith,mfu.getGroup(-1,"group"),2,1e-12) # Grp_dup and Grp are not equal considering connectivity only + mfu.getGroup(-1,"group_dup").checkGeoEquivalWith(mfu.getGroup(-1,"group"),12,1e-12)# Grp_dup and Grp are equal considering connectivity and coordinates + m_bis0 = mfu.getMeshAtLevel(-1) + m_desc, _, _, _, _ = m_bis0.buildDescendingConnectivity() + m_bis0.checkDeepEquivalOnSameNodesWith(mfu.getMeshAtLevel(-1), 2, 9.9999999) + + def testBuildInnerBoundary5(self): + """ Full 3D test with tetras only. In this case a tri from the group is not duplicated because it is made only + of non duplicated nodes. The tri in question is hence not part of the final new "dup" group. """ + coo = DataArrayDouble([200.0, 200.0, 0.0, 200.0, 200.0, 200.0, 200.0, 0.0, 200.0, 200.0, 0.0, 0.0, 0.0, 200.0, 0.0, 0.0, 200.0, 200.0, 0.0, 0.0, 0.0, 0.0, 0.0, + 200.0, 400.0, 200.0, 0.0, 400.0, 200.0, 200.0, 400.0, 0.0, 0.0, 400.0, 0.0, 200.0, 0.0, 100.00000000000016, 200.0, 63.15203310314546, 200.0, 200.0, 134.45205700643342, + 200.0, 200.0, 200.0, 100.00000000000016, 200.0, 63.15203310314546, 0.0, 200.0, 134.45205700643342, 0.0, 200.0, 0.0, 100.00000000000016, 0.0, 63.15203310314546, + 200.0, 0.0, 134.45205700643342, 200.0, 0.0, 200.0, 100.00000000000016, 0.0, 63.15203310314546, 0.0, 0.0, 134.45205700643342, 0.0, 0.0, 200.0, 200.0, 100.02130053568538, + 0.0, 200.0, 100.00938163175135, 200.0, 0.0, 100.02130053568538, 0.0, 0.0, 100.00938163175135, 299.3058739933347, 200.0, 200.0, 400.0, 98.68100542924483, + 200.0, 302.8923433403344, 0.0, 200.0, 302.8923433403344, 200.0, 0.0, 400.0, 100.00000000000016, 0.0, 302.8923433403344, 0.0, 0.0, 400.0, 200.0, 98.55126825835082, + 400.0, 0.0, 100.02162286181577, 99.31624553977466, 99.99999998882231, 200.0, 99.31624576683302, 100.00000010178034, 0.0, 99.31624560596512, 200.0, 100.0050761312483, + 99.31624560612883, 0.0, 100.00507613125338, 200.0, 99.99999995813045, 100.00950673487786, 0.0, 99.99999989928207, 100.0041870621175, 301.29063354383015, + 100.0000000093269, 0.0, 301.29063360689975, 0.0, 100.00957769061164, 140.52853868782435, 99.99999963972768, 100.00509135751312, 297.87779091770784, + 97.16750463405486, 97.18018457127863], 46, 3) + c0 = [14, 45, 31, 21, 42, 14, 37, 38, 20, 44, 14, 39, 36, 41, 44, 14, 5, 25, 12, 13, 14, 38, 36, 44, 41, 14, 21, 20, 24, 44, 14, 38, 25, 41, 19, 14, 37, 38, 44, 41, 14, 16, 27, + 39, 41, 14, 21, 45, 26, 40, 14, 39, 37, 44, 41, 14, 14, 15, 24, 44, 14, 25, 38, 41, 13, 14, 27, 18, 6, 22, 14, 38, 36, 41, 13, 14, 44, 14, 15, 36, 14, 44, 23, 39, 26, 14, + 21,26, 23, 44, 14, 38, 44, 14, 24, 14, 39, 37, 41, 22, 14, 21, 33, 45, 42, 14, 27, 22, 39, 41, 14, 23, 26, 21, 3, 14, 27, 18, 22, 41, 14, 39, 36, 44, 17, 14, 21, 26, 44, 40, + 14, 39, 37, 22, 23, 14, 37, 38, 41, 19, 14, 25, 12, 13, 41, 14, 30, 26, 43, 45, 14, 38, 36, 13, 14, 14, 12, 36, 13, 41, 14, 20, 44, 21, 37, 14, 16, 36, 12, 41, 14, 39, 36, + 17, 16, 14, 44, 20, 24, 38, 14, 27, 16, 12, 41, 14, 26, 15, 17, 44, 14, 19, 18, 41, 37, 14, 40, 45, 26, 15, 14, 37, 38, 19, 20, 14, 17, 15, 26, 2, 14, 39, 36, 16, 41, 14, + 24, 21, 44, 40, 14, 16, 7, 27, 12, 14, 22, 18, 37, 41, 14, 21, 31, 45, 24, 14, 44, 40, 15, 24, 14, 24, 45, 15, 28, 14, 44, 40, 26, 15, 14, 24, 20, 21, 0, 14, 38, 36, 14, + 44, 14, 39, 37, 23, 44, 14, 45, 31, 42, 32, 14, 25, 18, 19, 4, 14, 36, 44, 17, 15, 14, 25, 19, 18, 41, 14, 24, 15, 14, 1, 14, 45, 24, 34, 28, 14, 35, 45, 30, 43, 14, 17, + 44, 39, 26, 14, 44, 23, 21, 37, 14, 30, 45, 29, 15, 14, 45, 35, 33, 43, 14, 30, 15, 26, 45, 14, 31, 21, 0, 24, 14, 33, 35, 32, 10, 14, 29, 45, 34, 28, 14, 32, 45, 34, + 29, 14, 45, 31, 32, 34, 14, 33, 26, 45, 43, 14, 45, 31, 34, 24, 14, 33, 26, 21, 45, 14, 11, 30, 35, 29, 14, 33, 35, 45, 32, 14, 33, 45, 42, 32, 14, 32, 8, 34, 31, 14, + 21, 26, 33, 3, 14, 35, 45, 32, 29, 14, 29, 34, 9, 28, 14, 15, 45, 24, 40, 14, 29, 45, 28, 15, 14, 21, 24, 45, 40, 14, 24, 15, 1, 28, 14, 35, 45, 29, 30, 14, 26, 15, + 30, 2] + cI0 = [0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90, 95, 100, 105, 110, 115, 120, 125, 130, 135, 140, 145, 150, 155, 160, 165, 170, 175, 180, 185, + 190, 195, 200, 205, 210, 215, 220, 225, 230, 235, 240, 245, 250, 255, 260, 265, 270, 275, 280, 285, 290, 295, 300, 305, 310, 315, 320, 325, 330, 335, 340, 345, 350, 355, + 360, 365, 370, 375, 380, 385, 390, 395, 400, 405, 410, 415, 420, 425, 430] + m3 = MEDCouplingUMesh("3D", 3) + m3.setCoords(coo) + m3.setConnectivity(DataArrayInt(c0), DataArrayInt(cI0)) + m3.checkConsistency() + m2, _, _, _, _ = m3.buildDescendingConnectivity() + grpIds = DataArrayInt([36,74]); grpIds.setName("group") + mfu = MEDFileUMesh() + mfu.setMeshAtLevel(0, m3) + mfu.setMeshAtLevel(-1, m2) + grpIds3D = DataArrayInt([0,1]); grpIds3D.setName("group_3d") + mfu.setGroupsAtLevel(0, [grpIds3D]) # just to check preservation of 3D group + mfu.setGroupsAtLevel(-1, [grpIds]) + nNod = m3.getNumberOfNodes() + nodesDup, cells1, cells2 = mfu.buildInnerBoundaryAlongM1Group("group") + m3_bis = mfu.getMeshAtLevel(0) + m3_bis.checkConsistency() + m2_bis = mfu.getMeshAtLevel(-1) + m2_bis.checkConsistency() + self.assertEqual(nNod+1, mfu.getNumberOfNodes()) + self.assertEqual(nNod+1, m3_bis.getNumberOfNodes()) + self.assertEqual(nNod+1, m2_bis.getNumberOfNodes()) + self.assertEqual([3], nodesDup.getValues()) + self.assertEqual(m3_bis.getCoords()[3].getValues(), m3_bis.getCoords()[nNod:].getValues()) + self.assertEqual(set([22]), set(cells1.getValues())) + self.assertEqual(set([77]), set(cells2.getValues())) + self.assertEqual([36,74],mfu.getGroupArr(-1,"group").getValues()) + self.assertEqual([0,1],mfu.getGroupArr(0,"group_3d").getValues()) + self.assertEqual([213],mfu.getGroupArr(-1,"group_dup").getValues()) # here only one cell has been duplicated + m_bis0 = mfu.getMeshAtLevel(-1) + m_desc, _, _, _, _ = m_bis0.buildDescendingConnectivity() + m_bis0.checkDeepEquivalOnSameNodesWith(mfu.getMeshAtLevel(-1), 2, 9.9999999) + pass + def testBasicConstructors(self): fname="Pyfile18.med" m=MEDFileMesh.New(fname) @@ -2034,25 +2147,25 @@ class MEDLoaderTest3(unittest.TestCase): # mm=MEDFileUMesh() mm.setMeshAtLevel(0,m) - self.assertIn(mm.getHeapMemorySize(),xrange(3889-100,3889+100+10*strMulFac)) + self.assertIn(mm.getHeapMemorySize(),xrange(3889-100,4225+100+10*strMulFac)) ff=MEDFileField1TS() ff.setFieldNoProfileSBT(f) - self.assertIn(ff.getHeapMemorySize(),xrange(771-40,771+21+(4+1)*strMulFac)) + self.assertIn(ff.getHeapMemorySize(),xrange(771-40,871+21+(4+1)*strMulFac)) # fff=MEDFileFieldMultiTS() fff.appendFieldNoProfileSBT(f) - self.assertIn(fff.getHeapMemorySize(),xrange(815-50,815+30+(6+2)*strMulFac)) + self.assertIn(fff.getHeapMemorySize(),xrange(815-50,915+30+(6+2)*strMulFac)) f.setTime(1.,0,-1) fff.appendFieldNoProfileSBT(f) - self.assertIn(fff.getHeapMemorySize(),xrange(1594-90,1594+50+(10+1)*strMulFac)) - self.assertIn(fff[0,-1].getHeapMemorySize(),xrange(771-40,771+20+(4+1)*strMulFac)) + self.assertIn(fff.getHeapMemorySize(),xrange(1594-90,1794+50+(10+1)*strMulFac)) + self.assertIn(fff[0,-1].getHeapMemorySize(),xrange(771-40,871+20+(4+1)*strMulFac)) f2=f[:50] f2.setTime(2.,1,-1) pfl=DataArrayInt.Range(0,50,1) ; pfl.setName("pfl") fff.appendFieldProfile(f2,mm,0,pfl) - self.assertIn(fff.getHeapMemorySize(),xrange(2348-130,2348+100+(10+2)*strMulFac)) + self.assertIn(fff.getHeapMemorySize(),xrange(2348-130,2608+100+(10+2)*strMulFac)) self.assertIn(fff.getProfile("pfl").getHeapMemorySize(),xrange(204-10,204+10+2*strMulFac)) - self.assertIn(fff[1,-1].getHeapMemorySize(),xrange(738-50,738+30+4*strMulFac)) + self.assertIn(fff[1,-1].getHeapMemorySize(),xrange(738-50,838+30+4*strMulFac)) pass def testCurveLinearMesh1(self): @@ -2465,8 +2578,7 @@ class MEDLoaderTest3(unittest.TestCase): def testInt32InMEDFileFieldStar1(self): fname="Pyfile63.med" f1=MEDLoaderDataForTest.buildVecFieldOnCells_1(); - arr=f1.getArray().convertToIntArr() - f1.setArray(None) + f1=f1.convertToIntField() m1=f1.getMesh() mm1=MEDFileUMesh.New() mm1.setCoords(m1.getCoords()) @@ -2474,19 +2586,17 @@ class MEDLoaderTest3(unittest.TestCase): mm1.setName(m1.getName()) mm1.write(fname,2) ff1=MEDFileIntField1TS() - ff1.setFieldNoProfileSBT(f1,arr) - a,b=ff1.getFieldOnMeshAtLevel(0,ON_CELLS,mm1) - self.assertEqual(b.getInfoOnComponents(),['power [MW/m^3]','density [g/cm^3]','temperature [K]']) - self.assertTrue(b.isEqual(arr)) + ff1.setFieldNoProfileSBT(f1) + a=ff1.getFieldOnMeshAtLevel(0,ON_CELLS,mm1) + self.assertEqual(a.getArray().getInfoOnComponents(),['power [MW/m^3]','density [g/cm^3]','temperature [K]']) self.assertTrue(a.isEqual(f1,1e-12,1e-12)) ff1.write(fname,0) ff2=MEDFileAnyTypeField1TS.New(fname) self.assertEqual(ff2.getName(),"VectorFieldOnCells") self.assertEqual(ff2.getTime(),[0,1,2.0]) self.assertTrue(isinstance(ff2,MEDFileIntField1TS)) - a,b=ff1.getFieldOnMeshAtLevel(0,ON_CELLS,mm1) - self.assertEqual(b.getInfoOnComponents(),['power [MW/m^3]','density [g/cm^3]','temperature [K]']) - self.assertTrue(b.isEqual(arr)) + a=ff1.getFieldOnMeshAtLevel(ON_CELLS,0,mm1) + self.assertEqual(a.getArray().getInfoOnComponents(),['power [MW/m^3]','density [g/cm^3]','temperature [K]']) self.assertTrue(a.isEqual(f1,1e-12,1e-12)) ff2.setTime(1,2,3.) c=ff2.getUndergroundDataArray() ; c*=2 @@ -2495,42 +2605,40 @@ class MEDLoaderTest3(unittest.TestCase): self.assertEqual(ffs1.getTimeSteps(),[(0, 1, 2.0), (1, 2, 3.0)]) self.assertEqual(len(ffs1),2) self.assertTrue(isinstance(ffs1,MEDFileIntFieldMultiTS)) - a,b=ffs1[2.].getFieldOnMeshAtLevel(0,ON_CELLS,mm1) - self.assertTrue(b.isEqual(arr)) + a=ffs1[2.].getFieldOnMeshAtLevel(ON_CELLS,0,mm1) self.assertTrue(a.isEqual(f1,1e-12,1e-12)) - a,b=ffs1[2.].getFieldOnMeshAtLevel(0,ON_CELLS,mm1) - self.assertTrue(b.isEqual(arr)) + a=ffs1.getFieldOnMeshAtLevel(ON_CELLS,0,1,0,mm1) self.assertTrue(a.isEqual(f1,1e-12,1e-12)) it=ffs1.__iter__() ; it.next() ; ff2bis=it.next() - a,b=ff2bis.getFieldOnMeshAtLevel(0,ON_CELLS,mm1) - self.assertTrue(b.isEqual(2*arr)) - f1.setTime(3.,1,2) - self.assertTrue(a.isEqual(f1,1e-12,1e-12)) + a=ff2bis.getFieldOnMeshAtLevel(0,ON_CELLS,mm1) + self.assertTrue(a.getArray().isEqual(2*f1.getArray())) + f1.setTime(3.,1,2) ; f1.getArray()[:]*=2 + self.assertTrue(a.isEqual(f1,1e-12,1e-12)) ; f1.getArray()[:]/=2 bc=DataArrayInt(6,3) ; bc[:]=0 ; bc.setInfoOnComponents(['power [MW/m^3]','density [g/cm^3]','temperature [K]']) for it in ffs1: - a,b=it.getFieldOnMeshAtLevel(0,ON_CELLS,mm1) - bc+=b + a=it.getFieldOnMeshAtLevel(ON_CELLS,0,mm1) + bc+=a.getArray() pass - self.assertTrue(bc.isEqual(3*arr)) - nf1=MEDCouplingFieldDouble(ON_NODES) + self.assertTrue(bc.isEqual(3*f1.getArray())) + nf1=MEDCouplingFieldInt(ON_NODES) nf1.setTime(9.,10,-1) nf1.setMesh(f1.getMesh()) narr=DataArrayInt(12,2) ; narr.setInfoOnComponents(["aa [u1]","bbbvv [ppp]"]) ; narr[:,0]=range(12) ; narr[:,1]=2*narr[:,0] - nf1.setName("VectorFieldOnNodes") + nf1.setName("VectorFieldOnNodes") ; nf1.setArray(narr) nff1=MEDFileIntField1TS.New() - nff1.setFieldNoProfileSBT(nf1,narr) + nff1.setFieldNoProfileSBT(nf1) self.assertEqual(nff1.getInfo(),('aa [u1]','bbbvv [ppp]')) self.assertEqual(nff1.getTime(),[10,-1,9.0]) nff1.write(fname,0) # - nf2=MEDCouplingFieldDouble(ON_NODES) + nf2=MEDCouplingFieldInt(ON_NODES) nf2.setTime(19.,20,-11) nf2.setMesh(f1.getMesh()) narr2=DataArrayInt(8,2) ; narr.setInfoOnComponents(["aapfl [u1]","bbbvvpfl [ppp]"]) ; narr2[:,0]=range(8) ; narr2[:,0]+=10 ; narr2[:,1]=3*narr2[:,0] - nf2.setName("VectorFieldOnNodesPfl") ; narr2.setName(nf2.getName()) + nf2.setName("VectorFieldOnNodesPfl") ; narr2.setName(nf2.getName()) ; nf2.setArray(narr2) nff2=MEDFileIntField1TS.New() npfl=DataArrayInt([1,2,4,5,6,7,10,11]) ; npfl.setName("npfl") - nff2.setFieldProfile(nf2,narr2,mm1,0,npfl) + nff2.setFieldProfile(nf2,mm1,0,npfl) nff2.getFieldWithProfile(ON_NODES,0,mm1) a,b=nff2.getFieldWithProfile(ON_NODES,0,mm1) ; b.setName(npfl.getName()) self.assertTrue(b.isEqual(npfl)) @@ -2557,8 +2665,8 @@ class MEDLoaderTest3(unittest.TestCase): self.assertTrue(isinstance(ffs[2],MEDFileFieldMultiTS)) self.assertTrue(isinstance(ffs[3],MEDFileIntFieldMultiTS)) # - self.assertTrue(fs["VectorFieldOnCells"][0].getUndergroundDataArray().isEqualWithoutConsideringStr(arr)) - self.assertTrue(fs["VectorFieldOnCells"][1,2].getUndergroundDataArray().isEqualWithoutConsideringStr(2*arr)) + self.assertTrue(fs["VectorFieldOnCells"][0].getUndergroundDataArray().isEqualWithoutConsideringStr(f1.getArray())) + self.assertTrue(fs["VectorFieldOnCells"][1,2].getUndergroundDataArray().isEqualWithoutConsideringStr(2*f1.getArray())) self.assertTrue(fs["VectorFieldOnNodesPfl"][0].getUndergroundDataArray().isEqualWithoutConsideringStr(narr2)) self.assertTrue(fs["VectorFieldOnNodes"][9.].getUndergroundDataArray().isEqualWithoutConsideringStr(narr)) self.assertTrue(fs["VectorFieldOnNodesDouble"][29.].getUndergroundDataArray().isEqualWithoutConsideringStr(f1.getMesh().getCoords(),1e-12)) @@ -2969,7 +3077,7 @@ class MEDLoaderTest3(unittest.TestCase): self.assertTrue(not ff0.getUndergroundDataArray().isAllocated()) self.assertEqual(ff0.getUndergroundDataArray().getInfoOnComponents(),['X [km]','YY [mm]']) heap_memory_ref=ff0.getHeapMemorySize() - self.assertIn(heap_memory_ref,xrange(182,298+2*strMulFac)) + self.assertIn(heap_memory_ref,xrange(182,465+2*strMulFac)) ff0.loadArrays() ## arr=DataArrayDouble(140) ; arr.iota() ; arr.rearrange(2) self.assertTrue(ff0.getUndergroundDataArray().isEqualWithoutConsideringStr(arr,1e-14)) @@ -2978,7 +3086,7 @@ class MEDLoaderTest3(unittest.TestCase): ff0=MEDFileField1TS(fname,"FieldCellPfl",False) self.assertEqual(ff0.getUndergroundDataArray().getInfoOnComponents(),["XX [pm]","YYY [hm]"]) heap_memory_ref=ff0.getHeapMemorySize() - self.assertIn(heap_memory_ref,xrange(350,415+6*strMulFac)) + self.assertIn(heap_memory_ref,xrange(350,520+6*strMulFac)) ff0.loadArrays() ## arr=DataArrayDouble(100) ; arr.iota() ; arr.rearrange(2) self.assertTrue(ff0.getUndergroundDataArray().isEqualWithoutConsideringStr(arr,1e-14)) @@ -2996,7 +3104,7 @@ class MEDLoaderTest3(unittest.TestCase): self.assertEqual(ff0.getUndergroundDataArray().getIJ(30,1),5.5) self.assertTrue(not ff0.getUndergroundDataArray().isEqualWithoutConsideringStr(arr,1e-14)) heap_memory_ref=ff0.getHeapMemorySize() - self.assertIn(heap_memory_ref,xrange(1100,1215+2*strMulFac)) + self.assertIn(heap_memory_ref,xrange(1100,1384+2*strMulFac)) ff0.unloadArrays() hmd=ff0.getHeapMemorySize()-heap_memory_ref self.assertEqual(hmd,-800) # -50*8*2 @@ -3005,7 +3113,7 @@ class MEDLoaderTest3(unittest.TestCase): # ff0=MEDFileField1TS(fname,"FieldCellPfl",-1,-1,False) heap_memory_ref=ff0.getHeapMemorySize() - self.assertIn(heap_memory_ref,xrange(299,415+6*strMulFac)) + self.assertIn(heap_memory_ref,xrange(299,520+6*strMulFac)) ff0.loadArrays() ## self.assertTrue(ff0.getUndergroundDataArray().isEqualWithoutConsideringStr(arr,1e-14)) self.assertEqual(ff0.getHeapMemorySize()-heap_memory_ref,50*8*2) @@ -3022,14 +3130,14 @@ class MEDLoaderTest3(unittest.TestCase): # ff0=MEDFileAnyTypeFieldMultiTS.New(fname,fieldName,False) heap_memory_ref=ff0.getHeapMemorySize() - self.assertIn(heap_memory_ref,xrange(5536,5956+(80+26)*strMulFac)) + self.assertIn(heap_memory_ref,xrange(5536,8212+(80+26)*strMulFac)) ff0.loadArrays() self.assertEqual(ff0.getHeapMemorySize()-heap_memory_ref,20*70*8*2) del ff0 # ffs=MEDFileFields(fname,False) heap_memory_ref=ffs.getHeapMemorySize() - self.assertIn(heap_memory_ref,xrange(5335,6687+(80+50)*strMulFac)) + self.assertIn(heap_memory_ref,xrange(5335,9031+(80+50)*strMulFac)) ffs.loadArrays() self.assertEqual(ffs.getHeapMemorySize()-heap_memory_ref,20*70*8*2+70*8*2+50*8*2) pass @@ -4852,7 +4960,466 @@ class MEDLoaderTest3(unittest.TestCase): self.assertTrue(mm.getHiddenCppPointer()==mm2.getHiddenCppPointer()) # optimization pass + def testCheckCoherency(self): + m2 = MEDCouplingUMesh("2d", 2) + m2.setCoords(DataArrayDouble([(0.0, 1.0)] * 4, 4,2)) # whatever + m2.setConnectivity(DataArrayInt([NORM_TRI3, 0,1,2,NORM_TRI3, 1,2,3]), DataArrayInt(([0,4,8]))) + m1 , _, _ , _, _ = m2.buildDescendingConnectivity() + mum = MEDFileUMesh() + mum.setMeshAtLevel(0, m2) + mum.setMeshAtLevel(-1, m1) + mum.checkConsistency() + mum2 = mum.deepCopy() + + # Nodes + arr = DataArrayInt([2]*4) + mum.setFamilyFieldArr(1, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + arr = DataArrayInt([2]*4) + mum.setRenumFieldArr(1, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + mum.setRenumFieldArr(1, DataArrayInt([2]*4)) + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + arr = DataArrayAsciiChar(['tutu x']*4) + mum.setNameFieldAtLevel(1, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + + # 2D + mum=mum2; mum2=mum.deepCopy(); + arr = DataArrayInt([2]*2) + mum.setFamilyFieldArr(0, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + arr = DataArrayInt([2]*2) + mum.setRenumFieldArr(0, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + mum.setRenumFieldArr(0, DataArrayInt([2]*2)) + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + arr = DataArrayAsciiChar(['tutu x']*2) + mum.setNameFieldAtLevel(0, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + + # 1D + mum=mum2; mum2=mum.deepCopy(); + arr = DataArrayInt([2]*5) + mum.setFamilyFieldArr(-1, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + arr = DataArrayInt([2]*5) + mum.setRenumFieldArr(-1, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + mum.setRenumFieldArr(-1, DataArrayInt([2]*5)) + self.assertRaises(InterpKernelException, mum.checkConsistency) + mum=mum2; mum2=mum.deepCopy(); + arr = DataArrayAsciiChar(['tutu x']*5) + mum.setNameFieldAtLevel(-1, arr); arr.reAlloc(35); + self.assertRaises(InterpKernelException, mum.checkConsistency) + + def testCheckSMESHConsistency(self): + m2 = MEDCouplingUMesh("2d", 2) + m2.setCoords(DataArrayDouble([(0.0, 1.0)] * 4, 4,2)) # whatever + m2.setConnectivity(DataArrayInt([NORM_TRI3, 0,1,2,NORM_TRI3, 1,2,3]), DataArrayInt(([0,4,8]))) + m1 , _, _ , _, _ = m2.buildDescendingConnectivity() + mum = MEDFileUMesh() + mum.setMeshAtLevel(0, m2) + mum.setMeshAtLevel(-1, m1) + mum.checkConsistency() + mum.checkSMESHConsistency() + n2 = DataArrayInt(m2.getNumberOfCells(), 1); n2.iota(1) + n1 = DataArrayInt(m1.getNumberOfCells(), 1); n1.iota(1) + mum.setRenumFieldArr(0, n2) + mum.setRenumFieldArr(-1, n1) + self.assertRaises(InterpKernelException, mum.checkSMESHConsistency) + mum.setRenumFieldArr(-1, n1+100) + mum.checkSMESHConsistency() + pass + + def testClearNodeAndCellNumbers(self): + m2 = MEDCouplingUMesh("2d", 2) + m2.setCoords(DataArrayDouble([(0.0, 1.0)] * 4, 4,2)) # whatever + m2.setConnectivity(DataArrayInt([NORM_TRI3, 0,1,2,NORM_TRI3, 1,2,3]), DataArrayInt(([0,4,8]))) + m1 , _, _ , _, _ = m2.buildDescendingConnectivity() + mum = MEDFileUMesh() + mum.setMeshAtLevel(0, m2) + mum.setMeshAtLevel(-1, m1) + mum.checkConsistency() + n2 = DataArrayInt(m2.getNumberOfCells(), 1); n2.iota(1) + n1 = DataArrayInt(m1.getNumberOfCells(), 1); n1.iota(1) + mum.setRenumFieldArr(0, n2) + mum.setRenumFieldArr(-1, n1) + mum.clearNodeAndCellNumbers() + mum.checkSMESHConsistency() + pass + + def testCMeshSetFamilyFieldArrNull(self): + meshName="mesh" + fname="Pyfile99.med" + arrX=DataArrayDouble([0,1,2,3]) + arrY=DataArrayDouble([0,1,2]) + m=MEDCouplingCMesh() ; m.setCoords(arrX,arrY) ; m.setName(meshName) + mm=MEDFileCMesh() ; mm.setMesh(m) + famCellIds=DataArrayInt([0,-2,-2,-1,-2,0]) + famNodeIds=DataArrayInt([0,0,0,3,4,1,2,7,2,1,0,0]) + mm.setFamilyFieldArr(0,famCellIds) + mm.setFamilyFieldArr(1,famNodeIds) + mm.write(fname,2) + mm=MEDFileMesh.New(fname) + self.assertTrue(mm.getFamilyFieldAtLevel(0) is not None) + self.assertTrue(mm.getFamilyFieldAtLevel(1) is not None) + mm.setFamilyFieldArr(0,None)#<- bug was here + mm.setFamilyFieldArr(1,None)#<- bug was here + self.assertTrue(mm.getFamilyFieldAtLevel(0) is None) + self.assertTrue(mm.getFamilyFieldAtLevel(1) is None) + mm3=mm.deepCopy() + self.assertTrue(mm3.getFamilyFieldAtLevel(0) is None) + self.assertTrue(mm3.getFamilyFieldAtLevel(1) is None) + mm.write(fname,2) + mm2=MEDFileMesh.New(fname) + self.assertTrue(mm2.getFamilyFieldAtLevel(0) is None) + self.assertTrue(mm2.getFamilyFieldAtLevel(1) is None) + pass + + def testAppendFieldProfileOnIntField(self): + fname="Pyfile100.med" + arrX=DataArrayDouble([0,1,2,3]) + arrY=DataArrayDouble([0,1,2]) + mesh=MEDCouplingCMesh() ; mesh.setCoords(arrX,arrY) ; mesh.setName("Mesh") + mm=MEDFileCMesh() + mm.setMesh(mesh) + # + fmts=MEDFileIntFieldMultiTS() + pflName="PFL" + pfl=DataArrayInt([1,3,5]) ; pfl.setName(pflName) + f=MEDCouplingFieldInt(ON_CELLS) ; f.setMesh(mesh) + fieldName="FieldOnCell" + f.setTime(1.2,1,1) ; f.setName(fieldName) + arr=DataArrayInt([101,102,103]) ; f.setArray(arr) + fmts.appendFieldProfile(f,mm,0,pfl) + # + mm.write(fname,2) + fmts.write(fname,0) + # + mm=MEDFileMesh.New(fname) + fmts=MEDFileAnyTypeFieldMultiTS.New(fname) + self.assertTrue(isinstance(fmts,MEDFileIntFieldMultiTS)) + self.assertEqual(fmts.getName(),fieldName) + self.assertEqual(len(fmts),1) + f1ts=fmts[0] + ftest,pfltest=f1ts.getFieldWithProfile(ON_CELLS,0,mm) + self.assertEqual(pfltest.getName(),pflName) + self.assertEqual(ftest.getName(),fieldName) + self.assertTrue(ftest.isEqualWithoutConsideringStr(arr)) + ftest2=f1ts.getFieldOnMeshAtLevel(ON_CELLS,0,mm) + self.assertTrue(ftest2.getArray().isEqualWithoutConsideringStr(arr)) + self.assertEqual(ftest2.getTime(),f.getTime()) + self.assertEqual(ftest2.getMesh().getNumberOfCells(),len(arr)) + pass + + def testMEDFileFieldEasyField1(self): + """Check for all spatial discretization of field (cells,nodes,elno,gauss) for double field that all is OK. Here no profile and only top level is considered.""" + ## Basic test on cells on top level + fname="Pyfile101.med" + fieldName="field1" + mm=MEDFileUMesh() + coo=DataArrayDouble([(3,2,1),(8,7,6),(5,9,10)]) + m=MEDCouplingUMesh("mesh",2) ; m.setCoords(coo) + m.allocateCells() + m.insertNextCell(NORM_TRI3,[0,1,2]) + m.insertNextCell(NORM_TRI3,[3,4,5]) + m.insertNextCell(NORM_TRI3,[6,7,8]) + m.insertNextCell(NORM_TRI3,[9,10,11]) + m.insertNextCell(NORM_QUAD4,[100,101,102,103]) + m.insertNextCell(NORM_QUAD4,[104,105,106,107]) + mm[0]=m + mm.write(fname,2) + arr0=DataArrayDouble([10,11,12,13,100,101]) + f=MEDCouplingFieldDouble(ON_CELLS) ; f.setArray(arr0) ; f.setMesh(m) + f.setName(fieldName) ; f.setTime(2.,6,7) + f0=f.deepCopy() + ff=MEDFileFieldMultiTS() ; ff.appendFieldNoProfileSBT(f) + ff.write(fname,0) + arr2=arr0+1000 ; f.setArray(arr2) + f.setTime(3.,8,9) ; ff=MEDFileField1TS() ; ff.setFieldNoProfileSBT(f) + ff.write(fname,0) + f1=f.deepCopy() + ## + mm=MEDFileMesh.New(fname) + f1ts=MEDFileField1TS(fname,fieldName,6,7) + ftst0=f1ts.field(mm) + self.assertTrue(f0.isEqual(ftst0,1e-12,1e-12)) + f1ts=MEDFileField1TS(fname,fieldName,8,9) + ftst1=f1ts.field(mm) + self.assertTrue(f1.isEqual(ftst1,1e-12,1e-12)) + fmts=MEDFileFieldMultiTS(fname,fieldName) + self.assertTrue(f1.isEqual(fmts.field(8,9,mm),1e-12,1e-12)) + ## Basic test on nodes on top level + f2=MEDCouplingFieldDouble(ON_NODES) ; arr2=DataArrayDouble([200,201,202]) ; arr2.setInfoOnComponent(0,"tutu") ; f2.setArray(arr2) ; f2.setMesh(m) ; f2.setTime(22.,23,24) + f2.setName(fieldName) + mm.write(fname,2) + ff=MEDFileField1TS() ; ff.setFieldNoProfileSBT(f2) ; ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) + f1ts=MEDFileField1TS(fname,fieldName,23,24) + self.assertTrue(f2.isEqual(f1ts.field(mm),1e-12,1e-12)) + fmts=MEDFileFieldMultiTS(fname,fieldName) + self.assertTrue(f2.isEqual(fmts.field(23,24,mm),1e-12,1e-12)) + ## Node on elements + f3=MEDCouplingFieldDouble(ON_GAUSS_NE) ; f3.setMesh(m) ; arr3=DataArrayDouble([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19]) ; f3.setArray(arr3) ; f3.setTime(0.5,2,3) + f3.setName(fieldName) ; f3.checkConsistencyLight() + mm.write(fname,2) ; ff=MEDFileField1TS() ; ff.setFieldNoProfileSBT(f3) ; ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) + f1ts=MEDFileField1TS(fname,fieldName,2,3) + self.assertTrue(f3.isEqual(f1ts.field(mm),1e-12,1e-12)) + ## Gauss + f4=MEDCouplingFieldDouble(ON_GAUSS_PT) ; f4.setMesh(m) ; f4.setName(fieldName) + f4.setGaussLocalizationOnType(NORM_TRI3,[0.,0.,1.,0.,1.,1.],[0.1,0.1, 0.2,0.2, 0.3,0.3, 0.4,0.4, 0.5,0.5],[0.2,0.3,0.1,0.05,0.35]) + f4.setGaussLocalizationOnType(NORM_QUAD4,[0.,0.,1.,0.,1.,1.,0.,1.],[0.3,0.4, 0.6,0.7],[0.7,0.3]) ; f4.setTime(0.25,4,5) + arr4=DataArrayDouble([0,1,2,3,4 ,10,11,12,13,14, 20,21,22,23,24, 30,31,32,33,34, 45,46, 55,56]) ; arr4.setInfoOnComponent(0,"abc") ; f4.setArray(arr4) + f4.checkConsistencyLight() + mm.write(fname,2) ; ff=MEDFileField1TS() ; ff.setFieldNoProfileSBT(f4) ; ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) + f1ts=MEDFileField1TS(fname,fieldName,4,5) + self.assertTrue(f4.isEqual(f1ts.field(mm),1e-12,1e-12)) + pass + + def testMEDFileFieldEasyField2(self): + """Same thantestMEDFileFieldEasyField1 except that here intfields are considered. + Check for all spatial discretization of field (cells,nodes,elno,gauss) for int field that all is OK. Here no profile and only top level is considered.""" + ## Basic test on cells on top level + fname="Pyfile102.med" + fieldName="field1" + mm=MEDFileUMesh() + coo=DataArrayDouble([(3,2,1),(8,7,6),(5,9,10)]) + m=MEDCouplingUMesh("mesh",2) ; m.setCoords(coo) + m.allocateCells() + m.insertNextCell(NORM_TRI3,[0,1,2]) + m.insertNextCell(NORM_TRI3,[3,4,5]) + m.insertNextCell(NORM_TRI3,[6,7,8]) + m.insertNextCell(NORM_TRI3,[9,10,11]) + m.insertNextCell(NORM_QUAD4,[100,101,102,103]) + m.insertNextCell(NORM_QUAD4,[104,105,106,107]) + mm[0]=m + mm.write(fname,2) + arr0=DataArrayInt([10,11,12,13,100,101]) + f=MEDCouplingFieldInt(ON_CELLS) ; f.setArray(arr0) ; f.setMesh(m) + f.setName(fieldName) ; f.setTime(2.,6,7) + f0=f.deepCopy() + ff=MEDFileIntFieldMultiTS() ; ff.appendFieldNoProfileSBT(f) + ff.write(fname,0) + arr2=arr0+1000 ; f.setArray(arr2) + f.setTime(3.,8,9) ; ff=MEDFileIntField1TS() ; ff.setFieldNoProfileSBT(f) + ff.write(fname,0) + f1=f.deepCopy() + ## + mm=MEDFileMesh.New(fname) + f1ts=MEDFileIntField1TS(fname,fieldName,6,7) + ftst0=f1ts.field(mm) + self.assertTrue(f0.isEqual(ftst0,1e-12,1e-12)) + f1ts=MEDFileIntField1TS(fname,fieldName,8,9) + ftst1=f1ts.field(mm) + self.assertTrue(f1.isEqual(ftst1,1e-12,1e-12)) + fmts=MEDFileIntFieldMultiTS(fname,fieldName) + self.assertTrue(f1.isEqual(fmts.field(8,9,mm),1e-12,1e-12)) + ## Basic test on nodes on top level + f2=MEDCouplingFieldInt(ON_NODES) ; arr2=DataArrayInt([200,201,202]) ; arr2.setInfoOnComponent(0,"tutu") ; f2.setArray(arr2) ; f2.setMesh(m) ; f2.setTime(22.,23,24) + f2.setName(fieldName) + mm.write(fname,2) + ff=MEDFileIntField1TS() ; ff.setFieldNoProfileSBT(f2) ; ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) + f1ts=MEDFileIntField1TS(fname,fieldName,23,24) + self.assertTrue(f2.isEqual(f1ts.field(mm),1e-12,1e-12)) + fmts=MEDFileIntFieldMultiTS(fname,fieldName) + self.assertTrue(f2.isEqual(fmts.field(23,24,mm),1e-12,1e-12)) + ## Node on elements + f3=MEDCouplingFieldInt(ON_GAUSS_NE) ; f3.setMesh(m) ; arr3=DataArrayInt([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19]) ; f3.setArray(arr3) ; f3.setTime(0.5,2,3) + f3.setName(fieldName) ; f3.checkConsistencyLight() + mm.write(fname,2) ; ff=MEDFileIntField1TS() ; ff.setFieldNoProfileSBT(f3) ; ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) + f1ts=MEDFileIntField1TS(fname,fieldName,2,3) + self.assertTrue(f3.isEqual(f1ts.field(mm),1e-12,1e-12)) + ## Gauss + f4=MEDCouplingFieldInt(ON_GAUSS_PT) ; f4.setMesh(m) ; f4.setName(fieldName) + f4.setGaussLocalizationOnType(NORM_TRI3,[0.,0.,1.,0.,1.,1.],[0.1,0.1, 0.2,0.2, 0.3,0.3, 0.4,0.4, 0.5,0.5],[0.2,0.3,0.1,0.05,0.35]) + f4.setGaussLocalizationOnType(NORM_QUAD4,[0.,0.,1.,0.,1.,1.,0.,1.],[0.3,0.4, 0.6,0.7],[0.7,0.3]) ; f4.setTime(0.25,4,5) + arr4=DataArrayInt([0,1,2,3,4 ,10,11,12,13,14, 20,21,22,23,24, 30,31,32,33,34, 45,46, 55,56]) ; arr4.setInfoOnComponent(0,"abc") ; f4.setArray(arr4) + f4.checkConsistencyLight() + mm.write(fname,2) ; ff=MEDFileIntField1TS() ; ff.setFieldNoProfileSBT(f4) ; ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) + f1ts=MEDFileIntField1TS(fname,fieldName,4,5) + self.assertTrue(f4.isEqual(f1ts.field(mm),1e-12,1e-12)) + pass + + def testMEDFileFieldEasyField3(self): + """Here a multi level mesh. And field on cells lying on different level of this mesh. Show how "field" method deal with that. Here on field double are considered.""" + fname="Pyfile103.med" + fieldName="field1" + mm=MEDFileUMesh() + coo=DataArrayDouble([(3,2,1),(8,7,6),(5,9,10)]) + m=MEDCouplingUMesh("mesh",2) ; m.setCoords(coo) + m.allocateCells() + m.insertNextCell(NORM_TRI3,[0,1,2]) + m.insertNextCell(NORM_TRI3,[3,4,5]) + m.insertNextCell(NORM_TRI3,[6,7,8]) + m.insertNextCell(NORM_TRI3,[9,10,11]) + m.insertNextCell(NORM_QUAD4,[100,101,102,103]) + m.insertNextCell(NORM_QUAD4,[104,105,106,107]) + mm[-1]=m + m0=MEDCouplingUMesh("mesh",3) ; m0.setCoords(coo) + m0.allocateCells() + m0.insertNextCell(NORM_TETRA4,[3,2,5,0]) + m0.insertNextCell(NORM_TETRA4,[7,6,3,2]) + mm[0]=m0 + mm.write(fname,2) + # start slowly + f1=MEDCouplingFieldDouble(ON_CELLS) ; f1.setName(fieldName) ; f1.setArray(DataArrayDouble([(0,100),(1,101)])) ; f1.setMesh(mm[0]) ; f1.setTime(4.,1,2) + f1ts=MEDFileField1TS() ; f1ts.setFieldNoProfileSBT(f1) ; f1ts.write(fname,0) + # + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileField1TS(fname,fieldName,1,2) + self.assertTrue(f1.isEqual(f1ts.field(mm),1e-12,1e-12)) + # here f1 lying on level -1 not 0 check if "field" method detect it ! + f1=MEDCouplingFieldDouble(ON_CELLS) ; f1.setName(fieldName) ; f1.setArray(DataArrayDouble([(0,100),(1,101),(0,100),(1,101),(0,100),(1,101)])) + f1.setMesh(mm[-1]) # -1 is very important + f1.setTime(16.,3,4) + f1.checkConsistencyLight() + mm.write(fname,2) + f1ts=MEDFileField1TS() ; f1ts.setFieldNoProfileSBT(f1) ; f1ts.write(fname,0) + # + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileField1TS(fname,fieldName,3,4) + self.assertTrue(f1.isEqual(f1ts.field(mm),1e-12,1e-12)) + # nodes on elements + f3=MEDCouplingFieldDouble(ON_GAUSS_NE) + f3.setMesh(mm[-1]) # this line is important + arr3=DataArrayDouble([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19]) ; f3.setArray(arr3) ; f3.setTime(0.5,2,3) + f3.setName(fieldName) ; f3.checkConsistencyLight() + mm.write(fname,2) ; ff=MEDFileField1TS() ; ff.setFieldNoProfileSBT(f3) ; ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileField1TS(fname,fieldName,2,3) + self.assertTrue(f3.isEqual(f1ts.field(mm),1e-12,1e-12)) + # gauss + f4=MEDCouplingFieldDouble(ON_GAUSS_PT) + f4.setMesh(mm[-1]) # this line is important + f4.setName(fieldName) + f4.setGaussLocalizationOnType(NORM_TRI3,[0.,0.,1.,0.,1.,1.],[0.1,0.1, 0.2,0.2, 0.3,0.3, 0.4,0.4, 0.5,0.5],[0.2,0.3,0.1,0.05,0.35]) + f4.setGaussLocalizationOnType(NORM_QUAD4,[0.,0.,1.,0.,1.,1.,0.,1.],[0.3,0.4, 0.6,0.7],[0.7,0.3]) ; f4.setTime(0.25,4,5) + arr4=DataArrayDouble([0,1,2,3,4 ,10,11,12,13,14, 20,21,22,23,24, 30,31,32,33,34, 45,46, 55,56]) ; arr4.setInfoOnComponent(0,"abc") ; f4.setArray(arr4) + f4.checkConsistencyLight() + mm.write(fname,2) ; ff=MEDFileField1TS() ; ff.setFieldNoProfileSBT(f4) ; ff.write(fname,0) + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileField1TS(fname,fieldName,4,5) + self.assertTrue(f4.isEqual(f1ts.field(mm),1e-12,1e-12)) + pass + + def testMEDFileFieldEasyField4(self): + """ Same than testMEDFileFieldEasyField3 but with integers""" + fname="Pyfile104.med" + fieldName="field1" + mm=MEDFileUMesh() + coo=DataArrayDouble([(3,2,1),(8,7,6),(5,9,10)]) + m=MEDCouplingUMesh("mesh",2) ; m.setCoords(coo) + m.allocateCells() + m.insertNextCell(NORM_TRI3,[0,1,2]) + m.insertNextCell(NORM_TRI3,[3,4,5]) + m.insertNextCell(NORM_TRI3,[6,7,8]) + m.insertNextCell(NORM_TRI3,[9,10,11]) + m.insertNextCell(NORM_QUAD4,[100,101,102,103]) + m.insertNextCell(NORM_QUAD4,[104,105,106,107]) + mm[-1]=m + m0=MEDCouplingUMesh("mesh",3) ; m0.setCoords(coo) + m0.allocateCells() + m0.insertNextCell(NORM_TETRA4,[3,2,5,0]) + m0.insertNextCell(NORM_TETRA4,[7,6,3,2]) + mm[0]=m0 + mm.write(fname,2) + # start slowly + f1=MEDCouplingFieldInt(ON_CELLS) ; f1.setName(fieldName) ; f1.setArray(DataArrayInt([(0,100),(1,101)])) ; f1.setMesh(mm[0]) ; f1.setTime(4.,1,2) + f1ts=MEDFileIntField1TS() ; f1ts.setFieldNoProfileSBT(f1) ; f1ts.write(fname,0) + # + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileIntField1TS(fname,fieldName,1,2) + self.assertTrue(f1.isEqual(f1ts.field(mm),1e-12,1e-12)) + # here f1 lying on level -1 not 0 check if "field" method detect it ! + f1=MEDCouplingFieldInt(ON_CELLS) ; f1.setName(fieldName) ; f1.setArray(DataArrayInt([(0,100),(1,101),(0,100),(1,101),(0,100),(1,101)])) + f1.setMesh(mm[-1]) # -1 is very important + f1.setTime(16.,3,4) + f1.checkConsistencyLight() + mm.write(fname,2) + f1ts=MEDFileIntField1TS() ; f1ts.setFieldNoProfileSBT(f1) ; f1ts.write(fname,0) + # + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileIntField1TS(fname,fieldName,3,4) + self.assertTrue(f1.isEqual(f1ts.field(mm),1e-12,1e-12)) + # nodes on elements + f3=MEDCouplingFieldInt(ON_GAUSS_NE) + f3.setMesh(mm[-1]) # this line is important + arr3=DataArrayInt([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19]) ; f3.setArray(arr3) ; f3.setTime(0.5,2,3) + f3.setName(fieldName) ; f3.checkConsistencyLight() + mm.write(fname,2) ; ff=MEDFileIntField1TS() ; ff.setFieldNoProfileSBT(f3) ; ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileIntField1TS(fname,fieldName,2,3) + self.assertTrue(f3.isEqual(f1ts.field(mm),1e-12,1e-12)) + # gauss + f4=MEDCouplingFieldInt(ON_GAUSS_PT) + f4.setMesh(mm[-1]) # this line is important + f4.setName(fieldName) + f4.setGaussLocalizationOnType(NORM_TRI3,[0.,0.,1.,0.,1.,1.],[0.1,0.1, 0.2,0.2, 0.3,0.3, 0.4,0.4, 0.5,0.5],[0.2,0.3,0.1,0.05,0.35]) + f4.setGaussLocalizationOnType(NORM_QUAD4,[0.,0.,1.,0.,1.,1.,0.,1.],[0.3,0.4, 0.6,0.7],[0.7,0.3]) ; f4.setTime(0.25,4,5) + arr4=DataArrayInt([0,1,2,3,4 ,10,11,12,13,14, 20,21,22,23,24, 30,31,32,33,34, 45,46, 55,56]) ; arr4.setInfoOnComponent(0,"abc") ; f4.setArray(arr4) + f4.checkConsistencyLight() + mm.write(fname,2) ; ff=MEDFileIntField1TS() ; ff.setFieldNoProfileSBT(f4) ; ff.write(fname,0) + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileIntField1TS(fname,fieldName,4,5) + self.assertTrue(f4.isEqual(f1ts.field(mm),1e-12,1e-12)) + pass + + def testMEDFileFieldEasyField5(self): + """More and more difficult now look at how profiles are managed by "field" method.""" + fname="Pyfile105.med" + fieldName="field1" + mm=MEDFileUMesh() + coo=DataArrayDouble([(3,2,1),(8,7,6),(5,9,10)]) + m=MEDCouplingUMesh("mesh",2) ; m.setCoords(coo) + m.allocateCells() + m.insertNextCell(NORM_TRI3,[0,1,2]) + m.insertNextCell(NORM_TRI3,[3,4,5]) + m.insertNextCell(NORM_TRI3,[6,7,8]) + m.insertNextCell(NORM_TRI3,[9,10,11]) + m.insertNextCell(NORM_QUAD4,[100,101,102,103]) + m.insertNextCell(NORM_QUAD4,[104,105,106,107]) + mm[0]=m + mm.write(fname,2) + pfl=DataArrayInt([0,2,3,5]) ; pfl.setName("pfl") + m2=m.deepCopy()[pfl] ; m2.setName(m.getName()) + # + arr0=DataArrayDouble([10,11,12,13]) + f=MEDCouplingFieldDouble(ON_CELLS) ; f.setArray(arr0) ; f.setMesh(m2) + f.setName(fieldName) ; f.setTime(2.,6,7) ; f.checkConsistencyLight() + ff=MEDFileFieldMultiTS() ; ff.appendFieldProfile(f,mm,0,pfl) # ff is a field on profile + ff.write(fname,0) + # + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileField1TS(fname,fieldName,6,7) + self.assertTrue(f.isEqual(f1ts.field(mm),1e-12,1e-12)) + # more complicated -> multi level + m0=MEDCouplingUMesh("mesh",3) ; m0.setCoords(coo) + m0.allocateCells() + m0.insertNextCell(NORM_TETRA4,[3,2,5,0]) + m0.insertNextCell(NORM_TETRA4,[7,6,3,2]) + mm2=MEDFileUMesh() + mm2[0]=m0 ; mm2[-1]=m + # + ff=MEDFileField1TS() ; ff.setFieldProfile(f,mm2,-1,pfl) + # + mm=MEDFileMesh.New(fname) ; f1ts=MEDFileField1TS(fname,fieldName,6,7) + self.assertTrue(f.isEqual(f1ts.field(mm),1e-12,1e-12)) + pass + pass if __name__ == "__main__": - unittest.main() + unittest.main()