U vIdd@sddlmZddlmZddlmZmZddlmZddl m Z ddl m Z ddl Z dd lmZdd lmZmZe e jjd kZd d d dddZddZddZddZddZddZddZddZGdd d eZ d!d"Z!d#d$Z"d%d&Z#d'd(d)d*d*d+d,d-d.Z$dS)/) annotations)Enum)wrapssingledispatch)Callable)warn)versionN) SparseDataset)H5Group ZarrGrouptupleint)shapeaxis chunk_sizeccsp||}d}ddtt|D}|||krTt|||||<t|V||7}q"t|d||<t|VdS)a. Gives indexer tuples chunked along an axis. Params ------ shape Shape of array to be chunked axis Axis to chunk along chunk_size Size of chunk along axis Returns ------- An iterator of tuples for indexing into an array of passed shape. rcSsg|] }tdqS)N)slice).0irJ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/anndata/_io/utils.py +sz)idx_chunks_along_axis..N)rangelenrr)rrrtotalcurZ mutable_idxrrridx_chunks_along_axiss   rcCs*zt|WdStk r$YdSXdS)z Check whether string is float. See also -------- http://stackoverflow.com/questions/736043/checking-if-a-string-can-be-converted-to-float-in-python TFN)float ValueErrorstringrrris_float4s r"cCs*zt|WdStk r$YdSXdS)z Check whether string is integer.TFN)rrr rrris_intCs r#cCs |dkr dS|dkrdSdSdS)z Check whether string is boolean.True)TTFalse)TF)FFNrr rrr convert_boolLs r&cCsLt|rt|St|r t|St|dr8t|dS|dkrDdS|SdS)z%Convert string to int, float or bool.rNoneN)r#rr"rr&r rrrconvert_stringVs  r)cCs2t|}t|trt|St|d|ddS)zChecks that passed value is a valid h5py key. Should convert it if there is an obvious conversion path, error otherwise. of type z" is an invalid key. Should be str.N)type issubclassstr TypeError)keytyprrr check_keyds r1cOs ddlm}tdt|||S)Nr') read_elemzHThis internal function has been deprecated, please use read_elem instead)specsr2rDeprecationWarning)argskwargsr2rrrread_attributeys  r7cOs ddlm}tdt|||S)Nr') write_elemzIThis internal function has been deprecated, please use write_elem instead)r3r8rr4)r5r6r8rrrwrite_attributes  r9c@seZdZdZdS)AnnDataReadErrorz-Error caused while trying to read in AnnData.N)__name__ __module__ __qualname____doc__rrrrr:sr:cCsfz ddl}Wntk r$d}YnX|rDt||j|jfrD|j}nt|trZ|jjj }n|jj }|S)Nr) zarr ImportError isinstanceGroupZArraystorer groupfilename)elemr?parentrrr _get_parents    rIcs"ddtfdd}|S)a[ A decorator for zarr element reading which makes keys involved in errors get reported. Example ------- >>> import zarr >>> @report_read_key_on_error ... def read_arr(group): ... raise NotImplementedError() >>> z = zarr.open("tmp.zarr") >>> z["X"] = [1, 2, 3] >>> read_arr(z["X"]) # doctest: +SKIP cSs@t|tr|n,t|}td|jdt|d|d|dS)Nz%Above error raised while reading key r*z from .)rAr:rIrFr+)erGrHrrrre_raise_errors z0report_read_key_on_error..re_raise_errorc sbddlm}|D]}t||sq$qz ||WStk r\}z||W5d}~XYnXdS)Nr)Reader)anndata._io.specsrMrA Exception)r5r6rMrGrKfuncrLrr func_wrappers   z.report_read_key_on_error..func_wrapperrrQrRrrPrreport_read_key_on_errors  rUcs"ddtfdd}|S)af A decorator for zarr element reading which makes keys involved in errors get reported. Example ------- >>> import zarr >>> @report_write_key_on_error ... def write_arr(group, key, val): ... raise NotImplementedError() >>> z = zarr.open("tmp.zarr") >>> X = [1, 2, 3] >>> write_arr(z, "X", X) # doctest: +SKIP cSsDdt|krn0t|}t||d|dt|d||dS)Nz$Above error raised while writing keyz' Above error raised while writing key z of z to )formatrIr+)rKrGr/rHrrrrLs z1report_write_key_on_error..re_raise_errorc sddlm}tt|D]&}||}||d}t||sq@qz ||WStk rz}z|||W5d}~XYnXdS)Nr)Writerr')rNrWrrrArO)r5r6rWrrGr/rKrPrrrRs    z/report_write_key_on_error..func_wrapperrSrTrrPrreport_write_key_on_errors rX)Xvarvarm)attrszZarrGroup | H5Grouprdict)fread_df read_attrreturncCs|rBtdd|Dr>t|dr,d|jnd}t|d|Si}d|krfd|krf||d|d<d |krd |kr||d |d <d |krd |kr||d |d <|S) z| Backwards compat for reading legacy raw. Makes sure that no modern raw group coexists with legacy raw.* groups. css|]}|dVqdS)zraw.N) startswith)rkrrr sz#_read_legacy_raw..filenamezFile Storez) has both legacy and current raw formats.rYzraw.XrZzraw.varr[zraw.varm)anyhasattrrer)r^Z modern_rawr_r`r\whatrawrrr_read_legacy_raws rk)% __future__renumr functoolsrrtypingrwarningsr packagingrZh5pyZ_core.sparse_datasetr Zanndata.compatr r parse __version__majorZH5PY_V3rr"r#r&r)r1r7r9OSErrorr:rIrUrXrkrrrrs.         )7