ó ¦–Õ\c@sAddlmZd„Zdddd„Zeedd„ZdS(i(tcorecs,|j‰t‡fd†tˆjƒDƒƒS(Nc3s$|]}ˆjj|ƒjVqdS(N(tdomaintdimttile(t.0ti(tschema(s3lib/python2.7/site-packages/dask/array/tiledb_io.pys s(Rtlisttrangetndim(t tiledb_array((Rs3lib/python2.7/site-packages/dask/array/tiledb_io.pyt_tiledb_to_chunkss c Ks%ddl}|ptƒ}|jddƒ}t||jƒrH|}n!|j|d|d|d|ƒ}|jjr„t dƒ‚n|sÃ|jj dkr«t dƒ‚qÃ|jj d ƒj }n|jrÛt d ƒ‚n|pêt|ƒ}t|ƒ|jjks t‚tj||d d |ƒS( s”Load 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 -------- >>> # create a tiledb array >>> import tiledb, numpy as np, tempfile # doctest: +SKIP >>> uri = tempfile.NamedTemporaryFile().name # doctest: +SKIP >>> tiledb.from_numpy(uri, np.arange(0,9).reshape(3,3)) # doctest: +SKIP >>> # read back the array >>> import dask.array as da # doctest: +SKIP >>> tdb_ar = da.from_tiledb(uri) # doctest: +SKIP >>> tdb_ar.shape # doctest: +SKIP (3, 3) >>> tdb_ar.mean().compute() # doctest: +SKIP 4.0 iÿÿÿÿNtkeytattrtconfigs&Sparse TileDB arrays are not supportedisOkeyword 'attribute' must be providedwhen loading a multi-attribute TileDB arrayis%TileDB array must be open for readingtnames tiledb-%s(ttiledbtdicttpoptNonet isinstancetArraytopenRtsparset ValueErrortnattrt TypeErrorR Rt iswritableR tlenR tAssertionErrorRt from_array( turit attributetchunkststorage_optionstkwargsRt tiledb_configR ttdb((s3lib/python2.7/site-packages/dask/array/tiledb_io.pyt from_tiledb s"#  !  c Kseddl}|ptƒ}|jddƒ}tj|jƒsNtdƒ‚nt|t ƒr¹g|jD]} | d^qg} |jddƒ}|j ||d| d|d||} nlt||j ƒr|} |j | j ko|j | j ko|j| jks%tdƒ‚q%n td ƒ‚| jo4| jsFtd ƒ‚n|j| d td |d |ƒS(sÔSave 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()`` 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 # doctest: +SKIP >>> uri = tempfile.NamedTemporaryFile().name # doctest: +SKIP >>> data = da.random.random(5,5) # doctest: +SKIP >>> da.to_tiledb(data, uri) # doctest: +SKIP >>> import tiledb # doctest: +SKIP >>> tdb_ar = tiledb.open(uri) # doctest: +SKIP >>> all(tdb_ar == data) # doctest: +SKIP True .. _chunk specification: https://docs.tiledb.io/en/stable/tutorials/tiling-dense.html .. _tile extents: http://docs.dask.org/en/latest/array-chunks.html iÿÿÿÿNR s^Attempt to save array to TileDB with irregular chunking, please call `arr.rechunk(...)` first.iRRs>Target TileDB array layout is not compatible with source arraysc'uri' must be string pointing to supported TileDB store location or an open, writable TileDB array.s-Target TileDB array is not open and writable.tlocktcomputet return_stored(RRRRRt_check_regular_chunksR!RRtstrt empty_likeRtshapetdtypeR tisopenRtstoretFalse( tdarrayRR(R)R"R#RR$R tcR!R%((s3lib/python2.7/site-packages/dask/array/tiledb_io.pyt to_tiledbIs(2  * N(tRR RR&tTrueR1R4(((s3lib/python2.7/site-packages/dask/array/tiledb_io.pyts   ?