#!/usr/bin/python
# coding=utf-8
import numpy as np
import sys
from PyQt5 import QtWidgets, QtCore
from PyQt5.QtGui import QFont
from PyQt5.QtWidgets import QLineEdit, QApplication as qApp
from maya_widget import MayaviQWidget, myAction, IndexedQSlider
import utils
# usage: GUI for showing all models
class HumanShapeAnalysisDemo(QtWidgets.QMainWindow):
def __init__(self):
QtWidgets.QMainWindow.__init__(self)
self.statusBar().showMessage("Hello there")
self.mainBox = QtWidgets.QHBoxLayout()
container = QtWidgets.QWidget()
container.setWindowTitle("Embedding Mayavi in a PyQt5 Application")
layout = QtWidgets.QGridLayout(container)
self.viewer3D = MayaviQWidget(container)
layout.addWidget(self.viewer3D, 1, 1)
container.show()
self.mainBox.addWidget(container)
self.setWindowTitle("3D Human Body Reshaping with Anthropometric Modeling")
parentWidget = QtWidgets.QWidget()
self.box = QtWidgets.QVBoxLayout()
self.set_menu()
self.set_radio()
self.set_button()
self.set_slider()
self.mainBox.addLayout(self.box)
self.mainBox.addWidget(self.viewer3D)
self.set_dialog()
self.resize(650, 650)
self.viewer3D.setSizePolicy(
QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Expanding)
parentWidget.setLayout(self.mainBox)
self.setCentralWidget(parentWidget)
def set_menu(self):
menubar = self.menuBar()
fileMenu = menubar.addMenu('&File')
exit = QtWidgets.QAction("Exit", self)
exit.setShortcut("Ctrl+Q")
exit.setStatusTip('Exit application')
exit.triggered.connect(qApp.quit)
fileMenu.addAction(exit)
save = QtWidgets.QAction("Save", self)
save.setShortcut("Ctrl+S")
save.setStatusTip('save obj file')
save.triggered.connect(self.viewer3D.save)
fileMenu.addAction(save)
self.flag_ = 0
self.label_ = "female"
self.mode = {0:"global_mapping", 1:"local_with_mask", 2:"local_with_rfemat"}
for i in range(0, len(self.mode)):
mode = myAction(i, self.mode[i], self)
mode.myact.connect(self.select_mode)
#self.connect(mode, QtCore.SIGNAL('myact(int)'), self.select_mode)
fileMenu.addAction(mode)
self.setToolTip('This is a window, or <b>something</b>')
def set_radio(self):
self.radio1 = QtWidgets.QRadioButton('female')
self.radio2 = QtWidgets.QRadioButton('male')
self.radio1.setFont(QFont("Arial", 11))
self.radio2.setFont(QFont("Arial", 11))
self.radio1.setChecked(True)
self.radio1.toggled.connect(self.radio_act)
self.radio2.toggled.connect(self.radio_act)
radio_box = QtWidgets.QHBoxLayout()
radio_box.addWidget(self.radio1)
radio_box.addWidget(self.radio2)
self.box.addLayout(radio_box)
def radio_act(self):
if self.radio1.isChecked():
self.label_ = 'female'
else:
self.label_ = 'male'
self.viewer3D.select_mode(label=self.label_, flag=self.flag_)
def set_button(self):
self.button_box = QtWidgets.QHBoxLayout()
self.reset_button = QtWidgets.QPushButton("RESET")
self.reset_button.setStatusTip('reset input to mean value')
self.reset_button.setFont(QFont("Arial", 11))
self.reset_button.clicked.connect(self.reset)
self.button_box.addWidget(self.reset_button)
self.pre_button = QtWidgets.QPushButton("PREDICT")
self.pre_button.setToolTip('model your own shape')
self.pre_button.setFont(QFont("Arial", 11))
self.pre_button.clicked.connect(self.show_dialog)
self.button_box.addWidget(self.pre_button)
self.box.addLayout(self.button_box)
def set_slider(self):
self.slider = []
self.spin = []
self.label = []
for i in range(0, utils.M_NUM):
hbox = QtWidgets.QHBoxLayout()
slider = IndexedQSlider(i, QtCore.Qt.Horizontal, self)
slider.setStatusTip('%d. %s' % (i, utils.M_STR[i]))
slider.setRange(-30, 30)
slider.valueChangeForwarded.connect(
self.viewer3D.sliderForwardedValueChangeHandler)
slider.setFixedWidth(60)
self.slider.append(slider)
spinBox = QtWidgets.QSpinBox()
spinBox.setRange(-30, 30)
spinBox.valueChanged.connect(slider.setValue)
slider.valueChanged.connect(spinBox.setValue)
self.spin.append(spinBox)
label = QtWidgets.QLabel()
label.setText(utils.M_STR[i])
# label.setFont(QFont("Arial", 11, QFont.Bold))
label.setFont(QFont("Arial", 12))
label.setFixedWidth(190)
self.label.append(label)
hbox.addWidget(label)
hbox.addWidget(slider)
hbox.addWidget(spinBox)
self.box.addLayout(hbox)
def set_dialog(self):
self.pre_dialog = QtWidgets.QDialog()
self.dialogBox = QtWidgets.QVBoxLayout()
self.pre_dialog.setWindowTitle("Input")
self.editList = []
for i in range(0, utils.M_NUM):
edit = QtWidgets.QLineEdit()
self.editList.append(edit)
label = QtWidgets.QLabel()
if i == 0:
label.setText('{} (x2 kg)'.format(utils.M_STR[i]))
else:
label.setText('{} (cm)'.format(utils.M_STR[i]))
# label.setFont(QFont("Arial", 11, QFont.Bold))
label.setFont(QFont("Arial", 12))
label.setFixedHeight(20)
label.setFixedWidth(190)
box = QtWidgets.QHBoxLayout()
box.addWidget(label)
box.addWidget(edit)
self.dialogBox.addLayout(box)
dialogOK = QtWidgets.QPushButton("OK")
clearButton = QtWidgets.QPushButton("CLEAR")
dialogOK.setFont(QFont("Arial", 11, QFont.Bold))
clearButton.setFont(QFont("Arial", 11, QFont.Bold))
dialogOK.clicked.connect(self.predict)
clearButton.clicked.connect(self.clear)
box = QtWidgets.QHBoxLayout()
box.addWidget(dialogOK)
box.addWidget(clearButton)
self.dialogBox.addLayout(box)
self.pre_dialog.setLayout(self.dialogBox)
def predict(self):
try:
w = float(self.editList[0].text())
h = float(self.editList[1].text())
data = []
data.append(w ** (1.0 / 3.0) * 1000)
data.append(h * 10)
for i in range(2, len(self.editList)):
try:
tmp = float(self.editList[i].text())
data.append(tmp * 10)
except ValueError:
data.append(0)
data = np.array(data).reshape(utils.M_NUM, 1)
[t_data, value] = self.viewer3D.predict(data)
for i in range(2, len(self.editList)):
self.editList[i].setText("%f" % (value[i, 0] / 10))
for i in range(0, len(self.slider)):
self.slider[i].valueChangeForwarded.disconnect(
self.viewer3D.sliderForwardedValueChangeHandler)
self.slider[i].setValue(t_data[i] / 3.0 * 100.0)
self.slider[i].valueChangeForwarded.connect(
self.viewer3D.sliderForwardedValueChangeHandler)
except ValueError:
self.editList[0].setText("Please input.")
self.editList[1].setText("Please input.")
def closeEvent(self, event):
self.pre_dialog.close()
event.accept()
def reset(self):
for i in range(0, utils.M_NUM):
self.slider[i].setValue(0)
def clear(self):
for i in range(0, len(self.editList)):
self.editList[i].clear()
def show_dialog(self):
self.pre_dialog.show()
def select_mode(self, id):
self.flag_ = id
self.setWindowTitle(self.mode[id])
self.viewer3D.select_mode(label=self.label_, flag=self.flag_)
def show_app():
app = qApp(sys.argv)
win = HumanShapeAnalysisDemo()
win.show()
sys.exit(app.exec_())
if __name__ == "__main__":
show_app()
