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

Added PiCreature.get_mode, along with other changes to keep it working

Browse source
This commit is contained in:
Grant Sanderson 2018-05-16 18:41:48 -07:00
parent d8ed9d188b
commit 1c71a00073
1 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
for_3b1b_videos/pi_creature.py
View file

@ -47,6 +47,7 @@ class PiCreature(SVGMobject):
def __init__(self, mode="plain", **kwargs):
digest_config(self, kwargs)
self.mode = mode
self.parts_named = False
try:
svg_file = os.path.join(
@ -68,6 +69,13 @@ class PiCreature(SVGMobject):
if self.start_corner is not None:
self.to_corner(self.start_corner)
def align_data(self, mobject):
# This ensures that after a transform into a different mode,
# the pi creatures mode will be updated appropriately
SVGMobject.align_data(self, mobject)
if isinstance(mobject, PiCreature):
self.mode = mobject.get_mode()
def name_parts(self):
self.mouth = self.submobjects[MOUTH_INDEX]
self.body = self.submobjects[BODY_INDEX]
@ -104,18 +112,22 @@ class PiCreature(SVGMobject):
def change_mode(self, mode):
new_self = self.__class__(
mode = mode,
mode=mode,
)
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.shift(self.eyes.get_center() - new_self.eyes.get_center())
if hasattr(self, "purposeful_looking_direction"):
new_self.look(self.purposeful_looking_direction)
Transform(self, new_self).update(1)
self.mode = mode
return self
def get_mode(self):
return self.mode
def look(self, direction):
norm = np.linalg.norm(direction)
if norm == 0: