l j0dZddlmZgdZddlZddlmZddlmZddl Z ddl m Z m Z ddlmZdd lmZdd lmZerdd lmZd)dZd*dZd+dZd,dZe dfd-dZde zfd.d Z d/d0d'Zd1d(ZdS)2z7Utility functions for continuous animation of mobjects.) annotations)assert_is_mobject_methodalwaysf_always always_redraw always_shift always_rotateturn_animation_into_updatercycle_animationN)Callable) TYPE_CHECKING)DEGREESRIGHT)Mobject) OpenGLMobject) normalize) Animationmethodr returnNonec|tj|sJ|j}t|tt fsJdSN)inspectismethod__self__ isinstancerr)rmobjects h/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/animation/updaters/mobject_update_utils.pyrr sA  F # ### #oG g7 8 888 888rczt||j}|j|fd|S)Nc|gRiSr)margsfunckwargss rzalways..*s!$$q":4":":":6":":rrr__func__ add_updater)rr$r&rr%s `` @rrr&sJV$$$oG ?D ::::::;;; NrCallable[[Mobject], None]c~t||j}|jfd}|||S)z More functional version of always, where instead of taking in args, it takes in functions which output the relevant arguments. c8dD}|g|RidS)Nc"g|] }| Sr"r").0 arg_generators r z-f_always..updater..9sDDDM DDDrr")mobr$arg_generatorsr%r&s rupdaterzf_always..updater8s<DD^DDD S"4"""6"""""rr()rr3r&rr4r%s `` @rrr.sa V$$$oG ?D#######    Nrr%Callable[[], Mobject]cPfdS)aRedraw the mobject constructed by a function every frame. This function returns a mobject with an attached updater that continuously regenerates the mobject according to the specified function. Parameters ---------- func A function without (required) input arguments that returns a mobject. Examples -------- .. manim:: TangentAnimation class TangentAnimation(Scene): def construct(self): ax = Axes() sine = ax.plot(np.sin, color=RED) alpha = ValueTracker(0) point = always_redraw( lambda: Dot( sine.point_from_proportion(alpha.get_value()), color=BLUE ) ) tangent = always_redraw( lambda: TangentLine( sine, alpha=alpha.get_value(), color=YELLOW, length=4 ) ) self.add(ax, sine, point, tangent) self.play(alpha.animate.set_value(1), rate_func=linear, run_time=2) c>Sr)become)_r%r2s rr'zalways_redraw..iscjj00rr*)r%r2s`@rrr@s6P $&&COO00000111 Jrg?r directionnp.ndarray[np.float64]ratefloatc<|fd|S)a~A mobject which is continuously shifted along some direction at a certain rate. Parameters ---------- mobject The mobject to shift. direction The direction to shift. The vector is normalized, the specified magnitude is not relevant. rate Length in Manim units which the mobject travels in one second along the specified direction. Examples -------- .. manim:: ShiftingSquare class ShiftingSquare(Scene): def construct(self): sq = Square().set_fill(opacity=1) tri = Triangle() VGroup(sq, tri).arrange(LEFT) # construct a square which is continuously # shifted to the right always_shift(sq, RIGHT, rate=5) self.add(sq) self.play(tri.animate.set_fill(opacity=1)) cT||ztzSr)shiftr)r#dtr;r=s rr'zalways_shift..s$aggb4i)I:N:N.N&O&Orr:)rr;r=s ``rrrms-F OOOOOPPP Nrc <|fd|S)a&A mobject which is continuously rotated at a certain rate. Parameters ---------- mobject The mobject to be rotated. rate The angle which the mobject is rotated by over one second. kwags Further arguments to be passed to :meth:`.Mobject.rotate`. Examples -------- .. manim:: SpinningTriangle class SpinningTriangle(Scene): def construct(self): tri = Triangle().set_fill(opacity=1).set_z_index(2) sq = Square().to_edge(LEFT) # will keep spinning while there is an animation going on always_rotate(tri, rate=2*PI, about_point=ORIGIN) self.add(tri, sq) self.play(sq.animate.to_edge(RIGHT), rate_func=linear, run_time=1) c&|j|zfiSr)rotate)r#rBr&r=s rr'zalways_rotate..shahrDy&C&CF&C&Crr:)rr=r&s ``rr r s,: CCCCCDDD NrF animationrcyclebooldelayc j}d_| _dfd ||S)a Add an updater to the animation's mobject which applies the interpolation and update functions of the animation If cycle is True, this repeats over and over. Otherwise, the updater will be popped upon completion The ``delay`` parameter is the delay (in seconds) before the animation starts.. Examples -------- .. manim:: WelcomeToManim class WelcomeToManim(Scene): def construct(self): words = Text("Welcome to") banner = ManimBanner().scale(0.5) VGroup(words, banner).arrange(DOWN) turn_animation_into_updater(Write(words, run_time=0.9)) self.add(words) self.wait(0.5) self.play(banner.expand(), run_time=0.5) Fr#rrBr>c"jdkr}|dkrUd||dSj|z }r|dz}nGt j|dd}|dkr+|dS||xj|z c_dS)Nr) total_time get_run_time interpolateupdate_mobjectsfinishremove_updaternpclip)r#rBrun_time time_ratioalpharGrHupdates rrYz+turn_animation_into_updater..updates)  1 $ $ --//H1}}%%a((())"---  """  ((("-8J "Q Aq11A::$$&&&$$V,,,F  ! !% ( ( (  % %b ) ) )"r)r#rrBr>)rsuspend_mobject_updatingbeginrNr*)rGrHrJr&rrYs`` @rr r st8G).I& OO!6I########4  Nrc  t|fddi|S)NrHT)r )rGr&s rr r s &y G G G G GGr)rr rr)rr rr)rr+rr)r%r5rr)rrr;r<r=r>rr)rrr=r>rr)Fr)rGrrHrIrJr>rr)rGrrr)__doc__ __future__r__all__rcollections.abcr typingr numpyrTmanim.constantsrrmanim.mobject.mobjectr manim.openglrmanim.utils.space_opsrmanim.animation.animationrrrrrrr r r r"rrrhs==""""""   $$$$$$ ********))))))&&&&&&++++++43333339999 $****\;@s$$$$$N35w,D?@<<<<<~HHHHHHr