support multidimensional grids

JuliaApproximation · May 5, 2021 · 3c34e3d · 3c34e3d · dlfivefifty · May 5, 2021
1 parent 504a79a
commit 3c34e3d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/plotting.jl b/src/plotting.jl
@@ -12,12 +12,15 @@ _plotgrid(_, P) = grid(P)
 
 plotgrid(g) = _plotgrid(MemoryLayout(g), g)
 
+_split_svec(x) = (x,)
+_split_svec(x::AbstractArray{<:StaticVector{2}}) = (map(first,x), map(last,x))
+
 @recipe function f(g::AbstractQuasiVector)
     x = plotgrid(g)
-    x,g[x]
+    tuple(_split_svec(x)...,g[x])
 end
 
 @recipe function f(g::AbstractQuasiMatrix)
     x = plotgrid(g)
-    x,g[x,:]
+    tuple(_split_svec(x)...,g[x,:])
 end