TidZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z ddlm Z  dd lmZn"#e$rZejeZYdZ[ndZ[wwxYwd ZGd d ejZGd deZGddeZGddeZdde e ejdefdZdde e ejfdZ ddefdZdS)z? material.py ------------- Store visual materials as objects. N) exceptionsutil)tol)NDArrayOptionalcolorImagegh㈵>ceZdZdZejdZeejdZedZ e j dZ dZ dS)Materialc tdNzmust be subclassed!NotImplementedError)selfargskwargss Q/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/visual/material.py__init__zMaterial.__init__s!"7888c tdrrrs r__hash__zMaterial.__hash__s!"7888rcdSz The "average" color of this material. Returns --------- color : (4,) uint8 Average color of this material. Nrs r main_colorzMaterial.main_color#rc4t|dr|jSdS)N_name material_0)hasattrr#rs rnamez Material.name/s! 4 ! ! : |rc| ||_dSdSN)r#rvalues rr&z Material.name5s  DJJJ  rc*tj|Sr()copydeepcopyrs rr,z Material.copy:s}T"""rN) __name__ __module__ __qualname__rabcabstractmethodrpropertyr r&setterr,rrrrrs999 999  X X  [[#####rrceZdZdZ d dZdZd dZdZedZ edZ e j d Z d Z dS) SimpleMaterialz& Hold a single image texture. Nc ||_||_tj||_tj||_tj||_||_||_dSr() imager&r to_rgbaambientdiffusespecular glossinessr)rr8r;r:r<r=r&rs rrzSimpleMaterial.__init__Cs[  }W-- }W--  h// % rc6tj||jSr()r uv_to_colorr8)ruvs rto_colorzSimpleMaterial.to_color[s TZ000rctj|jdd}tj|jdd}tj|jdd}||j}d|dj|dj|dj|d|jdg}i}|j|jj}|d }|d | }| d |tj } |j | | | d | ||<d|d||d<||fS)a Convert the current material to an OBJ format material. Parameters ----------- name : str or None Name to apply to the material Returns ----------- tex_name : str Name of material mtl_name : str Name of mtl file in files files : dict Data as {file name : bytes} Nznewmtl zKa {:0.8f} {:0.8f} {:0.8f}zKd {:0.8f} {:0.8f} {:0.8f}zKs {:0.8f} {:0.8f} {:0.8f}zNs z0.8fpng.zmap_Kd )fpformatr zutf-8z.mtl)r to_floatr:r;r<r&rGr=r8lowerappendrBytesIOsaveseekreadjoinencode) rr&KaKdKsmtldata image_type image_namef_objs rto_objzSimpleMaterial.to_obj^s(^DL ) )"1" - ^DL ) )"1" - ^DM * *2A2 . <9D d   / ( / 4 / ( / 4 / ( / 4 ($/ ( ( (   : !*J!"  77:#3#3#5#577J JJ--- . . .LNNE JOOuZO 8 8 8 JJqMMM$zz||D "iinn33G<<]]]Tzrcxt|jdr't|j}nd}t|jdr)|t|jz}t|jdr)|t|jz}t|jdr)|t|jz}t|jtr'|tt|jdzz}|S)z Provide a hash of the material so we can detect duplicates. Returns ------------ hash : int Hash of image and parameters tobytesri) r%r8hashr\r:r;r< isinstancer=floatint)rhasheds rrzSimpleMaterial.__hash__s 4:y ) ) $*,,..//FFF 4< + + 3 d4<//1122 2F 4< + + 3 d4<//1122 2F 4=) , , 4 d4=002233 3F dou - - 8 d3t56677 7F rc|jS)z2 Return the most prominent color. )r;rs rr zSimpleMaterial.main_colors |rc4t|dr|jSdS)N _glossiness?)r%rdrs rr=zSimpleMaterial.glossinesss" 4 ' ' $# #src6|dSt||_dSr()r_rdr)s rr=zSimpleMaterial.glossinesss = F <<rcZd|jdzz dz}t||j|jS)z Convert the current simple material to a PBR material. Returns ------------ pbr : PBRMaterial Contains material information in PBR format. rg?)roughnessFactorbaseColorTexturebaseColorFactor)r= PBRMaterialr8r;)r roughnesss rto_pbrzSimpleMaterial.to_pbrs>$/A-.I> %!Z L    r)NNNNNNr() r.r/r0__doc__rrArZrr3r r=r4rmrrrr6r6>s  01118888t:X X (((      rr6cVeZdZd dZdZdZdZdZdZe dZ d Z d Z dS) MultiMaterialNc *| g|_dS||_dS)z Wrapper for a list of Materials. Parameters ---------- materials : Optional[List[Material]] List of materials with which the container to be initialized. N materials)rrsrs rrzMultiMaterial.__init__s   DNNN&DNNNrcvd|jD}t|dkrtS|dS)z" TODO : IMPLEMENT c<g|]}t|t|Srr^rk.0ms r z(MultiMaterial.to_pbr..s'GGGQJq+,F,FGqGGGrr)rslenrk)rpbrs rrmzMultiMaterial.to_pbrs<HG$.GGG s88q==== 1v rcxttjd|jDS)z Provide a hash of the multi material so we can detect duplicates. Returns ------------ hash : int Xor hash of the contained materials. c,g|]}t|Sr)r]rws rrzz*MultiMaterial.__hash__..s)J)J)Ja$q'')J)J)Jr)r`np bitwise_xorreducersrs rrzMultiMaterial.__hash__s32>(()J)J4>)J)J)JKKLLLrc*t|jSr()iterrsrs r__iter__zMultiMaterial.__iter__DN###rc*t|jSr()nextrsrs r__next__zMultiMaterial.__next__rrc*t|jSr()r{rsrs r__len__zMultiMaterial.__len__s4>"""rcdSrrrs rr zMultiMaterial.main_color r!rc:|j|dS)z Adds new material to the container. Parameters ---------- material : Material The material to be added. N)rsrK)rmaterials raddzMultiMaterial.adds  h'''''rc|j|S)z Get material by index. Parameters ---------- idx : int Index of the material to be retrieved. Returns ------- The material on the given index. rr)ridxs rgetzMultiMaterial.gets~c""rr() r.r/r0rrmrrrrr3r rrrrrrprps ' ' ' ' M M M$$$$$$###  X  ( ( ( # # # # #rrpc eZdZdZ d$dZedZejdZedZejdZed Z e jd Z ed Z e jd Z ed Z e jdZ edZ e jdZ edZ e jdZ edZejdZedZejdZedZejdZedZejdZedZejdZedZejdZdZd Zd!Zed"Zd#ZdS)%rkz Create a material for physically based rendering as specified by GLTF 2.0: https://git.io/fhkPZ Parameters with `Texture` in them must be PIL.Image objects NFc i|_||_||_||_||_| |_||_||_||_| |_ | |_ | |_ ||_ | |_ t|dkrYtjdd|dSdS)Nrzunsupported material keys: {}z, )_dataemissiveFactorrjmetallicFactorrh alphaCutoff normalTextureemissiveTextureocclusionTexturerimetallicRoughnessTexture doubleSidedr& alphaModer{rlogdebugrGrPkeys)rr&rrrjrrhrrrirrrrrs rrzPBRMaterial.__init__8s$ -.-.&+. 0 0(@%' " v;;?? HNN/66tyy7O7OPP      ?rc6|jdS)ac The factors for the emissive color of the material. This value defines linear multipliers for the sampled texels of the emissive texture. Returns ----------- emissiveFactor : (3,) float Ech element in the array MUST be greater than or equal to 0 and less than or equal to 1. rrrrs rrzPBRMaterial.emissiveFactoriz~~.///rc||jdddStj|tjd}|t krtd||jd<dS)NrdtyperCz$all factors must be greater than 0.0) rpoprarrayfloat64reshapemin_eps ValueError)rr*emissives rrzPBRMaterial.emissiveFactorxs = JNN+T 2 2 2 2 2xRZ888@@CCH||~~%% !GHHH+3DJ' ( ( (rc6|jdS)z The material alpha rendering mode enumeration specifying the interpretation of the alpha value of the base color. Returns ----------- alphaMode : str One of 'OPAQUE', 'MASK', 'BLEND' rrrs rrzPBRMaterial.alphaModesz~~k***rc||jdddSt|}|dvrt d|||jd<dS)Nr)OPAQUEMASKBLENDzincorrect alphaMode: %s)rrstrupperstriprr)s rrzPBRMaterial.alphaModesw = JNN; - - - - -JJ$$&&,,..E777 !:EBBB&+DJ{ # # #rc6|jdS)a$ Specifies the cutoff threshold when in MASK alpha mode. If the alpha value is greater than or equal to this value then it is rendered as fully opaque, otherwise, it is rendered as fully transparent. A value greater than 1.0 will render the entire material as fully transparent. This value MUST be ignored for other alpha modes. When alphaMode is not defined, this value MUST NOT be defined. Returns ----------- alphaCutoff : float Value of cutoff. rrrs rrzPBRMaterial.alphaCutoffs z~~m,,,rcr||jdddSt||jd<dS)Nrrrr_r)s rrzPBRMaterial.alphaCutoffs; = JNN=$ / / / / /(-e DJ} % % %rc6|jdS)z Specifies whether the material is double sided. Returns ----------- doubleSided : bool Specifies whether the material is double sided. rrrs rrzPBRMaterial.doubleSidedsz~~m,,,rcr||jdddSt||jd<dS)Nr)rrboolr)s rrzPBRMaterial.doubleSideds; = JNN=$ / / / / /(,U DJ} % % %rc6|jdS)a% The factor for the metalness of the material. This value defines a linear multiplier for the sampled metalness values of the metallic-roughness texture. Returns ----------- metallicFactor : float How metally is the material rrrs rrzPBRMaterial.metallicFactorrrcr||jdddSt||jd<dS)Nrrr)s rrzPBRMaterial.metallicFactors= = JNN+T 2 2 2 2 2+0<d2>)..00F rcDt|j|j|jS)z Get a copy of the current PBR material as a simple material. Returns ------------ simple : SimpleMaterial Contains material information in a simple manner )r8r;r&)r6rirjr&rs r to_simplezPBRMaterial.to_simples,"'(    rc8tj|j}|Sr()r r9rj)rresults rr zPBRMaterial.main_colorst344 rctdd|jDS)z Provide a hash of the material so we can detect duplicate materials. Returns ------------ hash : int Hash of image and parameters rc3fK|],}|tj|V-dSr()r asanyarrayr\)rxrs r z'PBRMaterial.__hash__..sA/01= a  ((**====r)r]rPrvaluesrs rrzPBRMaterial.__hash__sQ HH48J4E4E4G4G     r) NNNNNNNNNNFNN)r.r/r0rnrr3rr4rrrrrhrjrrrrirr&r,rArr rrrrrkrk/s!%////b 0 0X 0 4 4 4 + +X + , , ,--X-"555 - -X -444 0 0X 0888 1 1X 1999 1 1X 1AAA / /X /000 1 1X 1222 2 2X 2333 2 2X 2333 : :X :$;;%$;&&X& [''[' % % %&   .X      rrkr returnc>tt|S)z Return an empty material set to a single color Parameters ----------- color : None or (3,) uint8 RGB color Returns ------------- material : SimpleMaterial Image is a a four pixel RGB r r8)r6 color_imager s rempty_materialrs  % 8 8 8 9 9 99rctttjrtSt jgdtj}t j|dvr||dt|<tj t j |d d tjS)z Generate an image with one color. Parameters ---------- color Optional uint8 color Returns ---------- image A (2, 2) RGBA image with the specified color. )drrr))rCNr)rrr) r^r rExceptionWrapperrruint8shaper{ fromarraytilerastype)r singles rrrs%455 X***"( ; ; ;F x,&&$|U| ?2761--55i@@GGQQ R RRrT paddingc n +,-./012345ddl}ddlm+ddlm4+fd1+1fd,+fd.+1fd -d /+fd 0d }i334fd }|rW|t 55fdtDt 5} n4tj t d} t| dkrdtj |fSttj| tt#t|ksJt|tksJt%dD} d} | dkrE,fd| D} | rV.fd| D} -fd| D}/fd| D}0fd| D}g}t'| | |||D]}d|D}tjd|D}|d}||krK||z }tj||z tj}|t5||| |} || |} |||}|||}|||}ng}| D]}tj|j}||krK||z }tj||z tj}|||| \}}|-|jd|jdz|dzkr | dz} |dz}nn| dkE| r|| \}}t;d|Drd}d}n||\}}t;d|Dr||\}}nd}t%d|Dr||\}}nd}tj|jtj}i2t'| | |D]\} }}!tj|jdz |z }"|!|dz z }#dtj|jz }$| D]}%||%}&|&tj |&|$ k|&d |$zkxxd zcc<tj!|&|$d |$z }&|&|"z|#z}'tDj#rItIj%||%|}(tIj%|'|})tj&|(|)d!sJ|'2|%<ǐtj 2fd"t#t|D}*| rtO||||rgd#nddd$||%|*fStQ|&|*fS)'a Pack multiple materials with texture into a single material. UV coordinates outside of the 0.0-1.0 range will be coerced into this range using a "wrap" behavior (i.e. modulus). Alpha blending and backface culling settings are not supported! Returns a material with alpha values set, but alpha blending disabled. Parameters ----------- materials : (n,) Material List of multiple materials uvs : (n, m, 2) float Original UV coordinates padding : int Number of pixels to pad each image with. max_tex_size_individual : int Maximum size of each individual texture. max_tex_size_fused : int | None Maximum size of the combined texture. Individual texture size will be reduced to fit. Set to None to allow infinite size. Returns ------------ material : SimpleMaterial Combined material. uv : (p, 2) float Combined UV coordinates in the 0.0-1.0 range. rNr r)packingc|||Stj||tj|ztj}j|S)z0 Multiply an image by a factor. Nr)convertrrrroundrrr)imgfactormoder s rmultiply_factorzpack..multiply_factor+sr >;;t$$ $ Xckk$''rz : : :V C UWW VBH   us###rcd}t|trf|j|j|jd}n|jt j|jd}tjt j |dd|ddg}tj |dz tj }|j dksJt|}||jdkr}|j}tj|}|jd kr,tj|d |jdzkdd |d <n|jd ks|j d |vrd|d <j|}nTt+|dd|j}n;tj t+|dgdkrt|j}|t}|dS)I Logic for extracting a simple image from each material. NRGBA)rrrrCrrrr).rCrrAr8r;)r^rkrirjr rIrr concatenatelinear_to_srgbrrrrrrrrwhererrgetattrr8r;r)matrc_linearc_srgbcrr rs rget_base_color_texturez$pack..get_base_color_texture8s  c; ' '& +#/%o(1D6$0!>#*=>>FFqII)(2A2,77!A#G HVc\**11"(;;w$!!nn3=G#;#;xhsmm=F**"$(3v;39N+NPSUV"W"WCKK]h..#-2Gd{{&)F %eoc** S'4 ( ( 4)CC Xgc9b11 2 2d : :ck**C ;--C{{6"""rc d}t|tr|j|jjdkr-t j|jd}nt j|j}t|jdks|jddkrq|j g|jdddR}t j |t j |dddfdzt j |dddfgd }nE|jddkr4t j |t j |dddfgd }|j _t j|d tj|j ztj|d <|j_t j|d tj|jztj|d <j|}n|j d n|j }|jd n|j}t jt jd ||gtjdz}j|dtj}|S)rNBGRRGBrr .raxis).r).r grer)NN)r^rkrrGrrrr{rrr ones_like zeros_likerrrrrrhr)rrmetallicrlmetallic_roughnesssr s rget_metallic_roughness_texturez,pack..get_metallic_roughness_texturels  c; ' '% X+7/6%??(3#?#G#G#N#NOOCC(3#?@@Csy>>Q&&#)B-1*<*<%#+8sy!}8a888C.LS"1"W66<M#c2A2g,77  CCYr]a''.#r}Sbqb\/J/J)KRTUUUC%1"$(F **2:669KK##fRX&&K&2"$(F **2:669LL##fRX&&K&eoc**"%"4"<33#BT##>CCCDW ')hHc9h7rzJJJSP''#&eo&9*&E&L&LRX&V&VWW rcd}t|tr|j|j|jd}n|j=t j|j}j|d}nBjtjgdd tj }| dS)rNr)r r r)rrr)r r rC) r^rkrrr r9rrrrrr)rrr r rs rget_emissive_texturez"pack..get_emissive_textures  c; ' ' Y".%oc&93;MuUU#/M#"455%eoaii &;&;<<%eobjI&F&F&M&Mbh&W&WXX{{5!!!rc$t|ddSr)r)rs rget_normal_texturez pack..get_normal_texturessOT222rct|dd}|1jtjdggtj}n|d}|S)NrrrL)rrrrrr)rocclusion_texturer s rget_occlusion_texturez#pack..get_occlusion_textures`#C);TBB  $ /3%0Q0Q0Q R R   1 9 9# > >   rcg}t||D]G\}}||d||}||H|Sr()ziprKresize)imagessizesresizedrsizes r resize_imageszpack..resize_imagessiVU++ $ $IC{t$$$$jj&&s####rc tttd|D}|vsJ|vr|S|ddddt }||<|S)Nc,g|]}t|Sr)id)rxis rrzz-pack..pack_images..s 7 7 71A 7 7 7rFT* ) deduplicate power_resizeseed iterationsspacing)r]tuplesortedr"r`)r"keyrpackedrrs r pack_imageszpack..pack_imagess 5 7 7 7 7 78899::&    &==#;  LL    s  rcfg|]-\}}t||.Sr)r]rK)rxr*ry unique_idxs rrzzpack..s6HHH41aDGG  # #A & &HHHr)rr r c3@K|]}t|tVdSr(rvrws rrzpack..s,@@*Q ,,@@@@@@rc>g|]}|dSrr)rxgr rss rrzzpack..s-KKKa((1Q499KKKrc>g|]}|dSr<r)rxr=rrss rrzzpack..s:"""DE..y1??"""rc>g|]}|dSr<r)rxr=rrss rrzzpack..s-OOO!,,Yqt_==OOOrc>g|]}|dSr<r)rxr=rrss rrzzpack..s-LLLq)))AaD/::LLLrc>g|]}|dSr<r)rxr=rrss rrzzpack..s-QQQA..y1??QQQrcg|]}||Sr(rrxtexs rrzzpack..sGGGCsCrc@g|]}tj|jSr)rrr%rCs rrzzpack..s$%M%M%MSbhsx&8&8%M%M%Mrrc3jK|].}tj|dkV/dS)rN)rrmax)rxxs rrzpack..,s988!rx{{  A%888888rc3K|]}|duV dSr(r)rxns rrzpack..3s&..q}......rc3jK|].}tj|dkV/dS)rN)rrr)rxos rrzpack..:s9::1rx{{  3&::::::rrg?re)atolc g|] }| Srr)rxr*new_uvs rrzzpack..gs<<::F!./A>!R!R $}X~>>H#mG^<>$;;;$LLNN-DDE!x5(899@@JJH%%h////%V,,w  * 1  1 -0BA0EEE !Q & ! # ) # # m ! # #p#'2k2D&E&E# ! 88x888 8 8 6H!NN + H 5 5 NA ..g... . . ! +{733 M11 M :: ::: : : #!,Y!7!7 OQQ"O%*BJ777J F!'67;;  sFHSX&&*j8j1n- RXch'''  Aa&++--C  cTEk3#*+=>> ? ? ?3 F ? ? ?'#tS4Z00C8^y0Ez 54SVSAA4UEBB{3!444444F1II+ 0i<<<rsM %%%%%%%%++++ 'J ' * *EEEEEE+  # # # # #tx # # #F[ [ [ [ [ X[ [ [ |P#P#P#P#P#HP#P#P#f_ _ _ _ _ (_ _ _ D ::(728#45:::::&SSx 12SSSS8  {4{4 {4{4{4{4{4{4s3AA  A