通过按钮将 VTK 集成到 PyQt 中

Question

我需要创建一个简单的 QT 应用程序，允许用户使用 VTK 查看网格。所以基本上这个程序是一个带有框架和按钮的 window（目前）。我的布局是这样的，按钮必须填充 window 的一半长度，而框架（将显示网格）将填充另一半。为了实验我尝试了这里显示的 3D 球：。

我遇到的问题是我只希望 vtk 小部件填充 window 的一半而不是整个 window。我创建了 foo.py 来创建我的主窗口、框架和按钮。 init.py 将初始化 GUI 本身。

这是我的 foo.py 代码：

from PyQt5 import QtCore, QtGui, QtWidgets

class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(743, 430)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.horizontalLayout = QtWidgets.QHBoxLayout(self.centralwidget)
        self.horizontalLayout.setObjectName("horizontalLayout")
        self.pushButton = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton.setObjectName("pushButton")
        self.horizontalLayout.addWidget(self.pushButton)
        self.frame = QtWidgets.QFrame(self.centralwidget)
        self.frame.setFrameShape(QtWidgets.QFrame.StyledPanel)
        self.frame.setFrameShadow(QtWidgets.QFrame.Raised)
        self.frame.setObjectName("frame")
        self.horizontalLayout.addWidget(self.frame)
        MainWindow.setCentralWidget(self.centralwidget)


        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        self.pushButton.setText(_translate("MainWindow", "PushButton"))

这是我的 init.py 代码：

import vtk
import sys
from PyQt5 import QtCore, QtGui
from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
from PyQt5.QtWidgets import QMainWindow, QApplication, QDialog, QFileDialog
from foo import Ui_MainWindow
from PyQt5 import Qt

class MainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self, parent=None):
        super(MainWindow, self).__init__(parent)
        self.setupUi(self)
        self.pushButton.clicked.connect(self.OpenVTK)

    def OpenVTK(self):

        self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
        self.vl = Qt.QVBoxLayout() #I think the mistake might be here..
        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()
        self.iren.Start()

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

如果有人能帮助我，我将不胜感激。提前致谢！

Answer 1

您可以为 frame 和 pushButton 指定 stretch factors，以便它们按比例调整大小。此外，在您的 OpenVTK() 中，不要将垂直布局设置为中央小部件，因为它会隐藏按钮。

在foo.py中的setupUi()中添加以下行：

# Set the stretch factors for both the pushButton & the frame
self.horizontalLayout.setStretchFactor(self.frame,1)
self.horizontalLayout.setStretchFactor(self.pushButton,1)

注释掉或删除 init.py 中 OpenVTK() 中的以下行：

#self.setCentralWidget(self.frame)

结果如下：

通过按钮将 VTK 集成到 PyQt 中

VTK integration in PyQt with a button

python

pyqt

vtk

pyqt5