B [@sddlmZmZddlmZddlmZddlmZddl m Z m Z m Z ddl mZddlmZddlmZd gZed d d fed diffZeefZee e e fZddZefddZd%ddZeeddfddZddddfddZdZdd d!Zed"dd#defd$d Zd#S)&)print_functiondivision)Basic)Expr)Symbol)IntegerRationalFloat)default_sort_key)Add)MuldotprintZblueZellipse)colorshaperZblackcstttstSttkr4fddjD}n"ttkrPtjt d}nj}dtj d t t |fS)z: A string that follows obj = type(obj)(*obj.args) exactly csg|]}t|qS)getattr).0Zslot)xr1lib/python3.7/site-packages/sympy/printing/dot.py szpurestr..)keyz%s(%s)z, ) isinstancerstrtype slotClasses __slots__ sort_classessortedargsr __name__joinmappurestr)rrr)rrr"s   r"cCs0t}x$|D]\}}t||r ||q W|S)a Merge style dictionaries in order >>> 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'} )dictrupdate)exprstylesstyletypZstyrrrstyleof!s  r), cCs|ddt|DS)z Print a dictionary of attributes >>> from sympy.printing.dot import attrprint >>> print(attrprint({'color': 'blue', 'shape': 'ellipse'})) "color"="blue", "shape"="ellipse" css|]}d|VqdS)z "%s"="%s"Nr)ritemrrr =szattrprint..)r ritems)d delimiterrrr attrprint6sr0rTcCsdt||}t|tr(|js(t|jj}n||}||d<t|}|rT|dt|7}d|t|fS)z String defining a node >>> from sympy.printing.dot import dotnode >>> from sympy.abc import x >>> print(dotnode(x)) "Symbol(x)_()" ["color"="black", "label"="x", "shape"="ellipse"]; labelz_%sz "%s" [%s];) r)rrZis_Atomr __class__rr"r0)r%r& labelfuncposrepeatr'r1expr_strrrrdotnode?s r7cCs t|t S)N)rr)rrrrTsr8csd||r gSt|dd|jD}|rNdt7fddt|D}fdd|DSdS)ao List of strings for all expr->expr.arg pairs See the docstring of dotprint for explanations of the options. >>> 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,)"; cSsg|] }t|qSr)r")rargrrrrfszdotedges..z_%scs&g|]\}}|dt|fqS)z_%s)r)riarg_str)r4rrriscsg|]}d|fqS)z "%s" -> "%s";r)rr;)r6rrrjsN)r"rr enumerate)r%atomr4r5Zarg_strsr)r6r4rdotedgesTs r>z|digraph{ # Graph style %(graphstyle)s ######### # Nodes # ######### %(nodes)s ######### # Edges # ######### %(edges)s }ZTDout)ZrankdirZorderingcCs t|t S)N)rr)rrrrr8sNc sdt}||ggdfdd |dtt|dddddS) aO 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,)"; } rc s`t|dr(kr(dSt|d fddt|jDdS)N)r3r4r5)r=r4r5cs.g|]&\}}|s|d|fqS)r)rr:r9)r=depthr4traverserrrsz.dotprint..traverse..)appendr7extendr>r<r)erAr4)r=edgesr3maxdepthnodesr5r&rB)rAr4rrBs  zdotprint..traverser )r/) graphstylerHrF)r) _graphstylecopyr$templater0r )r%r&r=rGr5r3kwargsrJr)r=rFr3rGrHr5r&rBrr sA   )r*) Z __future__rrZsympy.core.basicrZsympy.core.exprrZsympy.core.symbolrZsympy.core.numbersrrr Zsympy.core.compatibilityr Zsympy.core.addr Zsympy.core.mulr __all__Zdefault_stylesrrr"r)r0rr7r>rMrKr rrrrs*          )