improve plotgrid

Author: dlfivefifty

src/ContinuumArrays.jl

@@ -21,7 +21,7 @@ import QuasiArrays: cardinality, checkindex, QuasiAdjoint, QuasiTranspose, Inclu
                     AbstractQuasiFill, UnionDomain
 import InfiniteArrays: Infinity, InfAxes
 
-export Spline, LinearSpline, HeavisideSpline, DiracDelta, Derivative, ℵ₁, Inclusion, Basis, WeightedBasis, grid, transform, affine, ..
+export Spline, LinearSpline, HeavisideSpline, DiracDelta, Derivative, ℵ₁, Inclusion, Basis, WeightedBasis, grid, plotgrid, transform, affine, ..
 
 
 

src/bases/bases.jl

@@ -123,13 +123,6 @@ copy(L::Ldiv{<:AbstractBasisLayout,BroadcastLayout{typeof(*)},<:Any,<:AbstractQu
 # expansion
 _grid(_, P) = error("Overload Grid")
 
-_mul_grid(_, args) = grid(first(args))
-_mul_grid(::Tuple{Any,PaddedLayout}, (P,c)) = grid(P[:,colsupport(c)])
-
-function _grid(lay::ApplyLayout{typeof(*)}, P)
-    args = arguments(lay,P)
-    _mul_grid(map(MemoryLayout,args), args)
-end
 _grid(::MappedBasisLayout, P) = invmap(parentindices(P)[1])[grid(demap(P))]
 _grid(::SubBasisLayout, P) = grid(parent(P))
 _grid(::WeightedBasisLayouts, P) = grid(unweightedbasis(P))

src/plotting.jl

@@ -1,4 +1,16 @@
-plotgrid(g) = grid(g)
+
+
+_mul_plotgrid(_, args) = grid(first(args))
+_mul_plotgrid(::Tuple{Any,PaddedLayout}, (P,c)) = plotgrid(P[:,colsupport(c)])
+
+function _plotgrid(lay::ApplyLayout{typeof(*)}, P)
+    args = arguments(lay,P)
+    _mul_plotgrid(map(MemoryLayout,args), args)
+end
+
+_plotgrid(_, P) = grid(P)
+
+plotgrid(g) = _plotgrid(MemoryLayout(g), g)
 
 @recipe function f(g::AbstractQuasiVector)
     x = plotgrid(g)