Cd<ddlmZddlmZdZd dZ d dZdS) ) annotations)corecl|jtfdtjDS)Nc3VK|]#}j|jV$dS)N)domaindimtile).0ischemas 4lib/python3.11/site-packages/dask/array/tiledb_io.py z$_tiledb_to_chunks..s6FFa !!!$$)FFFFFF)r listrangendim) tiledb_arrayr s @r _tiledb_to_chunksrs8  F FFFF53E3EFFF F FFrNc .ddl}|p t}|dd}t||jr|}n|||||}|jjrtd|s>|jj dkrtd|j dj }|j rtd|pt|}t||jjksJt#j||d |z S) aLoad array from the TileDB storage format See https://docs.tiledb.io for more information about TileDB. Parameters ---------- uri: TileDB array or str Location to save the data attribute: str or None Attribute selection (single-attribute view on multi-attribute array) Returns ------- A Dask Array Examples -------- >>> import tempfile, tiledb >>> import dask.array as da, numpy as np >>> uri = tempfile.NamedTemporaryFile().name >>> _ = tiledb.from_numpy(uri, np.arange(0,9).reshape(3,3)) # create a tiledb array >>> tdb_ar = da.from_tiledb(uri) # read back the array >>> tdb_ar.shape (3, 3) >>> tdb_ar.mean().compute() 4.0 rNkey)attrconfigrz&Sparse TileDB arrays are not supportedzOkeyword 'attribute' must be providedwhen loading a multi-attribute TileDB arrayz%TileDB array must be open for readingz tiledb-%s)name)tiledbdictpop isinstanceArrayopenr sparse ValueErrornattr TypeErrorrr iswritablerlenrr from_array) uri attributechunksstorage_optionskwargsr tiledb_configrtdbs r from_tiledbr/ s->MMM#-tvvM   E4 ( (C#v|$$Nkk#ImkMM zCABBB 0 : a  >   **/I ~B@AAA  -(--F v;;#*/ ) ) ) ) ?3[3-> ? ? ??rTFc 6ddl}|p t}|p|dd}tj|jst dt|tr%d|jD} |j ||f| ||d|} nVt||j r2|} |j | j kr|j | j kst dnt d| j r| jst d || d || S) aSave array to the TileDB storage format Save 'array' using the TileDB storage manager, to any TileDB-supported URI, including local disk, S3, or HDFS. See https://docs.tiledb.io for more information about TileDB. Parameters ---------- darray: dask.array A dask array to write. uri: Any supported TileDB storage location. storage_options: dict Dict containing any configuration options for the TileDB backend. see https://docs.tiledb.io/en/stable/tutorials/config.html compute, return_stored: see ``store()`` key: str or None Encryption key Returns ------- None Unless ``return_stored`` is set to ``True`` (``False`` by default) Notes ----- TileDB only supports regularly-chunked arrays. TileDB `tile extents`_ correspond to form 2 of the dask `chunk specification`_, and the conversion is done automatically for supported arrays. Examples -------- >>> import dask.array as da, tempfile >>> uri = tempfile.NamedTemporaryFile().name >>> data = da.random.random(5,5) >>> da.to_tiledb(data, uri) >>> import tiledb >>> tdb_ar = tiledb.open(uri) >>> all(tdb_ar == data) True .. _chunk specification: https://docs.tiledb.io/en/stable/tutorials/tiling-dense.html .. _tile extents: http://docs.dask.org/en/latest/array-chunks.html rNrz^Attempt to save array to TileDB with irregular chunking, please call `arr.rechunk(...)` first.cg|] }|d S)r)r cs r zto_tiledb..s...1!A$...r)r rrz>Target TileDB array layout is not compatible with source arrayzc'uri' must be string pointing to supported TileDB store location or an open, writable TileDB array.z-Target TileDB array is not open and writable.F)lockcompute return_stored)rrrr_check_regular_chunksr*r"rstr empty_likerdtyperisopenr%store) darrayr(r6r7r+rr,rr-r*r.s r to_tiledbr?JskvMMM#-tvvM /""5$//C  %fm 4 4  >   #s .. ...f  $]  GM   C & &  **1H1HP 2I  1   JJ3>JHIII <<% < V VVr)NNN)TFNN) __future__r dask.arrayrrr/r?r2rr rBs""""""GGG <@<@<@<@D  ]W]W]W]W]W]Wr