U vId +@sddlmZddlmZmZmZddlmZddlm Z m Z ddl m Z ddl mZmZddlmZddlmZmZdd lmZmZed d Gd d d ZGdddeZd dddZGdddZeZe d dddZeed dddZ eed dddZ!dd dddZ"Gd d!d!Z#Gd"d#d#Z$dd$dd%d&Z%e id'd(d)d$d*d+d,d-d.Z&d/e'd/e'd/fe(d0d1d2d3d4Z)e d)dd5d6Z*d/S)7) annotations)MappingCallableIterable) dataclass)singledispatchwraps)MappingProxyType)AnyUnion) _read_attr) StorageTypeGroupStorageType)report_write_key_on_errorreport_read_key_on_errorT)frozenc@seZdZUded<ded<dS)IOSpecstr encoding_typeencoding_versionN)__name__ __module__ __qualname____annotations__rrS/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/anndata/_io/specs/registry.pyrs rc@s8eZdZeddddZeddddddd d Zd S) IORegistryErrorreturncCs*d|d|}|r"|d|7}||S)Nz!No method registered for writing z into z with r)cls dest_typetyp modifiersmsgrrr_from_write_partssz!IORegistryError._from_write_partsrrr r)methodregistrysrc_typspecrcCs d|d|d|d}||S)NzNo z method registered for  from z6. You may need to update your installation of anndata.r)rr%r&r'r(r#rrr_from_read_parts s z IORegistryError._from_read_partsN)rrr classmethodr$r*rrrrrsrr(csddfdd }|S)Nrfunccstfdd}|S)Ncs>||f||}||jdj||jdj|S)N encoding-typeencoding-version)attrs setdefaultrr)gkargskwargsresult)r.r(rrwrapper2sz.write_spec..decorator..wrapper)r)r.r8r,r-r decorator1szwrite_spec..decoratorr)r(r9rr,r write_spec0s r:c@seZdZddZefddddddd Zefddd d d d Zddd d ddZefddddddZefddd dddZ efddd dddZ efddddddZ efddd dddZ dddddZ d S)! IORegistrycCsi|_i|_i|_i|_dSN)read read_partialwrite write_specs)selfrrr__init__?s zIORegistry.__init__typeztype | tuple[type, str]zIOSpec | Mapping[str, str]z Iterable[str])r src_typer(r"cspttjkrNjkrNj}tdd|dn j<fdd}|S)NzHCannot overwrite IO specifications. Attempted to overwrite encoding for r)z to cst|jf<|Sr<)r:r?r-r r"rAr(rDrr _register\sz,IORegistry.register_write.._register) proc_spec frozensetr@ TypeError)rAr rDr(r"Z current_specrFrrErregister_writeGs  zIORegistry.register_writefrozenset[str])r rDr"cCsJddl}||jkr|j}|||f|jkr8|j|||fSt|||dS)Nr)h5pyFileGroupr?rr$)rAr rDr"rLrrr get_writerbs  zIORegistry.get_writercCs|||f|jkSr<)r?)rAr rDr"rrr has_writerrszIORegistry.has_writer)rDr(r"cs&ttfdd}|S)Ncs|jf<|Sr<r=r-r"rAr(rDrrrFsz+IORegistry.register_read.._registerrGrHrArDr(r"rFrrRr register_readzszIORegistry.register_readrcCs6|||f|jkr |j|||fStdtj||dS)Nr=)r=rr* _REGISTRYrArDr(r"rrr get_readerszIORegistry.get_readercCs|||f|jkSr<rQrWrrr has_readerszIORegistry.has_readercs&ttfdd}|S)Ncs|jf<|Sr<)r>r-rRrrrFsz3IORegistry.register_read_partial.._registerrSrTrrRrregister_read_partialsz IORegistry.register_read_partialcCs6|||f|jkr |j|||fStdtj||dS)Nr>)r>rr*rVrWrrrget_partial_readerszIORegistry.get_partial_readerr elemrcCs<t|dr.t||jjf}||jkr.|j|S|jt|S)Ndtype)hasattrrCr^kindr@)rAr]r!rrrget_specs    zIORegistry.get_specN)rrrrBrHrJrOrPrUrXrYrZr[rarrrrr;>s"     r;rcCstdt|ddS)Nz proc_spec not defined for type: .)NotImplementedErrorrCr,rrrrGsrGcCs|Sr<rr,rrrproc_spec_specsrdcCstfdd|DS)NcSsi|]\}}|dd|qS)-_)replace).0r4vrrr s z%proc_spec_mapping..)ritemsr,rrrproc_spec_mappingsrlr r\cstfdddDS)Ncsi|]}|tj|dqS))r r1)rhr4r]rrrjszget_spec..)r/r0)rGrnrrnrras  rac@s<eZdZdddddddZeefdd d d d d ZdS)ReaderNr;zUnion[Callable, None]None)r&callbackrcCs||_||_dSr<r&rqrAr&rqrrrrBszReader.__init__r zfrozenset(str)r )r]r"rcCsbddlm}|jt|t|t|}|||d}|jdk rV|j||j|t|dS||SdS)zERead an element from a store. See exported function for more details.rpartial)_readerN)iospec) functoolsrur&rXrCrarHrqname)rAr]r"ruZ read_funcrrr read_elems   zReader.read_elem)N)rrrrBrrHrzrrrrros roc@s@eZdZd dddddZeeieddd d d d ZdS)WriterNr;zGUnion[Callable[[GroupStorageType, str, StorageType, dict], None], None]rrcCs||_||_dSr<rrrsrrrrBszWriter.__init__)dataset_kwargsr"rr)storer4c Csddlm}ddlm}t|}t|} |dkr8ddS||sVt||j|}|dkrh|n||krv||=t |dr|| |j j f|f|j j kr||j || |j j f||d} n||j || ||d} |jdk r|j| |||||j |d S| ||||d SdS) Nrrt) PurePosixPathc_sdSr<r)rf__rrrz#Writer.write_elem../r^)Z_writer)r|rwr|)rxrupathlibr~rC is_absoluterryclearr_r^r`r&r?rOrqra) rAr}r4r]r|r"rur~r tZ write_funcrrr write_elemsD       zWriter.write_elem)N)rrrrBrr rHrrrrrr{s r{r cCstt|S)a Read an element from a store. Assumes that the element is encoded using the anndata encoding. This function will determine the encoded type using the encoding metadata stored in elem's attributes. Params ------ elem The stored element. )rorVrzrnrrrrz:s rzrrrrrp)r}r4r]r|rcCsttj||||ddS)a Write an element to a storage group using anndata encoding. Params ------ store The group to write to. k The key to write to in the group. Note that absolute paths will be written from the root. elem The element to write. Typically an in-memory object, e.g. an AnnData, pandas dataframe, scipy sparse matrix, etc. dataset_kwargs Keyword arguments to pass to the stores dataset creation function. E.g. for zarr this would be `chunks`, `compressor`. rN)r{rVr)r}r4r]r|rrrrIsrN)rkindicesr"rK)r"cCs$tt|t|t||||dS)z.Read part of an element from an on disk store.)rkr)rVr[rCrarH)r]rkrr"rrrread_elem_partialesrcCs|jSr<)ryrnrrrelem_keyrsr)+ __future__rcollections.abcrrr dataclassesrrxrrtypesr typingr r Zanndata.compatr Zanndata._typesr rZanndata._io.utilsrrr Exceptionrr:r;rVrGregisterrdrlraror{rzrslicerHrrrrrrs@    { J