public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-11-14 15:17:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Small fixes to PMobject

Browse source
This commit is contained in:
Grant Sanderson 2021-10-03 10:27:34 -07:00
parent 5f878a2c1a
commit a4afbfd739
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
manimlib/mobject/types/point_cloud_mobject.py
View file

@ -20,6 +20,11 @@ class PMobject(Mobject):
self.data[key] = resize_array(self.data[key], size) self.data[key] = resize_array(self.data[key], size)
return self return self
def set_points(self, points):
super().set_points(points)
self.resize_points(len(points))
return self
def add_points(self, points, rgbas=None, color=None, opacity=None): def add_points(self, points, rgbas=None, color=None, opacity=None):
""" """
points must be a Nx3 numpy array, as must rgbas if it is not None points must be a Nx3 numpy array, as must rgbas if it is not None
@ -87,6 +92,11 @@ class PMobject(Mobject):
lower_index = int(a * pmobject.get_num_points()) lower_index = int(a * pmobject.get_num_points())
upper_index = int(b * pmobject.get_num_points()) upper_index = int(b * pmobject.get_num_points())
for key in self.data: for key in self.data:
if key == "bounding_box":
continue
# data_len = len(self.data[key])
# lower_index = int(a * data_len)
# upper_index = int(np.ceil(b * data_len))
self.data[key] = pmobject.data[key][lower_index:upper_index] self.data[key] = pmobject.data[key][lower_index:upper_index]
return self return self