public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-04-13 09:47:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Changed how pi creatures follow what happens

Browse source
This commit is contained in:
Grant Sanderson 2019-04-21 08:14:35 -07:00
parent 7774c191c6
commit 77a3b47a09
1 changed files with 24 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
manimlib/for_3b1b_videos/pi_creature_scene.py
View file

@ -1,11 +1,10 @@
import itertools as it import itertools as it
import random import random
import copy
from manimlib.animation.transform import ApplyMethod
from manimlib.animation.transform import ReplacementTransform from manimlib.animation.transform import ReplacementTransform
from manimlib.animation.transform import Transform from manimlib.animation.transform import Transform
from manimlib.animation.composition import LaggedStart from manimlib.animation.composition import LaggedStart
from manimlib.animation.update import UpdateFromAlphaFunc
from manimlib.constants import * from manimlib.constants import *
from manimlib.for_3b1b_videos.pi_creature import Mortimer from manimlib.for_3b1b_videos.pi_creature import Mortimer
from manimlib.for_3b1b_videos.pi_creature import PiCreature from manimlib.for_3b1b_videos.pi_creature import PiCreature
@ -18,6 +17,7 @@ from manimlib.mobject.svg.drawings import SpeechBubble
from manimlib.mobject.svg.drawings import ThoughtBubble from manimlib.mobject.svg.drawings import ThoughtBubble
from manimlib.mobject.types.vectorized_mobject import VGroup from manimlib.mobject.types.vectorized_mobject import VGroup
from manimlib.scene.scene import Scene from manimlib.scene.scene import Scene
from manimlib.utils.bezier import interpolate
from manimlib.utils.rate_functions import squish_rate_func from manimlib.utils.rate_functions import squish_rate_func
from manimlib.utils.rate_functions import there_and_back from manimlib.utils.rate_functions import there_and_back
from manimlib.utils.space_ops import get_norm from manimlib.utils.space_ops import get_norm
@ -157,34 +157,31 @@ class PiCreatureScene(Scene):
if not self.any_pi_creatures_on_screen(): if not self.any_pi_creatures_on_screen():
return animations return animations
non_pi_creature_anims = [anim for anim in animations if anim.mobject not in self.get_pi_creatures()] pi_creatures = self.get_on_screen_pi_creatures()
non_pi_creature_anims = [
anim
for anim in animations
if len(set(anim.mobject.get_family()).intersection(pi_creatures)) == 0
]
if len(non_pi_creature_anims) == 0: if len(non_pi_creature_anims) == 0:
return animations return animations
# Look at ending state # Get pi creatures to look at whatever
# is being animated
first_anim = non_pi_creature_anims[0] first_anim = non_pi_creature_anims[0]
first_anim_copy = copy.deepcopy(first_anim) main_mobject = first_anim.mobject
first_anim_copy.begin() animations += [
first_anim_copy.update(1) UpdateFromAlphaFunc(
point_of_interest = first_anim_copy.mobject.get_center() pi_creature,
lambda p, a: p.look_at(
for pi_creature in self.get_pi_creatures(): interpolate(
if pi_creature not in self.get_mobjects(): p.get_look_at_spot(),
continue main_mobject.get_center(),
if pi_creature in first_anim.mobject.get_family(): a,
continue )
anims_with_pi_creature = [anim for anim in animations if pi_creature in anim.mobject.get_family()] ),
for anim in anims_with_pi_creature: )
continue # TODO, this is broken for pi_creature in pi_creatures
if isinstance(anim, Transform): ]
index = anim.mobject.get_family().index(pi_creature)
target_family = anim.target_mobject.get_family()
target = target_family[index]
if isinstance(target, PiCreature):
target.look_at(point_of_interest)
if not anims_with_pi_creature:
animations.append(
ApplyMethod(pi_creature.look_at, point_of_interest)
)
return animations return animations
def blink(self): def blink(self):