UHj\dZddlmZddlZddlZddlZddlZddlZddlm Z ejjde e e jjddlmZmZddlmZmZmZej.dZddZdd Zdd Zddd Zdd Zdd Zedk(reyy)a6Scan scene records for pointer fields that look like coordinate tables. This is intentionally broader than extract_scene_events.py. Many hits are script/object argument tables rather than confirmed map triggers, but the output is useful when comparing repeated scene records and looking for true transitions. ) annotationsN)Path)load_map_sizesread_point_table)find_cns_strings offset_to_va read_sectionszmap\d+_\d+[a-z]\.cns$c6tjd||dS)N/home/exedev/hwanse/tools/probe_scene_coordinate_candidates.pydword_atrs   dD& 1! 44c4d|cxkxrdkncxr|dzdvS)Nri>r)values rlooks_like_scene_idrs u x  BUT\l$BBrctd|D]U}||dzz}|dzt|kDr|S|jt||}|s=tj |sS|cS|S)N)rangelengetrMAP_RE fullmatch)rstringshit max_dwordsindexoffnames rnext_map_field_indexr'"stq*%EAIo 7SY   {{8D#./ F$$T*L  rc t|}t||}t|}g}|jD]\}}tj |s|dd} |j | } | s6tjd|} d} |j| | } | dkrg| dz} | dzt|kDr0t|| dz}t|| dz}|dk7s t|s_t||| |}g}td|D]}| |dzz}|dzt|kDrnpt||}d |cxkrd ksn8t|||g| \}}t|d krX|j!|t#|||t|t||d |s t#|| }||j!| |d |d|| |d:|j%d|S)Nr r rri@i`) fieldIndexfieldVa pointTableVa rawPointCount pointCountpoints0x04x)mapsceneId sceneIdHexrecordVarecordFileOffset candidatesc|d|dfS)Nr9r6r)items rz$extract_candidates..qs4 #3T%["Ar)key)r rritemsrr rr packfindrrrr'rrappendrsort)r extract_dirr#sectionsr! map_sizesrecordsmap_vafilenamer&sizeneedlesearchr"scene_idzero scan_limitr;r$r%ptr_va raw_pointsr3 record_vas rextract_candidatesrT-sT"HtX.G{+IG#MMO<) }}}T" T6*))FF+CQw1WFRx#d)#cAg.HD#'*Dqy 3H =-dGS*MJJq*- EAIo7SY&!$,"f: :%5dHf%Tt%T" Fv;?!!&+#/##>(.),Z&)&k"(   *$Xs3I  NN'$&xn"5 )(+",  S<| LLALB Nrcgd}|D]}djd|dddD}g}|dddD]:}djd|d dd D}|jd |d d |<|jd|dd|dd|ddd|ddj|d |jddj|S)N)z# Scene Coordinate CandidateszbGenerated by scanning scene records for pointer fields that decode as in-bounds coordinate tables.rVzmThese are broad candidates, not confirmed transition triggers. Many short tables are script/object arguments.rVz0| map | scene | record | fields | first points |z| --- | --- | --- | --- | --- |, c38K|]}d|dd|dyw)ir. n=r2Nr.0 candidates r zmarkdown..s1  ,'(Il,C+D E sr;r,r c36K|]}|dd|dyw)x,yNr)r\points rr^zmarkdown..s+:,ac |,sr3rYr.z: z| r6z | r8z | `0xr908xz` | z; z |rV )joinrC)rHlinesr=fieldsr3r]coordss rmarkdownrlus E  !,/3  l+BQ/ CIXX&x0!4F MMAi 56bA B  C  e Sl!3 45z"3'tF83tyy7H6I M   LL 99U rc tjt}|jdtt d|jdtt d|jdtd |jd td |jd t d |jddd|j }t|jj|j|j}|jr]|jjjdd|jjt!j"|dddzd|j$rM|j$jjdd|j$jt'|d|Dcgc]#}|j(r|d|j(k(s"|%}}t+dt-|d|ddD]@}dj/d|d dd!D}t+|d"d#|dd$|d%d&d#|Bt-|dkDrt+d't-|dz d(yycc}w))N) descriptionz--exez Hwanse2.exe)typedefaultz --extract-dir extract_fldz--outzwrite JSON output)rohelpz--md-outzwrite Markdown summaryz --max-dwords(z--map map_filterz#only print records for one map stem)destrrT)parentsexist_okFr-) ensure_asciiindentrgzutf-8)encodingr6zfound z6 scene records with coordinate-like pointer candidatesxrWc3FK|]}d|dd|dd|ddyw)rYr.rZr2z ptr=0xr0rfNrr[s rr^zmain..sH  ,'(Il,C+DE~.s3 5 s!r;r8r_z record=0xr9rfz... z more)argparseArgumentParser__doc__ add_argumentrint parse_argsrTexe read_bytesrEr#outparentmkdir write_textjsondumpsmd_outrlrtprintrrh)parserargsrHr=shownsummarys rmainrsI  $ $ 9F dD4GH dDrs # 3tH~--/6678APP , -5CEP>.D zFr