-
Notifications
You must be signed in to change notification settings - Fork 33
/
PyQtCube.py
104 lines (88 loc) · 3.07 KB
/
PyQtCube.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#!/usr/bin/python
from __future__ import absolute_import, division, print_function, unicode_literals
''' Demo showing use of pi3d as a widget on a PyQt window. See PyQt documentation
for details on how to use that.
'''
import sys
import demo
import pi3d
import numpy as np
from PyQt4 import QtGui
from PyQt4 import QtCore
W, H = 640, 480
FPS = 20
class QtWindow(QtGui.QWidget):
def __init__(self):
super(QtWindow, self).__init__()
''' layer set to -128 to hide behind X desktop
'''
self.DISPLAY = pi3d.Display.create(w=W, h=H, layer=-128)
shader = pi3d.Shader('uv_light')
tex = pi3d.Texture('textures/PATRN.PNG')
self.cube = pi3d.Cuboid(z=2)
self.cube.set_draw_details(shader, [tex])
self.cube.show_flag = True # additional attribute stuck onto Cuboid instance
self.cube.last_x = None
self.cube.last_y = None
self.initUI() # not entirely sure why this is done in a different method, just copying others' code!
def initUI(self):
button1 = QtGui.QPushButton('Rotate X 5deg') # some jump rotations
button1.clicked.connect(self.rotX)
button2 = QtGui.QPushButton('Rotate Y 15deg')
button2.clicked.connect(self.rotY)
button3 = QtGui.QPushButton('Rotate Z 20deg')
button3.clicked.connect(self.rotZ)
message = QtGui.QLabel('Left click and\ndrag on image or\nclick these buttons')
vbox = QtGui.QVBoxLayout()
vbox.addWidget(button1)
vbox.addWidget(button2)
vbox.addWidget(button3)
vbox.addWidget(message)
vbox.addStretch(1)
self.img = QtGui.QLabel('image') # self. as needs to be accessed from outside this method
a = np.zeros((H, W, 4), dtype=np.uint8)
im = QtGui.QImage(a, W, H, QtGui.QImage.Format_RGB888)
self.img.setPixmap(QtGui.QPixmap.fromImage(im))
hbox = QtGui.QHBoxLayout()
hbox.addLayout(vbox)
hbox.addWidget(self.img)
hbox.addStretch(1)
self.setLayout(hbox)
self.setWindowTitle('PyQT4 pi3d demo')
self.show()
def pi3d_loop(self):
self.DISPLAY.loop_running()
if self.cube.show_flag:
self.cube.draw()
im = QtGui.QImage(pi3d.screenshot(), W, H, QtGui.QImage.Format_RGB888)
self.img.setPixmap(QtGui.QPixmap.fromImage(im))
self.cube.show_flag = False
def rotX(self):
self.cube.rotateIncX(5.0)
self.cube.show_flag = True
def rotY(self):
self.cube.rotateIncY(15.0)
self.cube.show_flag = True
def rotZ(self):
self.cube.rotateIncZ(20.0)
self.cube.show_flag = True
def mousePressEvent(self, QMouseEvent):
cursor = QtGui.QCursor()
self.cube.last_x = cursor.pos().x()
self.cube.last_y = cursor.pos().y()
def mouseMoveEvent(self, QMouseEvent):
cursor = QtGui.QCursor()
x = cursor.pos().x()
y = cursor.pos().y()
self.cube.rotateIncY(self.cube.last_x - x)
self.cube.rotateIncX(self.cube.last_y - y)
self.cube.last_x = x
self.cube.last_y = y
self.cube.show_flag = True
app = QtGui.QApplication(sys.argv)
win = QtWindow()
timer = QtCore.QTimer()
timer.timeout.connect(win.pi3d_loop)
msfps = int(1000/FPS) # FPS in milliseconds
timer.start(msfps)
sys.exit(app.exec_())