l0j<dZddlZddlZddlZddlmZmZ ddlmZn#e $r ddlmZYnwxYwGddej Z Gdd e Z Gd d e Z Gd d e ZGdde ZddZdS)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)urlparseceZdZdZejdZejdZejdZejdZ dZ dZ dZ d S) Resolverz' The base class for resolvers. c td)NzUse a resolver subclass!NotImplementedError)selfargskwargss T/home/wildlama/miniconda3/envs/lam/lib/python3.11/site-packages/trimesh/resolvers.py__init__zResolver.__init__!"<===ctNr r keys rgetz Resolver.get"s!###rc tdNz`write` not implemented!r r namedatas rwritezResolver.write&rrc td)Nz`namespaced` not implemented!r r namespaces r namespacedzResolver.namespaced*s!"ABBBrc,||Sr)rrs r __getitem__zResolver.__getitem__.sxx}}rc.|||Sr)rr rvalues r __setitem__zResolver.__setitem__1szz#u%%%rc.||vSr)keysrs r __contains__zResolver.__contains__4sdiikk!!rN) __name__ __module__ __qualname____doc__abcabstractmethodrrrr!r#r'r*rrrrs >>> $$$ >>> CCC&&&"""""rrc0eZdZdZdZdZdZdZdZdS)FilePathResolverz> Resolve files from a source path on the file system. ctjtjt |}||_tj|r||_n,tj|}|d|_tj|jstd|jddS)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 expanduserabspathstrcleanisdirparentsplit ValueError)r sourcer:r=s rrzFilePathResolver.__init__=s""27??3v;;#?#?@@ 7==   #DKKGMM%((E(DKw}}T[)) GEdkEEEFF F G Grc#K|j}tj|jD]Y\}}}||r|t |d}|D]$}tj||V%ZdS)z List all files available to be loaded. Yields ----------- name : str Name of a file which can be accessed. N)r<r5walk startswithlenr6join)r r<r6_namesrs rr)zFilePathResolver.keysVs gdk22 / /ND!Uv&& +CKKMM* / /gll4...... /  / /rcfttj|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. )r3r5r6rDr<rs rr!zFilePathResolver.namespacedhs$  T[) D DEEErctj|j|}tj|sHtj|jtj|d}t|d5}|}dddn #1swxYwY|S)z Get an asset. Parameters ------------- name : str Name of the asset Returns ------------ data : bytes Loaded data from asset rbN) r5r6rDr<stripexistsr=openread)r rr6frs rrzFilePathResolver.getysw||DK66w~~d## F7<< RW]]4-@-@-DEED $   6688D                s/CCCcttj|j|d5}t j||ddddS#1swxYwYdS)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)rMr5r6rDr<rKr write_encoded)r rrrOs rrzFilePathResolver.writes"',,t{DJJLL994 @ @ 7A   6 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7sA**A.1A.N) r+r,r-r.rr)r!rrr1rrr3r38soGGG2///$FFF",77777rr3c8eZdZdZd dZdZdZdZdZdZ dS) ZipResolverz- Resolve files inside a ZIP archive. Nc||_t|tr1|ddz|_dSd|_dS)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. /N)archive isinstancer9rKrstripr )r rYr s rrzZipResolver.__init__sP i % % "&__..55c::S@DNNN!DNNNrc|j<|jtfd|jDS|jS)z Get the available keys in the current archive. Returns ----------- keys : iterable Keys in the current archive. Nctg|]4}|rt|k*|d5SrrBrC).0klengthr s r z$ZipResolver.keys..sN<< **031vv&'' />r)r rCrYr))r rar s @@rr)zZipResolver.keysst > %I^^F**,,  |  """rc6|ji|_||j|<dS)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)rYr%s rrzZipResolver.writes% < DL! Srcv|dSt|dr|d}|j}||vr t||jD] }||vr|}n ||}t |t tfr|S|d| }|d|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) hasattrrerY nearby_namesr rZbytesr9seekrN)r rrYoptionobjrs rrzZipResolver.gets < F 4 " " (;;w''D, w  &tT^<<  W$$"DE %dm cE3< ( ( J  xxzz   rc.t|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. )rYr )rVrYrs rr!zZipResolver.namespaced s 4<9EEEErc4tj|jS)z Export the contents of the current archive as a ZIP file. Returns ------------ compressed : bytes Compressed data in ZIP format. )rcompressrYr s rexportzZipResolver.exports}T\***r)NN) r+r,r-r.rr)rrr!rrr1rrrVrVs""""(###, " " ",,,\FFF$ + + + + +rrVc*eZdZdZdZdZdZdZdS) WebResolverz+ Resolve assets from a remote URL. ct|dr|d}t|}d|jdD}t |dkrd}n=d|dvrd|d d}nd|}dd |jd z|j d| dfDdz|_ d |j t |jd zd vsJ|j dsJd S)z Resolve assets from a base URL. Parameters -------------- url : str Location where a mesh was stored or directory where mesh was stored rerfc8g|]}t|dk|S)rrCr_is rrbz(WebResolver.__init__..As#AAAqc!ffqjjjjjrrXr.rINc3@K|]}t|dk|VdS)rNrwrxs r z'WebResolver.__init__..Ns:q66A:::::rz:/z//) rgrerr6r=rCrDschemenetlocrKbase_urlendswith)r urlparsedr=r6s rrzWebResolver.__init__0se 3 ! ! &**W%%C#BAFK--c22AAA u::??DD E"I  88E#2#J''DD88E??D HH -$. 0C0CC0H0H$**UX//Z      4=V]););a)?)A)ABBBBB}%%c*******rc<ddl}|}|j|z}||d}|jdkr>|dr |dd}||j|zd}||jS)z Get a resource from the remote site. Parameters ------------- name : str Asset name, i.e. 'quadknot.obj.mtl' rNT)follow_redirectsi,./)httpxrKrr status_coderBraise_for_statuscontent)r rrrresponses rrzWebResolver.get[s  zz||md"99S4988  3 & &t$$ ABBxyy!5yMMH !!###rc2t|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)rtrrs rr!zWebResolver.namespaced|st}y89999rc td)Nzcan't write to remoter r%s rrzWebResolver.writes!"9:::rN)r+r,r-r.rrr!rr1rrrtrt+s\)+)+)+V   B:::";;;;;rrtcDeZdZddZdZdZedZdZdZ dS) GithubResolverNc|d|||_nd|||_|tj||_dSd|_dS)a Get files from a remote Github repository by downloading a zip file with the entire branch or a specific commit. Parameters ------------- repo : str In the format of `owner/repo` branch : str The remote branch you want to get files from. commit : str The full commit hash: pass either this OR branch. save : None or str A path if you want to save results locally. Nz9https://github.com/{repo}/archive/refs/heads/{branch}.zip)repobranchz.https://github.com/{repo}/archive/{commit}.zip)rcommit)formatrr DiskCachecache)r rrrsaves rrzGithubResolver.__init__st$ >Pf$vf.. HH NUU&VDH   *400DJJJDJJJrc4|jS)z List the available files in the repository. Returns ---------- keys : iterable Keys available to the resolved. )zippedr)rqs rr)zGithubResolver.keyss{!!!rc tdrr rs rrzGithubResolver.writes!"<===rcfd}tdrjSjj|}t t jt j|d_jS)zk - opened zip file - locally saved zip file - retrieve zip file and saved cddl}|j}|jst |j|jS)z4 Fetch the remote zip file. rN)requestsrrokr>r)rrr s rfetchz$GithubResolver.zipped..fetchsE OOO||DH--H; 3 !1222# #r_zipzip) file_type) rgrrrrrVr decompresswrap_as_stream)r rraws` rrzGithubResolver.zippeds $ $ $ $ $ 4  9 jnnTXu-- OD/44 F F F   yrc6|j|Sr)rrrs rrzGithubResolver.gets{s###rc6|j|S)rn)rr!rs rr!zGithubResolver.namespaceds {%%i000r)NNN) r+r,r-rr)rpropertyrrr!r1rrrrsB " " ">>>X<$$$11111rrc#@ Kd dd fd fd fdddg}|d }t}|D].}||}||vr||||zV/tj|d D]o\}}|||}||vr||||zV|||}||vrS||||zVpd |vr||d d d |d  }||d dd d d |Vd Sd Sd S)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. c^||r|t|dS|Srr^)prefixitems rtrimznearby_names..trims0 ??6 " " 'F & & rc|Srr1xs rznearby_names.. s!rc*|Sr)rKrs rrznearby_names..s!''))rcd|S)Nrr1rrs rrznearby_names..$$tQ--rcd|S)Nz.\r1rs rrznearby_names..s$$ua..rcd|S)N\r1rs rrznearby_names..rrcLtj|dS)NrI)r5r6r=rs rrznearby_names..s"'--""2&rc.|ddS)Nz%20 )replacers rrznearby_names..s!))E3''rNrzrz..rXrI) setadd itertools combinationsrKr=countextendrD) rr cleanershitrOcurrentabrKrs @rrhrhs"      &&''H  %%C ""!D'' c>>  '!!!!!&x33 " "1!AAdGG** c>>  '!!!!!AAdGG** c>>  '!!!!! t|| -$$**3//0C4::d3C3C2C0CD TZZ%%b)//44::3??@@@hhuoo |--rr)r.r/rr5rzrr urllib.parser ImportErrorABCrr3rVrtrrhr1rrrs  "%%%%%%%"""!!!!!!!!" """""tx""">e7e7e7e7e7xe7e7e7PH+H+H+H+H+(H+H+H+Vc;c;c;c;c;(c;c;c;Lb1b1b1b1b1Xb1b1b1J@@@@@@s  --