B [C@sdZddlmZmZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZd d Zd d ZddZdZdddZdS)z7Tools for setting up printing in interactive sessions. )print_functiondivisionN) LooseVersion)BytesIO)latex)preview) integer_types)debugc s.ddl}ddlmfdd}||_dS)z.Setup printing in Python interactive session. rN)builtinscs(|dk r$d_t|f|_dS)zPython's pretty-printer display hook. This function was adapted from: http://www.python.org/dev/peps/pep-0217/ N)_print)arg)r settingsstringify_func9lib/python3.7/site-packages/sympy/interactive/printing.py _displayhooksz+_init_python_printing.._displayhook)syssympy.core.compatibilityr displayhook)rrrrr)r rrr_init_python_printings  rc  syddlmWntk r$YnXd|r4d} nd} || fd} d} d}d | |||| f}|td td | pt fd d }fddfdd fdd fdd} fdd} fdd} fdd}ddltjdkrddl m }ddl m }dd l m}m}dd!lm}||tttttt|||g tt}|jjd"}x|D]}|||qW|jjd#}|d$krtd%x||D]}|||qWn`|d&krtd'xL|D]}|||qWn0td(x&|D]}||jkr|j|qW|jjd)}|d*krptd+xL|D]}|||qXWn0td,x4|D]}||jkr~|j|q~Wn |d-|dS).z/Setup printing in IPython interactive session. r) latex_to_pngz]\documentclass[%s]{article} \pagestyle{empty} \usepackage{amsmath,amsfonts}%s\begin{document}z\usepackage{euler}Ztightz0cm,0cmz-T %s -D %d -bg %s -fg %s -O %szinit_printing: DVIOPTIONS:zinit_printing: PREAMBLE:cs0|r||n|jj|dS)z*caller for pretty, for use in IPython 0.11N)textlibpretty)r pcycle)IPython_can_print_latexrrr _print_plainAsz,_init_ipython_printing.._print_plainc s\t}yt|dd|dWn4tk rR}ztddt|Wdd}~XYnX|S)Npngr)outputZviewerZ outputbufferpreamble dvioptionsz png printing:z"_preview_wrapper exception raised:)rr Exceptionr reprgetvalue)oZ exprbuffere)r%r$rr_preview_wrapperHs  z0_init_ipython_printing.._preview_wrapperc sV|dd}|dd}y|Stk rP}ztdt|dSd}~XYnXdS)Nz \operatornamerz \overlinez\barzmatplotlib exception caught:)replace ValueErrorr r')r)r*)rrr_matplotlib_wrapperUs  z3_init_ipython_printing.._matplotlib_wrappercsyddlm}ddlm}ddlm}m}ddlm}t t t t f}t |rtjdd|Dks~tjdd|Dkrd Stfd dDSt trtfd dDSt trd St |||||frd St ttfrrd Sd Stk rd SXd S)zReturn True if type o can be printed with LaTeX. If o is a container type, this is True if and only if every element of o can be printed with LaTeX. r)Basic) MatrixBase)VectorDyadic) NDimArraycss|] }|jVqdS)N)__str__).0irrr vszC_init_ipython_printing.._can_print_latex..css|] }|jVqdS)N)__repr__)r5r6rrrr7wsFc3s|]}|VqdS)Nr)r5r6)r rrr7ysc3s"|]}|o|VqdS)Nr)r5r6)r r)rrr7{sTN)sympyr/Zsympy.matricesr0sympy.physics.vectorr1r2sympy.tensor.arrayr3listtupleset frozenset isinstancetyper4r8alldictboolfloatr RuntimeError)r)r/r0r1r2r3Z builtin_types)r print_builtin)r)rr cs,       z0_init_ipython_printing.._can_print_latexc s||rx|fdi}y|Stk rv}z4tdt|ddkrb|fddi}|Sd}~XYnXdS)z A function that returns a png rendered by an external latex distribution, falling back to matplotlib rendering modezpreview failed with:z# Falling back to matplotlib backendinlineN)rFr r')r)sr*)r r.r+r latex_moderrr_print_latex_pngs z0_init_ipython_printing.._print_latex_pngcs(|r$|fddi}|SdS)zD A function that returns a png rendered by mathtext rHrINr)r)rJ)r r.rrrr_print_latex_matplotlibsz7_init_ipython_printing.._print_latex_matplotlibcs2|r.|fddi}|d}d|SdS)zW A function to generate the latex representation of sympy expressions. rHZplain$z$$%s$$N)strip)r)rJ)r rrrr_print_latex_texts z1_init_ipython_printing.._print_latex_textcs6|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)ZrcZpprintr r')selfr out)rrr_result_displays  z/_init_ipython_printing.._result_displayNz0.11)r/)r0)r1r2)r3z text/plainz 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 ImportErrorsplitr default_latexrV __version__Zsympy.core.basicr/Zsympy.matrices.matricesr0r:r1r2r;r3rEr=r<r>r?rCrZdisplay_formatterZ formattersZfor_typeZ type_printerspopZset_hook)ipr use_latexeuler forecolor backcolorfontsizerKrG latex_printerrZ addpackagesZ imagesizeoffsetZ resolutionZdvir!rLrMrPrTr/r0r1r2r3Zprintable_typesZplaintext_formatterclsZ png_formatterZlatex_formatterr) rr r.r+r%rrKrr$rGrrr_init_ipython_printing$sx    (                   rdc CsddtjkrdSyddlm}Wn:tk rXyddlm}Wntk rRdSXYnXt||S)z%Is a shell instance an IPython shell?rFr)InteractiveShell)rmodulesZIPython.core.interactiveshellrerUZ IPython.iplibr@)shellrerrr _is_ipythons  rhFTBlack Transparent10pt equation*c sddl}ddlm}|r4|dk r&|}qNddlm}n|dk rB|}n ddlm}d}|dkry t}Wntk rxYn X|dk }|r|st|}|rB|rBy@ddl }t |j dkrddl m }n ddlm }dd lm}Wntk rYnPXt|||fsBd d |jkrBdkr,td d |dkrBtdd }tsb|sb|jdn.||rfdd}nfdd}|r|dd}|rtdt||||| | | | | |f |n t|f|dS)a) Initializes pretty-printer depending on the environment. Parameters ========== pretty_print: boolean 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 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 If True, use unicode characters; if False, do not use unicode characters. use_latex: string, boolean, or None If True, use default latex rendering in GUI interfaces (png and mathjax); if False, do not use latex rendering; 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 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 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 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, optional, default='Black' DVI setting for foreground color. backcolor: string, optional, default='Transparent' DVI setting for background color. fontsize: string, optional, default='10pt' A font size to pass to the LaTeX documentclass function in the preamble. latex_mode: string, optional, default='equation*' 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(). 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 rN)Printer)r)sstrreprFz1.0)TerminalInteractiveShell)InteractiveConsolezipython-consolerz*init_printing: Setting use_unicode to TrueTz(init_printing: Setting use_latex to True)order use_unicode wrap_line num_columnscs|dS)N)rqrrrsrtr)expr)_stringify_funcrtrqrrrsrrszinit_printing..cs |dS)N)rqr)ru)rvrqrrrwsrHzMinit_printing: Mode is not able to be set due to internalsof IPython printing)rZsympy.printing.printerrmZsympy.printingrrnZ get_ipython NameErrorrhrrXrYZ!IPython.terminal.interactiveshellroZ*IPython.frontend.terminal.interactiveshellcoderprUr@joinargvr NO_GLOBALZset_global_settingsrZrdr)Z pretty_printrqrrr\rsrtZ no_globalr[r]r^r_r`rKrGZ str_printerZpretty_printerrarrrmrZ in_ipythonrrorpZmode_in_settingsr)rvrtrqrrrsr init_printingsdn           r})TNNNNNFNFrirjrkrlTNNN)__doc__Z __future__rrrZdistutils.versionrrXiorr9rrWrrrZsympy.utilities.miscr rrdrhr|r}rrrrs(      K