package extension for weights in DomainIntegrals.jl

Author: daanhb

Project.toml

@@ -1,6 +1,6 @@
 name = "GridArrays"
 uuid = "356ae075-8bc9-5ef9-a342-46c9ba26438c"
-version = "0.2.3"
+version = "0.2.4"
 
 [deps]
 CompositeTypes = "b152e2b5-7a66-4b01-a709-34e65c35f657"
@@ -12,9 +12,11 @@ StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [weakdeps]
+DomainIntegrals = "cc6bae93-f070-4015-88fd-838f9505a86c"
 RecipesBase = "3cdcf5f2-1ef4-517c-9805-6587b60abb01"
 
 [extensions]
+GridArraysDomainIntegralsExt = "DomainIntegrals"
 GridArraysRecipesBaseExt = "RecipesBase"
 
 [compat]

ext/GridArraysDomainIntegralsExt.jl

@@ -0,0 +1,52 @@
+module GridArraysDomainIntegralsExt
+
+using GridArrays, DomainIntegrals
+
+import DomainIntegrals:
+    measure,
+    laguerre_α,
+    jacobi_α,
+    jacobi_β
+
+measure(x::GridArrays.LaguerreNodes) = LaguerreWeight(x.α)
+measure(w::GridArrays.LaguerreWeights) = LaguerreWeight(w.α)
+
+laguerre_α(x::GridArrays.LaguerreNodes) = x.α
+laguerre_α(w::GridArrays.LaguerreWeights) = w.α
+
+measure(x::GridArrays.JacobiNodes) = JacobiWeight(x.α, x.β)
+measure(w::GridArrays.JacobiWeights) = JacobiWeight(w.α, w.β)
+
+jacobi_α(x::GridArrays.JacobiNodes) = x.α
+jacobi_α(w::GridArrays.JacobiWeights) = w.α
+jacobi_β(x::GridArrays.JacobiNodes) = x.β
+jacobi_β(w::GridArrays.JacobiWeights) = w.β
+
+measure(x::GridArrays.LegendreNodes{T}) where T = LegendreWeight{T}()
+measure(w::GridArrays.LegendreWeights{T}) where T = LegendreWeight{T}()
+
+jacobi_α(x::GridArrays.LegendreNodes{T}) where T = zero(T)
+jacobi_α(w::GridArrays.LegendreWeights{T}) where T = zero(T)
+jacobi_β(x::GridArrays.LegendreNodes{T}) where T = zero(T)
+jacobi_β(w::GridArrays.LegendreWeights{T}) where T = zero(T)
+
+measure(x::GridArrays.ChebyshevTNodes{T}) where T = ChebyshevTWeight{T}()
+measure(w::GridArrays.ChebyshevTWeights{T}) where T = ChebyshevTWeight{T}()
+
+jacobi_α(x::GridArrays.ChebyshevTNodes{T}) where T = -one(T)/2
+jacobi_α(w::GridArrays.ChebyshevTWeights{T}) where T = -one(T)/2
+jacobi_β(x::GridArrays.ChebyshevTNodes{T}) where T = -one(T)/2
+jacobi_β(w::GridArrays.ChebyshevTWeights{T}) where T = -one(T)/2
+
+measure(x::GridArrays.ChebyshevUNodes{T}) where T = ChebyshevUWeight{T}()
+measure(w::GridArrays.ChebyshevUWeights{T}) where T = ChebyshevUWeight{T}()
+
+jacobi_α(x::GridArrays.ChebyshevUNodes{T}) where T = -one(T)/2
+jacobi_α(w::GridArrays.ChebyshevUWeights{T}) where T = -one(T)/2
+jacobi_β(x::GridArrays.ChebyshevUNodes{T}) where T = -one(T)/2
+jacobi_β(w::GridArrays.ChebyshevUWeights{T}) where T = -one(T)/2
+
+measure(x::GridArrays.HermiteNodes{T}) where T = HermiteWeight{T}()
+measure(w::GridArrays.HermiteWeights{T}) where T = HermiteWeight{T}()
+
+end