3 d!@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_pajekccsh|jdkrd}n|j}d|jVt|}tt|tdt|d}x0|D]&}|jj|ij }|j dd}|j dd}yt |j d||}Wn2t k r} z| j d7_ WYd d } ~ XnX|||<|j d d } d jtt||||| f} xv|jD]j\} } t| trF| jdkrF| d t| d t| 7} n(tjd| dt| trbdnddqW| VqRW|jrdVndVx|jddD]\}} }|j }|j dd}d jtt|||| |f} xv|jD]j\} } t| tr*| jdkr*| d t| d t| 7} n(tjd| dt| trFdnddqW| VqWd 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 xgyidSPajek format requires 'id' to be an int(). Refer to the 'Relabeling nodes' section.NshapeZellipse zNode attribute z is not processed. zEmpty attributezNon-string attribute.z*arcsz*edgesT)dataweightg?zEdge attribute )r )nameorderlistdictziprangelennodesgetcopypopint ValueErrorargsjoinmap make_qstritems isinstancestrstripwarningswarnZ is_directededges)GrrZ nodenumbernnar r r er skvuZedgedatadvaluer4>> 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*pathencodingliner4r4r5rcsrbcsfdd|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|]}|jVqdS)N)decode).0r>)r=r4r5 szread_pajek..)r)r<r=linesr4)r=r5rscsxddl}t|tr t|jd}tdd|D}tj}gx2|rry t|}WnPYnX|jj dry|jdd\}}Wnt k rYn X||j d<qB|jj d ri}|j\}}xt t |D]}t|}y"d d|jt|jd D} Wn$tk r,|jt|} YnX| dd \} }j||j|||| <| |j|d <y6| d d\} } } |j|jt| t| | dWn YnXt| ddd | ddd }|j|j|qWqB|jj ds|jj dr8|jj drtj|}|jj dr2|j}x:|D]}y"dd|jt|jd D} Wn$tk r|jt|} YnXt| d krq:| dd \}}|j||}|j||}i}y&| d d}|jdt|diWn YnXt| ddd | ddd }|j||j||f|q:WqB|jj drBtj|}fddt|D}|j|qBW|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() rNr9cSsg|]}|jdqS)r9)rstrip)rAr>r4r4r5 szparse_pajek..z*networkrrz *verticescSsg|]}|jdqS)zutf-8)r@)rAr r4r4r5rEszutf-8r )r r r z*edgesz*arcsz*edgecSsg|]}|jdqS)zutf-8)r@)rAr r4r4r5rEsrz*matrixc3sN|]F\}}t|jD]0\}}t|dkr||dt|ifVqqdS)rrN) enumeratesplitr)rArowr>colr)labelsr4r5rB szparse_pajek..)shlexr$r%iterrLnxZ MultiDiGraphnextlower startswithrgraphrrr;AttributeErrorappendadd_noderupdatefloatrZ MultiGraphZ to_directedrrZadd_edgeZDiGraphrKZadd_edges_from)rCrPr*llabelrZ nodelabelsZnnodesiZ splitliner r r r Z extra_attruivir1r0Z edge_datawZadj_listr4)rOr5rs     "      "       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%)tr4r4r5r"s   r")r8)r8) __doc__r'ZnetworkxrRZnetworkx.utilsr__all__rrrrr"r4r4r4r5s  J   o