opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-03 04:04:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow for tracking which data keys should always have the same size as points

Browse Source
This commit is contained in:
Grant Sanderson
2023-01-13 09:47:06 -08:00
parent 108bb3da44
commit 440138aac5
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
manimlib/mobject/mobject.py
View File

@ -67,6 +67,7 @@ class Mobject(object):
shader_dtype: Sequence[Tuple[str, type, Tuple[int]]] = [
('point', np.float32, (3,)),
]
aligned_data_keys = ['points']
def __init__(
self,
@ -171,8 +172,8 @@ class Mobject(object):
new_length: int,
resize_func: Callable[[np.ndarray, int], np.ndarray] = resize_array
):
if new_length != len(self.data["points"]):
self.data["points"] = resize_func(self.data["points"], new_length)
for key in self.aligned_data_keys:
self.data[key] = resize_func(self.data[key], new_length)
self.refresh_bounding_box()
return self