Cách hiển thị đám mây điểm trong vtk với các màu khác nhau?

Question

Tôi có một đám mây điểm mà tôi muốn hình dung trong vtk. Tôi muốn nhìn thấy chúng trong các màu khác nhau tùy thuộc vào một số điều kiện. Tôi đang sử dụng python.

Bất kỳ đề xuất nào về cách thực hiện điều đó?

Cảm ơn rất nhiều

Answer 1

Đây là ví dụ đặt màu điểm dựa trên giá trị trục z.

import vtk 
from numpy import random 

class VtkPointCloud: 

    def __init__(self, zMin=-10.0, zMax=10.0, maxNumPoints=1e6): 
     self.maxNumPoints = maxNumPoints 
     self.vtkPolyData = vtk.vtkPolyData() 
     self.clearPoints() 
     mapper = vtk.vtkPolyDataMapper() 
     mapper.SetInputData(self.vtkPolyData) 
     mapper.SetColorModeToDefault() 
     mapper.SetScalarRange(zMin, zMax) 
     mapper.SetScalarVisibility(1) 
     self.vtkActor = vtk.vtkActor() 
     self.vtkActor.SetMapper(mapper) 

    def addPoint(self, point): 
     if self.vtkPoints.GetNumberOfPoints() < self.maxNumPoints: 
      pointId = self.vtkPoints.InsertNextPoint(point[:]) 
      self.vtkDepth.InsertNextValue(point[2]) 
      self.vtkCells.InsertNextCell(1) 
      self.vtkCells.InsertCellPoint(pointId) 
     else: 
      r = random.randint(0, self.maxNumPoints) 
      self.vtkPoints.SetPoint(r, point[:]) 
     self.vtkCells.Modified() 
     self.vtkPoints.Modified() 
     self.vtkDepth.Modified() 

    def clearPoints(self): 
     self.vtkPoints = vtk.vtkPoints() 
     self.vtkCells = vtk.vtkCellArray() 
     self.vtkDepth = vtk.vtkDoubleArray() 
     self.vtkDepth.SetName('DepthArray') 
     self.vtkPolyData.SetPoints(self.vtkPoints) 
     self.vtkPolyData.SetVerts(self.vtkCells) 
     self.vtkPolyData.GetPointData().SetScalars(self.vtkDepth) 
     self.vtkPolyData.GetPointData().SetActiveScalars('DepthArray') 

pointCloud = VtkPointCloud() 
for k in xrange(1000): 
    point = 20*(random.rand(3)-0.5) 
    pointCloud.addPoint(point) 
pointCloud.addPoint([0,0,0]) 
pointCloud.addPoint([0,0,0]) 
pointCloud.addPoint([0,0,0]) 
pointCloud.addPoint([0,0,0]) 

# Renderer 
renderer = vtk.vtkRenderer() 
renderer.AddActor(pointCloud.vtkActor) 
renderer.SetBackground(.2, .3, .4) 
renderer.ResetCamera() 

# Render Window 
renderWindow = vtk.vtkRenderWindow() 
renderWindow.AddRenderer(renderer) 

# Interactor 
renderWindowInteractor = vtk.vtkRenderWindowInteractor() 
renderWindowInteractor.SetRenderWindow(renderWindow) 

# Begin Interaction 
renderWindow.Render() 
renderWindowInteractor.Start()