U td@@sddlZddlZddlmZddlmZddlmZddlm Z ddZ d'd d Z d(d d Z d)ddZ d*ddZddZd+ddZd,ddZd-ddZd.ddZd/d!d"Zd0d#d$Zd1d%d&ZdS)2N) copyfileobj)warn) GraphBase)named_temporary_filec Csbddl}t|dr>t|dr>z |j}Wntk r<YdSX|j|\}}|}|dkr|j|\}}|}|dkrdS|dkrd S|d kr|d dS|d ks|d kr^t|d}|}|dkrdS| }t |dkrZ|}|dkrdS| }t |dkrT|}|dkr2dS| }t |dkrPdSdSdSndSdS)aV_identify_format(filename) Tries to identify the format of the graph stored in the file with the given filename. It identifies most file formats based on the extension of the file (and not on syntactic evaluation). The only exception is the adjacency matrix format and the edge list format: the first few lines of the file are evaluated to decide between the two. @note: Internal function, should not be called directly. @param filename: the name of the file or a file object whose C{name} attribute is set. @return: the format of the file as a string. rNnamereadz.gz.pickleZpicklez.graphmlZgraphmlz)z.dimacsz.dlz.dotz.edgez.edgesz .edgelistz.gmlr z .graphmlzz.gwz.lglz.lgrz.ncolz.netz.pajekrz.picklezz.svgz.txtz.datredgesZ adjacency) os.pathhasattrr Exceptionpathsplitextloweropenreadlinestripsplitlen) filenameosrootext_Zext2flinepartsr!H/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/io/files.py_identify_format sL        r##c Ost|trt|}gd}}|D]L} | } t| dkr:q | |rFq dd| |D} || |d7}q ||dkr|j |f||} n |j |f||\} } | | j |<| S)aConstructs a graph based on an adjacency matrix from the given file. Additional positional and keyword arguments not mentioned here are passed intact to L{Graph.Adjacency}. @param f: the name of the file to be read or a file object @param sep: the string that separates the matrix elements in a row. C{None} means an arbitrary sequence of whitespace characters. @param comment_char: lines starting with this string are treated as comments. @param attribute: an edge attribute name where the edge weights are stored in the case of a weighted adjacency matrix. If C{None}, no weights are stored, values larger than 1 are considered as edge multiplicities. @return: the created graphrcSsg|] }t|qSr!)float).0xr!r!r" sz8_construct_graph_from_adjacency_file..r N) isinstancestrrrr startswithrappendcloseZ AdjacencyZ_Weighted_Adjacencyes) clsrsepZ comment_char attributeargskwdsmatrixrirrowgraphweightsr!r!r"$_construct_graph_from_adjacency_fileds$       r9FcCsDddlm}t||||\}}}}||jd<||d<||d<|S)aReads a graph from a file conforming to the DIMACS minimum-cost flow file format. For the exact definition of the format, see U{http://lpsolve.sourceforge.net/5.5/DIMACS.htm}. Restrictions compared to the official description of the format are as follows: - igraph's DIMACS reader requires only three fields in an arc definition, describing the edge's source and target node and its capacity. - Source vertices are identified by 's' in the FLOW field, target vertices are identified by 't'. - Node indices start from 1. Only a single source and target node is allowed. @param f: the name of the file or a Python file handle @param directed: whether the generated graph should be directed. @return: the generated graph. The indices of the source and target vertices are attached as graph attributes C{source} and C{target}, the edge capacities are stored in the C{capacity} edge attribute. r)Graphcapacitysourcetarget)Zigraphr:superZ Read_DIMACSr.)r/rZdirectedr:r7r<r=capr!r!r"!_construct_graph_from_dimacs_files   r@c CsTtD}t|d}tt|d|W5QRX|j||dW5QRSQRXdS)aDReads a graph from a zipped GraphML file. @param f: the name of the file @param index: if the GraphML file contains multiple graphs, specified the one that should be loaded. Graph indices start from zero, so if you want to load the first graph, specify 0 here. @return: the loaded graph objectwbrb)indexN)rrrgzipGzipFileZ Read_GraphML)r/rrCtmpfileoutfr!r!r"#_construct_graph_from_graphmlz_files  rHc Csddl}t|dr||}nzt|d}Wn~tk rlz||}Wntk rftdYnXYnRtk rz||}Wntk rtdYnXYnX||}|t ||std|j |S)zReads a graph from Python pickled format @param fname: the name of the file, a stream to read from, or a string containing the pickled data. @return: the created graph object. rNrrBzJCannot load file. If fname is a file name, that filename may be incorrect.unpickled object is not a %s) picklerloadrUnicodeDecodeErrorloads TypeErrorIOErrorr-r)__name__)r/fnamerJresultfpr!r!r"!_construct_graph_from_pickle_files0    rTcCsnddl}t|dr@t|tjr*||}qR|tjd|d}n|t|d}t||sjtd|j|S)zReads a graph from compressed Python pickled format, uncompressing it on-the-fly. @param fname: the name of the file or a stream to read from. @return: the created graph object. rNrrBmodefileobjrI) rJrr)rDrErKrrNrP)r/rQrJrRr!r!r""_construct_graph_from_picklez_files    rXc Ost|tjrt|}|dkr$t|}z|j|d}Wn(ttfk r^tdt|YnX|dkrxtdt|t ||}||f||S)aUnified reading function for graphs. This method tries to identify the format of the graph given in the first parameter and calls the corresponding reader method. The remaining arguments are passed to the reader method without any changes. @param f: the file containing the graph to be loaded @param format: the format of the file (if known in advance). C{None} means auto-detection. Possible values are: C{"ncol"} (NCOL format), C{"lgl"} (LGL format), C{"graphdb"} (GraphDB format), C{"graphml"}, C{"graphmlz"} (GraphML and gzipped GraphML format), C{"gml"} (GML format), C{"net"}, C{"pajek"} (Pajek format), C{"dimacs"} (DIMACS format), C{"edgelist"}, C{"edges"} or C{"edge"} (edge list), C{"adjacency"} (adjacency matrix), C{"dl"} (DL format used by UCINET), C{"pickle"} (Python pickled format), C{"picklez"} (gzipped Python pickled format) @raises IOError: if the file format can't be identified and none was given. Nrunknown file format: %sz$no reader method for file format: %s r)rPathLiker*r#Z_format_mappingKeyError IndexErrorrOgetattr)r/rformatr2r3readerr!r!r"_construct_graph_from_files  ra  cOsVt|trt|d}|j||}|D]$}||tt|||q$|dS)aWrites the adjacency matrix of the graph to the given file All the remaining arguments not mentioned here are passed intact to L{Graph.get_adjacency}. @param f: the name of the file to be written. @param sep: the string that separates the matrix elements in a row @param eol: the string that separates the rows of the matrix. Please note that igraph is able to read back the written adjacency matrix if and only if this is a single newline character wN)r)r*rZ get_adjacencywritejoinmapr-)r7rr0eolr2r3r4r6r!r!r"_write_graph_to_adjacency_file*s    rir;cCs|dkr2z |d}Wntk r0tdYnX|dkrdz |d}Wntk rbtdYnXt|tr||krtd|dg|}t|||||S)aGWrites the graph in DIMACS format to the given file. @param f: the name of the file to be written or a Python file handle. @param source: the source vertex ID. If C{None}, igraph will try to infer it from the C{source} graph attribute. @param target: the target vertex ID. If C{None}, igraph will try to infer it from the C{target} graph attribute. @param capacity: the capacities of the edges in a list or the name of an edge attribute that holds the capacities. If there is no such edge attribute, every edge will have a capacity of 1. Nr<zlsource vertex must be provided in the 'source' graph attribute or in the 'source' argument of write_dimacs()r=zltarget vertex must be provided in the 'target' graph attribute or in the 'target' argument of write_dimacs()z"'%s' edge attribute does not existr ) r\ ValueErrorr)r*Zedge_attributesrZecountrZ write_dimacs)r7rr<r=r;r!r!r"_write_graph_to_dimacs_file?s$     rk c CsFt6}||t|d|}tt|d||W5QRXdS)avWrites the graph to a zipped GraphML file. The library uses the gzip compression algorithm, so the resulting file can be unzipped with regular gzip uncompression (like C{gunzip} or C{zcat} from Unix command line) or the Python C{gzip} module. Uses a temporary file to store intermediate GraphML data, so make sure you have enough free space to store the unzipped GraphML file as well. @param f: the name of the file to be written. @param compresslevel: the level of compression. 1 is fastest and produces the least compression, and 9 is slowest and produces the most compression.rArBN)rZ write_graphmlrDrErrr-)r7r compresslevelrFrGr!r!r"_write_graph_to_graphmlz_filefs  rncCsXddl}|dkr|||St|ds6d}t|d}nd}||||}|rT||S)aSaves the graph in Python pickled format @param fname: the name of the file or a stream to save to. If C{None}, saves the graph to a string and returns the string. @param version: pickle protocol version to be used. If -1, uses the highest protocol available @return: C{None} if the graph was saved successfully to the given file, or a string if C{fname} was C{None}. rNreTrAF)rJdumpsrrdumpr-r7rQversionrJZfile_was_openedrRr!r!r"_write_graph_to_pickle_file}s    rtcCsdddl}d}t|ds(d}t|d}nt|tjsFd}tjd|d}||||}|r`||S)aSaves the graph in Python pickled format, compressed with gzip. Saving in this format is a bit slower than saving in a Python pickle without compression, but the final file takes up much less space on the hard drive. @param fname: the name of the file or a stream to save to. @param version: pickle protocol version to be used. If -1, uses the highest protocol available @return: C{None} if the graph was saved successfully to the given file. rNFreTrArU)rJrrDrr)rErqr-rrr!r!r"_write_graph_to_picklez_files  ruc Ost|tjrt|}|dkr$t|}z|j|d}Wn(ttfk r^tdt|YnX|dkrxtdt|t ||}||f||S)aUnified writing function for graphs. This method tries to identify the format of the graph given in the first parameter (based on extension) and calls the corresponding writer method. The remaining arguments are passed to the writer method without any changes. @param f: the file containing the graph to be saved @param format: the format of the file (if one wants to override the format determined from the filename extension, or the filename itself is a stream). C{None} means auto-detection. Possible values are: - C{"adjacency"}: adjacency matrix format - C{"dimacs"}: DIMACS format - C{"dot"}, C{"graphviz"}: GraphViz DOT format - C{"edgelist"}, C{"edges"} or C{"edge"}: numeric edge list format - C{"gml"}: GML format - C{"graphml"} and C{"graphmlz"}: standard and gzipped GraphML format - C{"gw"}, C{"leda"}, C{"lgr"}: LEDA native format - C{"lgl"}: LGL format - C{"ncol"}: NCOL format - C{"net"}, C{"pajek"}: Pajek format - C{"pickle"}, C{"picklez"}: standard and gzipped Python pickled format - C{"svg"}: SVG format @raises IOError: if the file format can't be identified and none was given. Nr rYz$no writer method for file format: %srZ)r7rr_r2r3writerr!r!r"_write_graph_to_files,  rw)Nr$N)F)r)N)N)rbrc)NNr;)rl)Nro)Nro)N)rDrshutilrwarningsrZigraph._igraphrZ igraph.utilsrr#r9r@rHrTrXrarirkrnrtrurwr!r!r!r"s0    X + "  - %  '   !