← Back
Editing: _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$ � - �
Save File
Cancel