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

22
manimlib/utils/space_ops.py
View file

@ -53,13 +53,21 @@ def quaternion_conjugate(quaternion):
def rotate_vector(vector, angle, axis=OUT): def rotate_vector(vector, angle, axis=OUT):
quat = quaternion_from_angle_axis(angle, axis) if len(vector) == 2:
quat_inv = quaternion_conjugate(quat) # Use complex numbers...because why not
product = reduce( z = complex(*vector) * np.exp(complex(0, angle))
quaternion_mult, return np.array([z.real, z.imag])
[quat, np.append(0, vector), quat_inv] elif len(vector) == 3:
) # Use quaternions...because why not
return product[1:] quat = quaternion_from_angle_axis(angle, axis)
quat_inv = quaternion_conjugate(quat)
product = reduce(
quaternion_mult,
[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): def thick_diagonal(dim, thickness=2):