l jSybdZddlmZgdZddlZddlZddlmZmZm Z ddl m Z m Z ddl ZddlmZddlmZmZd d lmZd d lmZd d lmZmZmZmZmZd d lmZm Z d dl!m"Z"m#Z#d dl$m%Z%m&Z&e rd dl'm(Z(d dl m)Z)m*Z*GddeZ+Gdde+Z,Gdde+Z-Gdde+Z.Gdde+Z/Gdde+Z0Gdde0Z1Gd d!e+Z2Gd"d#e2Z3Gd$d%e3Z4Gd&d'e2Z5Gd(d)e2Z6Gd*d+e6Z7Gd,d-e2Z8Gd.d/e+Z9Gd0d1e3Z:Gd2d3e2Z;Gd4d5e+Z<Gd6d7e<Z=Gd8d9e+Z>Gd:d;e+Z?Gd<d=e?Z@dS)>z1Animations transforming one mobject into another.) annotations) TransformReplacementTransformTransformFromCopyClockwiseTransformCounterclockwiseTransform MoveToTarget ApplyMethodApplyPointwiseFunctionApplyPointwiseFunctionToCenter FadeToColor FadeTransformFadeTransformPieces ScaleInPlaceShrinkToCenterRestore ApplyFunction ApplyMatrixApplyComplexFunction CyclicReplaceSwapTransformAnimationsN)CallableIterableSequence) TYPE_CHECKINGAny)MethodWithArgs) OpenGLGroup OpenGLMobject)config) Animation)#DEFAULT_POINTWISE_FUNCTION_RUN_TIMEDEGREESORIGINOUT RendererType)GroupMobject)path_along_arcpath_along_circles)smoothsquish_rate_func)Scene) Point3DLikePoint3DLike_ArrayceZdZdZdddeddfd*fd Zed+dZejd,dZed-dZ e jd.dZ d/fd Z d0dZ d1fd Z d2d!Z d3d#Zd4d)ZxZS)5ruDA Transform transforms a Mobject into a target Mobject. Parameters ---------- mobject The :class:`.Mobject` to be transformed. It will be mutated to become the ``target_mobject``. target_mobject The target of the transformation. path_func A function defining the path that the points of the ``mobject`` are being moved along until they match the points of the ``target_mobject``, see :mod:`.utils.paths`. path_arc The arc angle (in radians) that the points of ``mobject`` will follow to reach the points of the target if using a circular path arc, see ``path_arc_centers``. See also :func:`manim.utils.paths.path_along_arc`. path_arc_axis The axis to rotate along if using a circular path arc, see ``path_arc_centers``. path_arc_centers The center of the circular arcs along which the points of ``mobject`` are moved by the transformation. If this is set and ``path_func`` is not set, then a ``path_along_circles`` path will be generated using the ``path_arc`` parameters and stored in ``path_func``. If ``path_func`` is set, this and the other ``path_arc`` fields are set as attributes, but a ``path_func`` is not generated from it. replace_mobject_with_target_in_scene Controls which mobject is replaced when the transformation is complete. If set to True, ``mobject`` will be removed from the scene and ``target_mobject`` will replace it. Otherwise, ``target_mobject`` is never added and ``mobject`` just takes its shape. Examples -------- .. manim :: TransformPathArc class TransformPathArc(Scene): def construct(self): def make_arc_path(start, end, arc_angle): points = [] p_fn = path_along_arc(arc_angle) # alpha animates between 0.0 and 1.0, where 0.0 # is the beginning of the animation and 1.0 is the end. for alpha in range(0, 11): points.append(p_fn(start, end, alpha / 10.0)) path = VMobject(stroke_color=YELLOW) path.set_points_smoothly(points) return path left = Circle(stroke_color=BLUE_E, fill_opacity=1.0, radius=0.5).move_to(LEFT * 2) colors = [TEAL_A, TEAL_B, TEAL_C, TEAL_D, TEAL_E, GREEN_A] # Positive angles move counter-clockwise, negative angles move clockwise. examples = [-90, 0, 30, 90, 180, 270] anims = [] for idx, angle in enumerate(examples): left_c = left.copy().shift((3 - idx) * UP) left_c.fill_color = colors[idx] right_c = left_c.copy().shift(4 * RIGHT) path_arc = make_arc_path(left_c.get_center(), right_c.get_center(), arc_angle=angle * DEGREES) desc = Text('%d°' % examples[idx]).next_to(left_c, LEFT) # Make the circles in front of the text in front of the arcs. self.add( path_arc.set_z_index(1), desc.set_z_index(2), left_c.set_z_index(3), ) anims.append(Transform(left_c, right_c, path_arc=angle * DEGREES)) self.play(*anims, run_time=2) self.wait() See also -------- :class:`~.ReplacementTransform`, :meth:`~.Mobject.interpolate`, :meth:`~.Mobject.align_data` NrFmobjectMobject | Nonetarget_mobject path_funcCallable | Nonepath_arcfloat path_arc_axis np.ndarraypath_arc_centers&Point3DLike | Point3DLike_Array | None$replace_mobject_with_target_in_sceneboolreturnNonec ||_||_||_|||_n'|j t ||j|j|_||_||n t |_tj |fi|dSN) r:r<r8r6r,r>r*r5super__init__) selfr3r5r6r8r:r<r>kwargs __class__s T/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/animation/transform.pyrEzTransform.__init__s*7HX'   '0DNN  " ./%"DN 1 1-8NNgii  ++F+++++c|jSrC) _path_arcrFs rIr8zTransform.path_arcs ~rJcT||_t|j|j|_dS)N) arc_angleaxis)rLr+r: _path_func)rFr8s rIr8zTransform.path_arcs.!(n#   rJSCallable[[Iterable[np.ndarray], Iterable[np.ndarray], float], Iterable[np.ndarray]]c|jSrCrQrMs rIr6zTransform.path_funcs rJc| ||_dSdSrCrT)rFr6s rIr6zTransform.path_funcs  'DOOO ! rJch||_|j|_tjt jkr |j |jn|j |jt dSrC) create_targetr5copy target_copyr"rendererr(OPENGLr3align_data_and_family align_datarDbeginrFrHs rIr^zTransform.begins#0022.3355 ?l1 1 1 L . .t/? @ @ @ @ L # #D$4 5 5 5  rJMobject | OpenGLMobjectc|jSrC)r5rMs rIrWzTransform.create_targets ""rJscener/ct||jr"||j|jdSdSrC)rDclean_up_from_scener>replacer3r5)rFrbrHs rIrdzTransform.clean_up_from_scenesO ##E***  4 = MM$,(; < < < < < = =rJSequence[Mobject]c6|j|j|j|jgSrC)r3starting_mobjectr5rYrMs rIget_all_mobjectszTransform.get_all_mobjectss# L  !       rJIterable[tuple]c|j|j|jg}tjt jkrtd|DddiStd|DddiS)Nc3>K|]}|VdSrC) get_family.0mobs rI z4Transform.get_all_families_zipped..s,::c))::::::rJstrictTc3>K|]}|VdSrC)family_members_with_pointsrns rIrqz4Transform.get_all_families_zipped..s.FF#S3355FFFFFFrJ)r3rhrYr"rZr(r[zip)rFmobss rIget_all_families_zippedz!Transform.get_all_families_zippedsp L  !    ?l1 1 1::T:::H4HH HFFFFFTtTTTrJ submobjectr*starting_submobjectrYalphac@|||||j|SrC) interpolater6)rFrxryrYrzs rIinterpolate_submobjectz Transform.interpolate_submobjects& 2KWWW rJ)r3r4r5r4r6r7r8r9r:r;r<r=r>r?r@rA)r@r9)r8r9r@rA)r@rR)r6rRr@rAr@rA)r@r`)rbr/r@rAr@rf)r@rj) rxr*ryr*rYr*rzr9r@r)__name__ __module__ __qualname____doc__r'rEpropertyr8setterr6r^rWrdrirwr} __classcell__rHs@rIrr9szJJ^*.%)$'CG5:,,,,,,,BX_   _ X((((      #### ======     UUUUrJrc$eZdZdZdfd ZxZS) ra>Replaces and morphs a mobject into a target mobject. Parameters ---------- mobject The starting :class:`~.Mobject`. target_mobject The target :class:`~.Mobject`. kwargs Further keyword arguments that are passed to :class:`Transform`. Examples -------- .. manim:: ReplacementTransformOrTransform :quality: low class ReplacementTransformOrTransform(Scene): def construct(self): # set up the numbers r_transform = VGroup(*[Integer(i) for i in range(1,4)]) text_1 = Text("ReplacementTransform", color=RED) r_transform.add(text_1) transform = VGroup(*[Integer(i) for i in range(4,7)]) text_2 = Text("Transform", color=BLUE) transform.add(text_2) ints = VGroup(r_transform, transform) texts = VGroup(text_1, text_2).scale(0.75) r_transform.arrange(direction=UP, buff=1) transform.arrange(direction=UP, buff=1) ints.arrange(buff=2) self.add(ints, texts) # The mobs replace each other and none are left behind self.play(ReplacementTransform(r_transform[0], r_transform[1])) self.play(ReplacementTransform(r_transform[1], r_transform[2])) # The mobs linger after the Transform() self.play(Transform(transform[0], transform[1])) self.play(Transform(transform[1], transform[2])) self.wait() r3r*r5r@rAc Btj||fddi|dS)Nr>TrDrErFr3r5rGrHs rIrEzReplacementTransform.__init__,sB ^  JN RX     rJr3r*r5r*r@rArrrrrErrs@rIrrsH--^          rJrc0eZdZdZd fd Zd fd ZxZS) rz^Preserves a copy of the original VMobject and transforms only it's copy to the target VMobjectr3r*r5r@rAc >tj||fi|dSrCrrs rIrEzTransformFromCopy.__init__5s*;;F;;;;;rJrzr9cPtd|z dSN)rDr|rFrzrHs rIr|zTransformFromCopy.interpolate8s% AI&&&&&rJrrzr9r@rA)rrrrrEr|rrs@rIrr2sahh<<<<<<''''''''''rJrc4eZdZdZej fd fd ZxZS) raTransforms the points of a mobject along a clockwise oriented arc. See also -------- :class:`.Transform`, :class:`.CounterclockwiseTransform` Examples -------- .. manim:: ClockwiseExample class ClockwiseExample(Scene): def construct(self): dl, dr = Dot(), Dot() sl, sr = Square(), Square() VGroup(dl, sl).arrange(DOWN).shift(2*LEFT) VGroup(dr, sr).arrange(DOWN).shift(2*RIGHT) self.add(dl, dr) self.wait() self.play( ClockwiseTransform(dl, sl), Transform(dr, sr) ) self.wait() r3r*r5r8r9r@rAc Btj||fd|i|dSNr8rrFr3r5r8rGrHs rIrEzClockwiseTransform.__init__Z1 .NN8NvNNNNNrJr3r*r5r*r8r9r@rArrrrnppirErrs@rIrr<saB5& OOOOOOOOOOOrJrc2eZdZdZejfd fd ZxZS) raTransforms the points of a mobject along a counterclockwise oriented arc. See also -------- :class:`.Transform`, :class:`.ClockwiseTransform` Examples -------- .. manim:: CounterclockwiseTransform_vs_Transform class CounterclockwiseTransform_vs_Transform(Scene): def construct(self): # set up the numbers c_transform = VGroup(DecimalNumber(number=3.141, num_decimal_places=3), DecimalNumber(number=1.618, num_decimal_places=3)) text_1 = Text("CounterclockwiseTransform", color=RED) c_transform.add(text_1) transform = VGroup(DecimalNumber(number=1.618, num_decimal_places=3), DecimalNumber(number=3.141, num_decimal_places=3)) text_2 = Text("Transform", color=BLUE) transform.add(text_2) ints = VGroup(c_transform, transform) texts = VGroup(text_1, text_2).scale(0.75) c_transform.arrange(direction=UP, buff=1) transform.arrange(direction=UP, buff=1) ints.arrange(buff=2) self.add(ints, texts) # The mobs move in clockwise direction for ClockwiseTransform() self.play(CounterclockwiseTransform(c_transform[0], c_transform[1])) # The mobs move straight up for Transform() self.play(Transform(transform[0], transform[1])) r3r*r5r8r9r@rAc Btj||fd|i|dSrrrs rIrEz"CounterclockwiseTransform.__init__rrJrrrs@rIrrds_$$T% OOOOOOOOOOOrJrc,eZdZdZdfd ZddZxZS) r aTransforms a mobject to the mobject stored in its ``target`` attribute. After calling the :meth:`~.Mobject.generate_target` method, the :attr:`target` attribute of the mobject is populated with a copy of it. After modifying the attribute, playing the :class:`.MoveToTarget` animation transforms the original mobject into the modified one stored in the :attr:`target` attribute. Examples -------- .. manim:: MoveToTargetExample class MoveToTargetExample(Scene): def construct(self): c = Circle() c.generate_target() c.target.set_fill(color=GREEN, opacity=0.5) c.target.shift(2*RIGHT + UP).scale(0.5) self.add(c) self.play(MoveToTarget(c)) r3r*r@rAc r||tj||jfi|dSrC)check_validity_of_inputrDrEtargetrFr3rGrHs rIrEzMoveToTarget.__init__s@ $$W---'.;;F;;;;;rJcDt|dstddS)Nrz8MoveToTarget called on mobjectwithout attribute 'target')hasattr ValueError)rFr3s rIrz$MoveToTarget.check_validity_of_inputs3w)) J   rJr3r*r@rA)rrrrrErrrs@rIr r s[2<<<<<<rJr c,eZdZd fd Zd fd ZxZS) _MethodAnimationr3r*methodslist[MethodWithArgs]r@rAcX||_t|dSrC)rrDrE)rFr3rrHs rIrEz_MethodAnimation.__init__s(  !!!!!rJc|jD]'}|jj|jg|jRi|j(t dSrC)rmethod__func__r3argsrGrDfinish)rFitemrHs rIrz_MethodAnimation.finishs^L J JD DK  I I I IT[ I I I I rJ)r3r*rrr@rAr~)rrrrErrrs@rIrrs[""""""rJrc4eZdZdZd fd Zd dZd d ZxZS) r a Animates a mobject by applying a method. Note that only the method needs to be passed to this animation, it is not required to pass the corresponding mobject. Furthermore, this animation class only works if the method returns the modified mobject. Parameters ---------- method The method that will be applied in the animation. args Any positional arguments to be passed when applying the method. kwargs Any keyword arguments passed to :class:`~.Transform`. rrr@rAc||||_||_tj|jfi|dSrC)rr method_argsrDrE__self__)rFrrrGrHs rIrEzApplyMethod.__init__sO $$V,,, 33F33333rJctj|stdt|jt t fsJdS)NzJWhoops, looks like you accidentally invoked the method you want to animate)inspectismethodr isinstancerr*r )rFrs rIrz#ApplyMethod.check_validity_of_inputsQ'' 1 &/G]+CDDDDDDDrJr*c|j}t|j}t|dkr0t |dt r|}ni}|j}|j |g|Ri||S)Nr) rlistrlenrdictpoprrXr)rFrr method_kwargsrs rIrWzApplyMethod.create_targetsD$%% t99q==ZR$77= HHJJMMM%%''7777777 rJ)rrr@rA)r@r*)rrrrrErrWrrs@rIr r ss$444444EEEE        rJr c(eZdZdZefd fd ZxZS) r aAnimation that applies a pointwise function to a mobject. Examples -------- .. manim:: WarpSquare :quality: low class WarpSquare(Scene): def construct(self): square = Square() self.play( ApplyPointwiseFunction( lambda point: complex_to_R3(np.exp(R3_to_complex(point))), square ) ) self.wait() functiontypes.MethodTyper3r*run_timer9r@rAc Ltj|j|fd|i|dS)Nr)rDrEapply_function)rFrr3rrGrHs rIrEzApplyPointwiseFunction.__init__ s5 /WWHWPVWWWWWrJ)rrr3r*rr9r@rA)rrrrr$rErrs@rIr r s\0> XXXXXXXXXXXrJr c,eZdZd fd Zd fd ZxZS) r rrr3r*r@rAc T||_tj|jfi|dSrC)rrDrEmove_torFrr3rGrHs rIrEz'ApplyPointwiseFunctionToCenter.__init__s1  33F33333rJc||jg|_t dSrC)rr3 get_centerrrDr^r_s rIr^z$ApplyPointwiseFunctionToCenter.begins= MM$,*A*A*C*CDDE  rJrrr3r*r@rAr~)rrrrEr^rrs@rIr r s[444444rJr c$eZdZdZd fd ZxZS) r aAnimation that changes color of a mobject. Examples -------- .. manim:: FadeToColorExample class FadeToColorExample(Scene): def construct(self): self.play(FadeToColor(Text("Hello World!"), color=RED)) r3r*colorstrr@rAc Htj|j|fi|dSrC)rDrE set_color)rFr3rrGrHs rIrEzFadeToColor.__init__-s-*E<tj|dfi|dSNrrrs rIrEzShrinkToCenter.__init__Qs*!..v.....rJrrrs@rIrrCsG  //////////rJrc$eZdZdZdfd ZxZS)raTTransforms a mobject to its last saved state. To save the state of a mobject, use the :meth:`~.Mobject.save_state` method. Examples -------- .. manim:: RestoreExample class RestoreExample(Scene): def construct(self): s = Square() s.save_state() self.play(FadeIn(s)) self.play(s.animate.set_color(PURPLE).set_opacity(0.5).shift(2*LEFT).scale(3)) self.play(s.animate.shift(5*DOWN).rotate(PI/4)) self.wait() self.play(Restore(s), run_time=2) r3r*r@rAc Ftj|jfi|dSrC)rDrErestorers rIrEzRestore.__init__ks*33F33333rJrrrs@rIrrUsG*4444444444rJrc(eZdZd fd Zd d ZxZS) rrrr3r*r@rAc J||_tj|fi|dSrC)rrDrErs rIrEzApplyFunction.__init__ps/  ++F+++++rJrc||j}t|tt fst d|S)NzDFunctions passed to ApplyFunction must return object of type Mobject)rr3rXrr*r TypeError)rFrs rIrWzApplyFunction.create_targettsTt|002233&7M":;; V  rJr)r@r)rrrrErWrrs@rIrrosQ,,,,,,rJrc0eZdZdZefd fd Zd d ZxZS) ra{Applies a matrix transform to an mobject. Parameters ---------- matrix The transformation matrix. mobject The :class:`~.Mobject`. about_point The origin point for the transform. Defaults to ``ORIGIN``. kwargs Further keyword arguments that are passed to :class:`ApplyPointwiseFunction`. Examples -------- .. manim:: ApplyMatrixExample class ApplyMatrixExample(Scene): def construct(self): matrix = [[1, 1], [0, 2/3]] self.play(ApplyMatrix(matrix, Text("Hello World!")), ApplyMatrix(matrix, NumberPlane())) matrixr;r3r* about_pointr@rAc x|fd}tj||fi|dS)NcDtj|z jzSrC)rdotT)prrs rIfuncz"ApplyMatrix.__init__..funcs 6!k/6844{B BrJ)initialize_matrixrDrE)rFrr3rrGrrHs ` ` rIrEzApplyMatrix.__init__sg''// C C C C C C w11&11111rJctj|}|jdkr"tjd}||ddddf<|}n|jdkrt d|S)N)r!r!r!)rrzMatrix has bad dimensions)rarrayshapeidentityr)rFr new_matrixs rIrzApplyMatrix.initialize_matrixsi&!! <6 ! !QJ!'Jrr2A2v FF \V # #899 9 rJ)rr;r3r*rr;r@rA)rr;r@r;)rrrrr&rErrrs@rIrr}se:#) 2 2 2 2 2 2 2rJrc,eZdZd fd Zd fd ZxZS) rrrr3r*r@rAc Z||_|j}tj||fi|dSrC)rapply_complex_functionrDrE)rFrr3rGrrHs rIrEzApplyComplexFunction.__init__s9  /44V44444rJc|td}tj|j|_t dSr)rcomplexrlogimagr8rD_init_path_func)rFfunc1rHs rIrz$ApplyComplexFunction._init_path_funcsH gajj))u *  !!!!!rJrr~)rrrrErrrs@rIrrs[555555 """"""""""rJrc8eZdZdZdezdd fd Zdd ZxZS)raAn animation moving mobjects cyclically. In particular, this means: the first mobject takes the place of the second mobject, the second one takes the place of the third mobject, and so on. The last mobject takes the place of the first one. Parameters ---------- mobjects List of mobjects to be transformed. path_arc The angle of the arc (in radians) that the mobjects will follow to reach their target. kwargs Further keyword arguments that are passed to :class:`.Transform`. Examples -------- .. manim :: CyclicReplaceExample class CyclicReplaceExample(Scene): def construct(self): group = VGroup(Square(), Circle(), Triangle(), Star()) group.arrange(RIGHT) self.add(group) for _ in range(4): self.play(CyclicReplace(*group)) Z)r8mobjectsr*r8r9r@rAcft||_tj|jfd|i|dSr)r)grouprDrE)rFr8rrGrHs rIrEzCyclicReplace.__init__s=H% AAhA&AAAAArJr)c|j}|dg|dd}t||jdD]\}}|||S)NrTrr)rrXrur)rFrcycled_targetsm1m2s rIrWzCyclicReplace.create_targetsf"" *3vcrc{3.$*TBBB  FB JJrNNNN rJ)rr*r8r9r@rA)r@r))rrrrr%rErWrrs@rIrrsw@57LBBBBBBBB rJrceZdZdS)rN)rrrrJrIrrsDrJrcBeZdZeefd fd Zd fd ZxZS)r start_animr#end_anim rate_funcrr@rAc ||_||_d|vr|d|_nt |j|j|_||fD]}||j|j|j|j|jkra|j|j||fD]=}t|tr&|j|j|j >tj |j|jfd|i||j|_|j |_dS)Nrr )r r rrmax set_run_timerhget_num_pointsr]rrr5rDrEr3)rFr r r rGanimrHs rIrEzTransformAnimations.__init__sx%    "JJz22DMM 3X5FGGDM( - -D   dm , , , ,  ' 3)5+::<<(7799::  ' 2 283L M M M"H, J JdI..J43H3T)44T5HIII   0  )>??EE'>#6#6#8#899E))&)))))rJc|j|_tj||j|j}}|d|df|d|dffD]\}}|||dS)aFInitial setup for the animation. The mobject to which this animation is bound is a group consisting of both the starting and the ending mobject. At the start, the ending mobject replaces the starting mobject (and is completely faded). In the end, it is set to be the other way around. rrN)r3rXending_mobjectr#r^rhghost_to)rFstartendm0rs rIr^zFadeTransform.beginQs#l//11*D,?sa%(+c!fc!f-=> " "FB MM"b ! ! ! ! " "rJc|s|jr"|||j|j|ddS)zReplaces the source by the target and sets the opacity to 0. If the provided target has no points, and thus a location of [0, 0, 0] the source will simply fade out where it currently is. )rrrN)r submobjectsrerr set_opacity)rFsourcers rIrzFadeTransform.ghost_tobs[  " " Yf&8 Y NN64r<sv77""""""   0 8888888888%%%%%%%%000000JJJJJJJJ++++++-,,,,,,,<<<<<<<<;;;;;;;;8######77777777@@@@@ @@@F3 3 3 3 3 93 3 3 l''''' '''%O%O%O%O%O%O%O%OP.O.O.O.O.O .O.O.Ob"""""9"""J|.....)...bXXXXX[XXX>%;=====+===$@@@@@;@@@$/////\///$44444k4444     I   00000(000f " " " " "; " " "+++++I+++\     =   %#%#%#%#%#)%#%#%#Pa-a-a-a-a-Ia-a-a-H''''''''''-''''''''''rJ