src/Plugins/MEDReader/plugin/Test/testMEDReader16.py

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