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

Added Mobject.get_pieces

Browse source
This commit is contained in:
Grant Sanderson 2018-08-20 15:49:31 -07:00
parent 7ab432f419
commit 1c348cbe95
1 changed files with 14 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
mobject/mobject.py
View file

@ -47,14 +47,14 @@ class Mobject(Container):
if self.name is None: if self.name is None:
self.name = self.__class__.__name__ self.name = self.__class__.__name__
self.updaters = [] self.updaters = []
self.init_points() self.reset_points()
self.generate_points() self.generate_points()
self.init_colors() self.init_colors()
def __str__(self): def __str__(self):
return str(self.name) return str(self.name)
def init_points(self): def reset_points(self):
self.points = np.zeros((0, self.dim)) self.points = np.zeros((0, self.dim))
def init_colors(self): def init_colors(self):
@ -773,6 +773,17 @@ class Mobject(Container):
def point_from_proportion(self, alpha): def point_from_proportion(self, alpha):
raise Exception("Not implemented") raise Exception("Not implemented")
def get_pieces(self, n_pieces):
template = self.copy()
template.submobjects = []
alphas = np.linspace(0, 1, n_pieces + 1)
return Group(*[
template.copy().pointwise_become_partial(
self, a1, a2
)
for a1, a2 in zip(alphas[:-1], alphas[1:])
])
# Family matters # Family matters
def __getitem__(self, value): def __getitem__(self, value):
@ -909,7 +920,7 @@ class Mobject(Container):
def push_self_into_submobjects(self): def push_self_into_submobjects(self):
copy = self.copy() copy = self.copy()
copy.submobjects = [] copy.submobjects = []
self.init_points() self.reset_points()
self.add(copy) self.add(copy)
return self return self