jBCdZddlZddlZddlZddlmZmZddlmZm Z m Z m Z  ddl m Z GddejZGd d eZGd d eZGd deZGddeZddZe ee fZy#e$r ddl m Z YZwxYw)z resolvers.py --------------- Provides a common interface to load assets referenced by name like MTL files, texture images, etc. Assets can be from ZIP archives, web assets, or a local file path. N)cachingutil)DictMappingOptionalUnion)urlparseceZdZdZej dZej dZej defdZ ej defdZ ej dZ d efd Z d efd Z d ed efd Zy)Resolverz' The base class for resolvers. ctd)NzUse a resolver subclass!NotImplementedError)selfargskwargss >/DATA/.local/lib/python3.12/site-packages/trimesh/resolvers.py__init__zResolver.__init__!"<==ctNrrkeys rgetz Resolver.get#s !##rnamectdNz`write` not implemented!rrrdatas rwritezResolver.write'rr namespacectd)Nz`namespaced` not implemented!rrr"s r namespacedzResolver.namespaced+s!"ABBrctd)Nz`keys` not implemented!rrs rkeysz Resolver.keys/s!";<> $$ >#>> CCCC ==s&s&"""rr cXeZdZdZdefdZdZdeddfdZdefd Zded e ee ffd Z y ) FilePathResolverz> Resolve files from a source path on the file system. sourcectjjtjjt |}||_tjj |r||_n)tjj|}|d|_tjj |jstd|jd||_ tjj||_ y)z Resolve files based on a source path. Parameters ------------ source : str File path where mesh was loaded from rzpath `z `not a directory!N) ospath expanduserabspathr8cleanisdirparentsplit ValueError file_pathbasename file_name)rr=rCrFs rrzFilePathResolver.__init__Bs""277??3v;#?@ 77== DKGGMM%(E(DKww}}T[[)vdkk]2DEF F))&1rc#K|j}tj|jD]N\}}}|j|r|t |d}|D]$}tj j ||&Pyw)z List all files available to be loaded. Yields ----------- name : str Name of a file which can be accessed. N)rEr?walk startswithlenr@join)rrEr@_namesrs rr(zFilePathResolver.keys^si ggdkk2ND!Uv&CKM*ggll4.. 3sA>Br"r/chttjj|j|S)aI Return a resolver which changes the root of the resolver by an added namespace. Parameters ------------- namespace : str Probably a subdirectory Returns -------------- resolver : FilePathResolver Resolver with root directory changed. )r<r?r@rOrEr$s rr%zFilePathResolver.namespacedps"  T[[) DEErrcxtjj|j|j }tjj |sGtjj|j|j j d}tjj |sJtjj|jtjj|d}t|d5}|j}ddd|S#1swYSxYw)z Get an asset. Parameters ------------- name : str Name of the asset Returns ------------ data : bytes Loaded data from asset /rbN) r?r@rOrEstripexistslstriprFopenread)rrr@fr s rrzFilePathResolver.getsww||DKK6ww~~d#77<< TZZ\-@-@-EFDww~~d#77<< RWW]]4-@-DED $ 668D  s D//D9r cttjj|j|j d5}t j||dddy#1swYyxYw)z Write an asset to a file path. Parameters ----------- name : str Name of the file to write data : str or bytes Data to write to the file wb)file_objstuffN)rZr?r@rOrErWr write_encoded)rrr r\s rr!zFilePathResolver.writesF"'',,t{{DJJL94 @A    6A @ @s A$$A-N) r2r3r4r5r8rr(r%rr bytesr!r:rrr<r<=sW2s28/$FCF,>F"07#7U3:%67rr<cpeZdZdZddeedeefdZdZdeddfd Z d ede fd Z deddfd Z de fd Z y) ZipResolverz- Resolve files inside a ZIP archive. Narchiver"c||_t|tr(|jj ddz|_yd|_y)a{ Resolve files inside a ZIP archive as loaded by trimesh.util.decompress Parameters ------------- archive : dict Contains resources as file object namespace : None or str If passed will only show keys that start with this value and this substring must be removed for any get calls. rTN)re isinstancer8rWrstripr")rrer"s rrzZipResolver.__init__s: i %&__.55c:S@DN!DNrc|ja|j}t|}|jjDcgc]&}|j |rt||kDr||d(c}S|jjScc}w)z Get the available keys in the current archive. Returns ----------- keys : iterable Keys in the current archive. N)r"rNrer(rM)rr"lengthks rr(zZipResolver.keyss >> %I^F **,,A<< *s1v&' ,  ||  "" s+Brr/cH|ji|_||j|<y)z Store a value in the current archive. Parameters ----------- key : hashable Key to store data under. value : str, bytes, file-like Value to store. N)rer,s rr!zZipResolver.writes# << DL! SrrcJ|yt|dr|jd}|j}||vr#t||jD] }||vs|}n||}t |t tfr|S|jd|j}|jd|S)z Get an asset from the ZIP archive. Parameters ------------- name : str Name of the asset Returns ------------- data : bytes Loaded data from asset Ndecodeutf-8r) hasattrrnre nearby_namesr"rgrbr8seekr[)rrreoptionobjr s rrzZipResolver.gets <  4 ";;w'D,, w &tT^^<W$"D =dm cE3< (J  xxz   rc0t|j|S) Return a "sub-resolver" with a root namespace. Parameters ------------- namespace : str The root of the key to clip off, i.e. if this resolver has key `a/b/c` you can get 'a/b/c' with resolver.namespaced('a/b').get('c') Returns ----------- resolver : Resolver Namespaced resolver. )rer")rdrer$s rr%zZipResolver.namespaceds 4<<9EErc@tj|jS)z Export the contents of the current archive as a ZIP file. Returns ------------ compressed : bytes Compressed data in ZIP format. )rcompressrer's rexportzZipResolver.export(s}}T\\**r)NN)r2r3r4r5rrr8rr(r!rbrr%ryr:rrrdrdsp""(3-"(#, " " ",,,\FCFMF$ + +rrdcTeZdZdZdefdZdedefdZdefdZdeddfd Z d Z d Z y ) WebResolverz+ Resolve assets from a remote URL. urlct|dr|jd}t|}|jj dDcgc]}t |dkDs|}}t |dk(rd}n-d|dvrdj |dd}ndj |}||_dj d |jd z|jjd|jdfDdz|_ d |jt |jd zdvsJ|jjdsJ|j dd|_ ycc}w) z Resolve assets from a base URL. Parameters -------------- url : str Location where a mesh was stored or directory where mesh was stored rnrorTr.rUNc3<K|]}t|dkDr|yw)rN)rN).0is r z'WebResolver.__init__..\s#ZAq6A:Zsz:/z//)rprnr r@rFrNrOr|schemenetlocrWbase_urlendswithrJ)rr|parsedrrFr@s rrzWebResolver.__init__:sP 3 !**W%C#"KK--c2A2qc!fqj2A u:?D E"I 88E#2J'D88E?D HH --$. 0C0CC0H$**UX/Z     4==V]]);a)?)ABBBB}}%%c***3+?Bs EErr/c0ddl}|j}|j|z}|j|d}|jdk\r6|j dr|dd}|j|j|zd}|j |jS)z Get a resource from the remote site. Parameters ------------- name : str Asset name, i.e. 'quadknot.obj.mtl' rNTfollow_redirectsi,./)httpxrWrr status_coderMraise_for_statuscontent)rrrr|responses rrzWebResolver.getks zz|mmd"99S498   3 &t$ABxyy!5yMH !!#rc|ddl}|j|jd}|j|jS)z Fetch the data at the full URL this resolver was instantiated with, i.e. `https://stuff.com/hi.glb` this will return the response. Returns -------- content The value at `self.url` rNTrrrr|rr)rrrs rget_basezWebResolver.get_bases7 99TXX9=!!#rr"c4t|j|zS)a Return a namespaced version of current resolver. Parameters ------------- namespace : str URL fragment Returns ----------- resolver : WebResolver With sub-url: `https://example.com/{namespace}` )r|)r{rr$s rr%zWebResolver.namespacedst}}y899rctd)Nz`WebResolver` is read-only!rr,s rr!zWebResolver.writes!"?@@rctd)Nz`WebResolver` can't list keysrr's rr(zWebResolver.keyss!"ABBrN) r2r3r4r5r8rrbrrr%r!r(r:rrr{r{5sS/,C/,b   B % $:C:M:"ACrr{c leZdZ d dedeedeedeefdZdZdZed e fd Z d Z d Z y)GithubResolverNrepobranchcommitsavec|d|d|d|_n|d|d|d|_n td|tj||_yd|_y)a Get files from a remote Github repository by downloading a zip file with the entire branch or a specific commit. Parameters ------------- repo In the format of `owner/repo` branch The remote branch you want to get files from. commit The full commit hash: pass either this OR branch. save A path if you want to save results locally. Nzhttps://github.com/z /archive/z.zipz/archive/refs/heads/z$`commit` or `branch` must be passed!)r|rGr DiskCachecache)rrrrrs rrzGithubResolver.__init__sh0  ,TF)F84HDH  ,TF2FvhdSDHCD D   **40DJDJrc6|jjS)z List the available files in the repository. Returns ---------- keys : iterable Keys available to the resolved. )zippedr(r's rr(zGithubResolver.keyss{{!!rctdrrrs rr!zGithubResolver.writes!"<==rr/c dtffd }tdr jSjj j |}t tjtj|djDcic]\}}|jddd|c}}_jScc}}w)zj - opened zip file - locally saved zip file - retrieve zip file and saved r/c~ddl}|jjd}|j|jS)z4 Fetch the remote zip file. rNTrr)rrrs rfetchz$GithubResolver.zipped..fetchs6 yyDyAH  % % '## #r_zipzip) file_typerTr) rbrprrrr|rdr decompresswrap_as_streamitemsrF)rrrawrkvs` rrzGithubResolver.zippeds $u $ 4 99 jjnnTXXu- !OO'',%' DAqQ"A%   yy s C c8|jj|Sr)rrrs rrzGithubResolver.gets{{s##rc8|jj|S)rv)rr%r$s rr%zGithubResolver.namespaceds {{%%i00r)NNN) r2r3r4r8rrr(r!propertyrdrrr%r:rrrrsu!% $" $$ $ $ sm $L ">! !!F$1rrc# Kd dd fd fd fdddg}|d }t}|D]'}||}||vr|j|||z)tj|d D][\}}|||}||vr|j|||z|||}||vrD|j|||z]d |vr||j d j d d |j d  }|j|j d dj d j d d j|y y y w)z Try to find nearby variants of a specified name. Parameters ------------ name : str Initial name. Yields ----------- nearby : str Name that is a lightly permutated version of the initial name. cD|j|r|t|dS|Sr)rMrN)prefixitems rtrimznearby_names..trim6s$ ??6 "F & & rc|Srr:xs rznearby_names..<s!rc"|jSr)rWrs rrznearby_names..=s !'')rcd|S)Nrr:rrs rrznearby_names..> $tQ-rcd|S)Nz.\r:rs rrznearby_names..?s $ua.rcd|S)N\r:rs rrznearby_names..@rrcFtjj|dS)NrU)r?r@rFrs rrznearby_names..As"''--"2&rc&|jddS)Nz%20 )replacers rrznearby_names..Bs!))E3'rNr~rz..rTrU) setadd itertools combinationsrWrFcountextendrO) rr"cleanershitr\currentabrWrs @rrqrq%so"  &'H  %C D' c>  '!! &&x31AdG* c>  '!!AdG* c>  '!!4 t| -$**3/0C4::d3C2CD TZZ%b)//4::3?@hhuo .|sEEr)r5r6rr?r~rrtypedrrrr urllib.parser ImportErrorABCr r<rdr{rrq ResolverLiker:rrrs  11"% "txx "Fj7xj7ZH+(H+V~C(~CBl1Xl1^@HXw&' i"!"sA77 BB