ó žÃÒYc@sZdZddlmZddlZddlmZedƒedƒdd„ƒƒZdS(s(Functions for finding chains in a graph.iÿÿÿÿ(tisliceN(tnot_implemented_fortdirectedt multigraphc cs«dd„}d„}|||ƒ\}}tƒ}xq|D]i}|j|ƒd„|j|ddƒDƒ}x4|D],\}} t|||| |ƒƒ} | VqsWq:WdS(sÁReturn the chain decomposition of a graph. The *chain decomposition* of a graph with respect a depth-first search tree is a set of cycles or paths derived from the set of fundamental cycles of the tree in the following manner. Consider each fundamental cycle with respect to the given tree, represented as a list of edges beginning with the nontree edge oriented away from the root of the tree. For each fundamental cycle, if it overlaps with any previous fundamental cycle, just take the initial non-overlapping segment, which is a path instead of a cycle. Each cycle or path is called a *chain*. For more information, see [1]_. Parameters ---------- G : undirected graph root : node (optional) A node in the graph `G`. If specified, only the chain decomposition for the connected component containing this node will be returned. This node indicates the root of the depth-first search tree. Yields ------ chain : list A list of edges representing a chain. There is no guarantee on the orientation of the edges in each chain (for example, if a chain includes the edge joining nodes 1 and 2, the chain may include either (1, 2) or (2, 1)). Raises ------ NodeNotFound If `root` is not in the graph `G`. Notes ----- The worst-case running time of this implementation is linear in the number of nodes and number of edges [1]_. References ---------- .. [1] Jens M. Schmidt (2013). "A simple test on 2-vertex- and 2-edge-connectivity." *Information Processing Letters*, 113, 241–244. Elsevier. cSsîtjƒ}g}xÏtj|d|ƒD]¸\}}}|dkr«||krr|j|ddƒ|j|ƒqà|j|d|ƒ|j||dtƒ|j|ƒq(|dkr(|||kr(|j||dtƒq(q(W||fS(s+Builds a directed graph composed of cycles from the given graph. `G` is an undirected simple graph. `root` is a node in the graph from which the depth-first search is started. This function returns both the depth-first search cycle graph (as a :class:`~networkx.DiGraph`) and the list of nodes in depth-first preorder. The depth-first search cycle graph is a directed graph whose edges are the edges of `G` oriented toward the root if the edge is a tree edge and away from the root if the edge is a non-tree edge. If `root` is not specified, this performs a depth-first search on each connected component of `G` and returns a directed forest instead. If `root` is not in the graph, this raises :exc:`KeyError`. tsourcetforwardtparenttnontreeN( tnxtDiGraphtdfs_labeled_edgestadd_nodetNonetappendtadd_edgetFalsetTrue(tGtroottHtnodestutvtd((sp/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/algorithms/chains.pyt_dfs_cycle_forestDs %  cssRx@||krB||fV|j|ƒ||j|d}}qW||fVdS(s¿Generate the chain starting from the given nontree edge. `G` is a DFS cycle graph as constructed by :func:`_dfs_cycle_graph`. The edge (`u`, `v`) is a nontree edge that begins a chain. `visited` is a set representing the nodes in `G` that have already been visited. This function yields the edges in an initial segment of the fundamental cycle of `G` starting with the nontree edge (`u`, `v`) that includes all the edges up until the first node that appears in `visited`. The tree edges are given by the 'parent' node attribute. The `visited` set is updated to add each node in an edge yielded by this function. RN(taddR(RRRtvisited((sp/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/algorithms/chains.pyt _build_chain|s   css*|] \}}}|r||fVqdS(N((t.0RRR((sp/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/algorithms/chains.pys ¤stdataRN(R tsetRt out_edgestlist( RRRRRRRRtedgesRtchain((sp/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/algorithms/chains.pytchain_decompositions3 8    ( t__doc__t itertoolsRtnetworkxRtnetworkx.utilsRR R#(((sp/private/var/folders/w6/vb91730s7bb1k90y_rnhql1dhvdd44/T/pip-build-w4MwvS/networkx/networkx/algorithms/chains.pyt s