HEX

File: //lib/python3.10/__pycache__/graphlib.cpython-310.pyc
o

E�he%�@s@ddgZdZdZGdd�d�ZGdd�de�ZGdd�d�ZdS)	�TopologicalSorter�
CycleError������c@seZdZdZdd�ZdS)�	_NodeInfo��node�
npredecessors�
successorscCs||_d|_g|_dS�N�r)�selfr�r
�/usr/lib/python3.10/graphlib.py�__init__
s
z_NodeInfo.__init__N)�__name__�
__module__�__qualname__�	__slots__rr
r
r
rrsrc@seZdZdZdS)raCSubclass of ValueError raised by TopologicalSorter.prepare if cycles
    exist in the working graph.

    If multiple cycles exist, only one undefined choice among them will be reported
    and included in the exception. The detected cycle can be accessed via the second
    element in the *args* attribute of the exception instance and consists in a list
    of nodes, such that each node is, in the graph, an immediate predecessor of the
    next node in the list. In the reported list, the first and the last node will be
    the same, to make it clear that it is cyclic.
    N)rrr�__doc__r
r
r
rrsc@sbeZdZdZddd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dS)rzFProvides functionality to topologically sort a graph of hashable nodesNcCsLi|_d|_d|_d|_|dur"|��D]\}}|j|g|�R�qdSdSr
)�
_node2info�_ready_nodes�_npassedout�
_nfinished�items�add)r�graphr�predecessorsr
r
rr*s�zTopologicalSorter.__init__cCs*|j�|�}durt|�|j|<}|S�N)r�getr)rr�resultr
r
r�
_get_nodeinfo4szTopologicalSorter._get_nodeinfocGsR|jdur	td��|�|�}|jt|�7_|D]
}|�|�}|j�|�qdS)a�Add a new node and its predecessors to the graph.

        Both the *node* and all elements in *predecessors* must be hashable.

        If called multiple times with the same node argument, the set of dependencies
        will be the union of all dependencies passed in.

        It is possible to add a node with no dependencies (*predecessors* is not provided)
        as well as provide a dependency twice. If a node that has not been provided before
        is included among *predecessors* it will be automatically added to the graph with
        no predecessors of its own.

        Raises ValueError if called after "prepare".
        Nz/Nodes cannot be added after a call to prepare())r�
ValueErrorr r�lenr	�append)rrr�nodeinfo�pred�	pred_infor
r
rr9s


�zTopologicalSorter.addcCsB|jdur	td��dd�|j��D�|_|��}|rtd|��dS)axMark the graph as finished and check for cycles in the graph.

        If any cycle is detected, "CycleError" will be raised, but "get_ready" can
        still be used to obtain as many nodes as possible until cycles block more
        progress. After a call to this function, the graph cannot be modified and
        therefore no more nodes can be added using "add".
        Nzcannot prepare() more than oncecSsg|]
}|jdkr|j�qS)r)rr)�.0�ir
r
r�
<listcomp>_sz-TopologicalSorter.prepare.<locals>.<listcomp>znodes are in a cycle)rr!r�values�_find_cycler)r�cycler
r
r�prepareTs
�
�zTopologicalSorter.preparecCsV|jdur	td��t|j�}|j}|D]}t||_q|j��|jt|�7_|S)a�Return a tuple of all the nodes that are ready.

        Initially it returns all nodes with no predecessors; once those are marked
        as processed by calling "done", further calls will return all new nodes that
        have all their predecessors already processed. Once no more progress can be made,
        empty tuples are returned.

        Raises ValueError if called without calling "prepare" previously.
        N�prepare() must be called first)	rr!�tupler�	_NODE_OUTr�clearrr")rr�n2irr
r
r�	get_readyjs



zTopologicalSorter.get_readycCs(|jdur	td��|j|jkpt|j�S)a�Return ``True`` if more progress can be made and ``False`` otherwise.

        Progress can be made if cycles do not block the resolution and either there
        are still nodes ready that haven't yet been returned by "get_ready" or the
        number of nodes marked "done" is less than the number that have been returned
        by "get_ready".

        Raises ValueError if called without calling "prepare" previously.
        Nr.)rr!rr�bool�rr
r
r�	is_active�s

zTopologicalSorter.is_activecCs|��Sr)r6r5r
r
r�__bool__�szTopologicalSorter.__bool__cGs�|jdur	td��|j}|D]b}|�|�}dur!td|�d���|j}|tkrJ|dkr4td|�d���|tkr@td|�d���Jd|�d	|����t|_|jD]}||}|jd
8_|jdkrh|j�|�qP|j	d
7_	qdS)a�Marks a set of nodes returned by "get_ready" as processed.

        This method unblocks any successor of each node in *nodes* for being returned
        in the future by a call to "get_ready".

        Raises :exec:`ValueError` if any node in *nodes* has already been marked as
        processed by a previous call to this method, if a node was not added to the
        graph by using "add" or if called without calling "prepare" previously or if
        node has not yet been returned by "get_ready".
        Nr.znode z was not added using add()rz% was not passed out (still not ready)z was already marked doneFz: unknown status �)
rr!rrrr0�
_NODE_DONEr	r#r)r�nodesr2rr$�stat�	successor�successor_infor
r
r�done�s0

�

��zTopologicalSorter.donec	Cs�|j}g}g}t�}i}|D]Y}||vrq	||vr,||vr+|||d�|gSn|�|�|�t||j�j�t|�||<|�|�|rez|d�}Wntyb||�	�=|�	�Ynw|sInqqdS)NTr)
r�setrr#�iterr	�__next__r"�
StopIteration�pop)rr2�stack�itstack�seen�node2stackirr
r
rr+�s<�



���zTopologicalSorter._find_cycleccs>�|��|��r|��}|EdH|j|�|��s	dSdS)aQReturns an iterable of nodes in a topological order.

        The particular order that is returned may depend on the specific
        order in which the items were inserted in the graph.

        Using this method does not require to call "prepare" or "done". If any
        cycle is detected, :exc:`CycleError` will be raised.
        N)r-r6r3r>)r�
node_groupr
r
r�static_order�s�	

�zTopologicalSorter.static_orderr)rrrrrr rr-r3r6r7r>r+rIr
r
r
rr's

/%N)�__all__r0r9rr!rrr
r
r
r�<module>s