🏷️ interpolate: stub the remaining private C- ext and cython modules (#717)

Author: jorenham

scipy-stubs/interpolate/_dierckx.pyi

@@ -0,0 +1,59 @@
+# defined in scipy/interpolate/src/_dierckxmodule.cc
+
+import numpy as np
+import optype.numpy as onp
+
+def _coloc(
+    x: onp.Array1D[np.float64], t: onp.Array1D[np.float64], k: int, ab: onp.Array2D[np.float64], offset: int = 0, /
+) -> None: ...  # undocumented
+def _coloc_nd(
+    xvals: onp.Array2D[np.float64],
+    _t: onp.Array2D[np.float64],
+    len_t: onp.Array1D[np.int64],
+    k: onp.Array1D[np.int64],
+    _indices_k1d: onp.Array2D[np.int64],
+    _cstrides: onp.Array1D[np.int64],
+    /,
+) -> None: ...  # undocumented
+def _norm_eq_lsq(
+    x: onp.Array1D[np.float64],
+    t: onp.Array1D[np.float64],
+    k: int,
+    y: onp.Array2D[np.float64],
+    w: onp.Array1D[np.float64],
+    ab: onp.Array2D[np.float64],
+    rhs: onp.Array2D[np.float64],
+    /,
+) -> None: ...  # undocumented
+
+#
+def data_matrix(
+    x: onp.Array1D[np.float64], t: onp.Array1D[np.float64], k: int, w: onp.Array1D[np.float64], extrapolate: bool = False, /
+) -> onp.Array2D[np.float64]: ...  # undocumented
+def evaluate_all_bspl(
+    t: onp.Array1D[np.float64], k: int, xval: float, m: int, mu: int = 0, /
+) -> onp.Array1D[np.float64]: ...  # undocumented
+def evaluate_ndbspline(
+    xi: onp.Array2D[np.float64],
+    t: onp.Array2D[np.float64],
+    len_t: onp.Array1D[np.int64],
+    k: onp.Array1D[np.int64],
+    nu: onp.Array1D[np.int64],
+    extrapolate: bool,
+    c1r: onp.Array1D[np.float64],
+    num_c_tr: int,
+    strides_c1r: onp.Array1D[np.int64],
+    indices_k1d: onp.Array2D[np.int64],
+    /,
+) -> onp.Array2D[np.float64]: ...  # undocumented
+def evaluate_spline(
+    t: onp.Array1D[np.float64], c: onp.Array2D[np.float64], k: int, xp: onp.Array1D[np.float64], nu: int, extrapolate: bool
+) -> onp.Array2D[np.float64]: ...  # undocumented
+def find_interval(t: onp.Array1D[np.float64], k: int, xval: float, prev_l: int, extrapolate: bool) -> int: ...  # undocumented
+def fpback(R: onp.Array2D[np.float64], nc: int, y: onp.Array2D[np.float64]) -> onp.Array2D[np.float64]: ...  # undocumented
+def fpknot(
+    x: onp.Array1D[np.float64], t: onp.Array1D[np.float64], k: int, residuals: onp.Array2D[np.float64]
+) -> float: ...  # undocumented
+def qr_reduce(
+    a: onp.Array2D[np.float64], offset: int, nc: int, y: onp.Array2D[np.float64], startrow: int = 1
+) -> None: ...  # undocumented

scipy-stubs/interpolate/_fitpack.pyi

@@ -0,0 +1,49 @@
+# defined in scipy/interpolate/src/_fitpackmodule.c
+
+from _typeshed import Incomplete
+
+import numpy as np
+import optype.numpy as onp
+
+def _insert(
+    iopt: int, t: onp.Array1D[np.float64], c: onp.Array2D[np.float64], k: int, x: onp.Array1D[np.float64], m: int, /
+) -> tuple[onp.Array1D[np.float64], onp.Array1D[np.float64], int]: ...
+def _parcur(
+    x: onp.Array1D[np.float64],
+    w: onp.Array1D[np.float64],
+    u: onp.Array1D[np.float64],
+    ub: onp.Array1D[np.float64],
+    ue: onp.Array1D[np.float64],
+    k: int,
+    iopt: int,
+    ipar: int,
+    s: onp.Array1D[np.float64],
+    t: onp.Array1D[np.float64],
+    nest: int,
+    wrk: onp.Array1D[np.float64],
+    iwrk: int,
+    per: float,
+    /,
+) -> tuple[onp.Array1D[np.float64], onp.Array1D[np.float64], dict[str, Incomplete]]: ...
+def _surfit(
+    x: onp.Array1D[np.float64],
+    y: onp.Array1D[np.float64],
+    z: onp.Array1D[np.float64],
+    w: onp.Array1D[np.float64],
+    xb: float,
+    xe: float,
+    yb: float,
+    ye: float,
+    kx: int,
+    ky: int,
+    iopt: int,
+    s: onp.Array1D[np.float64],
+    eps: float,
+    tx: onp.Array1D[np.float64],
+    ty: onp.Array1D[np.float64],
+    nxest: int,
+    nyest: int,
+    wrk: onp.Array1D[np.float64],
+    lwrk1: int,
+    lwrk2: int,
+) -> tuple[onp.Array1D[np.float64], onp.Array1D[np.float64], onp.Array1D[np.float64], dict[str, Incomplete]]: ...

scipy-stubs/interpolate/_ppoly.pyi

@@ -0,0 +1,43 @@
+# defined in scipy/interpolate/_ppoly.pyx
+
+import numpy as np
+import optype.numpy as onp
+
+def _croots_poly1(c: onp.Array1D[np.float64], w: onp.Array1D[np.complex128], y: float = 0.0) -> None: ...  # undocumented
+def evaluate(
+    c: onp.Array3D[np.float64 | np.complex128],
+    x: onp.Array1D[np.float64],
+    xp: onp.Array1D[np.float64],
+    dx: int,
+    extrapolate: bool,
+    out: onp.Array2D[np.float64 | np.complex128],
+) -> None: ...  # undocumented
+def evaluate_bernstein(
+    c: onp.Array3D[np.float64 | np.complex128],
+    x: onp.Array1D[np.float64],
+    xp: onp.Array1D[np.float64],
+    nu: int,
+    extrapolate: bool,
+    out: onp.Array2D[np.float64 | np.complex128],
+) -> None: ...  # undocumented
+def evaluate_nd(
+    c: onp.Array3D[np.float64 | np.complex128],
+    xs: tuple[onp.Array1D[np.float64], ...],
+    ks: onp.Array1D[np.int32 | np.int64],
+    xp: onp.Array2D[np.float64],
+    dx: onp.Array2D[np.int32 | np.int64],
+    extrapolate: bool,
+    out: onp.Array2D[np.float64 | np.complex128],
+) -> None: ...  # undocumented
+def fix_continuity(c: onp.Array3D[np.float64 | np.complex128], x: onp.Array1D[np.float64], order: int) -> None: ...
+def integrate(
+    c: onp.Array3D[np.float64 | np.complex128],
+    x: onp.Array1D[np.float64],
+    a: float,
+    b: float,
+    extrapolate: bool,
+    out: onp.Array1D[np.float64 | np.complex128],
+) -> None: ...  # undocumented
+def real_roots(
+    c: onp.Array3D[np.float64], x: onp.Array1D[np.float64], y: float, report_discont: bool, extrapolate: bool
+) -> list[onp.Array1D[np.float64]]: ...  # undocumented

scipy-stubs/interpolate/_rgi_cython.pyi

@@ -0,0 +1,21 @@
+# defined in scipy/interpolate/_rgi_cython.pyx
+
+from typing_extensions import TypeVar
+
+import numpy as np
+import optype.numpy as onp
+
+_Inexact64T = TypeVar("_Inexact64T", np.float64, np.complex128)
+
+###
+
+def evaluate_linear_2d(
+    values: onp.Array2D[_Inexact64T],
+    indices: onp.Array2D[np.intp],
+    norm_distances: onp.Array2D[np.float64],
+    grid: tuple[onp.Array1D[np.float64], ...],
+    out: onp.Array2D[_Inexact64T],
+) -> onp.Array2D[_Inexact64T]: ...
+def find_indices(
+    grid: tuple[onp.Array1D[np.float64], ...], xi: onp.Array2D[np.float64]
+) -> tuple[onp.Array2D[np.intp], onp.Array2D[np.float64]]: ...