src/Plugins/MEDReader/Test/testMEDReader16.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2014  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 (EDF R&D)
  21
  22 from MEDLoader import *
  23
  24 """ This test is a non regression test of EDF8662 : This bug revealed that ELNOMesh and ELNOPoints do not behave correctly after the call of ExtractGroup"""
  25
  26 fname="testMEDReader16.med"
  27 outImgName="testMEDReader16.png"
  28
  29 arr=DataArrayDouble([0,1,2])
  30 m=MEDCouplingCMesh() ; m.setCoords(arr,arr) ; m=m.buildUnstructured() ; m.setName("Mesh")
  31 mm=MEDFileUMesh() ; mm.setMeshAtLevel(0,m)
  32 grp0=DataArrayInt([0,1]) ; grp0.setName("grp0")
  33 grp1=DataArrayInt([2,3]) ; grp1.setName("grp1")
  34 grp2=DataArrayInt([1,2]) ; grp2.setName("grp2")
  35 grp3=DataArrayInt([0,1,2,3]) ; grp3.setName("grp3")
  36 mm.setGroupsAtLevel(0,[grp0,grp1,grp2,grp3])
  37 f=MEDCouplingFieldDouble(ON_GAUSS_NE) ; f.setMesh(m) ; f.setName("MyField") ; f.setTime(0.,0,0)
  38 arr2=DataArrayDouble(4*4*2) ; arr2.iota() ; arr2.rearrange(2) ; arr2.setInfoOnComponents(["aa","bbb"])
  39 f.setArray(arr2) ; arr2+=0.1 ; f.checkCoherency()
  40 mm.write(fname,2)
  41 MEDLoader.WriteFieldUsingAlreadyWrittenMesh(fname,f)
  42 #
  43 from paraview.simple import *
  44 from paraview import servermanager
  45
  46 paraview.simple._DisableFirstRenderCameraReset()
  47 reader=MEDReader(FileName=fname)
  48 ExpectedEntries=['TS0/Mesh/ComSup0/MyField@@][@@GSSNE','TS1/Mesh/ComSup0/Mesh@@][@@P0']
  49 assert(reader.GetProperty("FieldsTreeInfo")[::2]==ExpectedEntries)
  50 reader.AllArrays=['TS0/Mesh/ComSup0/MyField@@][@@GSSNE']
  51 ExtractGroup1 = ExtractGroup(Input=reader)
  52 ExtractGroup1.UpdatePipelineInformation()
  53 ExtractGroup1.AllGroups=["GRP_grp1"]
  54 ELNOMesh1=ELNOMesh(Input=ExtractGroup1)
  55 ELNOPoints1=ELNOPoints(Input=ExtractGroup1)
  56 ELNOPoints1.SelectSourceArray=['ELNO@MyField']
  57 for elt in [ELNOMesh1,ELNOPoints1]:
  58     elnoMesh=servermanager.Fetch(ELNOPoints1,0)
  59     vtkArrToTest=elnoMesh.GetBlock(0).GetPointData().GetArray("MyField")
  60     assert(vtkArrToTest.GetNumberOfTuples()==8)
  61     assert(vtkArrToTest.GetNumberOfComponents()==2)
  62     assert(vtkArrToTest.GetComponentName(0)==arr2.getInfoOnComponent(0))
  63     assert(vtkArrToTest.GetComponentName(1)==arr2.getInfoOnComponent(1))
  64     vals=[vtkArrToTest.GetValue(i) for i in xrange(16)]
  65     assert(arr2[8:].isEqualWithoutConsideringStr(DataArrayDouble(vals,8,2),1e-12))
  66     pass
  67 #
  68 ExtractGroup1.AllGroups=["GRP_grp2"]
  69 for elt in [ELNOMesh1,ELNOPoints1]:
  70     elnoMesh=servermanager.Fetch(ELNOMesh1)
  71     vtkArrToTest=elnoMesh.GetBlock(0).GetPointData().GetArray("MyField")
  72     assert(vtkArrToTest.GetNumberOfTuples()==8)
  73     assert(vtkArrToTest.GetNumberOfComponents()==2)
  74     assert(vtkArrToTest.GetComponentName(0)==arr2.getInfoOnComponent(0))
  75     assert(vtkArrToTest.GetComponentName(1)==arr2.getInfoOnComponent(1))
  76     vals=[vtkArrToTest.GetValue(i) for i in xrange(16)]
  77     assert(arr2[4:12].isEqualWithoutConsideringStr(DataArrayDouble(vals,8,2),1e-12))
  78     pass
  79 # important to check that if all the field is present that it is OK (check of the optimization)
  80 ExtractGroup1.AllGroups=["GRP_grp3"]
  81 for elt in [ELNOMesh1,ELNOPoints1]:
  82     elnoMesh=servermanager.Fetch(ELNOMesh1)
  83     vtkArrToTest=elnoMesh.GetBlock(0).GetPointData().GetArray("MyField")
  84     assert(vtkArrToTest.GetNumberOfTuples()==16)
  85     assert(vtkArrToTest.GetNumberOfComponents()==2)
  86     assert(vtkArrToTest.GetComponentName(0)==arr2.getInfoOnComponent(0))
  87     assert(vtkArrToTest.GetComponentName(1)==arr2.getInfoOnComponent(1))
  88     vals=[vtkArrToTest.GetValue(i) for i in xrange(32)]
  89     assert(arr2.isEqualWithoutConsideringStr(DataArrayDouble(vals,16,2),1e-12))
  90     pass
  91 ELNOMesh1=ELNOMesh(Input=reader)
  92 ELNOPoints1=ELNOPoints(Input=reader)
  93 ELNOPoints1.SelectSourceArray=['ELNO@MyField']
  94 for elt in [ELNOMesh1,ELNOPoints1]:
  95     elnoMesh=servermanager.Fetch(ELNOMesh1)
  96     vtkArrToTest=elnoMesh.GetBlock(0).GetPointData().GetArray("MyField")
  97     assert(vtkArrToTest.GetNumberOfTuples()==16)
  98     assert(vtkArrToTest.GetNumberOfComponents()==2)
  99     assert(vtkArrToTest.GetComponentName(0)==arr2.getInfoOnComponent(0))
 100     assert(vtkArrToTest.GetComponentName(1)==arr2.getInfoOnComponent(1))
 101     vals=[vtkArrToTest.GetValue(i) for i in xrange(32)]
 102     assert(arr2.isEqualWithoutConsideringStr(DataArrayDouble(vals,16,2),1e-12))
 103     pass