1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2016-2022 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, or (at your option) any later version.
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 : Anthony Geay (EDF R&D)
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
30 from medcoupling import *
31 from paraview.simple import *
32 from MEDReaderHelper import WriteInTmpDir,RetriveBaseLine
34 #### disable automatic camera reset on 'Show'
35 paraview.simple._DisableFirstRenderCameraReset()
38 fname="testMEDReader20.med"
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
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
49 nbCells=m1.getNumberOfCells() ; nbNodes=m.getNumberOfNodes()
52 f=MEDCouplingFieldDouble(ON_CELLS) ; f.setMesh(m)
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)
58 f=MEDCouplingFieldDouble(ON_CELLS) ; f.setMesh(m1)
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)
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)
73 def test(baseline_file):
74 fname = GenerateCase()
76 # create a new 'MED Reader'
77 testMEDReader20med = MEDReader(FileNames=[fname])
78 testMEDReader20med.FieldsStatus = ['TS0/mesh/ComSup0/Field@@][@@P0']
79 testMEDReader20med.TimesFlagsStatus = ['0000', '0001', '0002', '0003', '0004']
82 animationScene1 = GetAnimationScene()
84 # update animation scene based on data timesteps
85 animationScene1.UpdateAnimationUsingDataTimeSteps()
87 if '-D' not in sys.argv:
89 renderView1 = GetActiveViewOrCreate('RenderView')
90 # uncomment following to set a specific view size
91 # renderView1.ViewSize = [610, 477]
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
100 # reset view to fit data
101 renderView1.ResetCamera()
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]
108 # set scalar coloring
109 ColorBy(testMEDReader20medDisplay, ('CELLS', 'Field'))
111 # rescale color and/or opacity maps used to include current data range
112 testMEDReader20medDisplay.RescaleTransferFunctionToDataRange(True)
114 # do not show color bar/color legend
115 testMEDReader20medDisplay.SetScalarBarVisibility(renderView1, False)
117 # get color transfer function/color map for 'Field'
118 fieldLUT = GetColorTransferFunction('Field')
120 # get opacity transfer function/opacity map for 'Field'
121 fieldPWF = GetOpacityTransferFunction('Field')
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()
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
138 renderView1.ViewSize =[300,300]
142 #### saving camera placements for all active views
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
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()
158 if __name__ == "__main__":
159 outImgName="testMEDReader20.png"
160 baseline_file = RetriveBaseLine(outImgName)