l jdZddlmZdgZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZe r ddl mZdd lmZd Zd Zd Zed GddZddZdS)z)Utilities for processing LaTeX templates.) annotations TexTemplateN) dataclassfield)Path) TYPE_CHECKINGAny)Self)StrPathzE\usepackage[english]{babel} \usepackage{amsmath} \usepackage{amssymb}z\begin{document}z\end{document}T)eqc(eZdZUdZeddZded< dZded< dZded < d Z ded < d Z ded < e Z ded< dZ ded< dZded< ed)dZejd*dZed+d,dZd-d.d!Zd/d"Zd0d$Zd1d&Zd2d'Zd(S)3rzATeX templates are used to create ``Tex`` and ``MathTex`` objects.F)defaultinitstr_bodylatex tex_compiler descriptionz.dvi output_formatz#\documentclass[preview]{standalone} documentclasspreamble YourTextHereplaceholder_textpost_doc_commandsreturnc |jpFdtd|j|jt |j|jtgS)zThe entire TeX template. N) rjoinfilterrr_BEGIN_DOCUMENTrr _END_DOCUMENTselfs J/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/utils/tex.pybodyzTexTemplate.body9sUz  TYY &M#*)!      valueNonec||_dS)N)r)r$r(s r%r&zTexTemplate.bodyJs  r'tex_template.texfiler kwargsr r c f|di|}t|d|_|S)zCreate an instance by reading the content of a file. Using the ``add_to_preamble`` and ``add_to_document`` methods on this instance will have no effect, as the body is read from the file. zutf-8)encoding)r read_textr&)clsr,r-instances r% from_filezTexTemplate.from_fileNs83====T ,,g,>> r'txtprependboolc|jrtjdtd|r|dz|jz|_n|xjd|zz c_|S)aAdds text to the TeX template's preamble (e.g. definitions, packages). Text can be inserted at the beginning or at the end of the preamble. Parameters ---------- txt String containing the text to be added, e.g. ``\usepackage{hyperref}``. prepend Whether the text should be added at the beginning of the preamble, i.e. right after ``\documentclass``. Default is to add it at the end of the preamble, i.e. right before ``\begin{document}``. zeThis TeX template was created with a fixed body, trying to add text the preamble will have no effect. stacklevelr)rwarningswarn UserWarningr)r$r5r6s r%add_to_preamblezTexTemplate.add_to_preambleYsg :  Mw      ($J6DMM MMTCZ 'MM r'cl|jrtjdtd|xj|z c_|S)zAdds text to the TeX template just after \begin{document}, e.g. ``\boldmath``. Parameters ---------- txt String containing the text to be added. zeThis TeX template was created with a fixed body, trying to add text the document will have no effect.r9r:)rr<r=r>r)r$r5s r%add_to_documentzTexTemplate.add_to_documentpsK :  Mw     #% r' expressioncB|j|j|S)akInserts expression verbatim into TeX template. Parameters ---------- expression The string containing the expression to be typeset, e.g. ``$\sqrt{2}$`` Returns ------- :class:`str` LaTeX code based on current template, containing the given ``expression`` and ready for typesetting )r&replacer)r$rBs r%get_texcode_for_expressionz&TexTemplate.get_texcode_for_expressionsy  !6 CCCr' environmentct|\}}|j|jd|||gS)a*Inserts expression into TeX template wrapped in ``\begin{environment}`` and ``\end{environment}``. Parameters ---------- expression The string containing the expression to be typeset, e.g. ``$\sqrt{2}$``. environment The string containing the environment in which the expression should be typeset, e.g. ``align*``. Returns ------- :class:`str` LaTeX code based on template, containing the given expression inside its environment, ready for typesetting r)_texcode_for_environmentr&rDrr)r$rBrFbeginends r%!get_texcode_for_expression_in_envz-TexTemplate.get_texcode_for_expression_in_envsJ".k:: sy   !499eZ-E#F#F   r'c*tj|S)z0Create a deep copy of the TeX template instance.)copydeepcopyr#s r%rMzTexTemplate.copys}T"""r'N)rr)r(rrr))r+)r,r r-r rr )F)r5rr6r7rr )r5rrr )rBrrr)rBrrFrrr)rr )__name__ __module__ __qualname____doc__rr__annotations__rrrr_DEFAULT_PREAMBLErrrpropertyr&setter classmethodr4r?rArErKrMr0r'r%rrsKKr...E....0LTK'MR?M????`%H%%%%f*****Vo    X  [[[." D D D D    ,######r'rFrrtuple[str, str]c|dd}d|z}|ds|dz }tjd|d}d|d zdz}||fS) a>Processes the tex_environment string to return the correct ``\begin{environment}[extra]{extra}`` and ``\end{environment}`` strings. Parameters ---------- environment The tex_environment as a string. Acceptable formats include: ``{align*}``, ``align*``, ``{tabular}[t]{cccl}``, ``tabular}{cccl``, ``\begin{tabular}[t]{cccl}``. Returns ------- Tuple[:class:`str`, :class:`str`] A pair of strings representing the opening and closing of the tex environment, e.g. ``\begin{tabular}{cccl}`` and ``\end{tabular}`` z\begin{z\begin{)}]r[)maxsplitz\end{r) removeprefixendswithresplit)rFrIsplit_at_bracerJs r%rHrHs **955BB3GGK  $E >>* % %  Xc;;;;N ^A& & ,C #:r')rFrrrX)rR __future__r__all__rMrar< dataclassesrrpathlibrtypingrr r manim.typingr rTr!r"rrHr0r'r%rjs1//""""""  ((((((((%%%%%%%%%$$$$$$&!  dJ#J#J#J#J#J#J#J#Zr'