l j-hdZddlmZddlmZmZgdZddlZddl Z ddl m Z ddl m Z ddlmZdd lmZmZdd lmZmZdd lTdd lmZdd lmZddlmZmZddlmZddl m!Z!ddl"m#Z#dZ$GddeZ%Gdde%Z&Gddee#Z'Gdde&Z(Gdde(Z)Gdd e(Z*dS)!a;Mobjects representing text rendered using LaTeX. .. important:: See the corresponding tutorial :ref:`rendering-with-latex` .. note:: Just as you can use :class:`~.Text` (from the module :mod:`~.text_mobject`) to add text to your videos, you can use :class:`~.Tex` and :class:`~.MathTex` to insert LaTeX. ) annotations)BLACKParsableManimColor)SingleStringMathTexMathTexTex BulletedListTitleN)Iterable)reduce)dedent)AnySelf)configlogger)*)Line) SVGMobject)VGroupVMobject) TexTemplate)tex_to_svg_file)ConvertToOpenGL substringceZdZdZddddddedfd)fd Zd*dZed+dZej d,dZd-d Z d.d"Z d.d#Z d/d%Z d*d&Zd0d1d(ZxZS)2rzElementary building block for rendering text with LaTeX. Tests ----- Check that creating a :class:`~.SingleStringMathTex` object works:: >>> SingleStringMathTex('Test') # doctest: +SKIP SingleStringMathTex('Test') rTNFalign* tex_stringstr stroke_widthfloat should_centerboolheight float | Noneorganize_left_to_righttex_environment str | None tex_templateTexTemplate | None font_sizecolorParsableManimColor | Nonekwargsrc | tj} ||_||_||_| t d}||_||_t| ||j|j} tj d| |||| dddd| | |j |_| |j|_|jr|dSdS)Nr)) environmentr)T)should_subdivide_sharp_curvesshould_remove_null_curves) file_namer"r r$r,path_string_config)rr, _font_sizer&r'rr)rr_get_modified_expressionsuper__init__ init_colorsr$initial_heightr+#_organize_submobjects_left_to_right) selfrr r"r$r&r'r)r+r,r.r3 __class__s Y/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/mobject/text/tex_mobject.pyr9zSingleStringMathTex.__init__9s( =JJ$E#&<#.  !.1L)5$#  ) )* 5 5,*    '%15-1     #k >!_DN  & 7  4 4 6 6 6 6 6 7 7returncZt|jdt|jdSN()type__name__reprrr=s r?__repr__zSingleStringMathTex.__repr__m,t**%@@T_(=(=@@@@r@c0|j|jz tz S)z!The font size of the tex mobject.)r$r;SCALE_FACTOR_PER_FONT_POINTrJs r?r+zSingleStringMathTex.font_sizeps{T003NNNr@font_valNonec|dkrtd|jdkr|||jz dSdS)Nrz!font_size must be greater than 0.) ValueErrorr$scaler+)r=rOs r?r+zSingleStringMathTex.font_sizeusM q==@AA A [1__ JJx$.0 1 1 1 1 1 _r@c\|}|}||}|SN)strip_modify_special_strings)r=rresults r?r7z,SingleStringMathTex._get_modified_expressions,--f55 r@texc pttjdkdkdkdkdddg}|rd}|z d krd d krd d rd d fddD\}}||kr,dddd|dD]}d|zv}d|zv}||z rd S)Nz\overz \overlinez\sqrtz\sqrt{_^dotz{\quad}z \substackz\quad\\z\quad\\c3K|]8}td|ddDV9dS)c*g|]}||ddv|S)rz (){}[]|.\r5.0ss r? zISingleStringMathTex._modify_special_strings...s+PPPqQP1Q4<;O;O;O;O;Or@N)lensplit)rcsubstrrYs r? z>SingleStringMathTex._modify_special_strings..sc! !  PPCIIf--abb1PPP Q Q! ! ! ! ! ! r@)\left\rightrkz\bigrl)arrayz \begin{%s}z\end{%s})rVr opor_endswith startswithreplace_remove_stray_braces) r=rYshould_add_fillerfiller num_lefts num_rightscontextbegin_inend_ins ` r?rWz+SingleStringMathTex._modify_special_stringssiikk" Fx|#xy  S!! S!! U##     F 6MC ,  C "99C >>& ! ! 4++fl33C! ! ! ! /! ! !  :  " "++h00C++i11C'',,   G%/C7H!G+3F&   r@cb|d|dz |dz}|d|dz |dz}||krd|z}|dz }||k||kr|dz}|dz }||k|S)z Makes :class:`~.MathTex` resilient to unmatched braces. This is important when the braces in the TeX code are spread over multiple arguments as in, e.g., ``MathTex(r"e^{i", r"\tau} = 1")``. {z\{z\\{}z\}z\\}rf)count)r=rYrvrws r?rsz(SingleStringMathTex._remove_stray_bracessIIcNNSYYu%5%55 '8J8JJ YYs^^cii&6&6679K9KK 9$$)C NI9$$*$$)C !OJ*$$ r@rc2|d|S)Nc|dS)Nrr5)ps r?zISingleStringMathTex._organize_submobjects_left_to_right..s AaDr@)sortrJs r?r<z7SingleStringMathTex._organize_submobjects_left_to_rights ..!!! r@c|jSrU)rrJs r?get_tex_stringz"SingleStringMathTex.get_tex_strings r@propagate_colorsc|jD]~}|jtkr|j|_tjt jkr|Ntjt jkr|||S)N)r) submobjectsr,rrrenderer RendererTypeOPENGLr:CAIRO)r=r submobjects r?r:zSingleStringMathTex.init_colorss* J JJ5((#zJ ,"555&&((((L$666&&8H&III r@)rrr r!r"r#r$r%r&r#r'r(r)r*r+r!r,r-r.rrAr)rAr!)rOr!rArP)rrrAr)rYrrArrAr)T)rr#rAr)rH __module__ __qualname____doc__DEFAULT_FONT_SIZEr9rKpropertyr+setterr7rWrsr<rr: __classcell__r>s@r?rr.sE "#',&.+/,+/27272727272727hAAAAOOOXO 2 2 2 2 3333j$         r@rceZdZdZdddddd:fdZd;dZeddZ d?d!Z e d@d#Z e d@d$Z dAd&ZdBd)ZdCd,Z dDdEd2ZdFd5ZdGd7ZdHd9ZxZS)IrucA string compiled with LaTeX in math mode. Examples -------- .. manim:: Formula :save_last_frame: class Formula(Scene): def construct(self): t = MathTex(r"\int_a^b f'(x) dx = f(b)- f(a)") self.add(t) Notes ----- Double-brace notation ``{{ ... }}`` can be used to split a single string argument into multiple submobjects without having to pass separate strings:: MathTex(r"{{ a^2 }} + {{ b^2 }} = {{ c^2 }}") Each ``{{ ... }}`` group and every piece of text between groups becomes its own submobject, which is useful for :class:`~.TransformMatchingTex` animations. For ``{{`` to be recognised as a group opener it must appear either at the very start of the string or be immediately preceded by a whitespace character. ``{{`` that follows non-whitespace — such as in ``\frac{{{n}}}{k}`` or ``a^{{2}}`` — is left untouched, so ordinary nested-brace LaTeX is not accidentally split. To prevent an unintentional split, insert a space between the two braces: ``{{ ... }}`` → ``{ { ... } }``. Tests ----- Check that creating a :class:`~.MathTex` works:: >>> MathTex('a^2 + b^2 = c^2') # doctest: +SKIP MathTex('a^2 + b^2 = c^2') Check that double brace group splitting works correctly:: >>> t1 = MathTex('{{ a }} + {{ b }} = {{ c }}') # doctest: +SKIP >>> len(t1.submobjects) # doctest: +SKIP 5 >>> t2 = MathTex(r"\frac{1}{a+b\sqrt{2}}") # doctest: +SKIP >>> len(t2.submobjects) # doctest: +SKIP 1  Nr) arg_separatorsubstrings_to_isolatetex_to_color_mapr' tex_stringsrrrIterable[str] | Noner$dict[str, ParsableManimColor] | Noner'r(r.rc\|dtd|_||_|gnt ||_|i|_n||_|j|j||_ d|_ | ||_ g|_ ||j |j}tj|f|j |jd||j|j |_|n<#t($r/}|j r!t+jt/d|d}~wwxYw||j|jr|dSdS)Nr)F)r'r)a A group of double braces, {{ ... }}, was detected in your string. Manim splits TeX strings at the double braces, which might have caused the current compilation error. If you didn't use the double brace split intentionally, add spaces between the braces to avoid the automatic splitting: {{ ... }} --> { { ... } }. )poprr)rlistrrextendkeysr'brace_notation_split_occurred_prepare_tex_stringsrmatched_strings_and_ids)_join_tex_strings_with_unique_delimintersr8r9joinr_break_up_by_substringsrRrerrorr set_color_by_tex_to_color_mapr&r<) r=rrrr'rr. joined_stringcompilation_errorr>s r?r9zMathTex.__init__s#JJ~vn7MNN*'/BBT:O5P5P "  #CED ! !$4D ! "))$*?*D*D*F*FGGG.-2*44[AA>@$ $ JJ $"<M EGG   $ 4!.      #055d6FGGDO  ( ( * * * * $ $ $1       $ # $ **4+@AAA  & 7  4 4 6 6 6 6 6 7 7s9A?D99 E2*E--E2 Iterable[str]rA list[str]cd|D}g}|D],}||}||-t|t|krd|_d|DS)NcZg|](}t|tr|nt|)Sr5) isinstancerrcstrings r?rez0MathTex._prepare_tex_strings..Rs>! ! ! CIj-- >FF3v;;! ! ! r@Tc8g|]}t|dk|S)r)rgrs r?rez0MathTex._prepare_tex_strings..\s$RRR6#f++PQ/////r@)_split_double_bracesrrgr)r=rtex_strings_validatedtex_strings_validated_tworrhs r?rzMathTex._prepare_tex_stringsOs! ! MX! ! ! %'!/ 4 4J--j99E % , ,U 3 3 3 3 ( ) )C0E,F,F F F15D .RR%>RRRRr@rc*g}d}d}d}d}|t|kr_||dkrA|dzt|kr+||dz}|dks|dvr||||dzz }|dz }a|sf|||dzdkrD|dks||dz r!||d}d }d}|dz }n|||z }|dz }n||d kr|dz }|||z }|dz }nt||d kr6|dkr0|||dzd kr||d}d}|dz }n2||d kr|dz}|||z }|dz }n|||z }|dz }|t|k_|||S) abSplit *tex_string* on Manim's ``{{ ... }}`` double-brace notation. Rules that avoid false positives on ordinary LaTeX source: * ``{{`` is only treated as a group opener when it appears at the very start of the string or is immediately preceded by a whitespace character. Naturally-occurring ``{{`` in LaTeX is usually preceded by non-whitespace (e.g. ``\frac{{{n}}}{k}`` or ``a^{{2}}``), so the whitespace guard eliminates the most common false positives without any brace-depth bookkeeping on the outer string. * Inside an open group the depth of *real* LaTeX braces is tracked. ``}}`` only closes the Manim group when the inner depth is zero, so ``{{ a^{b^{c}} }}`` is handled correctly. * Escape sequences are consumed as two-character units in priority order: ``\\`` first (escaped backslash), then ``\{`` / ``\}`` (escaped braces). This ensures e.g. ``\\}}`` is read as an escaped backslash followed by a real ``}}`` rather than as ``\`` + ``\}`` + lone ``}``. r^rF\rfz{}rz{{Tr|r}z}})rgisspaceappend)rsegmentscurrenti inside_manim inner_depthnext_chs r?rzMathTex._split_double_braces^s .!   #j//!!!}$$QZ)@)@$QU+d??gooz!a!e)44GFA" a!a%i(D00FFjQ/7799FOOG,,, G#'L"#KFAAz!},GFAAa=C''1$Kz!},GFAAqMS((#q(("1q1u9-55OOG,,, G#(LFAA]c))1$Kz!},GFAAz!},GFA[#j//!!^    r@cd}d}t|D]\}}d|dd}|j|d|dft|}d} t |dkrV|||} | r$||| \} }| | z} |dz }n| |z} d}t |dkV|| z }|t |dz kr ||jz }|dz }||z}|S) Nr^r"\special{dvisvgm:raw }uniquerf\special{dvisvgm:raw }) enumeraterrrrg_locate_first_match _handle_matchr) r=rrrssIdxidxr string_partunprocessed_stringprocessed_string first_match processeds r?rz1MathTex._join_tex_strings_with_unique_deliminterss` (55 8 8OCNNNNNK  ( / /=Oc=O=O=O0P Q Q Q "%Z ! ())A--"66)+= ,484F4F{551I1(8)'C$QJEE'7:L'L$)+&())A-- + +KS%%)))t11 8 8K)K7MMr@rre.Match | Nonect|}d}d}|D]1}tjdtj|d|}|rmt|d|krG|}t|d}t|d}|rt|d|krl|t|dkrF|}t|d}t|d}3|S)Nrz(.*?)(z)(.*)rfr)rgrematchescapegroup)r=rrfirst_match_startfirst_match_lengthrrrs r?rzMathTex._locate_first_matchs0 233 . = =IHAbi &:&:AAACUVVE =U[[^^,,/@@@# $' A$7$7!%(Q%8%8"" =3u{{1~~..2CCC%EKKNN(;(;;;"'K(+EKKNN(;(;%),U[[^^)<)<&r@rintrre.Matchtuple[str, str]c|d}|d}|d}d|dtd}d}|j|d|dtf||z|z|z}|} || fS) Nrfrrrrrr)rMATHTEX_SUBSTRINGrr) r=rr pre_matchmatched_string post_match pre_string post_stringrrs r?rzMathTex._handle_matchs%%a(( $**1-- &&q)) U5 U U>O U U U 4  $++ DeDD1BDD E   %z1NB[P'!333r@list[tuple[str, str]]c$d|jDS)z4Return only the 'ss' (substring_to_isolate) matches.cPg|]#\}}|t||f$Sr5rprrcrYid_s r?rez.MathTex._substring_matches..sC   S||-.. #J   r@rrJs r?_substring_matcheszMathTex._substring_matches&   8    r@c$d|jDS)z(Return only the main tex_string matches.cPg|]#\}}|t||f$Sr5rrs r?rez)MathTex._main_matches..sC   S<< 122 #J   r@rrJs r? _main_matcheszMathTex._main_matchesrr@rcTg} |jD]I\}}t}||_|j|j|j||JnK#t$r>tj d|d|d||jdYnwxYw||_|S)z Reorganize existing submobjects one layer deeper based on the structure of tex_strings (as a list of tex_strings) z0MathTex: Could not find SVG group for tex part 'z' (id: z ). Using fallback to root group.root) r MathTexPartraddid_to_vgroup_dictrrKeyErrorrr)r=new_submobjectsr tex_string_idmtps r?rzMathTex._break_up_by_substringss +- C-1-? , ,) M!mm!+/ >JKK&&s++++  ,  C C C LF:FFVcFFF     " "4#9&#A B B B B B  C + sAAABBrY VGroup | Nonec J|jD]\}}||kr|j|cSdSrU)rr)r=rYr.tex_strmatch_ids r?get_part_by_texzMathTex.get_part_by_texs@!%!= 8 8 GX#~~-h7777tr@r,rc l|jD]+\}}||kr |j||,|SrU)rr set_color)r=rYr,r.rrs r?set_color_by_texzMathTex.set_color_by_tex!sH"&!= B B GX#~~&x0::5AAA r@?opacityr!remaining_opacityr%c ||||jD]+\}}||kr |j||,|S)a Sets the opacity of the tex specified. If 'remaining_opacity' is specified, then the remaining tex will be set to that opacity. Parameters ---------- tex The tex to set the opacity of. opacity Default 0.5. The opacity to set the tex to remaining_opacity Default None. The opacity to set the remaining tex to. If None, then the remaining tex will not be changed Nr) set_opacityrr)r=rYrrr.rrs r?set_opacity_by_texzMathTex.set_opacity_by_tex)sf*  (   %6  7 7 7!%!= F F GX#~~&x0<>uEEE F r@partcz|}||vrtd||S)Nz*Trying to get index of part not in MathTex)rhrRindex)r=r split_selfs r? index_of_partzMathTex.index_of_partNs=ZZ\\ z ! !IJJ J%%%r@rPc>|jddS)Nc*|SrU)r)ms r?rz-MathTex.sort_alphabetically..UsA,<,<,>,>r@)key)rrrJs r?sort_alphabeticallyzMathTex.sort_alphabeticallyTs& ">">?????r@) rrrrrrrrr'r(r.r)rrrAr)rrrAr)rrrrrAr)rrrrrAr)rrrrrAr)rArr)rYrr.rrAr)rYrr,rr.rrAr)rN) rYrrr!rr%r.rrAr)r r r.rrAr)rrrAr)rArP)rHrrrr9r staticmethodrrrrrrrrrrr rrrrrs@r?rrs00j!6:AE&. 7777777777777777r S S S SLLL\L\""""H( 4 4 4 4   X    X * *. 8&&&& @@@@@@@@r@rc"eZdZUded<ddZdS)rrrrAcZt|jdt|jdSrCrFrJs r?rKzMathTexPart.__repr__[rLr@Nr)rHrr__annotations__rKr5r@r?rrXs9OOOAAAAAAr@r) metaclassc,eZdZdZdddd fd ZxZS)raA string compiled with LaTeX in normal mode. The color can be set using the ``color`` argument. Any parts of the ``tex_string`` that are colored by the TeX commands ``\color`` or ``\textcolor`` will retain their original color. Tests ----- Check whether writing a LaTeX string works:: >>> Tex('The horse does not eat cucumber salad.') # doctest: +SKIP Tex('The horse does not eat cucumber salad.') r^centerrr'rrrr'r(r.rc@tj|||d|dS)Nr!)r8r9)r=rr'rr.r>s r?r9z Tex.__init__psB  '+        r@)rrrrr'r(r.r)rHrrrr9rrs@r?rr_sY& &.             r@rc8eZdZdZeddddfdZdddZxZS)r aA bulleted list. Examples -------- .. manim:: BulletedListExample :save_last_frame: class BulletedListExample(Scene): def construct(self): blist = BulletedList("Item 1", "Item 2", "Item 3", height=2, width=2) blist.set_color_by_tex("Item 1", RED) blist.set_color_by_tex("Item 2", GREEN) blist.set_color_by_tex("Item 3", BLUE) self.add(blist) rN)buffdot_scale_factorr'rrr$r!r%r'r(r.rc||_||_||_d|D}tj|d|i||D]e}t d|j}||dtt| |f| tt|jdS)Ncg|]}|dzS)r_r5rbs r?rez)BulletedList.__init__..s:::qF :::r@r'z\cdotr) aligned_edger$) r$r%r'r8r9rrSnext_toLEFT SMALL_BUFF add_to_backarrangeDOWN) r=r$r%r'rr.line_separated_itemsrr]r>s r?r9zBulletedList.__init__s 0.::E::: ! +     " "D(##))$*?@@C KKQz 2 2 2   S ! ! ! ! T49 =====r@rindex_or_string int | strrrArPcj|}t|tr+||}|td|dn5t|tr|j|}nt d||jD]3}||ur|d||4dS)Nz.Could not locate part by provided tex string 'z'.zExpected int or string, got rfr)rrr Exceptionrr TypeErrorset_fill)r=r0rargr other_parts r? fade_all_butzBulletedList.fade_all_buts c3   B-1-A-A#-F-FD|LSLLLS ! ! B#C(DD@3@@AA A* 5 5JT!!##A#....##G#4444  5 5r@) rrr$r!r%r!r'r(r.r)r)r0r1rr!rArP)rHrrrMED_LARGE_BUFFr9r8rrs@r?r r su(%"#&* >>>>>>>>.555555555r@r c.eZdZdZddeddfdZxZS)r apA mobject representing an underlined title. Examples -------- .. manim:: TitleExample :save_last_frame: import manim class TitleExample(Scene): def construct(self): banner = ManimBanner() title = Title(f"Manim version {manim.__version__}") self.add(banner, title) TF)include_underlinematch_underline_width_to_textunderline_buff text_partsrr;r#r<r=r!r.rc||_||_||_tj|i||t |jrtddz }ttt}| |t|j|jr| |n||_||||_dSdS)N frame_widthr)r$)r;r<r=r8r9to_edgeUPrrr*RIGHTr)r. match_widthwidthr underline) r=r;r<r=r>r.underline_widthrFr>s r?r9zTitle.__init__s"3-J*,*//// R  ! '$]3a7OT5))I   dDt/B  C C C1 2%%d++++"1  HHY   &DNNN ' 'r@) r>rr;r#r<r#r=r!r.r)rHrrrMED_SMALL_BUFFr9rrs@r?r r s\(#'.3 . ''''''''''''r@r )+r __future__rmanim.utils.colorrr__all__operatorrnrcollections.abcr functoolsr textwrapr typingrrmanimrrmanim.constantsmanim.mobject.geometry.linermanim.mobject.svg.svg_mobjectr&manim.mobject.types.vectorized_mobjectrrmanim.utils.texrmanim.utils.tex_file_writingropengl.opengl_compatibilityrrrrrrr r r5r@r?rYs  #"""""77777777    $$$$$$ ,,,,,,444444CCCCCCCC''''''888888999999rrrrr*rrrjr@r@r@r@r@!r@r@r@j AAAAA(oAAAA     '   @95959595953959595x('('('('('C('('('('('r@