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

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