cubic.h

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <iostream>
#include <cmath>
#include <variant>
 
#include "../config.h"
#include "../util/arrays.h"
#include "../util/misc.h"
#include "../util/spline.h"
 
namespace alfi::spline {
    template <typename Number = DefaultNumber, template <typename, typename...> class Container = DefaultContainer>
    class CubicSpline {
    public:
        struct Conditions final {
            struct Clamped final {
                Clamped(SizeT point_idx, Number df) : point_idx(std::move(point_idx)), df(std::move(df)) {}
                SizeT point_idx;
                Number df;
            };
            struct FixedSecond final {
                FixedSecond(SizeT point_idx, Number d2f) : point_idx(std::move(point_idx)), d2f(std::move(d2f)) {}
                SizeT point_idx;
                Number d2f;
            };
            struct FixedThird final {
                FixedThird(SizeT segment_idx, Number d3f) : segment_idx(std::move(segment_idx)), d3f(std::move(d3f)) {}
                SizeT segment_idx;
                Number d3f;
            };
            struct NotAKnot final {
                explicit NotAKnot(SizeT point_idx) : point_idx(std::move(point_idx)) {}
                SizeT point_idx;
            };
        };
 
        using Condition = std::variant<typename Conditions::Clamped,
                                       typename Conditions::FixedSecond,
                                       typename Conditions::FixedThird,
                                       typename Conditions::NotAKnot>;
 
        struct Types final {
            struct Natural final {};
            struct NotAKnot final {};
            struct Periodic final {};
            struct ParabolicEnds final {};
            struct Clamped final {
                Clamped(Number df_1, Number df_n) : df_1(std::move(df_1)), df_n(std::move(df_n)) {}
                Number df_1, df_n;
            };

            struct FixedSecond final {
                FixedSecond(Number d2f_1, Number d2f_n) : d2f_1(std::move(d2f_1)), d2f_n(std::move(d2f_n)) {}
                Number d2f_1, d2f_n;
            };

            struct FixedThird final {
                FixedThird(Number d3f_1, Number d3f_n) : d3f_1(std::move(d3f_1)), d3f_n(std::move(d3f_n)) {}
                Number d3f_1, d3f_n;
            };

            struct NotAKnotStart final {};
            struct NotAKnotEnd final {};
            struct SemiNotAKnot final {};
            struct Custom final {
                Custom(Condition condition1, Condition condition2) : cond1(std::move(condition1)), cond2(std::move(condition2)) {}
                Condition cond1, cond2;
            };
            using Default = Natural;
        };
 
        using Type = std::variant<typename Types::Natural,
                                  typename Types::NotAKnot,
                                  typename Types::Periodic,
                                  typename Types::ParabolicEnds,
                                  typename Types::Clamped,
                                  typename Types::FixedSecond,
                                  typename Types::FixedThird,
                                  typename Types::NotAKnotStart,
                                  typename Types::NotAKnotEnd,
                                  typename Types::SemiNotAKnot,
                                  typename Types::Custom>;
 
        static Container<Number> compute_coeffs(const Container<Number>& X, const Container<Number>& Y, Type type = typename Types::Default{}) {
            if (X.size() != Y.size()) {
                std::cerr << "Error in function " << __FUNCTION__
                          << ": Vectors X (of size " << X.size()
                          << ") and Y (of size " << Y.size()
                          << ") are not the same size. Returning an empty array..." << std::endl;
                return {};
            }
 
            const SizeT n = X.size();
 
            if (n <= 1) {
                return util::spline::simple_spline<Number,Container>(X, Y, 3);
            }
 
            if (std::holds_alternative<typename Types::Natural>(type)) {
                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::FixedSecond{0, 0}, typename Conditions::FixedSecond{n - 1, 0}});
            } else if (std::holds_alternative<typename Types::NotAKnot>(type)) {
                if (n <= 4) {
                    return util::spline::simple_spline<Number,Container>(X, Y, 3);
                }
                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::NotAKnot{1}, typename Conditions::NotAKnot{n - 2}});
            } else if (std::holds_alternative<typename Types::ParabolicEnds>(type)) {
                if (n <= 3) {
                    return util::spline::simple_spline<Number,Container>(X, Y, 3);
                }
                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::FixedThird{0, 0}, typename Conditions::FixedThird{n - 2, 0}});
            } else if (const auto* c = std::get_if<typename Types::Clamped>(&type)) {
                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::Clamped{0, c->df_1}, typename Conditions::Clamped{n - 1, c->df_n}});
            } else if (const auto* fs = std::get_if<typename Types::FixedSecond>(&type)) {
                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::FixedSecond{0, fs->d2f_1}, typename Conditions::FixedSecond{n - 1, fs->d2f_n}});
            } else if (const auto* ft = std::get_if<typename Types::FixedThird>(&type)) {
                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::FixedThird{0, ft->d3f_1}, typename Conditions::FixedThird{n - 2, ft->d3f_n}});
            } else if (std::holds_alternative<typename Types::NotAKnotStart>(type)) {
                if (n <= 4) {
                    return util::spline::simple_spline<Number,Container>(X, Y, 3);
                }
                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::NotAKnot{1}, typename Conditions::NotAKnot{2}});
            } else if (std::holds_alternative<typename Types::NotAKnotEnd>(type)) {
                if (n <= 4) {
                    return util::spline::simple_spline<Number,Container>(X, Y, 3);
                }
                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::NotAKnot{n - 3}, typename Conditions::NotAKnot{n - 2}});
            } else if (std::holds_alternative<typename Types::SemiNotAKnot>(type)) {
                if (n <= 4) {
                    return util::spline::simple_spline<Number,Container>(X, Y, 3);
                }
                return util::arrays::mean(compute_coeffs(X, Y, typename Types::NotAKnotStart{}), compute_coeffs(X, Y, typename Types::NotAKnotEnd{}));
            }
 
            Container<Number> coeffs(4 * (n - 1));
 
            // i1 - index of first constructed segment
            // i2 - index of first after last constructed segment
            SizeT i1 = 0, i2 = n - 1;
 
            const auto dX = util::arrays::diff(X), dY = util::arrays::diff(Y);
 
            if (std::holds_alternative<typename Types::Periodic>(type)) {
                if (n == 2) {
                    return util::spline::simple_spline<Number,Container>(X, Y, 3);
                }
 
                Container<Number> C(n);
                if (n == 3) {
                    C[0] = 3 * (dY[0]/dX[0] - dY[1]/dX[1]) / (dX[0] + dX[1]);
                    C[1] = -C[0];
                    C[2] = C[0];
                } else {
                    Container<Number> diag(n - 1), right(n - 1);
                    for (SizeT i = 0; i < n - 2; ++i) {
                        diag[i] = 2 * (dX[i] + dX[i+1]);
                        right[i] = 3 * (dY[i+1]/dX[i+1] - dY[i]/dX[i]);
                    }
                    diag[n-2] = 2 * (dX[n-2] + dX[0]);
                    right[n-2] = 3 * (dY[0]/dX[0] - dY[n-2]/dX[n-2]);
 
                    Container<Number> last_row(n - 2), last_col(n - 2);
                    last_row[0] = last_col[0] = dX[0];
                    for (SizeT i = 1; i < n - 3; ++i) {
                        last_row[i] = last_col[i] = 0;
                    }
                    last_row[n-3] = last_col[n-3] = dX[n-2];
 
                    for (SizeT i = 0; i < n - 3; ++i) {
                        const auto m1 = dX[i+1] / diag[i];
                        diag[i+1] -= m1 * dX[i+1];
                        last_col[i+1] -= m1 * last_col[i];
                        right[i+1] -= m1 * right[i];
                        const auto m2 = last_row[i] / diag[i];
                        last_row[i+1] -= m2 * dX[i+1];
                        diag[n-2] -= m2 * last_col[i];
                        right[n-2] -= m2 * right[i];
                    }
                    diag[n-2] -= last_row[n-3] / diag[n-3] * last_col[n-3];
                    right[n-2] -= last_row[n-3] / diag[n-3] * right[n-3];
 
                    C[n-1] = right[n-2] / diag[n-2];
                    C[n-2] = (right[n-3] - last_col[n-3] * C[n-1]) / diag[n-3];
                    for (SizeT i = n - 3; i >= 1; --i) {
                        C[i] = (right[i-1] - dX[i]*C[i+1] - last_col[i-1]*C[n-1]) / diag[i-1];
                    }
                    C[0] = C[n-1];
                }
 
                for (SizeT i = 0, index = 0; i < n - 1; ++i) {
                    coeffs[index++] = (C[i+1] - C[i]) / (3*dX[i]);
                    coeffs[index++] = C[i];
                    coeffs[index++] = dY[i]/dX[i] - dX[i] * ((2*C[i] + C[i+1]) / 3);
                    coeffs[index++] = Y[i];
                }
            } else if (auto* custom = std::get_if<typename Types::Custom>(&type)) {
                // special case
                if (std::holds_alternative<typename Conditions::NotAKnot>(custom->cond1)
                        && std::holds_alternative<typename Conditions::NotAKnot>(custom->cond2)
                        && n <= 3) {
                    return util::spline::simple_spline<Number,Container>(X, Y, 3);
                }
 
                // check conditions
                for (const auto* cond : {&custom->cond1, &custom->cond2}) {
                    if (const auto* c = std::get_if<typename Conditions::Clamped>(cond)) {
                        if (c->point_idx < 0 || c->point_idx >= n) {
                            std::cerr << "Error in function " << __FUNCTION__
                                      << ": point index for condition 'Clamped' is out of bounds: "
                                      << "expected to be non-negative and less than " << n << ", but got " << c->point_idx
                                      << ". Returning an empty array..." << std::endl;
                            return {};
                        }
                    } else if (const auto* fs = std::get_if<typename Conditions::FixedSecond>(cond)) {
                        if (fs->point_idx < 0 || fs->point_idx >= n) {
                            std::cerr << "Error in function " << __FUNCTION__
                                      << ": point index for condition 'FixedSecond' is out of bounds: "
                                      << "expected to be non-negative and less than " << n << ", but got " << fs->point_idx
                                      << ". Returning an empty array..." << std::endl;
                            return {};
                        }
                    } else if (const auto* ft = std::get_if<typename Conditions::FixedThird>(cond)) {
                        if (ft->segment_idx < 0 || ft->segment_idx >= n - 1) {
                            std::cerr << "Error in function " << __FUNCTION__
                                      << ": segment index for condition 'FixedThird' is out of bounds: "
                                      << "expected to be non-negative and less than " << n - 1 << ", but got " << ft->segment_idx
                                      << ". Returning an empty array..." << std::endl;
                            return {};
                        }
                    } else if (const auto* nak = std::get_if<typename Conditions::NotAKnot>(cond)) {
                        if (nak->point_idx < 1 || nak->point_idx >= n - 1) {
                            std::cerr << "Error in function " << __FUNCTION__
                                      << ": point index for condition 'NotAKnot' is out of bounds: "
                                      << "expected to be positive and less than " << n - 1 << ", but got " << nak->point_idx
                                      << ". Returning an empty array..." << std::endl;
                            return {};
                        }
                    } else {
                        std::cerr << "Error in function " << __FUNCTION__ << ": Unknown condition of 'Custom' type. This should not have happened. "
                                  << "Please report this to the developers. Returning an empty array..." << std::endl;
                        return {};
                    }
                }
 
                // check if same conditions on one point and return a somewhat reasonable compromise solution
                if (const auto c1 = std::get_if<typename Conditions::Clamped>(&custom->cond1),
                            c2 = std::get_if<typename Conditions::Clamped>(&custom->cond2);
                            c1 && c2) {
                    if (c1->point_idx == c2->point_idx) {
                        const auto df = (c1->df + c2->df) / 2;
                        if (c1->point_idx < n / 2) {
                            return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::Clamped{c1->point_idx, df}, typename Conditions::NotAKnot{n - 2}});
                        } else {
                            return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::NotAKnot{1}, typename Conditions::Clamped{c1->point_idx, df}});
                        }
                    }
                } else if (const auto fs1 = std::get_if<typename Conditions::FixedSecond>(&custom->cond1),
                            fs2 = std::get_if<typename Conditions::FixedSecond>(&custom->cond2);
                            fs1 && fs2) {
                    if (fs1->point_idx == fs2->point_idx) {
                        const auto d2f = (fs1->d2f + fs2->d2f) / 2;
                        if (fs1->point_idx < n / 2) {
                            return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::FixedSecond{fs1->point_idx, d2f}, typename Conditions::NotAKnot{n - 2}});
                        } else {
                            return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::NotAKnot{1}, typename Conditions::FixedSecond{fs1->point_idx, d2f}});
                        }
                    }
                } else if (const auto ft1 = std::get_if<typename Conditions::FixedThird>(&custom->cond1),
                            ft2 = std::get_if<typename Conditions::FixedThird>(&custom->cond2);
                            ft1 && ft2) {
                    if (ft1->segment_idx == ft2->segment_idx) {
                        if (n == 2) {
                            const auto i = ft1->segment_idx;
                            const auto c = -dX[i] * (ft1->d3f + ft2->d3f) / 8;
                            coeffs[4*i] = -2*c / (3*dX[i]);
                            coeffs[4*i+1] = c;
                            coeffs[4*i+2] = dY[i]/dX[i] - dX[i] * c / 3;
                            coeffs[4*i+3] = Y[i];
                            i1 = i;
                            i2 = i + 1;
                            goto post_main_block;
                        } else {
                            const auto d3f = (ft1->d3f + ft2->d3f) / 2;
                            if (ft1->segment_idx < n / 2) {
                                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::FixedThird{ft1->segment_idx, d3f}, typename Conditions::NotAKnot{n - 2}});
                            } else {
                                return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::NotAKnot{1}, typename Conditions::FixedThird{ft1->segment_idx, d3f}});
                            }
                        }
                    }
                } else if (const auto nak1 = std::get_if<typename Conditions::NotAKnot>(&custom->cond1),
                            nak2 = std::get_if<typename Conditions::NotAKnot>(&custom->cond2);
                            nak1 && nak2) {
                    if (nak1->point_idx == nak2->point_idx) {
                        if (nak1->point_idx < n / 2) {
                            return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::NotAKnot{nak1->point_idx}, typename Conditions::NotAKnot{n - 2}});
                        } else {
                            return compute_coeffs(X, Y, typename Types::Custom{typename Conditions::NotAKnot{1}, typename Conditions::NotAKnot{nak1->point_idx}});
                        }
                    }
                }
 
                const auto set_segment_indices = [&]() {
                    std::visit(util::misc::overload{
                        [&](const typename Conditions::Clamped& c) { i1 = c.point_idx; },
                        [&](const typename Conditions::FixedSecond& fs) { i1 = fs.point_idx; },
                        [&](const typename Conditions::FixedThird& ft) { i1 = ft.segment_idx; },
                        [&](const typename Conditions::NotAKnot& nak) { i1 = nak.point_idx - 1; },
                    }, custom->cond1);
                    std::visit(util::misc::overload{
                        [&](const typename Conditions::Clamped& c) { i2 = c.point_idx; },
                        [&](const typename Conditions::FixedSecond& fs) { i2 = fs.point_idx; },
                        [&](const typename Conditions::FixedThird& ft) { i2 = ft.segment_idx + 1; },
                        [&](const typename Conditions::NotAKnot& nak) { i2 = nak.point_idx + 1; },
                    }, custom->cond2);
                };
                set_segment_indices();
                // swap if wrong order and set again
                if (i2 <= i1) {
                    std::swap(custom->cond1, custom->cond2);
                    set_segment_indices();
                }
 
                // special case: both conditions on one point
                // only possible for Clamped and FixedSecond
                if (i2 <= i1) {
                    assert(i1 == i2);
                    const auto i = i1;
                    const auto* c
                        = std::holds_alternative<typename Conditions::Clamped>(custom->cond1)
                        ? std::get_if<typename Conditions::Clamped>(&custom->cond1)
                        : std::get_if<typename Conditions::Clamped>(&custom->cond2);
                    const auto* fs
                        = std::holds_alternative<typename Conditions::FixedSecond>(custom->cond1)
                        ? std::get_if<typename Conditions::FixedSecond>(&custom->cond1)
                        : std::get_if<typename Conditions::FixedSecond>(&custom->cond2);
                    assert(c && fs);
                    if (i > 0) {
                        coeffs[4*(i-1)] = (fs->d2f/2 - c->df/dX[i-1] + dY[i-1]/dX[i-1]/dX[i-1]) / dX[i-1];
                        coeffs[4*(i-1)+1] = 3*c->df/dX[i-1] - 3*dY[i-1]/dX[i-1]/dX[i-1] - fs->d2f;
                        coeffs[4*(i-1)+2] = 3*dY[i-1]/dX[i-1] + fs->d2f*dX[i-1]/2 - 2*c->df;
                        coeffs[4*(i-1)+3] = Y[i-1];
                        i1 = i - 1;
                    }
                    if (i < n - 1) {
                        coeffs[4*i] = (dY[i]/dX[i]/dX[i] - c->df/dX[i] - fs->d2f/2) / dX[i];
                        coeffs[4*i+1] = fs->d2f/2;
                        coeffs[4*i+2] = c->df;
                        coeffs[4*i+3] = Y[i];
                        i2 = i + 1;
                    }
                    goto post_main_block;
                }
 
                // number of points in the "subspline"
                const auto m = i2 - i1 + 1;
 
                // common equations
                Container<Number> lower(m), diag(m), upper(m), right(m);
                for (SizeT i = i1 + 1; i < i1 + m - 1; ++i) {
                    const auto j = i - i1;
                    lower[j] = dX[i-1];
                    diag[j] = 2 * (dX[i-1] + dX[i]);
                    upper[j] = dX[i];
                    right[j] = 3 * (dY[i]/dX[i] - dY[i-1]/dX[i-1]);
                }
                lower[0] = NAN;
                upper[m-1] = NAN;
 
                // unique equations
                std::visit(util::misc::overload{
                    [&](const typename Conditions::Clamped& c) {
                        diag[0] = 2*dX[i1];
                        upper[0] = dX[i1];
                        right[0] = 3 * (dY[i1]/dX[i1] - c.df);
                    },
                    [&](const typename Conditions::FixedSecond& fs) {
                        diag[0] = 1;
                        upper[0] = 0;
                        right[0] = fs.d2f / 2;
                    },
                    [&](const typename Conditions::FixedThird& ft) {
                        diag[0] = 1;
                        upper[0] = -1;
                        right[0] = -dX[i1] * ft.d3f / 2;
                    },
                    [&](const typename Conditions::NotAKnot&) {
                        diag[0] = dX[i1] - dX[i1+1];
                        upper[0] = 2*dX[i1] + dX[i1+1];
                        right[0] = dX[i1] / (dX[i1]+dX[i1+1]) * 3 * (dY[i1+1]/dX[i1+1] - dY[i1]/dX[i1]);
                    },
                }, custom->cond1);
                std::visit(util::misc::overload{
                    [&](const typename Conditions::Clamped& c) {
                        lower[m-1] = dX[i1+m-2];
                        diag[m-1] = 2*dX[i1+m-2];
                        right[m-1] = 3 * (c.df - dY[i1+m-2]/dX[i1+m-2]);
                    },
                    [&](const typename Conditions::FixedSecond& fs) {
                        lower[m-1] = 0;
                        diag[m-1] = 1;
                        right[m-1] = fs.d2f / 2;
                    },
                    [&](const typename Conditions::FixedThird& ft) {
                        lower[m-1] = -1;
                        diag[m-1] = 1;
                        right[m-1] = dX[i1+m-2] * ft.d3f / 2;
                    },
                    [&](const typename Conditions::NotAKnot&) {
                        lower[m-1] = 2*dX[i1+m-2] + dX[i1+m-3];
                        diag[m-1] = dX[i1+m-2] - dX[i1+m-3];
                        right[m-1] = dX[i1+m-2] / (dX[i1+m-2]+dX[i1+m-3]) * 3 * (dY[i1+m-2]/dX[i1+m-2] - dY[i1+m-3]/dX[i1+m-3]);
                    },
                }, custom->cond2);
 
                const auto C = util::linalg::tridiag_solve<Number,Container>(
                    std::move(lower), std::move(diag), std::move(upper), std::move(right));
 
                for (SizeT i = 0, index = 4 * i1; i < m - 1; ++i) {
                    const auto j = i + i1;
                    coeffs[index++] = (C[i+1] - C[i]) / (3*dX[j]);
                    coeffs[index++] = C[i];
                    coeffs[index++] = dY[j]/dX[j] - dX[j] * ((2*C[i] + C[i+1]) / 3);
                    coeffs[index++] = Y[j];
                }
 
                // additional corrections
                if (const auto* ft = std::get_if<typename Conditions::FixedThird>(&custom->cond1)) {
                    coeffs[4*i1] = ft->d3f / 6;
                }
                if (const auto* ft = std::get_if<typename Conditions::FixedThird>(&custom->cond2)) {
                    coeffs[4*(i2-1)] = ft->d3f / 6;
                }
            } else {
                std::cerr << "Error in function " << __FUNCTION__ << ": Unknown type. This should not have happened. "
                          << "Please report this to the developers. Returning an empty array..." << std::endl;
                return {};
            }
 
        post_main_block:
            for (SizeT iter = 0; iter < i1; ++iter) {
                const auto i = i1 - 1 - iter;
                coeffs[4*i] = (coeffs[4*(i+1)+1] - coeffs[4*(i+1)+2]/dX[i] + dY[i]/dX[i]/dX[i]) / dX[i];
                coeffs[4*i+1] = coeffs[4*(i+1)+1] - 3 * coeffs[4*i] * dX[i];
                coeffs[4*i+2] = dY[i]/dX[i] - coeffs[4*i]*dX[i]*dX[i] - coeffs[4*i+1]*dX[i];
                coeffs[4*i+3] = Y[i];
            }
 
            for (SizeT i = i2; i < n - 1; ++i) {
                coeffs[4*i+3] = Y[i];
                coeffs[4*i+2] = 3 * coeffs[4*(i-1)]*dX[i-1]*dX[i-1] + 2 * coeffs[4*(i-1)+1]*dX[i-1] + coeffs[4*(i-1)+2];
                coeffs[4*i+1] = 3 * coeffs[4*(i-1)]*dX[i-1] + coeffs[4*(i-1)+1];
                coeffs[4*i] = (dY[i]/dX[i]/dX[i] - coeffs[4*i+1] - coeffs[4*i+2]/dX[i]) / dX[i];
            }
 
            return coeffs;
        }
 
        CubicSpline() = default;
 
        template <typename ContainerXType>
        CubicSpline(ContainerXType&& X, const Container<Number>& Y, Type type = typename Types::Default{}) {
            construct(std::forward<ContainerXType>(X), Y, type);
        }
 
        CubicSpline(const CubicSpline& other) = default;
        CubicSpline(CubicSpline&& other) noexcept = default;
 
        CubicSpline& operator=(const CubicSpline& other) = default;
        CubicSpline& operator=(CubicSpline&& other) noexcept = default;
 
        template <typename ContainerXType>
        void construct(ContainerXType&& X, const Container<Number>& Y, Type type = typename Types::Default{}) {
            if (X.size() != Y.size()) {
                std::cerr << "Error in function " << __FUNCTION__
                          << ": Vectors X (of size " << X.size()
                          << ") and Y (of size " << Y.size()
                          << ") are not the same size. Doing nothing..." << std::endl;
                return;
            }
            auto coeffs = compute_coeffs(X, Y, type);
            if (coeffs.empty() && !X.empty()) {
                std::cerr << "Error in function " << __FUNCTION__
                          << ": failed to construct coefficients. Not changing the object..." << std::endl;
                return;
            }
            _type = type;
            _X = std::forward<ContainerXType>(X);
            _coeffs = std::move(coeffs);
        }
 
        Number eval(Number x) const {
            return eval(x, std::distance(_X.begin(), util::misc::first_leq_or_begin(_X.begin(), _X.end(), x)));
        }
        Number eval(Number x, SizeT segment) const {
            if (_coeffs.empty()) {
                return NAN;
            } else if (_coeffs.size() == 1) {
                return _coeffs[0];
            }
            segment = std::clamp(segment, static_cast<SizeT>(0), static_cast<SizeT>(_X.size() - 2));
            x = x - _X[segment];
            return ((_coeffs[4*segment] * x + _coeffs[4*segment+1]) * x + _coeffs[4*segment+2]) * x + _coeffs[4*segment+3];
        }
 
        Container<Number> eval(const Container<Number>& xx, bool sorted = true) const {
            Container<Number> result(xx.size());
            if (sorted) {
                for (SizeT i = 0, i_x = 0; i < xx.size(); ++i) {
                    const Number x = xx[i];
                    while (i_x + 1 < _X.size() && x >= _X[i_x+1])
                        ++i_x;
                    result[i] = eval(x, i_x);
                }
            } else {
                for (SizeT i = 0; i < xx.size(); ++i) {
                    result[i] = eval(xx[i]);
                }
            }
            return result;
        }
 
        Number operator()(Number x) const {
            return eval(x);
        }
        Container<Number> operator()(const Container<Number>& xx) const {
            return eval(xx);
        }
 
        Type type() const {
            return _type;
        }
 
        const Container<Number>& X() const & {
            return _X;
        }
        Container<Number>&& X() && {
            return std::move(_X);
        }
 
        const Container<Number>& coeffs() const & {
            return _coeffs;
        }
        Container<Number>&& coeffs() && {
            return std::move(_coeffs);
        }
 
        static std::pair<SizeT, SizeT> segment(SizeT index) {
            return {4*index, 4*(index+1)};
        }
 
    private:
        Type _type = typename Types::Default{};
        Container<Number> _X = {};
        Container<Number> _coeffs = {};
    };
}