U ZʹjR @sddlmZddlmZddlmZddlmZmZm Z ddl m Z dgZ eddd fed d iffZ eeee fZd#d dZe fddZd$ddZe eddfddZddddfddZdZdddZe d dd!defd"dZd!S)%)Basic)Expr)Symbol)IntegerRationalFloat)sreprdotprintblueZellipse)colorshaper blackFcCsdd}t|tst|}n<|js(t|}n,|j}ttt|}dt|j d |f}|r`||f}|S)aNA string that follows ```obj = type(obj)(*obj.args)``` exactly. Parameters ========== with_args : boolean, optional If ``True``, there will be a second argument for the return value, which is a tuple containing ``purestr`` applied to each of the subnodes. If ``False``, there will not be a second argument for the return. Default is ``False`` Examples ======== >>> from sympy import Float, Symbol, MatrixSymbol >>> from sympy import Integer # noqa: F401 >>> from sympy.core.symbol import Str # noqa: F401 >>> from sympy.printing.dot import purestr Applying ``purestr`` for basic symbolic object: >>> code = purestr(Symbol('x')) >>> code "Symbol('x')" >>> eval(code) == Symbol('x') True For basic numeric object: >>> purestr(Float(2)) "Float('2.0', precision=53)" For matrix symbol: >>> code = purestr(MatrixSymbol('x', 2, 2)) >>> code "MatrixSymbol(Str('x'), Integer(2), Integer(2))" >>> eval(code) == MatrixSymbol('x', 2, 2) True With ``with_args=True``: >>> purestr(Float(2), with_args=True) ("Float('2.0', precision=53)", ()) >>> purestr(MatrixSymbol('x', 2, 2), with_args=True) ("MatrixSymbol(Str('x'), Integer(2), Integer(2))", ("Str('x')", 'Integer(2)', 'Integer(2)')) z%s(%s), ) isinstancerstrargsrtuplemappurestrtype__name__join)x with_argsZsargsrvrrrQ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/printing/dot.pyrs1   rcCs*i}|D]\}}t||r||q|S)a Merge style dictionaries in order Examples ======== >>> from sympy import Symbol, Basic, Expr, S >>> from sympy.printing.dot import styleof >>> styles = [(Basic, {'color': 'blue', 'shape': 'ellipse'}), ... (Expr, {'color': 'black'})] >>> styleof(Basic(S(1)), styles) {'color': 'blue', 'shape': 'ellipse'} >>> x = Symbol('x') >>> styleof(x + 1, styles) # this is an Expr {'color': 'black', 'shape': 'ellipse'} )rupdate)exprstylesstyletypZstyrrrstyleofNs    r"rcCs|ddt|DS)z Print a dictionary of attributes Examples ======== >>> from sympy.printing.dot import attrprint >>> print(attrprint({'color': 'blue', 'shape': 'ellipse'})) "color"="blue", "shape"="ellipse" css|]}d|VqdS)z "%s"="%s"Nr).0itemrrr qszattrprint..)rsorteditems)d delimiterrrr attrprintgs r*rTcCsdt||}t|tr(|js(t|jj}n||}||d<t|}|rT|dt|7}d|t|fS)z String defining a node Examples ======== >>> from sympy.printing.dot import dotnode >>> from sympy.abc import x >>> print(dotnode(x)) "Symbol('x')_()" ["color"="black", "label"="x", "shape"="ellipse"]; label_%sz "%s" [%s];) r"rrZis_Atomr __class__rrr*)rr labelfuncposrepeatr r+expr_strrrrdotnodets r2cCs t|t SNrrrrrrr6cs\||r gSt|dd\}|rFdt7fddt|D}fdd|DSdS)a 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,)"; T)rr,cs&g|]\}}d|t|ffqS)z%s_%s)r)r#ia)r/rr szdotedges..csg|]}d|fqS)z "%s" -> "%s";r)r#r9)r1rrr:sN)rr enumerate)ratomr/r0Zarg_strsr)r1r/rdotedgess r=z|digraph{ # Graph style %(graphstyle)s ######### # Nodes # ######### %(nodes)s ######### # Edges # ######### %(edges)s }ZTDout)ZrankdirZorderingcCs t|t Sr3r4r5rrrr6r7Nc sdt}||ggdfdd |dtt|dddddS) a DOT description of a SymPy expression tree Parameters ========== styles : list of lists composed of (Class, mapping), optional Styles for different classes. The default is .. code-block:: python ( (Basic, {'color': 'blue', 'shape': 'ellipse'}), (Expr, {'color': 'black'}) ) atom : function, optional Function used to determine if an arg is an atom. A good choice is ``lambda x: not x.args``. The default is ``lambda x: not isinstance(x, Basic)``. maxdepth : integer, optional The maximum depth. The default is ``None``, meaning no limit. repeat : boolean, optional Whether to use different nodes for common subexpressions. The default is ``True``. For example, for ``x + x*y`` with ``repeat=True``, it will have two nodes for ``x``; with ``repeat=False``, it will have one node. .. warning:: Even if a node appears twice in the same object like ``x`` in ``Pow(x, x)``, it will still only appear once. Hence, with ``repeat=False``, the number of arrows out of an object might not equal the number of args it has. labelfunc : function, optional A function to create a label for a given leaf node. 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)``. **kwargs : optional Additional keyword arguments are included as styles for the graph. Examples ======== >>> from sympy 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)r.r/r0)r<r/r0cs.g|]&\}}|s|d|fqS)r)r#r8arg)r<depthr/traverserrr:!sz.dotprint..traverse..)appendr2extendr=r;r)erAr/r<edgesr.maxdepthnodesr0rrB)rAr/rrBs  zdotprint..traverser )r)) graphstylerIrG)r) _graphstylecopyrtemplater*r)rrr<rHr0r.kwargsrKrrFrr s]   )F)r)Zsympy.core.basicrZsympy.core.exprrZsympy.core.symbolrZsympy.core.numbersrrrZsympy.printing.reprr__all__default_stylesZ slotClassesrr"r*rr2r=rNrLr rrrrs.        ?