ó
î&]\c        
   @` sß  d  d l  m Z m Z m Z d  d l Z d  d l Z d  d l Z d d l m Z d d l m	 Z	 d d l
 m Z d  d l m Z e j ƒ  Z [ e d j d	 d
 ƒ e d <e j e ƒ Z d d d d d d d d g Z e d d e j d d „ ƒ Z d e j d d „ Z e d d d d d e d! i  d „ ƒ Z e d d d d e d „ ƒ Z e d d d d d d e d „ ƒ Z e d d d d e d „ ƒ Z e d d d d e d „ ƒ Z d „  Z e d" e d d d d e d  „ ƒ Z d S(#   i    (   t   divisiont   print_functiont   absolute_importNi   (   t   _ni_support(   t	   _nd_image(   t   docdict(   t   doccert   modes   Default is 'reflect's   Default is 'constant't   spline_filter1dt   spline_filtert   geometric_transformt   map_coordinatest   affine_transformt   shiftt   zoomt   rotatei   iÿÿÿÿt   mirrorc         C` sÉ   | d k  s | d k r' t  d ƒ ‚ n  t j |  ƒ }  t j |  ƒ rT t d ƒ ‚ n  t j | |  ƒ } | d k rˆ t j |  ƒ | d <n= t j | ƒ } t j	 | |  j
 ƒ } t j |  | | | | ƒ | S(   s+  
    Calculate a one-dimensional spline filter along the given axis.

    The lines of the array along the given axis are filtered by a
    spline filter. The order of the spline must be >= 2 and <= 5.

    Parameters
    ----------
    %(input)s
    order : int, optional
        The order of the spline, default is 3.
    axis : int, optional
        The axis along which the spline filter is applied. Default is the last
        axis.
    output : ndarray or dtype, optional
        The array in which to place the output, or the dtype of the returned
        array. Default is `numpy.float64`.
    %(mode)s

    Returns
    -------
    spline_filter1d : ndarray
        The filtered input.

    Notes
    -----
    All functions in `ndimage.interpolation` do spline interpolation of
    the input image. If using b-splines of `order > 1`, the input image
    values have to be converted to b-spline coefficients first, which is
    done by applying this one-dimensional filter sequentially along all
    axes of the input. All functions that require b-spline coefficients
    will automatically filter their inputs, a behavior controllable with
    the `prefilter` keyword argument. For functions that accept a `mode`
    parameter, the result will only be correct if it matches the `mode`
    used when filtering.
    i    i   s   spline order not supporteds   Complex type not supportedi   .(   i    i   (   t   RuntimeErrort   numpyt   asarrayt   iscomplexobjt	   TypeErrorR   t   _get_outputt   arrayt   _extend_mode_to_codet   _check_axist   ndimR   R   (   t   inputt   ordert   axist   outputR   (    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyR   6   s    'c      	   C` sÒ   | d k  s | d k r' t  d ƒ ‚ n  t j |  ƒ }  t j |  ƒ rT t d ƒ ‚ n  t j | |  ƒ } | d
 k rÀ |  j d k rÀ xJ t |  j ƒ D]( } t	 |  | | d | d | ƒ| }  q‘ Wn |  d	 | d	 <| S(   sñ  
    Multi-dimensional spline filter.

    For more details, see `spline_filter1d`.

    See Also
    --------
    spline_filter1d

    Notes
    -----
    The multi-dimensional filter is implemented as a sequence of
    one-dimensional spline filters. The intermediate arrays are stored
    in the same data type as the output. Therefore, for output types
    with a limited precision, the results may be imprecise because
    intermediate results may be stored with insufficient precision.

    i   i   s   spline order not supporteds   Complex type not supportedi    i   R   R   .(   i    i   (
   R   R   R   R   R   R   R   R   t   rangeR   (   R   R   R   R   R   (    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyR	   l   s    t   constantg        c
         C` s%  | d k  s | d k r' t  d ƒ ‚ n  t j |  ƒ }  t j |  ƒ rT t d ƒ ‚ n  | d	 k rl |  j } n  |  j d k  s t | ƒ d k  rœ t  d ƒ ‚ n  t	 j
 | ƒ } | rØ | d k rØ t |  | d t j ƒ}
 n |  }
 t	 j | |  d | ƒ} t j |
 | d	 d	 d	 | | | | | |	 ƒ | S(
   sú  
    Apply an arbitrary geometric transform.

    The given mapping function is used to find, for each point in the
    output, the corresponding coordinates in the input. The value of the
    input at those coordinates is determined by spline interpolation of
    the requested order.

    Parameters
    ----------
    %(input)s
    mapping : {callable, scipy.LowLevelCallable}
        A callable object that accepts a tuple of length equal to the output
        array rank, and returns the corresponding input coordinates as a tuple
        of length equal to the input array rank.
    output_shape : tuple of ints, optional
        Shape tuple.
    %(output)s
    order : int, optional
        The order of the spline interpolation, default is 3.
        The order has to be in the range 0-5.
    %(mode)s
    %(cval)s
    %(prefilter)s
    extra_arguments : tuple, optional
        Extra arguments passed to `mapping`.
    extra_keywords : dict, optional
        Extra keywords passed to `mapping`.

    Returns
    -------
    output : ndarray
        The filtered input.

    See Also
    --------
    map_coordinates, affine_transform, spline_filter1d


    Notes
    -----
    This function also accepts low-level callback functions with one
    the following signatures and wrapped in `scipy.LowLevelCallable`:

    .. code:: c

       int mapping(npy_intp *output_coordinates, double *input_coordinates,
                   int output_rank, int input_rank, void *user_data)
       int mapping(intptr_t *output_coordinates, double *input_coordinates,
                   int output_rank, int input_rank, void *user_data)

    The calling function iterates over the elements of the output array,
    calling the callback function at each element. The coordinates of the
    current output element are passed through ``output_coordinates``. The
    callback function must return the coordinates at which the input must
    be interpolated in ``input_coordinates``. The rank of the input and
    output arrays are given by ``input_rank`` and ``output_rank``
    respectively.  ``user_data`` is the data pointer provided
    to `scipy.LowLevelCallable` as-is.

    The callback function must return an integer error status that is zero
    if something went wrong and one otherwise. If an error occurs, you should
    normally set the python error status with an informative message
    before returning, otherwise a default error message is set by the
    calling function.

    In addition, some other low-level function pointer specifications
    are accepted, but these are for backward compatibility only and should
    not be used in new code.

    Examples
    --------
    >>> import numpy as np
    >>> from scipy.ndimage import geometric_transform
    >>> a = np.arange(12.).reshape((4, 3))
    >>> def shift_func(output_coords):
    ...     return (output_coords[0] - 0.5, output_coords[1] - 0.5)
    ...
    >>> geometric_transform(a, shift_func)
    array([[ 0.   ,  0.   ,  0.   ],
           [ 0.   ,  1.362,  2.738],
           [ 0.   ,  4.812,  6.187],
           [ 0.   ,  8.263,  9.637]])

    >>> b = [1, 2, 3, 4, 5]
    >>> def shift_func(output_coords):
    ...     return (output_coords[0] - 3,)
    ...
    >>> geometric_transform(b, shift_func, mode='constant')
    array([0, 0, 0, 1, 2])
    >>> geometric_transform(b, shift_func, mode='nearest')
    array([1, 1, 1, 1, 2])
    >>> geometric_transform(b, shift_func, mode='reflect')
    array([3, 2, 1, 1, 2])
    >>> geometric_transform(b, shift_func, mode='wrap')
    array([2, 3, 4, 1, 2])

    i    i   s   spline order not supporteds   Complex type not supportedi   s!   input and output rank must be > 0R   t   shapeN(   R   R   R   R   R   t   NoneR!   R   t   lenR   R   R	   t   float64R   R   R
   (   R   t   mappingt   output_shapeR   R   R   t   cvalt	   prefiltert   extra_argumentst   extra_keywordst   filtered(    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyR
   Ž   s$    g!c   	      C` sl  | d k  s | d k r' t  d ƒ ‚ n  t j |  ƒ }  t j |  ƒ rT t d ƒ ‚ n  t j | ƒ } t j | ƒ r t d ƒ ‚ n  | j d } |  j d k  s¯ t | ƒ d k  r¾ t  d ƒ ‚ n  | j d |  j k rã t  d ƒ ‚ n  t j	 | ƒ } | r| d k rt
 |  | d t j ƒ} n |  } t j | |  d	 | ƒ} t j | d
 | d
 d
 | | | | d
 d
 ƒ | S(   so  
    Map the input array to new coordinates by interpolation.

    The array of coordinates is used to find, for each point in the output,
    the corresponding coordinates in the input. The value of the input at
    those coordinates is determined by spline interpolation of the
    requested order.

    The shape of the output is derived from that of the coordinate
    array by dropping the first axis. The values of the array along
    the first axis are the coordinates in the input array at which the
    output value is found.

    Parameters
    ----------
    %(input)s
    coordinates : array_like
        The coordinates at which `input` is evaluated.
    %(output)s
    order : int, optional
        The order of the spline interpolation, default is 3.
        The order has to be in the range 0-5.
    %(mode)s
    %(cval)s
    %(prefilter)s

    Returns
    -------
    map_coordinates : ndarray
        The result of transforming the input. The shape of the output is
        derived from that of `coordinates` by dropping the first axis.

    See Also
    --------
    spline_filter, geometric_transform, scipy.interpolate

    Examples
    --------
    >>> from scipy import ndimage
    >>> a = np.arange(12.).reshape((4, 3))
    >>> a
    array([[  0.,   1.,   2.],
           [  3.,   4.,   5.],
           [  6.,   7.,   8.],
           [  9.,  10.,  11.]])
    >>> ndimage.map_coordinates(a, [[0.5, 2], [0.5, 1]], order=1)
    array([ 2.,  7.])

    Above, the interpolated value of a[0.5, 0.5] gives output[0], while
    a[2, 1] is output[1].

    >>> inds = np.array([[0.5, 2], [0.5, 4]])
    >>> ndimage.map_coordinates(a, inds, order=1, cval=-33.3)
    array([  2. , -33.3])
    >>> ndimage.map_coordinates(a, inds, order=1, mode='nearest')
    array([ 2.,  8.])
    >>> ndimage.map_coordinates(a, inds, order=1, cval=0, output=bool)
    array([ True, False], dtype=bool)

    i    i   s   spline order not supporteds   Complex type not supportedi   s!   input and output rank must be > 0s"   invalid shape for coordinate arrayR   R!   N(   R   R   R   R   R   R!   R   R#   R   R   R	   R$   R   R   R
   R"   (	   R   t   coordinatesR   R   R   R'   R(   R&   R+   (    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyR   
  s,    ?!	c	         C` s¢  | d k  s | d k r' t  d ƒ ‚ n  t j |  ƒ }  t j |  ƒ rT t d ƒ ‚ n  | d k rl |  j } n  |  j d k  s t | ƒ d k  rœ t  d ƒ ‚ n  t	 j
 | ƒ } | rØ | d k rØ t |  | d t j ƒ}	 n |  }	 t	 j | |  d | ƒ} t j | d	 t j ƒ} | j d k s0| j d d k  r?t  d ƒ ‚ n  | j d
 k rD| j d |  j d k rD| j d |  j |  j d g k rD| j d |  j d k rd g |  j d g }
 t j | |  j |
 k ƒ sd | j |  j |
 f } t | ƒ ‚ qn  | d |  j … |  j f } | d |  j … d |  j … f } n  | j d |  j k rit  d ƒ ‚ n  | j d
 k r| j d | j k rt  d ƒ ‚ n  | j j s¸| j ƒ  } n  t	 j | |  j ƒ } t j | d	 t j ƒ} | j d k s| j d d k  rt  d ƒ ‚ n  | j j s1| j ƒ  } n  | j d k rst j d ƒ t j |	 | | | | | | | ƒ n+ t j |	 d d | | | | | | d d ƒ | S(   s£
  
    Apply an affine transformation.

    Given an output image pixel index vector ``o``, the pixel value
    is determined from the input image at position
    ``np.dot(matrix, o) + offset``.

    Parameters
    ----------
    %(input)s
    matrix : ndarray
        The inverse coordinate transformation matrix, mapping output
        coordinates to input coordinates. If ``ndim`` is the number of
        dimensions of ``input``, the given matrix must have one of the
        following shapes:

            - ``(ndim, ndim)``: the linear transformation matrix for each
              output coordinate.
            - ``(ndim,)``: assume that the 2D transformation matrix is
              diagonal, with the diagonal specified by the given value. A more
              efficient algorithm is then used that exploits the separability
              of the problem.
            - ``(ndim + 1, ndim + 1)``: assume that the transformation is
              specified using homogeneous coordinates [1]_. In this case, any
              value passed to ``offset`` is ignored.
            - ``(ndim, ndim + 1)``: as above, but the bottom row of a
              homogeneous transformation matrix is always ``[0, 0, ..., 1]``,
              and may be omitted.

    offset : float or sequence, optional
        The offset into the array where the transform is applied. If a float,
        `offset` is the same for each axis. If a sequence, `offset` should
        contain one value for each axis.
    output_shape : tuple of ints, optional
        Shape tuple.
    %(output)s
    order : int, optional
        The order of the spline interpolation, default is 3.
        The order has to be in the range 0-5.
    %(mode)s
    %(cval)s
    %(prefilter)s

    Returns
    -------
    affine_transform : ndarray
        The transformed input.

    Notes
    -----
    The given matrix and offset are used to find for each point in the
    output the corresponding coordinates in the input by an affine
    transformation. The value of the input at those coordinates is
    determined by spline interpolation of the requested order. Points
    outside the boundaries of the input are filled according to the given
    mode.

    .. versionchanged:: 0.18.0
        Previously, the exact interpretation of the affine transformation
        depended on whether the matrix was supplied as a one-dimensional or
        two-dimensional array. If a one-dimensional array was supplied
        to the matrix parameter, the output pixel value at index ``o``
        was determined from the input image at position
        ``matrix * (o + offset)``.

    References
    ----------
    .. [1] https://en.wikipedia.org/wiki/Homogeneous_coordinates
    i    i   s   spline order not supporteds   Complex type not supportedi   s!   input and output rank must be > 0R   R!   t   dtypei   s    no proper affine matrix providedso   Expected homogeneous transformation matrix with shape %s for image shape %s, but bottom row was not equal to %sNs&   affine matrix has wrong number of rowss)   affine matrix has wrong number of columnss   no proper offset provideds}   The behaviour of affine_transform with a one-dimensional array supplied for the matrix parameter has changed in scipy 0.18.0.(   i   i   (   R   R   R   R   R   R"   R!   R   R#   R   R   R	   R$   R   t   allt
   ValueErrort   flagst
   contiguoust   copyt   _normalize_sequencet   warningst   warnR   t
   zoom_shiftR
   (   R   t   matrixt   offsetR&   R   R   R   R'   R(   R+   t   exptdt   msg(    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyR   b  s`    I!	")#%%"c   	      C` sK  | d k  s | d k r' t  d ƒ ‚ n  t j |  ƒ }  t j |  ƒ rT t d ƒ ‚ n  |  j d k  rr t  d ƒ ‚ n  t j | ƒ } | r® | d k r® t |  | d t j	 ƒ} n |  } t j
 | |  ƒ } t j | |  j ƒ } g  | D] } | ^ qâ } t j | d t j	 ƒ} | j j s(| j ƒ  } n  t j | d	 | | | | | ƒ | S(
   s­  
    Shift an array.

    The array is shifted using spline interpolation of the requested order.
    Points outside the boundaries of the input are filled according to the
    given mode.

    Parameters
    ----------
    %(input)s
    shift : float or sequence
        The shift along the axes. If a float, `shift` is the same for each
        axis. If a sequence, `shift` should contain one value for each axis.
    %(output)s
    order : int, optional
        The order of the spline interpolation, default is 3.
        The order has to be in the range 0-5.
    %(mode)s
    %(cval)s
    %(prefilter)s

    Returns
    -------
    shift : ndarray
        The shifted input.

    i    i   s   spline order not supporteds   Complex type not supportedi   s!   input and output rank must be > 0R   R-   N(   R   R   R   R   R   R   R   R   R	   R$   R   R3   R0   R1   R2   R   R6   R"   (	   R   R   R   R   R   R'   R(   R+   t   ii(    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyR   ä  s&    c         C` s  | d k  s | d k r' t  d ƒ ‚ n  t j |  ƒ }  t j |  ƒ rT t d ƒ ‚ n  |  j d k  rr t  d ƒ ‚ n  t j | ƒ } | r® | d k r® t |  | d t j	 ƒ} n |  } t j
 | |  j ƒ } t g  t |  j | ƒ D]" \ } }	 t t | |	 ƒ ƒ ^ qß ƒ }
 t g  t |  j | ƒ D] \ } }	 t | |	 ƒ ^ q ƒ } |
 | k rdt j d t ƒ n  t j |
 t ƒ d } t j t j |  j ƒ d | d	 t j |  j d
 t j	 ƒd | d k ƒ} t j | |  d |
 ƒ} t j | ƒ } t j | | d | | | | ƒ | S(   s%  
    Zoom an array.

    The array is zoomed using spline interpolation of the requested order.

    Parameters
    ----------
    %(input)s
    zoom : float or sequence
        The zoom factor along the axes. If a float, `zoom` is the same for each
        axis. If a sequence, `zoom` should contain one value for each axis.
    %(output)s
    order : int, optional
        The order of the spline interpolation, default is 3.
        The order has to be in the range 0-5.
    %(mode)s
    %(cval)s
    %(prefilter)s

    Returns
    -------
    zoom : ndarray
        The zoomed input.

    Examples
    --------
    >>> from scipy import ndimage, misc
    >>> import matplotlib.pyplot as plt

    >>> fig = plt.figure()
    >>> ax1 = fig.add_subplot(121)  # left side
    >>> ax2 = fig.add_subplot(122)  # right side
    >>> ascent = misc.ascent()
    >>> result = ndimage.zoom(ascent, 3.0)
    >>> ax1.imshow(ascent)
    >>> ax2.imshow(result)
    >>> plt.show()

    >>> print(ascent.shape)
    (512, 512)

    >>> print(result.shape)
    (1536, 1536)
    i    i   s   spline order not supporteds   Complex type not supportedi   s!   input and output rank must be > 0R   s˜   From scipy 0.13.0, the output shape of zoom() is calculated with round() instead of int() - for these inputs the size of the returned array has changed.t   outR-   t   whereR!   N(   R   R   R   R   R   R   R   R   R	   R$   R3   t   tuplet   zipR!   t   intt   roundR4   R5   t   UserWarningR   t   floatt   dividet	   ones_likeR   t   ascontiguousarrayR   R6   R"   (   R   R   R   R   R   R'   R(   R+   R;   t   jjR&   t   output_shape_oldt   zoom_div(    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyR     s:    />8
	c         C` sž   |  d | d k  r% |  d | d <n  |  d | d k rJ |  d | d <n  |  d | d k  ro |  d | d <n  |  d | d k r” |  d | d <n  | | f S(   Ni    i   (    (   t   coort   minct   maxc(    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyt   _minmaxm  s    c	   "      C` s  t  j |  ƒ }  t | ƒ } |  j }	 | d d k  rG | d c |	 7<n  | d d k  rj | d c |	 7<n  | d d k  sª | d d k  sª | d |	 k sª | d |	 k r¹ t d ƒ ‚ n  | d | d k rä | d | d f } n  t  j d | } t j | ƒ }
 t j | ƒ } t j | ƒ } t j | ƒ } t  j	 |
 | g | | g g d t  j
 ƒ} |  j | d } |  j | d } | r“t  j	 |
 | g | | g g d t  j
 ƒ} d d g } d d g } t  j | d | g ƒ } t | | | ƒ \ } } t  j | | d g ƒ } t | | | ƒ \ } } t  j | | | g ƒ } t | | | ƒ \ } } t | d | d d ƒ } t | d | d d ƒ } n" |  j | d } |  j | d } t  j d d t  j
 ƒ} t | ƒ d d | d <t | ƒ d d | d <t  j | | ƒ } t  j d d t  j
 ƒ} t | ƒ d d | d <t | ƒ d d | d <| | } t |  j ƒ } | | | d <| | | d <t | ƒ } t j | |  d	 | ƒ} |  j d k rät |  | | | | | | | | ƒ	 n—g  } t  j |  j d
 d ƒ} | |  j | d } | |  j | d } xL t |  j ƒ D]; } | | k r^| j d ƒ q<| j t d d d ƒ ƒ q<Wt t |  j ƒ ƒ } | j ƒ  | j | d ƒ | j | d ƒ | | d | | d f } x  t | ƒ D]’ } |  t | ƒ } | t | ƒ }  t | | | | |  | | | | ƒ	 xG | D]? }! | |! |  j |! d k  ri| |! c d 7<Pq4d | |! <q4WqåW| S(   sQ  
    Rotate an array.

    The array is rotated in the plane defined by the two axes given by the
    `axes` parameter using spline interpolation of the requested order.

    Parameters
    ----------
    %(input)s
    angle : float
        The rotation angle in degrees.
    axes : tuple of 2 ints, optional
        The two axes that define the plane of rotation. Default is the first
        two axes.
    reshape : bool, optional
        If `reshape` is true, the output shape is adapted so that the input
        array is contained completely in the output. Default is True.
    %(output)s
    order : int, optional
        The order of the spline interpolation, default is 3.
        The order has to be in the range 0-5.
    %(mode)s
    %(cval)s
    %(prefilter)s

    Returns
    -------
    rotate : ndarray
        The rotated input.

    i    i   s    invalid rotation plane specifiedi´   R-   g      à?i   g       @R!   R   (   i   (   i   N(   R   R   t   listR   R   t   pit   matht   cost   sinR   R$   R!   t   dotRM   R@   t   zerosRC   R>   R   R   R   t   productR   t   appendt   sliceR"   t   reverset   remove("   R   t   anglet   axest   reshapeR   R   R   R'   R(   t   rankt   m11t   m12t   m21t   m22R7   t   iyt   ixt   mtrxRK   RL   RJ   t   oyt   oxR8   t   tmpR&   R,   t   sizeR;   t	   iter_axest   ost   iat   oaRG   (    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyR   y  s˜    "	@
	

(    (   i   i    (    t
   __future__R    R   R   RP   R   R4   t    R   R   t   _ni_docstringsR   t
   scipy.miscR   R2   t   docdict_copyt   replacet   filldoct	   docfillert   __all__R$   R   R	   R"   t   TrueR
   R   R   R   R   RM   R   (    (    (    s:   lib/python2.7/site-packages/scipy/ndimage/interpolation.pyt   <module>   sP   	4"	xV2S	