qframe 绘图_【三维绘图】对vtk python接口做一个封装实现三维绘图

pyqt5 + vtk(python3+qt5)

代码

import sys

import vtk

from PyQt5 import QtCore, QtGui

from PyQt5 import Qt

from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor

class MainWindow(Qt.QMainWindow):

def __init__(self, parent = None):

Qt.QMainWindow.__init__(self, parent)

self.frame = Qt.QFrame()

self.vl = Qt.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()

ma