o h""@sdZddlZddlZddlmZgdZddZeddd dd d Zedd d ej ddddddZ ej dddddZ 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 cs>|jdkr d}n|j}d|Vt|}tt|tdt|d}|D]}|j|i }| dd}| dd}z t | d||}Wnt y_} z| j d 7_ d } ~ ww|||<| d d } d tt||||| f} |D]1\} } t| tr| dkr| d t| d t| 7} q{td| dt| trdnddq{| Vq(|rdVndV|jddD]Z\}} }| }| dd}d tt|||| |f} |D]4\} } t| tr| dkr| d t| d t| 7} qtd| dt| trdnddq| 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. NetworkXz *vertices xgyid)zSPajek format requires 'id' to be an int(). Refer to the 'Relabeling nodes' section.Nshapeellipse zNode attribute z is not processed. zEmpty attributezNon-string attribute.*arcs*edgesT)dataweightg?zEdge attribute )nameorderlistdictziprangelennodesgetcopypopint ValueErrorargsjoinmap make_qstritems isinstancestrstripwarningswarn is_directededges)Grr nodenumbernnar r r errr skvuedgedatadvaluer:l/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/networkx/readwrite/pajek.pyrsV      rr wb)modeUTF-8cCs*t|D]}|d7}|||qdS)aWrite 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.  N)rwriteencode)r.pathencodingliner:r:r;rcs rrbT)graphs returns_graphcsfdd|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.0rDrCr:r; szread_pajek..)r)rBrClinesr:rLr;rsrc sZddl}t|trt|d}tdd|D}t}g|rzt|}WnY|S| drRz |dd\}}Wn t yJYn_w||j d<nW| d ri}|\}}t t |D]|}t|}zd d|t|d D} Wnty|t|} Ynw| dd \} }|||||| <| |j|d <z| d d\} } } |j|t| t| | dWnYt| ddd | ddd }|j||qgn| ds| dr| drt|}| dr |}|D]{}zdd|t|d D} Wnty3|t|} Ynwt| d kr=q| dd \}}|||}|||}i}z| d d}|dt|diWnYt| ddd | ddd }|||j||fi|qn| drt|}fddt|D}|||s"|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?cSg|]}|dqS)r?)rstriprJr:r:r; szparse_pajek..z*networkr rz *verticescSrOutf-8rHrKr r:r:r;rQ rSr )r r r rrz*edgecSrOrRrHrTr:r:r;rQrUrz*matrixc3sP|]#\}}t|D]\}}t|dkr ||dt|ifVq qdS)rrN) enumeratesplitr )rKrowrDcolrlabelsr:r;rM s  zparse_pajek..)shlexr'r(iterr\nx MultiDiGraphnextlower startswithr!graphrr rAAttributeErrorappendadd_noderupdatefloatr MultiGraph to_directedrradd_edgeDiGraphr[add_edges_from)rNrar.llabelr nodelabelsnnodesi splitliner r r r extra_attruivir6r5 edge_datawadj_listr:r_r;rs  O               SrcCs*t|ts t|}d|vrd|d}|S)zcReturns the string representation of t. Add outer double-quotes if the string has a space. r")r'r()tr:r:r;r%s  r%)r>) __doc__r*networkxrcnetworkx.utilsr__all__rr _dispatchablerrr%r:r:r:r;s  J    o