U :v f!@stdZddlZddlZddlmZddddgZddZed d d dd dZeddd dddZ ddZ ddZ dS)a  ***** Pajek ***** Read graphs in Pajek format. This implementation handles directed and undirected graphs including those with self loops and parallel edges. Format ------ See http://vlado.fmf.uni-lj.si/pub/networks/pajek/doc/draweps.htm for format information. N) open_file read_pajek parse_pajekgenerate_pajek write_pajekc csV|jdkrd}n|j}d|Vt|}tt|tdt|d}|D]"}|j|i }| dd}| dd}zt | d||}Wn2t k r} z| j d 7_ W5d } ~ XYnX|||<| d d } d tt||||| f} |D]j\} } t| tr@| dkr@| d t| d t| 7} n(td| dt| tr\dnddq| VqN|rdVndV|jddD]\}} }| }| dd}d tt|||| |f} |D]j\} } t| tr| dkr| d t| d t| 7} n(td| dt| tr8dnddq| Vqd S)zGenerate lines in Pajek graph format. Parameters ---------- G : graph A Networkx graph References ---------- See http://vlado.fmf.uni-lj.si/pub/networks/pajek/doc/draweps.htm for format information. ZNetworkXz *vertices xgyid)zSPajek format requires 'id' to be an int(). Refer to the 'Relabeling nodes' section.NshapeZellipse zNode attribute z is not processed. zEmpty attributezNon-string attribute.*arcs*edgesT)dataweightg?zEdge attribute )nameorderlistdictziprangelennodesgetcopypopint ValueErrorargsjoinmap make_qstritems isinstancestrstripwarningswarnZ is_directededges)GrrZ nodenumbernnar r r errr skvuZedgedatadvaluer5>> G = nx.path_graph(4) >>> nx.write_pajek(G, "test.net") Warnings -------- Optional node attributes and edge attributes must be non-empty strings. Otherwise it will not be written into the file. You will need to convert those attributes to strings if you want to keep them. References ---------- See http://vlado.fmf.uni-lj.si/pub/networks/pajek/doc/draweps.htm for format information.  N)rwriteencode)r+pathencodingliner5r5r6rcs rbcsfdd|D}t|S)aZRead graph in Pajek format from path. Parameters ---------- path : file or string File or filename to write. Filenames ending in .gz or .bz2 will be uncompressed. Returns ------- G : NetworkX MultiGraph or MultiDiGraph. Examples -------- >>> G = nx.path_graph(4) >>> nx.write_pajek(G, "test.net") >>> G = nx.read_pajek("test.net") To create a Graph instead of a MultiGraph use >>> G1 = nx.Graph(G) References ---------- See http://vlado.fmf.uni-lj.si/pub/networks/pajek/doc/draweps.htm for format information. c3s|]}|VqdS)Ndecode.0r?r>r5r6 szread_pajek..)r)r=r>linesr5rEr6rsc sjddl}t|tr t|d}tdd|D}t}g|rfz t|}WnYqfYnX| drz|dd\}}Wnt k rYn X||j d<q>| d ri}|\}}t t |D]}t|}z"d d|t|d D} Wn$tk r(|t|} YnX| dd \} }|||||| <| |j|d <z6| d d\} } } |j|t| t| | dWn YnXt| ddd | ddd }|j||qq>| ds| dr,| drt|}| dr,|}|D]}z"dd|t|d D} Wn$tk rz|t|} YnXt| d krq0| dd \}}|||}|||}i}z&| d d}|dt|diWn YnXt| ddd | ddd }|||j||f|q0q>| dr>t|}fddt|D}||q>|S)zParse Pajek format graph from string or iterable. Parameters ---------- lines : string or iterable Data in Pajek format. Returns ------- G : NetworkX graph See Also -------- read_pajek rNr:cSsg|]}|dqS)r:)rstriprCr5r5r6 szparse_pajek..z*networkrrz *verticescSsg|]}|dqSutf-8rArDr r5r5r6rIsrKr )r r r rrz*edgecSsg|]}|dqSrJrArLr5r5r6rIsrz*matrixc3sN|]F\}}t|D]0\}}t|dkr||dt|ifVqqdS)rrN) enumeratesplitr)rDrowr?colrlabelsr5r6rF s  zparse_pajek..)shlexr%r&iterrSnxZ MultiDiGraphnextlower startswithrgraphrrr<AttributeErrorappendadd_noderupdatefloatrZ MultiGraphZ to_directedrrZadd_edgeZDiGraphrRZadd_edges_from)rGrXr+llabelrZ nodelabelsZnnodesiZ splitliner r r r Z extra_attruivir2r1Z edge_datawZadj_listr5rVr6rs                   cCs*t|tst|}d|kr&d|d}|S)zcReturns the string representation of t. Add outer double-quotes if the string has a space. r ")r%r&)tr5r5r6r#s   r#)r9)r9) __doc__r(ZnetworkxrZZnetworkx.utilsr__all__rrrrr#r5r5r5r6s  J   o