From 0b7ca0db0b33419f4a07c7f000ae5eb1ddfc11e7 Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauldmccarthy@gmail.com>
Date: Mon, 2 Sep 2024 13:56:37 +0100
Subject: [PATCH] ENH: New function to copy spline coefficients
---
splinterpolator.h | 19 +++++++++++++++++--
1 file changed, 17 insertions(+), 2 deletions(-)
diff --git a/splinterpolator.h b/splinterpolator.h
index 538e489..5e16c8e 100644
--- a/splinterpolator.h
+++ b/splinterpolator.h
@@ -64,8 +64,7 @@ public:
ExtrapolationType et=Zeros,
unsigned int order=3,
bool copy_low_order=true,
- Utilities::NoOfThreads
- nthr=Utilities::NoOfThreads(1),
+ Utilities::NoOfThreads nthr=Utilities::NoOfThreads(1),
double prec=1e-8,
bool data_are_coefs=false)
: _valid(false), _own_coef(false), _coef(0), _cptr(0), _ndim(0), _nthr(nthr._n)
@@ -84,6 +83,22 @@ public:
Splinterpolator& operator=(const Splinterpolator& src)
{ if(_own_coef) delete [] _coef; assign(src); return(*this); }
+ // Copy the spline coefficients into dest.
+ void Copy(std::vector<T>& dest)
+ {
+ unsigned int N = 1;
+ for (auto d : _dim) {
+ N *= d;
+ }
+ dest.resize(N);
+
+ auto coefs = coef_ptr();
+
+ for (unsigned int i = 0; i < N; i++) {
+ dest[i] = coefs[i];
+ }
+ }
+
// Set new data in Splinterpolator.
void Set(const T *data_or_coefs,
const std::vector<unsigned int>& dim,
--
GitLab