QuasiArrays v0.13

Author: dlfivefifty

Project.toml

@@ -22,15 +22,20 @@ QuasiArrays = "c4ea9172-b204-11e9-377d-29865faadc5c"
 RecurrenceRelationships = "807425ed-42ea-44d6-a357-6771516d7b2c"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
-Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+
+[weakdeps]
+StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
+
+[extensions]
+MultivariateOrthogonalPolynomialsStatsBaseExt = "StatsBase"
 
 [compat]
 ArrayLayouts = "1.11"
 BandedMatrices = "1"
 BlockArrays = "1.0"
 BlockBandedMatrices = "0.13"
 ClassicalOrthogonalPolynomials = "0.15.8"
-ContinuumArrays = "0.19.6"
+ContinuumArrays = "0.20"
 DomainSets = "0.7"
 FastTransforms = "0.17"
 FillArrays = "1.0"
@@ -39,19 +44,21 @@ InfiniteArrays = "0.15"
 InfiniteLinearAlgebra = "0.9, 0.10"
 LazyArrays = "2.3.1"
 LazyBandedMatrices = "0.11.3"
-QuasiArrays = "0.12.2"
+QuasiArrays = "0.13"
 Random = "1.11.0"
 RecurrenceRelationships = "0.2"
 SpecialFunctions = "1, 2"
 StaticArrays = "1"
+StatsBase = "0.34"
 julia = "1.10"
 
 [extras]
 Base64 = "2a0f44e3-6c83-55bd-87e4-b1978d98bd5f"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 RecipesBase = "3cdcf5f2-1ef4-517c-9805-6587b60abb01"
+StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Base64", "ForwardDiff", "Random", "RecipesBase", "Test"]
+test = ["Base64", "ForwardDiff", "Random", "RecipesBase", "StatsBase", "Test"]

ext/MultivariateOrthogonalPolynomialsStatsBaseExt.jl

@@ -0,0 +1,18 @@
+module MultivariateOrthogonalPolynomialsStatsBaseExt
+using MultivariateOrthogonalPolynomials, StatsBase
+using MultivariateOrthogonalPolynomials.StaticArrays
+using MultivariateOrthogonalPolynomials.QuasiArrays
+using MultivariateOrthogonalPolynomials: ExpansionLayout, KronOPLayout
+import MultivariateOrthogonalPolynomials.QuasiArrays: sample_layout
+
+function sample_layout(::ExpansionLayout{KronOPLayout{2}}, f::AbstractQuasiVector, n::Integer)
+    F = reshape(f)
+    x = sample(sum(F; dims=2), n)
+    # x = sample(F[:,y]) # TODO: this should work
+    y = [sample(F[x,:]) for x in x]
+    map(SVector, x, y)
+end
+
+sample_layout(lay::ExpansionLayout{KronOPLayout{2}}, f::AbstractQuasiVector) = only(sample_layout(lay, f, 1))
+
+end

src/MultivariateOrthogonalPolynomials.jl

@@ -21,7 +21,7 @@ import LazyArrays: arguments, paddeddata, LazyArrayStyle, LazyLayout, PaddedLayo
 import LazyBandedMatrices: LazyBandedBlockBandedLayout, AbstractBandedBlockBandedLayout, AbstractLazyBandedBlockBandedLayout, _krontrav_axes, DiagTravLayout, invdiagtrav, ApplyBandedBlockBandedLayout, krontrav
 import InfiniteArrays: InfiniteCardinal, OneToInf
 
-import ClassicalOrthogonalPolynomials: jacobimatrix, Weighted, orthogonalityweight, HalfWeighted, WeightedBasis, pad, recurrencecoefficients, clenshaw, weightedgrammatrix, Clenshaw, OPLayout, sample, sample_layout
+import ClassicalOrthogonalPolynomials: jacobimatrix, Weighted, orthogonalityweight, HalfWeighted, WeightedBasis, pad, recurrencecoefficients, clenshaw, weightedgrammatrix, Clenshaw, OPLayout
 import HarmonicOrthogonalPolynomials: BivariateOrthogonalPolynomial, MultivariateOrthogonalPolynomial, Plan,
                                           AngularMomentum, angularmomentum, BlockOneTo, BlockRange1, interlace,
                                           MultivariateOPLayout, AbstractMultivariateOPLayout, MAX_PLOT_BLOCKS

src/rect.jl

@@ -203,13 +203,3 @@ function vec_layout(::KronExpansionLayout{OPLayout, OPLayout}, f)
     A,C,Bc = arguments(f)
     RectPolynomial(A, Bc') * DiagTrav(C)
 end
-
-function sample_layout(::ExpansionLayout{MultivariateOrthogonalPolynomials.KronOPLayout{2}}, f::AbstractQuasiVector, n)
-    F = reshape(f)
-    x = sample(sum(F; dims=2), n)
-    # x = sample(F[:,y]) # TODO: this should work
-    y = [sample(F[x,:]) for x in x]
-    map(SVector, x, y)
-end
-
-sample_layout(lay::ExpansionLayout{MultivariateOrthogonalPolynomials.KronOPLayout{2}}, f::AbstractQuasiVector) = only(sample_layout(lay, f, 1))

test/test_rect.jl

@@ -1,5 +1,5 @@
-using MultivariateOrthogonalPolynomials, ClassicalOrthogonalPolynomials, StaticArrays, LinearAlgebra, BlockArrays, FillArrays, Base64, LazyBandedMatrices, ArrayLayouts, Random, Test
-using ClassicalOrthogonalPolynomials: expand, coefficients, recurrencecoefficients, sample
+using MultivariateOrthogonalPolynomials, ClassicalOrthogonalPolynomials, StaticArrays, LinearAlgebra, BlockArrays, FillArrays, Base64, LazyBandedMatrices, ArrayLayouts, Random, StatsBase, Test
+using ClassicalOrthogonalPolynomials: expand, coefficients, recurrencecoefficients
 using MultivariateOrthogonalPolynomials: weaklaplacian, ClenshawKron
 using ContinuumArrays: plotgridvalues, ExpansionLayout
 using Base: oneto
@@ -251,6 +251,7 @@ Random.seed!(3242)
         @test sum(F; dims=2)[0.1,1] ≈ 2.1748993079723618
 
         x,y = coordinates(P)
+        @test sample(f) isa SVector
         @test sum(sample(f, 100_000))/100_000 ≈ [sum(x .* f)/sum(f),sum(y .* f)/sum(f)] rtol=1E-1
     end
 end