o  h@s\ddlmZddlZddlZddlZdddZddZddd Zd d Z d d Z ddZ dS))IterableNFcCs||dkrdS|dkr dS|dvrdS|dkrdS|d krd S|d kr&|r&dS|d kr,d S|d kr4|r4d S|d kr:dStd)zAConvert an extension mode to the corresponding integer code. nearestrwrap)reflectz grid-mirrormirrorconstantz grid-wrapz grid-constantzboundary mode not supported) RuntimeError)mode is_filterrM/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/ndimage/_ni_support.py_extend_mode_to_code%s&  rcCsJt|t}|st|rt|}t||krd}t||S|g|}|S)zIf input is a scalar, create a sequence of length equal to the rank by duplicating the input. If input is a sequence, check if its length is equal to the length of array. z6sequence argument must have length equal to input rank) isinstancestrnpiterablelistlenr)inputrankis_str normalizederrrrr_normalize_sequence>s   rcCs.|dur|j}|dur)|stj||jjd}|St|jtj}tj||d}|St|ttjfrR|rIt|j dkrIt j dddt|tj}tj||d}|St|t r{t|}|rg|j dkrgt dt|jtjsrt dtj||d}|St|}|j|krt d|r|jj dkrt d|S) N)dtypecz+promoting specified output dtype to complexr ) stacklevelzoutput must have complex dtypezoutput must have numeric dtypezoutput shape not correct)shaperzerosr name promote_types complex64rtypekindwarningswarnrr issubclassnumberasarray)outputrr#complex_output complex_typerrr _get_outputNs:  r2cs|dur ttSt|rt|f}n8t|trH|D]}tdd|D}| ks3|dkr;td|dqtfdd|D}nd}t|t tt |t |kr^td|S) Ncss|]}t|VqdSN)operatorindex.0axrrr usz_check_axes..rzspecified axis: z is out of rangec3s$|] }|dkr |n|VqdS)rNrr6ndimrrr9xs"z6axes must be an integer, iterable of integers, or Nonezaxes must be unique) tuplerangerisscalarr4r5rr ValueErrorrset)axesr;r8messagerr:r _check_axesns    rCcCsBt|trdSt|turt|tjrdS|St|tjrdS|S)z'array or dtype' polymorphism. Return None for np.int8, dtype('float32') or 'f' etc arg for np.empty(3) etc N)rrr(r,rgenericr argrrr_skip_if_dtypes  rGcCs|dus t|tr dS|Sr3)rintrErrr _skip_if_intsrI)F)NF) collections.abcrr4r*numpyrrrr2rCrGrIrrrrs