Use unweighted transforms in weighted plotting (#187)

* Use unweighted transforms in weighted plotting

* Update Project.toml

* Update test_recipesbaseext.jl

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "ContinuumArrays"
 uuid = "7ae1f121-cc2c-504b-ac30-9b923412ae5c"
-version = "0.18.3"
+version = "0.18.4"
 
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"

src/plotting.jl

@@ -36,6 +36,12 @@ plotvalues_size(::Tuple{InfiniteCardinal{1},Int}, g, x=plotgrid(g)) = g[x,:]
 plotvalues_layout(lay, g, x...) = plotvalues_size(size(g), g, x...)
 # plotvalues_layout(::WeightedBasisLayouts, wP, n...) = plotvalues(unweighted(wP), n...)
 plotvalues_layout(::ExpansionLayout{MappedBasisLayout}, g, x...) = plotvalues(demap(g))
+function plotvalues_layout(::ExpansionLayout{<:WeightedBasisLayout}, g, x...)
+    f = unweighted(g)
+    w = weight(basis(g))
+    x = plotgrid(g)
+    w[x] .* plotvalues(f)
+end
 # plotvalues_layout(::SubBasisLayout, P::AbstractQuasiMatrix, n) = plotvalues(parent(P), maximum(parentindices(P)[2][n]))
 # plotvalues_layout(::SubBasisLayout, P::AbstractQuasiMatrix) = plotvalues(parent(P), maximum(parentindices(P)[2]))
 

test/test_recipesbaseext.jl

@@ -34,7 +34,8 @@ import RecipesBase
         rep = RecipesBase.apply_recipe(Dict{Symbol, Any}(), u)
         @test rep[1].args == (grid(T), u[grid(T)])
         wrep = RecipesBase.apply_recipe(Dict{Symbol, Any}(), v)
-        @test wrep[1].args == (grid(wT), v[grid(wT)])
+        @test wrep[1].args[1] == grid(wT)
+        @test wrep[1].args[2] ≈ v[grid(wT)]
 
         @test plotgrid(v) == plotgrid(u) == grid(T) == grid(wT) == plotgrid_layout(MemoryLayout(v), v) == plotgrid_layout(MemoryLayout(u), u)
         y = affine(0..1, x)