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

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2016-2022  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 # non regression test that emulates https://ageay@git.salome-platform.org/gitpub/samples/datafiles.git Med/ResOK_0000.med
  23 # This test point error during commit efd9331a9455785d0f04b75 in PARAVIS
  24 # Commit of the correction : a4e89b15c2faff6341ab9c3d78abc in PARAVIS
  25 # Due to mistake in MEDReader, the family field array on nodes was deleted twice when changing time step
  26
  27 import os
  28 import sys
  29
  30 from medcoupling import *
  31 from paraview.simple import *
  32 from MEDReaderHelper import WriteInTmpDir,RetriveBaseLine
  33
  34 #### disable automatic camera reset on 'Show'
  35 paraview.simple._DisableFirstRenderCameraReset()
  36
  37 def GenerateCase():
  38     fname="testMEDReader20.med"
  39     nb=10
  40     arrX=DataArrayDouble(nb+1) ; arrX.iota()
  41     arrY=DataArrayDouble([0.,1.])
  42     m=MEDCouplingCMesh() ; m.setCoords(arrX,arrY) ; m=m.buildUnstructured(); m.setName("mesh") ; m.simplexize(0)
  43     mm=MEDFileUMesh() ; mm[0]=m
  44     m1=m.computeSkin() ; mm[-1]=m1
  45     #
  46     f0=DataArrayInt(m1.getNumberOfCells()) ; f0.iota() ; mm.setFamilyFieldArr(-1,f0)
  47     f1=DataArrayInt(m1.getNumberOfNodes()) ; f1.iota() ; mm.setFamilyFieldArr(1,f1) # <- very important the bug can be shown here
  48     #
  49     nbCells=m1.getNumberOfCells() ; nbNodes=m.getNumberOfNodes()
  50     mm.write(fname,2)
  51     for i in range(5):
  52         f=MEDCouplingFieldDouble(ON_CELLS) ; f.setMesh(m)
  53         f.setName("Field")
  54         arr=DataArrayInt(2*nb) ; arr.iota(i) ; arr%=nb ; arr=arr.convertToDblArr()
  55         f.setArray(arr) ; f.setTime(float(i),i,0)
  56         WriteFieldUsingAlreadyWrittenMesh(fname,f)
  57         #
  58         f=MEDCouplingFieldDouble(ON_CELLS) ; f.setMesh(m1)
  59         f.setName("Field")
  60         arr=DataArrayInt(nbCells) ; arr.iota(i) ; arr%=nbCells ; arr=arr.convertToDblArr()
  61         f.setArray(arr) ; f.setTime(float(i),i,0)
  62         WriteFieldUsingAlreadyWrittenMesh(fname,f)
  63         #
  64         f=MEDCouplingFieldDouble(ON_NODES) ; f.setMesh(m)
  65         f.setName("FieldNode")
  66         arr=DataArrayDouble(nbNodes) ; arr[:]=float(i)
  67         f.setArray(arr) ; f.setTime(float(i),i,0)
  68         WriteFieldUsingAlreadyWrittenMesh(fname,f)
  69         pass
  70     return fname
  71
  72 @WriteInTmpDir
  73 def test(baseline_file):
  74     fname = GenerateCase()
  75     #####################
  76     # create a new 'MED Reader'
  77     testMEDReader20med = MEDReader(FileName=fname)
  78     testMEDReader20med.AllArrays = ['TS0/mesh/ComSup0/Field@@][@@P0']
  79     testMEDReader20med.AllTimeSteps = ['0000', '0001', '0002', '0003', '0004']
  80
  81     # get animation scene
  82     animationScene1 = GetAnimationScene()
  83
  84     # update animation scene based on data timesteps
  85     animationScene1.UpdateAnimationUsingDataTimeSteps()
  86
  87     if '-D' not in sys.argv:
  88         # get active view
  89         renderView1 = GetActiveViewOrCreate('RenderView')
  90         # uncomment following to set a specific view size
  91         # renderView1.ViewSize = [610, 477]
  92
  93         # show data in view
  94         testMEDReader20medDisplay = Show(testMEDReader20med, renderView1)
  95         # trace defaults for the display properties.
  96         testMEDReader20medDisplay.ColorArrayName = [None, '']
  97         testMEDReader20medDisplay.GlyphType = 'Arrow'
  98         testMEDReader20medDisplay.ScalarOpacityUnitDistance = 4.664739046219201
  99
 100         # reset view to fit data
 101         renderView1.ResetCamera()
 102
 103         #changing interaction mode based on data extents
 104         renderView1.InteractionMode = '2D'
 105         renderView1.CameraPosition = [5.0, 0.5, 10000.0]
 106         renderView1.CameraFocalPoint = [5.0, 0.5, 0.0]
 107
 108         # set scalar coloring
 109         ColorBy(testMEDReader20medDisplay, ('CELLS', 'Field'))
 110
 111         # rescale color and/or opacity maps used to include current data range
 112         testMEDReader20medDisplay.RescaleTransferFunctionToDataRange(True)
 113
 114         # do not show color bar/color legend
 115         testMEDReader20medDisplay.SetScalarBarVisibility(renderView1, False)
 116
 117         # get color transfer function/color map for 'Field'
 118         fieldLUT = GetColorTransferFunction('Field')
 119
 120         # get opacity transfer function/opacity map for 'Field'
 121         fieldPWF = GetOpacityTransferFunction('Field')
 122
 123         animationScene1.GoToNext() # <- very important to see the bug play with time steps...
 124         animationScene1.GoToNext()
 125         animationScene1.GoToNext()
 126         animationScene1.GoToNext()
 127         animationScene1.GoToPrevious()
 128         animationScene1.GoToPrevious()
 129
 130         # current camera placement for renderView1
 131         renderView1.InteractionMode = '2D'
 132         renderView1.CameraPosition = [5.0, 0.5, 10000.0]
 133         renderView1.CameraFocalPoint = [5.0, 0.5, 0.0]
 134         renderView1.CameraParallelScale = 5.024937810560445
 135
 136         #
 137
 138         renderView1.ViewSize =[300,300]
 139         Render()
 140         #WriteImage(png)
 141
 142         #### saving camera placements for all active views
 143
 144         # current camera placement for renderView1
 145         renderView1.InteractionMode = '2D'
 146         renderView1.CameraPosition = [5.0, 0.5, 10000.0]
 147         renderView1.CameraFocalPoint = [5.0, 0.5, 0.0]
 148         renderView1.CameraParallelScale = 5.024937810560445
 149
 150         # compare with baseline image
 151         import vtk.test.Testing
 152         from vtk.util.misc import vtkGetTempDir
 153         vtk.test.Testing.VTK_TEMP_DIR = vtk.util.misc.vtkGetTempDir()
 154         vtk.test.Testing.compareImage(GetActiveView().GetRenderWindow(), baseline_file, threshold=1)
 155         vtk.test.Testing.interact()
 156         pass
 157
 158 if __name__ == "__main__":
 159   outImgName="testMEDReader20.png"
 160   baseline_file = RetriveBaseLine(outImgName)
 161   test(baseline_file)
 162   pass