Mercurial > repos > public > sbplib_julia

comparison src/SbpOperators/laplace/secondderivative.jl @ 550:dfcf0e506ba9 refactor/tensor_index_coupling

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge
author Jonatan Werpers <jonatan@werpers.com>
date Sun, 29 Nov 2020 12:57:15 +0100
parents 9330338d6ab5
children e71f2f81b5f8
comparison
equal deleted inserted replaced
549:ace46ca8513f 550:dfcf0e506ba9
19 19
20 LazyTensors.range_size(D2::SecondDerivative) = D2.size 20 LazyTensors.range_size(D2::SecondDerivative) = D2.size
21 LazyTensors.domain_size(D2::SecondDerivative) = D2.size 21 LazyTensors.domain_size(D2::SecondDerivative) = D2.size
22 22
23 # Apply for different regions Lower/Interior/Upper or Unknown region 23 # Apply for different regions Lower/Interior/Upper or Unknown region
24 function LazyTensors.apply(D2::SecondDerivative{T}, v::AbstractVector{T}, I::Index{Lower}) where T 24 function LazyTensors.apply(D2::SecondDerivative{T}, v::AbstractVector{T}, i::Index{Lower}) where T
25 return @inbounds D2.h_inv*D2.h_inv*apply_stencil(D2.closureStencils[Int(I)], v, Int(I)) 25 return @inbounds D2.h_inv*D2.h_inv*apply_stencil(D2.closureStencils[Int(i)], v, Int(i))
26 end 26 end
27 27
28 function LazyTensors.apply(D2::SecondDerivative{T}, v::AbstractVector{T}, I::Index{Interior}) where T 28 function LazyTensors.apply(D2::SecondDerivative{T}, v::AbstractVector{T}, i::Index{Interior}) where T
29 return @inbounds D2.h_inv*D2.h_inv*apply_stencil(D2.innerStencil, v, Int(I)) 29 return @inbounds D2.h_inv*D2.h_inv*apply_stencil(D2.innerStencil, v, Int(i))
30 end 30 end
31 31
32 function LazyTensors.apply(D2::SecondDerivative{T}, v::AbstractVector{T}, I::Index{Upper}) where T 32 function LazyTensors.apply(D2::SecondDerivative{T}, v::AbstractVector{T}, i::Index{Upper}) where T
33 N = length(v) # TODO: Use domain_size here instead? N = domain_size(D2,size(v)) 33 N = length(v) # TODO: Use domain_size here instead? N = domain_size(D2,size(v))
34 return @inbounds D2.h_inv*D2.h_inv*apply_stencil_backwards(D2.closureStencils[N-Int(I)+1], v, Int(I)) 34 return @inbounds D2.h_inv*D2.h_inv*apply_stencil_backwards(D2.closureStencils[N-Int(i)+1], v, Int(i))
35 end 35 end
36 36
37 function LazyTensors.apply(D2::SecondDerivative{T}, v::AbstractVector{T}, i) where T 37 function LazyTensors.apply(D2::SecondDerivative{T}, v::AbstractVector{T}, i) where T
38 N = length(v) # TODO: Use domain_size here instead? 38 N = length(v) # TODO: Use domain_size here instead?
39 r = getregion(i, closuresize(D2), N) 39 r = getregion(i, closuresize(D2), N)
40 I = Index(i, r) 40 return LazyTensors.apply(D2, v, Index(i, r))
41 return LazyTensors.apply(D2, v, I)
42 end 41 end
43 42
44 closuresize(D2::SecondDerivative{T,N,M,K}) where {T<:Real,N,M,K} = M 43 closuresize(D2::SecondDerivative{T,N,M,K}) where {T<:Real,N,M,K} = M