src/Plugins/MEDReader/Test/testMEDReader15.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2017  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 MEDLoader import *
  23
  24 """ This test is a non regression test that checks the behaviour of MEDReader when a mesh has the same name than a field.
  25 """
  26
  27 fname="testMEDReader15.med"
  28 outImgName="testMEDReader15.png"
  29
  30 zeName="zeName"
  31 c=DataArrayDouble([(0.,0.,0.),(1.,0.,0.),(0.,1.,0.)])
  32 m=MEDFileUMesh()
  33 m.setCoords(c)
  34 m.setName(zeName)
  35 m.write(fname,2)
  36 f=MEDCouplingFieldDouble(ON_NODES)
  37 f.setName(zeName)
  38 f.setArray(DataArrayDouble([(-1.,1.,0.),(0.,1.,0.),(1.,1.,0.)]))
  39 tmp=MEDCouplingUMesh.Build0DMeshFromCoords(m.getCoords()) ; tmp.setName(zeName)
  40 f.setMesh(tmp)
  41 WriteFieldUsingAlreadyWrittenMesh(fname,f)
  42 #
  43
  44 from paraview.simple import *
  45
  46 paraview.simple._DisableFirstRenderCameraReset()
  47 reader=MEDReader(FileName=fname)
  48 ExpectedEntries=['TS0/zeName/ComSup0/zeName@@][@@P1','TS0/zeName/ComSup0/MESH@zeName@@][@@P1']
  49 assert(reader.GetProperty("FieldsTreeInfo")[::2]==ExpectedEntries)
  50
  51 #
  52 glyph1=Glyph(Input=reader,GlyphType='Arrow',Scalars='FamilyIdNode',Vectors='zeName',GlyphMode='All Points',ScaleFactor=0.1,GlyphTransform='Transform2')
  53
  54 renderView1=GetActiveViewOrCreate('RenderView')
  55 renderView1.InteractionMode='3D'
  56 zeNameLUT = GetColorTransferFunction('zeName')
  57 zeNameLUT.RGBPoints = [1.0, 0.231373, 0.298039, 0.752941, 1.2071067811865475, 0.865003, 0.865003, 0.865003, 1.4142135623730951, 0.705882, 0.0156863, 0.14902]
  58 zeNameLUT.ScalarRangeInitialized = 1.
  59 zeNameLUT.VectorMode = 'Component'
  60
  61 glyph1Display=Show(glyph1,renderView1)
  62 glyph1Display.ColorArrayName = ['POINTS', 'FamilyIdNode']
  63 glyph1Display.LookupTable = zeNameLUT
  64 # set scalar coloring
  65 ColorBy(glyph1Display, ('POINTS', 'zeName'))
  66 # rescale color and/or opacity maps used to include current data range
  67 glyph1Display.RescaleTransferFunctionToDataRange(True)
  68 # do not show color bar/color legend
  69 glyph1Display.SetScalarBarVisibility(renderView1, False)
  70 #
  71 renderView1.ViewSize =[300,300]
  72 renderView1.GetRenderWindow().DoubleBufferOff()
  73 Render()
  74
  75 # compare with baseline image
  76 import os
  77 import sys
  78 try:
  79   baselineIndex = sys.argv.index('-B')+1
  80   baselinePath = sys.argv[baselineIndex]
  81 except:
  82   print("Could not get baseline directory. Test failed.")
  83   exit(1)
  84 baseline_file = os.path.join(baselinePath, "testMEDReader15.png")
  85 import vtk.test.Testing
  86 vtk.test.Testing.VTK_TEMP_DIR = vtk.util.misc.vtkGetTempDir()
  87 vtk.test.Testing.compareImage(GetActiveView().GetRenderWindow(), baseline_file,
  88                                                             threshold=1)
  89 vtk.test.Testing.interact()