doc/user/images/extract_feilds.py

   1 # -*- coding: utf-8 -*-
   2
   3 ###
   4 ### This script is intended to be launched in a new SALOME study
   5 ###
   6
   7 import os
   8 import salome
   9
  10 salome.salome_init()
  11
  12 import iparameters
  13 ipar = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1), True)
  14
  15 #Set up visual properties:
  16 ipar.setProperty("AP_ACTIVE_VIEW", "ParaView_0_0")
  17 ipar.setProperty("AP_WORKSTACK_INFO", "0000000100000000000000020100000001000003b5000000040000000100000001000000080000001800500061007200610056006900650077005f0030005f00300000000102")
  18 ipar.setProperty("AP_ACTIVE_MODULE", "ParaViS")
  19 ipar.setProperty("AP_SAVEPOINT_NAME", "GUI state: 1")
  20 #Set up lists:
  21 # fill list AP_VIEWERS_LIST
  22 ipar.append("AP_VIEWERS_LIST", "ParaView_1")
  23 # fill list ParaView_1
  24 ipar.append("ParaView_1", "ParaView scene:2 - viewer:1")
  25 ipar.append("ParaView_1", "empty")
  26 # fill list AP_MODULES_LIST
  27 ipar.append("AP_MODULES_LIST", "ParaViS")
  28
  29 medfile = os.path.join( os.getenv("MEDCOUPLING_ROOT_DIR"),"share","resources","med", "pointe.med")
  30
  31 from MEDLoader import ReadField, WriteField, WriteMesh
  32 f=ReadField(medfile,"fieldnodeint") # field on 19 nodes
  33 f4 = f[(range(19/2))]
  34 f4.getMesh().translate( [5,0,0] )
  35
  36 import tempfile
  37 medfile2=tempfile.NamedTemporaryFile().name + ".med"
  38
  39 WriteMesh(medfile2,f4.getMesh(), True )
  40 WriteField(medfile2,f4,False)
  41
  42 # ###
  43 ### PARAVIS component
  44 ###
  45
  46 import pvsimple
  47 pvsimple.ShowParaviewView()
  48 #### import the simple module from the paraview
  49 from pvsimple import *
  50 #### disable automatic camera reset on 'Show'
  51 pvsimple._DisableFirstRenderCameraReset()
  52
  53 # create a new 'MED Reader'
  54 pointemed = MEDReader(FileName=medfile)
  55
  56 # Properties modified on pointemed
  57 pointemed.AllArrays = ['TS0/maa1/ComSup0/fieldnodeint@@][@@P1']
  58
  59 # get active view
  60 renderView1 = GetActiveViewOrCreate('RenderView')
  61 # uncomment following to set a specific view size
  62 # renderView1.ViewSize = [935, 561]
  63
  64 # show data in view
  65 pointemedDisplay = Show(pointemed, renderView1)
  66
  67 # trace defaults for the display properties.
  68 pointemedDisplay.Representation = 'Surface'
  69
  70 # reset view to fit data
  71 renderView1.ResetCamera()
  72
  73 # set scalar coloring
  74 ColorBy(pointemedDisplay, ('POINTS', 'fieldnodeint', 'comp1'))
  75
  76 # rescale color and/or opacity maps used to include current data range
  77 pointemedDisplay.RescaleTransferFunctionToDataRange(True, False)
  78
  79 # show color bar/color legend
  80 pointemedDisplay.SetScalarBarVisibility(renderView1, True)
  81
  82 # get color transfer function/color map for 'fieldnodeint'
  83 fieldnodeintLUT = GetColorTransferFunction('fieldnodeint')
  84
  85
  86 # create a new 'MED Reader'
  87 partmed = MEDReader(FileName='/data/eap/S8/MEDCOUPLING_BUILD/doc/part.med')
  88
  89 # Properties modified on partmed
  90 partmed.AllArrays = ['TS0/maa1/ComSup0/fieldnodeint@@][@@P1', 'TS0/maa1/ComSup0/maa1@@][@@P0']
  91
  92 # show data in view
  93 partmedDisplay = Show(partmed, renderView1)
  94
  95 # trace defaults for the display properties.
  96 partmedDisplay.Representation = 'Surface'
  97
  98 # update the view to ensure updated data information
  99 renderView1.Update()
 100
 101 # hide data in view
 102 Hide(partmed, renderView1)
 103
 104 # destroy partmed
 105 Delete(partmed)
 106 del partmed
 107
 108 # create a new 'MED Reader'
 109 partmed = MEDReader(FileName='/data/eap/S8/MEDCOUPLING_BUILD/doc/part.med')
 110
 111 # Properties modified on partmed
 112 partmed.AllArrays = ['TS0/maa1/ComSup0/fieldnodeint@@][@@P1', 'TS0/maa1/ComSup0/maa1@@][@@P0']
 113
 114 # show data in view
 115 partmedDisplay = Show(partmed, renderView1)
 116
 117 # trace defaults for the display properties.
 118 partmedDisplay.Representation = 'Surface'
 119
 120 # update the view to ensure updated data information
 121 renderView1.Update()
 122
 123 # destroy partmed
 124 Delete(partmed)
 125 del partmed
 126
 127 # create a new 'MED Reader'
 128 partmed = MEDReader(FileName=medfile2)
 129
 130 # Properties modified on partmed
 131 partmed.AllArrays = ['TS0/maa1/ComSup0/fieldnodeint@@][@@P1']
 132
 133 # show data in view
 134 partmedDisplay = Show(partmed, renderView1)
 135
 136 # trace defaults for the display properties.
 137 partmedDisplay.Representation = 'Surface'
 138
 139 # update the view to ensure updated data information
 140 renderView1.Update()
 141
 142 # set scalar coloring
 143 ColorBy(partmedDisplay, ('POINTS', 'fieldnodeint', 'comp1'))
 144
 145 # rescale color and/or opacity maps used to include current data range
 146 partmedDisplay.RescaleTransferFunctionToDataRange(True, False)
 147
 148 # show color bar/color legend
 149 partmedDisplay.SetScalarBarVisibility(renderView1, True)
 150
 151 #### saving camera placements for all active views
 152
 153 # current camera placement for renderView1
 154 renderView1.CameraPosition = [10.652899780874208, -21.421142047941988, 4.509321505454196]
 155 renderView1.CameraFocalPoint = [-1.941979639934524, 14.064247480193199, 0.8656523817352476]
 156 renderView1.CameraViewUp = [-0.033512904575591126, 0.09030855881511615, 0.9953498125943683]
 157 renderView1.CameraParallelScale = 3.774917217635375
 158
 159
 160 if salome.sg.hasDesktop():
 161   salome.sg.updateObjBrowser(True)
 162   iparameters.getSession().restoreVisualState(1)