ó Ħĵ™\c@ sjddlmZmZddlmZddlmZddlmZddl m Z m Z m Z ddl mZddlmZddlmZd gZeid d 6d d 6feidd 6ffZeefZee e e fZd„Zed„Zdd„Zeeded„Zd„ded„ZdZidd6dd6Zed„deed„Z!dS( i˙˙˙˙(tprint_functiontdivision(tBasic(tExpr(tSymbol(tIntegertRationaltFloat(tdefault_sort_key(tAdd(tMultdotprinttbluetcolortellipsetshapetblackcC sŻt|tƒst|ƒSt|ƒtkrSg|jD]}t||ƒ^q5}n3t|ƒtkr}t|j dt ƒ}n |j }dt|ƒj dj t t|ƒƒfS(s: A string that follows obj = type(obj)(*obj.args) exactly tkeys%s(%s)s, (t isinstanceRtstrttypet slotClassest __slots__tgetattrt sort_classestsortedtargsRt__name__tjointmaptpurestr(txtslotR((s1lib/python2.7/site-packages/sympy/printing/dot.pyRs ( cC sCtƒ}x3|D]+\}}t||ƒr|j|ƒqqW|S(sÔ Merge style dictionaries in order Examples ======== >>> from sympy import Symbol, Basic, Expr >>> from sympy.printing.dot import styleof >>> styles = [(Basic, {'color': 'blue', 'shape': 'ellipse'}), ... (Expr, {'color': 'black'})] >>> styleof(Basic(1), styles) {'color': 'blue', 'shape': 'ellipse'} >>> x = Symbol('x') >>> styleof(x + 1, styles) # this is an Expr {'color': 'black', 'shape': 'ellipse'} (tdictRtupdate(texprtstyleststylettyptsty((s1lib/python2.7/site-packages/sympy/printing/dot.pytstyleof!s  s, cC s#|jd„t|jƒƒDƒƒS(sÙ Print a dictionary of attributes Examples ======== >>> from sympy.printing.dot import attrprint >>> print(attrprint({'color': 'blue', 'shape': 'ellipse'})) "color"="blue", "shape"="ellipse" cs s|]}d|VqdS(s "%s"="%s"N((t.0titem((s1lib/python2.7/site-packages/sympy/printing/dot.pys Cs(RRtitems(tdt delimiter((s1lib/python2.7/site-packages/sympy/printing/dot.pyt attrprint9s cC st||ƒ}t|tƒr=|j r=t|jjƒ}n ||ƒ}||d>> from sympy.printing.dot import dotnode >>> from sympy.abc import x >>> print(dotnode(x)) "Symbol(x)_()" ["color"="black", "label"="x", "shape"="ellipse"]; tlabels_%ss "%s" [%s];( R(RRtis_AtomRt __class__RRR.(R#R$t labelfunctpostrepeatR%R/texpr_str((s1lib/python2.7/site-packages/sympy/printing/dot.pytdotnodeEs    cC st|tƒ S(N(RR(R((s1lib/python2.7/site-packages/sympy/printing/dot.pyt]tc C sş||ƒrgSt|ƒ}g|jD]}t|ƒ^q&}|r•|dt|ƒ7}gt|ƒD]'\}}|dt||fƒ^qe}ng|D]}d||f^qœSdS(sŠ List of strings for all expr->expr.arg pairs See the docstring of dotprint for explanations of the options. Examples ======== >>> from sympy.printing.dot import dotedges >>> from sympy.abc import x >>> for e in dotedges(x+2): ... print(e) "Add(Integer(2), Symbol(x))_()" -> "Integer(2)_(0,)"; "Add(Integer(2), Symbol(x))_()" -> "Symbol(x)_(1,)"; s_%ss "%s" -> "%s";N(RRRt enumerate( R#tatomR3R4R5targtarg_strstitarg_str((s1lib/python2.7/site-packages/sympy/printing/dot.pytdotedges]s  "=s|digraph{ # Graph style %(graphstyle)s ######### # Nodes # ######### %(nodes)s ######### # Edges # ######### %(edges)s }tTDtrankdirtouttorderingcC st|tƒ S(N(RR(R((s1lib/python2.7/site-packages/sympy/printing/dot.pyR7sc s”tjƒ}|j|ƒg‰g‰d‡‡‡‡‡‡‡‡fd†‰ˆ|dƒtit|ddƒd6djˆƒd6djˆƒd6S( sO DOT description of a SymPy expression tree Options are ``styles``: Styles for different classes. The default is:: [(Basic, {'color': 'blue', 'shape': 'ellipse'}), (Expr, {'color': 'black'})]`` ``atom``: Function used to determine if an arg is an atom. The default is ``lambda x: not isinstance(x, Basic)``. Another good choice is ``lambda x: not x.args``. ``maxdepth``: The maximum depth. The default is None, meaning no limit. ``repeat``: Whether to different nodes for separate common subexpressions. The default is True. For example, for ``x + x*y`` with ``repeat=True``, it will have two nodes for ``x`` and with ``repeat=False``, it will have one (warning: even if it appears twice in the same object, like Pow(x, x), it will still only appear only once. Hence, with repeat=False, the number of arrows out of an object might not equal the number of args it has). ``labelfunc``: How to label leaf nodes. The default is ``str``. Another good option is ``srepr``. For example with ``str``, the leaf nodes of ``x + 1`` are labeled, ``x`` and ``1``. With ``srepr``, they are labeled ``Symbol('x')`` and ``Integer(1)``. Additional keyword arguments are included as styles for the graph. Examples ======== >>> from sympy.printing.dot import dotprint >>> from sympy.abc import x >>> print(dotprint(x+2)) # doctest: +NORMALIZE_WHITESPACE digraph{ # Graph style "ordering"="out" "rankdir"="TD" ######### # Nodes # ######### "Add(Integer(2), Symbol(x))_()" ["color"="black", "label"="Add", "shape"="ellipse"]; "Integer(2)_(0,)" ["color"="black", "label"="2", "shape"="ellipse"]; "Symbol(x)_(1,)" ["color"="black", "label"="x", "shape"="ellipse"]; ######### # Edges # ######### "Add(Integer(2), Symbol(x))_()" -> "Integer(2)_(0,)"; "Add(Integer(2), Symbol(x))_()" -> "Symbol(x)_(1,)"; } c  s°ˆjt|ˆdˆd|dˆƒƒˆr>|ˆkr>dSˆjt|dˆd|dˆƒƒgt|jƒD]5\}}ˆ|ƒssˆ||d||fƒ^qsdS(NR2R3R4R:i(tappendR6textendR?R9R(tetdepthR3R=R;(R:tedgesR2tmaxdepthtnodesR4R$ttraverse(s1lib/python2.7/site-packages/sympy/printing/dot.pyRKÓs (%iR-s t graphstyleRJRH((t _graphstyletcopyR"ttemplateR.R(R#R$R:RIR4R2tkwargsRL((R:RHR2RIRJR4R$RKs1lib/python2.7/site-packages/sympy/printing/dot.pyR sA  ' N((("t __future__RRtsympy.core.basicRtsympy.core.exprRtsympy.core.symbolRtsympy.core.numbersRRRtsympy.core.compatibilityRtsympy.core.addR tsympy.core.mulR t__all__tdefault_stylesRRRR(R.RtTrueR6R?RORMtNoneR (((s1lib/python2.7/site-packages/sympy/printing/dot.pyts*    ,