o q+WZ@sddlmZddlZddlZddlZddlmZddZddZd d Z e fd d Z e e e fd dZ ej dej ddejeefddZdS))print_functionN) findMachinescCsd|ddS)Nz"{}""z\")formatreplacesr 4/usr/lib/python3/dist-packages/automat/_visualize.py_gvquote sr cCs d|S)Nz<{}>)rrr r r _gvhtmls r cOs6dddt|D}d|}dj|||dS)z? Construct a string from the HTML element description.  css(|]\}}d|tt|VqdS)z{}={}N)rr str).0keyvaluer r r szelementMaker..z#<{name} {attrs}>{children})nameattrschildren)joinsorteditemsr)rrrformattedAttrsformattedChildrenr r r elementMakers   rcsi}|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)facepurple)colorportz point-size9cs(g|]}dd|fiddqS)rr pink)r#r )r outputLabel_E pointSizer r 0s ztableMaker..trtable)rlenappend) inputLabel outputLabelsr$r)rinputLabelCelloutputLabelCellsrowsr r(r tableMakers,  r5cstjdddddiddid}|D]}||jurd}d}nd }d}|j|||d |d d qt|D]H\}} | \} } } } d |}|| }d}t|fdd| D|d}|j|t |ddd|j || d||dd| d|||| q6|S)zA Produce a L{graphviz.Digraph} object from an automaton. true100)packdpifontnameMenlo) graph_attr node_attr edge_attrboldz Menlo-Boldrellipseblue)fontameshapestyler#zt{} tableportcsg|]}|qSr r )r outputSymboloutputAsStringr r r+\s zmakeDigraph..)r$z0.2none)labelmarginrCz{}:{}:w) arrowheadz{}:{}:e) graphvizDigraphstates initialStatenode enumerateallTransitionsrr5r edge) automaton inputAsStringrH stateAsStringdigraphstate stateShapefontNameneachTransitioninState inputSymboloutState outputSymbolsthisTransitionr0r$r-r rGr makeDigraph?sN        rccCsrd}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)rhdefaultactionz--dot-directoryz-dzWhere to write out .dot files.z.automat_visualize)rhrjz--image-directoryz-izWhere to write out image files.z --image-typez-tzThe image format.png)rhchoicesrjz--viewz-vz.View rendered graphs with default image viewercWsdS)Nr )argsr r r _printsztool.._printz ...discoveredz{}.dot)filename directoryz...wrote dot into)rprqviewcleanupz...wrote image intoz...wrote image and dot intoN)insertargparseArgumentParser add_argumentrMFORMATS parse_args dot_directoryimage_directoryquietrg asDigraphsaver image_typerenderrr) _progname_argv_syspath _findMachinesro DESCRIPTIONEPILOGargumentParserrnexplicitlySaveDotrgmachinerX deleteDotmsgr r r toolls        r) __future__rrusysrM _discoverrr r rr5reprrcargvpathprintrr r r r s&     -