1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2013 CEA/DEN, EDF R&D
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.
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.
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
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 # Author : Edward AGAPOV (eap)
22 from MEDLoader import *
24 from MEDLoaderDataForTest import MEDLoaderDataForTest
26 class SauvLoaderTest(unittest.TestCase):
28 def testSauv2Med(self):
29 # get a file containing all types of readable piles
30 self.assertTrue( os.getenv("MED_ROOT_DIR") )
31 sauvFile = os.path.join( os.getenv("MED_ROOT_DIR"), "share","salome",
32 "resources","med","allPillesTest.sauv")
33 self.assertTrue( os.access( sauvFile, os.F_OK))
35 # read SAUV and write MED
36 medFile = "SauvLoaderTest.med"
37 sr=SauvReader.New(sauvFile);
38 d2=sr.loadInMEDFileDS();
42 self.assertEqual(1,d2.getNumberOfMeshes())
43 self.assertEqual(8+97,d2.getNumberOfFields())
45 m = mm.getMeshAtPos(0)
46 self.assertEqual(17,len(m.getGroupsNames()))
51 def testMed2Sauv(self):
53 self.assertTrue( os.getenv("MED_ROOT_DIR") )
54 medFile = os.path.join( os.getenv("MED_ROOT_DIR"), "share","salome",
55 "resources","med","pointe.med")
56 self.assertTrue( os.access( medFile, os.F_OK))
57 pointeMed = MEDFileData.New( medFile )
59 # add 3 faces to pointeMed
60 pointeMedMesh = pointeMed.getMeshes().getMeshAtPos(0)
61 pointeM1D = MEDCouplingUMesh.New()
62 pointeM1D.setCoords( pointeMedMesh.getCoords() )
63 pointeM1D.setMeshDimension( 2 )
64 pointeM1D.allocateCells( 3 )
65 pointeM1D.insertNextCell( NORM_TRI3, 3, [0,1,2])
66 pointeM1D.insertNextCell( NORM_TRI3, 3, [0,1,3])
67 pointeM1D.insertNextCell( NORM_QUAD4, 4, [10,11,12,13])
68 pointeM1D.finishInsertingCells()
69 pointeMedMesh.setMeshAtLevel( -1, pointeM1D )
70 pointeMed.getMeshes().setMeshAtPos( 0, pointeMedMesh )
72 # add a field on 2 faces to pointeMed
73 ff1=MEDFileFieldMultiTS.New()
74 f1=MEDCouplingFieldDouble.New(ON_GAUSS_NE,ONE_TIME)
76 f1.setName("Field on 2 faces")
77 d=DataArrayDouble.New()
79 d.setInfoOnComponent(0,"sigX [MPa]")
80 d.setInfoOnComponent(1,"sigY [GPa]")
81 d.setValues([311,312,321,322,331,332,411,412,421,422,431,432,441,442],3+4,2)
84 da.setValues([0,2],2,1)
86 pointeM1D_2=pointeM1D[da]
87 f1.setMesh( pointeM1D_2 )
88 ff1.appendFieldProfile(f1,pointeMedMesh,-1,da)
89 pointeMed.getFields().pushField( ff1 )
92 pointeFields = pointeMed.getFields()
93 for i in range( pointeFields.getNumberOfFields() ):
94 if pointeFields.getFieldAtPos(i).getName() == "fieldnodeint":
95 pointeFields.destroyFieldAtPos( i )
98 # write pointeMed to SAUV
99 sauvFile = "SauvLoaderTest.sauv"
101 sw.setMEDFileDS(pointeMed);
104 # read SAUV and check
105 sr=SauvReader.New(sauvFile);
106 d2=sr.loadInMEDFileDS();
107 self.assertEqual(1,d2.getNumberOfMeshes())
108 self.assertEqual(4,d2.getNumberOfFields())
109 m = d2.getMeshes().getMeshAtPos(0)
110 self.assertEqual("maa1",m.getName())
111 self.assertEqual(6,len(m.getGroupsNames()))
112 self.assertEqual(3,m.getMeshDimension())
113 groups = m.getGroupsNames()
114 self.assertTrue( "groupe1" in groups )
115 self.assertTrue( "groupe2" in groups )
116 self.assertTrue( "groupe3" in groups )
117 self.assertTrue( "groupe4" in groups )
118 self.assertTrue( "groupe5" in groups )
119 self.assertTrue( "maa1" in groups )
120 self.assertEqual(16,m.getSizeAtLevel(0))
121 um0 = m.getGenMeshAtLevel(0)
122 self.assertEqual(12, um0.getNumberOfCellsWithType( NORM_TETRA4 ))
123 self.assertEqual(2, um0.getNumberOfCellsWithType( NORM_PYRA5 ))
124 self.assertEqual(2, um0.getNumberOfCellsWithType( NORM_HEXA8 ))
125 um1 = m.getGenMeshAtLevel(-1)
126 self.assertEqual(2, um1.getNumberOfCellsWithType( NORM_TRI3 ))
127 pointeUM0 = pointeMedMesh.getGenMeshAtLevel(0)
128 self.assertTrue(m.getCoords().isEqualWithoutConsideringStr(pointeMedMesh.getCoords(),1e-12))
129 self.assertEqual( um0.getMeasureField(0).accumulate(0),
130 pointeUM0.getMeasureField(0).accumulate(0),1e-12)
133 fieldnodedoubleTS1 = pointeMed.getFields().getFieldWithName("fieldnodedouble")
134 fieldnodedoubleTS2 = d2.getFields().getFieldWithName("fieldnodedouble")
135 self.assertEqual( fieldnodedoubleTS1.getInfo(), fieldnodedoubleTS2.getInfo())
136 self.assertEqual( fieldnodedoubleTS1.getNumberOfTS(), fieldnodedoubleTS2.getNumberOfTS())
137 io1 = fieldnodedoubleTS1.getIterations()
138 io2 = fieldnodedoubleTS2.getIterations()
139 for i in range(fieldnodedoubleTS1.getNumberOfTS() ):
140 fnd1 = fieldnodedoubleTS1.getFieldOnMeshAtLevel(ON_NODES, io1[i][0],io1[i][1],pointeUM0)
141 fnd2 = fieldnodedoubleTS2.getFieldOnMeshAtLevel(ON_NODES, io2[i][0],io2[i][1],um0)
142 self.assertTrue( fnd1.getArray().isEqual( fnd2.getArray(), 1e-12 ))
143 # fieldcelldoublevector
144 fieldnodedoubleTS1 = pointeMed.getFields().getFieldWithName("fieldcelldoublevector")
145 fieldnodedoubleTS2 = d2.getFields().getFieldWithName("fieldcelldoublevector")
146 self.assertEqual( fieldnodedoubleTS1.getInfo(), fieldnodedoubleTS2.getInfo())
147 self.assertEqual( fieldnodedoubleTS1.getNumberOfTS(), fieldnodedoubleTS2.getNumberOfTS())
148 io1 = fieldnodedoubleTS1.getIterations()
149 io2 = fieldnodedoubleTS2.getIterations()
150 for i in range(fieldnodedoubleTS1.getNumberOfTS() ):
151 fnd1 = fieldnodedoubleTS1.getFieldOnMeshAtLevel(ON_CELLS, io1[i][0],io1[i][1],pointeUM0)
152 fnd2 = fieldnodedoubleTS2.getFieldOnMeshAtLevel(ON_CELLS, io2[i][0],io2[i][1],um0)
153 self.assertAlmostEqual( fnd1.accumulate(0), fnd2.accumulate(0) )
154 self.assertAlmostEqual( fnd1.accumulate(1), fnd2.accumulate(1) )
155 self.assertAlmostEqual( fnd1.accumulate(2), fnd2.accumulate(2) )
157 fieldOnFaces = d2.getFields().getFieldWithName(f1.getName())
158 io1 = fieldOnFaces.getIterations()
159 fof = fieldOnFaces.getFieldOnMeshAtLevel(f1.getTypeOfField(),io1[i][0],io1[i][1],um1)
160 self.assertTrue( d.isEqual( fof.getArray(), 1e-12 ))
162 os.remove( sauvFile )
166 def testSauv2MedWONodeFamilyNum(self):
167 """test for issue 0021673: [CEA 566] Bug in SauvWriter when writing meshes
168 having no family ids on nodes."""
170 myCoords=DataArrayDouble.New([-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 ],9,2)
171 targetConn=[0,3,4,1, 1,4,2, 4,5,2, 6,7,4,3, 7,8,5,4];
172 targetMesh=MEDCouplingUMesh.New("BugInSauvWriter",2);
173 targetMesh.allocateCells(5);
174 targetMesh.insertNextCell(NORM_TRI3,3,targetConn[4:7]);
175 targetMesh.insertNextCell(NORM_TRI3,3,targetConn[7:10]);
176 targetMesh.insertNextCell(NORM_QUAD4,4,targetConn[0:4]);
177 targetMesh.insertNextCell(NORM_QUAD4,4,targetConn[10:14]);
178 targetMesh.insertNextCell(NORM_QUAD4,4,targetConn[14:18]);
179 targetMesh.finishInsertingCells();
180 targetMesh.setCoords(myCoords);
183 m.setMeshAtLevel(0,targetMesh)
185 fam=DataArrayInt.New(targetMesh.getNumberOfNodes())
187 #m.setFamilyFieldArr(1,fam)
190 ms=MEDFileMeshes.New()
192 meddata=MEDFileData.New()
193 meddata.setMeshes(ms)
195 medFile = "BugInSauvWriter.sauv"
197 sw.setMEDFileDS(meddata);
203 def testSauv2MedOnPipe1D(self):
204 """test for issue 0021745: [CEA 600] Some missing groups in mesh after reading a SAUV file with SauvReader."""
205 sauvFile="Test_sauve_1D.sauv"
206 # Make a sauve file with a qudratic 1D mesh
207 m=MEDCouplingUMesh.New("pipe1D",1)
209 targetConn=[0,2,1, 2,4,3]
210 m.insertNextCell(NORM_SEG3,3,targetConn[0:3])
211 m.insertNextCell(NORM_SEG3,3,targetConn[3:6])
212 m.finishInsertingCells();
214 coords=[ 0.,1.,2.,4.,5. ];
215 c=DataArrayDouble.New()
216 c.setValues(coords,5,1)
219 mm=MEDFileUMesh.New()
220 mm.setName(m.getName())
221 mm.setDescription("1D mesh")
223 mm.setMeshAtLevel(0,m);
225 mfd1 = MEDFileData.New()
226 ms=MEDFileMeshes.New(); ms.setMeshAtPos(0,mm)
230 sw.setMEDFileDS(mfd1)
232 # Check connectivity read from the sauv file
233 sr = SauvReader.New(sauvFile)
234 mfd2 = sr.loadInMEDFileDS()
235 mfMesh = mfd2.getMeshes()[0]
236 mesh = mfMesh.getMeshAtLevel(0)
237 self.assertTrue(mesh.getNodalConnectivity().isEqual(m.getNodalConnectivity()))
242 def testMissingGroups(self):
243 """test for issue 0021749: [CEA 601] Some missing groups in mesh after reading a SAUV file with SauvReader."""
244 self.assertTrue( os.getenv("MED_ROOT_DIR") )
245 sauvFile = os.path.join( os.getenv("MED_ROOT_DIR"), "share","salome",
246 "resources","med","BDC-714.sauv")
247 self.assertTrue( os.access( sauvFile, os.F_OK))
248 name_of_group_on_cells='Slice10:ABSORBER'
249 name_of_group_on_cells2='Slice10:00LR'
250 sr=SauvReader.New(sauvFile)
251 mfd2=sr.loadInMEDFileDS()
252 mfMesh=mfd2.getMeshes()[0]
254 self.assertTrue(name_of_group_on_cells in mfMesh.getGroupsNames())
255 self.assertTrue(name_of_group_on_cells2 in mfMesh.getGroupsNames())
256 self.assertEqual(270,len(mfMesh.getGroupsNames()))
258 ids1=mfMesh.getGroupArr(0,name_of_group_on_cells)
259 ids2=mfMesh.getGroupArr(0,name_of_group_on_cells2)
262 self.assertTrue(ids1.isEqual(ids2))