view src/BoundaryConditions/sat.jl @ 1163:c9fdfb1efba8 feature/boundary_conditions

Merge with default
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Wed, 12 Oct 2022 10:45:47 +0200
parents 05b1d6fd6401
children fd80e9a0ef99
line wrap: on
line source

"""
    sat_tensors(op, grid, bc::BoundaryCondition{T}, params...)

Returns the `LazyTensor`s used to construct a SAT for the SBP operator `op` on 
`grid` associated with the boundary condition `bc`.
"""
function sat_tensors end

"""
    sat(op, grid, bc::BoundaryCondition{ConstantBoundaryData{T}}, params...)

Simultaneous-Approximation-Term for constant boundary data. f = sat(op, grid, bc) returns
an anonymous function, such that f(u) is a `LazyTensorApplication` weakly imposing the BC.
"""
function sat(op, grid, bc::BoundaryCondition{ConstantBoundaryData{T}}, params...) where T
    closure, penalty = sat_tensors(op, grid, bc, params...)
    b_sz = size(boundary_grid(grid, bc.id))
    g = fill(data(bc), b_sz)
    if iszero(g)
        return u -> closure(u)
    else
        return u -> closure(u) + penalty(g)
    end
end

"""
    sat(op, grid, bc::BoundaryCondition{SpaceDependentBoundaryData{T}}, params...)

Simultaneous-Approximation-Term for space-dependent boundary data. f = sat(op, grid, bc) returns
an anonymous function, such that f(u) is a `LazyTensorApplication` weakly imposing the BC.
"""
function sat(op, grid, bc::BoundaryCondition{SpaceDependentBoundaryData{T}}, params...) where T
    closure, penalty = sat_tensors(op, grid, bc, params...)
    g = data(bc)
    return u -> closure(u) + penalty(g)
end

"""
    sat(op, grid, bc::BoundaryCondition{SpaceDependentBoundaryData{T}}, params...)

Simultaneous-Approximation-Term for time-dependent boundary data. f = sat(op, grid, bc) returns
an anonymous function, such that f(u,t) is a `LazyTensorApplication` weakly imposing the BC at time t.
"""
function sat(op, grid, bc::BoundaryCondition{TimeDependentBoundaryData{T}}, params...) where T
    closure, penalty = sat_tensors(op, grid, bc, params...)
    b_sz = size(boundary_grid(grid, bc.id))
    b_vec = ones(eltype(grid), b_sz)
    g = data(bc)
    return (u,t) -> closure(u) + g(t)*penalty(b_vec)
end

"""
    sat(op, grid, bc::BoundaryCondition{SpaceDependentBoundaryData{T}}, params...)

Simultaneous-Approximation-Term for space-time-dependent boundary data. f = sat(op, grid, bc) returns
an anonymous function, such that f(u,t) is a `LazyTensorApplication` weakly imposing the BC at time t.
"""
function sat(op, grid, bc::BoundaryCondition{SpaceTimeDependentBoundaryData{T}}, params...) where T
    closure, penalty = sat_tensors(op, grid, bc, params...)
    g = data(bc)
    return (u,t) -> closure(u) + penalty(g(t))
end