Интеграция VTK в PyQt с помощью кнопки - PullRequest
0 голосов
/ 09 мая 2018

Мне нужно создать простое приложение QT, которое позволяет пользователю просматривать сетки с использованием VTK.Так что в основном программа представляет собой окно с рамкой и кнопкой (пока).Мой макет таков, что кнопка должна заполнить половину длины окна, а рамка (которая будет отображать сетку) заполнит другую половину.Для экспериментов я попробовал 3D шар, показанный здесь: Встраивание объекта VTK в окно PyQT5 .

Проблема, которую я получаю, состоит в том, что я хочу, чтобы только виджет vtk заполнил половину окна вместо всего окна.Я создал foo.py, чтобы создать главное окно, рамку и кнопку.init.py инициализирует сам графический интерфейс.

Вот мой код для 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_())

Я был бы очень признателен, если бы кто-нибудь мог мне помочь.Заранее спасибо!

1 Ответ

0 голосов
/ 09 мая 2018

Вы можете указать коэффициенты растяжения для frame и pushButton, чтобы они пропорционально меняли размер. Кроме того, в вашем OpenVTK() не делайте вертикальный макет в качестве центрального виджета, поскольку он будет скрывать кнопку.

Добавьте следующие строки в setupUi() в foo.py:

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

Закомментируйте или удалите следующую строку из OpenVTK() в init.py:

#self.setCentralWidget(self.frame)

Результат выглядит следующим образом:

Qt VTK

...