a kh:T@svdZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd Zd d Zd d ZdZdddZdS)z7Tools for setting up printing in interactive sessions. )BytesIO)latex)preview)debug) Printable) import_modulec s*ddl}ddlfdd}||_dS)z.Setup printing in Python interactive session. rNcs,|dur(d_t|fi|_dS)zPython's pretty-printer display hook. This function was adapted from: https://www.python.org/dev/peps/pep-0217/ N)_print)argbuiltinssettingsstringify_funcH/var/www/auris/lib/python3.9/site-packages/sympy/interactive/printing.py _displayhooksz+_init_python_printing.._displayhook)sysr displayhook)rr rrrr r_init_python_printing s rc  sPtdddzddlm Wnty.Yn0 durl|j} | dkrPd n| d kr^d nd td |d kr~d ndd} d}td}d| || |f}|d}dd |gtdtd| pt fdd} fdd fdd fdd d!fd"d# fd$d%} fd&d'} fd(d)} fd*d+}t t t ttttg}|jjd,}|tgD]}|||q|jjd-}|d.vrtd/|D]}|||q|t_n4td0|D]}||jvr|j|qtjt_|jjd1}|d2vrttd3|D]}|||qX|t_nf|d4krtd5|D]}|||q|t_n4td6|D]}||jvr|j|qtjt_|jjd7}|d8vrtd9|D]}|||q|t_n4td:|D]}||jvr$|j|q$tjt_dS);z/Setup printing in IPython interactive session. IPythonz1.0)Zmin_module_versionr) latex_to_pngNZlightbgZBlacklinuxWhiteZGrayz*init_printing: Automatic foreground color:svgz \special{color %s}Ztightz0cm,0cmz-T %s -D %d -bg %s -fg %s -O %sg@z --no-fontsz --scale={}zinit_printing: DVIOPTIONS:zinit_printing: DVIOPTIONS_SVG:cs0|r||n|jj|dS)z*caller for pretty, for use in IPython 0.11N)textlibpretty)r pcycle)r _can_printrrr _print_plainPsz,_init_ipython_printing.._print_plainc sbt}zt|dd|dWn6tyX}ztddt|WYd}~n d}~00|S)NpngroutputZviewereulerZ outputbufferextra_preamble dvioptionsfontsizez png printing:"_preview_wrapper exception raised:)rr ExceptionrreprgetvalueoZ exprbuffere)r(r&r'r)rr_preview_wrapperWs  z0_init_ipython_printing.._preview_wrapperc sht}zt|dd|dWn6tyX}ztddt|WYd}~n d}~00|dS)Nrrr$z svg printing:r*zutf-8)rrr+rr,r-decoder.)dvioptions_svgr&r'r)rr _svg_wrapperds  z,_init_ipython_printing.._svg_wrapperc snz4z|dWWSty0|YWS0Wn4tyh}ztdt|WYd}~dSd}~00dS)N)colorscalezmatplotlib exception caught:) TypeError ValueErrorrr,)r/r0) forecolorrr6rr_matplotlib_wrapperqs z3_init_ipython_printing.._matplotlib_wrapper)Z_latexZ _sympystrZ_prettyZ _sympyreprcszttttf}t|rftjdd|DvsHtjdd|DvrNWdStfddDWStt rtfddDWStt rWdStt rWdSt fddDrWdStt tfrވrWdSWdStyYdS0d S) zReturn True if type o can be printed with one of the SymPy printers. If o is a container type, this is True if and only if every element of o can be printed in this way. css|] }|jVqdSN)__str__.0irrr z=_init_ipython_printing.._can_print..css|] }|jVqdSr;)__repr__r=rrrr@rAFc3s|]}|VqdSr;rr=)r!rrr@rAc3s"|]}|o|VqdSr;rr=)r!r/rrr@rATc3s|]}t|VqdSr;)hasattr)r>hookr/rrr@rAN)listtupleset frozenset isinstancetyper<rBalldictboolranyfloatint RuntimeError)r/Z builtin_types)r! print_builtinprinting_hooksrErr!s*     z*_init_ipython_printing.._can_printc s|r|fdi}dkr,d|}z |WSty}z@tdt|ddkrr|fddi}|WYd}~Sd}~00dS)z A function that returns a png rendered by an external latex distribution, falling back to matplotlib rendering modeplain$\displaystyle %s$preview failed with:z# Falling back to matplotlib backendinlineNrRrr,r/sr0)r!r:r1r latex_moder rr_print_latex_pngs  z0_init_ipython_printing.._print_latex_pngc sp|rl|fdi}dkr,d|}z |WStyj}ztdt|dWYd}~n d}~00dS)zz A function that returns a svg rendered by an external latex distribution, no fallback available. rUrVrWrXz No fallback available.NrZr[)r!r4rr]r rr_print_latex_svgs  z0_init_ipython_printing.._print_latex_svgcs(|r$|fddi}|SdS)zD A function that returns a png rendered by mathtext rUrYNrr/r\)r!r:rr rr_print_latex_matplotlibsz7_init_ipython_printing.._print_latex_matplotlibcs4|r0|fdi}dkr,d|S|SdS)zW A function to generate the latex representation of SymPy expressions. rUrVrWNrr`)r!rr]r rr_print_latex_texts z1_init_ipython_printing.._print_latex_textz text/plainz image/svg+xml)rz"init_printing: using svg formatterz*init_printing: not using any svg formatterz image/png)Tr#z"init_printing: using png formatterZ matplotlibz)init_printing: using matplotlib formatterz*init_printing: not using any png formatterz text/latex)TZmathjaxz&init_printing: using mathjax formatterz-init_printing: not using text/latex formatter)rZIPython.lib.latextoolsr ImportErrorcolorslowerrroundsplitformat default_latexrPrGrFrHrIrMrQZdisplay_formatter formattersrZfor_typeZ _repr_svg_Z type_printerspopZ_repr_disabledZ _repr_png_Z _repr_latex_)ipr use_latexr&r9 backcolorr)r]rS latex_printerr6r r5Z imagesizeoffset resolutionZdviZ svg_scaler"r^r_rarbZprintable_typesZplaintext_formatterclsZ svg_formatterZ png_formatterZlatex_formatterr)rr!r:r1r4r(r3r&r'r)r9rr]rrSrTr6r rr_init_ipython_printing!s           &           rsc Csnddlm}d|vrdSzddlm}Wn:tybzddlm}Wnty\YYdS0Yn0t||S)z%Is a shell instance an IPython shell?r)modulesrF)InteractiveShell)rrtZIPython.core.interactiveshellrurcZ IPython.iplibrJ)shellrtrurrr _is_ipythons   rwFTN Transparent10ptrV?c sddl}ddlm}|r4|dur&|}qNddlm}n|durB|}n ddlm}d}|durz t}WntyvYn 0|du}|r|st|}|r|rzddl m }ddl m }Wnt yYnP0t|||fsd d |jvrdurtd d |durtd d }ts:|s:|jdn.||rZfdd}nfdd}|r|dd}|rtdt||||| | | | | ||f i|nt|fi|dS)a, Initializes pretty-printer depending on the environment. Parameters ========== pretty_print : bool, default=True If ``True``, use :func:`~.pretty_print` to stringify or the provided pretty printer; if ``False``, use :func:`~.sstrrepr` to stringify or the provided string printer. order : string or None, default='lex' There are a few different settings for this parameter: ``'lex'`` (default), which is lexographic order; ``'grlex'``, which is graded lexographic order; ``'grevlex'``, which is reversed graded lexographic order; ``'old'``, which is used for compatibility reasons and for long expressions; ``None``, which sets it to lex. use_unicode : bool or None, default=None If ``True``, use unicode characters; if ``False``, do not use unicode characters; if ``None``, make a guess based on the environment. use_latex : string, bool, or None, default=None If ``True``, use default LaTeX rendering in GUI interfaces (png and mathjax); if ``False``, do not use LaTeX rendering; if ``None``, make a guess based on the environment; if ``'png'``, enable LaTeX rendering with an external LaTeX compiler, falling back to matplotlib if external compilation fails; if ``'matplotlib'``, enable LaTeX rendering with matplotlib; if ``'mathjax'``, enable LaTeX text generation, for example MathJax rendering in IPython notebook or text rendering in LaTeX documents; if ``'svg'``, enable LaTeX rendering with an external latex compiler, no fallback wrap_line : bool If True, lines will wrap at the end; if False, they will not wrap but continue as one line. This is only relevant if ``pretty_print`` is True. num_columns : int or None, default=None If ``int``, number of columns before wrapping is set to num_columns; if ``None``, number of columns before wrapping is set to terminal width. This is only relevant if ``pretty_print`` is ``True``. no_global : bool, default=False If ``True``, the settings become system wide; if ``False``, use just for this console/session. ip : An interactive console This can either be an instance of IPython, or a class that derives from code.InteractiveConsole. euler : bool, optional, default=False Loads the euler package in the LaTeX preamble for handwritten style fonts (https://www.ctan.org/pkg/euler). forecolor : string or None, optional, default=None DVI setting for foreground color. ``None`` means that either ``'Black'``, ``'White'``, or ``'Gray'`` will be selected based on a guess of the IPython terminal color setting. See notes. backcolor : string, optional, default='Transparent' DVI setting for background color. See notes. fontsize : string or int, optional, default='10pt' A font size to pass to the LaTeX documentclass function in the preamble. Note that the options are limited by the documentclass. Consider using scale instead. latex_mode : string, optional, default='plain' The mode used in the LaTeX printer. Can be one of: ``{'inline'|'plain'|'equation'|'equation*'}``. print_builtin : boolean, optional, default=True If ``True`` then floats and integers will be printed. If ``False`` the printer will only print SymPy types. str_printer : function, optional, default=None A custom string printer function. This should mimic :func:`~.sstrrepr`. pretty_printer : function, optional, default=None A custom pretty printer. This should mimic :func:`~.pretty`. latex_printer : function, optional, default=None A custom LaTeX printer. This should mimic :func:`~.latex`. scale : float, optional, default=1.0 Scale the LaTeX output when using the ``'png'`` or ``'svg'`` backends. Useful for high dpi screens. settings : Any additional settings for the ``latex`` and ``pretty`` commands can be used to fine-tune the output. Examples ======== >>> from sympy.interactive import init_printing >>> from sympy import Symbol, sqrt >>> from sympy.abc import x, y >>> sqrt(5) sqrt(5) >>> init_printing(pretty_print=True) # doctest: +SKIP >>> sqrt(5) # doctest: +SKIP ___ \/ 5 >>> theta = Symbol('theta') # doctest: +SKIP >>> init_printing(use_unicode=True) # doctest: +SKIP >>> theta # doctest: +SKIP \u03b8 >>> init_printing(use_unicode=False) # doctest: +SKIP >>> theta # doctest: +SKIP theta >>> init_printing(order='lex') # doctest: +SKIP >>> str(y + x + y**2 + x**2) # doctest: +SKIP x**2 + x + y**2 + y >>> init_printing(order='grlex') # doctest: +SKIP >>> str(y + x + y**2 + x**2) # doctest: +SKIP x**2 + x + y**2 + y >>> init_printing(order='grevlex') # doctest: +SKIP >>> str(y * x**2 + x * y**2) # doctest: +SKIP x**2*y + x*y**2 >>> init_printing(order='old') # doctest: +SKIP >>> str(x**2 + y**2 + x + y) # doctest: +SKIP x**2 + x + y**2 + y >>> init_printing(num_columns=10) # doctest: +SKIP >>> x**2 + x + y**2 + y # doctest: +SKIP x + y + x**2 + y**2 Notes ===== The foreground and background colors can be selected when using ``'png'`` or ``'svg'`` LaTeX rendering. Note that before the ``init_printing`` command is executed, the LaTeX rendering is handled by the IPython console and not SymPy. The colors can be selected among the 68 standard colors known to ``dvips``, for a list see [1]_. In addition, the background color can be set to ``'Transparent'`` (which is the default value). When using the ``'Auto'`` foreground color, the guess is based on the ``colors`` variable in the IPython console, see [2]_. Hence, if that variable is set correctly in your IPython console, there is a high chance that the output will be readable, although manual settings may be needed. References ========== .. [1] https://en.wikibooks.org/wiki/LaTeX/Colors#The_68_standard_colors_known_to_dvips .. [2] https://ipython.readthedocs.io/en/stable/config/details.html#terminal-colors See Also ======== sympy.printing.latex sympy.printing.pretty rN)Printer)r)sstrreprF)TerminalInteractiveShell)InteractiveConsolezipython-consolerz*init_printing: Setting use_unicode to TrueTz(init_printing: Setting use_latex to Trueorder use_unicode wrap_line num_columnscs|fd|S)Nrrexprr _stringify_funcrrrrrrszinit_printing..cs|fdi|S)Nrrr)rrrrrs rUzMinit_printing: Mode is not able to be set due to internalsof IPython printing)rZsympy.printing.printerr{Zsympy.printingrr|Z get_ipython NameErrorrwZ!IPython.terminal.interactiveshellr}coder~rcrJjoinargvr NO_GLOBALZset_global_settingsrkrsr)Z pretty_printrrrmrrZ no_globalrlr&r9rnr)r]rSZ str_printerZpretty_printerror6r rr{rZ in_ipythonr}r~Zmode_in_settingsrrr init_printing1sl             r)TNNNNNFNFNrxryrVTNNNrz)__doc__iorZsympy.printing.latexrriZsympy.printing.previewrZsympy.utilities.miscrZsympy.printing.defaultsrZsympy.externalrrrsrwrrrrrrs&      |