Mercurial > repos > public > sbplib_julia
view src/SbpOperators/quadrature/diagonal_quadrature.jl @ 546:09ae5b519b4c feature/quadrature_as_outer_product
Update documentation
author | Vidar Stiernström <vidar.stiernstrom@it.uu.se> |
---|---|
date | Fri, 27 Nov 2020 12:33:10 +0100 |
parents | 576c6d1acc28 |
children | 3c18a15934a7 |
line wrap: on
line source
""" diagonal_quadrature(g,quadrature_closure) Constructs the diagonal quadrature operator `H` on a grid of `Dim` dimensions as a `TensorMapping`. The one-dimensional operator is a `DiagonalQuadrature`, while the multi-dimensional operator is the outer-product of the one-dimensional operators in each coordinate direction. """ function diagonal_quadrature(g::EquidistantGrid{Dim}, quadrature_closure) where Dim H = DiagonalQuadrature(restrict(g,1), quadrature_closure) for i ∈ 2:Dim H = H⊗DiagonalQuadrature(restrict(g,i), quadrature_closure) end return H end export diagonal_quadrature """ DiagonalQuadrature{T,M} <: TensorMapping{T,1,1} Implements the one-dimensional diagonal quadrature operator as a `TensorMapping` The quadrature is defined by the quadrature interval length `h`, the quadrature closure weights `closure` and the number of quadrature intervals `size`. The interior stencil has the weight 1. """ struct DiagonalQuadrature{T,M} <: TensorMapping{T,1,1} h::T closure::NTuple{M,T} size::Tuple{Int} end export DiagonalQuadrature """ DiagonalQuadrature(g, quadrature_closure) Constructs the `DiagonalQuadrature` `H` on the `EquidistantGrid` `g` with `H.closure` specified by `quadrature_closure`. """ function DiagonalQuadrature(g::EquidistantGrid{1}, quadrature_closure) return DiagonalQuadrature(spacing(g)[1], quadrature_closure, size(g)) end LazyTensors.range_size(H::DiagonalQuadrature) = H.size LazyTensors.domain_size(H::DiagonalQuadrature) = H.size """ apply(H::DiagonalQuadrature{T}, v::AbstractVector{T}, I::Index) where T Implements the application `(H*v)[i]` an `Index{R}` where `R` is one of the regions `Lower`,`Interior`,`Upper`,`Unknown`. """ function LazyTensors.apply(H::DiagonalQuadrature{T}, v::AbstractVector{T}, I::Index{Lower}) where T return @inbounds H.h*H.closure[Int(I)]*v[Int(I)] end function LazyTensors.apply(H::DiagonalQuadrature{T},v::AbstractVector{T}, I::Index{Upper}) where T N = length(v); return @inbounds H.h*H.closure[N-Int(I)+1]*v[Int(I)] end function LazyTensors.apply(H::DiagonalQuadrature{T}, v::AbstractVector{T}, I::Index{Interior}) where T return @inbounds H.h*v[Int(I)] end function LazyTensors.apply(H::DiagonalQuadrature{T}, v::AbstractVector{T}, I::Index{Unknown}) where T N = length(v); r = getregion(Int(I), closure_size(H), N) i = Index(Int(I), r) return LazyTensors.apply(H, v, i) end """ apply(H::DiagonalQuadrature{T}, v::AbstractVector{T}, I::Index) where T Implements the application (H'*v)[I]. The operator is self-adjoint. """ LazyTensors.apply_transpose(H::DiagonalQuadrature, v::AbstractVector, I) = LazyTensors.apply(H,v,I) """ closure_size(H) Returns the size of the closure stencil of a DiagonalQuadrature `H`. """ closure_size(H::DiagonalQuadrature{T,M}) where {T,M} = M export closure_size