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
|
"""Quaternion module
This module provides conversion routines between Matrices, Quaternions (rotations around
an axis) and Eulers.
(c) 2000, onk@section5.de """
# NON PUBLIC XXX
from math import sin, cos, acos
from util import vect
reload(vect)
Vector = vect.Vector
Matrix = vect.Matrix
class Quat:
"""Simple Quaternion class
Usually, you create a quaternion from a rotation axis (x, y, z) and a given
angle 'theta', defining the right hand rotation:
q = fromRotAxis((x, y, z), theta)
This class supports multiplication, providing an efficient way to
chain rotations"""
def __init__(self, w = 1.0, x = 0.0, y = 0.0, z = 0.0):
self.v = (w, x, y, z)
def asRotAxis(self):
"""returns rotation axis (x, y, z) and angle phi (right hand rotation)"""
phi2 = acos(self.v[0])
if phi2 == 0.0:
return Vector(0.0, 0.0, 1.0), 0.0
else:
s = 1 / (sin(phi2))
v = Vector(s * self.v[1], s * self.v[2], s * self.v[3])
return v, 2.0 * phi2
def __mul__(self, other):
w1, x1, y1, z1 = self.v
w2, x2, y2, z2 = other.v
w = w1*w2 - x1*x2 - y1*y2 - z1*z2
x = w1*x2 + x1*w2 + y1*z2 - z1*y2
y = w1*y2 - x1*z2 + y1*w2 + z1*x2
z = w1*z2 + x1*y2 - y1*x2 + z1*w2
return Quat(w, x, y, z)
def asMatrix(self):
w, x, y, z = self.v
v1 = Vector(1.0 - 2.0 * (y*y + z*z), 2.0 * (x*y + w*z), 2.0 * (x*z - w*y))
v2 = Vector(2.0 * (x*y - w*z), 1.0 - 2.0 * (x*x + z*z), 2.0 * (y*z + w*x))
v3 = Vector(2.0 * (x*z + w*y), 2.0 * (y*z - w*x), 1.0 - 2.0 * (x*x + y*y))
return Matrix(v1, v2, v3)
# def asEuler1(self, transp = 0):
# m = self.asMatrix()
# if transp:
# m = m.transposed()
# return m.asEuler()
def asEuler(self, transp = 0):
from math import atan, asin, atan2
w, x, y, z = self.v
x2 = x*x
z2 = z*z
tmp = x2 - z2
r = (w*w + tmp - y*y )
phi_z = atan2(2.0 * (x * y + w * z) , r)
phi_y = asin(2.0 * (w * y - x * z))
phi_x = atan2(2.0 * (w * x + y * z) , (r - 2.0*tmp))
return phi_x, phi_y, phi_z
def fromRotAxis(axis, phi):
"""computes quaternion from (axis, phi)"""
phi2 = 0.5 * phi
s = sin(phi2)
return Quat(cos(phi2), axis[0] * s, axis[1] * s, axis[2] * s)
#def fromEuler1(eul):
#qx = fromRotAxis((1.0, 0.0, 0.0), eul[0])
#qy = fromRotAxis((0.0, 1.0, 0.0), eul[1])
#qz = fromRotAxis((0.0, 0.0, 1.0), eul[2])
#return qz * qy * qx
def fromEuler(eul):
from math import sin, cos
e = eul[0] / 2.0
cx = cos(e)
sx = sin(e)
e = eul[1] / 2.0
cy = cos(e)
sy = sin(e)
e = eul[2] / 2.0
cz = cos(e)
sz = sin(e)
w = cx * cy * cz - sx * sy * sz
x = sx * cy * cz - cx * sy * sz
y = cx * sy * cz + sx * cy * sz
z = cx * cy * sz + sx * sy * cz
return Quat(w, x, y, z)
|