File: surfaceData.py

package info (click to toggle)
esys-particle 2.3.5%2Bdfsg2-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 13,132 kB
  • sloc: cpp: 81,480; python: 5,872; makefile: 1,259; sh: 313; perl: 225
file content (51 lines) | stat: -rw-r--r-- 1,783 bytes parent folder | download | duplicates (4)
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
#############################################################
##                                                         ##
## 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              ##
##                                                         ##
#############################################################

import vtk as kwvtk
from esys.lsm.vis import core

class SurfaceData(core.SurfaceData):
    def __init__(self, data, pointExtractor):
        core.SurfaceData.__init__(self, data, pointExtractor)

    def getVtkPolyData(self):
        vtkPolyData = kwvtk.vtkPolyData()
        vtkPolyData.SetPoints(
            self.getPointExtractor().getVtkPoints(self.getData())
        )
        return vtkPolyData

    def getVtkDelaunay(self):
        vtkDelny = kwvtk.vtkDelaunay2D()
        vtkDelny.SetInput(self.getVtkPolyData())
        return vtkDelny

    def getVtkPolyDataNormals(self):
        vtkNormals = kwvtk.vtkPolyDataNormals()
        vtkNormals.SetInput(self.getVtkDelaunay().GetOutput())
        return vtkNormals

    def getVtkMapper(self):
        vtkMapper = kwvtk.vtkPolyDataMapper()
        vtkMapper.SetInput(self.getVtkPolyDataNormals().GetOutput())
        return vtkMapper

    def getActor(self):
        vtkActor = kwvtk.vtkActor()
        vtkActor.SetMapper(self.getVtkMapper())

        return vtkActor

    def addActor(self, addTo):
        addTo.AddActor(self.getActor())