l j(@^UdZddlmZddlmZmZmZddlmZm Z ddl Z ddl m Z ddlmZmZddlmZdd lmZmZdd lmZmZdd lmZdd lmZdd lmZddlm Z er ddl!m"Z"ddl#m$Z$gdZ%dZ&de'd<GddeZ(Gdde(Z)Gdde(Z*Gdde*Z+dS)z1Tools for displaying multiple animations at once.) annotations)CallableIterableSequence) TYPE_CHECKINGAnyN)config) Animationprepare_animation) RendererType)GroupMobject) OpenGLGroup OpenGLMobject)Scene)remove_list_redundancies)flatten_iterable_parameters)linear) OpenGLVGroup)VGroup)AnimationGroup Succession LaggedStartLaggedStartMapg?floatDEFAULT_LAGGED_START_LAG_RATIOcxeZdZdZddeddd"fdZd#dZd$dZd%dZd$dZ d%dZ d&dZ d'dZ d$dZ d(d!ZxZS))raPlays a group or series of :class:`~.Animation`. Parameters ---------- animations Sequence of :class:`~.Animation` objects to be played. group A group of multiple :class:`~.Mobject`. run_time The duration of the animation in seconds. rate_func The function defining the animation progress based on the relative runtime (see :mod:`~.rate_functions`) . lag_ratio Defines the delay after which the animation is applied to submobjects. A lag_ratio of ``n.nn`` means the next animation will play when ``nnn%`` of the current animation has played. Defaults to 0.0, meaning that all animations will be played together. This does not influence the total runtime of the animation. Instead the runtime of individual animations is adjusted so that the complete animation has the defined run time. Nr)grouprun_time rate_func lag_ratio animationsAnimation | Iterable[Animation]r2Group | VGroup | OpenGLGroup | OpenGLVGroup | Noner float | Noner Callable[[float], float]r!rkwargsrct|}d|D|_||_|Wtd|jD}tdt jkrt||_nt||_n||_tj |jf|j|d|| ||_ dS)Nc,g|]}t|S)r .0anims V/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/animation/composition.py z+AnimationGroup.__init__..@s!HHHt,T22HHHcDg|]}||jSr*) is_introducermobjectr+s r.r/z+AnimationGroup.__init__..Ds+VVV$ASASAUAUVVVVr0renderer)r r!)rr"r rr r OPENGLrrr super__init__ init_run_timer) selfrrr r!r"r'arg_animmobjects __class__s r.r7zAnimationGroup.__init__6s/z::HHxHHH" =/VV$/VVVHj!\%888JUK #H- DJ J "&.I  IO    $11(;; r0return!Sequence[Mobject | OpenGLMobject]c*t|jSN)listrr9s r.get_all_mobjectszAnimationGroup.get_all_mobjectsSsDJr0Nonec|jstd|dd|_|jr|j|jD]}|dS)NTrying to play Q without animations, this is not supported. Please add at least one subanimation.)r" ValueErroranim_group_timesuspend_mobject_updatingrsuspend_updatingbeginr9r-s r.rMzAnimationGroup.beginVs 8$888  #  ( * J ' ' ) ) )O  D JJLLLL  r0scenercD|jD]}||dSr@)r" _setup_scener9rOr-s r.rQzAnimationGroup._setup_scenebs4O % %D   e $ $ $ $ % %r0c|jD]}|d|jdd<d|jdd<|jr|jdSdS)NT)r"finish anims_begunanims_finishedrKrresume_updatingrNs r.rTzAnimationGroup.finishfssO  D KKMMMM"!%AAA  ( ) J & & ( ( ( ( ( ) )r0c|||jD]*}|jr |j|_||+dSr@) _on_finishr"removerclean_up_from_scenerRs r.r[z"AnimationGroup.clean_up_from_scenensY O , ,D| ,#|  $ $U + + + + , ,r0dtcx|jd|j|jzD]}||dS)Nr-)anims_with_timingsrUrVupdate_mobjects)r9r\r-s r.r_zAnimationGroup.update_mobjectsusR+F3   33 3  % %D   $ $ $ $ % %r0c|t|jdd|_||jn|S)a(Calculates the run time of the animation, if different from ``run_time``. Parameters ---------- run_time The duration of the animation in seconds. Returns ------- run_time The duration of the animation in seconds. endr)default)build_animations_with_timingsmaxr^ max_end_time)r9rs r.r8zAnimationGroup.init_run_time{sH **,,,  7 >JJJ$,$4t  (Br0ctjd|jD}|jd}gd}tj|||_tj|t |_tj|t |_|dkrdS|dd|j z}|j|jd<tj ||jdd d<|jd|z|jd <dS) zDCreates a list of triplets of the form (anim, start_time, end_time).cg|] }|j Sr*)rr+s r.r/z@AnimationGroup.build_animations_with_timings..sHHHdmHHHr0r))r-O)startf8)rarj)dtypeNr-rira) nparrayr"shapezerosr^boolrUrVr!add accumulate)r9 run_timesnum_animationsrklagss r.rcz,AnimationGroup.build_animations_with_timingssHHHHHHII "+???.0h~U.S.S.S')xd'K'K'K*,(>*N*N*N Q   F"~.*./'/1v/@/@/F/F(,)-)@)II)U&&&r0alphac|||jz}||jk}|j}||dk}||dk}||j|z|j|zz}|d|dz }|dk} d|| <||dz |z } |r d| | dk| z<n d| | dk| z<t |d| dD]\} } | | ||_||_||_dS)Nrirarrmr-T)strict)r rerJr^rUrVzip interpolate) r9rxrJtime_goes_backawt new_begun new_finished to_updateruwith_zero_run_time sub_alphasanim_to_update sub_alphas r.r|zAnimationGroup.interpolatesP ..//$2CC(4+??%#s7|3 &U3    )t/B.Bl].R S e$y'99 &!^() $%% '(::iG  B@AJ Q*<< = =@AJ Q*<< =), f z$* * *  2 2 %NI  & &y 1 1 1 1.$*r0) r"r#rr$rr%r r&r!rr'r)r=r>r=rD)rOrr=rDr\rr=rD)rr%r=rrxrr=rD)__name__ __module__ __qualname____doc__rr7rCrMrQrTr[r_r8rcr| __classcell__r<s@r.rrs4EI!%.4 <<<<<<<<:        %%%%)))),,,,%%%% CCCC.VVVV !+!+!+!+!+!+!+!+r0rcbeZdZdZdddfd Zdd ZddZddZddZddZ ddZ ddZ xZ S) raPlays a series of animations in succession. Parameters ---------- animations Sequence of :class:`~.Animation` objects to be played. lag_ratio Defines the delay after which the animation is applied to submobjects. A lag_ratio of ``n.nn`` means the next animation will play when ``nnn%`` of the current animation has played. Defaults to 1.0, meaning that the next animation will begin when 100% of the current animation has played. This does not influence the total runtime of the animation. Instead the runtime of individual animations is adjusted so that the complete animation has the defined run time. Examples -------- .. manim:: SuccessionExample class SuccessionExample(Scene): def construct(self): dot1 = Dot(point=LEFT * 2 + UP * 2, radius=0.16, color=BLUE) dot2 = Dot(point=LEFT * 2 + DOWN * 2, radius=0.16, color=MAROON) dot3 = Dot(point=RIGHT * 2 + DOWN * 2, radius=0.16, color=GREEN) dot4 = Dot(point=RIGHT * 2 + UP * 2, radius=0.16, color=YELLOW) self.add(dot1, dot2, dot3, dot4) self.play(Succession( dot1.animate.move_to(dot2), dot2.animate.move_to(dot3), dot3.animate.move_to(dot4), dot4.animate.move_to(dot1) )) rmr!r"r r!rr'rc>tj|d|i|dSNr!r6r7r9r!r"r'r<s r.r7zSuccession.__init__s**D DVDDDDDr0r=rDcd|jstd|d|ddS)NrFrGr)r"rIupdate_active_animationrBs r.rMzSuccession.beginsP 8$888  $$Q'''''r0cN|j||jdSdSr@)active_animationnext_animationrBs r.rTzSuccession.finishs4#/    ! ! !#/////r0r\cL|jr|j|dSdSr@)rr_)r9r\s r.r_zSuccession.update_mobjectss4   6  ! 1 1" 5 5 5 5 5 6 6r0rO Scene | Nonec|dS|r?|jD]7}|s!|j||j8||_dSr@)r2r"r3rsrOrRs r.rQzSuccession._setup_scenesl = F      , , ,))++, 0HIIdl+++ r0indexintcf||_|t|jkrd|_d|_d|_dS|j||_|j|j|j|j |d|_|j |d|_dS)Nrira) active_indexlenr"ractive_start_timeactive_end_timerQrOrMr^)r9rs r.rz"Succession.update_active_animations! C(( ( (6:D !37D "15D $(OE$:D !  ! . .tz : : :  ! ' ' ) ) )%)%tj|d|i|dSrrrs r.r7zLaggedStart.__init__Vs, *D DVDDDDDr0rrrrrrr7rrs@r.rr)sc**^:EEEEEEEEEEEEr0rc,eZdZdZddefdfd ZxZS)raTPlays a series of :class:`~.Animation` while mapping a function to submobjects. Parameters ---------- animation_class :class:`~.Animation` to apply to mobject. mobject :class:`~.Mobject` whose submobjects the animation, and optionally the function, are to be applied. arg_creator Function which will be applied to :class:`~.Mobject`. run_time The duration of the animation in seconds. lag_ratio Defines the delay after which the animation is applied to submobjects. A lag_ratio of ``n.nn`` means the next animation will play when ``nnn%`` of the current animation has played. Defaults to 0.05, meaning that the next animation will begin when 5% of the current animation has played. This does not influence the total runtime of the animation. Instead the runtime of individual animations is adjusted so that the complete animation has the defined run time. kwargs Further keyword arguments that are passed to `animation_class`. Examples -------- .. manim:: LaggedStartMapExample class LaggedStartMapExample(Scene): def construct(self): title = Tex("LaggedStartMap").to_edge(UP, buff=LARGE_BUFF) dots = VGroup( *[Dot(radius=0.16) for _ in range(35)] ).arrange_in_grid(rows=5, cols=7, buff=MED_LARGE_BUFF) self.add(dots, title) # Animate yellow ripple effect for mob in dots, title: self.play(LaggedStartMap( ApplyMethod, mob, lambda m : (m.set_color, YELLOW), lag_ratio = 0.1, rate_func = there_and_back, run_time = 2 )) Nanimation_classtype[Animation]r3r arg_creator)Callable[[Mobject], Iterable[Any]] | Nonerrr!r'rc  d d}|fd|D}t| d vr d fd|D} tj| ||ddS) Nmobrr=c|Sr@r*)rs r.identityz)LaggedStartMap.__init__..identitys r0c&g|] }|Sr*r*)r,submobrs r.r/z+LaggedStartMap.__init__..s#???V[[((???r0r!c g|] }|i Sr*r*)r,args anim_kwargsrs r.r/z+LaggedStartMap.__init__..s(RRRoot;{;;RRRr0)rr!)rrr=r)dictpopr6r7) r9rr3rrr!r'r args_listr"rr<s ` ` @r.r7zLaggedStartMap.__init__s      #K????w??? 6ll + % % OOK ( ( (RRRRR RRR *x9MMMMMMr0) rrr3rrrrrr!rr'rrrs@r.rr_se..hBF9 NNNNNNNNNNNr0r),r __future__rcollections.abcrrrtypingrrnumpyrn manim._configr manim.animation.animationr r manim.constantsr manim.mobject.mobjectr r#manim.mobject.opengl.opengl_mobjectrrmanim.scene.scenermanim.utils.iterablesrmanim.utils.parameter_parsingrmanim.utils.rate_functionsr.manim.mobject.opengl.opengl_vectorized_mobjectr&manim.mobject.types.vectorized_mobjectr__all__r__annotations__rrrrr*r0r.rsB777""""""8888888888%%%%%%%% BBBBBBBB((((((00000000JJJJJJJJ######::::::EEEEEE------>KKKKKK====== K K K)-,,,,e+e+e+e+e+Ye+e+e+P`8`8`8`8`8`8`8`8F3E3E3E3E3E.3E3E3ElFNFNFNFNFN[FNFNFNFNFNr0