File: //lib64/python3.9/__pycache__/graphlib.cpython-39.opt-2.pyc
a
�DOge% � @ s@ d dgZ dZdZG dd� d�ZG dd� de�ZG dd � d �ZdS ) �TopologicalSorter�
CycleError������c @ s e Zd ZdZdd� ZdS )� _NodeInfo��node�
npredecessors�
successorsc C s || _ d| _g | _d S �N� r )�selfr � r
� /usr/lib64/python3.9/graphlib.py�__init__
s z_NodeInfo.__init__N)�__name__�
__module__� __qualname__� __slots__r r
r
r
r r s r c @ s e Zd ZdS )r N)r r r r
r
r
r r s c @ s^ e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dS )r Nc C sH i | _ d | _d| _d| _|d urD|�� D ]\}}| j|g|�R � q(d S r
)�
_node2info� _ready_nodes� _npassedout�
_nfinished�items�add)r Zgraphr � predecessorsr
r
r r * s zTopologicalSorter.__init__c C s* | j �|� }d u r&t|� | j |< }|S �N)r �getr )r r �resultr
r
r �
_get_nodeinfo4 s z TopologicalSorter._get_nodeinfoc G sR | j d urtd��| �|�}| jt|�7 _|D ]}| �|�}|j�|� q2d S )Nz/Nodes cannot be added after a call to prepare())r �
ValueErrorr r �lenr �append)r r r �nodeinfoZpredZ pred_infor
r
r r 9 s
zTopologicalSorter.addc C sB | j d urtd��dd� | j�� D �| _ | �� }|r>td|��d S )Nz cannot prepare() more than oncec S s g | ]}|j d kr|j�qS )r )r r )�.0�ir
r
r �
<listcomp>_ s z-TopologicalSorter.prepare.<locals>.<listcomp>znodes are in a cycle)r r r �values� _find_cycler )r �cycler
r
r �prepareT s
�zTopologicalSorter.preparec C sV | j d u rtd��t| j �}| j}|D ]}t|| _q&| j �� | jt|�7 _|S �N� prepare() must be called first) r r �tupler � _NODE_OUTr �clearr r )r r �n2ir r
r
r � get_readyj s
zTopologicalSorter.get_readyc C s( | j d u rtd��| j| jk p&t| j �S r* )r r r r �bool�r r
r
r � is_active� s
zTopologicalSorter.is_activec C s | � � S r )r3 r2 r
r
r �__bool__� s zTopologicalSorter.__bool__c G s� | j d u rtd��| j}|D ]�}|�|� }d u rBtd|�d���|j}|tkr�|dkrjtd|�d���n|tkr�td|�d���n t|_|jD ]0}|| }| jd8 _|jdkr�| j �|� q�| j d7 _ q d S )Nr+ znode z was not added using add()r z% was not passed out (still not ready)z was already marked done� )
r r r r r r- �
_NODE_DONEr r! r )r Znodesr/ r r" �stat� successorZsuccessor_infor
r
r �done� s*
�
zTopologicalSorter.donec C s� | j }g }g }t� }i }|D ]�}||v r*q ||v rV||v r�||| d � |g S n6|�|� |�t|| j�j� t|�||< |�|� |r�z|d � }W q*W q� ty� ||� � = |� � Y q�0 q�q q*q d S )Nr )
r �setr r! �iterr �__next__r �
StopIteration�pop)r r/ �stackZitstack�seenZ node2stackir r
r
r r' � s0
z TopologicalSorter._find_cyclec c s2 | � � | �� r.| �� }|E d H | j|� qd S r )r) r3 r0 r9 )r Z
node_groupr
r
r � static_order� s
z TopologicalSorter.static_order)N)
r r r r r r r) r0 r3 r4 r9 r' rA r
r
r
r r ' s
/%N)�__all__r- r6 r r r r r
r
r
r �<module> s