Python VTK多邊形數據未更新

Question

我正在嘗試使用VTK繪制點，然后以給定的點位置集交互地更新其位置。

我可以交互地使用polydata對象來繪制點，但是當我調用self.polydata.Update()時，它們不會更新。 當我調用self.polydata.GetCellData().SetScalars(someCharArray)時，這些點將更新

這是VTK中的錯誤，還是我沒有正確更新點坐標？

我提供了一個示例腳本。 如果在self.polydata.GetCellData().SetScalars(someCharArray)注釋了self.polydata.GetCellData().SetScalars(someCharArray) ，則在使用滑塊時，繪圖將不會更新該點的坐標。 但是，如果您保留該行，它們將更新。

謝謝！

import numpy as np
import vtk
from vtk.qt4.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
from PyQt4 import QtGui
import sys

class ViewerWithScrollBar(QtGui.QMainWindow):
    def __init__(self, parent=None):
        super(ViewerWithScrollBar, self).__init__(parent)
        # Define the renderer and Qt window ------------------------
        self.frame = QtGui.QFrame()

        self.hl = QtGui.QHBoxLayout()
        self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
        self.hl.addWidget(self.vtkWidget)

        self.ren = vtk.vtkRenderer()
        self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
        self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
        self.iren.SetInteractorStyle(vtk.vtkInteractorStyleTrackballCamera())
        self.ren.ResetCamera()

        self.frame.setLayout(self.hl)
        self.setCentralWidget(self.frame)

        # Point coordinate data ---------------------------------
        self.coordData = {}
        self.coordData[0] = np.array([[0,0,0], [1,0,0], [1,1,0]])
        self.coordData[1] = self.coordData[0] + np.array([[0.2, 0.1, -0.05], [0,0,0], [0,0,0]])
        self.coordData[2] = self.coordData[1] + np.array([[0.2, 0.1, -0.05], [0,0,0], [0,0,0]])

        # Define the slider bar and add it to the window ---------------
        slider = QtGui.QSlider()
        slider.setAccessibleName('Time index')
        slider.setRange(0, len(self.coordData)-1)
        slider.valueChanged.connect(self.sliderCallback)

        self.hl.addWidget(slider)

        # Create the polydata object -----------------------------
        points = vtk.vtkPoints()
        points.SetNumberOfPoints(len(self.coordData[0]))
        self.polydata = vtk.vtkPolyData()

        for i in range(len(self.coordData[0])):
            points.SetPoint(i, self.coordData[0][i])
        self.polydata.SetPoints(points)

        ptsFilter = vtk.vtkVertexGlyphFilter()
        ptsFilter.SetInputConnection(self.polydata.GetProducerPort())
        ptsMapper = vtk.vtkPolyDataMapper()
        ptsMapper.SetInputConnection(ptsFilter.GetOutputPort())
        ptsActor = vtk.vtkActor()
        ptsActor.SetMapper(ptsMapper)
        ptsActor.GetProperty().SetPointSize(10)

        self.ren.AddActor(ptsActor)

        self.show()
        self.iren.Initialize()

    def sliderCallback(self):
        index = self.sender().value() # The index that the slider bar is currently on
        someCharArray = vtk.vtkUnsignedCharArray()
        points = self.polydata.GetPoints()
        for i in range(len(self.coordData[index])):
            points.SetPoint(i, self.coordData[index][i])
        self.polydata.GetCellData().SetScalars(someCharArray) # For some reason the polydata won't update unless this is called.
        # self.polydata.Update()
        self.iren.Render()
        return

if __name__ == "__main__":

    app = QtGui.QApplication(sys.argv)

    window = ViewerWithScrollBar()

    sys.exit(app.exec_())

Answer 1

在lib的建議下，我修改了代碼。 在sliderCallback方法中調用self.polydata.Modified（）解決了該問題

import numpy as np
import vtk
from vtk.qt4.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
from PyQt4 import QtGui
import sys

class ViewerWithScrollBar(QtGui.QMainWindow):
    def __init__(self, parent=None):
        super(ViewerWithScrollBar, self).__init__(parent)
        # Define the renderer and Qt window ------------------------
        self.frame = QtGui.QFrame()

        self.hl = QtGui.QHBoxLayout()
        self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
        self.hl.addWidget(self.vtkWidget)

        self.ren = vtk.vtkRenderer()
        self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
        self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
        self.iren.SetInteractorStyle(vtk.vtkInteractorStyleTrackballCamera())
        self.ren.ResetCamera()

        self.frame.setLayout(self.hl)
        self.setCentralWidget(self.frame)

        # Point coordinate data ---------------------------------
        self.coordData = {}
        self.coordData[0] = np.array([[0,0,0], [1,0,0], [1,1,0]])
        self.coordData[1] = self.coordData[0] + np.array([[0.2, 0.1, -0.05], [0,0,0], [0,0,0]])
        self.coordData[2] = self.coordData[1] + np.array([[0.2, 0.1, -0.05], [0,0,0], [0,0,0]])

        # Define the slider bar and add it to the window ---------------
        slider = QtGui.QSlider()
        slider.setAccessibleName('Time index')
        slider.setRange(0, len(self.coordData)-1)
        slider.valueChanged.connect(self.sliderCallback)

        self.hl.addWidget(slider)

        # Create the polydata object -----------------------------
        points = vtk.vtkPoints()
        points.SetNumberOfPoints(len(self.coordData[0]))
        self.polydata = vtk.vtkPolyData()

        for i in range(len(self.coordData[0])):
            points.SetPoint(i, self.coordData[0][i])
        self.polydata.SetPoints(points)

        ptsFilter = vtk.vtkVertexGlyphFilter()
        ptsFilter.SetInputData(self.polydata)
        ptsMapper = vtk.vtkPolyDataMapper()
        ptsMapper.SetInputConnection(ptsFilter.GetOutputPort())
        ptsActor = vtk.vtkActor()
        ptsActor.SetMapper(ptsMapper)
        ptsActor.GetProperty().SetPointSize(10)

        self.ren.AddActor(ptsActor)

        self.show()
        self.iren.Initialize()

    def sliderCallback(self):
        index = self.sender().value() # The index that the slider bar is currently on
        points = self.polydata.GetPoints()
        for i in range(len(self.coordData[index])):
            points.SetPoint(i, self.coordData[index][i])
        self.polydata.Modified()
        self.iren.Render()
        return

if __name__ == "__main__":

    app = QtGui.QApplication(sys.argv)

    window = ViewerWithScrollBar()

    sys.exit(app.exec_())

Answer 2

我將其修訂為PyQt5版本，並在Windows 10 PyCharm 2018.1（社區版）下使用最新的vtk 8.1.0和PyQt5 5.10.1運行並測試了以上代碼。 看起來需要修改“點”而不是“ self.polydata”。 否則，將不會顯示更新的點。

import numpy as np
import vtk
from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
from PyQt5 import Qt
import sys

class ViewerWithScrollBar(Qt.QMainWindow):
    def __init__(self, parent=None):
    super(ViewerWithScrollBar, self).__init__(parent)
    # Define the renderer and Qt window ------------------------
    self.frame = Qt.QFrame()

    self.hl = Qt.QHBoxLayout()
    self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
    self.hl.addWidget(self.vtkWidget)

    self.ren = vtk.vtkRenderer()
    self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
    self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
    self.iren.SetInteractorStyle(vtk.vtkInteractorStyleTrackballCamera())
    self.ren.ResetCamera()

    self.frame.setLayout(self.hl)
    self.setCentralWidget(self.frame)

    # Point coordinate data ---------------------------------
    self.coordData = {}
    self.coordData[0] = np.array([[0,0,0], [1,0,0], [1,1,0]])
    self.coordData[1] = self.coordData[0] + np.array([[0.2, 0.0, -0.05], [0,2,0], [0,0,3.5]])
    self.coordData[2] = self.coordData[1] + np.array([[0.2, 10.0, -0.05], [0,5.0,0], [0,0,0]])

    # Define the slider bar and add it to the window ---------------
    slider = Qt.QSlider()
    slider.setAccessibleName('Time index')
    slider.setRange(0, len(self.coordData)-1)
    slider.valueChanged.connect(self.sliderCallback)

    self.hl.addWidget(slider)

    # Create the polydata object -----------------------------
    points = vtk.vtkPoints()
    points.SetNumberOfPoints(len(self.coordData[0]))
    self.polydata = vtk.vtkPolyData()

    for i in range(len(self.coordData[0])):
        points.SetPoint(i, self.coordData[0][i])
    self.polydata.SetPoints(points)

    self.ptsFilter = vtk.vtkVertexGlyphFilter()
    self.ptsFilter.SetInputData(self.polydata)
    ptsMapper = vtk.vtkPolyDataMapper()
    ptsMapper.SetInputConnection(self.ptsFilter.GetOutputPort())
    self.ptsActor = vtk.vtkActor()
    self.ptsActor.SetMapper(ptsMapper)
    self.ptsActor.GetProperty().SetPointSize(10)

    self.ren.AddActor(self.ptsActor)

    self.show()
    self.iren.Initialize()
    self.iren.Start()

def sliderCallback(self):
    index = self.sender().value() # The index that the slider bar is currently on
    points = self.polydata.GetPoints()
    for i in range(len(self.coordData[index])):
        points.SetPoint(i, self.coordData[index][i])
    points.Modified() # Here you need to call Modified for points
    self.show()
    self.iren.Render()

    return

if __name__ == "__main__":
    app = Qt.QApplication(sys.argv)
    window = ViewerWithScrollBar()
    sys.exit(app.exec_())