def initPoints( self, **args ):
if isNone(self.np_points_data):
wait = args.get( 'wait', True )
if wait: self.waitForData( ExecutionDataPacket.POINTS )
else: return
vtk_points_data = numpy_support.numpy_to_vtk( self.np_points_data )
vtk_points_data.SetNumberOfComponents( 3 )
vtk_points_data.SetNumberOfTuples( int( self.np_points_data.size / 3 ) )
self.vtk_planar_points = vtk.vtkPoints()
self.vtk_planar_points.SetData( vtk_points_data )
self.createPolydata( **args )
def updateScalars( self, **args ):
if isNone(self.vardata):
wait = args.get( 'wait', True )
if wait: self.waitForData( ExecutionDataPacket.VARDATA )
else: return
vtk_color_data = numpy_support.numpy_to_vtk( self.vardata )
vtk_color_data.SetName( 'vardata' )
self.polydata.GetPointData().SetScalars( vtk_color_data )
self.polydata.Modified()
self.mapper.Modified()
self.actor.Modified()
def updateVertices( self, **args ):
self.vertices = vtk.vtkCellArray()
if isNone(self.np_index_seq):
wait = args.get( 'wait', False )
if wait: self.waitForData( ExecutionDataPacket.INDICES )
else: return
cell_sizes = numpy.ones_like( self.np_index_seq )
self.np_cell_data = numpy.dstack( ( cell_sizes, self.np_index_seq ) ).flatten()
self.vtk_cell_data = numpy_support.numpy_to_vtkIdTypeArray( self.np_cell_data )
self.vertices.SetCells( cell_sizes.size, self.vtk_cell_data )
self.polydata.SetVerts(self.vertices)
def initPoints(self, **args):
if isNone(self.np_points_data):
wait = args.get('wait', True)
if wait: self.waitForData(ExecutionDataPacket.POINTS)
else: return
vtk_points_data = numpy_support.numpy_to_vtk(self.np_points_data)
vtk_points_data.SetNumberOfComponents(3)
vtk_points_data.SetNumberOfTuples(int(self.np_points_data.size / 3))
self.vtk_planar_points = vtk.vtkPoints()
self.vtk_planar_points.SetData(vtk_points_data)
self.createPolydata(**args)
def updateScalars(self, **args):
if isNone(self.vardata):
wait = args.get('wait', True)
if wait: self.waitForData(ExecutionDataPacket.VARDATA)
else: return
vtk_color_data = numpy_support.numpy_to_vtk(self.vardata)
vtk_color_data.SetName('vardata')
self.polydata.GetPointData().SetScalars(vtk_color_data)
self.polydata.Modified()
self.mapper.Modified()
self.actor.Modified()
def updateVertices(self, **args):
self.vertices = vtk.vtkCellArray()
if isNone(self.np_index_seq):
wait = args.get('wait', False)
if wait: self.waitForData(ExecutionDataPacket.INDICES)
else: return
cell_sizes = numpy.ones_like(self.np_index_seq)
self.np_cell_data = numpy.dstack(
(cell_sizes, self.np_index_seq)).flatten()
self.vtk_cell_data = numpy_support.numpy_to_vtkIdTypeArray(
self.np_cell_data)
self.vertices.SetCells(cell_sizes.size, self.vtk_cell_data)
self.polydata.SetVerts(self.vertices)
