fpGridPolar.hpp

Go to the documentation of this file.

/***************************************************************************************************
!                                ____________________  ___   ________ __
!                               / ____/  _/_  __/ __ \/   | / ____/ //_/
!                              / /_   / /  / / / /_/ / /| |/ /   / ,<
!                             / __/ _/ /  / / / ____/ ___ / /___/ /| |
!                            /_/   /___/ /_/ /_/   /_/  |_\____/_/ |_|
!
!                                     A Curve Fitting Package
!
!   fpGridPolar.hpp (class fpGridPolar)
!> @brief Standalone C++ wrapper for fitpack_grid_polar (no fortran-arrays dependency)
!
!   @author Federico Perini
!   @date   2026-08-27
!
!   References :
!     - C. De Boor, "On calculating with b-splines", J Approx Theory 6 (1972) 50-62
!     - M. G. Cox, "The numerical evaluation of b-splines", J Inst Maths Applics 10 (1972) 134-149
!     - P. Dierckx, "Curve and surface fitting with splines", Monographs on numerical analysis,
!                    Oxford university press, 1993.
!
! **************************************************************************************************/
 
#ifndef FPGRIDPOLAR_HPP_INCLUDED
#define FPGRIDPOLAR_HPP_INCLUDED
 
#include "fitpack_config.h"
 
#if HAVE_FXARRAY
#include "fxArrays.hpp"
#endif
 
#include "fitpack_grid_polar_c.h"
#include "fpFitter.hpp"
#include <string>
#include <vector>
#include <cstdint>
#include <memory>
#include <stdexcept>
#include <variant>
#include <optional>
#include <array>
 
static_assert(sizeof(fitpack_grid_polar_c) == sizeof(fitpack_fitter_c),
    "C descriptor layout mismatch: fitpack_grid_polar_c vs fitpack_fitter_c");

class fpGridPolar : public fpFitter {
public:
    // ===========================================================================================
    // Constructors and Destructor
    // ===========================================================================================

    fpGridPolar() : fpFitter(NoAlloc{}) {
        fitpack_grid_polar_c_allocate(as<fitpack_grid_polar_c>(), nullptr);
    }

    ~fpGridPolar() override {
        fitpack_grid_polar_c_destroy(as<fitpack_grid_polar_c>(), nullptr);
    }

    fpGridPolar(const fpGridPolar& other) : fpFitter(NoAlloc{}) {
        *as<fitpack_grid_polar_c>() = fitpack_grid_polar_c_null;
        fitpack_grid_polar_c_copy(as<fitpack_grid_polar_c>(), other.as<fitpack_grid_polar_c>(), false, nullptr);
    }

    fpGridPolar& operator=(const fpGridPolar& other) {
        if (this != &other) {
            fitpack_grid_polar_c_destroy(as<fitpack_grid_polar_c>(), nullptr);
            fitpack_grid_polar_c_copy(as<fitpack_grid_polar_c>(), other.as<fitpack_grid_polar_c>(), false, nullptr);
        }
        return *this;
    }

    fpGridPolar(fpGridPolar&& other) noexcept : fpFitter(NoAlloc{}) {
        *as<fitpack_grid_polar_c>() = fitpack_grid_polar_c_null;
        fitpack_grid_polar_c_move_alloc(as<fitpack_grid_polar_c>(), other.as<fitpack_grid_polar_c>(), nullptr);
    }

    fpGridPolar& operator=(fpGridPolar&& other) noexcept {
        if (this != &other) {
            fitpack_grid_polar_c_destroy(as<fitpack_grid_polar_c>(), nullptr);
            fitpack_grid_polar_c_move_alloc(as<fitpack_grid_polar_c>(), other.as<fitpack_grid_polar_c>(), nullptr);
        }
        return *this;
    }

    explicit fpGridPolar(fitpack_grid_polar_c& c_wrapper, bool move = false) : fpFitter(NoAlloc{}) {
        if (move) {
            fitpack_grid_polar_c_move_alloc(as<fitpack_grid_polar_c>(), &c_wrapper, nullptr);
        } else {
            *as<fitpack_grid_polar_c>() = fitpack_grid_polar_c_null;
            fitpack_grid_polar_c_copy(as<fitpack_grid_polar_c>(), &c_wrapper, false, nullptr);
        }
    }

    explicit fpGridPolar(fitpack_grid_polar_c& c_wrapper, ViewTag) : fpFitter(NoAlloc{}) {
        *as<fitpack_grid_polar_c>() = c_wrapper;
        as<fitpack_grid_polar_c>()->is_pointer = true;
    }
 
    // ===========================================================================================
    // Utility Methods
    // ===========================================================================================

    bool is_allocated() const override {
        return as<fitpack_grid_polar_c>()->cptr != nullptr;
    }

    bool is_pointer() const override {
        return as<fitpack_grid_polar_c>()->is_pointer;
    }

    const char* c_type_name() const override {
        return fitpack_grid_polar_c_c_type_name(*as<fitpack_grid_polar_c>());
    }

    const char* cpp_type_name() const override {
        return "fpGridPolar";
    }

    fitpack_grid_polar_c& c_handle() {
        return *as<fitpack_grid_polar_c>();
    }

    const fitpack_grid_polar_c& c_handle() const {
        return *as<fitpack_grid_polar_c>();
    }

    static fpGridPolar make_view() {
        return fpGridPolar(NoAlloc{});
    }

    fpFitter& as_parent() {
        return static_cast<fpFitter&>(*this);
    }
    const fpFitter& as_parent() const {
        return static_cast<const fpFitter&>(*this);
    }
 
    // ===========================================================================================
    // Method Wrappers (standalone — no fxArray dependency)
    // ===========================================================================================

    virtual void new_points(std::vector<double>& u, std::vector<double>& v, double r, int32_t z_n1, int32_t z_n2, std::vector<double>& z, double* z0 = nullptr) {
        int32_t n = static_cast<int32_t>(u.size());
        fitpack_grid_polar_c_new_points(as<fitpack_grid_polar_c>(), n, u.data(), v.data(), r, z_n1, z_n2, z.data(), z0);
    }
 

    virtual int32_t new_fit(std::vector<double>& u, std::vector<double>& v, double r, int32_t z_n1, int32_t z_n2, std::vector<double>& z, double* z0 = nullptr, double* smoothing = nullptr) {
        int32_t n = static_cast<int32_t>(u.size());
        return fitpack_grid_polar_c_new_fit(as<fitpack_grid_polar_c>(), n, u.data(), v.data(), r, z_n1, z_n2, z.data(), z0, smoothing);
    }
 
 
    virtual void set_origin_BC(double* z0 = nullptr, bool* exact = nullptr, bool* differentiable = nullptr) {
        fitpack_grid_polar_c_set_origin_BC(as<fitpack_grid_polar_c>(), z0, exact, differentiable);
    }
 
    virtual int32_t fit(double* smoothing = nullptr, bool* keep_knots = nullptr) {
        return fitpack_grid_polar_c_fit(as<fitpack_grid_polar_c>(), smoothing, keep_knots);
    }
 
    virtual int32_t least_squares(double* smoothing = nullptr, bool* reset_knots = nullptr) {
        return fitpack_grid_polar_c_least_squares(as<fitpack_grid_polar_c>(), smoothing, reset_knots);
    }
 
    virtual int32_t interpolate(bool* reset_knots = nullptr) {
        return fitpack_grid_polar_c_interpolate(as<fitpack_grid_polar_c>(), reset_knots);
    }
 
    virtual double eval(double u, double v, int32_t* ierr = nullptr) {
        return fitpack_grid_polar_c_gridded_eval_one(as<fitpack_grid_polar_c>(), u, v, ierr);
    }

    virtual std::vector<double> eval(std::vector<double>& u, std::vector<double>& v, int32_t* ierr = nullptr) {
        int32_t n = static_cast<int32_t>(u.size());
        std::vector<double> result(n * n);
        int32_t n_result = 0;
        fitpack_grid_polar_c_gridded_eval_many(as<fitpack_grid_polar_c>(), n, u.data(), v.data(), ierr, result.data(), &n_result, n * n);
        result.resize(n_result);
        return result;
    }
 
 
    int32_t comm_size() const override {
        return fitpack_grid_polar_c_comm_size(as<fitpack_grid_polar_c>());
    }

    void comm_pack(std::vector<double>& buffer) const override {
        int32_t n = static_cast<int32_t>(buffer.size());
        fitpack_grid_polar_c_comm_pack(as<fitpack_grid_polar_c>(), n, buffer.data());
    }
 

    void comm_expand(std::vector<double>& buffer) override {
        int32_t n = static_cast<int32_t>(buffer.size());
        fitpack_grid_polar_c_comm_expand(as<fitpack_grid_polar_c>(), n, buffer.data());
    }
 
 
    virtual void write(const char* filename) {
        fitpack_grid_polar_c_write(as<fitpack_grid_polar_c>(), filename);
    }
 
    double mse() const override {
        return fitpack_grid_polar_c_mse(as<fitpack_grid_polar_c>());
    }
 
    int32_t core_comm_size() const override {
        return fitpack_grid_polar_c_core_comm_size(as<fitpack_grid_polar_c>());
    }

    void core_comm_pack(std::vector<double>& buffer) const override {
        int32_t n = static_cast<int32_t>(buffer.size());
        fitpack_grid_polar_c_core_comm_pack(as<fitpack_grid_polar_c>(), n, buffer.data());
    }
 

    void core_comm_expand(std::vector<double>& buffer) override {
        int32_t n = static_cast<int32_t>(buffer.size());
        fitpack_grid_polar_c_core_comm_expand(as<fitpack_grid_polar_c>(), n, buffer.data());
    }
 
 
    void destroy_base() override {
        fitpack_grid_polar_c_destroy_base(as<fitpack_grid_polar_c>());
    }
 
    // ===========================================================================================
    // Component Array Accessors
    // ===========================================================================================

    std::vector<double> u_vector() const {
        double* raw = nullptr;
        int64_t extents[1] = {0};
        fitpack_grid_polar_c_getcomp_u(as<fitpack_grid_polar_c>(), &raw, extents);
        const int64_t total = extents[0];
        if (raw == nullptr || total <= 0) return {};
        const double* first = reinterpret_cast<const double*>(raw);
        return std::vector<double>(first, first + total);
    }
 
#if HAVE_FXARRAY
    fxArray<double> u() const {
        double* raw = nullptr;
        int64_t extents[1] = {0};
        fitpack_grid_polar_c_getcomp_u(as<fitpack_grid_polar_c>(), &raw, extents);
        FX_SIZE bounds[2];   // (lower, upper) per dimension, Fortran lbound 1
        for (int k = 0; k < 1; ++k) {
            bounds[2 * k] = 1;
            bounds[2 * k + 1] = static_cast<FX_SIZE>(extents[k]);
        }
        array_c descr = array_c_null;
        array_c_from_ptr(&descr, "u", static_cast<void*>(raw),
                         getCFITypeFlag<double>(),
                         static_cast<FX_SIZE>(sizeof(double)),
                         static_cast<FX_RANK>(1), bounds);
        return fxArray<double>(descr);
    }
#endif // HAVE_FXARRAY

    std::vector<double> v_vector() const {
        double* raw = nullptr;
        int64_t extents[1] = {0};
        fitpack_grid_polar_c_getcomp_v(as<fitpack_grid_polar_c>(), &raw, extents);
        const int64_t total = extents[0];
        if (raw == nullptr || total <= 0) return {};
        const double* first = reinterpret_cast<const double*>(raw);
        return std::vector<double>(first, first + total);
    }
 
#if HAVE_FXARRAY
    fxArray<double> v() const {
        double* raw = nullptr;
        int64_t extents[1] = {0};
        fitpack_grid_polar_c_getcomp_v(as<fitpack_grid_polar_c>(), &raw, extents);
        FX_SIZE bounds[2];   // (lower, upper) per dimension, Fortran lbound 1
        for (int k = 0; k < 1; ++k) {
            bounds[2 * k] = 1;
            bounds[2 * k + 1] = static_cast<FX_SIZE>(extents[k]);
        }
        array_c descr = array_c_null;
        array_c_from_ptr(&descr, "v", static_cast<void*>(raw),
                         getCFITypeFlag<double>(),
                         static_cast<FX_SIZE>(sizeof(double)),
                         static_cast<FX_RANK>(1), bounds);
        return fxArray<double>(descr);
    }
#endif // HAVE_FXARRAY

    std::vector<double> z_vector() const {
        double* raw = nullptr;
        int64_t extents[2] = {0, 0};
        fitpack_grid_polar_c_getcomp_z(as<fitpack_grid_polar_c>(), &raw, extents);
        const int64_t total = extents[0] * extents[1];
        if (raw == nullptr || total <= 0) return {};
        const double* first = reinterpret_cast<const double*>(raw);
        return std::vector<double>(first, first + total);
    }

    std::array<int64_t, 2> z_shape() const {
        double* raw = nullptr;
        int64_t extents[2] = {0, 0};
        fitpack_grid_polar_c_getcomp_z(as<fitpack_grid_polar_c>(), &raw, extents);
        return { extents[0], extents[1] };
    }
 
#if HAVE_FXARRAY
    fxArray<double> z() const {
        double* raw = nullptr;
        int64_t extents[2] = {0, 0};
        fitpack_grid_polar_c_getcomp_z(as<fitpack_grid_polar_c>(), &raw, extents);
        FX_SIZE bounds[4];   // (lower, upper) per dimension, Fortran lbound 1
        for (int k = 0; k < 2; ++k) {
            bounds[2 * k] = 1;
            bounds[2 * k + 1] = static_cast<FX_SIZE>(extents[k]);
        }
        array_c descr = array_c_null;
        array_c_from_ptr(&descr, "z", static_cast<void*>(raw),
                         getCFITypeFlag<double>(),
                         static_cast<FX_SIZE>(sizeof(double)),
                         static_cast<FX_RANK>(2), bounds);
        return fxArray<double>(descr);
    }
#endif // HAVE_FXARRAY

    std::vector<double> t_vector() const {
        double* raw = nullptr;
        int64_t extents[2] = {0, 0};
        fitpack_grid_polar_c_getcomp_t(as<fitpack_grid_polar_c>(), &raw, extents);
        const int64_t total = extents[0] * extents[1];
        if (raw == nullptr || total <= 0) return {};
        const double* first = reinterpret_cast<const double*>(raw);
        return std::vector<double>(first, first + total);
    }

    std::array<int64_t, 2> t_shape() const {
        double* raw = nullptr;
        int64_t extents[2] = {0, 0};
        fitpack_grid_polar_c_getcomp_t(as<fitpack_grid_polar_c>(), &raw, extents);
        return { extents[0], extents[1] };
    }
 
#if HAVE_FXARRAY
    fxArray<double> t() const {
        double* raw = nullptr;
        int64_t extents[2] = {0, 0};
        fitpack_grid_polar_c_getcomp_t(as<fitpack_grid_polar_c>(), &raw, extents);
        FX_SIZE bounds[4];   // (lower, upper) per dimension, Fortran lbound 1
        for (int k = 0; k < 2; ++k) {
            bounds[2 * k] = 1;
            bounds[2 * k + 1] = static_cast<FX_SIZE>(extents[k]);
        }
        array_c descr = array_c_null;
        array_c_from_ptr(&descr, "t", static_cast<void*>(raw),
                         getCFITypeFlag<double>(),
                         static_cast<FX_SIZE>(sizeof(double)),
                         static_cast<FX_RANK>(2), bounds);
        return fxArray<double>(descr);
    }
#endif // HAVE_FXARRAY
 
    // ===========================================================================================
    // Scalar Property Accessors
    // ===========================================================================================
 
    double& r() {
        return *fitpack_grid_polar_c_ref_r(as<fitpack_grid_polar_c>());
    }
    const double& r() const {
        return *fitpack_grid_polar_c_ref_r(as<fitpack_grid_polar_c>());
    }
 
    double& z0() {
        return *fitpack_grid_polar_c_ref_z0(as<fitpack_grid_polar_c>());
    }
    const double& z0() const {
        return *fitpack_grid_polar_c_ref_z0(as<fitpack_grid_polar_c>());
    }
 
    bool z0_present() const {
        return fitpack_grid_polar_c_get_z0_present(as<fitpack_grid_polar_c>());
    }
    void set_z0_present(bool value) {
        fitpack_grid_polar_c_set_z0_present(as<fitpack_grid_polar_c>(), value);
    }
 
    bool z0_exact() const {
        return fitpack_grid_polar_c_get_z0_exact(as<fitpack_grid_polar_c>());
    }
    void set_z0_exact(bool value) {
        fitpack_grid_polar_c_set_z0_exact(as<fitpack_grid_polar_c>(), value);
    }
 
    bool z0_zero_gradient() const {
        return fitpack_grid_polar_c_get_z0_zero_gradient(as<fitpack_grid_polar_c>());
    }
    void set_z0_zero_gradient(bool value) {
        fitpack_grid_polar_c_set_z0_zero_gradient(as<fitpack_grid_polar_c>(), value);
    }
 
    int32_t& nmax() {
        return *fitpack_grid_polar_c_ref_nmax(as<fitpack_grid_polar_c>());
    }
    const int32_t& nmax() const {
        return *fitpack_grid_polar_c_ref_nmax(as<fitpack_grid_polar_c>());
    }
 
    int32_t& bc_continuity_origin() {
        return *fitpack_grid_polar_c_ref_bc_continuity_origin(as<fitpack_grid_polar_c>());
    }
    const int32_t& bc_continuity_origin() const {
        return *fitpack_grid_polar_c_ref_bc_continuity_origin(as<fitpack_grid_polar_c>());
    }
 
    int32_t& bc_boundary() {
        return *fitpack_grid_polar_c_ref_bc_boundary(as<fitpack_grid_polar_c>());
    }
    const int32_t& bc_boundary() const {
        return *fitpack_grid_polar_c_ref_bc_boundary(as<fitpack_grid_polar_c>());
    }
 
protected:
    explicit fpGridPolar(NoAlloc tag) : fpFitter(tag) {}
 
};
 
#endif /* FPGRIDPOLAR_HPP_INCLUDED */