o o hR @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)sreprdotprintblueellipse)colorshaper blackFcCsdd}t|ts t|}n|jst|}n|j}ttt|}dt|j d |f}|r0||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_argssargsrvrrrf/var/www/html/construction_image-detection-poc/venv/lib/python3.10/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)exprstylesstyletypstyrrrstyleofNs    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|jst|jj}n||}||d<t|}|r*|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%rris_Atomr __class__rrr-)r r! labelfuncposrepeatr"r.expr_strrrrdotnodets r6cC t|t SNrrrrrr r;csX||rgSt|dd\}|r#dt7fddt|D}fdd|DS)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)r3rr szdotedges..csg|]}d|fqS)z "%s" -> "%s";r)r&r>)r5rrr?s)rr enumerate)r atomr3r4arg_strsr)r5r3rdotedgess rCz|digraph{ # Graph style %(graphstyle)s ######### # Nodes # ######### %(nodes)s ######### # Edges # ######### %(edges)s }TDout)rankdirorderingcCr7r8r9r:rrrr;r<Nc sdt}||ggd fdd |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|drkrdSt|d fddt|jDdS)N)r2r3r4)rAr3r4cs.g|]\}}|s|d|fqS)r)r&r=arg)rAdepthr3traverserrr?!s.z.dotprint..traverse..)appendr6extendrCr@r)erJr3rAedgesr2maxdepthnodesr4r!rK)rJr3rrKs  "zdotprint..traverser )r,) graphstylerRrPN)r) _graphstylecopyrtemplater-r)r r!rArQr4r2kwargsrTrrOrr s]   )F)r)sympy.core.basicrsympy.core.exprrsympy.core.symbolrsympy.core.numbersrrrsympy.printing.reprr__all__default_styles slotClassesrr%r-rr6rCrWrUr rrrrs*        ?