public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Only recompute VMobject.get_unit_normal as needed

Browse source
This commit is contained in:
Grant Sanderson 2024-08-21 12:26:42 -05:00
parent 9a5386b022
commit 941513d68c
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
manimlib/mobject/types/vectorized_mobject.py
View file

@ -116,6 +116,7 @@ class VMobject(Mobject):
self.needs_new_triangulation = True
self.triangulation = np.zeros(0, dtype='i4')
self.needs_new_joint_angles = True
self.needs_new_unit_normal = True
self.outer_vert_indices = np.zeros(0, dtype='i4')
super().__init__(**kwargs)
@ -873,21 +874,22 @@ class VMobject(Mobject):
if self.get_num_points() < 3:
return OUT
if not self.needs_new_unit_normal:
return self.data["base_normal"][1, :]
area_vect = self.get_area_vector()
area = get_norm(area_vect)
if area > 0:
normal = area_vect / area
else:
points = self.get_points()
normal = get_unit_normal(
points[1] - points[0],
points[2] - points[1],
)
p = self.get_points()
normal = get_unit_normal(p[1] - p[0], p[2] - p[1])
self.data["base_normal"][1::2] = normal
self.needs_new_unit_normal = False
return normal
def refresh_unit_normal(self) -> Self:
self.get_unit_normal()
self.needs_new_unit_normal = True
return self
def rotate(
@ -1236,14 +1238,14 @@ class VMobject(Mobject):
self.refresh_triangulation()
if refresh_joints:
self.get_joint_angles(refresh=True)
self.get_unit_normal()
self.refresh_unit_normal()
return self
@triggers_refreshed_triangulation
def append_points(self, points: Vect3Array) -> Self:
assert len(points) % 2 == 0
super().append_points(points)
self.get_unit_normal()
self.refresh_unit_normal()
return self
@triggers_refreshed_triangulation