
    _Mhf              	       6   d Z ddlZddlZddlZddlmZmZmZm	Z	m
Z
 ddlmZ dZdZd Zd	 Zd
 Zd ZddddddddZddddddddZd Zd Z G d d          Zd Z G d d          ZdddZd ZdddddddddZdddddddddZddddddddddZdS ) a   Replicate FITPACK's logic for constructing smoothing spline functions and curves.

    Currently provides analogs of splrep and splprep python routines, i.e.
    curfit.f and parcur.f routines (the drivers are fpcurf.f and fppara.f, respectively)

    The Fortran sources are from
    https://github.com/scipy/scipy/blob/maintenance/1.11.x/scipy/interpolate/fitpack/

    .. [1] P. Dierckx, "Algorithms for smoothing data with periodic and
        parametric splines, Computer Graphics and Image Processing",
        20 (1982) 171-184.
        :doi:`10.1016/0146-664X(82)90043-0`.
    .. [2] P. Dierckx, "Curve and surface fitting with splines", Monographs on
         Numerical Analysis, Oxford University Press, 1993.
    .. [3] P. Dierckx, "An algorithm for smoothing, differentiation and integration
         of experimental data using spline functions",
         Journal of Computational and Applied Mathematics, vol. I, no 3, p. 165 (1975).
         https://doi.org/10.1016/0771-050X(75)90034-0
    N   )_not_a_knotmake_interp_splineBSplinefpcheck_lsq_solve_qr)_dierckxgMbP?   c                     |dz  }t          | ||||          \  }}}t          j        |          }t          |||          }t	          | ||           |          S )N   )r   npascontiguousarrayr   _compute_residuals)	xytkww2_cspls	            `/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/interpolate/_fitpack_repro.py_get_residualsr   *   sc    	
AB Aq!Q**GAq!
QA
!Q

Cb##a&&!,,,    c                 D    ||z
  dz                       d          }| |z  S )Nr   r   axis)sum)r   splxr   deltas       r   r   r   ?   s*    Qh]Q''E:r   c                     t          j        | |||          }t          j        ||          }t          j        |d|         |||d         f         }|S )ar  Add a new knot.

    (Approximately) replicate FITPACK's logic:
      1. split the `x` array into knot intervals, ``t(j+k) <= x(i) <= t(j+k+1)``
      2. find the interval with the maximum sum of residuals
      3. insert a new knot into the middle of that interval.

    NB: a new knot is in fact an `x` value at the middle of the interval.
    So *the knots are a subset of `x`*.

    This routine is an analog of
    https://github.com/scipy/scipy/blob/v1.11.4/scipy/interpolate/fitpack/fpcurf.f#L190-L215
    (cf _split function)

    and https://github.com/scipy/scipy/blob/v1.11.4/scipy/interpolate/fitpack/fpknot.f
    N)r	   fpknotr   searchsortedr_)r   r   r   	residualsnew_knotidx_tt_news          r   add_knotr*   D   sR    " q!Q	22HOAx((EE!FUF)Xqy01ELr   c                    t          j        | t                    } t          j        |t                    }|t          j        | t                    }net          j        |t                    }|j        dk    rt          d|j        d          |dk                                     rt          d          |j        dk    s|j        dk    rt          d	|j        d
          t          |          }|r'|j        dk    rt          d	|j        d|d          n2|j        dk    rt          d	|j        d|d          |dddf         }|j        d         | j        d         k    r t          d|j        d| j         d          | j        d         |j        d         k    r t          d| j        d|j        d          | j        dk    s(| dd         | dd         k                                     rt          d          t          j
        |          }|dk     rt          d|          |t          |           }|t          |           }| ||||||fS )zACommon input validations for generate_knots and make_splrep.
    dtypeNr   	w.ndim = z not implemented yet.r   zWeights must be non-negativer   z	y.ndim = z  not supported (must be 1 or 2.)z% != 2 not supported with parametric =.z% != 1 not supported with parametric =z"Weights is incompatible: w.shape =z != z Data is incompatible: x.shape = z and y.shape = z(Expect `x` to be an ordered 1D sequence.z"`s` must be non-negative. Got s = )r   asarrayfloat	ones_likendim
ValueErroranyboolshapeoperatorindexminmax)r   r   r   r   sxbxe
parametrics           r   _validate_inputsrA   \   s}    	
1E"""A

1E"""AyL%(((Jq&&&6Q;;@@@@AAAE;;== 	=;<<<v{{afqjjGAFGGGHHHj!!J 6Q;;SSSJSSSTTT  6Q;;SSSJSSSTTTaaagJwqzQWQZNQWNNAGNNNOOOwqzQWQZP!'PP17PPPQQQv{{quq"v~**,,{CDDDqA1uu@!@@AAA	zVV	zVVaAq"b  r      r   r>   r?   r   r=   nestc             #     K   |dk    r)||t          d          t          | |          }|V  dS t          | ||||||t          j        |          dk              \  } }}}}}}t          | |||||||          E d{V  dS )a
  Replicate FITPACK's constructing the knot vector.

    Parameters
    ----------
    x, y : array_like
        The data points defining the curve ``y = f(x)``.
    w : array_like, optional
        Weights.
    xb : float, optional
        The boundary of the approximation interval. If None (default),
        is set to ``x[0]``.
    xe : float, optional
        The boundary of the approximation interval. If None (default),
        is set to ``x[-1]``.
    k : int, optional
        The spline degree. Default is cubic, ``k = 3``.
    s : float, optional
        The smoothing factor. Default is ``s = 0``.
    nest : int, optional
        Stop when at least this many knots are placed.

    Yields
    ------
    t : ndarray
        Knot vectors with an increasing number of knots.
        The generator is finite: it stops when the smoothing critetion is
        satisfied, or when then number of knots exceeds the maximum value:
        the user-provided `nest` or `x.size + k + 1` --- which is the knot vector
        for the interpolating spline.

    Examples
    --------
    Generate some noisy data and fit a sequence of LSQ splines:

    >>> import numpy as np
    >>> import matplotlib.pyplot as plt
    >>> from scipy.interpolate import make_lsq_spline, generate_knots
    >>> rng = np.random.default_rng(12345)
    >>> x = np.linspace(-3, 3, 50)
    >>> y = np.exp(-x**2) + 0.1 * rng.standard_normal(size=50)

    >>> knots = list(generate_knots(x, y, s=1e-10))
    >>> for t in knots[::3]:
    ...     spl = make_lsq_spline(x, y, t)
    ...     xs = xs = np.linspace(-3, 3, 201)
    ...     plt.plot(xs, spl(xs), '-', label=f'n = {len(t)}', lw=3, alpha=0.7)
    >>> plt.plot(x, y, 'o', label='data')
    >>> plt.plot(xs, np.exp(-xs**2), '--')
    >>> plt.legend()

    Note that increasing the number of knots make the result follow the data
    more and more closely.

    Also note that a step of the generator may add multiple knots:

    >>> [len(t) for t in knots]
    [8, 9, 10, 12, 16, 24, 40, 48, 52, 54]

    Notes
    -----
    The routine generates successive knots vectors of increasing length, starting
    from ``2*(k+1)`` to ``len(x) + k + 1``, trying to make knots more dense
    in the regions where the deviation of the LSQ spline from data is large.

    When the maximum number of knots, ``len(x) + k + 1`` is reached
    (this happens when ``s`` is small and ``nest`` is large), the generator
    stops, and the last output is the knots for the interpolation with the
    not-a-knot boundary condition.

    Knots are located at data sites, unless ``k`` is even and the number of knots
    is ``len(x) + k + 1``. In that case, the last output of the generator
    has internal knots at Greville sites, ``(x[1:] + x[:-1]) / 2``.

    .. versionadded:: 1.15.0

    r   Nzs == 0 is interpolation onlyr   r@   rC   )r5   r   rA   r   r4   _generate_knots_impl)	r   r   r   r>   r?   r   r=   rD   r   s	            r   generate_knotsrH      s      Z 	Avvq};<<<1,	1aAr2"'!**/  Aq!Q2r $AqA"qADQQQQQQQQQQQQr   c          	   #     K   |t           z  }| j        }	|t          |	|z   dz   d|z  dz             }n(|d|dz   z  k     rt          d|dd|dz   z   d          d|dz   z  }
|	|z   dz   }t	          j        |g|dz   z  |g|dz   z  z   t                    }|j        d         }d	}d	}t          |	          D ]}|V  |}t          | ||||
          }|
                                }||z
  }t          |          |k     s|dk     r d S ||
k    rd}nJ||z
  }||k    rt          ||z  |z            n|dz  }t          |dz  t          ||dz  d                    }t          |          D ]j}t          | |||          }|j        d         }||k    rt          | |          }|V    d S ||k    r|V    d S ||dz
  k     rt          | ||||
          }k!d S )Nr   r   rB   `nest` too small: nest =  < 2*(k+1) = r/   r,   r   g        r   )TOLsizer<   r5   r   r1   r2   r8   ranger   r   absintr;   r*   r   )r   r   r   r>   r?   r   r=   rD   accmnminnmaxr   nfpfpoldr   r&   fpmsnplusr!   npl1js                          r   rG   rG      sg     
c'C	A| 1q519acAg&&!QU)R$RR1Q3RRRSSSa!e9Dq519D 	
B41:ac
*%888A	
A	BE 1XX /< /< "1aA333	]]__Av IIOOFF
 99EEBJE053ut|e+,,,qDaT5!8Q!7!788E u 	< 	<AAq),,A 
A Dyy1%% Dyy 519}}*1aA;;;	+	<0 Fr   c                 t    d}t          |dz             D ]"}||z   |k    r|| |         | ||z            z
  z  }#|S )Ng      ?r   )rO   )r   ir\   r   resr=   s         r   proddr`   L  sM    
C1Q3ZZ # #q5A::AaD1QqS6M"CJr   c           	         | j         d         }| ||z
  dz
           | |         z
  }|d|z  z
  dz
  }t          j        |dz
  |dz   ft                    }t	          |dz
            D ]S}||z   dz   }t	          |dz             D ]6}||z   }	| |	|z   dz            | |	         z
  t          | |	||          z  |||f<   7T|||z  |z  z  }t          j        d t	          |dz
            D             t          j                  }
||z
  dz
  }||
|fS )a  Discontinuity matrix: jumps of k-th derivatives of b-splines at internal knots.

    See Eqs. (9)-(10) of Ref. [1], or, equivalently, Eq. (3.43) of Ref. [2].

    This routine assumes internal knots are all simple (have multiplicity =1).

    Parameters
    ----------
    t : ndarray, 1D, shape(n,)
        Knots.
    k : int
        The spline degree

    Returns
    -------
    disc : ndarray, shape(n-2*k-1, k+2)
        The jumps of the k-th derivatives of b-splines at internal knots,
        ``t[k+1], ...., t[n-k-1]``.
    offset : ndarray, shape(2-2*k-1,)
        Offsets
    nc : int

    Notes
    -----

    The normalization here follows FITPACK:
    (https://github.com/scipy/scipy/blob/maintenance/1.11.x/scipy/interpolate/fitpack/fpdisc.f#L36)

    The k-th derivative jumps are multiplied by a factor::

        (delta / nrint)**k / k!

    where ``delta`` is the length of the interval spanned by internal knots, and
    ``nrint`` is one less the number of internal knots (i.e., the number of
    subintervals between them).

    References
    ----------
    .. [1] Paul Dierckx, Algorithms for smoothing data with periodic and parametric
           splines, Computer Graphics and Image Processing, vol. 20, p. 171 (1982).
           :doi:`10.1016/0146-664X(82)90043-0`

    .. [2] Tom Lyche and Knut Morken, Spline methods,
        http://www.uio.no/studier/emner/matnat/ifi/INF-MAT5340/v05/undervisningsmateriale/

    r   r   r   r,   c                     g | ]}|S  rc   ).0r^   s     r   
<listcomp>zdisc.<locals>.<listcomp>  s    111Qq111r   )r8   r   emptyr2   rO   r`   arrayint64)r   r   rV   r!   nrintmatrjjr\   iir^   offsetncs               r   discro   T  sI   ^ 	

A a!eaiL1Q4E!GaKE8UQYA&e444DEAI E EFQJA,, 	E 	EBRAa!eaiL1Q4/5Aq!3D3DDDRLL	E 	UE\AD X11%a..111BBBF	
QBr   c                   (    e Zd ZdZdddddZd ZdS )Fa   The r.h.s. of ``f(p) = s``.

    Given scalar `p`, we solve the system of equations in the LSQ sense:

        | A     |  @ | c | = | y |
        | B / p |    | 0 |   | 0 |

    where `A` is the matrix of b-splines and `b` is the discontinuity matrix
    (the jumps of the k-th derivatives of b-spline basis elements at knots).

    Since we do that repeatedly while minimizing over `p`, we QR-factorize
    `A` only once and update the QR factorization only of the `B` rows of the
    augmented matrix |A, B/p|.

    The system of equations is Eq. (15) Ref. [1]_, the strategy and implementation
    follows that of FITPACK, see specific links below.

    References
    ----------
    [1] P. Dierckx, Algorithms for Smoothing Data with Periodic and Parametric Splines,
        COMPUTER GRAPHICS AND IMAGE PROCESSING vol. 20, pp 171-184 (1982.)
        https://doi.org/10.1016/0146-664X(82)90043-0

    N)RYc                   || _         || _        || _        || _        |t	          j        |t                    n|}|j        dk    rt          d|j        d          || _	        || _
        |j        dk    rt          d|j        d          t          ||          \  }	}
}||t          |||||          \  }}}|j        d         |z
  dz
  }|dz   }|j        d         |k    r$t          d	|j        d         d
|dz   d          t	          j        |	j        d         |j        d         ft                    }t          j        |d |         |f         | _        t	          j        ||	j        d         z   | j        dz   ft                    }|d |d d f         |d |d |f<   || _        t          j        t	          j        |t          j                  |
f         | _        || _        |	| _        d S )Nr,   r   r.   z != 1.r   z&F: expected y.ndim == 2, got y.ndim = z	 instead.r   zInternal error: R.shape[1] =z	 != k+1 =r/   )r   r   r   r   r   r3   r2   r4   r5   r   r=   ro   r   r8   zerosr%   YYAAarangerh   rm   rn   b)selfr   r   r   r   r=   r   rr   rs   ry   b_offsetb_ncr   rn   nzzrw   s                    r   __init__z
F.__init__  s   ,-IBL%((((16Q;;11112226Q;;QQVQQQRRR !AJJ8T 9#Aq!Q22GAq! WQZ!^aU71:J
JJ!A#JJJKKK Hagaj!'!*-U;;;%#2#	" XrAGAJq1???"aaay3B38eBIb9998CDr   c                    | j                                         }| j                                        }| j        }| j        |z  ||d d d f<   | j                                        }t          j        |||||           t          j        |||          }t          | j
        || j                  }t          | j        dz   || j                  | j                  }|                                }	|| _        |	| j        z
  S )N)startrowr   )rw   copyrm   rn   ry   rv   r	   	qr_reducefpbackr   r   r   r   r   r   r   r   r   r=   )
rz   pABrm   rn   QYr   r   r&   rW   s
             r   __call__z
F.__call__  s     W\\^^!!##WVaZ2336
W\\^^ 	2vr2;;;; OBB''dfa((&tvqy##df++tvFF	]]__DF{r   )N)__name__
__module____qualname____doc__r   r   rc   r   r   rq   rq     sP         0/44 / / / / /b    r   rq   c                     |||z
  z  }|||z
  z  }|||z
  z  }|t           j        k    r||z  | |z  z    |z  S | |z  |z  ||z  |z  z   | |z  |z  z    | |z  ||z  z   ||z  z   z  S )a  The root of r(p) = (u*p + v) / (p + w) given three points and values,
    (p1, f2), (p2, f2) and (p3, f3).

    The FITPACK analog adjusts the bounds, and we do not
    https://github.com/scipy/scipy/blob/maintenance/1.11.x/scipy/interpolate/fitpack/fprati.f

    NB: FITPACK uses p < 0 to encode p=infinity. We just use the infinity itself.
    Since the bracket is ``p1 <= p2 <= p3``, ``p3`` can be infinite (in fact,
    this is what the minimizer starts with, ``p3=inf``).
    )r   inf)	p1f1p2f2p3f3h1h2h3s	            r   fpratir     s     
rBwB	rBwB	rBwB	RV||BB"$$U2X2b 2b58+,2220EFFr   c                       e Zd Zd ZdS )Bunchc                 *     | j         j        di | d S )Nrc   )__dict__update)rz   kwargss     r   r   zBunch.__init__  s#    &&v&&&&&r   N)r   r   r   r   rc   r   r   r   r     s#        ' ' ' ' 'r   r   a(  error. a theoretically impossible result was found during
the iteration process for finding a smoothing spline with
fp = s. probably causes : s too small.
there is an approximation returned but the corresponding
weighted sum of squared residuals does not satisfy the
condition abs(fp-s)/s < tol.
a7  error. the maximal number of iterations maxit (set to 20
by the program) allowed for finding a smoothing spline
with fp=s has been reached. probably causes : s too small
there is an approximation returned but the corresponding
weighted sum of squared residuals does not satisfy the
condition abs(fp-s)/s < tol.
)r   rB   c           	         d}d}d}d\  }}|\  \  }	}
\  }}|}t          t                    D ]}| | |          }}t          |          |k     rd\  }} n|dk    r,||z
  |k    r|}|}||z  }||	k    r|	|z  ||z  z   }S|dk     rd}|dk    r<|
|z
  |k    r+|}	|}
||z  }|t          j        k    r||k    r||z  ||z  z   }|dk    rd}|
|k    s||k    rd\  }} n)t          |	|
||||          }|dk     r||}}||}
}	d	\  }}|dk    r.t          j        t          t          |                   d
           t          ||||          S )a  Solve `f(p) = 0` using a rational function approximation.

    In a nutshell, since the function f(p) is known to be monotonically decreasing, we
       - maintain the bracket (p1, f1), (p2, f2) and (p3, f3)
       - at each iteration step, approximate f(p) by a rational function
         r(p) = (u*p + v) / (p + w)
         and make a step to p_new to the root of f(p): r(p_new) = 0.
         The coefficients u, v and w are found from the bracket values p1..3 and f1...3

    The algorithm and implementation follows
    https://github.com/scipy/scipy/blob/maintenance/1.11.x/scipy/interpolate/fitpack/fpcurf.f#L229
    and
    https://github.com/scipy/scipy/blob/maintenance/1.11.x/scipy/interpolate/fitpack/fppara.f#L290

    Note that the latter is for parametric splines and the former is for 1D spline
    functions. The minimization is indentical though [modulo a summation over the
    dimensions in the computation of f(p)], so we reuse the minimizer for both
    d=1 and d>1.
    g?g?g{Gz?)r   r   )r   Tr   r   )r   F)rB   Fr   )
stacklevel)	convergedroot
iterationsier)rO   MAXITrP   r   r   r   warningswarnRuntimeWarning_iermesgr   )fp0bracketrR   con1con9con4ich1ich3r   r   r   r   r   itr   r   r   r   s                      r   	root_ratir   -  s   , DDD JD$!HRhr2
AEll 4" 4"AAaDDB r77S==$NCE 199Bw#~~dF77D2d7*Q66D199Bw#~~dF<<AGGD2d7*Q66D 88rRxx%NCE 2r2r2r** 66BBBB "Y
axxnXc]33BBBB91EEEEr   r   r>   r?   r   r=   r   rD   c          
         |t           z  }	| j        }
|t          |
|z   dz   d|z  dz             }n9|d|dz   z  k     rt          d|dd|dz   z   d          |t          d          |-t	          | |||||||	          }t          |          d
         }nt          | ||           |j        d         d|dz   z  k    r(t          | ||||          \  }}}t          |||          S t          | ||||          \  }}}|j        d         |z
  dz
  }||dddf         
                                z  }t          | ||||          }|
                                }||z
  }t          | |t          j        |g|dz   z  |g|dz   z  z             ||          }|
                                }||z
  }d|ft          j        |ff}t          | |||||||          }t!          ||||	          }|j        S )z3Shared infra for make_splrep and make_splprep.
    Nr   r   rB   rJ   rK   r/   zEither supply `t` or `nest`.)r   r   r=   r>   r?   rD   r0   r   rL   )r   r=   r   rr   rs   )rM   rN   r<   r5   rG   listr   r8   r   r   r   r   r   rg   r   rq   r   r   )r   r   r   r>   r?   r   r=   r   rD   rR   rS   genr   r   rr   rs   rn   r   r&   rW   fpinffp0r   r   s                           r   _make_splrep_implr     s8    c'C	A| 1q519acAg&&!QU)R$RR1Q3RRRSSS=;<<<y"1a1Q2"4PPPIIbM1awqzQ!a%[  1aA..1aq!Q Aq!Q**GAq!	
Q	B
Qqqq!tW[[]]A
 q!QQ///I	BFE q!RXrdAaCjB41:.E%F%F1MMI
--//C
'C #h'G	!QQ!qA+++A!Q%%A 5Lr   c                    |dk    r'|||t          d          t          | ||          S t          | ||||||d          \  } }}}}}}t          | ||||||||	  	        }	|	j        dddf         |	_        |	S )a  Find the B-spline representation of a 1D function.

    Given the set of data points ``(x[i], y[i])``, determine a smooth spline
    approximation of degree ``k`` on the interval ``xb <= x <= xe``.

    Parameters
    ----------
    x, y : array_like, shape (m,)
        The data points defining a curve ``y = f(x)``.
    w : array_like, shape (m,), optional
        Strictly positive 1D array of weights, of the same length as `x` and `y`.
        The weights are used in computing the weighted least-squares spline
        fit. If the errors in the y values have standard-deviation given by the
        vector ``d``, then `w` should be ``1/d``.
        Default is ``np.ones(m)``.
    xb, xe : float, optional
        The interval to fit.  If None, these default to ``x[0]`` and ``x[-1]``,
        respectively.
    k : int, optional
        The degree of the spline fit. It is recommended to use cubic splines,
        ``k=3``, which is the default. Even values of `k` should be avoided,
        especially with small `s` values.
    s : float, optional
        The smoothing condition. The amount of smoothness is determined by
        satisfying the conditions::

            sum((w * (g(x)  - y))**2 ) <= s

        where ``g(x)`` is the smoothed fit to ``(x, y)``. The user can use `s`
        to control the tradeoff between closeness to data and smoothness of fit.
        Larger `s` means more smoothing while smaller values of `s` indicate less
        smoothing.
        Recommended values of `s` depend on the weights, `w`. If the weights
        represent the inverse of the standard deviation of `y`, then a good `s`
        value should be found in the range ``(m-sqrt(2*m), m+sqrt(2*m))`` where
        ``m`` is the number of datapoints in `x`, `y`, and `w`.
        Default is ``s = 0.0``, i.e. interpolation.
    t : array_like, optional
        The spline knots. If None (default), the knots will be constructed
        automatically.
        There must be at least ``2*k + 2`` and at most ``m + k + 1`` knots.
    nest : int, optional
        The target length of the knot vector. Should be between ``2*(k + 1)``
        (the minimum number of knots for a degree-``k`` spline), and
        ``m + k + 1`` (the number of knots of the interpolating spline).
        The actual number of knots returned by this routine may be slightly
        larger than `nest`.
        Default is None (no limit, add up to ``m + k + 1`` knots).

    Returns
    -------
    spl : a `BSpline` instance
        For `s=0`,  ``spl(x) == y``.
        For non-zero values of `s` the `spl` represents the smoothed approximation
        to `(x, y)`, generally with fewer knots.

    See Also
    --------
    generate_knots : is used under the hood for generating the knots
    make_splprep : the analog of this routine for parametric curves
    make_interp_spline : construct an interpolating spline (``s = 0``)
    make_lsq_spline : construct the least-squares spline given the knot vector
    splrep : a FITPACK analog of this routine

    References
    ----------
    .. [1] P. Dierckx, "Algorithms for smoothing data with periodic and
        parametric splines, Computer Graphics and Image Processing",
        20 (1982) 171-184.
    .. [2] P. Dierckx, "Curve and surface fitting with splines", Monographs on
        Numerical Analysis, Oxford University Press, 1993.

    Notes
    -----
    This routine constructs the smoothing spline function, :math:`g(x)`, to
    minimize the sum of jumps, :math:`D_j`, of the ``k``-th derivative at the
    internal knots (:math:`x_b < t_i < x_e`), where

    .. math::

        D_i = g^{(k)}(t_i + 0) - g^{(k)}(t_i - 0)

    Specifically, the routine constructs the spline function :math:`g(x)` which
    minimizes

    .. math::

            \sum_i | D_i |^2 \to \mathrm{min}

    provided that

    .. math::

           \sum_{j=1}^m (w_j \times (g(x_j) - y_j))^2 \leqslant s ,

    where :math:`s > 0` is the input parameter.

    In other words, we balance maximizing the smoothness (measured as the jumps
    of the derivative, the first criterion), and the deviation of :math:`g(x_j)`
    from the data :math:`y_j` (the second criterion).

    Note that the summation in the second criterion is over all data points,
    and in the first criterion it is over the internal spline knots (i.e.
    those with ``xb < t[i] < xe``). The spline knots are in general a subset
    of data, see `generate_knots` for details.

    Also note the difference of this routine to `make_lsq_spline`: the latter
    routine does not consider smoothness and simply solves a least-squares
    problem

    .. math::

        \sum w_j \times (g(x_j) - y_j)^2 \to \mathrm{min}

    for a spline function :math:`g(x)` with a _fixed_ knot vector ``t``.

    .. versionadded:: 1.15.0
    r   Ns==0 is for interpolation only)r   FrF   r   )r5   r   rA   r   r   )
r   r   r   r>   r?   r   r=   r   rD   r   s
             r   make_splrepr     s    n 	Avv=AMT-==>>>!!Q!,,,,,Q1aBuUUUAq!Q2r
AqA"qA
N
N
NC E!!!Q$KCEJr   )r   uubuer   r=   r   rD   c                R   t          j        | d          } |w| ddddf         | ddddf         z
  dz  }	t          j        |	                    d                                                    }t           j        d||d         z  f         }|dk    r/|||t          d          t          || j        |d          |fS t          || |||||d	
          \  }} }}}}}t          || |||||||	  	        }
|
j        j        }t          |
j        ||
j        d          }||fS )aG  
    Find a smoothed B-spline representation of a parametric N-D curve.

    Given a list of N 1D arrays, `x`, which represent a curve in
    N-dimensional space parametrized by `u`, find a smooth approximating
    spline curve ``g(u)``.

    Parameters
    ----------
    x : array_like, shape (m, ndim)
        Sampled data points representing the curve in ``ndim`` dimensions.
        The typical use is a list of 1D arrays, each of length ``m``.
    w : array_like, shape(m,), optional
        Strictly positive 1D array of weights.
        The weights are used in computing the weighted least-squares spline
        fit. If the errors in the `x` values have standard deviation given by
        the vector d, then `w` should be 1/d. Default is ``np.ones(m)``.
    u : array_like, optional
        An array of parameter values for the curve in the parametric form.
        If not given, these values are calculated automatically, according to::

            v[0] = 0
            v[i] = v[i-1] + distance(x[i], x[i-1])
            u[i] = v[i] / v[-1]

    ub, ue : float, optional
        The end-points of the parameters interval. Default to ``u[0]`` and ``u[-1]``.
    k : int, optional
        Degree of the spline. Cubic splines, ``k=3``, are recommended.
        Even values of `k` should be avoided especially with a small ``s`` value.
        Default is ``k=3``
    s : float, optional
        A smoothing condition.  The amount of smoothness is determined by
        satisfying the conditions::

            sum((w * (g(u) - x))**2) <= s,

        where ``g(u)`` is the smoothed approximation to ``x``.  The user can
        use `s` to control the trade-off between closeness and smoothness
        of fit.  Larger ``s`` means more smoothing while smaller values of ``s``
        indicate less smoothing.
        Recommended values of ``s`` depend on the weights, ``w``.  If the weights
        represent the inverse of the standard deviation of ``x``, then a good
        ``s`` value should be found in the range ``(m - sqrt(2*m), m + sqrt(2*m))``,
        where ``m`` is the number of data points in ``x`` and ``w``.
    t : array_like, optional
        The spline knots. If None (default), the knots will be constructed
        automatically.
        There must be at least ``2*k + 2`` and at most ``m + k + 1`` knots.
    nest : int, optional
        The target length of the knot vector. Should be between ``2*(k + 1)``
        (the minimum number of knots for a degree-``k`` spline), and
        ``m + k + 1`` (the number of knots of the interpolating spline).
        The actual number of knots returned by this routine may be slightly
        larger than `nest`.
        Default is None (no limit, add up to ``m + k + 1`` knots).

    Returns
    -------
    spl : a `BSpline` instance
        For `s=0`,  ``spl(u) == x``.
        For non-zero values of ``s``, `spl` represents the smoothed approximation
        to ``x``, generally with fewer knots.
    u : ndarray
        The values of the parameters

    See Also
    --------
    generate_knots : is used under the hood for generating the knots
    make_splrep : the analog of this routine 1D functions
    make_interp_spline : construct an interpolating spline (``s = 0``)
    make_lsq_spline : construct the least-squares spline given the knot vector
    splprep : a FITPACK analog of this routine

    Notes
    -----
    Given a set of :math:`m` data points in :math:`D` dimensions, :math:`\vec{x}_j`,
    with :math:`j=1, ..., m` and :math:`\vec{x}_j = (x_{j; 1}, ..., x_{j; D})`,
    this routine constructs the parametric spline curve :math:`g_a(u)` with
    :math:`a=1, ..., D`, to minimize the sum of jumps, :math:`D_{i; a}`, of the
    ``k``-th derivative at the internal knots (:math:`u_b < t_i < u_e`), where

    .. math::

        D_{i; a} = g_a^{(k)}(t_i + 0) - g_a^{(k)}(t_i - 0)

    Specifically, the routine constructs the spline function :math:`g(u)` which
    minimizes

    .. math::

            \sum_i \sum_{a=1}^D | D_{i; a} |^2 \to \mathrm{min}

    provided that

    .. math::

        \sum_{j=1}^m \sum_{a=1}^D (w_j \times (g_a(u_j) - x_{j; a}))^2 \leqslant s

    where :math:`u_j` is the value of the parameter corresponding to the data point
    :math:`(x_{j; 1}, ..., x_{j; D})`, and :math:`s > 0` is the input parameter.

    In other words, we balance maximizing the smoothness (measured as the jumps
    of the derivative, the first criterion), and the deviation of :math:`g(u_j)`
    from the data :math:`x_j` (the second criterion).

    Note that the summation in the second criterion is over all data points,
    and in the first criterion it is over the internal spline knots (i.e.
    those with ``ub < t[i] < ue``). The spline knots are in general a subset
    of data, see `generate_knots` for details.

    .. versionadded:: 1.15.0

    References
    ----------
    .. [1] P. Dierckx, "Algorithms for smoothing data with periodic and
        parametric splines, Computer Graphics and Image Processing",
        20 (1982) 171-184.
    .. [2] P. Dierckx, "Curve and surface fitting with splines", Monographs on
        Numerical Analysis, Oxford University Press, 1993.
    r   r   Nr0   r   r   r   )r   r   TrF   r   )r   stacksqrtr   cumsumr%   r5   r   TrA   r   r   r   r   r   )r   r   r   r   r   r   r=   r   rD   dpr   ccspl1s                r   make_splprepr   O  sO   t 	A 	yAAAh3B36"Q&GRHH!H$$%%,,..E!Q2Y,Avv=AMT-==>>>!!QSAA66699,Q1aBtTTTAq!Q2r
AqA"qA
N
N
NC 
B35"ce!,,,D7Nr   )r   r   r9   numpyr   	_bsplinesr   r   r   r   r    r	   rM   r   r   r   r*   rA   rH   rG   r`   ro   rq   r   r   r   r   r   r   r   rc   r   r   <module>r      s   &                          
- - -*  
  0.! .! .!b #tQT XR XR XR XR XRv %)Tda14 J J J J JH  F F FRc c c c c c c cLG G G&' ' ' ' ' ' ' ' $ZF ZF ZFz "&$41TPT = = = = =@  DTQ!t$ B B B B BJ $41TPT P P P P P P Pr   