public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Make sure rotate_vector works on 2d vectors

Browse source
This commit is contained in:
Grant Sanderson 2019-01-04 12:48:05 -08:00
parent 56f331e8c4
commit 662598fcef
1 changed files with 15 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
manimlib/utils/space_ops.py
View file

@ -53,6 +53,12 @@ def quaternion_conjugate(quaternion):
def rotate_vector(vector, angle, axis=OUT):
if len(vector) == 2:
# Use complex numbers...because why not
z = complex(*vector) * np.exp(complex(0, angle))
return np.array([z.real, z.imag])
elif len(vector) == 3:
# Use quaternions...because why not
quat = quaternion_from_angle_axis(angle, axis)
quat_inv = quaternion_conjugate(quat)
product = reduce(
@ -60,6 +66,8 @@ def rotate_vector(vector, angle, axis=OUT):
[quat, np.append(0, vector), quat_inv]
)
return product[1:]
else:
raise Exception("vector must be of dimension 2 or 3")
def thick_diagonal(dim, thickness=2):