3j3zdZddlmZddlmZmZddlmZddlm Z m Z m Z m Z eGddZ Gdd Zy ) uanu_v3.batch_dependency_matrix — track dependency + overlap checkers. Authority: task-2553+17.md §4(3,4,5) + §12 9-R.1. Provides: * dependency matrix (declared blocking edges between tracks) * expected_files overlap checker * forbidden_write_targets overlap checker * independence query (does track X completion block track Y?) Pure stdlib; never mutates existing tracked files. ) annotations) dataclassfield) combinations)DictListSequenceTupleceZdZUded<eeZded<eeZded<eeZded<eeZ ded<y ) TrackSpecstrtrack_id)default_factoryz List[str]expected_filesforbidden_write_targets depends_on own_artifactsN) __name__ __module__ __qualname____annotations__rlistrrrr@/home/jay/workspace/scripts/../anu_v3/batch_dependency_matrix.pyr r sDM %d ;NI;).t)DYD!$7J 7$T:M9:rr c\eZdZd dZd dZddZddZddZddZddZ ddZ dd Z dd Z y )BatchDependencyMatrixci|_|D]K}|j|jvrtd|j||j|j<M|jy)Nzduplicate track )_tracksr ValueError_validate_edges)selftracksts r__init__zBatchDependencyMatrix.__init__s_-/  )AzzT\\) #3AJJ>!BCC'(DLL $ ) rc|jjD]d}|jD]S}||jvrtd|jd|||jk(s a depends on (is blocked by) b.)rrr)r"idsabs rmatrixzBatchDependencyMatrix.matrix2s]4<<   cBAa3333B B  B s A#AAAcr||j|jvxr||j|jvS)z7True if neither track blocks the other (parallel-safe).rr)r"r+r,s ris_independentz$BatchDependencyMatrix.is_independent:s: T\\!_// / 4a333 rc8||j|jvS)z:Does *candidate_track* have to wait for *completed_track*?r/)r"completed_trackcandidate_tracks rblockszBatchDependencyMatrix.blocksAs$,,"?"J"JJJrcdt|j}tg}|r|jDcgc]%\}}t fd|j Dr|'}}}|s t d|jt||D]}j|||=|r|Scc}}w)z=Topological layers; tracks in the same layer run in parallel.c3&K|]}|v ywNr).0ddones r z>BatchDependencyMatrix.parallelizable_groups..Ns7QqDy7szdependency cycle detected) dictrsetitemsallrr appendsortedadd)r" remaininglayerstidr$layerr:s @rparallelizable_groupsz+BatchDependencyMatrix.parallelizable_groupsEs& "$(oo/C7!,,77E  !<== MM&- ( # cN # s*B,cg}t|jdD]j\}}tt|j|jt|j|jz}|sW|j |||fl|S)N)rrrAr=rr@)r"outr+r,shareds rexpected_files_overlapz,BatchDependencyMatrix.expected_files_overlapYs~02 q1 +DAqDLLO223dll1o4456F Aq&>*  + rcg}|jD]}|j|}t|jt|jz}|jD]}}||k(r |j|}t|jt|jz}t ||zt|jz }|sj|j |||f|S)aReal cross-track write violation = a track writing a file that another track *owns* and/or has declared no-write, where the file is NOT the writer's own artifact. Returns (writer, owner, files). A track legitimately writing its own artifact that merely appears in a sibling's forbidden list is NOT a violation (that is the sibling declaring "I won't touch this").)rr=rrrrAr@) r"rJr+writer writer_writesr,owner protectedrKs rforbidden_write_overlapz-BatchDependencyMatrix.forbidden_write_overlapds13 /A\\!_F 5 56$$:M\\ /6 Q 3 34s118  "Y.&../0JJ1f~. /  /( rcXt|jxs|jSr7)boolrLrR)r"s rhas_boundary_conflictz+BatchDependencyMatrix.has_boundary_conflicts)  ' ' ) KT-I-I-K  rc ,t|j|j|j|j Dcgc] \}}}|||dc}}}|j Dcgc] \}}}|||dc}}}dScc}}}wcc}}}w)N)r+r,rK)rNrPrK)r#r-rGrLrR)rrr-rGrLrR)r"r+r,ss rto_dictzBatchDependencyMatrix.to_dicts4<<(kkm%)%?%?%A"&!''Q1a1-' "&!=!=!?((Q1qA6(  '(s B0BN)r#zSequence[TrackSpec]returnNone)rYrZ)rYzDict[str, Dict[str, bool]])r+r r,r rYrT)r2r r3r rYrT)rYzList[List[str]])rYz List[Tuple[str, str, List[str]]])rYrT)rYzDict[str, object]) rrrr%r!r-r0r4rGrLrRrUrXrrrrrs6N  K( >  rrN)__doc__ __future__r dataclassesrr itertoolsrtypingrrr r r rrrrr`s@ #(".. ;; ;w w r