U Acjp@s:ddlmZddlZddlZddlmZddlmZmZddl Z ddl m Z m Z m Z mZddlmZddlmZdd lmZmZmZd d d d d Zd d d ddZd d d d dddZefd dd dd dddZdddddZeeefdddddd d!d"Zejdejddejee fd ddd#d$d%d&d'd(Z!dS))) annotationsNwraps)CallableIterator) AutomatonInputOutputState) findMachines)MethodicalMachine) TypeMachine InputProtocolCorestr)sreturncCsd|ddS)Nz"{}""z\")formatreplacerr6/tmp/pip-unpacked-wheel-phmf5huh/automat/_visualize.py_gvquotesrcCs d|S)Nz<{}>)rrrrr_gvhtmlsr)namechildrenattrsrcOs6dddt|D}d|}dj|||dS)z? Construct a string from the HTML element description.  css&|]\}}d|tt|VqdS)z{}={}N)rrr).0keyvaluerrr szelementMaker..z#<{name} {attrs}>{children})rrr)joinsorteditemsr)rrrZformattedAttrsZformattedChildrenrrr elementMakers   r(z list[str]zCallable[..., str]) inputLabel outputLabelsport_Ercsi}|rtt||d<dd|ddfd|d|}dd ifd d |D}d |g}|rx|d|d|S)z> Construct an HTML table to label a state transition. colspantdfontz menlo-italic)faceZpurple)colorr+z point-size9cs$g|]}dd|fddqS)r.r/Zpink)r1r)r Z outputLabelr,Z pointSizerr <sztableMaker..trtable)r5)r6)rlenappend)r)r*r+r,r-ZinputLabelCellZoutputLabelCellsZrowsrr3r tableMaker&s(    r9zCallable[[State], str])xrcs tdddfdd }|S)Nr r)trcs|ddddS)Nz>)r)r;r:rrimplJszescapify..implr)r:r?rr>rescapifyIsr@zAutomaton[State, Input, Output]zCallable[[Input], str]zCallable[[Output], str]zgraphviz.Digraph) automaton inputAsStringoutputAsString stateAsStringrcst|}tt|}tjdddddiddid}|D]8}||jkrVd}d}nd }d}|j|||d |d d q>t|D]\}} | \} } } } d |}|| }d}t |fdd| D|d}|j|t |ddd|j || d||dd| d|||| q|S)zA Produce a L{graphviz.Digraph} object from an automaton. trueZ100)packZdpiZfontnameZMenlo)Z graph_attrZ node_attrZ edge_attrZboldz Menlo-Boldr$ZellipseZblue)Zfontameshapestyler1zt{}Z tableportcsg|] }|qSrr)r Z outputSymbolrCrrr4{szmakeDigraph..)r+z0.2none)labelmarginrGz{}:{}:w)Z arrowheadz{}:{}:e) r@graphvizZDigraphZstatesZ initialStatenode enumerateZallTransitionsrr9rZedge)rArBrCrDdigraphstateZ stateShapeZfontNamenZeachTransitionZinStateZ inputSymbolZoutStateZ outputSymbolsZthisTransitionr)r+r6rrIr makeDigraphQsN      rSz[Callable[[str], Iterator[tuple[str, MethodicalMachine | TypeMachine[InputProtocol, Core]]]]zCallable[(Ellipsis, None)]None) _progname_argv_syspath _findMachines_printrcCsxd}d}|dr|ddtj|||d}|jddd|jd d d d d d|jddddd|jddddd|jdddtjdd|jdddd d d||}|jo|j p|j|jk} |j rdd }||j D]\} } || d!| } | r| j d" | |jd#|| d$|j|jr|j p.| } |j| _ | jd" | |j|j| d%| r`d&}nd'}|| ||jqd(S))z/ Entry point for command line utility. zB Visualize automat.MethodicalMachines as graphviz graphs. zz You must have the graphviz tool suite installed. Please visit http://www.graphviz.org for more information. rr$)prog descriptionepilogfqpnz@A Fully Qualified Path name representing where to find machines.)helpz--quietz-qzsuppress outputF store_true)r^defaultactionz--dot-directoryz-dzWhere to write out .dot files.z.automat_visualize)r^r`z--image-directoryz-izWhere to write out image files.z --image-typez-tzThe image format.Zpng)r^choicesr`z--viewz-vz.View rendered graphs with default image viewercWsdS)Nr)argsrrrrYsztool.._printz ...discoveredz{}.dot)filename directoryz...wrote dot into)rdreviewcleanupz...wrote image intoz...wrote image and dot intoN)insertargparseArgumentParser add_argumentrMZFORMATS parse_argsZ dot_directoryZimage_directoryquietr]Z asDigraphsaverZ image_typerenderrf)rUrVrWrXrYZ DESCRIPTIONZEPILOGZargumentParserrcZexplicitlySaveDotr]machinerPZ deleteDotmsgrrrtools   rr)" __future__rrisys functoolsrtypingrrrMZ_corerr r r Z _discoverr Z _methodicalr Z_typedrrrrrr(r9r@reprrSargvpathprintrrrrrrs2    # ;