͗j>2ddlZddlmZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZmZmZddlmZmZeGdd ZeGd d Zy) N)Path) dataclass)BinaryIOAnyListDictTupleUnionOptional) PCFVersion AttributeTypeATTRIBUTE_VALUES)read_cstring_stream write_cstringcNeZdZUdZeed<eed<eed<eeee e ffed<y) PCFElementaRepresents a particle system element in a PCF file. Each element contains attributes that define particle behavior, appearance, and other properties. Attributes: type_name_index: Index into the string dictionary for the element type. element_name: Name of this specific element instance. data_signature: 16-byte signature for data integrity. attributes: Dictionary mapping attribute names to (type, value) tuples. type_name_index element_namedata_signature attributesN) __name__ __module__ __qualname____doc__int__annotations__bytesrr rrVC:\Users\pivot\Documents\cueki\casual-preloader\Lib\site-packages\valve_parsers/pcf.pyrr s5 UE-"45566r rceZdZdZddeeefdefdZdede de dd fd Z dede fd Z d eeefddfd Z dZdedeefdZdedeefdZeddedefdZdedede fdZy )PCFFilea2Parser for Valve's PCF format. PCF files define particle systems used in Source engine games. They contain particle effects like smoke, fire, explosions, etc. Args: input_file: Path to the PCF file to parse. version: PCF format version. Defaults to "DMX_BINARY2_PCF1". Available versions are in the PCFVersion enum. Example: >>> pcf = PCFFile("particles.pcf") >>> pcf.decode() >>> print(f"Found {len(pcf.elements)} particle systems") >>> pcf.encode("modified_particles.pcf") input_fileversioncN||_g|_g|_t||_yN)r%string_dictionaryelementsrr$)selfr$r%s r!__init__zPCFFile.__init__2s$ .0*, z*r file attr_typevaluereturnNctj|std||tjk(r0t |t r|jdd}t||y|tjk(r@|D]:}|jtjtj|g|<y|jtjjk\rt|jtjtj|t|t|jdz }|D]}|j!|||y|tj"tj$tj&tj(fvr9|jtjtj|g|y|jtjtj||y)a4Write attribute data to file in the correct format. Args: file: File object to write to. attr_type: Type of the attribute (from AttributeType enum). value: The value to write. Raises: ValueError: If the attribute type is unsupported. Unsupported attribute type: asciireplaceerrorsN)rget ValueErrorrSTRING isinstancestrencoderMATRIXwritestructpackr. ELEMENT_ARRAYlen_write_attribute_dataCOLORVECTOR2VECTOR3VECTOR4)r*r,r-r.row base_typeitems r!rCzPCFFile._write_attribute_data8s ##I.;I;GH H  ,, ,%% WY ? $ &   ,, , 6;;'7';';I'FMMN  ??m99?? ? JJv{{#3#7#7 #BCJO P%ioo&:;I**4DA       ! !  ! !  ! !   JJv{{#3#7#7 #BKUK L  6;;/33I>FGr c |tjtjtjfvr;t j t j||jddS|tjk(rt|jddS|tjk(r t|S|tjk(rLt j t j||jdd}|j|S|tjk(r%t j d|jdS|tjk(r%t j d|jdS|tj k(r%t j d|jdS|tj"k(r%t j d |jd S|tj$k(r U U7 {CDDK Vs ,K'=K, output_pathc t|d5}tt|j}t ||d|j t jdt|j|jD]}|j |dz|j t jdt|j|jD]j}|j t jd|j|j |jdz|j |jl|jD]}|j t jdt|j|jjD]\}\}}|jj!|} |j t jd| |j t jd||j#||| ddd|S#1swY|SxYw)a5Write the PCF data to a file. Args: output_path: Where to save the PCF file. Returns: Self for method chaining. Raises: IOError: If the output file cannot be written. Example: >>> pcf.encode("modified_particles.pcf") wb r?r@rBr(r)rrrritemsindexrC) r*r^r,version_stringstringelement attr_namer- attr_value name_indexs r!r<zPCFFile.encodes+t $$Z>N $>"2" 5 6 JJv{{4T-C-C)DE F00 6G+,1 JJv{{4T]]);< === 6;;tW-D-DEF 7//'9: 7112)  == 6;;tS1C1C-DEF:A:L:L:R:R:T6I6 :!%!7!7!=!=i!HJJJv{{4<=JJv{{3 :;..tY K ;U))%8 9%8 s HH66Ic t|jd5}t|}|jdd}t t D]5}|j dstt |}||dk(s.||_ntd|tjd|jd d }t|D](}t|}|jj|*tjd |jd d } t| D]p}tjd|jd d } t|} |jd } t!| | | i} |j"j| r|j"D]} tjd |jd d }t|D]y}tjd|jd d } t%|jdd }|j| }|j'||}||f| j(|<{ ddd|S#1swY|SxYw)aParse the PCF file and populate string dictionary and elements. Returns: Self for method chaining. Raises: IOError: If the file cannot be read. ValueError: If the PCF format version is unsupported. Example: >>> pcf = PCFFile("particles.pcf").decode() >>> for element in pcf.elements: ... print(f"Particle system: {element.element_name}") rbr2r3r4DMX_razUnsupported PCF version: rbrrPrLrO)rrrrr N)rer$rdecodedirr startswithrfr%r8r?rTrUrYr(appendrr)rrZr)r*r,header header_strver_attrr%r]r\rj element_countrrrrkattribute_countr-rlrms r!rszPCFFile.decodes!$//4 (D(.FwyAJ O&&v.%j(;G!y^3'/  ,!#I $ 6 6 GI!%!:!:4!KJ5> 4KG&&y1 0)M)` a)` sAII3GII$namecr|jdd}|jD]}|j|k(s|cSy)aFind a particle element by its name. Args: name: Name of the element to find. Returns: The PCFElement if found, None otherwise. Example: >>> element = pcf.find_element_by_name("my_explosion_effect") >>> if element: ... print(f"Found element with {len(element.attributes)} attributes") r2r3r4N)r<r)r)r*r| name_bytesrks r!find_element_by_namezPCFFile.find_element_by_names<[[[; }}G##z1%r type_namec|jdd}g}|jD]2}|j|j}||k(s"|j |4|S)a6Get all elements of a specific type. Args: type_name: Type name to search for (e.g., 'DmeParticleSystem', 'DmeParticleOperator'). Returns: List of matching elements. Example: >>> operators = pcf.get_elements_by_type('DmeParticleOperator') r2r3r4)r<r)r(rrv)r*r type_bytesresultsrk element_types r!get_elements_by_typezPCFFile.get_elements_by_types]%%gi%@ }}G11'2I2IJLz)w'% r rkrlcp|jdd}||jvr|j|\}}|S|S)aRGet an attribute value from an element with type conversion. Args: element: The PCFElement to read from. attr_name: Name of the attribute. default: Default value if attribute not found. Returns: The attribute value, or default if not found. Example: >>> element = pcf.find_element_by_name("my_effect") >>> if element: ... radius = pcf.get_attribute_value(element, "radius", 5.0) ... color = pcf.get_attribute_value(element, "color", (255, 255, 255, 255)) r2r3r4)r<r)rkrldefault attr_bytesr-rms r!get_attribute_valuezPCFFile.get_attribute_value2sG$%%gi%@ ++ +$+$6$6z$B !Iz r c|jdd}||jvr|jj|||f|j|<y)awSet an attribute value on an element. Args: element: The PCFElement to modify. attr_name: Name of the attribute. value: New value for the attribute. attr_type: Type of the attribute (from AttributeType enum). Example: >>> from valve_parsers import AttributeType >>> element = pcf.find_element_by_name("my_effect") >>> if element: ... pcf.set_attribute_value(element, "radius", 10.0, AttributeType.FLOAT) ... pcf.set_attribute_value(element, "color", (255, 0, 0, 255), AttributeType.COLOR) r2r3r4N)r<r(rvr)r*rkrlr.r-rs r!set_attribute_valuezPCFFile.set_attribute_valueJsR$%%gi%@  T33 3  " " ) )* 5*3U);:&r )DMX_BINARY2_PCF1r')rrrrr rr;r+rrrrCrZr<rsr rrrr staticmethodrrrr r!r#r#s"+5s#3+c+ ,H,H)6,H?B,H ,H\2E2Em2Eh+%c "2+y+Z?B*1E(cd:6F,ZC.<!<.1rsU !DDDBB3  77 7& B<B< B