ContinuumArrays v0.17 (#172)

Author: dlfivefifty

Project.toml

@@ -25,16 +25,16 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [compat]
 ArrayLayouts = "1.0.9"
-BandedMatrices = "0.17.30, 1"
+BandedMatrices = "1"
 BlockArrays = "0.16.14"
 BlockBandedMatrices = "0.12.5"
-ClassicalOrthogonalPolynomials = "0.11.10"
-ContinuumArrays = "0.16.3"
+ClassicalOrthogonalPolynomials = "0.12"
+ContinuumArrays = "0.17"
 DomainSets = "0.6, 0.7"
 FastTransforms = "0.15.11"
 FillArrays = "1.0"
 HarmonicOrthogonalPolynomials = "0.5"
-InfiniteArrays = "0.12, 0.13"
+InfiniteArrays = "0.13"
 InfiniteLinearAlgebra = "0.7"
 LazyArrays = "1.0"
 LazyBandedMatrices = "0.9"

src/MultivariateOrthogonalPolynomials.jl

@@ -11,7 +11,7 @@ import Base.Broadcast: Broadcasted, broadcasted, DefaultArrayStyle
 import DomainSets: boundary, EuclideanDomain
 
 import QuasiArrays: LazyQuasiMatrix, LazyQuasiArrayStyle, domain
-import ContinuumArrays: @simplify, Weight, weight, grid, plotgrid, TransformFactorization, ExpansionLayout, plotvalues, unweighted, plan_grid_transform, checkpoints, transform_ldiv, AbstractBasisLayout, basis_axes, Inclusion, grammatrix, weaklaplacian
+import ContinuumArrays: @simplify, Weight, weight, grid, plotgrid, TransformFactorization, ExpansionLayout, plotvalues, unweighted, plan_transform, checkpoints, transform_ldiv, AbstractBasisLayout, basis_axes, Inclusion, grammatrix, weaklaplacian
 
 import ArrayLayouts: MemoryLayout, sublayout, sub_materialize
 import BlockArrays: block, blockindex, BlockSlice, viewblock, blockcolsupport, AbstractBlockStyle, BlockStyle

src/disk.jl

@@ -256,7 +256,7 @@ end
 *(P::ZernikeTransform{T}, f::Matrix{T}) where T = ModalTrav(P.disk2cxf \ (P.analysis * f))
 *(P::ZernikeITransform, f::AbstractVector) = P.synthesis * (P.disk2cxf * ModalTrav(f).matrix)
 
-plan_grid_transform(Z::Zernike{T}, B::Tuple{Block{1}}, dims=1:1) where T = grid(Z,B[1]), ZernikeTransform{T}(Int(B[1]), Z.a, Z.b)
+plan_transform(Z::Zernike{T}, (N,)::Tuple{Block{1}}, dims=1) where T = ZernikeTransform{T}(Int(N), Z.a, Z.b)
 
 ##
 # Laplacian

src/rect.jl

@@ -98,14 +98,17 @@ function checkpoints(P::RectPolynomial)
     SVector.(x, y')
 end
 
-function plan_grid_transform(P::KronPolynomial{d,<:Any,<:Fill}, B::Tuple{Block{1}}, dims=1:1) where d
+function plan_transform(P::KronPolynomial{d,<:Any,<:Fill}, B::Tuple{Block{1}}, dims=1:1) where d
     @assert dims == 1
 
     T = first(P.args)
-    x, F = plan_grid_transform(T, tuple(Fill(Int(B[1]),d)...))
     @assert d == 2
-    x̃ = Vector(x)
-    SVector.(x̃, x̃'), ApplyPlan(DiagTrav, F)
+    ApplyPlan(DiagTrav, plan_transform(T, tuple(Fill(Int(B[1]),d)...)))
+end
+
+function grid(P::RectPolynomial, B::Block{1})
+    x,y = grid.(P.args, Int(B))
+    SVector.(x, y')
 end
 
 function plotgrid(P::RectPolynomial, B::Block{1})
@@ -114,14 +117,12 @@ function plotgrid(P::RectPolynomial, B::Block{1})
 end
 
 
-function plan_grid_transform(P::KronPolynomial{d}, B::Tuple{Block{1}}, dims=1:1) where d
+function plan_transform(P::KronPolynomial{d}, B::Tuple{Block{1}}, dims=1:1) where d
     @assert dims == 1 || dims == 1:1 || dims == (1,)
     @assert d == 2
     N = Int(B[1])
-    xF = [plan_grid_transform(P.args[k], (N,N), k) for k=1:length(P.args)]
-    x,y = map(first,xF)
-    Fx,Fy = map(last,xF)
-    SVector.(x, y'), ApplyPlan(DiagTrav, TensorPlan(Fx,Fy))
+    Fx,Fy = plan_transform(P.args[1], (N,N), 1),plan_transform(P.args[2], (N,N), 2)
+    ApplyPlan(DiagTrav, TensorPlan(Fx,Fy))
 end
 
 applylayout(::Type{typeof(*)}, ::Lay, ::DiagTravLayout) where Lay <: AbstractBasisLayout = ExpansionLayout{Lay}()

src/triangle.jl

@@ -784,10 +784,9 @@ TriPlan{T}(N::Block{1}, a, b, c) where T = TriPlan{T}(Matrix{T}(undef, Int(N), I
 
 *(T::TriPlan, F::AbstractMatrix) = DiagTrav(tridenormalize!(T.tri2cheb\(T.grid2cheb*F),T.a,T.b,T.c))
 
-function plan_grid_transform(P::JacobiTriangle, Bs::Tuple{Block{1}}, dims=1:1)
+function plan_transform(P::JacobiTriangle, (N,)::Tuple{Block{1}}, dims=1)
     T = eltype(P)
-    N = Bs[1]
-    grid(P, N), TriPlan{T}(N, P.a, P.b, P.c)
+    TriPlan{T}(N, P.a, P.b, P.c)
 end
 
 struct TriIPlan{T}