Add Plotting (#91)

* Add Plotting

* Add Poisson example with plotting

* add tests

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "ContinuumArrays"
 uuid = "7ae1f121-cc2c-504b-ac30-9b923412ae5c"
-version = "0.7.2"
+version = "0.7.3"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"
@@ -13,6 +13,7 @@ IntervalSets = "8197267c-284f-5f27-9208-e0e47529a953"
 LazyArrays = "5078a376-72f3-5289-bfd5-ec5146d43c02"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 QuasiArrays = "c4ea9172-b204-11e9-377d-29865faadc5c"
+RecipesBase = "3cdcf5f2-1ef4-517c-9805-6587b60abb01"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 
 [compat]
@@ -25,6 +26,7 @@ Infinities = "0.1"
 IntervalSets = "0.5"
 LazyArrays = "0.21"
 QuasiArrays = "0.5.1"
+RecipesBase = "1.0"
 StaticArrays = "0.12, 1"
 julia = "1.5"
 

examples/poisson.jl

@@ -0,0 +1,18 @@
+using ContinuumArrays, Plots
+
+###
+# Dirichlet
+####
+
+L = LinearSpline(range(0,1; length=10_000))[:,2:end-1]
+x = axes(L,1)
+D = Derivative(x)
+Δ = -((D*L)'D*L)
+M = L'L
+f = L \ exp.(x)
+u = L * (Δ \ (M*f))
+plot(u)
+
+# we can also use cholesky
+ũ = L * (cholesky(Symmetric(-Δ)) \ -(M*f))
+plot(ũ)

src/ContinuumArrays.jl

@@ -1,5 +1,5 @@
 module ContinuumArrays
-using IntervalSets, LinearAlgebra, LazyArrays, FillArrays, BandedMatrices, QuasiArrays, Infinities, InfiniteArrays, StaticArrays, BlockArrays
+using IntervalSets, LinearAlgebra, LazyArrays, FillArrays, BandedMatrices, QuasiArrays, Infinities, InfiniteArrays, StaticArrays, BlockArrays, RecipesBase
 import Base: @_inline_meta, @_propagate_inbounds_meta, axes, getindex, convert, prod, *, /, \, +, -, ==, ^,
                 IndexStyle, IndexLinear, ==, OneTo, _maybetail, tail, similar, copyto!, copy, diff,
                 first, last, show, isempty, findfirst, findlast, findall, Slice, union, minimum, maximum, sum, _sum,
@@ -88,4 +88,6 @@ include("operators.jl")
 include("bases/bases.jl")
 include("basisconcat.jl")
 
+include("plotting.jl")
+
 end

src/bases/bases.jl

@@ -122,6 +122,7 @@ copy(L::Ldiv{<:AbstractBasisLayout,BroadcastLayout{typeof(*)},<:Any,<:AbstractQu
 
 # expansion
 _grid(_, P) = error("Overload Grid")
+_grid(lay::ApplyLayout{typeof(*)}, P) = grid(first(arguments(lay,P)))
 _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

@@ -0,0 +1,11 @@
+plotgrid(g) = grid(g)
+
+@recipe function f(g::AbstractQuasiVector)
+    x = plotgrid(g)
+    x,g[x]
+end
+
+@recipe function f(g::AbstractQuasiMatrix)
+    x = plotgrid(g)
+    x,g[x,:]
+end

test/runtests.jl

@@ -1,4 +1,4 @@
-using ContinuumArrays, QuasiArrays, LazyArrays, IntervalSets, FillArrays, LinearAlgebra, BandedMatrices, FastTransforms, InfiniteArrays, Test, Base64
+using ContinuumArrays, QuasiArrays, LazyArrays, IntervalSets, FillArrays, LinearAlgebra, BandedMatrices, FastTransforms, InfiniteArrays, Test, Base64, RecipesBase
 import ContinuumArrays: ℵ₁, materialize, AffineQuasiVector, BasisLayout, AdjointBasisLayout, SubBasisLayout, ℵ₁,
                         MappedBasisLayout, AdjointMappedBasisLayout, MappedWeightedBasisLayout, TransformFactorization, Weight, WeightedBasisLayout, SubWeightedBasisLayout, WeightLayout,
                         Expansion, basis, invmap, Map, checkpoints
@@ -596,3 +596,13 @@ ContinuumArrays.invmap(::InvQuadraticMap{T}) where T = QuadraticMap{T}()
 end
 
 include("test_basisconcat.jl")
+
+@testset "Plotting" begin
+    L = LinearSpline(0:5)
+    rep = RecipesBase.apply_recipe(Dict{Symbol, Any}(), L)
+    @test rep[1].args == (L.points,L[L.points,:])
+
+    u = L*randn(6)
+    rep = RecipesBase.apply_recipe(Dict{Symbol, Any}(), u)
+    @test rep[1].args == (L.points,u[L.points])
+end