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

Bug fix to VMobject.set_sheen

Browse source
This commit is contained in:
Grant Sanderson 2018-08-13 14:13:52 -07:00
parent be52b795df
commit d80dbb71d8
1 changed files with 14 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
mobject/types/vectorized_mobject.py
View file

@ -67,12 +67,9 @@ class VMobject(Mobject):
opacity=self.background_stroke_opacity, opacity=self.background_stroke_opacity,
family=self.propagate_style_to_family, family=self.propagate_style_to_family,
) )
self.set_sheen_direction(
self.sheen_direction,
family=self.propagate_style_to_family
)
self.set_sheen( self.set_sheen(
self.sheen, factor=self.sheen,
direction=self.sheen_direction,
family=self.propagate_style_to_family family=self.propagate_style_to_family
) )
return self return self
@ -111,7 +108,9 @@ class VMobject(Mobject):
# one. 99% of the time they'll be the same. # one. 99% of the time they'll be the same.
curr_rgbas = getattr(self, array_name) curr_rgbas = getattr(self, array_name)
if len(curr_rgbas) < len(rgbas): if len(curr_rgbas) < len(rgbas):
curr_rgbas = stretch_array_to_length(len(rgbas)) curr_rgbas = stretch_array_to_length(
curr_rgbas, len(rgbas)
)
setattr(self, array_name, curr_rgbas) setattr(self, array_name, curr_rgbas)
elif len(rgbas) < len(curr_rgbas): elif len(rgbas) < len(curr_rgbas):
rgbas = stretch_array_to_length(len(curr_rgbas)) rgbas = stretch_array_to_length(len(curr_rgbas))
@ -257,12 +256,16 @@ class VMobject(Mobject):
self.sheen_direction = direction self.sheen_direction = direction
return self return self
def set_sheen(self, factor, family=True): def set_sheen(self, factor, direction=None, family=True):
if family: if family:
for submob in self.submobject_family(): for submob in self.submobjects:
submob.sheen = factor submob.set_sheen(factor, direction, family)
else: self.sheen = factor
self.sheen = factor if direction is not None:
# family set to false because recursion will
# already be handled above
self.set_sheen_direction(direction, family=False)
# Reset color to put sheen into effect
self.set_stroke(self.get_stroke_color(), family=family) self.set_stroke(self.get_stroke_color(), family=family)
self.set_fill(self.get_fill_color(), family=family) self.set_fill(self.get_fill_color(), family=family)
return self return self