class BVHPlayerPy(QMainWindow):
def __init__(self, pathCD):
super().__init__()
self.setMaximumSize(800, 500)
self.pathCurrentDir = pathCD
self.pathMotionFileDir = pathCD.rstrip(os.path.basename(pathCD))
self.setCentralWidget(self.initComponent())
menuBar = self.menuBar()
menuBar.setNativeMenuBar(False)
fileMenu = menuBar.addMenu("&File")
loadAction = QAction("&Open...", self)
loadAction.triggered.connect(self.loadFile)
loadAction.setShortcut("Ctrl+l")
fileMenu.addAction(loadAction)
quitAction = QAction("&Quit...", self)
quitAction.triggered.connect(self.quit)
quitAction.setShortcut("Ctrl+q")
fileMenu.addAction(quitAction)
self.setMenuBar(menuBar)
self.setWindowTitle("BVH Player")
def initComponent(self):
self.drawPanel = GLWidget(self)
self.infoPanel = InfoWidget(self)
self.controlPanel = ControlWidget(self)
self.splitterPanel = SplitWidget(self)
controlLayout = QVBoxLayout()
controlLayout.addWidget(self.infoPanel)
controlLayout.addWidget(self.controlPanel)
controlLayout.addWidget(self.splitterPanel)
mainLayout = QHBoxLayout()
mainLayout.addWidget(self.drawPanel)
mainLayout.addLayout(controlLayout)
mainWidget = QWidget()
mainWidget.setLayout(mainLayout)
return mainWidget
def quit(self):
sys.exit()
def loadFile(self):
filePath = QFileDialog.getOpenFileName(self, "Choose Motion File...", self.pathMotionFileDir, "Biovision Hierarchy (*.bvh)")
if filePath[0] == "":
# print("Error: Motion file is not given")
pass
else:
root, motion, frames, frameTime = BVH.readBVH(filePath[0])
self.pathMotionFileDir = os.path.dirname(filePath[0])
self.drawPanel.setMotion(root, motion, frames, frameTime)
self.infoPanel.initInfo(os.path.basename(filePath[0]), frameTime, frames)
self.controlPanel.setPlayMode(True)
self.splitterPanel.setActive()
self.splitterPanel.initMotionData(os.path.basename(filePath[0]), root, motion, frameTime)
def keyPressEvent(self, event:QKeyEvent):
if event.key() == Qt.Key_Escape:
self.quit()
elif event.key() == Qt.Key_S:
if self.drawPanel.motion is not None:
self.drawPanel.isPlaying = not self.drawPanel.isPlaying
self.controlPanel.setPlayMode(self.drawPanel.isPlaying)
elif event.key() == Qt.Key_F:
self.drawPanel.fastRatio *= 2.0
elif event.key() == Qt.Key_D:
self.drawPanel.fastRatio /= 2.0
elif event.key() == Qt.Key_Right:
if self.drawPanel.frames is not None:
self.drawPanel.frameCount += 1
if self.drawPanel.frameCount >= self.drawPanel.frames:
self.drawPanel.frameCount = 0
elif event.key() == Qt.Key_Left:
if self.drawPanel.frames is not None:
self.drawPanel.frameCount -= 1
if self.drawPanel.frameCount < 0:
self.drawPanel.frameCount = self.drawPanel.frames - 1
else:
pass
