Mercurial > repos > public > sbplib_julia
changeset 881:aa4875f9a530 feature/variable_derivatives
Start implementing the variable second derivative
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Thu, 20 Jan 2022 15:18:14 +0100 |
parents | f74189c954d6 |
children | 9098fc936776 |
files | src/SbpOperators/SbpOperators.jl src/SbpOperators/volumeops/derivatives/second_derivative_variable.jl test/SbpOperators/volumeops/derivatives/second_derivative_variable_test.jl |
diffstat | 3 files changed, 85 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- a/src/SbpOperators/SbpOperators.jl Thu Jan 20 15:17:37 2022 +0100 +++ b/src/SbpOperators/SbpOperators.jl Thu Jan 20 15:18:14 2022 +0100 @@ -14,6 +14,7 @@ include("volumeops/volume_operator.jl") include("volumeops/constant_interior_scaling_operator.jl") include("volumeops/derivatives/second_derivative.jl") +include("volumeops/derivatives/second_derivative_variable.jl") include("volumeops/laplace/laplace.jl") include("volumeops/inner_products/inner_product.jl") include("volumeops/inner_products/inverse_inner_product.jl")
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/SbpOperators/volumeops/derivatives/second_derivative_variable.jl Thu Jan 20 15:18:14 2022 +0100 @@ -0,0 +1,62 @@ +export SecondDerivativeVariable + +# """ +# SecondDerivativeVariable(grid, inner_stencil, closure_stencils, parity, direction) + +# Creates a volume operator on a `Dim`-dimensional grid acting along the +# specified coordinate `direction`. The action of the operator is determined by +# the stencils `inner_stencil` and `closure_stencils`. When `Dim=1`, the +# corresponding `SecondDerivativeVariable` tensor mapping is returned. When `Dim>1`, the +# returned operator is the appropriate outer product of a one-dimensional +# operators and `IdentityMapping`s, e.g for `Dim=3` the volume operator in the +# y-direction is `I⊗op⊗I`. +# """ +# function volume_operator(grid::EquidistantGrid, inner_stencil, closure_stencils, parity, direction) +# #TODO: Check that direction <= Dim? + +# # Create 1D volume operator in along coordinate direction +# op = SecondDerivativeVariable(restrict(grid, direction), inner_stencil, closure_stencils, parity) +# # Create 1D IdentityMappings for each coordinate direction +# one_d_grids = restrict.(Ref(grid), Tuple(1:dimension(grid))) +# Is = IdentityMapping{eltype(grid)}.(size.(one_d_grids)) +# # Formulate the correct outer product sequence of the identity mappings and +# # the volume operator +# parts = Base.setindex(Is, op, direction) +# return foldl(⊗, parts) +# end + +""" + SecondDerivativeVariable{T,N,M,K} <: TensorOperator{T,1} +Implements a one-dimensional constant coefficients volume operator +""" +struct SecondDerivativeVariable{T,N,M,K} <: TensorMapping{T,1,1} + inner_stencil::NestedStencil{T,N} + closure_stencils::NTuple{M,NestedStencil{T,K}} + size::NTuple{1,Int} +end + +function SecondDerivativeVariable(grid::EquidistantGrid{1}, inner_stencil, closure_stencils) + return SecondDerivativeVariable(inner_stencil, Tuple(closure_stencils), size(grid)) +end + +closure_size(::SecondDerivativeVariable{T,N,M}) where {T,N,M} = M + +LazyTensors.range_size(op::SecondDerivativeVariable) = op.size +LazyTensors.domain_size(op::SecondDerivativeVariable) = op.size + +function LazyTensors.apply(op::SecondDerivativeVariable{T}, v::AbstractVector{T}, i::Index{Lower}) where T + return @inbounds apply_stencil(op.closure_stencils[Int(i)], v, Int(i)) +end + +function LazyTensors.apply(op::SecondDerivativeVariable{T}, v::AbstractVector{T}, i::Index{Interior}) where T + return apply_stencil(op.inner_stencil, v, Int(i)) +end + +function LazyTensors.apply(op::SecondDerivativeVariable{T}, v::AbstractVector{T}, i::Index{Upper}) where T + return @inbounds Int(op.parity)*apply_stencil_backwards(op.closure_stencils[op.size[1]-Int(i)+1], v, Int(i)) +end + +function LazyTensors.apply(op::SecondDerivativeVariable{T}, v::AbstractVector{T}, i) where T + r = getregion(i, closure_size(op), op.size[1]) + return LazyTensors.apply(op, v, Index(i, r)) +end
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test/SbpOperators/volumeops/derivatives/second_derivative_variable_test.jl Thu Jan 20 15:18:14 2022 +0100 @@ -0,0 +1,22 @@ +using Test + +using Sbplib.Grids +using Sbplib.LazyTensors +using Sbplib.SbpOperators +using Sbplib.SbpOperators: NestedStencil, CenteredNestedStencil + + + +@testset "SecondDerivativeVariable" begin + g = EquidistantGrid(11, 0., 1.) + + interior_stencil = CenteredNestedStencil((-1/2, -1/2, 0.),( 1/2, 1., 1/2),( 0., -1/2, -1/2)) + closure_stencils = [ + NestedStencil(( 1/2, 1/2, 0.),(-1/2, -1/2, 0.), center = 1), + NestedStencil((-1/2, -1/2, 0.),( 1/2, 1., 1/2), center = 2), + ] + + @test SecondDerivativeVariable(interior_stencil,Tuple(closure_stencils), (4,)) isa TensorMapping + @test SecondDerivativeVariable(g, interior_stencil, closure_stencils) isa TensorMapping +end +