Support plots (#36)

* Support plots

* Update ultrasphericalspectralmethod.jl

* Update Project.toml

* Update test_chebyshev.jl

Author: dlfivefifty

Project.toml

@@ -25,11 +25,11 @@ QuasiArrays = "c4ea9172-b204-11e9-377d-29865faadc5c"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 
 [compat]
-ArrayLayouts = "0.6.2"
+ArrayLayouts = "0.6.2, 0.7"
 BandedMatrices = "0.16.5"
 BlockArrays = "0.15"
 BlockBandedMatrices = "0.10"
-ContinuumArrays = "0.7.2"
+ContinuumArrays = "0.7.3"
 DomainSets = "0.4, 0.5"
 FFTW = "1.1"
 FastGaussQuadrature = "0.4.3"

examples/ultrasphericalspectralmethod.jl

@@ -1,14 +1,11 @@
-using ContinuumArrays, FillArrays, InfiniteArrays, Plots
+using ClassicalOrthogonalPolynomials, Plots
 
 T = Chebyshev()
 C = Ultraspherical(2)
 D = Derivative(axes(T,1))
 A = (C\(D^2*T))+100(C\T)
 
-n = 100
-c = [T[[-1,1],1:n]; A[1:n-2,1:n]] \ [1;2;zeros(n-2)]
-u = T*Vcat(c,Zeros(∞))
-
-xx = range(-1,1;length=1000)
-plot(xx,u[xx])
+c = Vcat(T[[-1,1],:], A) \ [1;2;zeros(∞)]
+u = T*c
 
+plot(u)

src/ClassicalOrthogonalPolynomials.jl

@@ -28,7 +28,7 @@ import QuasiArrays: cardinality, checkindex, QuasiAdjoint, QuasiTranspose, Inclu
 
 import InfiniteArrays: OneToInf, InfAxes, Infinity, AbstractInfUnitRange, InfiniteCardinal, InfRanges
 import ContinuumArrays: Basis, Weight, basis, @simplify, Identity, AbstractAffineQuasiVector, ProjectionFactorization,
-    inbounds_getindex, grid, transform, transform_ldiv, TransformFactorization, QInfAxes, broadcastbasis, Expansion,
+    inbounds_getindex, grid, plotgrid, transform, transform_ldiv, TransformFactorization, QInfAxes, broadcastbasis, Expansion,
     AffineQuasiVector, AffineMap, WeightLayout, WeightedBasisLayout, WeightedBasisLayouts, demap, AbstractBasisLayout, BasisLayout,
     checkpoints, weight, unweightedbasis, MappedBasisLayouts, __sum
 import FastTransforms: Λ, forwardrecurrence, forwardrecurrence!, _forwardrecurrence!, clenshaw, clenshaw!,

src/classical/jacobi.jl

@@ -207,6 +207,12 @@ function grid(Pn::SubQuasiArray{T,2,<:AbstractJacobi,<:Tuple{Inclusion,AbstractU
     ChebyshevGrid{1,T}(maximum(jr))
 end
 
+function plotgrid(Pn::SubQuasiArray{T,2,<:AbstractJacobi,<:Tuple{Inclusion,AbstractUnitRange}}) where T
+    kr,jr = parentindices(Pn)
+    ChebyshevGrid{2,T}(40maximum(jr))
+end
+
+
 function ldiv(::Legendre{V}, f::AbstractQuasiVector) where V
     T = ChebyshevT{V}()
     [cheb2leg(paddeddata(T \ f)); zeros(V,∞)]

test/test_chebyshev.jl

@@ -356,6 +356,10 @@ import ContinuumArrays: MappedWeightedBasisLayout, Map
         V = view(T,:,[1,3,4])
         @test (U\(D*V))[1:5,:] == (U \ (V'D')')[1:5,:] == (U\(D*T))[1:5,[1,3,4]] 
     end
+
+    @testset "plot" begin
+        @test ContinuumArrays.plotgrid(ChebyshevT()[:,1:5]) == ChebyshevGrid{2}(200)
+    end
 end
 
 struct QuadraticMap{T} <: Map{T} end