Merge pull request #91 from jlazar17/patch-1

add in-place `evaluate!`
kbarbary · Sep 27, 2024 · 3103e28 · 3103e28
2 parents a5406e7 + 0c62fb1
commit 3103e28
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/Dierckx.jl b/src/Dierckx.jl
@@ -952,10 +952,10 @@ function evaluate(spline::Spline2D, x::AbstractVector, y::AbstractVector)
     return z
 end
 
-function evaluate(spline::Spline2D, x::Real, y::Real)
+function evaluate!(wrk::Vector{Float64}, Vspline::Spline2D, x::Real, y::Real)
     ier = Ref{Int32}()
     lwrk = spline.kx + spline.ky + 2
-    wrk = Vector{Float64}(undef, lwrk)
+    length(wrk) == lwrk || throw(ArgumentError("Length of work array not equal to required length of `spline.kx + spline.ky + 2 = $(spline.kx + spline.ky + 2)`"))
     z = Ref{Float64}()
     ccall((:bispeu_, libddierckx), Nothing,
           (Ref{Float64}, Ref{Int32},  # ty, ny
@@ -974,6 +974,12 @@ function evaluate(spline::Spline2D, x::Real, y::Real)
     return z[]
 end
 
+function evaluate(spline::Spline2D, x::Real, y::Real)
+    lwrk = spline.kx + spline.ky + 2
+    wrk = Vector{Float64}(undef, lwrk)
+    evaluate!(wrk, spline, x, y)
+end
+
 # Evaluate spline on the grid spanned by the input arrays.
 function evalgrid(spline::Spline2D, x::AbstractVector, y::AbstractVector)
     mx = length(x)