fixed pi creature color bug

This commit is contained in:
Ben Hambrecht 2018-04-09 19:27:13 +02:00
parent 15d582e673
commit c24f7b6926

View file

@ -98,14 +98,15 @@ class PiCreature(SVGMobject):
def set_color(self, color): def set_color(self, color):
self.body.set_fill(color) self.body.set_fill(color)
self.color = color
return self return self
def change_mode(self, mode): def change_mode(self, mode):
new_self = self.__class__( new_self = self.__class__(
mode=mode, mode = mode,
color=self.color
) )
new_self.scale_to_fit_height(self.get_height()) new_self.match_style(self)
new_self.match_height(self)
if self.is_flipped() ^ new_self.is_flipped(): if self.is_flipped() ^ new_self.is_flipped():
new_self.flip() new_self.flip()
new_self.shift(self.eyes.get_center() - new_self.eyes.get_center()) new_self.shift(self.eyes.get_center() - new_self.eyes.get_center())