public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Removed condition from Mobject.highlight

Browse source
This commit is contained in:
Grant Sanderson 2017-10-26 21:30:59 -07:00
parent 0d8a713950
commit b9aec1e907
2 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
mobject/mobject.py
View file

@ -399,11 +399,17 @@ class Mobject(object):
## Color functions ## Color functions
def highlight(self, color = YELLOW_C, family = True, condition = None): def highlight(self, color = YELLOW_C, family = True):
""" """
Condition is function which takes in one arguments, (x, y, z). Condition is function which takes in one arguments, (x, y, z).
Here it just recurses to submobjects, but in subclasses this
should be further implemented based on the the inner workings
of color
""" """
raise Exception("Not implemented") if family:
for submob in self.submobjects:
submob.highlight(color, family = family)
return self
def gradient_highlight(self, *colors): def gradient_highlight(self, *colors):
self.submobject_gradient_highlight(*colors) self.submobject_gradient_highlight(*colors)

6
mobject/point_cloud_mobject.py
View file

@ -30,14 +30,10 @@ class PMobject(Mobject):
self.rgbas = np.append(self.rgbas, rgbas, axis = 0) self.rgbas = np.append(self.rgbas, rgbas, axis = 0)
return self return self
def highlight(self, color = YELLOW_C, family = True, condition = None): def highlight(self, color = YELLOW_C, family = True):
rgba = color_to_rgba(color) rgba = color_to_rgba(color)
mobs = self.family_members_with_points() if family else [self] mobs = self.family_members_with_points() if family else [self]
for mob in mobs: for mob in mobs:
if condition:
to_change = np.apply_along_axis(condition, 1, mob.points)
mob.rgbas[to_change, :] = rgba
else:
mob.rgbas[:,:] = rgba mob.rgbas[:,:] = rgba
return self return self