TiBC.dZddlZddlZddlZddlmZmZddlmZm Z m Z m Z  ddl m Z n#e$r ddl m Z YnwxYwGddejZGd d eZGd d eZGd deZGddeZddZe ee fZdS)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)urlparseceZdZdZejdZejdZejdefdZ ejdefdZ ejdZ d efd Z d efd Z d ed efd ZdS)Resolverz' The base class for resolvers. c td)NzUse a resolver subclass!NotImplementedError)selfargskwargss K/DATA/AppData/hermes/venv/lib/python3.11/site-packages/trimesh/resolvers.py__init__zResolver.__init__!"<===ctNrrkeys rgetz Resolver.get#s!###rnamec tdNz`write` not implemented!rrrdatas rwritezResolver.write'rr namespacec td)Nz`namespaced` not implemented!rrr"s r namespacedzResolver.namespaced+s!"ABBBrc td)Nz`keys` not implemented!rrs rkeysz Resolver.keys/s!";<<>> $$$ >#>>>> CCCCCC ===s&s&&&&""""""""rr c`eZdZdZdefdZdZdeddfdZdefd Zded e ee ffd Z d S) FilePathResolverz> Resolve files from a source path on the file system. sourcectjtjt |}||_tj|r||_n,tj|}|d|_tj|jstd|jd||_ tj ||_ dS)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""27??3v;;#?#?@@ 7==   #DKKGMM%((E(DKw}}T[)) GEdkEEEFF F))&11rc#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)rEr?walk startswithlenr@join)rrEr@_namesrs rr(zFilePathResolver.keys^s gdk22 / /ND!Uv&& +CKKMM* / /gll4...... /  / /rr"r/cfttj|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[) D DEEErrctj|j|}tj|sJtj|j|d}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) r?r@rOrEstripexistslstriprFopenread)rrr@fr s rrzFilePathResolver.getsw||DK66w~~d## G7<< TZZ\\-@-@-E-EFFDw~~d## F7<< RW]]4-@-@-DEED $   6688D                sD99D=D=r cttj|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)rZr?r@rOrErWr write_encoded)rrr r\s rr!zFilePathResolver.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) r2r3r4r5r8rr(r%rr bytesr!r:rrr<r<=s2s22228///$FCF,>FFFF"07#7U3:%6777777rr<c~eZdZdZddeedeefdZdZdeddfd Z d ede fd Z deddfd Z de fd Z dS) ZipResolverz- Resolve files inside a ZIP archive. Narchiver"c||_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. rTN)re isinstancer8rWrstripr")rrer"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*|d5SrrMrN).0klengthr"s r z$ZipResolver.keys..sN<< **031vv&'' />r)r"rNrer()rrnr"s @@rr(zZipResolver.keysst > %I^^F**,,  |  """rrr/c6|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)rer,s rr!zZipResolver.writes% < DL! Srrcv|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) hasattrrrre nearby_namesr"rgrbr8seekr[)rrreoptionobjr s 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. )rer")rdrer$s rr%zZipResolver.namespaceds 4<9EEEErc4tj|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%r}r:rrrdrds"""(3-""""(###, " " " " " ",,,,,,\FCFMFFFF$ + + + + + + +rrdcVeZdZdZdefdZdedefdZdefdZdeddfd Z d Z d Z d S) WebResolverz+ Resolve assets from a remote URL. urlct|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 dsJ|dd|_ d S)z Resolve assets from a base URL. Parameters -------------- url : str Location where a mesh was stored or directory where mesh was stored rrrsc8g|]}t|dk|S)rrNrlis rroz(WebResolver.__init__..Js#AAAqc!ffqjjjjjrrTr.rUNc3@K|]}t|dk|VdS)rNrrs r z'WebResolver.__init__..\s:q66A:::::rz:/z//)rtrrr r@rFrNrOrschemenetlocrWbase_urlendswithrJ)rrparsedrFr@s rrzWebResolver.__init__:s 3 ! ! &**W%%C#AAFK--c22AAA u::??DD E"I  88E#2#J''DD88E??D HH -$. 0C0CC0H0H$**UX//Z      4=V]););a)?)A)ABBBBB}%%c*****3+rrr/c<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' rNTfollow_redirectsi,./)httpxrWrr status_coderMraise_for_statuscontent)rrrrresponses rrzWebResolver.getks  zz||md"99S4988  3 & &t$$ ABBxyy!5yMMH !!###rcxddl}||jd}||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` rNTrrrrrr)rrrs rget_basezWebResolver.get_basesA  99TX9==!!###rr"c2t|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)rrr$s rr%zWebResolver.namespacedst}y89999rc td)Nz`WebResolver` is read-only!rr,s rr!zWebResolver.writes!"?@@@rc td)Nz`WebResolver` can't list keysrr's rr(zWebResolver.keyss!"ABBBrN) r2r3r4r5r8rrbrrr%r!r(r:rrrr5s/,C/,/,/,/,b      B %    $:C:M::::"AAACCCCCrrc eZdZ d dedeedeedeefdZdZdZed e fd Z d Z d Z dS)GithubResolverNrepobranchcommitsavec|d|d|d|_n |d|d|d|_ntd|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 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!)rrGr DiskCachecache)rrrrrs rrzGithubResolver.__init__s~0  HTHHFHHHDHH  STSSvSSSDHHCDD D   *400DJJJDJJJrc4|jS)z List the available files in the repository. Returns ---------- keys : iterable Keys available to the resolved. )zippedr(r's rr(zGithubResolver.keyss{!!!rc tdrrrs rr!zGithubResolver.writes!"<===rr/cFdtffd }tdrjSjj|}t dtjtj |d D_jS)zj - opened zip file - locally saved zip file - retrieve zip file and saved r/czddl}|jd}||jS)z4 Fetch the remote zip file. rNTrr)rrrs rfetchz$GithubResolver.zipped..fetchs@ LLLyyDyAAH  % % ' ' '# #r_zipcNi|]"\}}|ddd|#S)rTr)rF)rlrmvs r z)GithubResolver.zipped..s?   AqQ"A   rzip) file_type) rbrtrrrrrdr decompresswrap_as_streamitems)rrraws` rrzGithubResolver.zippeds $u $ $ $ $ $ $ 4  9 jnnTXu--    O',,%''       yrc6|j|Sr)rrrs rrzGithubResolver.gets{s###rc6|j|S)rz)rr%r$s rr%zGithubResolver.namespaceds {%%i000r)NNN) r2r3r4r8rrr(r!propertyrdrrr%r:rrrrs!% $" $$$ $ $ sm $$$$L " " ">>>! !!!X!F$$$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|Srrk)prefixitems rtrimznearby_names..trim6s0 ??6 " " 'F & & rc|Srr:xs rznearby_names..<s!rc*|Sr)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..@rrcLtj|dS)NrU)r?r@rFrs rrznearby_names..As"'--""2&rc.|ddS)Nz%20 )replacers rrznearby_names..Bs!))E3''rNrrz..rTrU) setadd itertools combinationsrWrFcountextendrO) rr"cleanershitr\currentabrWrs @rruru%s"      &&''H  %%C ""!D'' c>>  '!!!!!&x33 " "1!AAdGG** c>>  '!!!!!AAdGG** c>>  '!!!!! t|| -$$**3//0C4::d3C3C2C0CD TZZ%%b)//44::3??@@@hhuoo |--rr)r5r6rr?rrrtypedrrrr urllib.parser ImportErrorABCr r<rdrrru ResolverLiker:rrrs  111111111111"%%%%%%%"""!!!!!!!!" " " " " "tx " " "Fj7j7j7j7j7xj7j7j7ZH+H+H+H+H+(H+H+H+V~C~C~C~C~C(~C~C~CBl1l1l1l1l1Xl1l1l1^@@@@HXw&' s + 99