Esempio n. 1
0
 def __init__(self, renderer):
     self.__renderer = renderer
     self.__reader = vtkStructuredGridReader()
     self.__mapper = vtkDataSetMapper()
     self.__lookup_table = vtkLookupTable()
     self.__property = vtkProperty()
     self.__actor = vtkActor()
Esempio n. 2
0
 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()
Esempio n. 3
0
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()
Esempio n. 4
0
 def __init__(self, renderer):
     self.__renderer = renderer
     self.__mapper = vtkPolyDataMapper()
     self.__lookup_table = vtkLookupTable()
     self.__property = vtkProperty()
     self.__actor = vtkActor()