forked from sympy/sympy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
plot_camera.py
125 lines (106 loc) · 3.71 KB
/
plot_camera.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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
from pyglet.gl import *
from plot_rotation import get_spherical_rotatation
from util import get_model_matrix
from util import screen_to_model, model_to_screen
from util import vec_subs
class PlotCamera(object):
min_dist = 0.05
max_dist = 500.0
min_ortho_dist = 100.0
max_ortho_dist = 10000.0
_default_dist = 6.0
_default_ortho_dist = 600.0
rot_presets = {
'xy': (0, 0, 0),
'xz': (-90, 0, 0),
'yz': (0, 90, 0),
'perspective': (-45, 0, -45)
}
def __init__(self, window, ortho=False):
self.window = window
self.axes = self.window.plot.axes
self.ortho = ortho
self.reset()
def init_rot_matrix(self):
glPushMatrix()
glLoadIdentity()
self._rot = get_model_matrix()
glPopMatrix()
def set_rot_preset(self, preset_name):
self.init_rot_matrix()
try:
r = self.rot_presets[preset_name]
except AttributeError:
raise ValueError("%s is not a valid rotation preset." % preset_name)
try:
self.euler_rotate(r[0], 1, 0, 0)
self.euler_rotate(r[1], 0, 1, 0)
self.euler_rotate(r[2], 0, 0, 1)
except AttributeError:
pass
def reset(self):
self._dist = 0.0
self._x, self._y = 0.0, 0.0
self._rot = None
if self.ortho:
self._dist = self._default_ortho_dist
else:
self._dist = self._default_dist
self.init_rot_matrix()
def mult_rot_matrix(self, rot):
glPushMatrix()
glLoadMatrixf(rot)
glMultMatrixf(self._rot)
self._rot = get_model_matrix()
glPopMatrix()
def setup_projection(self):
glMatrixMode(GL_PROJECTION)
glLoadIdentity()
if self.ortho:
# yep, this is pseudo ortho (don't tell anyone)
gluPerspective(0.3, float(self.window.width)/float(self.window.height),
self.min_ortho_dist-0.01, self.max_ortho_dist+0.01)
else:
gluPerspective(30.0, float(self.window.width)/float(self.window.height),
self.min_dist-0.01, self.max_dist+0.01)
glMatrixMode(GL_MODELVIEW)
def _get_scale(self):
return 1.0, 1.0, 1.0
def apply_transformation(self):
glLoadIdentity()
glTranslatef(self._x, self._y, -self._dist)
if self._rot is not None:
glMultMatrixf(self._rot)
glScalef(*self._get_scale())
def spherical_rotate(self, p1, p2, sensitivity=1.0):
mat = get_spherical_rotatation(p1, p2, self.window.width,
self.window.height, sensitivity)
if mat is not None:
self.mult_rot_matrix(mat)
def euler_rotate(self, angle, x, y, z):
glPushMatrix()
glLoadMatrixf(self._rot)
glRotatef(angle, x, y, z)
self._rot = get_model_matrix()
glPopMatrix()
def zoom_relative(self, clicks, sensitivity):
if self.ortho:
dist_d = clicks * sensitivity * 50.0
min_dist = self.min_ortho_dist
max_dist = self.max_ortho_dist
else:
dist_d = clicks * sensitivity
min_dist = self.min_dist
max_dist = self.max_dist
new_dist = (self._dist-dist_d)
if (clicks < 0 and new_dist < max_dist) or new_dist > min_dist:
self._dist = new_dist
def mouse_translate(self, x, y, dx, dy):
glPushMatrix()
glLoadIdentity()
glTranslatef(0, 0, -self._dist)
z = model_to_screen(0, 0, 0)[2]
d = vec_subs(screen_to_model(x, y, z), screen_to_model(x-dx, y-dy, z))
glPopMatrix()
self._x += d[0]
self._y += d[1]