o 8VaY@sdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z d d Zd d Zd dZdZ       dddZdS)z7Tools for setting up printing in interactive sessions. N) version_tuple)BytesIO)latex)preview)debug) Printablec s*ddl}ddlfdd}||_dS)z.Setup printing in Python interactive session. rNcs0|durd_t|fi|_dSdS)zPython's pretty-printer display hook. This function was adapted from: http://www.python.org/dev/peps/pep-0217/ N)_print)argbuiltinssettingsstringify_func._displayhook)sysr displayhook)rr rrrr r_init_python_printing s rc  szddlm Wn tyYnwdur0|j} | dkr"dn | dkr)dndtd d |r7d } nd } |d krC| d} || f d}d}td}d||||f}|d}dd|gtdtdtd | p}t fdd} fdd fdd fdd d!  fd"d# fd$d%} fd&d'}fd(d)} fd*d+}fd,d-}ddl t j t d.krt ttttttg}|jjd/}|tgD]}|||q|jjd0}|d1vrtd2|D] }|||q|t_ntd3|D]}||jvr1|j|q#tjt_|jjd4}|d5vrVtd6|D] }|||qH|t_n3|d7krotd8|D] }|||qa|t_ntd9|D]}||jvr|j|qutjt_|jjd:}|d;vrtd<|D] }|||q|t_dStd=|D]}||jvr|j|qtjt_dS|d>|dS)?z/Setup printing in IPython interactive session. r) latex_to_pngNZlightbgZBlackZlinuxZWhiteZGrayz*init_printing: Automatic foreground color:zW\documentclass[varwidth,%s]{standalone} \usepackage{amsmath,amsfonts}%s\begin{document}z\usepackage{euler}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:zinit_printing: PREAMBLE:cs2|r ||dS|jj|dS)z*caller for pretty, for use in IPython 0.11N)textlibpretty)r pcycle)IPython _can_printrrr _print_plainWsz,_init_ipython_printing.._print_plainc sPt}zt|dd|dW|Sty'}z tddt|d}~ww)NpngroutputZviewerZ outputbufferpreamble dvioptionsz png printing:"_preview_wrapper exception raised:)rr ExceptionrreprgetvalueoZ exprbuffere)r%r$rr_preview_wrapper^sz0_init_ipython_printing.._preview_wrapperc sXt}z t|dd|dWnty$}z tddt|d}~ww|dS)Nrrr"z svg printing:r&zutf-8)rrr'rr(r)decoder*)dvioptions_svgr$rr _svg_wrapperks z,_init_ipython_printing.._svg_wrapperc s~|dd}|dd}zz |dWWSty$|YWSwty>}ztdt|WYd}~dSd}~ww)Nz \operatornamerz \overlinez\bar)colorscalezmatplotlib exception caught:)replace TypeError ValueErrorrr()r+r,) forecolorrr2rr_matplotlib_wrapperxs   z3_init_ipython_printing.._matplotlib_wrapper)Z_latexZ _sympystrZ_prettyZ _sympyreprcszqttttf}t|r3tjdd|Dvs$tjdd|Dvr'WdStfddDWStt rEtfddDWStt rMWdStt rUWdSt fddDrcWdStt tfroroWdSWdSty{YdSw) 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. cs|]}|jVqdSN)__str__.0irrr z=_init_ipython_printing.._can_print..csr8r9)__repr__r;rrrr>r?Fc3s|]}|VqdSr9rr;)rrrr>sc3s$|] }|o |VqdSr9rr;)rr+rrr>s"Tc3s|]}t|VqdSr9)hasattr)r<hookr+rrr>s)listtupleset frozenset isinstancetyper:r@alldictboolranyfloatint RuntimeError)r+Z builtin_types)r print_builtinprinting_hooksrCrrs,      z*_init_ipython_printing.._can_printc s|rH|fdi}dkrd|}z|WStyG}z tdt|ddkr9|fddi}|WYd}~Sd}~wwdS)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 backendinlineNrPrr(r+sr,)rr7r-r latex_moder rr_print_latex_pngs   z0_init_ipython_printing.._print_latex_pngc sr|r7|fdi}dkrd|}z|WSty6}ztdt|dWYd}~dSd}~wwdS)zz A function that returns a svg rendered by an external latex distribution, no fallback available. rSrTrUrVz No fallback available.NrXrY)rr0rr[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)rr7rr rr_print_latex_matplotlibsz7_init_ipython_printing.._print_latex_matplotlibcs4|r|fdi}dkrd|S|SdS)zW A function to generate the latex representation of sympy expressions. rSrTrUNrr^)rrr[r rr_print_latex_texts z1_init_ipython_printing.._print_latex_textcs:|jjr|}d|vrtt|dStt|dS)zIPython's pretty-printer display hook, for use in IPython 0.10 This function was adapted from: ipython/IPython/hooks.py:155  N)ZrcZpprintr 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 ImportErrorZcolorslowerrroundsplitformat default_latexrr __version__rNrErDrFrGrKrOZdisplay_formatterZ formattersrZfor_typeZ _repr_svg_Z type_printerspopZ_repr_disabledZ _repr_png_Z _repr_latex_Zset_hook)ipr use_latexeulerr6 backcolorfontsizer[rQ latex_printerr2r r1Z addpackagesZ imagesizeoffsetZ resolutionZdviZ svg_scaler r\r]r_r`rdZprintable_typesZplaintext_formatterclsZ svg_formatterZ png_formatterZlatex_formatterr)rrr7r-r0r%r/r6rr[rr$rQrRr2r rr_init_ipython_printing"s            &                 ruc CsddtjvrdSzddlm}Wnty,zddlm}Wn ty)YYdSwYnwt||S)z%Is a shell instance an IPython shell?rFr)InteractiveShell)rmodulesZIPython.core.interactiveshellrvreZ IPython.iplibrH)shellrvrrr _is_ipython?s    ryFT Transparent10ptrT?c sddl}ddlm}|r|dur|}nddlm}n |dur!|}nddlm}d}|dur@zt}Wn ty;Ynw|du}|rH|sHt|}|r|rz"ddl }t |j t dkraddl m }nddlm }dd lm}Wn tywYn$wt|||fsd d |jvrdurtd d |durtdd }ts|s|jdn||rfdd}nfdd}|r|dd}|rtdt||||| | | | | ||f i|dSt|fi|dS)a Initializes pretty-printer depending on the environment. Parameters ========== pretty_print : boolean, default=True If True, use pretty_print to stringify or the provided pretty printer; if False, use 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 : boolean 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, boolean, 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 : boolean 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 : boolean, 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 : boolean, optional, default=False Loads the euler package in the LaTeX preamble for handwritten style fonts (http://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, 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 sympy.printing.sstrrepr(). pretty_printer : function, optional, default=None A custom pretty printer. This should mimic sympy.printing.pretty(). latex_printer : function, optional, default=None A custom LaTeX printer. This should mimic sympy.printing.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_funcrrrrrr(szinit_printing..cs|fdi|S)Nrrr)rrrrr0srSzMinit_printing: Mode is not able to be set due to internalsof IPython printing)rZsympy.printing.printerr}Zsympy.printingrr~Z get_ipython NameErrorryrrrkZ!IPython.terminal.interactiveshellrZ*IPython.frontend.terminal.interactiveshellcoderrerHjoinargvr NO_GLOBALZset_global_settingsrlrur)Z pretty_printrrrnrrZ no_globalrmror6rprqr[rQZ str_printerZpretty_printerrrr2r rr}rZ in_ipythonrrrZmode_in_settingsrrr init_printingSsv          r)TNNNNNFNFNrzr{rTTNNNr|)__doc__rZsympy.external.importtoolsriorZsympyrrjrZsympy.utilities.miscrZsympy.printing.defaultsrrruryrrrrrrs,