Skip to content

Commit 46c3768

Browse files
committed
Merge branch 'master' of https://github.com/jonwright/ImageD11
2 parents 75d11f7 + e92b81b commit 46c3768

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

ImageD11/forward_model/ori_converter.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -109,9 +109,14 @@ def quat2u(q):
109109

110110
@njit
111111
def rod2quat(r):
112-
113-
s = 1 / np.sqrt(1 + np.linalg.norm(r)**2)
114-
return np.array([s, s * r[0], s * r[1], s * r[2]])
112+
norm_sq = r[0]**2 + r[1]**2 + r[2]**2
113+
s = 1 / np.sqrt(1 + norm_sq)
114+
q = np.empty(4, dtype = np.float64)
115+
q[0] = s
116+
q[1] = s*r[0]
117+
q[2] = s*r[1]
118+
q[3] = s*r[2]
119+
return q
115120

116121

117122
@njit
@@ -466,4 +471,4 @@ def u2euler(mat, axes='rzxz'):
466471
if az < 0:
467472
az = az + 2*np.pi
468473

469-
return np.rad2deg(ax), np.rad2deg(ay), np.rad2deg(az)
474+
return np.rad2deg(ax), np.rad2deg(ay), np.rad2deg(az)

0 commit comments

Comments
 (0)