ó ,µ[c@s°dZddlZddlZddlmZmZmZddddgZd„Z ed d d ƒd d „ƒZ edd dƒd d„ƒZ d„Z d„Z d„ZdS(s  ***** 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. iÿÿÿÿN(tis_string_liket open_filetmake_strt read_pajekt parse_pajektgenerate_pajekt write_pajekc csµ|jdkrd}n |j}d|jƒVt|ƒ}tt|tdt|ƒdƒƒƒ}x8|D]0}|jj|iƒj ƒ}|j ddƒ}|j ddƒ}t |j d||ƒƒ}|||<|j d d ƒ} d j t t||||| fƒƒ} x„|jƒD]v\} } t| ƒri| jƒdkri| d t| ƒt| ƒf7} qtjd | t| ƒr‡dndfƒqW| VqkW|jƒr³dVndVxö|jdtƒD]â\} } }|j ƒ}|j ddƒ}d j t t|| || |fƒƒ} x„|jƒD]v\} } t| ƒr{| jƒdkr{| d t| ƒt| ƒf7} q.tjd| t| ƒr™dndfƒq.W| VqËWdS(sðGenerate 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. ttNetworkXs *vertices %sitxgtytidtshapetellipset s %s %ss'Node attribute %s is not processed. %s.sEmpty attributesNon-string attributes*arcss*edgestdatatweightgð?s'Edge attribute %s is not processed. %s.N(tnametordertlisttdicttziptrangetlentnodestgettcopytpoptinttjointmapt make_qstrtitemsRtstriptwarningstwarnt is_directedtedgestTrue(tGRRt nodenumbertntnaR R R R tstktvtutedgedatatdtvalue((s7lib/python2.7/site-packages/networkx/readwrite/pajek.pyR!sJ    (  '#   " )# itmodetwbsUTF-8cCs;x4t|ƒD]&}|d7}|j|j|ƒƒq WdS(s°Write graph in Pajek format to path. Parameters ---------- G : graph A Networkx graph path : file or string File or filename to write. Filenames ending in .gz or .bz2 will be compressed. Examples -------- >>> 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. s N(Rtwritetencode(R'tpathtencodingtline((s7lib/python2.7/site-packages/networkx/readwrite/pajek.pyRds itrbcs ‡fd†|Dƒ}t|ƒS(sTRead 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|]}|jˆƒVqdS(N(tdecode(t.0R8(R7(s7lib/python2.7/site-packages/networkx/readwrite/pajek.pys ¢s(R(R6R7tlines((R7s7lib/python2.7/site-packages/networkx/readwrite/pajek.pyR…scs ddl}t|ƒr0t|jdƒƒ}ntg|D]}|jdƒ^q:ƒ}tjƒ}g‰x/|r›yt|ƒ}WnPnX|jƒj dƒrãy|jddƒ\}}Wnt k rÒq˜X||j dg|jt|ƒjdƒƒD]} | jdƒ^qW} Wn&tk r›|jt|ƒƒ} nX| d d !\} }ˆj|ƒ|j|ƒ||| <| |j|d g|jt|ƒjdƒƒD]} | jdƒ^q%} Wn&tk ri|jt|ƒƒ} nXt| ƒd kr‚qýn| d d !\}}|j||ƒ}|j||ƒ}i}y/| d d!}|jit|d ƒd6ƒWnnXt| ddd …| ddd …ƒ}|j|ƒ|j|||qýWqm|jƒj dƒrmtj|ƒ}‡fd†t |ƒDƒ}|j!|ƒqmqmW|S(sîParse Pajek format graph from string or iterable. Parameters ---------- lines : string or iterable Data in Pajek format. Returns ------- G : NetworkX graph See Also -------- read_pajek() iÿÿÿÿNs s*networkiRs *verticessutf-8iiR iR R R is*edgess*arcss*edgeiRis*matrixc3sj|]`\}}t|jƒƒD]A\}}t|ƒdkrˆ|ˆ|it|ƒd6fVqqdS(iRN(t enumeratetsplitR(R;trowR8tcolR(tlabels(s7lib/python2.7/site-packages/networkx/readwrite/pajek.pys s ("tshlexRtiterR>trstriptnxt MultiDiGraphtnexttlowert startswithtNonet ValueErrortgraphRRRR5R:tAttributeErrortstrtappendtadd_nodeRtupdatetfloatRt MultiGrapht to_directedRRtadd_edgetDiGraphR=tadd_edges_from(R<RBR8R'tltlabelRt nodelabelstnnodestiR t splitlineR R R t extra_attrtuitviR.R-t edge_datatwtadj_list((RAs7lib/python2.7/site-packages/networkx/readwrite/pajek.pyR¦s  (    ;     )* ;  ")  cCs8t|ƒst|ƒ}nd|kr4d|}n|S(sbReturn the string representation of t. Add outer double-quotes if the string has a space. Rs"%s"(RRN(tt((s7lib/python2.7/site-packages/networkx/readwrite/pajek.pyRs    cCsddl}|jdƒdS(Niÿÿÿÿstest.net(tostunlink(tmoduleRe((s7lib/python2.7/site-packages/networkx/readwrite/pajek.pytteardown_modules (t__doc__R"tnetworkxREtnetworkx.utilsRRRt__all__RRRRRRh(((s7lib/python2.7/site-packages/networkx/readwrite/pajek.pyts   C  j