;i& B d Z ddlZddlZddlmZ G d de ZdS )a
altgraph.Dot - Interface to the dot language
============================================
The :py:mod:`~altgraph.Dot` module provides a simple interface to the
file format used in the
`graphviz `_
program. The module is intended to offload the most tedious part of the process
(the **dot** file generation) while transparently exposing most of its
features.
To display the graphs or to generate image files the
`graphviz `_
package needs to be installed on the system, moreover the :command:`dot` and
:command:`dotty` programs must be accesible in the program path so that they
can be ran from processes spawned within the module.
Example usage
-------------
Here is a typical usage::
from altgraph import Graph, Dot
# create a graph
edges = [ (1,2), (1,3), (3,4), (3,5), (4,5), (5,4) ]
graph = Graph.Graph(edges)
# create a dot representation of the graph
dot = Dot.Dot(graph)
# display the graph
dot.display()
# save the dot representation into the mydot.dot file
dot.save_dot(file_name='mydot.dot')
# save dot file as gif image into the graph.gif file
dot.save_img(file_name='graph', file_type='gif')
Directed graph and non-directed graph
-------------------------------------
Dot class can use for both directed graph and non-directed graph
by passing ``graphtype`` parameter.
Example::
# create directed graph(default)
dot = Dot.Dot(graph, graphtype="digraph")
# create non-directed graph
dot = Dot.Dot(graph, graphtype="graph")
Customizing the output
----------------------
The graph drawing process may be customized by passing
valid :command:`dot` parameters for the nodes and edges. For a list of all
parameters see the `graphviz `_
documentation.
Example::
# customizing the way the overall graph is drawn
dot.style(size='10,10', rankdir='RL', page='5, 5' , ranksep=0.75)
# customizing node drawing
dot.node_style(1, label='BASE_NODE',shape='box', color='blue' )
dot.node_style(2, style='filled', fillcolor='red')
# customizing edge drawing
dot.edge_style(1, 2, style='dotted')
dot.edge_style(3, 5, arrowhead='dot', label='binds', labelangle='90')
dot.edge_style(4, 5, arrowsize=2, style='bold')
.. note::
dotty (invoked via :py:func:`~altgraph.Dot.display`) may not be able to
display all graphics styles. To verify the output save it to an image file
and look at it that way.
Valid attributes
----------------
- dot styles, passed via the :py:meth:`Dot.style` method::
rankdir = 'LR' (draws the graph horizontally, left to right)
ranksep = number (rank separation in inches)
- node attributes, passed via the :py:meth:`Dot.node_style` method::
style = 'filled' | 'invisible' | 'diagonals' | 'rounded'
shape = 'box' | 'ellipse' | 'circle' | 'point' | 'triangle'
- edge attributes, passed via the :py:meth:`Dot.edge_style` method::
style = 'dashed' | 'dotted' | 'solid' | 'invis' | 'bold'
arrowhead = 'box' | 'crow' | 'diamond' | 'dot' | 'inv' | 'none'
| 'tee' | 'vee'
weight = number (the larger the number the closer the nodes will be)
- valid `graphviz colors
`_
- for more details on how to control the graph drawing process see the
`graphviz reference
`_.
N)
GraphErrorc j e Zd ZdZ ddZd Zdd
Zd Zd Zd
Z d Z
d Z ddZ ddZ
dS )Dota0
A class providing a **graphviz** (dot language) representation
allowing a fine grained control over how the graph is being
displayed.
If the :command:`dot` and :command:`dotty` programs are not in the current
system path their location needs to be specified in the contructor.
NGdotdottyneatodigraphc @ |i c| _ | _ |
dv sJ |
| _ d| _ d| _ ||| c| _ | _ | _ i i c| _ | _ |||}|||fd}|d}t |D ]A} |i } n || } | -i | j | < | j | fi | | B|dD ]c}
fd ||
D D ]G}|i }n ||
| }|2|
| j vr
i | j |
< i | j |
|< | j
|
|fi | HbdS dS )z!
Initialization.
)graphr
z tmp_dot.dotz tmp_neo.dotNc , | | S N)out_nbrs)noder s @C:\PYTHON\MyICR_Workspace\venv\Lib\site-packages\altgraph/Dot.pyedgefnz Dot.__init__..edgefn s ~~d+++ c 3 $ K | ]
}|v |V d S r r ).0nseens r z Dot.__init__.. s' B B1T Q B Br )nameattrtypetemp_dottemp_neor r r nodesedgesset
node_styleadd
edge_style)selfr r r nodevisitor edgevisitorr r r r graphtyper styleheadtail edgestyler s @r __init__z Dot.__init__ s $R 49 00000 %
%
+.u($*dj "$R
DJ E #(
,
,
,
, =Euu D"# D)) #%
4 .....
A
A B B B Bt B B B A AD"*$& $/Kd$;$; , tz11/1DJt ,13
4(.'d@@i@@@ A
A
Ar c || _ dS )z+
Changes the overall style
N)r )r% r s r r) z Dot.style s r c |dk rH| | j | j d| j d| j }t j | n| | j | j d| j }t j | dS )z6
Displays the current graph via dotty
r -o N)save_dotr r r ossystemr )r% mode neato_cmdplot_cmds r displayz Dot.display s
7?? MM$- ( ( ()-T]]]DMMRI Ii MM$- ( ( ( "jjj$--8
(r c @ || j vr
i | j |< || j |< dS )zB
Modifies a node style to the dot representation.
N)r r )r% r kwargss r r" zDot.node_style s/ tz ! ! !DJt !
4r c 6 | j D ]} | j |fi | dS )z*
Modifies all node styles
N)r r" )r% r: r s r all_node_stylezDot.all_node_style s< J , ,D DOD + +F + + + + , ,r c || j vrt d| || j | vri | j | |< || j | |< dS # t $ r t d|d|d w xY w)zC
Modifies an edge style to the dot representation.
z
invalid node zinvalid edge z -> r1 N)r r r KeyError)r% r* r+ r: s r r$ zDot.edge_style s tz ! ! *$$ 899 9 G4:d+++)+
4 &%+DJt T " " " G G G *$$$ EFF F Gs /A !A/c # & K | j dk rd| j dV n0| j dk rd| j dV nt d| j t | j D ]\ }}|d|dV d V d
}d }t | j D ]A\ }}d |d
V t | D ]\ }}|||fz V |V Bt | j D ]}t | j | D ]i}| j dk r d |d|d
V n d |d|d
V t | j | | D ]\ }}|||fz V |V jdV d S )Nr
zdigraph z {
r zgraph zunsupported graphtype z="z";
z%s="%s",z];
z "z" [z" -> "z" -- "z}
)r r r sortedr itemsr r ) r% attr_name
attr_valuecpattepatt node_name node_attrr* r+ s r iterdotz Dot.iterdot sF 9 ! ! !&*iii1 1 1 1 1
Y'
!
!
!$(III/ / / / / *499 FGG G &,DIOO,=,=%>%> 7 7 !Iz ) :::6 6 6 6 6
%+4:+;+;+=+=$>$> Iyy ) + + + +)/ 0A0A)B)B
6
6% :y* 555555KKKK 4:&& D tz$ /00
9 )))04ddd;;;;; * 15ddd;;;;-3DJt4DT4J4P4P4R4R-S-S : :)Iz 9j"999999
r c * | S r )rI )r% s r __iter__z Dot.__iter__ s ||~~ r c |s#t j t dd | j }t |d 5 }| D ]}| | ddd dS # 1 swxY w Y dS )zD
Saves the current graph representation into a file
always pass a file_name
stacklevelwN)warningswarnDeprecationWarningr openrI write)r% file_namefpchunks r r2 z Dot.save_dot s
& M,.GTU V V V V
I
)S
!
! R
s -A11A58A5gifc |s t j t dd d}|dk rO| | j | j d| j d| j }t j | | j }n!| | j | j }|d|}|d |d| j d|}t j | d
S ) z5
Saves the dot file as an image file
rM rN rO outr r0 r1 .z -TN)
rR rS rT r2 r r r r3 r4 r )r% rW file_typer5 r6 r7
create_cmds r save_imgz Dot.save_img( s
M,.GTU V V V V I 7?? MM$- ( ( ()-T]]]DMMRI Ii xHH MM$- ( ( (xH (yy))4 HH II MMM I
* r )
NNNNNr r r r r
)r r )NrZ r )__name__
__module__ __qualname____doc__r- r) r8 r" r< r$ rI rK r2 r` r r r r r u s
=A =A =A =A~
" " ", , , G G G & & &P r r )rd r3 rR altgraphr objectr r r r rg sv m m\
L L L L L & L L L L L r