Mercurial > repos > public > sbplib_julia
changeset 56:27a8d3021a1c cell_based_test
Convert apply functions to cell-based
author | Ylva Rydin <ylva.rydin@telia.com> |
---|---|
date | Tue, 15 Jan 2019 12:26:06 +0100 |
parents | c62ea0112d4d |
children | 178a203f3e6d |
files | diffOp.jl plotDerivative.jl plotDerivative2d.jl sbpD2.jl |
diffstat | 4 files changed, 39 insertions(+), 48 deletions(-) [+] |
line wrap: on
line diff
--- a/diffOp.jl Tue Jan 15 10:25:56 2019 +0100 +++ b/diffOp.jl Tue Jan 15 12:26:06 2019 +0100 @@ -1,6 +1,6 @@ abstract type DiffOp end -function apply!(D::DiffOp, u::AbstractVector, v::AbstractVector) +function apply(D::DiffOp, v::AbstractVector, i::Int) error("not implemented") end @@ -32,6 +32,21 @@ error("not implemented") end +# DiffOp must have a grid!!! +function apply!(D::DiffOp, u::AbstractVector, v::AbstractVector) + for i ∈ 1:Grid.numberOfPoints(D.grid) + u[i] = apply(D, v, i) + end + + return nothing +end + +function apply(D::DiffOp, v::AbstractVector)::AbstractVector + u = zeros(eltype(v), size(v)) + apply!(D,v,u) + return u +end + # Differential operator for a*d^2/dx^2 struct Laplace1D <: DiffOp grid @@ -40,11 +55,10 @@ end # u = L*v -function apply!(L::Laplace1D, u::AbstractVector, v::AbstractVector) +function apply(L::Laplace1D, v::AbstractVector, i::Int) h = Grid.spacings(L.grid)[1] - apply!(L.op, u, v, h) - u .= L.a * u - return nothing + uᵢ = L.a * apply(L.op, h, v, i) + return uᵢ end @@ -56,34 +70,17 @@ end # u = L*v -function apply!(L::Laplace2D, u::AbstractVector, v::AbstractVector) - u .= 0*u +function apply(L::Laplace2D, v::AbstractVector, i::Int) h = Grid.spacings(L.grid) li = LinearIndices(L.grid.numberOfPointsPerDim) - n_x, n_y = L.grid.numberOfPointsPerDim - - - # For each x - temp = zeros(eltype(u), n_y) - for i ∈ 1:n_x - - v_i = view(v, li[i,:]) - apply!(L.op, temp, v_i, h[2]) - - u[li[i,:]] += temp - end + ci = CartesianIndices(L.grid.numberOfPointsPerDim) + I = ci[i] - # For each y - temp = zeros(eltype(u), n_x) - for i ∈ 1:n_y - v_i = view(v, li[:,i]) - apply!(L.op, temp, v_i, h[1]) + # 2nd x-derivative + uᵢ = apply(L.op, h[1], view(v, li[:,I[2]]), I[1]) + # 2nd y-derivative + uᵢ += apply(L.op, h[2], view(v, li[I[1],:]), I[2]) - u[li[:,i]] += temp - end - - u .= L.a*u - - return nothing + return uᵢ end
--- a/plotDerivative.jl Tue Jan 15 10:25:56 2019 +0100 +++ b/plotDerivative.jl Tue Jan 15 12:26:06 2019 +0100 @@ -2,7 +2,7 @@ op =sbp.readOperator("d2_4th.txt","h_4th.txt") Laplace = sbp.Laplace1D(g,1,op) -init(x) = sin(x) +init(x) = cos(x) v = sbp.Grid.evalOn(g,init) u = zeros(length(v))
--- a/plotDerivative2d.jl Tue Jan 15 10:25:56 2019 +0100 +++ b/plotDerivative2d.jl Tue Jan 15 12:26:06 2019 +0100 @@ -9,8 +9,8 @@ sbp.apply!(Laplace,u,v) -@show u -@show u'*u +#@show u +#@show u'*u sbp.Grid.plotgridfunction(g,u)
--- a/sbpD2.jl Tue Jan 15 10:25:56 2019 +0100 +++ b/sbpD2.jl Tue Jan 15 12:26:06 2019 +0100 @@ -1,24 +1,18 @@ abstract type ConstantStencilOperator end -function apply!(op::ConstantStencilOperator, u::AbstractVector, v::AbstractVector, h::Real) - N = length(v) +function apply(op::ConstantStencilOperator, h::Real, v::AbstractVector, i::Int) cSize = closureSize(op) + N = length(v) - for i ∈ range(1; length=cSize) - u[i] = apply(op.closureStencils[i], v, i)/h^2 + if i ∈ range(1; length=cSize) + uᵢ = apply(op.closureStencils[i], v, i)/h^2 + elseif i ∈ range(N - cSize+1, length=cSize) + uᵢ = Int(op.parity)*apply(flip(op.closureStencils[N-i+1]), v, i)/h^2 + else + uᵢ = apply(op.innerStencil, v, i)/h^2 end - innerStart = 1 + cSize - innerEnd = N - cSize - for i ∈ range(innerStart, stop=innerEnd) - u[i] = apply(op.innerStencil, v, i)/h^2 - end - - for i ∈ range(innerEnd+1, length=cSize) - u[i] = Int(op.parity)*apply(flip(op.closureStencils[N-i+1]), v, i)/h^2 - end - - return nothing + return uᵢ end @enum Parity begin