a uDf@sddlmZddlmZddlmZddlZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZdd lmZd dZddZddZe d*ddZeeZe d+ddZddZ e j!dddee ej"ddZ#e j!dddee ej$ddZ%e j!dd dee ej&ddZ'eeZ(e(ej'ej&d!Z&e j!dd"dee ej)Z*e(ej+ej)d!Z,e*j-dure*j-.d#d$e*_-e*j-.d%d&e*_-d'd"Z+d(d)Z)e*j-e+_-e,j-e)_-dS),) annotationspartial)productN)curry)array_creation_dispatch)Arraynormalize_chunks)meta_from_array)tokenize) blockwise)ArrayChunkShapeDep)funcnamecCst|tjr|}t|ttfs(|f}|dd}|dd}|dd}|durl||g|Ri|j}t|}t|||d}|pt |dt ||||||}|||||dS)Nnamechunksautodtyper-)shaperkwargsrr) isinstancenpZndarraytolisttuplelistpoprr rr )funcargsrrrrrr\/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/dask/array/wrap.py_parse_wrap_argss(      r!c Osd|vr |d|dd}}n |d}t|trcz"wrap_func_like..csg|]}qSrr)r-_)rrr r.fr/c3s,|]$\}}tfdi|fVqdS)rNr)r-ks)rrrrr ir/z!wrap_func_like..)r$) r r)rr!rr enumeratezipdictrZastype)rrrxr$rr*rrkeysZshapeskwir2valsZdskr)rrrrr wrap_func_likeTs$r<cKsb|durt||fi|}nt||fi|}d}|jdur^|d|ji|j|_d|j|_|S)Nz Blocked variant of %(name)s Follows the signature of %(name)s exactly except that it also features optional keyword arguments ``chunks: int, tuple, or dict`` and ``name: str``. Original signature follows below. rZblocked_)r__doc____name__)Z wrap_funcr func_likerftemplaterrr wrapps  rBrcOs2|dkr dnd}t||g|Rd|i||S)Nrr#r)rZ broadcast_to)rrr$rrZ null_shaperrr _broadcast_trick_innersrCcCst|}|j|_|j|_|S)a Provide a decorator to wrap common numpy function with a broadcast trick. Dask arrays are currently immutable; thus when we know an array is uniform, we can replace the actual data by a single value and have all elements point to it, thus reducing the size. >>> x = np.broadcast_to(1, (100,100,100)) >>> x.base.nbytes 8 Those array are not only more efficient locally, but dask serialisation is aware of the _real_ size of those array and thus can send them around efficiently and schedule accordingly. Note that those array are read-only and numpy will refuse to assign to them, so should be safe. )rCr=r>)rinnerrrr broadcast_tricksrEnumpyones)backendrZf8rzerosempty)r?fullz*array([0.1, 0.1, 0.1, 0.1, 0.1, 0.1])z%array([0.1, 0.1, 0.1, 0.1, 0.1, 0.1])z >>> np.full_like(y, [0, 0, 255])zB>>> np.full_like(y, [0, 0, 255]) # doctest: +NORMALIZE_WHITESPACEcOsjt|dkr$tdt|jd|dddurVt|drJ|j|d<n t||d<t|||d|S)Nr$fill_value must be scalar. Received instead.r)r fill_value) rndim ValueErrortyper>r)hasattrr_full)rrNrrrrr rKs   cOs8t|dkr$tdt|jdt|||d|S)NrrLrM)arN)rrOrPrQr> _full_like)rTrNrrrrr full_likesrV)N)r)/ __future__r functoolsr itertoolsrrFrZtlzrZdask.array.backendsrZdask.array.corerr Zdask.array.utilsr Z dask.baser Zdask.blockwiser r(Z dask.layersr Z dask.utilsrr!r,r<rBwrCrEZregister_inplaceZ ones_likerGZ zeros_likerIZ empty_likerJZw_likerVrSrKrUr=replacerrrr sx          $