l j(dZddlmZddgZddlmZddlZddlm Z m Z ddl m Z m Z d d lmZd d lmZd d lmZmZd d lmZmZddlmZddlmZmZddlmZmZerd dl m!Z!GddeZ"Gdde"Z#Gdde"Z$dS)zQAnimations that try to transform Mobjects while keeping track of identical parts.) annotationsTransformMatchingShapesTransformMatchingTex) TYPE_CHECKINGN) OpenGLGroup OpenGLMobject) OpenGLVGroupOpenGLVMobject)config) RendererType)GroupMobject)VGroupVMobject)AnimationGroup)FadeInFadeOut)FadeTransformPieces Transform)ScenecleZdZdZ ddfd ZddZddZeddZeddZ xZ S)TransformMatchingAbstractBasea(Abstract base class for transformations that keep track of matching parts. Subclasses have to implement the two static methods :meth:`~.TransformMatchingAbstractBase.get_mobject_parts` and :meth:`~.TransformMatchingAbstractBase.get_mobject_key`. Basically, this transformation first maps all submobjects returned by the ``get_mobject_parts`` method to certain keys by applying the ``get_mobject_key`` method. Then, submobjects with matching keys are transformed into each other. Parameters ---------- mobject The starting :class:`~.Mobject`. target_mobject The target :class:`~.Mobject`. transform_mismatches Controls whether submobjects without a matching key are transformed into each other by using :class:`~.Transform`. Default: ``False``. fade_transform_mismatches Controls whether submobjects without a matching key are transformed into each other by using :class:`~.FadeTransform`. Default: ``False``. key_map Optional. A dictionary mapping keys belonging to some of the starting mobject's submobjects (i.e., the return values of the ``get_mobject_key`` method) to some keys belonging to the target mobject's submobjects that should be transformed although the keys don't match. kwargs All further keyword arguments are passed to the submobject transformations. Note ---- If neither ``transform_mismatches`` nor ``fade_transform_mismatches`` are set to ``True``, submobjects without matching keys in the starting mobject are faded out in the direction of the unmatched submobjects in the target mobject, and unmatched submobjects in the target mobject are faded in from the direction of the unmatched submobjects in the start mobject. FNmobjectrtarget_mobjecttransform_mismatchesboolfade_transform_mismatcheskey_map dict | Nonec zt|trt}nAt|trt}n$t|t rt }nt}||}||} |i}|} |} t| | D]8} | || | | | 9t| | fi|g} |}|}| D]o\}}||vrf|| vrb| ||| | |||d| |dpt|dkr!| t#||fi||}|}t|| D]} | || t| |D]} | | | |}|r+d|vrd|d<| t||fi|nh|r"| t#||fi|nD| t)|fd|i|| t+|fd|i|t-j| ||g|_||_dS)Nr$replace_mobject_with_target_in_sceneTtarget_position) isinstancer r rrrrr get_shape_mapset intersectionaddritemspoplenappendr differencecopyrrsuper__init__ to_removeto_add)selfrrrrr kwargs group_type source_map target_maptransform_sourcetransform_targetkeyanimskey_mapped_sourcekey_mapped_targetkey1key2 fade_source fade_targetfade_target_copy __class__s c/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/animation/transform_matching_parts.pyr1z&TransformMatchingAbstractBase.__init__Fs g~ . . %JJ  / / $JJ  * * JJJ''00 ''77 ?G &:<<%:<<z??// ;; 2 2C  C 1 1 1  C 1 1 1 1+-=HHHHI'JLL&JLL!--// + +JD$z!!dj&8&8!%%j&6777!%%j&6777tT***tT***  ! !A % % LL#$57HSSFSS   !jll  jll z??--j99 - -C OOJsO , , , ,z??--j99 - -C OOJsO , , , ,&++--  5VCCAE=> LL; FFvFF G G G G &  LL,[+PPPP Q Q Q Q LLTTkTVTT U U U LL'OOOOO    %  !#34$ returndictci}||D]t}||}||vr>tdtjkrt ||<nt ||<|||u|S)Nrenderer)get_mobject_partsget_mobject_keyr r OPENGLr rr))r4r shape_mapsmr;s rEr&z+TransformMatchingAbstractBase.get_shape_maps ((11 # #B&&r**C)##*%)<<<%1^^IcNN%+XXIcN cN  r " " " "rFscenerNonec|jD]}|d||j|j|j||jdS)Nr) animations interpolateremoverr2r)r3)r4rPanims rEclean_up_from_scenez1TransformMatchingAbstractBase.clean_up_from_scenesiO  D   Q     T\""" dn%% $+rFc tdNzTo be implemented in subclass.NotImplementedErrorrs rErKz/TransformMatchingAbstractBase.get_mobject_parts!"BCCCrFc tdrYrZr\s rErLz-TransformMatchingAbstractBase.get_mobject_keyr]rFFFN rrrrrrrrr r!)rrrGrH)rPrrGrQ)rr) __name__ __module__ __qualname____doc__r1r&rW staticmethodrKrL __classcell__rDs@rErrs))^&+*/# G%G%G%G%G%G%G%R    DDD\DDDD\DDDDDrFrc\eZdZdZ ddfd ZeddZeddZxZS)raAn animation trying to transform groups by matching the shape of their submobjects. Two submobjects match if the hash of their point coordinates after normalization (i.e., after translation to the origin, fixing the submobject height at 1 unit, and rounding the coordinates to three decimal places) matches. See also -------- :class:`~.TransformMatchingAbstractBase` Examples -------- .. manim:: Anagram class Anagram(Scene): def construct(self): src = Text("the morse code") tar = Text("here come dots") self.play(Write(src)) self.wait(0.5) self.play(TransformMatchingShapes(src, tar, path_arc=PI/2)) self.wait(0.5) FNrrrrrrr r!c Ftj||f|||d|dSN)rrr r0r1r4rrrrr r5rDs rEr1z TransformMatchingShapes.__init__O    "6&?         rFrG list[Mobject]c*|SN)family_members_with_pointsr\s rErKz)TransformMatchingShapes.get_mobject_partss11333rFintc&|||dtj|jddz}t |}||S)Nr)heightg) save_statecenterr'nproundpointshashtobytesrestore)rrounded_pointsresults rErLz'TransformMatchingShapes.get_mobject_keys~ 1 '.!44s:n,,..// rFr_r`rrrGrn)rrrGrr rarbrcrdr1rerKrLrfrgs@rErrs@&+*/#        $444\4\rFc\eZdZdZ ddfd ZeddZeddZxZS)raA transformation trying to transform rendered LaTeX strings. Two submobjects match if their ``tex_string`` matches. See also -------- :class:`~.TransformMatchingAbstractBase` Examples -------- .. manim:: MatchingEquationParts class MatchingEquationParts(Scene): def construct(self): variables = VGroup(MathTex("a"), MathTex("b"), MathTex("c")).arrange_submobjects().shift(UP) eq1 = MathTex("{{x}}^2", "+", "{{y}}^2", "=", "{{z}}^2") eq2 = MathTex("{{a}}^2", "+", "{{b}}^2", "=", "{{c}}^2") eq3 = MathTex("{{a}}^2", "=", "{{c}}^2", "-", "{{b}}^2") self.add(eq1) self.wait(0.5) self.play(TransformMatchingTex(Group(eq1, variables), eq2)) self.wait(0.5) self.play(TransformMatchingTex(eq2, eq3)) self.wait(0.5) FNrrrrrrr r!c Ftj||f|||d|dSrjrkrls rEr1zTransformMatchingTex.__init__ rmrFrGrnct|ttttfrd|jDSt |dsJ|jS)NcLg|]!}t|D]}|"S)rrK).0sps rE z:TransformMatchingTex.get_mobject_parts..sM-??BBrF tex_string)r%rrrr submobjectshasattrr\s rErKz&TransformMatchingTex.get_mobject_partssb gv{LI J J ' ,  7L11 1 11& &rFstrc|jSrp)rr\s rErLz$TransformMatchingTex.get_mobject_key's !!rFr_r`r)rrrGrrrgs@rErrsD&+*/#        $ ' ' '\ '"""\"""""rF)%rd __future__r__all__typingrnumpyrx#manim.mobject.opengl.opengl_mobjectrr.manim.mobject.opengl.opengl_vectorized_mobjectr r _configr constantsr mobject.mobjectrr mobject.types.vectorized_mobjectrr compositionrfadingrr transformrr scene.scenerrrrrrFrErsWW"""""" $&< = JJJJJJJJWWWWWWWW$$$$$$,,,,,,,,????????''''''########55555555$######ODODODODODNODODODd;;;;;;;;;|?"?"?"?"?"8?"?"?"?"?"rF