͗j"ddlZddlmZddlmZmZddlmZmZdZ e ddZ dZ d Z d Zd Zd Zd ZdZdZdZdZGddZy)N)Path)ListUnion)read_cstring_atread_fixed_stringsIDST,2 L@c|eZdZdZddeeefdefdZddZ ddd e ed edd fd Z ddd e ed edd fdZ ddZ y )MDLFilea(A (tiny) parser for Source engine MDL (model) files. Currently scoped to reading the materials[] and materialDirectories[] blocks, and rewriting materialDirectories[] in place using a append, the new strings are appended at EOF, the int32 offset table is repointed at them, and dataLength is updated. Args: mdl_path: Path to the MDL file. Accepts str or Path objects. auto_parse: Whether to automatically parse on initialization. Defaults to True for convenience. Example: >>> mdl = MDLFile("path/to/model.mdl") >>> print(mdl.materials) # ['my_texture', 'my_texture_blue'] >>> print(mdl.material_dirs) # ['models\\player\\demo\\'] >>> mdl.rewrite_material_dirs(['console\\models\\player\\demo\\']) Tmdl_path auto_parsect||_d|_d|_d|_d|_d|_g|_g|_d|_ d|_ d|_ |r|jyy)NrF) strrversionchecksumname data_length file_size materials material_dirs_material_offset_material_dir_offset_parsedparse)selfrrs VC:\Users\pivot\Documents\cueki\casual-preloader\Lib\site-packages\valve_parsers/mdl.py__init__zMDLFile.__init__1shH     !$&(*%&)*!  JJL returnct|jj}|ttdztk7rt d|jt jd|t\}|tvrt d|d|jt jd|t\}t|td}t jd|t\}t jd|t\}t jd|t\}t jd|t \}t jd|t"\} g} t%|D]E} || t&zz} t jd|| \} | j)t+|| | zGg}t%|D]<} t jd|| | dzz\}|j)t+||>||_||_||_||_t5||_| |_||_||_| |_d|_ |S)zParse the MDL header and populate materials / material_dirs. Returns: Self for method chaining. Raises: ValueError: If the file is not an MDL or is an unsupported version. r not an MDL file:  ?"    &S"* *$.!  r))update_data_lengthnew_dirsrJNcX|jtt|jj }|t t dzt k7rtd|jtjd|t\}tjd|t\}g}|D]?}|jt||j|jddzAt||k(r4t!|D]#\}} tj"d|||dzz| %|} nxt|} |D]'} |jtj$d| )tj"d|tt|tj"d|t| |r%tj"d|t&t|t|jj)|t+||_| |_t||_|rt||_yy)aReplace materialDirectories[] with new_dirs. Appends each new string at EOF. When `len(new_dirs) == dir_count`, the existing int32 offset table is overwritten in place. When the lengths differ (growing or shrinking), a fresh int32 table is appended at EOF and the header's count + offset fields are updated to point at it; the old table bytes become orphaned. Args: new_dirs: Replacement material directories. Length may differ from the existing materialDirectories[]. update_data_length: Whether to update the dataLength header field to match the new file size. Defaults to True. Raises: ValueError: If the file is no longer a valid MDL. r r,r-asciiN)_ensure_parsed bytearrayrrr.r/r0r1r2r3r;r<r?r@extendencode enumerate pack_intopackr8 write_byteslistr!r#rr) r&rKrJrArDrE new_offsetssrFnew_offnew_dir_offsets r'rewrite_material_dirszMDLFile.rewrite_material_dirsts. T]]+6689 ww{ #w .0@A A))$5LM **46NO !# A   s3x ( JJqxx(72 3 x=I %' 4 7  sJQ,>H5'N XN& 6;;tW56'   T3(?X O   T3(@. Q    T3(8#c( C T]]'',!(^$2!S "3xD  r) new_materialscj|jtt|jj }|t t dzt k7rtd|jtjd|t\}tjd|t\}t||k7rtdt|d|t|D]Z\}}||tzz}t|} |j|j!ddztj"d||| |z \|r%tj"d|t$t|t|jj'|t)||_t||_|rt||_yy) a.Replace per-material name strings in place. Appends each new string at EOF and rewrites each material entry's nameIndex (int32, relative to that entry's own start). No existing bytes are shifted, so all other offsets in the file remain valid. Useful when relocating model materials on disk: some MDLs are compiled with `$cdmaterials` blank and the full path baked into the material name itself (yielding an empty material_dir entry paired with a path-rooted name like `models/foo/bar`). Moving the on-disk folder requires re-prefixing those names too. Args: new_materials: Replacement material name strings. Must have the same length as the existing materials[]. update_data_length: Whether to update the dataLength header field to match the new file size. Defaults to True. Raises: ValueError: If new_materials length doesn't match the existing count, or if the file is no longer a valid MDL. r r,r-znew_materials has z entries but MDL has rMrNN)rOrPrrr.r/r0r1r2r3r9r:r@rSr>rQrRrTr8rVrWr rr) r&r]rJrArBrCrFrrGrZs r'rewrite_materialszMDLFile.rewrite_materialssv8 T]]+6689 ww{ #w .0@A A))$5HI **46JK  }  *$S%7$88Mi[Y !/GAt"Q)=%==I#hG JJt{{7+g5 6   T3 7Y3F G 0    T3(8#c( C T]]'',m,S "3xD  r)c>|js|jyy)N)r$r%)r&s r'rOzMDLFile._ensure_parseds|| JJLr))T)r*r)r*N)__name__ __module__ __qualname____doc__rrrboolr(r%rr\r_rOr)r'rrs& sDy!1 t 2p$( 9(s)9(! 9(  9(~$( 8(Cy8(! 8(  8(tr)r)r2pathlibrtypingrr_iorrr0r=r5r/r4r6r7r8r9r:r;r<r>rrfr)r'rjsl 3 Bm    NNr)