o i@s|ddlZddlZddZddZddZdd Zdd d Zd dfddZddZddZ ddZ  dddZ ddZ dS)NcCs<|dur |jjdvSt|tjr|jjdvSt|jdvS)Niu)dtypekind isinstancecupyndarray)outputinputr V/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupyx/scipy/ndimage/_util.py_is_integer_outputs    r cCs*|dkr|rt|stddSdSdS)Nconstantz@Non-finite cval is not supported for outputs with integer dtype.)risfiniteNotImplementedError)modecvalinteger_outputr r r _check_cvals rcCs0|jjdkrt|jjtjSt|jjtjS)zInitialize filter weights based on the input array. This helper is only used during initialization of some internal filters like prewitt and sobel to avoid costly double-precision computation. c)rrr promote_typesreal complex64float32)r r r r _init_weights_dtypes rcCsN|jjdks |jjdkrt|jjtjS|jjdvrtjSt|jjtjS)Nriub)rrrrrrfloat64r)r weightsr r r _get_weights_dtypes  rFcCs|dur|jn|}|dur$|rt|jtj}n|j}tj||d}|St|ttjfrK|rBt|jdkrBt dt|tj}tj||d}|St|t rit|}|r`|jdkr`t dtj||d}|S|j|krrt d|r~|jjdkr~t d|S)N)rrz+promoting specified output dtype to complexzoutput must have complex dtypezoutput shape not correct) shaperrrremptyrtyperwarningswarnstr RuntimeError)rr rcomplex_output_dtyper r r _get_output(s0   r'cCs|S)Nr )xr r r Asr)cszt|tr |g|Szt|}Wnty"|g|YSwfdd|D}t||kr;d|}t||S)Ncsg|]}|qSr r ).0r(convr r Hsz%_fix_sequence_arg..z'{} must have length equal to input rank)rr#iter TypeErrorlenformatr$)argndimnamer,lstmsgr r+r _fix_sequence_argAs     r7cCs4t|}|d|dks|d||krtd|S)Nrzinvalid origin)int ValueError)originwidthr r r _check_originOs r=cCs |dvrd|d}t||S)N)reflectr nearestmirrorwrap grid-mirror grid-wrapz grid-reflectz%boundary mode not supported (actual: ))r$)rr6r r r _check_modeVs rEcCs4tddt|j|jD|jj}|dkrdSdS)Ncss$|] \}}|dt|VqdS)N)abs)r*r(strider r r cs"z_get_inttype..lr9 ptrdiff_t)sumziprstridesritemsize)r nbytesr r r _get_inttype^s  rPr9cCs|rdnd}|r dnd}|dvrdj|||d}|S|dkr(d j|||d}|S|d kr>d j|||||d kr8d nd d}|S|dkrKdj||d}|S|dkrYdj|||d}|S|dvrddj||d}|S)Nfminminfmaxmax)r>rBz if ({ix} < 0) {{ {ix} = - 1 -{ix}; }} {ix} %= {xsize} * 2; {ix} = {min}({ix}, 2 * {xsize} - 1 - {ix});)ixxsizerRr@a  if ({xsize} == 1) {{ {ix} = 0; }} else {{ if ({ix} < 0) {{ {ix} = -{ix}; }} {ix} = 1 + ({ix} - 1) % (({xsize} - 1) * 2); {ix} = {min}({ix}, 2 * {xsize} - 2 - {ix}); }}r?zD {ix} = {min}({max}(({T}){ix}, ({T})0), ({T})({xsize} - 1));r9z long long)rUrVrRrTTrCz] {ix} %= {xsize}; while ({ix} < 0) {{ {ix} += {xsize}; }})rUrVrAz if ({ix} < 0) {{ {ix} += ({sz} - 1) * (({int_t})(-{ix} / ({sz} - 1)) + 1); }} else if ({ix} > ({sz} - 1)) {{ {ix} -= ({sz} - 1) * ({int_t})({ix} / ({sz} - 1)); }};)rUszint_t)r z grid-constantzP if (({ix} < 0) || {ix} >= {xsize}) {{ {ix} = -1; }}r1)rrUrVrYfloat_ixmin_funcmax_funcopsr r r _generate_boundary_condition_opshsF  + # r_cs@dfddt|dddD}djd|dd S) Nz<{type} ind_{j} = _i % ysize_{j} - {offset}; _i /= ysize_{j};cs g|] }j||dqS))r joffsetrZ)r*r`codeint_typeoffsetsr r r-sz)_generate_indices_ops..rFrz3{type} _i = i; {body} {type} ind_0 = _i - {offset}; )r bodyra)ranger1join)r3rdrerhr rbr _generate_indices_opssrk)NF)r9F) r!rr rrrr'r7r=rErPr_rkr r r r s   3