File: //proc/self/root/usr/lib/python3/dist-packages/automat/__pycache__/_visualize.cpython-38.pyc
U
��Y� � @ s� d dl mZ d dlZd dlZd dlZddlmZ dd� Zdd� Zd d
� Z e fd d �Z
e e e fd
d�Z ej
d ej
dd� ejeefdd�ZdS )� )�print_functionN� )� findMachinesc C s d� | �dd��S )Nz"{}"�"z\")�format�replace��s� r
�4/usr/lib/python3/dist-packages/automat/_visualize.py�_gvquote
s r c C s
d� | �S )Nz<{}>)r r r
r
r �_gvhtml s r
c O s6 d� dd� t|�� �D ��}d� |�}dj| ||d�S )z?
Construct a string from the HTML element description.
� c s s& | ] \}}d � |tt|���V qdS )z{}={}N)r r �str)�.0�key�valuer
r
r � <genexpr> s �z elementMaker.<locals>.<genexpr>� z#<{name} {attrs}>{children}</{name}>)�name�attrs�children)�join�sorted�itemsr )r r r ZformattedAttrsZformattedChildrenr
r
r � elementMaker s
�
�r c s� i }|rt t|��|d< � d� d| dd�fd|d�|��}dd i�� �fd
d �|D �}� d |�g}|rx|�� d|�� � � d|�� S )z>
Construct an HTML table to label a state transition.
�colspan�td�fontz menlo-italic)ZfaceZpurple)�color�portz
point-size�9c s$ g | ] }� d � d|f��dd��qS )r r Zpink)r r
)r Z outputLabel��_EZ pointSizer
r �
<listcomp>0 s ����z tableMaker.<locals>.<listcomp>�tr�table)r% )r& )r �len�append)�
inputLabelZ outputLabelsr r# r ZinputLabelCellZoutputLabelCellsZrowsr
r" r �
tableMaker s, ���� � r* c s t jddd�ddiddid�}| �� 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-Boldr ZellipseZblue)Zfontame�shapeZstyler zt{}Z tableportc s g | ] }� |��qS r
r
)r Z outputSymbol��outputAsStringr
r r$ \ s �z makeDigraph.<locals>.<listcomp>)r z0.2Znone)ZlabelZmarginr, z{}:{}:w)Z arrowheadz{}:{}:e) �graphvizZDigraphZstatesZ initialStateZnode� enumerateZallTransitionsr r* r
Zedge)Z automatonZ
inputAsStringr. Z
stateAsString�digraph�stateZ
stateShapeZfontName�nZeachTransitionZinStateZ inputSymbolZoutStateZ
outputSymbolsZthisTransitionr) r r& r
r- r � makeDigraph? sR ��
�
�� �
��r4 c C sx d}d}|d r |� dd� tj| ||d�}|jddd� |jd d
d d d
d� |jddddd� |jddddd� |jdddtjdd� |jd d d d d
d� |�|�}|jo�|j p�|j|jk} |j r�d d � }||j
�D ]�\}
} ||
d!� | � � } | �r | j d"�
|
�|jd#� ||
d$|j� |jr�|j �p.| }
|j| _
| jd"�
|
�|j|j|
d%� |
�r`d&}nd'}||
||j� q�d(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.
r r )�progZ descriptionZepilog�fqpnz@A Fully Qualified Path name representing where to find machines.)�helpz--quietz-qzsuppress outputF�
store_true)r7 �default�actionz--dot-directoryz-dz Where to write out .dot files.z.automat_visualize)r7 r9 z--image-directoryz-iz Where to write out image files.z --image-typez-tzThe image format.Zpng)r7 �choicesr9 z--viewz-vz.View rendered graphs with default image viewerc W s d S )Nr
)�argsr
r
r �_print� s ztool.<locals>._printz
...discoveredz{}.dot)�filename� directoryz...wrote dot into)r> r? �viewZcleanupz...wrote image intoz...wrote image and dot intoN)�insert�argparse�ArgumentParser� add_argumentr/ ZFORMATS�
parse_argsZ
dot_directoryZimage_directory�quietr6 Z asDigraphZsaver Z
image_typeZrenderr@ )Z _prognameZ_argvZ_syspathZ
_findMachinesr= Z DESCRIPTIONZEPILOGZargumentParserr< ZexplicitlySaveDotr6 �machiner1 Z deleteDot�msgr
r
r �tooll s| �������
�
� �rI )Z
__future__r rB �sysr/ Z _discoverr r r
r r* �reprr4 �argv�path�printrI r
r
r
r �<module> s$
�
- �