Mercurial > repos > public > sbplib_julia

annotate src/Grids/mapped_grid.jl @ 1776:265a740a49da feature/grids/curvilinear

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Remove reivew comment regarding implementation of normal since it requires more computation and we know of no other benefits
author Jonatan Werpers <jonatan@werpers.com>
date Sun, 15 Sep 2024 18:07:58 +0200
parents ecec2b0eea0f
children 819ab806960f 43c0bfc13de3
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
1 """
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
2 MappedGrid{T,D} <: Grid{T,D}
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
3
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
4 A grid defined by a coordinate mapping from a logical grid to some physical
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
5 coordinates. The physical coordinates and the Jacobian are stored as grid
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
6 functions corresponding to the logical grid.
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
7
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
8 See also: [`logical_grid`](@ref), [`jacobian`](@ref), [`metric_tensor`](@ref).
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
9 """
1773
08e52f442872 Replace AbstractArray{<:Any,2} with AbstractMatrix{<:Any}
Jonatan Werpers <jonatan@werpers.com>
parents: 1768
diff changeset
10 struct MappedGrid{T,D, GT<:Grid{<:Any,D}, CT<:AbstractArray{T,D}, JT<:AbstractArray{<:AbstractMatrix{<:Any}, D}} <: Grid{T,D}
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
11 logical_grid::GT
1430
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
12 physicalcoordinates::CT
1505
63101a1cd0e6 Remove some comments
Jonatan Werpers <jonatan@werpers.com>
parents: 1504
diff changeset
13 jacobian::JT
1743
49bd573ab07f Implement checking of sizes for inputs to MappedGrid
Jonatan Werpers <jonatan@werpers.com>
parents: 1741
diff changeset
14
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
15 """
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
16 MappedGrid(logical_grid, physicalcoordinates, jacobian)
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
17
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
18 A MappedGrid with the given physical coordinates and jacobian.
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
19 """
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
20 function MappedGrid(logical_grid::GT, physicalcoordinates::CT, jacobian::JT) where {T,D, GT<:Grid{<:Any,D}, CT<:AbstractArray{T,D}, JT<:AbstractArray{<:AbstractMatrix{<:Any}, D}}
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
21 if !(size(logical_grid) == size(physicalcoordinates) == size(jacobian))
1743
49bd573ab07f Implement checking of sizes for inputs to MappedGrid
Jonatan Werpers <jonatan@werpers.com>
parents: 1741
diff changeset
22 throw(ArgumentError("Sizes must match"))
49bd573ab07f Implement checking of sizes for inputs to MappedGrid
Jonatan Werpers <jonatan@werpers.com>
parents: 1741
diff changeset
23 end
49bd573ab07f Implement checking of sizes for inputs to MappedGrid
Jonatan Werpers <jonatan@werpers.com>
parents: 1741
diff changeset
24
1744
c38eead8be17 Add checks for the size of the jacobian
Jonatan Werpers <jonatan@werpers.com>
parents: 1743
diff changeset
25 if size(first(jacobian)) != (length(first(physicalcoordinates)),D)
c38eead8be17 Add checks for the size of the jacobian
Jonatan Werpers <jonatan@werpers.com>
parents: 1743
diff changeset
26 throw(ArgumentError("The size of the jacobian must match the dimensions of the grid and coordinates"))
c38eead8be17 Add checks for the size of the jacobian
Jonatan Werpers <jonatan@werpers.com>
parents: 1743
diff changeset
27 end
c38eead8be17 Add checks for the size of the jacobian
Jonatan Werpers <jonatan@werpers.com>
parents: 1743
diff changeset
28
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
29 return new{T,D,GT,CT,JT}(logical_grid, physicalcoordinates, jacobian)
1743
49bd573ab07f Implement checking of sizes for inputs to MappedGrid
Jonatan Werpers <jonatan@werpers.com>
parents: 1741
diff changeset
30 end
1430
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
31 end
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
32
1705
4870fc3faa25 Add tests for equallity of mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1704
diff changeset
33 function Base.:(==)(a::MappedGrid, b::MappedGrid)
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
34 same_logical_grid = logical_grid(a) == logical_grid(b)
1705
4870fc3faa25 Add tests for equallity of mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1704
diff changeset
35 same_coordinates = collect(a) == collect(b)
4870fc3faa25 Add tests for equallity of mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1704
diff changeset
36 same_jacobian = jacobian(a) == jacobian(b)
4870fc3faa25 Add tests for equallity of mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1704
diff changeset
37
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
38 return same_logical_grid && same_coordinates && same_jacobian
1705
4870fc3faa25 Add tests for equallity of mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1704
diff changeset
39 end
4870fc3faa25 Add tests for equallity of mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1704
diff changeset
40
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
41 """
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
42 logical_grid(g::MappedGrid)
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
43
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
44 The logical grid of `g`.
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
45 """
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
46 logical_grid(g::MappedGrid) = g.logical_grid
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
47
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
48 """
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
49 jacobian(g::MappedGrid)
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
50
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
51 The Jacobian matrix of `g` as a grid function.
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
52 """
1506
535f32316637 Rename from curvilinear to mapped
Jonatan Werpers <jonatan@werpers.com>
parents: 1505
diff changeset
53 jacobian(g::MappedGrid) = g.jacobian
1431
6adf31ba6cfd Add `jacobian` and `logicalgrid`
Jonatan Werpers <jonatan@werpers.com>
parents: 1430
diff changeset
54
1430
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
55
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
56 # Indexing interface
1506
535f32316637 Rename from curvilinear to mapped
Jonatan Werpers <jonatan@werpers.com>
parents: 1505
diff changeset
57 Base.getindex(g::MappedGrid, I::Vararg{Int}) = g.physicalcoordinates[I...]
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
58 Base.eachindex(g::MappedGrid) = eachindex(g.logical_grid)
1432
64b60b42d367 Implement indexing interface
Jonatan Werpers <jonatan@werpers.com>
parents: 1431
diff changeset
59
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
60 Base.firstindex(g::MappedGrid, d) = firstindex(g.logical_grid, d)
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
61 Base.lastindex(g::MappedGrid, d) = lastindex(g.logical_grid, d)
1430
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
62
1450
647c8b18b84f Implement iterator interface
Jonatan Werpers <jonatan@werpers.com>
parents: 1449
diff changeset
63 # Iteration interface
1506
535f32316637 Rename from curvilinear to mapped
Jonatan Werpers <jonatan@werpers.com>
parents: 1505
diff changeset
64 Base.iterate(g::MappedGrid) = iterate(g.physicalcoordinates)
535f32316637 Rename from curvilinear to mapped
Jonatan Werpers <jonatan@werpers.com>
parents: 1505
diff changeset
65 Base.iterate(g::MappedGrid, state) = iterate(g.physicalcoordinates, state)
1430
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
66
1506
535f32316637 Rename from curvilinear to mapped
Jonatan Werpers <jonatan@werpers.com>
parents: 1505
diff changeset
67 Base.IteratorSize(::Type{<:MappedGrid{<:Any, D}}) where D = Base.HasShape{D}()
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
68 Base.length(g::MappedGrid) = length(g.logical_grid)
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
69 Base.size(g::MappedGrid) = size(g.logical_grid)
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
70 Base.size(g::MappedGrid, d) = size(g.logical_grid, d)
1430
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
71
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
72 boundary_identifiers(g::MappedGrid) = boundary_identifiers(g.logical_grid)
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
73 boundary_indices(g::MappedGrid, id::TensorGridBoundary) = boundary_indices(g.logical_grid, id)
1430
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
74
1768
672897f64d58 More reivew comments
Vidar Stiernström <vidar.stiernstrom@gmail.com>
parents: 1752
diff changeset
75 # Review: Error when calling plot(boundary_grid(g, id))
672897f64d58 More reivew comments
Vidar Stiernström <vidar.stiernstrom@gmail.com>
parents: 1752
diff changeset
76 # Currently need to collect first, i.e., plot(collect(boundary_grid(g, id)))
1506
535f32316637 Rename from curvilinear to mapped
Jonatan Werpers <jonatan@werpers.com>
parents: 1505
diff changeset
77 function boundary_grid(g::MappedGrid, id::TensorGridBoundary)
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
78 b_indices = boundary_indices(g.logical_grid, id)
1503
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
79
1504
976f5784d7b9 Spelling
Jonatan Werpers <jonatan@werpers.com>
parents: 1503
diff changeset
80 # Calculate indices of needed jacobian components
1503
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
81 D = ndims(g)
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
82 all_indices = SVector{D}(1:D)
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
83 free_variable_indices = deleteat(all_indices, grid_id(id))
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
84 jacobian_components = (:, free_variable_indices)
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
85
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
86 # Create grid function for boundary grid jacobian
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
87 boundary_jacobian = componentview((@view g.jacobian[b_indices...]) , jacobian_components...)
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
88 boundary_physicalcoordinates = @view g.physicalcoordinates[b_indices...]
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
89
1506
535f32316637 Rename from curvilinear to mapped
Jonatan Werpers <jonatan@werpers.com>
parents: 1505
diff changeset
90 return MappedGrid(
1774
035af82f559a Rename logicalgrid to logical_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1773
diff changeset
91 boundary_grid(g.logical_grid, id),
1503
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
92 boundary_physicalcoordinates,
704a84eef8b6 Add tests for boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1502
diff changeset
93 boundary_jacobian,
1451
2e08f3444354 Implement boundary_identifiers, and a wishful implementation of boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1450
diff changeset
94 )
2e08f3444354 Implement boundary_identifiers, and a wishful implementation of boundary_grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1450
diff changeset
95 end
1430
9fc3c1af33e5 Add testsets and a few tests
Jonatan Werpers <jonatan@werpers.com>
parents: 1426
diff changeset
96
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
97
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
98 """
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
99 mapped_grid(x, J, size::Vararg{Int})
1527
69790e9d1652 Remove tests for refine and coarsen
Jonatan Werpers <jonatan@werpers.com>
parents: 1506
diff changeset
100
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
101 A `MappedGrid` with a default logical grid on a unit hyper box. `x` and `J`
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
102 are functions to be evaluated on the logical grid and `size` determines the
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
103 size of the logical grid.
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
104 """
1706
11640aa3e348 Add method and refactor mapped_grid()
Jonatan Werpers <jonatan@werpers.com>
parents: 1705
diff changeset
105 function mapped_grid(x, J, size::Vararg{Int})
1502
a2dc80396808 Add functions `curvilinear_grid`
Jonatan Werpers <jonatan@werpers.com>
parents: 1501
diff changeset
106 D = length(size)
1570
063a2bfb03da Fix bugs in calls to equidistant_grid after merge with default
Jonatan Werpers <jonatan@werpers.com>
parents: 1560
diff changeset
107 lg = equidistant_grid(ntuple(i->0., D), ntuple(i->1., D), size...)
1706
11640aa3e348 Add method and refactor mapped_grid()
Jonatan Werpers <jonatan@werpers.com>
parents: 1705
diff changeset
108 return mapped_grid(lg, x, J)
11640aa3e348 Add method and refactor mapped_grid()
Jonatan Werpers <jonatan@werpers.com>
parents: 1705
diff changeset
109 end
11640aa3e348 Add method and refactor mapped_grid()
Jonatan Werpers <jonatan@werpers.com>
parents: 1705
diff changeset
110
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
111 """
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
112 mapped_grid(lg::Grid, x, J)
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
113
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
114 A `MappedGrid` with logical grid `lg`. Physical coordinates and Jacobian are
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
115 determined by the functions `x` and `J`.
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
116 """
1706
11640aa3e348 Add method and refactor mapped_grid()
Jonatan Werpers <jonatan@werpers.com>
parents: 1705
diff changeset
117 function mapped_grid(lg::Grid, x, J)
1506
535f32316637 Rename from curvilinear to mapped
Jonatan Werpers <jonatan@werpers.com>
parents: 1505
diff changeset
118 return MappedGrid(
1502
a2dc80396808 Add functions `curvilinear_grid`
Jonatan Werpers <jonatan@werpers.com>
parents: 1501
diff changeset
119 lg,
a2dc80396808 Add functions `curvilinear_grid`
Jonatan Werpers <jonatan@werpers.com>
parents: 1501
diff changeset
120 map(x,lg),
a2dc80396808 Add functions `curvilinear_grid`
Jonatan Werpers <jonatan@werpers.com>
parents: 1501
diff changeset
121 map(J,lg),
a2dc80396808 Add functions `curvilinear_grid`
Jonatan Werpers <jonatan@werpers.com>
parents: 1501
diff changeset
122 )
a2dc80396808 Add functions `curvilinear_grid`
Jonatan Werpers <jonatan@werpers.com>
parents: 1501
diff changeset
123 end
1560
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
124
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
125 """
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
126 metric_tensor(g::MappedGrid)
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
127
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
128 The metric tensor of `g` as a grid function.
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
129 """
1682
4ce9cb21220c Fix naming of metrix tensor functions
Jonatan Werpers <jonatan@werpers.com>
parents: 1681
diff changeset
130 function metric_tensor(g::MappedGrid)
1560
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
131 return map(jacobian(g)) do ∂x∂ξ
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
132 ∂x∂ξ'*∂x∂ξ
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
133 end
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
134 end
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
135
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
136 """
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
137 metric_tensor_inverse(g::MappedGrid)
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
138
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
139 The inverse of the metric tensor of `g` as a grid function.
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
140 """
1682
4ce9cb21220c Fix naming of metrix tensor functions
Jonatan Werpers <jonatan@werpers.com>
parents: 1681
diff changeset
141 function metric_tensor_inverse(g::MappedGrid)
1560
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
142 return map(jacobian(g)) do ∂x∂ξ
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
143 inv(∂x∂ξ'*∂x∂ξ)
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
144 end
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
145 end
5d32ecb98db8 Add functions for jacobian_determiant, geometric_tensor, geometric_tensor_inverse
Jonatan Werpers <jonatan@werpers.com>
parents: 1527
diff changeset
146
1684
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
147 function min_spacing(g::MappedGrid{T,1} where T)
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
148 n, = size(g)
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
149
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
150 ms = Inf
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
151 for i ∈ 1:n-1
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
152 ms = min(ms, norm(g[i+1]-g[i]))
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
153 end
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
154
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
155 return ms
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
156 end
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
157
1688
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
158 function min_spacing(g::MappedGrid{T,2} where T)
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
159 n, m = size(g)
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
160
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
161 ms = Inf
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
162 for i ∈ 1:n-1, j ∈ 1:m-1 # loop over each cell of the grid
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
163
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
164 ms = min(
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
165 ms,
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
166 norm(g[i+1,j]-g[i,j]),
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
167 norm(g[i,j+1]-g[i,j]),
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
168
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
169 norm(g[i+1,j]-g[i+1,j+1]),
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
170 norm(g[i,j+1]-g[i+1,j+1]),
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
171
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
172 norm(g[i+1,j+1]-g[i,j]),
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
173 norm(g[i+1,j]-g[i,j+1]),
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
174 )
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
175 # NOTE: This could be optimized to avoid checking all interior edges twice.
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
176 end
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
177
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
178 return ms
72776d3d5fd6 Add min_spacing for 2D mapped grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1684
diff changeset
179 end
1684
22a9992471be Add min_spacing for mapped 1d grid
Jonatan Werpers <jonatan@werpers.com>
parents: 1682
diff changeset
180
1659
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
181 """
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
182 normal(g::MappedGrid, boundary)
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
183
1746
44faa334adc6 Add docs
Jonatan Werpers <jonatan@werpers.com>
parents: 1745
diff changeset
184 The outward pointing normal as a grid function on the corresponding boundary grid.
1659
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
185 """
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
186 function normal(g::MappedGrid, boundary)
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
187 b_indices = boundary_indices(g, boundary)
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
188 σ =_boundary_sign(component_type(g), boundary)
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
189 return map(jacobian(g)[b_indices...]) do ∂x∂ξ
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
190 ∂ξ∂x = inv(∂x∂ξ)
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
191 k = grid_id(boundary)
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
192 σ*∂ξ∂x[k,:]/norm(∂ξ∂x[k,:])
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
193 end
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
194 end
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
195
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
196 function _boundary_sign(T, boundary)
1736
863385aae454 Merge default
Jonatan Werpers <jonatan@werpers.com>
parents: 1706
diff changeset
197 if boundary_id(boundary) == UpperBoundary()
1659
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
198 return one(T)
1736
863385aae454 Merge default
Jonatan Werpers <jonatan@werpers.com>
parents: 1706
diff changeset
199 elseif boundary_id(boundary) == LowerBoundary()
1659
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
200 return -one(T)
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
201 else
1736
863385aae454 Merge default
Jonatan Werpers <jonatan@werpers.com>
parents: 1706
diff changeset
202 throw(ArgumentError("The boundary identifier must be either `LowerBoundary()` or `UpperBoundary()`"))
1659
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
203 end
3bbcd496e021 Add function for computing the normal at the boundary of a mapped grid as a grid function
Jonatan Werpers <jonatan@werpers.com>
parents: 1645
diff changeset
204 end