UHjZdZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZmZmZe ej%j&dZedz Zd Zdd Zdd Zdd Zdd Zd dZd!dZ d"dZd#dZd$dZd%dZ d&dZ! d'dZ"d(dZ#d(dZ$d)dZ%d)dZ&efd*dZ'd+dZ(e)dk(re(yy),zGScan save-selector script streams for selection-buffer readers/writers.) annotationsN)Counter defaultdict)Path)find_cns_strings read_sections va_to_offsetout c"|r t|dSdS)Nintvalues F/home/exedev/hwanse/tools/summarize_save_selector_selection_writers.py parse_hexrs"3ub>,,cd|dS)N0x08xrs rhex32r c{ rcd|dS)Nr02xrrs rhex8rrrcvt||}||dzt|kDrytjd||dS)N|D]}|ddk(s |cStd)Nname.dataz Hwanse2.exe has no .data section) ValueError)r&sections r data_sectionr/(s0 6?g %N 7 88rc|dz}|dvry|dz dz}|dz dz}|dk(rdnd}|dk(rd d ||d t|d d Sdd|d|dk(rdnd|dk(rdt|dd Sdt|dd S)N>rrprimaryBranchStatesecondaryBranchStater2reader0x11z[selectionBuffer[z]] == 1) operation opcodeHex stateTablemeaningwriterrrr3zactive branch-state tablezruntime slot matchzselectionBuffer[z%] = selected active branch-state slotz] = matching runtime slot)r)ropcode stream_plus_1 stream_plus_2 state_tables rstream_operationrC/s T\F ''aZ4'Mb[D(M*71*<&BXK ~!%%&7]8K7LGT   &&5;t^1I]~tM233X Y  $D$7#88QR  rc i}|D]}|jdxsg}|st|d}|-|j|t|ggggd}|jd}|r||dvr|dj ||jd}|r||dvr|dj ||jds|jd s|jd|jd |jd d } | |d vs |d j |  |D]} t| jd }| |j|t|ggggd}| jd| jd| jdd} | |dvr|dj | | jd xsgD]} | |d vs |d j |  t |} | |fS)NpathHex)leafPointerHexlabels resources fieldRecords frontierPairslabelrHresourcerI sceneMatchedfilename sceneIdHexmaprOrPrJrGsourcetargetselector)rSrTrUrK)getr setdefaultrappendsorted) selector_references leaf_streamscontextsrefpathleafcontextrLrMrecordstreampairleavess rbuild_leaf_indexreIs4!#H"7wwy!'R b" < %% "'+ "!#      U'("33 H  $ $U +77:&  (<< K ' ' 1 77> "swwz':wwz*GGJ/!ggl3F W^44'..v6=7@7$456 < %% "'+ "!#    jj*jj* :.  w/ / O $ + +D 1jj06B 7FW^44'..v6 7+70H F 8 rc i}|D]}|jd}t|ts&|j|t |gggd}|jd}|r||dvr|dj ||jd}|r||dvr|dj ||jddk(s|jd s|jd|jd |jd d }||d vs|d j |t |}t|D]F\} }| d zt|kr|| d znd} t |||d<| t | nd||d<H||fS)NselectedPointer)selectedPointerHexrHrIrJrLrHrMrIkindfieldMaprOrPrQrJr rangeStartHex rangeEndHex) rV isinstancerrWrrXrY enumerater") rZr\r]rootr`rLrMrarootsindex next_roots rbuild_root_indexrss "H"7ww()$$ %% &+Dk "     U'("33 H  $ $U +77:&  (<< K ' ' 1 776?j (SWWZ-@wwz*GGJ/!ggl3F W^44'..v6576 8 E '\ t(- CJ(>E%!)$D */+'"G%(##7G  (G  Nrctj||dz }|dkry||}|dzt|kr||dznd}|||k\ryt||}d||z d|d<||z |d<|Sru)ryrzr"r{)r'rpr\rqrorrr`s rroot_context_for_vars   r *Q .E qy "G%(d3#8G  "T G  Nrc jtt}|D]}|jdxsgD]}t|jd}| ||j |jd|jd|jd|jd|jdd|S) N branchSteps streamVaHexrSrT conditionfallthroughVaHexbranchTargetHex)rSrTrrr)rlistrVrrX)frontier_brancheslookupedgebranchr's rfrontier_branch_lookuprs$/$5F! hh}-3 F6::m45Bz 2J  ((8,((8,#ZZ 4$*JJ/A$B#)::.?#@     MrcBt|}t||}t||\}} t|\} } t |} t } g}|d}|d|dzdz }t ||dD]9}tjd||d}t|}|s,|d|z|dz }|dz d z}|d z d z}| |d ||d fxxd z cc<|tk7rjt|||dz}t|t||d |dt|t|||d |d| t|nd||j|ndt||| |t!|| | | j|gd}|d dk(r|t||d<||vr d|||d<|j#|<|Dcgc] }|d dk(s |}}|Dcgc] }|d dk(s |}}t%| j'dDcgc]\\}}}}|t||||d}}}}}g} |D]%}|jds| j#|'| Dchc]7}|jdr$|jdxsijd9}!}|!j)d|Dcgc]*}|jdxsijd|!vr|,}"}t+d| Dd}#|"Dcgc],}|jd dk(r|#t-|dd |#kr|.}$}dd d!d"d#d$d%ttd&|d'||t/|t/|| t%|!|"t1d(|"Dt1d)|"D|$d* Scc}wcc}wcc}}}}wcc}wcc}wcc}w)+Nrawraw_sizer r!rr'r5r1rr:r<r r;r=)vaHexvalueHexr:r;streamPlus1HexselectionBufferOffsetHexselectionBufferOffsetr<r= nextDwordHex nextDwordCnsselectorContextselectorRootContextfrontierBranchesr8rzcns:branchTargetKindr>c0|dd|dd|ddfS)Nrr r)items rz!scan_operations..s%d1gaj$q'!*d1gajAr)key)r:rrr<countrrrhc3:K|]}t|ddyw)rrNr.0rows r z"scan_operations..0s;3S\2 ;s)defaultrr,dword 0x0040b4e6 0x0040b55f 0x0040b696)r90x120x13 0x000000a8)contextFieldHexinterestingOffsetHex)r. alignmentopcodeHandlersselectionBuffercontextWindowBytesc32K|]}|ddk(s dyw)r:r>r Nrrs rrz"scan_operations..Q4t3WZ[fWgksWsQ4t c32K|]}|ddk(s dyw)r:r8r Nrrs rrz"scan_operations..Rrr) scancountsByOffsetinterestingOffsetRowswriterCountFor0x20readerCountFor0x20currentFrontierRowscurrentFrontierRootHexescurrentFrontierRootRows%currentFrontierRootWriterCountFor0x20%currentFrontierRootReaderCountFor0x20+currentFrontierRootWritersBeforeFirstReader)r/rrersrrranger#r$rCINTERESTING_OFFSETr)rrrVr~rrXrYitemsdiscardminrr"sum)%r%r&rZr[rcontext_windowr.stringsrdr\rp root_contextsfrontier_lookupcountsinteresting_rowsstartendr(ropr'r@rA next_valuer writer_rows reader_rowsr:selection_offsetrBrcounts_by_offsetcurrent_frontiercurrent_root_hexescurrent_root_rowsfirst_frontier_va$current_root_writers_before_frontiers% rscan_operationsrs8$GsH-G'(;\JFH+,?@E=,->?O,3IF ENE %.7:. . 2CsA&!%""4f5a8 e $  T]V #gen 4!t+ ", ;<0@ABaGB . . c8R!V4 2Ye KK"=1(,](;%2\*)}1;1GE*-T7A7MGKK 3SW3B.Y#6r5-#P / 3 3B ;  k?h &:+A%*:%6C! "W$,01D0E*F&'$C!%F#3S3c+6F(6R3SKS"2S3c+6F(6R3SKSBH LLNAB    > 6Y(+ #(,-=(>%5%    ) 77% &  # #C () $  77( ) & ' -2223GH t$$  GG) * 0b 5 56J KOa a  ;*:; %,  77; 8 +  ) G b !$5 5 ,(, $$$ $0(,-?(@ #1  +!1!+.!+./$*+=$>#4144tBS4t1t144tBS4t1t7[1 UTS " ,s0- N;N NN:N .`s)V4#dhhuo*>)V$&r rwzleaf rG-; )rVrXjoin)rr`partsfieldss r row_contextrWsgg'(.BG E{{8 Y'(*;BQ*?!@@A{{; \CHHW[-A"1-E$FFG [[ ( .BF  Y)V6RTST:)V!VVW{{&' Ww{{+;SAAGL_D``a 99U  "s"rc |jdxsi}|sy|jdxsg}d|jddz|jdr'ddj|jdxsgznd |rd djd |dd Dznd |jd d g}djd|DxsdS)NrrrJzroot=rhrHrrrc3PK|]}t|jd ywrrrs rrz#row_root_context..nsIdS%1Irrwrc3&K|] }|s| yw)Nr)rparts rrz#row_root_context..qs4dtT4s)rVr)rr`rrs rrow_root_contextrfsgg+,2G  [[ ( .BF'++2C88=D[[=R CHHW[[28b99XZMS CHHIfRajIIIY[ ',  E 994e4 4 ;;rc|d}ddddd|dddd |dd dd |dd dd |dddd|ddd|ddd|jddddt|jdxsgddddddg}|jdxsgD]j}djd |jd!xsgDxsd"}|jd#|d$d%|d&d%|d'd%|d(d%|d%t |d) l|jds|jd*|j gd+|jd,xsgD]D}|jd#|d$d%|d&d-|d.d%|d'd%t |d%t |d) F|jd,s|jd*|d/Dcgc] }|d.d0k(s |}}|j gd1|dd2D]2}|jd#|d$d%|d'd%|d3d%t |d) 4|s|jd4n/t|d2kDr!|jd5t|d2z d6|j gd7|d8D]A}|d9d:kDr |d9tk7r|jd#|d.d%|d;d%|d<d%|d=d) C|jdd>j|Scc}w)?Nrz!# Save Selector Selection WritersrzcDword-aligned scan for script opcodes that read/write `context+0xa8` selection-buffer byte offsets.z!- opcode `0x11`: reader handler `rr9z`.z!- opcode `0x12`: writer handler `rz.- opcode `0x13`: runtime-slot writer handler `rz!- focus offset: `selectionBuffer[rrz]`.z - writer candidates for `0x20`: r.z - reader candidates for `0x20`: rz,- current selector-root writers for `0x20`: rrz>- current selector-root writers before first frontier reader: rz## Current Frontierz0| va | op | value | meaning | branch | context |z%| --- | --- | --- | --- | --- | --- |rrc3K|];}|jdd|jdd|jd=ywrSz->rT rN)rVrs rrzmarkdown..sG xx!""TXXh%7$8$((;:O9P Q sAArrz| rz | r;rr=z |z| - | - | - | - | - |)rz(## Current Selector Root 0x20 Operationsrz1| va | op | value | root context | leaf context |z| --- | --- | --- | --- | --- |rrr:rr>)rz## 0x20 Writersrz#| va | value | stream+1 | context || --- | --- | --- | --- |xrz| - | - | - | - |z| ... | ... | ... | z( additional rows omitted from markdown |)rz## Counts By Offsetrz&| operation | offset | table | count |rrr0rr<r )rVr"rrXrextendrr)summaryrlinesrbranchesrrs rmarkdownrts  6?D+ m +D1A,B6,J+K2N +D1A,B6,J+K2N 8>N9OPV9W8XXZ[ +D1B,CDZ,[+\\_` *73G+H*IK *73G+H*IK 6w{{Cjlm7n6oopq HW[[ZGNHNNLNJOIPPQ R  :/# E&{{017R 99  239r     Wc#k"2!33s:6Gs9~c(3{3/?.@ D   ;;, - ,- LL{{45;  Wc#k"2!31S5E4Fc#j/IZZ]$%SS)9(:" >  ;;0 1 ,-")*A"Bc3c+FVZbFb3cKc LL4C   Wc#j/!2#c:J6K5LCP[\_P`Oaac d    () [ C  +C ,c 3K|]N}tj|jdd|jdd|jdPywr)htmlescaperVrs rrz0html_page..table_rows..sQ# txx12"TXXh5G4H$((S^J_I`ab#sAArrzrzr;rr: r= r)rrVrXrrrr)rowsrenderedrrs r table_rowszhtml_page..table_rowss- C{{#GG$67=2#  OO![[W678![[[)9:;<![[Z9:;{{3{#3456{{3y>234j!{{#3C#89:;{{;s#3456    "yy""rrr:r>r8rc 3K|]l}|ddks |dtk(rVdtj|ddtj|ddtj|dd|d d ny w) rrzr:zrr r<r rr N)rrrrs rrzhtml_page..s   ' (D 0D9P4QUg4g {{4 ,-./[[&@!ABCD{{4 -./0G}o   sA2A4rzzzz zF z0 Save Selector Selection Writersz zzz*

Save Selector Selection Writers

zw

Dword-aligned scan for script opcodes that read/write context+0xa8 selection-buffer byte offsets.

z

Handlers: 0x11 reader 0x0040b4e6; 0x12 writer 0x0040b55f; 0x13 runtime-slot writer 0x0040b696.

z9

Focus: selectionBuffer[0x20]. Writers: rz . Readers: rz.

z$

Current selector-root writers: rrz(. Writers before first frontier reader: rz

Current Frontier

z z rz/z
vaopvaluekindmeaningbranchroot contextleaf context
No frontier rows.
z0

Current Selector Root 0x20 Operations

rz4No selector-root rows.z

0x20 Writers

i,z-No writer rows.z

0x20 Readers

z-No reader rows.z

Counts By Offset

z^ zzr)r  list[dict]returnr)rrVr")rrrrr count_rowss r html_pagerst#*#**A"Bc3c+FVZbFb3cKc")*A"Bc3c+FVZbFb3cKc ,-  J 99''' ' # ' Q ' ; ' ' i' U' X' D' '' ' ' ' 5!'" B#'$ @%'& DGL`DaCbbmnuwKoLnMMR S''( /w{{;bde/f.ghPQTU\U`U`aNUOUUSUQVPWW\ ])'* &+', f-'. Jw{{+@AGRHLAAQ R/'0 ;1'2 f3'4 Jw{{+DEKLIQIJJZ [5'6 "7'8 f9': J{4C01f5fggwx;'< "='> f?'@ J{4C01f5fggwxA'B &C'D iE'F J</0G'H I'J K'L M'' 'dcs EE E E c|jdd|dz jtj|dddzd y) NT)parentsexist_ok$save_selector_selection_writers.jsonF)r:) ensure_ascii separatorsrutf-8encoding)mkdir write_textjsondumps)rout_dirs r write_outputsr#sD MM$M. 55AA 7:FMBrc tjt}|jdtt dz |jdtt dz |jdtt dz |jd tt d z |jd tt |jd td |j}|jj}t|t|tj|jj!dtj|j"j!dtj|j$j!d|j&}t)||j*t-d|dd|dd|j*dz y)N) descriptionz--exez Hwanse2.exe)typerz--selector-referencesz#save_scene_selector_references.jsonz--leaf-streamszsave_selector_leaf_streams.jsonz--frontier-branchesz$save_selector_frontier_branches.jsonz --out-dirz--context-windowirrzwrote selection writer scan (rz writers, rz readers for 0x20) -> r)argparseArgumentParser__doc__ add_argumentrROOTOUTr parse_argsr% read_bytesrrr loadsrZ read_textr[rrr#r"print)parserargsr%rs rmainr4s  $ $ 9F dD=4HI /dCJoDop (tSCd=de -D#HnBno  $< *eD    D ((   C c 4++55w5GH 4$$...@A 4))33W3EF  G'4<<(  ( ) **W=Q5R4STllCC D Fr__main__)rz str | Noner int | None)rrrr)r%bytesr&rr'rrr6)r&rrr{)rrr dict | None)rZrr[rr!tuple[list[int], dict[int, dict]])rZrrr9) r'rrd list[int]r\dict[int, dict]r|rrr8)r'rrpr:r\r;rr8)rrrzdict[int, list[dict]])r%r7r&rrZrr[rrrrrrr{)rr{rr)rr{rr)rr{r"rrNone)rr<)*r) __future__rr'ryrr r# collectionsrrpathlibrprobe_exe_scene_tablesrrr __file__resolverr+r,rrrrr)r/rCrersr~rrrrrrrr#r4__name__rrrrDs2M" ,PP H~''* Ul-494>#>>'>B"J  "x xx$x x " x  x xv # <OdIX254 zFr
operationoffsettablecount