Function derMatrix

cadquery/occ_impl/nurbs.py:1085–1128

Create a sparse design matrix and corresponding derivative matrices.

(u: Array, order: int, dorder: int, knots: Array)

@njit
def derMatrix(u: Array, order: int, dorder: int, knots: Array) -> list[COO]:
    """
    Create a sparse design matrix and corresponding derivative matrices.
    """

    # number of parameter values
    nu = np.size(u)

    # chunk size
    n = order + 1

    # temp chunk storage
    temp = np.zeros((dorder + 1, n))

    # initialize the empty matrix
    rv = []

    for _ in range(dorder + 1):
        rv.append(
            COO(
                i=np.empty(n * nu, dtype=np.int64),
                j=np.empty(n * nu, dtype=np.int64),
                v=np.empty(n * nu),
                shape=(nu, len(knots) - 1 - order),
            )
        )

    # loop over parameter values
    for i in range(nu):
        ui = u[i]

        # find the supporting span
        span = nbFindSpan(ui, order, knots)

        # evaluate non-zero functions
        nbBasisDer(span, ui, order, dorder, knots, temp)

        # update the matrices
        for di in range(dorder + 1):
            rv[di].i[i * n : (i + 1) * n] = i
            rv[di].j[i * n : (i + 1) * n] = span - order + np.arange(n)
            rv[di].v[i * n : (i + 1) * n] = temp[di, :]

    return rv


@njit