Mercurial > repos > public > sbplib_julia

annotate SbpOperators/src/d2.jl @ 254:4ca3794fffef boundary_conditions

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add apply_quadrature to SbpOperators
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 27 Jun 2019 16:10:25 +0200
parents 396eadb652bd
children f89718833620
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
253
396eadb652bd Rename function closureSize to closuresize
Jonatan Werpers <jonatan@werpers.com>
parents: 249
diff changeset
1 export D2, closuresize, readOperator, apply_e, apply_d, apply_e_T, apply_d_T
249
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
2
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
3 @enum Parity begin
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
4 odd = -1
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
5 even = 1
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
6 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
7
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
8 struct D2{T,N,M,K} <: ConstantStencilOperator
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
9 quadratureClosure::NTuple{M,T}
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
10 innerStencil::Stencil{T,N}
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
11 closureStencils::NTuple{M,Stencil{T,K}}
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
12 eClosure::Stencil{T,M}
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
13 dClosure::Stencil{T,M}
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
14 parity::Parity
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
15 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
16
253
396eadb652bd Rename function closureSize to closuresize
Jonatan Werpers <jonatan@werpers.com>
parents: 249
diff changeset
17 function closuresize(D::D2)::Int
249
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
18 return length(D.quadratureClosure)
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
19 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
20
254
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
21 apply_quadrature(op::D2{T}, h::Real, v::T, i::Integer, N::Integer, ::Type{Lower}) where T = v*h*op.quadratureClosure[i]
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
22 apply_quadrature(op::D2{T}, h::Real, v::T, i::Integer, N::Integer, ::Type{Upper}) where T = v*h*op.quadratureClosure[N-i+1]
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
23 apply_quadrature(op::D2{T}, h::Real, v::T, i::Integer, N::Integer, ::Type{Interior}) where T = v*h
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
24
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
25 function apply_quadrature(op::D2{T}, h::Real, v::T, i::Integer, N::Integer) where T
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
26 r = getregion(i, closuresize(op), N)
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
27 return apply_quadrature(op, h, v, i, N, r)
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
28 end
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
29 export apply_quadrature
4ca3794fffef Add apply_quadrature to SbpOperators
Jonatan Werpers <jonatan@werpers.com>
parents: 253
diff changeset
30
249
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
31 function apply_e_T(op::D2, v::AbstractVector, ::Type{Lower})
253
396eadb652bd Rename function closureSize to closuresize
Jonatan Werpers <jonatan@werpers.com>
parents: 249
diff changeset
32 @boundscheck if length(v) < closuresize(op)
249
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
33 throw(BoundsError())
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
34 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
35 apply(op.eClosure,v,1)
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
36 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
37
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
38 function apply_e_T(op::D2, v::AbstractVector, ::Type{Upper})
253
396eadb652bd Rename function closureSize to closuresize
Jonatan Werpers <jonatan@werpers.com>
parents: 249
diff changeset
39 @boundscheck if length(v) < closuresize(op)
249
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
40 throw(BoundsError())
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
41 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
42 apply(flip(op.eClosure),v,length(v))
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
43 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
44
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
45
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
46 function apply_e(op::D2, v::Number, N::Integer, i::Integer, ::Type{Lower})
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
47 @boundscheck if !(0<length(i) <= N)
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
48 throw(BoundsError())
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
49 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
50 op.eClosure[i-1]*v
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
51 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
52
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
53 function apply_e(op::D2, v::Number, N::Integer, i::Integer, ::Type{Upper})
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
54 @boundscheck if !(0<length(i) <= N)
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
55 throw(BoundsError())
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
56 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
57 op.eClosure[N-i]*v
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
58 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
59
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
60 function apply_d_T(op::D2, h_inv::Real, v::AbstractVector, ::Type{Lower})
253
396eadb652bd Rename function closureSize to closuresize
Jonatan Werpers <jonatan@werpers.com>
parents: 249
diff changeset
61 @boundscheck if length(v) < closuresize(op)
249
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
62 throw(BoundsError())
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
63 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
64 h_inv*apply(op.dClosure,v,1)
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
65 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
66
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
67 function apply_d_T(op::D2, h_inv::Real, v::AbstractVector, ::Type{Upper})
253
396eadb652bd Rename function closureSize to closuresize
Jonatan Werpers <jonatan@werpers.com>
parents: 249
diff changeset
68 @boundscheck if length(v) < closuresize(op)
249
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
69 throw(BoundsError())
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
70 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
71 -h_inv*apply(flip(op.dClosure),v,length(v))
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
72 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
73
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
74 function apply_d(op::D2, h_inv::Real, v::Number, N::Integer, i::Integer, ::Type{Lower})
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
75 @boundscheck if !(0<length(i) <= N)
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
76 throw(BoundsError())
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
77 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
78 h_inv*op.dClosure[i-1]*v
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
79 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
80
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
81 function apply_d(op::D2, h_inv::Real, v::Number, N::Integer, i::Integer, ::Type{Upper})
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
82 @boundscheck if !(0<length(i) <= N)
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
83 throw(BoundsError())
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
84 end
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
85 -h_inv*op.dClosure[N-i]*v
7cb4492ccd60 Refactor package SbpOperators
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
86 end