1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
#############################################################
## ##
## Copyright (c) 2003-2017 by The University of Queensland ##
## Centre for Geoscience Computing ##
## http://earth.uq.edu.au/centre-geoscience-computing ##
## ##
## Primary Business: Brisbane, Queensland, Australia ##
## Licensed under the Open Software License version 3.0 ##
## http://www.apache.org/licenses/LICENSE-2.0 ##
## ##
#############################################################
from __future__ import division
from esys.lsm.vis import core
import vtk as kwvtk
class EdgeData(core.EdgeData):
def __init__(self, data, edgeExtractor):
core.EdgeData.__init__(self, data, edgeExtractor)
def getVtkData(self):
return self.getEdgeExtractor().getVtkData(self.getData())
def getVtkTube(self):
vtkData = self.getVtkData()
vtkTube = kwvtk.vtkTubeFilter()
vtkTube.SetVaryRadiusToVaryRadiusByScalar()
vtkTube.SetNumberOfSides(16)
vtkTube.SetInput(vtkData)
radiusRange = \
vtkData.GetPointData().GetArray(
self.getEdgeExtractor().getRadiusFieldName()
).GetRange(
self.getEdgeExtractor().getRadiusFieldComponentIndex()
)
vtkTube.SetRadius(radiusRange[0])
vtkTube.SetRadiusFactor(radiusRange[0]/float(radiusRange[1]))
return vtkTube
def getVtkMapper(self):
vtkMapper = kwvtk.vtkPolyDataMapper()
vtkMapper.SetInput(self.getVtkTube().GetOutput())
vtkMapper.ScalarVisibilityOn()
vtkMapper.SetScalarModeToUsePointFieldData()
vtkMapper.ColorByArrayComponent(
self.getEdgeExtractor().getColorFieldName(),
self.getEdgeExtractor().getColorFieldComponentIndex(),
)
return vtkMapper
def getActor(self):
vtkActor = kwvtk.vtkActor()
vtkActor.SetMapper(self.getVtkMapper())
return vtkActor
def addActor(self, addTo):
addTo.AddActor(self.getActor())
|
