U Y@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 _gvhtmlsr 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)rrrZformattedAttrsZformattedChildrenr r r elementMakers   rcsi}|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)ZfaceZpurple)colorportz point-size9cs$g|]}dd|fddqS)rrZpink)rr )rZ outputLabel_EZ pointSizer r 0sztableMaker..trtable)r%)r&)rlenappend) inputLabelZ outputLabelsr r#rZinputLabelCellZoutputLabelCellsZrowsr r"r tableMakers,  r*cstjdddddiddid}|D]8}||jkr>d}d}nd }d}|j|||d |d d q&t|D]\}} | \} } } } d |}|| }d}t|fdd| D|d}|j|t |ddd|j || d||dd| d|||| ql|S)zA Produce a L{graphviz.Digraph} object from an automaton. trueZ100)ZpackZdpiZfontnameZMenlo)Z graph_attrZ node_attrZ edge_attrZboldz Menlo-BoldrZellipseZblue)ZfontameshapeZstylerzt{}Z tableportcsg|] }|qSr r )rZ outputSymboloutputAsStringr r r$\szmakeDigraph..)r z0.2Znone)ZlabelZmarginr,z{}:{}:w)Z arrowheadz{}:{}:e) graphvizZDigraphZstatesZ initialStateZnode enumerateZallTransitionsrr*r Zedge)Z automatonZ inputAsStringr.Z stateAsStringdigraphstateZ stateShapeZfontNamenZeachTransitionZinStateZ inputSymbolZoutStateZ outputSymbolsZthisTransitionr)r r&r r-r makeDigraph?sR       r4cCsxd}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)progZ descriptionZepilogfqpnz@A Fully Qualified Path name representing where to find machines.)helpz--quietz-qzsuppress outputF store_true)r7defaultactionz--dot-directoryz-dzWhere to write out .dot files.z.automat_visualize)r7r9z--image-directoryz-izWhere to write out image files.z --image-typez-tzThe image format.Zpng)r7choicesr9z--viewz-vz.View rendered graphs with default image viewercWsdS)Nr )argsr r r _printsztool.._printz ...discoveredz{}.dot)filename directoryz...wrote dot into)r>r?viewZcleanupz...wrote image intoz...wrote image and dot intoN)insertargparseArgumentParser add_argumentr/ZFORMATS parse_argsZ dot_directoryZimage_directoryquietr6Z asDigraphZsaverZ image_typeZrenderr@)Z _prognameZ_argvZ_syspathZ _findMachinesr=Z DESCRIPTIONZEPILOGZargumentParserr<ZexplicitlySaveDotr6machiner1Z deleteDotmsgr r r toolls|       rI)Z __future__rrBsysr/Z _discoverrr r rr*reprr4argvpathprintrIr r r r s$    -