o ip@sddlmZddlZddlZddlmZddlmZmZddl Z ddl m Z m Z m Z mZddlmZddlmZdd lmZmZmZd6d dZd6ddZd7ddZefd8ddZd9d d!Zeeefd:d*d+Zejdejddejee fd;d4d5Z!dS)<) annotationsNwraps)CallableIterator) AutomatonInputOutputState) findMachines)MethodicalMachine) TypeMachine InputProtocolCoresstrreturncCsd|ddS)Nz"{}""z\")formatreplacerrXC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\automat/_visualize.py_gvquotesrcCs d|S)Nz<{}>)rrrrr_gvhtmls rnamechildrenattrscOs6dddt|D}d|}dj|||dS)z? Construct a string from the HTML element description.  css(|]\}}d|tt|VqdS)z{}={}N)rrr).0keyvaluerrr s  zelementMaker..z#<{name} {attrs}>{children})rrrN)joinsorteditemsr)rrrZformattedAttrsZformattedChildrenrrr elementMakers   r( inputLabel outputLabels list[str]port_ECallable[..., str]csi}|r tt||d<dd|ddfd|d|}dd ifd d |D}d |g}|r>|d g|Rd g|RS)z> Construct an HTML table to label a state transition. colspantdfontz menlo-italic)faceZpurple)colorr,z point-size9cs(g|]}dd|fiddqS)r0r1Zpink)r3r)r Z outputLabelr-Z pointSizerr <sztableMaker..trtableN)rlenappend)r)r*r,r-r/ZinputLabelCellZoutputLabelCellsrowsrr5r tableMaker&s(    r<xCallable[[State], str]cstdfdd }|S)Ntr rrcs|ddddS)Nz>)r)r?r=rrimplJszescapify..impl)r?r rrr)r=rCrrBrescapifyIsrD automatonAutomaton[State, Input, Output] inputAsStringCallable[[Input], str]outputAsStringCallable[[Output], str] stateAsStringgraphviz.Digraphcst|}tt|}tjdddddiddid}|D]}||jur+d}d}nd }d}|j|||d |d d qt|D]H\}} | \} } } } d |}|| }d}t |fdd| D|d}|j|t |ddd|j || d||dd| d|||| qB|S)zA Produce a L{graphviz.Digraph} object from an automaton. trueZ100)packZdpiZfontnameZMenlo)Z graph_attrZ node_attrZ edge_attrZboldz Menlo-Boldr$Zellipseblue)Zfontameshapestyler3zt{}Z tableportcsg|]}|qSrr)r Z outputSymbolrIrrr6{szmakeDigraph..)r,z0.2none)labelmarginrPz{}:{}:w)Z arrowheadz{}:{}:eN) rDgraphvizZDigraphZstatesZ initialStatenode enumerateZallTransitionsrr<rZedge)rErGrIrKdigraphstateZ stateShapeZfontNamenZeachTransitionZinStateZ inputSymbolZoutStateZ outputSymbolsZthisTransitionr)r,r8rrRr makeDigraphQsN      r\ _progname_argv_syspath _findMachines[Callable[[str], Iterator[tuple[str, MethodicalMachine | TypeMachine[InputProtocol, Core]]]]_printCallable[..., None]NonecCsrd}d}|dr|ddtj|||d}|jddd|jd d d d d d|jddddd|jddddd|jdddtjdd|jdddd d d||}|jo_|j p_|j|jk} |j rgdd }||j D]J\} } || d!| } | r| j d" | |jd#|| d$|j|jr|j p| } |j| _ | jd" | |j|j| d%| rd&}nd'}|| ||jqld(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)ridefaultactionz--dot-directoryz-dzWhere to write out .dot files.z.automat_visualize)rirkz--image-directoryz-izWhere to write out image files.z --image-typez-tzThe image format.Zpng)richoicesrkz--viewz-vz.View rendered graphs with default image viewercWsdS)Nr)argsrrrrbsztool.._printz ...discoveredz{}.dot)filename directoryz...wrote dot into)rorpviewcleanupz...wrote image intoz...wrote image and dot intoN)insertargparseArgumentParser add_argumentrVZFORMATS parse_argsZ dot_directoryZimage_directoryquietrhZ asDigraphsaverZ image_typerenderrq)r]r^r_r`rbZ DESCRIPTIONZEPILOGZargumentParserrnZexplicitlySaveDotrhmachinerYZ deleteDotmsgrrrtools     r})rrrr)rrrrrrrr) r)rr*r+r,rr-r.rr)r=r>rr>) rErFrGrHrIrJrKr>rrL) r]rr^r+r_r+r`rarbrcrrd)" __future__rrtsys functoolsrtypingrrrVZ_corerr r r Z _discoverr Z _methodicalr Z_typedrrrrrr(r<rDreprr\argvpathprintr}rrrrs4        #  ;