Mercurial > repos > public > sbplib_julia
comparison src/SbpOperators/quadrature/quadrature.jl @ 345:2fcc960836c6
Merge branch refactor/combine_to_one_package.
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Sat, 26 Sep 2020 15:22:13 +0200 |
| parents | 01b851161018 |
| children | 0844069ab5ff |
comparison
equal
deleted
inserted
replaced
| 343:e12c9a866513 | 345:2fcc960836c6 |
|---|---|
| 1 export Quadrature | |
| 2 """ | |
| 3 Quadrature{Dim,T<:Real,N,M,K} <: TensorMapping{T,Dim,Dim} | |
| 4 | |
| 5 Implements the quadrature operator `Q` of Dim dimension as a TensorMapping | |
| 6 The multi-dimensional tensor operator consists of a tuple of 1D DiagonalInnerProduct H | |
| 7 tensor operators. | |
| 8 """ | |
| 9 struct Quadrature{Dim,T<:Real,M} <: TensorOperator{T,Dim} | |
| 10 H::NTuple{Dim,DiagonalInnerProduct{T,M}} | |
| 11 end | |
| 12 | |
| 13 LazyTensors.domain_size(Q::Quadrature{Dim}, range_size::NTuple{Dim,Integer}) where {Dim} = range_size | |
| 14 | |
| 15 function LazyTensors.apply(Q::Quadrature{Dim,T}, v::AbstractArray{T,Dim}, I::Vararg{Index,Dim}) where {T,Dim} | |
| 16 error("not implemented") | |
| 17 end | |
| 18 | |
| 19 function LazyTensors.apply(Q::Quadrature{1,T}, v::AbstractVector{T}, I::Index) where T | |
| 20 @inbounds q = apply(Q.H[1], v , I) | |
| 21 return q | |
| 22 end | |
| 23 | |
| 24 function LazyTensors.apply(Q::Quadrature{2,T}, v::AbstractArray{T,2}, I::Index, J::Index) where T | |
| 25 # Quadrature in x direction | |
| 26 @inbounds vx = view(v, :, Int(J)) | |
| 27 @inbounds qx = apply(Q.H[1], vx , I) | |
| 28 # Quadrature in y-direction | |
| 29 @inbounds vy = view(v, Int(I), :) | |
| 30 @inbounds qy = apply(Q.H[2], vy, J) | |
| 31 return qx*qy | |
| 32 end | |
| 33 | |
| 34 LazyTensors.apply_transpose(Q::Quadrature{Dim,T}, v::AbstractArray{T,Dim}, I::Vararg{Index,Dim}) where {Dim,T} = LazyTensors.apply(Q,v,I...) |