o i@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. Znearestrwrap)Zreflectz grid-mirrorZmirrorZconstantz grid-wrapz grid-constantzboundary mode not supported) RuntimeError)modeZ is_filterr S/home/kim/smarthome/.venv/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 )inputZrankZis_str normalizederrr r r _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)shaperZzerosrnameZ promote_typesZ complex64rtypekindwarningswarnrr issubclassnumberZasarray)outputrrZcomplex_outputZ complex_typer r r _get_outputNs:  r%cs|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.0axr r r usz_check_axes..rzspecified axis: z is out of rangec3s$|] }|dkr |n|VqdS)rNr r)ndimr r r,xs"z6axes must be an integer, iterable of integers, or Nonezaxes must be unique) tuplerangerZisscalarr'r(rr ValueErrorrset)Zaxesr.r+messager r-r _check_axesns    r4cCsBt|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)rrrr"rZgenericrargr r r _skip_if_dtypes  r7cCs|dus t|tr dS|Sr&)rintr5r r r _skip_if_intsr9)F)NF) collections.abcrr'r numpyrrrr%r4r7r9r r r r s