3
d
                 @   sN   d Z ddgZddlZddlmZ ddlmZ eddddd
dZdd Z	dS )z
Read graphs in LEDA format.

LEDA is a C++ class library for efficient data types and algorithms.

Format
------
See http://www.algorithmic-solutions.info/leda_guide/graphs/leda_native_graph_fileformat.html

	read_leda
parse_leda    N)NetworkXError)	open_filerb)modeUTF-8c                s    fdd| D }t |}|S )a  Read graph in LEDA format from path.

    Parameters
    ----------
    path : file or string
       File or filename to read.  Filenames ending in .gz or .bz2  will be
       uncompressed.

    Returns
    -------
    G : NetworkX graph

    Examples
    --------
    G=nx.read_leda('file.leda')

    References
    ----------
    .. [1] http://www.algorithmic-solutions.info/leda_guide/graphs/leda_native_graph_fileformat.html
    c             3   s   | ]}|j  V  qd S )N)decode).0line)encoding ;/tmp/pip-build-7vycvbft/networkx/networkx/readwrite/leda.py	<genexpr>+   s    zread_leda.<locals>.<genexpr>)r   )pathr   linesGr   )r   r   r      s    c             C   sp  t | trt| jd} tdd | D } xtdD ]}t|  q4W tt| }|dkrbtj }ntj	 }tt| }i }x@td|d D ].}t| j
 jd}|dkrt|}|||< qW |jdd |j D  tt| }xt|D ]~}yt| j \}}	}
}W n8 tk
r> } ztd	|d  |W Y d
d
}~X nX |j|t| |t|	 |dd d qW |S )a|  Read graph in LEDA format from string or iterable.

    Parameters
    ----------
    lines : string or iterable
       Data in LEDA format.

    Returns
    -------
    G : NetworkX graph

    Examples
    --------
    G=nx.parse_leda(string)

    References
    ----------
    .. [1] http://www.algorithmic-solutions.info/leda_guide/graphs/leda_native_graph_fileformat.html
    
c             S   s2   g | ]*}|j d p"|j dp"|dks|jdqS )#r    )
startswithrstrip)r
   r   r   r   r   
<listcomp>H   s   zparse_leda.<locals>.<listcomp>      z|{}|  r   c             S   s   g | ]\}}|qS r   r   )r
   isr   r   r   r   _   s    z"Too few fields in LEDA.GRAPH edge N   )label)
isinstancestritersplitrangenextintnxZDiGraphZGraphr   stripZadd_nodes_fromitemsBaseExceptionr   Zadd_edge)r   r   Zdur   nnodesymbolmr   tZreversalr   er   r   r   r   0   s6    


&,)r   )
__doc____all__Znetworkxr(   Znetworkx.exceptionr   Znetworkx.utilsr   r   r   r   r   r   r   <module>
   s   
