TableInterpolator.hpp

Go to the documentation of this file.

// Vulcan Table Interpolation Wrapper
// Thin wrappers around janus::Interpolator and janus::ScatteredInterpolator
// for atmospheric/aero property lookups
#pragma once
 
#include <janus/math/Interpolate.hpp>
#include <janus/math/ScatteredInterpolator.hpp>
#include <vector>
 
namespace vulcan {
 
// Re-export RBF kernel types for convenience
using janus::RBFKernel;
 
// ============================================================================
// Table1D - One-dimensional table interpolation
// ============================================================================

class Table1D {
  private:
    janus::Interpolator m_interp;
 
  public:
    Table1D(
        const janus::NumericVector &x, const janus::NumericVector &y,
        janus::InterpolationMethod method = janus::InterpolationMethod::Linear)
        : m_interp(x, y, method) {}

    template <janus::JanusScalar Scalar>
    Scalar operator()(const Scalar &x) const {
        return m_interp(x);
    }

    template <typename Derived>
    auto operator()(const Eigen::MatrixBase<Derived> &x) const {
        return m_interp(x);
    }

    janus::InterpolationMethod method() const { return m_interp.method(); }

    bool valid() const { return m_interp.valid(); }
};
 
// ============================================================================
// TableND - N-dimensional table interpolation
// ============================================================================

class TableND {
  private:
    janus::Interpolator m_interp;
    int m_dims;
 
  public:
    TableND(
        const std::vector<janus::NumericVector> &grid_points,
        const janus::NumericVector &values,
        janus::InterpolationMethod method = janus::InterpolationMethod::Linear)
        : m_interp(grid_points, values, method),
          m_dims(static_cast<int>(grid_points.size())) {}

    template <janus::JanusScalar Scalar>
    Scalar operator()(const janus::JanusVector<Scalar> &x) const {
        return m_interp(x);
    }

    template <typename Derived>
    auto operator()(const Eigen::MatrixBase<Derived> &x) const {
        return m_interp(x);
    }

    int dims() const { return m_dims; }

    janus::InterpolationMethod method() const { return m_interp.method(); }

    bool valid() const { return m_interp.valid(); }
};
 
// ============================================================================
// ScatteredTable1D - One-dimensional scattered data interpolation
// ============================================================================

class ScatteredTable1D {
  private:
    janus::ScatteredInterpolator m_interp;
 
  public:
    ScatteredTable1D(const janus::NumericVector &x,
                     const janus::NumericVector &y, int grid_resolution = 50,
                     RBFKernel kernel = RBFKernel::ThinPlateSpline)
        : m_interp(x, y, grid_resolution, kernel) {}

    template <janus::JanusScalar Scalar>
    Scalar operator()(const Scalar &x) const {
        return m_interp(x);
    }

    double reconstruction_error() const {
        return m_interp.reconstruction_error();
    }

    int dims() const { return m_interp.dims(); }

    bool valid() const { return m_interp.valid(); }
};
 
// ============================================================================
// ScatteredTableND - N-dimensional scattered data interpolation
// ============================================================================

class ScatteredTableND {
  private:
    janus::ScatteredInterpolator m_interp;
 
  public:
    ScatteredTableND(
        const janus::NumericMatrix &points, const janus::NumericVector &values,
        int grid_resolution = 20, RBFKernel kernel = RBFKernel::ThinPlateSpline,
        double epsilon = 1.0,
        janus::InterpolationMethod method = janus::InterpolationMethod::Linear)
        : m_interp(points, values, grid_resolution, kernel, epsilon, method) {}

    ScatteredTableND(
        const janus::NumericMatrix &points, const janus::NumericVector &values,
        const std::vector<janus::NumericVector> &grid_points,
        RBFKernel kernel = RBFKernel::ThinPlateSpline, double epsilon = 1.0,
        janus::InterpolationMethod method = janus::InterpolationMethod::Linear)
        : m_interp(points, values, grid_points, kernel, epsilon, method) {}

    template <janus::JanusScalar Scalar>
    Scalar operator()(const janus::JanusVector<Scalar> &x) const {
        return m_interp(x);
    }

    double reconstruction_error() const {
        return m_interp.reconstruction_error();
    }

    int dims() const { return m_interp.dims(); }

    bool valid() const { return m_interp.valid(); }
};
 
} // namespace vulcan