Mercurial > repos > public > sbplib_julia

annotate src/SbpOperators/boundaryops/normal_derivative.jl @ 1786:1f42944d4a72 feature/sbp_operators/laplace_curvilinear

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix errors introduced by merge
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 16 Sep 2024 11:03:37 +0200
parents 29b96fc75bee
children 835b1dcee38e
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
620
bfc893d03cbf Add NormalDerivative as a BoundaryOperator and reintroduce tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
1 """
1329
e94ddef5e72f Clean up documentation for changed types in SbpOperatorClean up documentation for changed types in SbpOperatorss
Jonatan Werpers <jonatan@werpers.com>
parents: 1283
diff changeset
2 normal_derivative(g, stencil_set::StencilSet, boundary)
e94ddef5e72f Clean up documentation for changed types in SbpOperatorClean up documentation for changed types in SbpOperatorss
Jonatan Werpers <jonatan@werpers.com>
parents: 1283
diff changeset
3 normal_derivative(g::TensorGrid, stencil_set::StencilSet, boundary::TensorGridBoundary)
e94ddef5e72f Clean up documentation for changed types in SbpOperatorClean up documentation for changed types in SbpOperatorss
Jonatan Werpers <jonatan@werpers.com>
parents: 1283
diff changeset
4 normal_derivative(g::EquidistantGrid, stencil_set::StencilSet, boundary)
620
bfc893d03cbf Add NormalDerivative as a BoundaryOperator and reintroduce tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
5
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 947
diff changeset
6 Creates the normal derivative boundary operator `d` as a `LazyTensor`
620
bfc893d03cbf Add NormalDerivative as a BoundaryOperator and reintroduce tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
7
1347
08f06bfacd5c Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1329
diff changeset
8 `d` computes the normal derivative at `boundary` of a grid function on `g` using the
08f06bfacd5c Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1329
diff changeset
9 'd1' stencil in `stencil_set`. `d'` is the prolongation of the normal
08f06bfacd5c Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1329
diff changeset
10 derivative of a grid function to the whole of `g` using the same stencil. On a
08f06bfacd5c Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1329
diff changeset
11 one-dimensional grid, `d` is a `BoundaryOperator`. On a multi-dimensional
08f06bfacd5c Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1329
diff changeset
12 grid, `d` is the inflation of a `BoundaryOperator`.
947
38d1752a9aff Reformat "See also:"
Jonatan Werpers <jonatan@werpers.com>
parents: 936
diff changeset
13
1104
c0ab81e4c39c Update docs
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1100
diff changeset
14 See also: [`BoundaryOperator`](@ref), [`LazyTensors.inflate`](@ref).
620
bfc893d03cbf Add NormalDerivative as a BoundaryOperator and reintroduce tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
15 """
1329
e94ddef5e72f Clean up documentation for changed types in SbpOperatorClean up documentation for changed types in SbpOperatorss
Jonatan Werpers <jonatan@werpers.com>
parents: 1283
diff changeset
16 function normal_derivative end
e94ddef5e72f Clean up documentation for changed types in SbpOperatorClean up documentation for changed types in SbpOperatorss
Jonatan Werpers <jonatan@werpers.com>
parents: 1283
diff changeset
17
e94ddef5e72f Clean up documentation for changed types in SbpOperatorClean up documentation for changed types in SbpOperatorss
Jonatan Werpers <jonatan@werpers.com>
parents: 1283
diff changeset
18
1283
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
19 function normal_derivative(g::TensorGrid, stencil_set::StencilSet, boundary::TensorGridBoundary)
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
20 op = normal_derivative(g.grids[grid_id(boundary)], stencil_set, boundary_id(boundary))
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
21 return LazyTensors.inflate(op, size(g), grid_id(boundary))
620
bfc893d03cbf Add NormalDerivative as a BoundaryOperator and reintroduce tests.
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
diff changeset
22 end
925
6b47a9ee1632 Add functionality for creating operators from a stencil set. Note: Tests are not updated yet!
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 871
diff changeset
23
1283
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
24 function normal_derivative(g::EquidistantGrid, stencil_set::StencilSet, boundary)
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
25 closure_stencil = parse_stencil(stencil_set["d1"]["closure"])
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
26 h_inv = inverse_spacing(g)
925
6b47a9ee1632 Add functionality for creating operators from a stencil set. Note: Tests are not updated yet!
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 871
diff changeset
27
1283
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
28 scaled_stencil = scale(closure_stencil,h_inv)
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
29 return BoundaryOperator(g, scaled_stencil, boundary)
54c3ed752730 Make tests for normal_derivative work
Jonatan Werpers <jonatan@werpers.com>
parents: 1104
diff changeset
30 end
1654
f4dc17cfafce Start adding normal derivative for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1347
diff changeset
31
f4dc17cfafce Start adding normal derivative for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1347
diff changeset
32 function normal_derivative(g::MappedGrid, stencil_set::StencilSet, boundary)
1657
9c84e97de895 Fix bug in normal derivative
Jonatan Werpers <jonatan@werpers.com>
parents: 1656
diff changeset
33 k = grid_id(boundary)
1658
cc9d18a5ff2d Simplify normal_derivative
Jonatan Werpers <jonatan@werpers.com>
parents: 1657
diff changeset
34 b_indices = boundary_indices(g, boundary)
1656
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
35
1657
9c84e97de895 Fix bug in normal derivative
Jonatan Werpers <jonatan@werpers.com>
parents: 1656
diff changeset
36 # Compute the weights for the logical derivatives
1696
29b96fc75bee Merge feature/grids/manifolds
Jonatan Werpers <jonatan@werpers.com>
parents: 1680
diff changeset
37 g⁻¹ = metric_tensor_inverse(g)
1656
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
38 α = map(CartesianIndices(g⁻¹)[b_indices...]) do I # TODO: Fix iterator here
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
39 gᵏⁱ = g⁻¹[I][k,:]
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
40 gᵏᵏ = g⁻¹[I][k,k]
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
41
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
42 gᵏⁱ./sqrt(gᵏᵏ)
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
43 end
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
44
1680
b30db2ea34ed Add concistency tests for normal_derivative and and fix bug regarding sign of boundary
Jonatan Werpers <jonatan@werpers.com>
parents: 1658
diff changeset
45 σ = ScalingTensor(
b30db2ea34ed Add concistency tests for normal_derivative and and fix bug regarding sign of boundary
Jonatan Werpers <jonatan@werpers.com>
parents: 1658
diff changeset
46 Grids._boundary_sign(component_type(g), boundary),
b30db2ea34ed Add concistency tests for normal_derivative and and fix bug regarding sign of boundary
Jonatan Werpers <jonatan@werpers.com>
parents: 1658
diff changeset
47 size(boundary_grid(g,boundary)),
b30db2ea34ed Add concistency tests for normal_derivative and and fix bug regarding sign of boundary
Jonatan Werpers <jonatan@werpers.com>
parents: 1658
diff changeset
48 )
b30db2ea34ed Add concistency tests for normal_derivative and and fix bug regarding sign of boundary
Jonatan Werpers <jonatan@werpers.com>
parents: 1658
diff changeset
49
b30db2ea34ed Add concistency tests for normal_derivative and and fix bug regarding sign of boundary
Jonatan Werpers <jonatan@werpers.com>
parents: 1658
diff changeset
50
1657
9c84e97de895 Fix bug in normal derivative
Jonatan Werpers <jonatan@werpers.com>
parents: 1656
diff changeset
51 # Assemble difference operator
1656
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
52 mapreduce(+,1:ndims(g)) do i
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
53 if i == k
1786
1f42944d4a72 Fix errors introduced by merge
Jonatan Werpers <jonatan@werpers.com>
parents: 1696
diff changeset
54 ∂_ξᵢ = normal_derivative(logical_grid(g), stencil_set, boundary)
1656
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
55 else
1786
1f42944d4a72 Fix errors introduced by merge
Jonatan Werpers <jonatan@werpers.com>
parents: 1696
diff changeset
56 e = boundary_restriction(logical_grid(g), stencil_set, boundary)
1f42944d4a72 Fix errors introduced by merge
Jonatan Werpers <jonatan@werpers.com>
parents: 1696
diff changeset
57 ∂_ξᵢ = σ ∘ e ∘ first_derivative(logical_grid(g), stencil_set, i)
1656
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
58 end
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
59
89456aa6fa80 Flesh out normal_derivative implementation some more.
Jonatan Werpers <jonatan@werpers.com>
parents: 1654
diff changeset
60 αᵢ = componentview(α,i)
1657
9c84e97de895 Fix bug in normal derivative
Jonatan Werpers <jonatan@werpers.com>
parents: 1656
diff changeset
61 DiagonalTensor(αᵢ) ∘ ∂_ξᵢ
1654
f4dc17cfafce Start adding normal derivative for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1347
diff changeset
62 end
f4dc17cfafce Start adding normal derivative for mapped grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1347
diff changeset
63 end