Skip to content

Instantly share code, notes, and snippets.

@paskino
Created July 27, 2017 11:30
Show Gist options
  • Star 14 You must be signed in to star a gist
  • Fork 3 You must be signed in to fork a gist
  • Save paskino/7b7ed6e4541d682bd5ce5d521505dd8f to your computer and use it in GitHub Desktop.
Save paskino/7b7ed6e4541d682bd5ce5d521505dd8f to your computer and use it in GitHub Desktop.
Minimal working example of VTK embedding in PyQt with Qt5
#!/usr/bin/env python
import sys
import vtk
from PyQt5 import QtCore, QtWidgets
from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
class MainWindow(QtWidgets.QMainWindow):
def __init__(self, parent = None):
QtWidgets.QMainWindow.__init__(self, parent)
self.frame = QtWidgets.QFrame()
self.vl = QtWidgets.QVBoxLayout()
self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
self.vl.addWidget(self.vtkWidget)
self.ren = vtk.vtkRenderer()
self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
# Create source
source = vtk.vtkSphereSource()
source.SetCenter(0, 0, 0)
source.SetRadius(5.0)
# Create a mapper
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(source.GetOutputPort())
# Create an actor
actor = vtk.vtkActor()
actor.SetMapper(mapper)
self.ren.AddActor(actor)
self.ren.ResetCamera()
self.frame.setLayout(self.vl)
self.setCentralWidget(self.frame)
self.show()
self.iren.Initialize()
if __name__ == "__main__":
app = QtWidgets.QApplication(sys.argv)
window = MainWindow()
sys.exit(app.exec_())
@paskino
Copy link
Author

paskino commented Jul 27, 2017

Based on http://www.vtk.org/Wiki/VTK/Examples/Python/Widgets/EmbedPyQt

Working on:

  • Python 3.5
  • VTK 7.1
  • Qt 5.8

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment