`iL&ddlZddlZddlZddlmZddlmZddlmZdZejZddZ ddZ ddZ dd Z dd Z dd Zdd ZddZ ddZejd ddZejd ddZejd ddZejd ddZejd ddZdS) N)_spline_prefilter_core)_spline_kernel_weights)_utilzS // workaround for HIP: line begins with #include #include c g}|d|dkrd|nd}t|D]!}|d|d|d|d"|S) a Extract target coordinate from coords array (for map_coordinates). Notes ----- Assumes the following variables have been initialized on the device:: coords (ndarray): array of shape (ncoords, ndim) containing the target coordinates. c_j: variables to hold the target coordinates computes:: c_j = coords[i + j * ncoords]; ncoords is determined by the size of the output array, y. y will be indexed by the CIndexer, _ind. Thus ncoords = _ind.size(); z ptrdiff_t ncoords = _ind.size();r + (W) W c_z = coords[i + z * ncoords];)appendrangendimnprepadopsprejs w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/ndimage/_interp_kernels.py_get_coord_maprs( CJJ1222 '! 7   C 4[[22 1 1111(+111 2 2 2 2 Jcg}|dkrd|nd}t|D]'}|d|d|d|d|d|d (|S) aCompute target coordinate based on a shift followed by a zoom. This version zooms from the center of the edge pixels. Notes ----- Assumes the following variables have been initialized on the device:: in_coord[ndim]: array containing the source coordinate zoom[ndim]: array containing the zoom for each axis shift[ndim]: array containing the zoom for each axis computes:: c_j = zoom[j] * (in_coord[j] - shift[j]) rrrr  = zoom[] * ((W)in_coord[ ] - shift[z])r r r r s r_get_coord_zoom_and_shiftr/s$ C '! 7   C 4[[DD C CCCC()CC56CC:=CCC D D D D Jrc g}|dkrd|nd}t|D]$}|d|d|d|d|d %|S) aCompute target coordinate based on a shift followed by a zoom. This version zooms from the outer edges of the grid pixels. Notes ----- Assumes the following variables have been initialized on the device:: in_coord[ndim]: array containing the source coordinate zoom[ndim]: array containing the zoom for each axis shift[ndim]: array containing the zoom for each axis computes:: c_j = zoom[j] * (in_coord[j] - shift[j] + 0.5) - 0.5 rrrr rrz] - shift[j] + 0.5) - 0.5r rr s r_get_coord_zoom_and_shift_gridrIs$ C '! 7   C 4[[NN M MMMM()MMDGMMM N N N N Jrc g}|dkrd|nd}t|D]$}|d|d|d|d|d %|S) a|Compute target coordinate based on a zoom. This version zooms from the center of the edge pixels. Notes ----- Assumes the following variables have been initialized on the device:: in_coord[ndim]: array containing the source coordinate zoom[ndim]: array containing the zoom for each axis computes:: c_j = zoom[j] * in_coord[j] rrrr r] * (W)in_coord[]r rr s r_get_coord_zoomr!cs" C '! 7   C 4[[55 4 4444'(44+.444 5 5 5 5 Jrc g}|dkrd|nd}t|D]$}|d|d|d|d|d %|S) aCompute target coordinate based on a zoom (grid_mode=True version). This version zooms from the outer edges of the grid pixels. Notes ----- Assumes the following variables have been initialized on the device:: in_coord[ndim]: array containing the source coordinate zoom[ndim]: array containing the zoom for each axis computes:: c_j = zoom[j] * (in_coord[j] + 0.5) - 0.5 rrrr rrz] + 0.5) - 0.5r rr s r_get_coord_zoom_gridr#|s" C '! 7   C 4[[CC B BBBB()BB9<BBB C C C C Jrc g}|dkrd|nd}t|D]$}|d|d|d|d|d %|S) aCCompute target coordinate based on a shift. Notes ----- Assumes the following variables have been initialized on the device:: in_coord[ndim]: array containing the source coordinate shift[ndim]: array containing the zoom for each axis computes:: c_j = in_coord[j] - shift[j] rrrr z = (W)in_coord[rr r rr s r_get_coord_shiftr%s C '! 7   C 4[[66 5 5555()55,/555 6 6 6 6 Jrc <g}|dkrd|nd}|dz}t|D]w}|d|dt|D]'}|d|d||z|zd |d (|d|d||z|zd |d x|S) a{Compute target coordinate based on a homogeneous transformation matrix. The homogeneous matrix has shape (ndim, ndim + 1). It corresponds to affine matrix where the last row of the affine is assumed to be: ``[0] * ndim + [1]``. Notes ----- Assumes the following variables have been initialized on the device:: mat(array): array containing the (ndim, ndim + 1) transform matrix. in_coords(array): coordinates of the input For example, in 2D: c_0 = mat[0] * in_coords[0] + mat[1] * in_coords[1] + aff[2]; c_1 = mat[3] * in_coords[0] + mat[4] * in_coords[1] + aff[5]; rrrz W c_z = (W)0.0;z c_z += mat[r];r r r)rrrrncolrks r_get_coord_affiner+s?( C '! 7   C !8D 4[[66 !!!! " " "t A AA JJ@@@(Q,@@89@@@ A A A A 555(T/55,/555 6 6 6 6 Jr unsigned intct|}d|d|d|dg}t|dz ddD]$}|d||d |d %|d d |S) z declare a multi-index array in_coord and unravel the 1D index, i into it. This code assumes that the array is a C-ordered array. z z in_coord[z ]; z s, t, idx = i;r'rz s = z(; t = idx / s; in_coord[z!] = idx - t * s; idx = t;z in_coord[0] = idx; )lenr r join)shapeuint_trcoders r_unravel_loop_indexr5s u::D #  # # # #  # # # $D4!8Q # #  1X      KK 99T??rrFc |g} |rdnd} | | d|rd} d}nd} d}t|D]!}| d|d |d |d "| d| d |d z dt|d z ddD]'}| d| d |d z d|d|d (| s#| t|| | ||| z} |tjurd}n/|tjkrd}n|tj krd}nd| d|}|dkrHddt|D}| d|d|d|dkr|dkr| d t|D]}|dkr| d!|dn"| d"|d#|d$|d%|d& |dkre|dkrd'}d(}nd)|}d*}| tj||d+||||dkr| d"|d#|d$|d,| d-|d.|d/|d0|d d1d2t|D}|d3krPdd4t|D}| d5|d6|d7| d8|d9 n| d:| d8|d n|d krt|D]}| d-|d#|d$|d%|d;|d<|d/|d=|d>|d?|d@|dA|dkr| dB|dC|dDn(| d"|dE|d/|dF|dG|dH|d |dkr|dkrdI}d(}ndJ|}d*}| tj||d+||||dkrdKndL|}| tj||d+||||dkr(| dM|dE|d$|dN|dG|d$|dO | dPgdQ|dR|dS|dT|dU|dV|d.|dW|dX|dY|dZ|d[|d\|d0|d]|d^|d_|d[|d`|d0|daĐn|d kr|d3krd}n|dbkrdb}ntj |}| dc|ddt|D]}| de|df|d zd | t| ||g|dhvrJ| di|dd'}| tj||d+||d(ndj|}|d zrdk}ndl}| | |||dmzn| |do|df|d zd t|d zD]T}dp|df|dq}| d"|dr|d| tj||d+||U| ds|dt|d.|du|dv|dw|dx|dy|dz|d{|d||d}|d{|d~|dt|dkrddt|D}d1dt|D}|d3ks |d krd|dkr^ddt|D}| d5|d|d| d|d| d| d8|d| d|dn%| d-| d| d8|d| d|d | d|z|dkr| d|r| dn| dd| }| dd}d ||||dd|D}| dkr|dz }||fS)aQ Args: coord_func (function): generates code to do the coordinate transformation. See for example, `_get_coord_shift`. ndim (int): The number of dimensions. large_int (bool): If true use Py_ssize_t instead of int for indexing. yshape (tuple): Shape of the output array. mode (str): Signal extension mode to use at the array boundaries cval (float): constant value used when `mode == 'constant'`. name (str): base name for the interpolation kernel integer_output (bool): boolean indicating whether the output has an integer type. nprepad (int): integer indicating the amount of prepadding at the boundaries. Returns: operation (str): code body for the ElementwiseKernel name (str): name for the ElementwiseKernel doubleYz out = 0.0;size_t ptrdiff_tr,intzconst z xsize_z = x.shape()[r(z sx_r'z = 1;rr.z = sx_z * xsize_r z (Y)CUDART_NANz (Y)CUDART_INFz(Y)(-CUDART_INF)()constantz || c &g|]}d|d|d|dS)z(c_z < 0) || (c_z > xsize_z - 1).0rs r z+_generate_interp_custom..s4 L L L1 61 6 6! 6 6a 6 6 6 L L Lrz if (z) { out = z"; } else {wrapzdouble dcoord;z dcoord = c_z z cf_z = (z)floor((double)c_z + 0.5);dcoordTcf_Fxsize_z)floor(dcoord + 0.5);z z ic_z = cf_z * sx_z + cg|]}d|Sic_r@rAs rrCz+_generate_interp_custom..E @ @ @qq @ @ @rz grid-constantcg|]}d|d Sz(ic_z < 0)r@rAs rrCz+_generate_interp_custom..G F F FQ F F Frz if (z) { out = z.; } else { out = (z)x[z]; }z out = (z); z cc_z + 1; z n_z = (c_z == cf_z)) ? 1 : 2; // points needed z$ double dcoordf = c_z%; double dcoordc = c_z + 1;z cf_bounded_z; z cc_bounded_z = cc_dcoordf cf_bounded_dcoordc cc_bounded_z z&)floor(dcoordf);; z*)floor(dcoordf + 1);; rz for (int s_z = 0; s_z < n_z; s_z.++) { W w_z; z; if (s_z7 == 0) { w_z = (W)cc_z - c_z; ic_z = cf_bounded_zM; } else { w_z = c_z - (W)cf_z = cc_bounded_z; }nearestz# W wx, wy; z start;z W weights_[)rorder)rDzdouble dcoord = c_z (double)c_zA start = ({int_t})floor({coord_var}) - {order_2};zG start = ({int_t})floor({coord_var} + 0.5) - {order_2};)int_t coord_varorder_2z ci_ci_r z = start + z W w_z; z; for (int k_z = 0; k_z <= z; k_z,++) { w_z = weights_z[k_z]; ic_z = ci_z] * sx_z * cg|]}d|S)w_r@rAs rrCz+_generate_interp_custom..s<<<1h1hh<<.rKrcg|]}d|d SrMr@rAs rrCz+_generate_interp_custom..rNrz) { out += z * (z)(z(); } else { z val = (z!]; out += val * (z); }z]; out += val * (z);}zy = (Y)rint((double)out);z y = (Y)out;r/-_z5cupyx_scipy_ndimage_interpolate_{}_order{}_{}_{}d_y{}cg|]}|Sr@r@rAs rrCz+_generate_interp_custom..s.F.F.F!!v.F.F.Fr_i64)r r r5numpynaninfr1r _generate_boundary_condition_opsr_get_spline_modespline_weights_inlineformatreplace) coord_funcr large_intyshapemodecvalrUnameinteger_outputr omit_in_coordrinternal_dtyper3rWr_condixvarfloat_ix _coord_idx spline_moderXop_strr*_weight operationmode_strs r_generate_interp_customr~s8. C!/8XXSNJJ.---...4[[AA ?E??!??!???@@@@JJ333D1H333444 4!8Q # #GG EFEEAEEQEEEEEFFFF 8 &vv66777  4)) )C uy    %)  !*>**D** z L Ld L L LNN      zz 6>> JJ' ( ( (t 2 2Av~~ ####$$$$ IIIII#(II;<IIIJJJz!!6>>$E#HH%!IIE$H :e\a\\5(DDEEE6>>JJ B B B B B#( B B BCCC JJ1 1111!"11*+111 2 2 2 2ZZ @ @E$KK @ @ @AA ? " "KK F F%++ F F FGGE JJ' ,6      JJ8"88'1888 9 9 9 9 !t6 6 A JJ $78 !"!"+,    v~~ /$%//$%///0000  444$%44-.4444$%44-.444555z!!6>>%E#HH-!--E$H :e\a\\5(DDEEE&*V^^ 9Jq9J9J :e\a\\5(DDEEE6>>JJ()/4()/4 JJ          #$    +,    23            !"             ()    01        ./    78        $%    01        ./    78          S6 n  ? " "$KK Y  #KK0A$GGK     t- - A JJ+++ 19+++ , , , JJ,U3::Qe:LL M M Mx 4444555$  :i!udDDEEEE-,, qy JDJ JJ 9eqj    JJE::q::519::: ; ; ;519%% B B&a! ***#$***+++ :#ULQLL%AABBBB JJ $%+067   &'  ,-    "#  ()  23      qyy**<<d <<<==ZZ @ @E$KK @ @ @AA ? " "uqyyTZ5G5GKK F F%++ F F FGGE JJ"04; *8 =G  .  29      JJ; ;;%3;;8B;;);;-4;;; < < < 3: z 3" .//// =!!! #I||C%%H B I I eXtSXX.F.Fv.F.F.F%G%G  D  d?rT)for_each_devicer'c d}d} tt||||||d||d \} } tj|| | | tS)Nzraw X x, raw W coordsY ymapT) rlrrmrnrorprUrqrrrrspreamble)r~rcupyElementwiseKernelmath_constants_preamble rrmrnrorprUrrr in_params out_paramsr|rqs r_get_map_kernelrsr(IJ-!    %   OIt  !)ZD+B D D DDrc d}d} tt||||||d|| \} } tj|| | | tS)Nzraw X x, raw W shiftrshift rlrrmrnrorprUrqrrrr)r~r%rrrrs r_get_shift_kernelrso'IJ-#    %   OIt  !)ZD+B D D DDrc d} d} |rt} nt} t| |||||||rdnd|| \} } tj| | | | t S)Nz raw X x, raw W shift, raw W zoomrzoom_shift_grid zoom_shiftrr)rrr~rrr)rrmrnrorprUrr grid_moderrrzoom_shift_funcr|rqs r_get_zoom_shift_kernelrs3IJ483-"   "+ =  %   OIt  !)ZD+B D D DDrc d} d} t|rtnt|||||||rdnd|| \} } tj| | | | t S)Nzraw X x, raw W zoomr zoom_gridzoomrr)r~r#r!rrr) rrmrnrorprUrrrrrrr|rqs r_get_zoom_kernelr/s&IJ-+4I''/   % 1[[6%   OIt  !)ZD+B D D DDrc d}d} tt||||||d|| \} } tj|| | | tS)Nzraw X x, raw W matraffinerr)r~r+rrrrs r_get_affine_kernelrDso%IJ-$    %   OIt  !)ZD+B D D DDr)r)r,)rFrF)rr'Fr)rr'FFr)rdrcupy._core.internalcupyx.scipy.ndimagerrrrrirrrr!r#r%r+r5r~memoizerrrrrr@rrrsr 666666666666%%%%%% /D:4422.D*KL*/FFFFRD))CD23DDD*)D*D))EF45DDD*)D(D))JKJKDDD*)D0D))DEDEDDD*)D(D))FG56DDD*)DDDr