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)