a
XC?he% @ s@ d dgZ dZdZG dd dZG dd deZG dd d ZdS ) TopologicalSorter
CycleErrorc @ s e Zd ZdZdd ZdS ) _NodeInfonode
npredecessors
successorsc C s || _ d| _g | _d S N r )selfr r
-/opt/alt/python39/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 ZdddZdd 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
_nfinisheditemsadd)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 urtddd | j D | _ | }|r>td|d S )Nz cannot prepare() more than oncec S s g | ]}|j d kr|jqS )r )r r ).0ir
r
r
_ s z-TopologicalSorter.prepare..znodes are in a cycle)r r r values _find_cycler )r cycler
r
r prepareT s
zTopologicalSorter.preparec C sV | j d u rtdt| 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 boolr 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|dn|tkrtd|dn 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" statZ 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|| jj t|||< || |rz|d }W q*W q ty || = | Y q0 qq q*q d S )Nr )
r setr r! iterr __next__r
StopIterationpop)r r/ stackZitstackseenZ node2stackir r
r
r r' s0
z TopologicalSorter._find_cyclec c s2 | | r.| }|E d H | j| qd S r )r) r3 r0 r8 )r Z
node_groupr
r
r static_order s
z TopologicalSorter.static_order)N)
r r r r r r r) r0 r3 r4 r8 r' r@ r
r
r
r r ' s
/%N)__all__r- r6 r r r r r
r
r
r s