dhpddlmZddlZddlZddlmZddlmZmZddl Z ddl m Z m Z m Z mZddlmZddlmZdd lmZmZmZd/d Zd/dZd0dZefd1dZd2dZeeefd3d%Zejdejddejee fd4d.Z!dS)5) annotationsNwraps)CallableIterator) AutomatonInputOutputState) findMachines)MethodicalMachine) TypeMachine InputProtocolCoresstrreturncTd|ddS)Nz"{}""z\")formatreplacers b/home/jenkins/workspace/simtester-sanitize/venv/lib/python3.11/site-packages/automat/_visualize.py_gvquoters" ==3.. / //c,d|S)Nz<{}>)rrs r_gvhtmlrs ==  rnamechildrenattrscddt|D}d|}d|||S)z? Construct a string from the HTML element description.  c 3|K|]7\}}d|tt|V8dS)z{}={}N)rrr).0keyvalues r zelementMaker..sV C sHSZZ0011rz#<{name} {attrs}>{children})rr!r )joinsorteditemsr)rr r!formattedAttrsformattedChildrens r elementMakerr/s|XX //N)) 0 7 7 2C 8  r inputLabel outputLabels list[str]port_ECallable[..., str]c i}|rtt||d<dd|dfd|d|}dd ifd |D}d |g}|r|d g|Rd g|RS) z> Construct an HTML table to label a state transition. colspantdfontz menlo-italic)facepurple)colorr3z point-size9c <g|]}dd|fidS)r8r9pink)r<)r% outputLabelr4 pointSizes r ztableMaker..<sN  4FK55955VDDDrtrtable)rlenappend) r0r1r3r4r7inputLabelCelloutputLabelCellsrowsrBs ` @r tableMakerrK&sG4 \!2!233 R  6:N333    Ns#I' Bt^ $ $ %D1 BBt/.///000 2g    rxCallable[[State], str]c>tdfd }|S)Ntr rrcj|ddddS)Nz>)r)rOrLs rimplzescapify..implJs/qtt||C((00f===r)rOr rrr)rLrSs` rescapifyrTIs8 1XX>>>>>X> Kr automatonAutomaton[State, Input, Output] inputAsStringCallable[[Input], str]outputAsStringCallable[[Output], str] stateAsStringgraphviz.Digraphc.t|}tt|}tjdddddiddi}|D]7}||jurd}d}nd }d}||||d |d 8t |D]\}} | \} } } } d |}|| }d}t|fd| D|}||t|dd| || d||d| d|||| |S)zA Produce a L{graphviz.Digraph} object from an automaton. true100)packdpifontnameMenlo) graph_attr node_attr edge_attrboldz Menlo-Boldr)ellipseblue)fontameshapestyler<zt{} tableportc&g|] }|Sr@r@)r% outputSymbolrYs rrCzmakeDigraph..{s# L L Ll^^L ) ) L L Lr)r3z0.2none)labelmarginrkz{}:{}:w) arrowheadz{}:{}:e) rTgraphvizDigraphstates initialStatenode enumerateallTransitionsrrKredge)rUrWrYr[digraphstate stateShapefontNameneachTransitioninState inputSymboloutState outputSymbolsthisTransitionr0r3rEs ` r makeDigraphrQs]++Mn--N]++M"511w'w'G !!##    I* * *J#HHJH M%      'y'?'?'A'ABBVV>8F5h a"];//   L L L Lm L L L     ^75>>%v VVV M' " "   ^T 2 2     Y%%nd;;]]8=T=TUUUU Nr _progname_argv_syspath _findMachines[Callable[[str], Iterator[tuple[str, MethodicalMachine | TypeMachine[InputProtocol, Core]]]]_printCallable[..., None]Nonecd}d}|dr|ddtj|||}|dd|d d d d d |dddd|dddd|dddtjd|dddd d ||}|jo|j p|j|jk} |j rd}||j D]\} } || d | } | rA| d! | |j"|| d#|j|jre|j p| } |j| _ | d! | |j|j| $| rd%}nd&}|| ||jd'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)rdefaultactionz--dot-directoryz-dzWhere to write out .dot files.z.automat_visualize)rrz--image-directoryz-izWhere to write out image files.z --image-typez-tzThe image format.png)rchoicesrz--viewz-vz.View rendered graphs with default image viewercdS)Nr@)argss rrztool.._prints Drz ...discoveredz{}.dot)filename directoryz...wrote dot into)rrviewcleanupz...wrote image intoz...wrote image and dot intoN)insertargparseArgumentParser add_argumentrtFORMATS parse_args dot_directoryimage_directoryquietr asDigraphsaver image_typerenderr)rrrrr DESCRIPTIONEPILOGargumentParserrexplicitlySaveDotrmachiner| deleteDotmsgs rtoolrsKF{2, KN R 4/|  -$    .$        @    $ $U + +D*   ND$8Drs"""""" %%%%%%%%222222222222######******44444444440000    $*     F-1.2,0 77777vXa[x|( "'[4[4[4[4[4[4[4r