Mercurial > repos > public > sbplib_julia
diff src/SbpOperators/quadrature/inverse_quadrature.jl @ 356:0844069ab5ff refactor/remove_dynamic_size_tensormapping
Reinclude SbpOperators and fix most of the code and tests there.
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Sun, 27 Sep 2020 22:51:31 +0200 |
parents | 01b851161018 |
children | 0546cb279fc2 |
line wrap: on
line diff
--- a/src/SbpOperators/quadrature/inverse_quadrature.jl Sun Sep 27 21:08:18 2020 +0200 +++ b/src/SbpOperators/quadrature/inverse_quadrature.jl Sun Sep 27 22:51:31 2020 +0200 @@ -2,14 +2,27 @@ """ InverseQuadrature{Dim,T<:Real,M,K} <: TensorMapping{T,Dim,Dim} -Implements the inverse quadrature operator `Qi` of Dim dimension as a TensorOperator +Implements the inverse quadrature operator `Qi` of Dim dimension as a TensorMapping The multi-dimensional tensor operator consists of a tuple of 1D InverseDiagonalInnerProduct tensor operators. """ -struct InverseQuadrature{Dim,T<:Real,M} <: TensorOperator{T,Dim} +struct InverseQuadrature{Dim,T<:Real,M} <: TensorMapping{T,Dim,Dim} Hi::NTuple{Dim,InverseDiagonalInnerProduct{T,M}} end + +function InverseQuadrature(g::EquidistantGrid{Dim}, quadratureClosure) where Dim + Hi = () + for i ∈ 1:Dim + Hi = (Hi..., InverseDiagonalInnerProduct(subgrid(g,i), quadratureClosure)) + end + + return InverseQuadrature(Hi) +end + +LazyTensors.range_size(Hi::InverseQuadrature) = getindex.(range_size.(Hi.Hi),1) +LazyTensors.domain_size(Hi::InverseQuadrature) = getindex.(domain_size.(Hi.Hi),1) + LazyTensors.domain_size(Qi::InverseQuadrature{Dim}, range_size::NTuple{Dim,Integer}) where Dim = range_size function LazyTensors.apply(Qi::InverseQuadrature{Dim,T}, v::AbstractArray{T,Dim}, I::Vararg{Index,Dim}) where {T,Dim}