Add convenience wrapper for solve! method of spline solver.

JuliaPlasma · Jun 27, 2022 · c07e71a · c07e71a · michakraus · Jun 28, 2022
1 parent e6fec74
commit c07e71a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/poisson_splines.jl b/src/poisson_splines.jl
@@ -87,6 +87,15 @@ function solve!(p::PoissonSolverPBSplines{DT}, x::AbstractVector{DT}, w::Abstrac
     return p
 end
 
+function solve!(p::PoissonSolverPBSplines{DT}, x::AbstractMatrix{DT}, w::AbstractMatrix{DT}) where {DT}
+    @assert size(x, 1) == 1
+    @assert size(w, 1) == 1
+
+    X = reshape(x, size(x,2))
+    W = reshape(w, size(w,2))
+
+    solve!(p, X, W)
+end
 
 function eval_density(p::PoissonSolverPBSplines{DT}, x::DT) where {DT}
     eval_PBSBasis(p.ρ, p.bspl, x)