def __init__(self, renderer): self.__renderer = renderer self.__reader = vtkStructuredGridReader() self.__mapper = vtkDataSetMapper() self.__lookup_table = vtkLookupTable() self.__property = vtkProperty() self.__actor = vtkActor()
def __init__(self, renderer): self.__renderer = renderer self.__reader = vtkStructuredGridReader() self.__mapper = vtkDataSetMapper() self.__lookup_table = vtkLookupTable() self.__property = vtkProperty() self.__actor = vtkActor() self.__points = vtkPointSource() self.__rungeKutta = vtkRungeKutta4() self.__streamer = vtkStreamTracer()
def create_vtk(xx, yy, zz, points, min_val, max_val, num_colors): point_cloud = VtkPointCloud() for point in points: point_cloud.add_point(point) mapper = point_cloud.actor.GetMapper() lut = vtk.vtkLookupTable() lut.SetTableRange(min_val, max_val) lut.SetHueRange(0.6667, 0.0) lut.Build() mapper.SetLookupTable(lut) mapper.SetScalarRange(min_val, max_val) cube = vtk.vtkCubeSource() xlen = 0.1 ylen = 0.1 zlen = 0.1 cube.SetXLength(xlen) cube.SetYLength(ylen) cube.SetZLength(zlen) glyph = vtk.vtkGlyph3D() glyph.SetInputData(point_cloud.pd) point_cloud.pd.GetPointData().SetActiveScalars("Obj") point_cloud.pd.GetPointData().SetScalars(point_cloud.obj) glyph.SetSourceConnection(cube.GetOutputPort()) glyph.ScalingOff() cb_actor = vtk.vtkScalarBarActor() cb_actor.SetTitle("Obj") cb_actor.SetLookupTable(lut) cb_actor.SetWidth(0.05) cb_actor.SetPosition(0.92, 0.15) cb_actor.VisibilityOn() lut.SetNumberOfColors(num_colors) glyph.Update() mapper.SetInputConnection(glyph.GetOutputPort()) axes_actor = vtk.vtkAxesActor() axes_actor.SetXAxisLabelText("x") axes_actor.SetYAxisLabelText("y") axes_actor.SetZAxisLabelText("z") axes_actor.GetXAxisCaptionActor2D().GetTextActor().SetTextScaleModeToNone() axes_actor.GetYAxisCaptionActor2D().GetTextActor().SetTextScaleModeToNone() axes_actor.GetZAxisCaptionActor2D().GetTextActor().SetTextScaleModeToNone() label_size = 20 axes_actor.GetXAxisCaptionActor2D().GetCaptionTextProperty().SetFontSize( label_size) axes_actor.GetYAxisCaptionActor2D().GetCaptionTextProperty().SetFontSize( label_size) axes_actor.GetZAxisCaptionActor2D().GetCaptionTextProperty().SetFontSize( label_size) renderer = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() renWin.AddRenderer(renderer) iren = vtk.vtkRenderWindowInteractor() iren.SetRenderWindow(renWin) renderer.AddActor(point_cloud.actor) renderer.AddActor(cb_actor) renderer.AddActor(axes_actor) renWin.Render() camera = vtk.vtkCamera() renderer.SetActiveCamera(camera) camera.SetParallelProjection(True) camera.SetPosition(20, 0, 0) iren.Start()
def __init__(self, renderer): self.__renderer = renderer self.__mapper = vtkPolyDataMapper() self.__lookup_table = vtkLookupTable() self.__property = vtkProperty() self.__actor = vtkActor()