Cdt*ddlmZddlZddlmZddlmZddlmZddlm Z m Z ddl Z ddl mZddlmZdd lmZdd lmZmZmZmZmZmZmZmZmZdd lmZmZdd l m!Z!dd l"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.d;ddZ3dZ4 d?d Z5ej6d!dddd"d#Z7e-e dd$d%d&Z8e9dfd'Z:ddde;fd(Z<e-e d@d)Z=e-e dAd+Z>e-e dde;dfdd,d-Z?e-e dBd.Z@e-e d;d/ZAe-e d0ZBd1ZCd2ZDd3ZEd4ZFd5ZGd6ZHd7ZId8ZJe-e dCd:ZKdS)D) annotationsN)Sequence)partial)product)IntegralNumber)sliding_window)chunk)array_creation_dispatch) Arrayasarrayblock blockwisebroadcast_arrays broadcast_to concatenatenormalize_chunksstack) greater_equalrint)meta_from_array)emptyfulloneszeros)tokenize)HighLevelGraph) cached_cumsum derived_from is_cupy_typexr backend str | Nonec b|p tj}tj|}|j|fi|S)aMove an Array collection to a new backend Parameters ---------- x : Array The input Array collection. backend : str, Optional The name of the new backend to move to. The default is the current "array.backend" configuration. Returns ------- dask.Array A new Array collection with the backend specified by ``backend``. )r r"dispatch to_backend)r!r"kwargsbackend_entrypoints 3lib/python3.11/site-packages/dask/array/creation.pyr&r&!s?$808G09'BB (  ( 5 5f 5 55CcDt|d}t|||\}}tj|r5|t tj|p|jSt||p|j||||j S)a Return a new array with the same shape and type as a given array. Parameters ---------- a : array_like The shape and data-type of `a` define these same attributes of the returned array. dtype : data-type, optional Overrides the data type of the result. order : {'C', 'F'}, optional Whether to store multidimensional data in C- or Fortran-contiguous (row- or column-wise) order in memory. chunks : sequence of ints The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. name : str, optional An optional keyname for the array. Defaults to hashing the input keyword arguments. shape : int or sequence of ints, optional. Overrides the shape of the result. Returns ------- out : ndarray Array of uninitialized (arbitrary) data with the same shape and type as `a`. See Also -------- ones_like : Return an array of ones with shape and type of input. zeros_like : Return an array of zeros with shape and type of input. empty : Return a new uninitialized array. ones : Return a new array setting values to one. zeros : Return a new array setting values to zero. Notes ----- This function does *not* initialize the returned array; to do that use `zeros_like` or `ones_like` instead. It may be marginally faster than the functions that do set the array values. Fnamedtyper0orderchunksr.meta) r _get_like_function_shapes_chunksnpisnanany map_blocksr empty_liker0r_metaar0r2r3r.shapes r)r:r::sX A4QFFME6 xN||GBM%:J17LLLMMM    W    r*cDt|d}t|||\}}tj|r5|t tj|p|jSt||p|j||||j S)a Return an array of ones with the same shape and type as a given array. Parameters ---------- a : array_like The shape and data-type of `a` define these same attributes of the returned array. dtype : data-type, optional Overrides the data type of the result. order : {'C', 'F'}, optional Whether to store multidimensional data in C- or Fortran-contiguous (row- or column-wise) order in memory. chunks : sequence of ints The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. name : str, optional An optional keyname for the array. Defaults to hashing the input keyword arguments. shape : int or sequence of ints, optional. Overrides the shape of the result. Returns ------- out : ndarray Array of ones with the same shape and type as `a`. See Also -------- zeros_like : Return an array of zeros with shape and type of input. empty_like : Return an empty array with shape and type of input. zeros : Return a new array setting values to zero. ones : Return a new array setting values to one. empty : Return a new uninitialized array. Fr-r/r1) r r5r6r7r8r9r ones_liker0rr;r<s r)r@r@xsJ A4QFFME6 xM||GBL9I!'KKKLLL    W    r*cDt|d}t|||\}}tj|r5|t tj|p|jSt||p|j||||j S)a Return an array of zeros with the same shape and type as a given array. Parameters ---------- a : array_like The shape and data-type of `a` define these same attributes of the returned array. dtype : data-type, optional Overrides the data type of the result. order : {'C', 'F'}, optional Whether to store multidimensional data in C- or Fortran-contiguous (row- or column-wise) order in memory. chunks : sequence of ints The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. name : str, optional An optional keyname for the array. Defaults to hashing the input keyword arguments. shape : int or sequence of ints, optional. Overrides the shape of the result. Returns ------- out : ndarray Array of zeros with the same shape and type as `a`. See Also -------- ones_like : Return an array of ones with shape and type of input. empty_like : Return an empty array with shape and type of input. zeros : Return a new array setting values to zero. ones : Return a new array setting values to one. empty : Return a new uninitialized array. Fr-r/r1) r r5r6r7r8r9r zeros_liker0rr;r<s r)rBrBsJ A4QFFME6 xN||GBM%:J17LLLMMM    W    r*c Ht|d}t|||\}}tj|r6|t tj|p|j|St|||p|j||||j S)a Return a full array with the same shape and type as a given array. Parameters ---------- a : array_like The shape and data-type of `a` define these same attributes of the returned array. fill_value : scalar Fill value. dtype : data-type, optional Overrides the data type of the result. order : {'C', 'F'}, optional Whether to store multidimensional data in C- or Fortran-contiguous (row- or column-wise) order in memory. chunks : sequence of ints The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. name : str, optional An optional keyname for the array. Defaults to hashing the input keyword arguments. shape : int or sequence of ints, optional. Overrides the shape of the result. Returns ------- out : ndarray Array of `fill_value` with the same shape and type as `a`. See Also -------- zeros_like : Return an array of zeros with shape and type of input. ones_like : Return an array of ones with shape and type of input. empty_like : Return an empty array with shape and type of input. zeros : Return a new array setting values to zero. ones : Return a new array setting values to one. empty : Return a new uninitialized array. full : Fill a new array. Fr-r/r1) r r5r6r7r8r9r full_liker0rr;)r= fill_valuer2r0r3r.r>s r)rDrDsR A4QFFME6 xY||GBL9I!'KKKZXXX    W   r*c8||j}||j}n|d}||fS)zb Helper function for finding shapes and chunks for *_like() array creation functions. Nauto)r>r3)r=r3r>s r)r5r5"s1  } >XF  &=r*2TFrGc t|}|tjdddj}t ||f|}||z }|r|dz n|}|dkrd}t ||z } dt ||||||fz} i} |} t|dD]E\} }|r|dz n|}| || zz}ttj||| ||f}| | |zz} || | | f<F|rt| | ||| fSt| | ||S)a- Return `num` evenly spaced values over the closed interval [`start`, `stop`]. Parameters ---------- start : scalar The starting value of the sequence. stop : scalar The last value of the sequence. num : int, optional Number of samples to include in the returned dask array, including the endpoints. Default is 50. endpoint : bool, optional If True, ``stop`` is the last sample. Otherwise, it is not included. Default is True. retstep : bool, optional If True, return (samples, step), where step is the spacing between samples. Default is False. chunks : int The number of samples on each block. Note that the last block will have fewer samples if `num % blocksize != 0` dtype : dtype, optional The type of the output array. Returns ------- samples : dask array step : float, optional Only returned if ``retstep`` is True. Size of spacing between samples. See Also -------- dask.array.arange Nrr/z linspace-)endpointr0) intr6linspacer0rfloatr enumeraterr r )startstopnumrKretstepr3r0range_divstepr.dsk blockstartibsbs_space blockstoptasks r)rMrM0s`N c((C } Aq!$$* fsfE : : :F E\F (377SC axx ==3 D 5$Xvu"MNN ND CJ6!9%%  2%-2662(T/2 ENXU C C C       4"9- T1I5S$e444d::S$e4444r*numpy)r3liker0cdt|dkr d}|d}d}nOt|dkr|d}|d}d}n)t|dkr|\}}}ntdttt j||z |z d}|t |nd} |"t j|||r||zn|j}t||f|}|r7tdd | zd t|||||fz} i} d} t|dD]E\} }|| |zz}|| |z|zz}ttj| |||||f}|| | | f<| |z } Ft!| | ||| S) a Return evenly spaced values from `start` to `stop` with step size `step`. The values are half-open [start, stop), so including start and excluding stop. This is basically the same as python's range function but for dask arrays. When using a non-integer step, such as 0.1, the results will often not be consistent. It is better to use linspace for these cases. Parameters ---------- start : int, optional The starting value of the sequence. The default is 0. stop : int The end of the interval, this value is excluded from the interval. step : int, optional The spacing between the values. The default is 1 when not specified. The last value of the sequence. chunks : int The number of samples on each block. Note that the last block will have fewer samples if ``len(array) % chunks != 0``. Defaults to "auto" which will automatically determine chunk sizes. dtype : numpy.dtype Output dtype. Omit to infer it from start, stop, step Defaults to ``None``. like : array type or ``None`` Array to extract meta from. Defaults to ``None``. Returns ------- samples : dask array See Also -------- dask.array.linspace rJrzT arange takes 3 positional arguments: arange([start], stop, [step]) Nr/z"Unexpected keyword argument(s): %s,zarange-r_)r0r4)len TypeErrorrLmaxr6ceilraranger0rjoinkeysrrOrr r )r3r_r0argsr'rPrQrVrRr4r.rW elem_countrYrZrXr\r]s r)riri}sN 4yyA~~Aw TaQAw Ta tTT    c"'4%<4/00!44 5 5C$($4?4 $D } %S'BtczzdCCI fsfE : : :F Xzmeshgrid..s ! ! !'!** ! ! !r*c6g|]}|Sru)flattenrvs r)ryzmeshgrid..s " " "!!))++ " " "r*rnrJr) boolNotImplementedErrorrf ValueErrorrerangeslicetupleappendr)rorpxir'gridrYsrs r)meshgridrsi &\\F !"8999 DBCCC|##>??? ! !b ! ! !B " "r " " "B4CGGaKK!ube 1r!u D 3r77^^ GGtf T{{! !HH qE!H A '&4CGGaKK7DGQa Kr*c t|}tj|}t|||}t |t |krt dg}t t |D]4}|t|||||f5g}t|r t|ddi}|rt|}n&tt |f|z|d|z}|S)a Implements NumPy's ``indices`` for Dask Arrays. Generates a grid of indices covering the dimensions provided. The final array has the shape ``(len(dimensions), *dimensions)``. The chunks are used to specify the chunking for axis 1 up to ``len(dimensions)``. The 0th axis always has chunks of length 1. Parameters ---------- dimensions : sequence of ints The shape of the index grid. dtype : dtype, optional Type to use for the array. Default is ``int``. chunks : sequence of ints, str The size of each block. Must be one of the following forms: - A blocksize like (500, 1000) - A size in bytes, like "100 MiB" which will choose a uniform block-like shape - The word "auto" which acts like the above, but uses a configuration value ``array.chunk-size`` for the chunk size Note that the last block will have fewer samples if ``len(array) % chunks != 0``. Returns ------- grid : dask array r>r0z)Need same number of chunks as dimensions.r0r3rprsrJ) rr6r0rrer~rrriallrrr) dimensionsr0r3rrYrs r)indicesrs>z""J HUOOE fJe D D DF :#f++%%DEEE B 3z?? # #KK &AeVAYLIIIJJJJ D :,+d++ YT{{c*oo'*4E$QW-XXX Kr*ci}||}|t}t|ttfst dt |||f|\}}|d}t |||||}d|z} t|D]t\} } t|D]_\} } | | z dz |z|cxkr| | z dz|zkr$nn!tj | | || | z |zz |f|| | | f<Htj | | f|f|| | | f<`ut|| ||f||f|S)a Return a 2-D Array with ones on the diagonal and zeros elsewhere. Parameters ---------- N : int Number of rows in the output. chunks : int, str How to chunk the array. Must be one of the following forms: - A blocksize like 1000. - A size in bytes, like "100 MiB" which will choose a uniform block-like shape - The word "auto" which acts like the above, but uses a configuration value ``array.chunk-size`` for the chunk size M : int, optional Number of columns in the output. If None, defaults to `N`. k : int, optional Index of the diagonal: 0 (the default) refers to the main diagonal, a positive value refers to an upper diagonal, and a negative value to a lower diagonal. dtype : data-type, optional Data-type of the returned array. Returns ------- I : Array of shape (N,M) An array where all elements are equal to zero, except for the `k`-th diagonal, whose values are equal to one. Nzchunks must be an int or stringrrzeye-rJ)r>r3r0) rN isinstancerLstrr~rrrOr6eyerr )Nr3Mkr0rvchunkshchunkstokenname_eyerYvchunkjhchunks r)rr2s}> Cy  } fsCj ) )<:;;;'q!fEJJJGW QZF Q1e , ,E~Hw'' J J 6"7++ J JIAvA V#q@@@@QUQY&,@@@@@@FQ&(( 'HaN##(*x&&1A5&IHaN## J hq!fff5EU S S SSr*ct|tjs4t|tst dt |dt ||zt||jdkrdnd}t|tjs%t|drt|ts|jdkrFt|}|j d|zf|j d|zff}ddftj ||fi}ny|jdkr_td| }t|j d|j d|z }||z }|dkrdn|ff}dftj ||fi}ntdt||| S|jdkr|jdkrtd|dkr|jd|jdkrdfd t#|D}t'j||g } t| |jdf| St+||S|dkr|jd} |} i}t#| D]r\} }t#| D]]\} }| | f}| | krtj | | f||<(tj||ff||<t/tj||f |f||<^st'j||g } t| | | f| S|dkr%t3t|d|g|dggd S|dkr't3t|| dgd| ggd SdS)Nz+v must be a dask array or numpy array, got zdiag-rJra__array_function__rrzArray must be 1d or 2d onlyr4cDi|]\}}|ftj||fSru)r6diag)rwrYrowr.s r) zdiag..s;17Cq BGSV,r* dependencies)r>constant)mode)rr6ndarrayr rftyperrndimhasattrabsr>rrgminr~r3rO __dask_keys__rfrom_collectionsdiagonalrrrBpad)vrr4mr3rWkdiag_row_startkdiag_row_stop len_kdiaggraph chunks_1dblocksrYrnkeyr.s @r)rros a $ $QZ5-A-AQOd1ggOOPPP Xa^^ #D 116Q;;aaA 6 6D!RZ  3'((31;Au1E1E3 6Q;;AAwqzA~'!'!*q.):;F!Q<"'1a1CC Vq[[!!aRjjO QWQZ!^<> !AvvHQK ""i(( L LDAq!),, L L1Ql66 "3CHH "1a&1CH ' aV D D DdKCHH  L/cLLLUD9i"8tDDDD Q477aVaV,:>>>> Q477qb!Wq1"g.Z@@@@ r*rJc4 '(dt|||z}jdkrtdd}d}||dj}||dj}||krtdt|}||kr||}}| }t t j||hz }t tfd |D} t|} td | } td |} tj |j ||z } | | z }|d krt}tjrd dl}|}|j||d z(t#}| D]U't%'(fd t | D}t'|jj |dzf||f'zdz<Vt-| dz}t/||(|Stjj|}tj|d}d |d <tjj|}tj|d}d |d <tjj|}tjj|}|| k| |kz}|| k| |kz}||\}||\}t#}d }d}| j |krF| j |kr4j||j||}!} | ||z} | ||z} | d kr| n| }t| |!|z }"|"| z }#| D]W''d||fz'||dz z|fz'|dz dz}$'|fz}%tjjf|$z|||f||f|%z<X||#fz }|dz }|"||z} t|!| |z||z} | ||kr|dzn|}| ||kr|dzn|}| j |kr| j |k4|j|||fz(t=j||g}&t-| dz}t/|&|(|S)Nz diagonal-raz1diag requires an array of at least two dimensionsc|dkr8||z}|dkr+d}tj|||||}|S)Nrz6{}: axis {} is out of bounds for array of dimension {})r6 AxisErrorformat)axisr.rtmsgs r) _axis_fmtzdiagonal.._axis_fmtsL !88t A1uuNl3::dD$#?#?@@@D r*ct|}||||t|SN)listpopr)r3axis1axis2s r)pop_axeszdiagonal..pop_axess=f 5 5V}}r*rrz"axis1 and axis2 cannot be the samec3LK|]}tj|VdSr)r numblocks)rwrYr=s r) zdiagonal..s1!K!KA% A"7"7!K!K!K!K!K!Kr*rrc3@K|]}||VdSrru)rwrfree_idx out_chunkss r)rzdiagonal..sB59 4 $0r*r/rrJrrur) rrr~r setrrrrergrr>r6r r;cupyr3dictrrrr0rr cumsumrollrirrr.rr))r=offsetrrr.rrr free_axes free_indices ndims_freerkdiag_col_startrrxprrWr>r4 row_stops_ row_starts col_stops_ col_starts row_blockid col_blockid row_filter col_filterIJrY kdiag_chunksnrowsncols kdiag_row_end kdiag_len input_idx output_idxrrrs)` @@r)rrst !VUE:: :DvzzLMMM IeWaf - -E IeWaf - -E ~~=>>> AA u}}eu GE!&MM""eU^3I!K!K!K!K!K!K!KLMMLYJ!aRjjO!QiiO!);<.Is, L L LAF1E! , , , L L Lr*rprsrur fromfunction) rNrrrreziprsum itertoolsrepeatr) funcr3r>r0r'indsarrsrlress ` r)rrBs NUE fe5 9 9 9F s5zz"" # #D L L L LUF9K9K L L LD T )D ) )D s4)$//00" 5 5D D$ F F F F^ Fv F FC Jr*cjdkrdntdt|tstdj cxkrdkrnn jz n)dcxkrjdz ksnt dz|dkr3t fdt jDS|dkrStjd}g}td |D]m\}}tj ||| d}td |D].\}} || kr#| t|| /ntdddfd |D}fd |D} g} | D]} t| j} t!| dksJ| d|zf| <t | } | tj|| | j }| |t)|  S)NrJrz!Must supply an integer axis valuez%Only integer valued repeats supportedzaxis(=%d) out of boundsc3`K|](}|krtdntdV)dS)Nr)r)rwdrs r)rzrepeat..ds;UU!a4iiuT{{{U1XXUUUUUUr*T) initial_zeroracHg|]}fz|fzfjz dz zzSr)r)rwrr= all_slicers r)ryzrepeat..qsKLM tqd"i\QVd]Q5F%GGr*c g|] }| Sruru)rwslcr=s r)ryzrepeat..us & & &QsV & & &r*)rr3r0r)rr}rrr~rrrr3r r6rMroundrrrrer9rr0r)r=repeatsrcchunksslicesc_startc_stoplsls_startls_stopslabsoutslabr3resultrs` ` @r)rrSs | 6Q;;DD%&IJJ J gx ( (K!"IJJJ w$  $ $ $ $ $!&1* $ $ $ $2T9:::!||UUUUuQV}}UUUUUVV AAHTN>>>G F)!W5588 [&' 2 2 8 8 ; ;!/2!6!6 8 8 Hg7"" eHg66777 8dD$''IQWF ' & & &v & & &E Cdk""6$<  A%%%%t Q'13t v IwT& !   6 s & & &&r*c: t|}n#t$r|f}YnwxYwtd|Drtdt |}t |r#|dddD]}||gz} t |St|}||jkrd|j|z z|z}|j|krd||jz z|j z}n|j }tdt||D}t||j S)Nc3"K|] }|dkV dS)rNrurwrYs r)rztile..s&  Q1q5      r*z Negative `reps` are not allowed.rc3&K|] \}}||zV dSrru)rwrrs r)rztile..s*881a!e888888r*r) rrfr8r~r rrrerr>rrr0)Arepstuprnreprr> shape_outs r)tiler#sADkk g  #   =;<<< A 3xx"I  Ds AAQxx CA16zzafqj!C'vzzAF #ag-88E388888I y 0 0 00s  ""c\t|tst|dddkr|j||ffz}nt|trGt d|Dr.t |dkr|j|d|dffz}nt|trFt |dkr3t d|Dr|jt|fz}n;t|tr}t ||jkret d|DrLt d|Dr3t d |Drtd |D}nt|trt |dkrrt|dtrWt |ddkr>t d |dDr|jt|dfz}ntd |S) Nrrc3@K|]}t|tVdSrrrrwpws r)rz#expand_pad_value..,;;2 2v&&;;;;;;r*rJrac3@K|]}t|tVdSrr&r's r)rz#expand_pad_value..r)r*c3@K|]}t|tVdSr)rrr's r)rz#expand_pad_value..s,==R 2x((======r*c3<K|]}t|dkVdSraN)rer's r)rz#expand_pad_value..s,332RA333333r*c3HK|]}td|DVdS)c3@K|]}t|tVdSrr&)rwws r)rz-expand_pad_value...s,66aJq&))666666r*N)rr's r)rz#expand_pad_value..s9KK266266666KKKKKKr*c34K|]}t|VdSrrr's r)rz#expand_pad_value..s(88%))888888r*c3@K|]}t|tVdSrr&r's r)rz#expand_pad_value..s,>>2 2v&&>>>>>>r*z6`pad_value` must be composed of integral typed values.) rrgetattrrrrrerrf)array pad_values r)expand_pad_valuer7sP)V$$R 64(H(HA(M(MJ9i"8!:: 9h''R ;;;;; ; ;R  NNa  J9Q<1">!@@ 9h''R  NNa   ;;;;; ; ; J% "2"2!44 9h''R  NNej ( ( ==9=== = = ) 33333 3 3 ) KKKKK K K ) 88i88888 9h'' R  NNa   y|X . .  !   " " >>1>>> > > #J% ! "5"5!77 PQQQ r*cft|jt|jg}t|jt|jg}|D]C}tdD]1}||||||<|||f|||<2Dd|D}d|D}||fS)zD Helper function for finding shapes and chunks of end pads. rac,g|]}t|Srur2)rwrs r)ryz)get_pad_shapes_chunks..///q%((///r*c,g|]}t|Srur2)rwrs r)ryz)get_pad_shapes_chunks..r:r*)rr>r3r)r5 pad_widthaxes pad_shapes pad_chunksrrYs r)get_pad_shapes_chunksr@s u{##T%+%6%67Ju|$$d5<&8&89J 22q 2 2A(|AJqM!  )! Q1JqM!   20/J///J//J///J z !!r*c|dz}t|j}|||<t|}tj|j}tj|||}tj|jD]d} t| } td| |<t| } tj|| |||dddd||| <e|S)z> Helper function to find the linear ramp for a chunk. rJrNr/) rr>rr6r0r:ndindexrrM) rPrQrRdimrVnum1r>r0rrYrs r)linear_ramp_chunkrEs 7D   EE#J %LLE HU[ ! !E ]5U ; ; ;F Z $ $OO GGt# !HHKa$EBBB122FvvvNq Mr*c  fd|D} tjD] t | f\}} g}|dkrAddlm |d } fd|D}dt |||D}n |dvr jtd gz jtd gzg}td d d |d <td d d |d  <d |D} fd |D}|dkrdt |||D}ni|dkr@|d } fdtt || | |D}n"|dkrfdt ||D}t|d |d g   S)zu Helper function for padding edges. Handles the cases where the only the values on the edge are needed. c8i|]\}}|t|Sru)r7)rwrrr5s r)rzpad_edge..s* G G G1a!%++ G G Gr*rr) asarray_safeconstant_valuescPg|]"}|tj#S))r_r0)rr0)rwrr5rHrs r)ryzpad_edge..sD Q_U%;%;6<PPPr*c8g|]\}}}t|||Srur)rwrrrs r)ryzpad_edge..s:Aq!Q1%%r*)edge linear_rampNrJrc,g|]}t|Srur2)rwsls r)ryzpad_edge.. s999%))999r*c g|] }| Sruru)rwrPrs r)ryzpad_edge..s:::&*:::r*rMc8g|]\}}}t|||SrurL)rwr=rrs r)ryzpad_edge..s:1a!Aq))r*rN end_valuesc xg|]6\}\}}}}|t|||jd|zdz 7S)rarJ)r3r0rCrV)r9rEr0)rwrYr=evr(rrrs r)ryzpad_edge..si   *>Ar2qLL) $l!eai!   r*rcFg|]\}}t|j|S))r>r0r3)r:r0)rwrrr5s r)ryzpad_edge..(s@Aq5QGGGr*r ) itemsrrr@dask.array.utilsrHrrrOr) r5r<rr'r>r? pad_arraysrI pad_slicesrSrHrrs ` @@@r)pad_edger[sH G G G  G G GF F 5:  5M5M!6vy1$!O!O Jf% :   5 5 5 5 5 5$%67:O(O "?J KKJJ, , , +t 5v{eDkk]7RSJ$T1d33JqM! $Rt44JqM! 99j999J::::z:::Jv~~#&z:z#J#J &&#L1!4      .7J IaL*MM..    W__ J77J jmVZ]C!LLL Mr*c |dvr@|dd}|dkrtd|dkrtdtj|jdzt }tj|jD]}g}g}t||j|D]_\} } } |d kr | d d d } | d kr_|dkr.| td | d d zd n| td | d d n| d krh|dkr4| t| | d z d z | d z d nP| t| | d z d d n"| td | d kr*|dvr&| td d d =| td at|}t|}|d krtd|D}|||||<t|}|S)z Helper function for padding boundaries with values in the array. Handles the cases where the padding is constructed from values in the array. Namely by reflecting them or tiling them to create periodic boundary constraints. >reflect symmetricr]evenoddz/`pad` does not support `reflect_type` of `odd`.zBunsupported value for reflect_type, must be one of (`even`, `odd`)rbr/wrapNrrJr)r]r^c3 K|] }d|z V dSr-rurs r)rzpad_reuse..bs&++!A++++++r*)getr}r~r6rrobjectrBr>rrrrrtolist) r5r<rr' reflect_typeridxselectorientrYrr(s r) pad_reuserk2s{ '''zz)V44 5 %&WXX X 6 ! !T Xej4'v 6 6 6Fz&,'',,Ci88 + +HAq"v~~"X1uu9$$MM%2a519d";";<<<<MM%beT":":;;;;Q9$$MM%BqE A q1ud"C"CDDDDMM%BqE 4">">???? eDkk***Avv$"::: eD$334444 eDkk****vv 6>>++s+++++CFmF+s 6==?? # #F Mr*c 6|dkrtdt||}tj|jdzt }tj|jD]}g}g}g}g} tt||j|j ||D]T\} \} } } }}| dkrv| | | td|dd| |d| |d| dkrz| | | t| |dz dd| |d| |d| td| | | | Vt|}t|}t|}t| } ||}|r|dkr||d }n;|d kr||d }n|d kr||d }t#||| }|d krMtj|jtjrt+|}||j}|||<t/|}|S)a6 Helper function for padding boundaries with statistics from the array. In cases where the padding requires computations of statistics from part or all of the array, this function helps compute those statistics as requested and then adds those statistics onto the boundaries of the array. medianz*`pad` does not support `mode` of `median`.rar/rJNrmaximumT)rkeepdimsmeanminimum)r3)r}r7r6rrrerBr>rOrr3rrrrgrprr issubdtyper0integerrrrrf)r5r<r stat_lengthrrhr=ri pad_shaper?rrYrrr0l result_idxs r) pad_statsrxks x!"NOOO"5+66K Xej4'v 6 6 6Fz&,'',!,!  "+ U[%, ; G G# #  % % A1aA1uu A eD!A$55666  1&&&!!!A$''''Q A eA!HdD99:::  1&&&!!!A$'''' eDkk***  ###!!!$$$$T{{v)$$ :&& 6]  <y  '^^^EE '__$_FF ""'^^^EE %j)JOOOJv~~=bj992!%j!1!1J'..u{;;  s 6==?? # #F Mr*ctj|}tj|jd||j|dzdzD]?}|d|t dfz||dz}||||||||<@|S)NrJ)r6r:rBr>r)r5pad_funciaxis_pad_widthiaxispad_func_kwargsrrYs r)wrapped_pad_funcr~s ]5 ! !F Z FUF+ek%!)++.FF G GPP fufIt &566 2HU1XOOq Mr*c ht||dd}|j}t|jD]}||d||j||dzfz||dzdz}|td|j|||||}||}|S)a9 Helper function for padding boundaries with a user defined function. In cases where the padding requires a custom user defined function be applied to the array, this function assists in the prepping and application of this function to the Dask Array to construct the desired boundaries. rr)rINrJr)r.r0rzr{r|r}) r[r3rrrechunkr>r9r~r0)r5r<rr'rr3rs r)pad_udfrseY A F F FF ]F 6;  (( 2A2J&,q1q5y13 3fQUWWo E  "" ,%aL"#  '' Mr*rc 6t|}t||}t|rt|||fi|Sgggdgdgdgdgdgdgdgdgd } t |t ||z }n&#t $r}t d|d|d}~wwxYw|r#t d|||d vrD|dtd |j D}t||||S|d kr%| dd t|||fi|S|d kr%| dd t|||fi|S|dvrt|||S|dvrt|||fi|Std)NrIrSrtrg) rrMrbrrNrnrprmrqr]r^zmode 'z' is not supportedz/unsupported keyword arguments for mode '{}': {}>rprmrnrqc3K|]}||fV dSrru)rwrs r)rzpad..s&5R5Rq!f5R5R5R5R5R5Rr*rrrN>rMr)r]r^rb unreachable)r r7callablerrKeyErrorr~rrdrr>rx setdefaultr[rk RuntimeError)r5r<rr'allowed_kwargsunsupported_kwargsrxrts r)rrs6 ENNE  22I~~9ui88888&'$~!? /!?"#$%  NC [[3~d/C+D+DD CCC:$:::;;BC  = D D(      777jj5R5Rek5R5R5R0R0RSS  4===   +Q///y$99&999   ,***y$99&999 " " "y$/// 1 1 1 4::6::: } % %%s%A;; BBBr)r!r r"r#)Nr+NNN)r+NNNN)rHTFrGNr)rrrJ)rGNN)r)L __future__rrcollections.abcr functoolsrrnumbersrrr^r6tlzr dask.arrayr dask.array.backendsr dask.array.corer r rrrrrrrdask.array.ufuncrrrXrdask.array.wraprrrr dask.baserdask.highlevelgraphr dask.utilsrrr r&r:r@rBrDr5rMregister_inplacerirrLrrNrrrrrrr#r7r@rEr[rkrxr~rrrur*r)rs""""""$$$$$$$$$$$$$$777777                      10000000,,,,,,444444444444......@@@@@@@@@@666662;;;;|4444n4444n9999x   MQJ5J5J5J5Z*)'22d$U<U<U<U<32U