i .UdZddlZddlmZmZmZmZddhdhdhddhdZeeee fe d<d e d e d e d ee ee ffd Z d e dee ee fd eee fdZ deee efdee ee effd eeeeffdZdedeeeefd eeefdZy)u엔티티/씬 이미지 생성 순서 결정 — 시각적 의존성 기반. Visual dependency graph를 구축하여 엔티티 참조 이미지를 올바른 순서로 생성하고, 독립적인 엔티티는 병렬로 처리한다. 씬 이미지도 동일한 토폴로지 정렬을 사용하여 같은 배경을 공유하는 씬은 순차 처리, 독립 씬은 병렬 처리한다. N)AnyDictListSetidentitytransformation possession)) characterr )r prop)r r )locationr VISUAL_REFERENCE_RULESentities relations participantsreturncP|Dcic] }|d|d }}|Dcic]}|dt}}i}|D]&}|j|dgj|(|D]}|j|dg} t | dkr'|d} t | D]\} } | | dzdD]o} | d| d}}||vs||vr||||}}||f}||f}t j|xst j|}|sW| |vs\||j|q|Scc}wcc}w) auBuild dependency graph based on visual reference rules only. Args: entities: list of dicts with at least {"id", "entity_type"}. relations: list of dicts with {"id", "relation_family"}. participants: list of dicts with {"relation_id", "canon_id"}. Returns: {entity_id: set of entity_ids it depends on (must be generated first)} id entity_type relation_idrelation_familyNcanon_id)set setdefaultappendgetlen enumerater add)rrre entity_typesdeps rel_partsprelpartsfamilyip1p2id1id2type1type2 type_pair type_pair_revalloweds S/Users/manta/Documents/Projects/TheRoad-I1/backend/app/modules/entity_dependency.pybuild_visual_dependency_graphr4swHP#Px!AdGQ}-=$=xL#P9A BA4#%D B"$I Q}-r299!< c$i, u:> &'u%EArAEGnj>2j>Sl*c.E+C0,s2Cu"EN !& 044?+// >v0IMM#&!%&4 KE$Q Bs DD#r#c|Dchc]}|d }}t}g}|rg}t|D]?}|j|t}|j|s/|j |A|s|j t| |S|j ||j ||t|z}|r|Scc}w)a\Topological sort into batches (each batch can run in parallel). Args: entities: list of dicts with at least {"id"}. deps: dependency graph from build_visual_dependency_graph(). Returns: List of batches. Each batch is a list of entity IDs whose dependencies have all been completed in earlier batches. r)rsortedrissubsetrupdate) rr#r! remaining completedbatchesreadyeid entity_depss r3topological_sort_entitiesr?Qs-55Hq1T7HI5%I!G )$C((3.K##I. S!%  NN6), -  N uSZ  " N+6s Cstills entity_lookupc.tt|Dcic] }|t}}i}t|D]2\}} t j |j dd}g}|D]}t|tr&|j dxs|j dd} nt|tr|} nM|j | } | sa| j ddk(sv|j| |D]"} | |vr||j|| ||| <$|j d} | st|D]0\} }|j d| k(s||j| 25|Scc}w#tjtf$rg}Y.wxYw) aBuild scene dependency graph based on shared background locations. Scenes sharing the same location must be generated sequentially (for visual continuity from previous scene reference). Scenes at independent locations can be generated in parallel. Args: stills: list of still dicts (ordered by still_index). Each must have "visible_entities_json". entity_lookup: entity_id -> entity dict (with "entity_type"). Returns: {scene_index: set of scene_indices it depends on} visible_entities_jsonz[]r entity_idrr dependent_scene_id)rangerrrjsonloadsrJSONDecodeError TypeError isinstancedictstrrr )r@rAsir#location_to_latest_scene still_data visible_ids location_idsvr=entityloc_id dep_scene_idsjothers r3build_scene_dependency_graphrZzs$6;3v;5G H5GrSU5GD H/1#F+J **Z^^4KT%RSK#% A!T"eeDk;QUU;%;As#"&&s+F&**]3zA##C(#F11R 5f=>/1 $V ,# "~~&:; &v. E99T?l2HLL$/;,D KK!I$$i0 K sE0%E55FF scene_countctt|}t}g}|rg}t|D]?}|j|t}|j |s/|j |A|s|j t| |S|j ||j ||t|z}|r|S)a[Topological sort scene indices into batches for concurrent generation. Args: scene_count: total number of scenes. deps: dependency graph from build_scene_dependency_graph(). Returns: List of batches. Each batch is a list of scene indices whose dependencies have all been completed in earlier batches. )rrGr6rr7rr8)r[r#r9r:r;r<rO scene_depss r3topological_sort_scenesr^seK01I%I!G #B"ce,J""9- R $  NN6), -  N uSZ   N)__doc__rHtypingrrrrr tuplerN__annotations__listr4r?intrZr^r_r3rgs: '' ",-= >(>(>/<1UCH_-1111 #s3x- 1h## sCH} # $s)_#R7 c3h 7T#s(^+,7 #s3x-7t"" sCH} " $s)_"r_