j UdZddlmZmZmZmZddlmZddlm Z m Z m Z m Z e dZ GddeZGd d eZGd d eZGd deZGddeZeegefZede eeffZiZe ee eeffed<de de ddfdZdede deegeffdZde e eefde defdZeddd ede e eeffd!Zed"d#d ede efd$Zy)%z gltf_extensions.py ------------------ Extension registry for glTF import/export with scope-based handlers. Each scope has a TypedDict defining the context passed to handlers. )AnyCallable OrderedDict TypedDict)log)DictListLiteralOptional)materialtexture_source primitiveprimitive_preprocessprimitive_exportcNeZdZUdZeeefed<edeeeffed<e ed<y)MaterialContextz$Context for material scope handlers.data.parse_texturesimagesN) __name__ __module__ __qualname____doc__r strr__annotations__rr M/DATA/.local/lib/python3.12/site-packages/trimesh/exchange/gltf/extensions.pyrr0s0. sCx.S$sCx.011 Lrrc&eZdZUdZeeefed<y)TextureSourceContextz*Context for texture_source scope handlers.rN)rrrrr rrrrrrr!r!8s4 sCx.rr!cDeZdZUdZeeefed<eed<eed<eed<y)PrimitiveContextz1Context for primitive scope handlers (post-load).rr mesh_kwargs accessorsN rrrrr rrrr rrrr#r#>s#; sCx.OOrr#cDeZdZUdZeeefed<eed<eed<eed<y)PrimitivePreprocessContextz;Context for primitive_preprocess scope handlers (pre-load).rrr%viewsNr&rrrr(r(Gs"E sCx.OO Krr(cNeZdZUdZeed<eed<eed<eed<eed<e ed<y) PrimitiveExportContextz>> @register_handler("MY_extension", scope="material") ... def my_handler(context: MaterialContext) -> Optional[Dict]: ... data = context["data"] ... images = context["images"] ... return {"baseColorFactor": [1, 0, 0, 1]} funcr5c |t<|S)N)r2)r?r-r=s r decoratorz#register_handler..decorators!% % r)r2Handler)r-r=rAs`` rregister_handlerrCzs20 I %G r extensionsc |r|tvr|jdsiSdSi}|jD]0\}}|t|vr d|i|}t|||x}|||<2|jdr#t t|jdS|dk(rud|vrq|d} |jD]Y} t| ts| jD]3\} } t| tr| | vri| | <t| | | /| | | <5[|S#t$r&}t j d|d|Yd}~ d}~wwxYw)aw Process extensions dict for a given scope, calling registered handlers. Parameters ---------- extensions The "extensions" dict from a glTF element, or None. scope Handler scope to invoke. **kwargs Scope-specific arguments that will be combined with extension data into a typed context dict. Required kwargs by scope: - material: parse_textures, images - texture_source: (none) - primitive: primitive, mesh_kwargs, accessors - primitive_preprocess: primitive, accessors, views - primitive_export: mesh, name, tree, buffer_items, primitive, include_normals Returns ------- results Dict of {extension_name: result} for most scopes. For scopes ending in "_source", returns first non-None result. For "primitive" scope, automatically merges results into mesh_kwargs. _sourceNrzfailed to process extension z: rr$) r2endswithr7 Exceptionrwarningnextitervaluesr8r9r:) rDr=kwargsresultsext_namercontextresulter$ ext_resultr;r<s rhandle_extensionsrTsn> i/ 2r<<G$**,$ 9U+ +  Ht.v.G#E*84W==J$*!- ~~i D)*D11   7]+ !..*Jj$/(..0 UeT*+-+- C( C 0%8',K$ 1 + N- H KK6xj1#F G G HsD ED??E#KHR_materials_pbrSpecularGlossinessr )r=rPc 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.glossrWrHrdebug)rPrWs r_specular_glossinessr[sK3Q!:)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. rr4)get)rPs r_texture_webp_sourcer_s 6?  x ((r)rtypingrrrr constantsrtypedr r r r Scoperr!r#r(r+rBr ParseTexturesr2rr:rCrTr[intr_rrrrfs9822 Y 8i9 yY C5#: d38n,- ,. 4T#w,'' (-  d t (3u7)W:L1MDAc3h(A A  AR7zJ/htCH~6NK.$,<=)"6)8C=)>)r