1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2017 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
22 from MEDLoader import *
24 """ This test generate a simple multi time field with a very aggressive time steps triplets. Neither dt, nor iteration nor order is considered.
25 In this case only the rank is considered.
28 fname="testMEDReader7.med"
29 outImgName="testMEDReader7.png"
31 arr=DataArrayDouble([(0,0,0),(1,0,0),(2,0,0),(3,0,0),(0,1,0),(1,1,0),(2,1,0),(3,1,0),(0,2,0),(1,2,0),(2,2,0),(3,2,0),(0,3,0),(1,3,0),(2,3,0),(3,3,0)])
32 m0=MEDCouplingUMesh("mesh",2) ; m0.setCoords(arr) ; m0.allocateCells()
33 for elt in [[2,3,6],[3,7,6],[6,9,5],[6,10,9]]:
34 m0.insertNextCell(NORM_TRI3,elt)
36 for elt in [[0,4,5,1],[5,6,2,1],[4,8,9,5],[6,10,11,7],[8,12,13,9],[9,13,14,10],[10,14,15,11]]:
37 m0.insertNextCell(NORM_QUAD4,elt)
40 mm.setMeshAtLevel(0,m0)
41 grp0=DataArrayInt([0,1,4,5,7,10]) ; grp0.setName("grp0")
42 mm.setGroupsAtLevel(0,[grp0])
43 fmts=MEDFileFieldMultiTS()
45 fNode=MEDCouplingFieldDouble(ON_NODES) ; fNode.setName("fNode")
47 fNode.setArray(DataArrayDouble([3,2,1,0,3.16,2.23,1.41,1,3.6,2.82,2.23,2,4.24,3.6,3.16,3]))
48 fNode.getArray().setInfoOnComponent(0,"C0")
49 fNode.setTime(0.5,1,1)
50 f1ts=MEDFileField1TS() ; f1ts.setFieldNoProfileSBT(fNode) ; fmts.pushBackTimeStep(f1ts)
52 fNode.getArray().reverse()
53 fNode.setTime(0.5,1,2)
54 f1ts=MEDFileField1TS() ; f1ts.setFieldNoProfileSBT(fNode) ; fmts.pushBackTimeStep(f1ts)
56 fNode.getArray().reverse()
57 fNode.setTime(0.5,2,1)
58 f1ts=MEDFileField1TS() ; f1ts.setFieldNoProfileSBT(fNode) ; fmts.pushBackTimeStep(f1ts)
60 fNode.getArray().reverse()
61 fNode.setTime(0.5,2,2)
62 f1ts=MEDFileField1TS() ; f1ts.setFieldNoProfileSBT(fNode) ; fmts.pushBackTimeStep(f1ts)
66 ################### MED write is done -> Go to MEDReader
67 from paraview.simple import *
69 myMedReader=MEDReader(FileName=fname)
70 myMedReader.AllArrays = ['TS0/mesh/ComSup0/fNode@@][@@P1']
71 assert(list(myMedReader.TimestepValues)==[0.,1.,2.,3.])
73 RenderView1 = GetRenderView()
74 RenderView1.CameraFocalPoint = [1.5, 1.5, 0.0]
75 RenderView1.CameraPosition = [1.5, 1.5, 10000.0]
76 RenderView1.InteractionMode = '3D'
77 RenderView1.CameraPosition = [1.5, 1.5, 8.196152422706632]
78 RenderView1.CameraParallelScale = 2.1213203435596424
79 RenderView1.CenterOfRotation = [1.5, 1.5, 0.0]
81 DataRepresentation4 = Show()
82 DataRepresentation4.EdgeColor = [0.0, 0.0, 0.5000076295109483]
83 DataRepresentation4.SelectionPointFieldDataArrayName = 'fNode'
84 DataRepresentation4.ScaleFactor = 0.3182729169726372
86 a1_fGauss_PVLookupTable = GetLookupTableForArray( "fNode", 1, RGBPoints=[0.22, 0.23, 0.299, 0.754, 2.95, 0.706, 0.016, 0.15], VectorMode='Magnitude', NanColor=[0.25, 0.0, 0.0], ColorSpace='Diverging', ScalarRangeInitialized=1.0, AllowDuplicateScalars=1 )
87 a1_fGauss_PiecewiseFunction = CreatePiecewiseFunction( Points=[0.0, 0.0, 0.5, 0.0, 1.0, 1.0, 0.5, 0.0] )
88 DataRepresentation4.ColorArrayName = 'fNode'
89 DataRepresentation4.LookupTable = a1_fGauss_PVLookupTable
90 a1_fGauss_PVLookupTable.ScalarOpacityFunction = a1_fGauss_PiecewiseFunction
92 RenderView1.ViewTime = 1.0 #### Important # red is in right bottom
93 RenderView1.CacheKey = 1.0
94 RenderView1.UseCache = 1
95 RenderView1.ViewSize=[300,300]
98 # compare with baseline image
102 baselineIndex = sys.argv.index('-B')+1
103 baselinePath = sys.argv[baselineIndex]
105 print("Could not get baseline directory. Test failed.")
107 baseline_file = os.path.join(baselinePath, "testMEDReader7.png")
108 import vtk.test.Testing
109 from vtk.util.misc import vtkGetTempDir
110 vtk.test.Testing.VTK_TEMP_DIR = vtk.util.misc.vtkGetTempDir()
111 vtk.test.Testing.compareImage(GetActiveView().GetRenderWindow(), baseline_file,
113 vtk.test.Testing.interact()