U ZʹjX@svdZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd Zd d Zd d ZdZdddZdS)z7Tools for setting up printing in interactive sessions. ) version_tuple)BytesIO)latex)preview)debug) Printablec s*ddl}ddlfdd}||_dS)z.Setup printing in Python interactive session. rNcs(|dk r$d_t|f|_dS)zPython's pretty-printer display hook. This function was adapted from: https://www.python.org/dev/peps/pep-0217/ N)_print)argbuiltinssettingsstringify_funcY/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/interactive/printing.py _displayhooksz+_init_python_printing.._displayhook)sysr displayhook)rr rrrr r_init_python_printing s rc  s|zddlm Wntk r$YnX dkrb|j} | dkrFd n| dkrTd nd td |d krtd nd d } d}td}d| || |f}|d}dd|gtdtd| pt fdd} fdd fdd fdddfdd  fd!d"} fd#d$} fd%d&} fd'd(}fd)d*}ddl t j t d+krlt ttttttg}|jjd,}|tgD]}|||q|jjd-}|d.kr td/|D]}|||q|t_n4td0|D]}||jkr,|j|q,tjt_|jjd1}|d2krtd3|D]}|||qv|t_nf|d4krtd5|D]}|||q|t_n4td6|D]}||jkr|j|qtjt_|jjd7}|d8kr6td9|D]}|||q|t_n4td:|D]}||jkrB|j|qBtjt_n |d;|dS)._print_plainc s`t}zt|dd|dWn4tk rV}ztddt|W5d}~XYnX|S)NpngroutputZviewereulerZ outputbufferextra_preamble dvioptionsfontsizez png printing:"_preview_wrapper exception raised:)rr ExceptionrreprgetvalueoZ exprbuffere)r'r%r&r(rr_preview_wrapperVs  z0_init_ipython_printing.._preview_wrapperc sft}zt|dd|dWn4tk rV}ztddt|W5d}~XYnX|dS)Nrrr#z svg printing:r)zutf-8)rrr*rr+r,decoder-)dvioptions_svgr%r&r(rr _svg_wrappercs  z,_init_ipython_printing.._svg_wrapperc srz6z|dWWStk r2|YWSXWn6tk rl}ztdt|WYdSd}~XYnXdS)N)colorscalezmatplotlib exception caught:) TypeError ValueErrorrr+)r.r/) forecolorrr5rr_matplotlib_wrapperpsz3_init_ipython_printing.._matplotlib_wrapper)Z_latexZ _sympystrZ_prettyZ _sympyreprcszttttf}t|rftjdd|DksHtjdd|DkrNWdStfddDWStt rtfddDWStt rWdStt rWdSt fddDrWdStt tfrވrWdSWdStk rYdSXd 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 sz=_init_ipython_printing.._can_print..css|] }|jVqdSr:)__repr__r<rrrr?sFc3s|]}|VqdSr:rr<)r rrr?sc3s"|]}|o|VqdSr:rr<)r r.rrr?sTc3s|]}t|VqdSr:)hasattr)r=hookr.rrr?sN)listtupleset frozenset isinstancetyper;r@alldictboolranyfloatint RuntimeError)r.Z builtin_types)r print_builtinprinting_hooksrCrr s*     z*_init_ipython_printing.._can_printc s|r|fdi}dkr,d|}z |WStk r}z._print_latex_pngc sn|rj|fdi}dkr,d|}z |WStk rh}ztdt|dW5d}~XYnXdS)zz A function that returns a svg rendered by an external latex distribution, no fallback available. rSrTrUrVz No fallback available.NrXrY)r r3rr[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 rSrWNrr.rZ)r r9rr 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. rSrTrUNrr^)r rr[r rr_print_latex_texts z1_init_ipython_printing.._print_latex_textcs8|jjr(|}d|krtt|n tt|dS)zIPython's pretty-printer display hook, for use in IPython 0.10 This function was adapted from: ipython/IPython/hooks.py:155  N)rcpprintr r+)selfr out)rrr_result_displays  z/_init_ipython_printing.._result_displayz0.11z 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 formatterZresult_display)ZIPython.lib.latextoolsr ImportErrorcolorslowerrroundsplitformat default_latexrr __version__rNrErDrFrGrKrOdisplay_formatter formattersrZfor_typeZ _repr_svg_Z type_printerspopZ_repr_disabledZ _repr_png_Z _repr_latex_Zset_hook)ipr use_latexr%r8 backcolorr(r[rQ latex_printerr5r r4Z imagesizeoffset resolutionZdviZ svg_scaler!r\r]r_r`rfZprintable_typesZplaintext_formatterclsZ svg_formatterZ png_formatterZlatex_formatterr)rr r9r0r3r'r2r%r&r(r8rr[rrQrRr5r rr_init_ipython_printing!s        &             ryc Csrddlm}d|krdSzddlm}Wn>tk rfzddlm}Wntk r`YYdSXYnXt||S)z%Is a shell instance an IPython shell?r)modulesrF)InteractiveShell)rrzZIPython.core.interactiveshellr{rgZ IPython.iplibrH)shellrzr{rrr _is_ipython3s r}FTN Transparent10ptrT?c sddl}ddlm}|r4|dk r&|}qNddlm}n|dk rB|}n ddlm}d}|dkrz t}Wntk rxYn X|dk }|r|st|}|rF|rFzDddl }t |j t dkrddl m }n ddlm }dd lm}Wntk rYnPXt|||fsFd d |jkrFdkr0td d |dkrFtdd }tsf|sf|jdn.||rfdd}nfdd}|r|dd}|rtdt||||| | | | | ||f |n t|f|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)sstrreprFz1.0)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)rrrrr%s rSzMinit_printing: Mode is not able to be set due to internalsof IPython printing)rZsympy.printing.printerrZsympy.printingrr get_ipython NameErrorr}rrrnZ!IPython.terminal.interactiveshellrZ*IPython.frontend.terminal.interactiveshellcoderrgrHjoinargvr NO_GLOBALZset_global_settingsrqryr)Z pretty_printrrrsrrZ no_globalrrr%r8rtr(r[rQZ str_printerZpretty_printerrur5r rrrZ in_ipythonrrrZmode_in_settingsrrr init_printingHs~           r)TNNNNNFNFNr~rrTTNNNr)__doc__Zsympy.external.importtoolsriorZsympy.printing.latexrrmZsympy.printing.previewrZsympy.utilities.miscrZsympy.printing.defaultsrrryr}rrrrrrs: