src/Plugins/MEDReader/Test/testMEDReader20.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2016  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 from MEDLoader import *
  28 fname="testMEDReader20.med"
  29 png="testMEDReader20.png"
  30 nb=10
  31 arrX=DataArrayDouble(nb+1) ; arrX.iota()
  32 arrY=DataArrayDouble([0.,1.])
  33 m=MEDCouplingCMesh() ; m.setCoords(arrX,arrY) ; m=m.buildUnstructured(); m.setName("mesh") ; m.simplexize(0)
  34 mm=MEDFileUMesh() ; mm[0]=m
  35 m1=m.computeSkin() ; mm[-1]=m1
  36 #
  37 f0=DataArrayInt(m1.getNumberOfCells()) ; f0.iota() ; mm.setFamilyFieldArr(-1,f0)
  38 f1=DataArrayInt(m1.getNumberOfNodes()) ; f1.iota() ; mm.setFamilyFieldArr(1,f1) # <- very important the bug can be shown here
  39 #
  40 nbCells=m1.getNumberOfCells() ; nbNodes=m.getNumberOfNodes()
  41 mm.write(fname,2)
  42 for i in range(5):
  43     f=MEDCouplingFieldDouble(ON_CELLS) ; f.setMesh(m)
  44     f.setName("Field")
  45     arr=DataArrayInt(2*nb) ; arr.iota(i) ; arr%=nb ; arr=arr.convertToDblArr()
  46     f.setArray(arr) ; f.setTime(float(i),i,0)
  47     WriteFieldUsingAlreadyWrittenMesh(fname,f)
  48     #
  49     f=MEDCouplingFieldDouble(ON_CELLS) ; f.setMesh(m1)
  50     f.setName("Field")
  51     arr=DataArrayInt(nbCells) ; arr.iota(i) ; arr%=nbCells ; arr=arr.convertToDblArr()
  52     f.setArray(arr) ; f.setTime(float(i),i,0)
  53     WriteFieldUsingAlreadyWrittenMesh(fname,f)
  54     #
  55     f=MEDCouplingFieldDouble(ON_NODES) ; f.setMesh(m)
  56     f.setName("FieldNode")
  57     arr=DataArrayDouble(nbNodes) ; arr[:]=float(i)
  58     f.setArray(arr) ; f.setTime(float(i),i,0)
  59     WriteFieldUsingAlreadyWrittenMesh(fname,f)
  60     pass
  61 #####################
  62 from paraview.simple import *
  63 #### disable automatic camera reset on 'Show'
  64 paraview.simple._DisableFirstRenderCameraReset()
  65
  66 # create a new 'MED Reader'
  67 testMEDReader20med = MEDReader(FileName=fname)
  68 testMEDReader20med.AllArrays = ['TS0/mesh/ComSup0/Field@@][@@P0']
  69 testMEDReader20med.AllTimeSteps = ['0000', '0001', '0002', '0003', '0004']
  70
  71 # get animation scene
  72 animationScene1 = GetAnimationScene()
  73
  74 # update animation scene based on data timesteps
  75 animationScene1.UpdateAnimationUsingDataTimeSteps()
  76
  77 # get active view
  78 renderView1 = GetActiveViewOrCreate('RenderView')
  79 # uncomment following to set a specific view size
  80 # renderView1.ViewSize = [610, 477]
  81
  82 # show data in view
  83 testMEDReader20medDisplay = Show(testMEDReader20med, renderView1)
  84 # trace defaults for the display properties.
  85 testMEDReader20medDisplay.ColorArrayName = [None, '']
  86 testMEDReader20medDisplay.GlyphType = 'Arrow'
  87 testMEDReader20medDisplay.ScalarOpacityUnitDistance = 4.664739046219201
  88 testMEDReader20medDisplay.SelectUncertaintyArray = [None, '']
  89 testMEDReader20medDisplay.UncertaintyTransferFunction = 'PiecewiseFunction'
  90 testMEDReader20medDisplay.OpacityArray = [None, '']
  91 testMEDReader20medDisplay.RadiusArray = [None, '']
  92 testMEDReader20medDisplay.RadiusRange = [0.0, 10.0]
  93 testMEDReader20medDisplay.ConstantRadius = 10.0
  94 testMEDReader20medDisplay.PointSpriteDefaultsInitialized = 1
  95 testMEDReader20medDisplay.SelectInputVectors = [None, '']
  96 testMEDReader20medDisplay.WriteLog = ''
  97
  98 # reset view to fit data
  99 renderView1.ResetCamera()
 100
 101 #changing interaction mode based on data extents
 102 renderView1.InteractionMode = '2D'
 103 renderView1.CameraPosition = [5.0, 0.5, 10000.0]
 104 renderView1.CameraFocalPoint = [5.0, 0.5, 0.0]
 105
 106 # set scalar coloring
 107 ColorBy(testMEDReader20medDisplay, ('FIELD', 'vtkBlockColors'))
 108
 109 # show color bar/color legend
 110 testMEDReader20medDisplay.SetScalarBarVisibility(renderView1, True)
 111
 112 # get color transfer function/color map for 'vtkBlockColors'
 113 vtkBlockColorsLUT = GetColorTransferFunction('vtkBlockColors')
 114
 115 # get opacity transfer function/opacity map for 'vtkBlockColors'
 116 vtkBlockColorsPWF = GetOpacityTransferFunction('vtkBlockColors')
 117
 118 # set scalar coloring
 119 ColorBy(testMEDReader20medDisplay, ('CELLS', 'Field'))
 120
 121 # rescale color and/or opacity maps used to include current data range
 122 testMEDReader20medDisplay.RescaleTransferFunctionToDataRange(True)
 123
 124 # show color bar/color legend
 125 testMEDReader20medDisplay.SetScalarBarVisibility(renderView1, True)
 126
 127 # get color transfer function/color map for 'Field'
 128 fieldLUT = GetColorTransferFunction('Field')
 129
 130 # get opacity transfer function/opacity map for 'Field'
 131 fieldPWF = GetOpacityTransferFunction('Field')
 132
 133 animationScene1.GoToNext() # <- very important to see the bug play with time steps...
 134 animationScene1.GoToNext()
 135 animationScene1.GoToNext()
 136 animationScene1.GoToNext()
 137 animationScene1.GoToPrevious()
 138 animationScene1.GoToPrevious()
 139
 140 # current camera placement for renderView1
 141 renderView1.InteractionMode = '2D'
 142 renderView1.CameraPosition = [5.0, 0.5, 10000.0]
 143 renderView1.CameraFocalPoint = [5.0, 0.5, 0.0]
 144 renderView1.CameraParallelScale = 5.024937810560445
 145
 146 #
 147
 148 renderView1.ViewSize =[300,300]
 149 Render()
 150 #WriteImage(png)
 151
 152 #### saving camera placements for all active views
 153
 154 # current camera placement for renderView1
 155 renderView1.InteractionMode = '2D'
 156 renderView1.CameraPosition = [5.0, 0.5, 10000.0]
 157 renderView1.CameraFocalPoint = [5.0, 0.5, 0.0]
 158 renderView1.CameraParallelScale = 5.024937810560445
 159 # compare with baseline image
 160 import os
 161 import sys
 162 try:
 163   baselineIndex = sys.argv.index('-B')+1
 164   baselinePath = sys.argv[baselineIndex]
 165 except:
 166   print "Could not get baseline directory. Test failed."
 167   exit(1)
 168 baseline_file = os.path.join(baselinePath,png)
 169 import vtk.test.Testing
 170 vtk.test.Testing.VTK_TEMP_DIR = vtk.util.misc.vtkGetTempDir()
 171 vtk.test.Testing.compareImage(GetActiveView().GetRenderWindow(), baseline_file, threshold=25)
 172 vtk.test.Testing.interact()