:vj: UdZddlmZmZddlmZmZmZmZddl Z ddl m Z ddl mZddlmZed Zeed <Gd d eZGd deZGddeZGddeZGddeZeegefZeed<edeeeffZeed<iZeeeeeffed<dededdfdZdededeegeffdZd eededefd!Z dd"d eeefdzded#edzdefd$Z!ed%d&'d(edeeefdzfd)Z"ed*d+'d(ede#dzfd,Z$d-Z%d.Z&d/Z'ed0d1'd(eddfd2Z(ed0d3'd(ede)dzfd4Z*y)5z gltf_extensions.py ------------------ Extension registry for glTF import/export with scope-based handlers. Each scope has a TypedDict defining the context passed to handlers. )CallableIterable)AnyLiteral TypeAlias TypedDictN)log) IndexedDict)NDArray)materialtexture_source primitiveprimitive_preprocessprimitive_exportScopecNeZdZUdZeeefed<edeeeffed<e ed<y)MaterialContextz$Context for material scope handlers.data.parse_texturesimagesN) __name__ __module__ __qualname____doc__dictstrr__annotations__rlistq/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/exchange/gltf/extensions.pyrr4s0. sCx.S$sCx.011 Lr!rc&eZdZUdZeeefed<y)TextureSourceContextz*Context for texture_source scope handlers.rN)rrrrrrrrr r!r"r$r$<s4 sCx.r!r$cDeZdZUdZeeefed<eed<eed<eed<y)PrimitiveContextz1Context for primitive scope handlers (post-load).rr mesh_kwargs accessorsN rrrrrrrrrr r!r"r&r&Bs#; sCx.OOr!r&cDeZdZUdZeeefed<eed<eed<eed<y)PrimitivePreprocessContextz;Context for primitive_preprocess scope handlers (pre-load).rrr(viewsNr)r r!r"r+r+Ks"E sCx.OO Kr!r+cXeZdZUdZeed<eed<eed<eed<eed<ee e fed<y) PrimitiveExportContextzr>osOlln U eT "sf}F3KQU9V s U + F3K %r!r0scopecRtvr it<dtdtffd }|S)ax Decorator to register a handler for a glTF extension. Parameters ---------- name Extension name, e.g. "KHR_materials_pbrSpecularGlossiness". scope Handler scope, e.g. "material", "texture_source", "primitive". Returns ------- decorator Function that registers the handler and returns it unchanged. Example ------- >>> @register_handler("MY_extension", scope="material") ... def my_handler(context: MaterialContext) -> dict | None: ... data = context["data"] ... images = context["images"] ... return {"baseColorFactor": [1, 0, 0, 1]} funcr:c |t<|S)N)r7)rCr0rAs r" decoratorz#register_handler..decorators!% % r!)r7r5)r0rArEs`` r"register_handlerrFs20 I %G r! extensionscbt|tj|ijz S)aK Find extension names with no registered handler for a scope. Parameters ---------- extensions Extension names, i.e. the keys of a glTF "extensions" dict. scope Handler scope to check against. Returns ------- missing Extension names with no handler registered for the scope. )setr7getkeys)rGrAs r" unregisteredrLs' z?Y]]5"5::< <)rGrArMkwargsresultsext_namercontextresulter' ext_resultr?r@s r"handle_extensionsr^sD i/ 2r<<G$**,$ 9U+ +  Ht.v.G#E*84W==J$*!- ~~i D)*D11   7]+ !..*Jj$/(..0 UeT*+-+- C( C 0%8',K$ 1 + N1 H! 8$ KK6xj1#F G G HsD E.EE#KHR_materials_pbrSpecularGlossinessr )rArZc ddlm}|d i|d|dS#t$rtjddYy wxYw) a' Convert specular-glossiness material to PBR metallic-roughness. Parameters ---------- context MaterialContext with extension data, parse_textures function, and images. Returns ------- pbr_dict PBR metallic-roughness parameters, or None on failure. r )specular_to_pbrrr)rz%failed to convert specular-glossinessT)exc_infoNr ) visual.glossrarQr debug)rZras r"_specular_glossinessresK3Q!:)9!:!PQQ  9DIs AAEXT_texture_webprc*|djdS)z Return image source index from EXT_texture_webp. Parameters ---------- context TextureSourceContext with extension data. Returns ------- source_index Index into glTF images array, or None if not present. rr9)rJ)rZs r"_texture_webp_sourcerhs 6?  x ((r!))COLOR_0colorsCOLOR) TEXCOORD_0 tex_coord TEX_COORD)NORMALnormalsroKHR_draco_mesh_compressionrcddl}|d}|d}|djdi}|j|d|d}|djDcic]\}}|| }}}|jD]&} |j| d }||vs| d|||<(|djd } t |d d} | | | || <yyycc}}w) a Replace a primitive's placeholder accessors with decompressed draco data. The accessors of a draco-compressed primitive have no `bufferView`, so the loader filled them with zeros before calling us. All of the geometry is in a single opaque blob, and the extension carries the indirection we need to unpack it: a mapping of glTF attribute name to draco attribute id. Parameters ---------- context PrimitivePreprocessContext, whose `accessors` we mutate in-place. rNrr(r attributesr, bufferView unique_idindicesfaces)DracoPyrJdecoder<rugetattr) rZrzrr(rudecodedr0identnamesattrrxrys r" draco_decoder@s 6?D $I%)),;JnnWW-d<.@ABG-1,>,D,D,F G,F[T5UD[,FE G""yyk*+ : *.v,Ij& '# k"&&y1G GWd +Eu0" ' 1 Hs Crc tddl}ddlm}|d}|d}|d}gi}}tD]}\}} } |j |j |} | *|j || f| j tjk(r| n| jtj|| <||d} ||d } |jd| | d ttd |}|j|}t|j t| k7s!t|j"t| k7rt%j&d y|j(Dcic] }|d |d }}dg|D cic]\}} ||t+|j,| !}}} ||d||d|j/did<y cc}wcc} }w)a[ Compress a primitive's geometry into a single draco buffer. Every array in `arrays` is absorbed, so the exporter left their accessors with no `bufferView` rather than storing the same data twice. Returning `None` makes it store them after all, so a failure here exports the primitive uncompressed rather than pointing at accessors full of zeros. Parameters ---------- context PrimitiveExportContext, whose `primitive` and `buffer_items` we mutate. Returns ------- compressed True if the geometry is now inside a draco buffer, None if not. rN)_buffer_appendrrur3POSITIONrxT)pointsrypreserve_orderquantization_bitscompression_levelz:draco round trip changed the vertex count, not compressingattribute_typerw)rrr2)rvrurGrsr )rzr_draco_optionalrJappenddtypenpuint8astypefloat64encode_DRACO_QUANTIZATION_DRACO_COMPRESSIONr{lenrryr rSrur| AttributeType setdefault)rZrzrrrur3absorboptionalr0keywordkindr@positionrxbuffercheckridsabsorbeds r" draco_encoderhs(  $I<(JX F2HF.gt :>>$/0 =  tTl#%*[[BHH%rFrIrLr^rer4rhrrrrboolrr r!r"rsO/55$Yy8i9 y Y "seSj))$Cc3h$78 y8,. 4T#w,'' (-  d t (3u7)W:L1MD=Xc]=5=S=. FS#X%F F $J F  F\7zJ/d38nt6KK.$,<=)"6)3:)>)&.6LM$#4$#$#N$#N.6HIP0PTD[PJPr!