Mercurial > repos > public > sbplib_julia

annotate src/Grids/manifolds.jl @ 1931:bb2c708d724d feature/grids/manifolds

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Implement connections for unstructured atlas
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 07 Feb 2025 09:18:09 +0100
parents c5affb342e73
children ac63820222f8
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
1 """
1581
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
2 Chart{D}
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
3
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
4 A parametrized description of a manifold or part of a manifold.
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
5 """
1581
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
6 struct Chart{D, PST<:ParameterSpace{D}, MT}
1564
35fe4375b35f Export things and fix ConcreteChart and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1558
diff changeset
7 mapping::MT
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 parameterspace::PST
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
10
1781
a73838c9ef94 Let Chart implement Base.ndims instead of domain_dim
Jonatan Werpers <jonatan@werpers.com>
parents: 1780
diff changeset
11 Base.ndims(::Chart{D}) where D = D
1581
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
12 (c::Chart)(ξ) = c.mapping(ξ)
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
13 parameterspace(c::Chart) = c.parameterspace
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
14
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
15 """
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
16 jacobian(c::Chart, ξ)
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17
1581
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
18 The jacobian of the mapping evaluated at `ξ`. This defers to the
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
19 implementation of `jacobian` for the mapping itself. If no implementation is
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
20 available one can easily be specified for either the mapping function or the
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
21 chart itself.
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
22 ```julia
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
23 c = Chart(f, ps)
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
24 jacobian(f::typeof(f), ξ) = f′(ξ)
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
25 ```
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
26 or
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
27 ```julia
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
28 c = Chart(f, ps)
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
29 jacobian(c::typeof(c),ξ) = f′(ξ)
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
30 ```
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
31 which will both allow calling `jacobian(c,ξ)`.
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
32 """
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
33 jacobian(c::Chart, ξ) = jacobian(c.mapping, ξ)
1644
e213bd857f3f Add some todos and tbds
Jonatan Werpers <jonatan@werpers.com>
parents: 1625
diff changeset
34 # TBD: Can we register a error hint for when jacobian is called with a function that doesn't have a registered jacobian?
1581
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
35
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
36
1867
de4b4f2aee4f Add some tests for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1844
diff changeset
37 # TBD: Should Charts, parameterspaces, Atlases, have boundary names?
1644
e213bd857f3f Add some todos and tbds
Jonatan Werpers <jonatan@werpers.com>
parents: 1625
diff changeset
38
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
39 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
40 Atlas
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
41
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42 A collection of charts and their connections.
1844
1987347752ef Add stub for connections(::Atlas)
Jonatan Werpers <jonatan@werpers.com>
parents: 1782
diff changeset
43 Should implement methods for `charts` and `connections`.
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
44 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
45 abstract type Atlas end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
46
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
48 charts(::Atlas)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
50 The colloction of charts in the atlas.
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
51 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
52 function charts end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
53
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
54 """
1844
1987347752ef Add stub for connections(::Atlas)
Jonatan Werpers <jonatan@werpers.com>
parents: 1782
diff changeset
55 connections(::Atlas)
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
56
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
57 TBD: What exactly should this return?
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
58 """
1844
1987347752ef Add stub for connections(::Atlas)
Jonatan Werpers <jonatan@werpers.com>
parents: 1782
diff changeset
59 function connections end
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
60
1927
fe18cdad2b27 Test construction of CartesianAtlas with different type on the charts
Jonatan Werpers <jonatan@werpers.com>
parents: 1925
diff changeset
61 struct CartesianAtlas{D,C<:Chart} <: Atlas
fe18cdad2b27 Test construction of CartesianAtlas with different type on the charts
Jonatan Werpers <jonatan@werpers.com>
parents: 1925
diff changeset
62 charts::AbstractArray{C,D}
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
63 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
64
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
65 charts(a::CartesianAtlas) = a.charts
1919
71d218593cac Add implementation for size to CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1914
diff changeset
66
1920
ae83c91286a2 Implement connections for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1919
diff changeset
67 function connections(a::CartesianAtlas)
ae83c91286a2 Implement connections for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1919
diff changeset
68 c = Tuple{MultiBlockBoundary, MultiBlockBoundary}[]
ae83c91286a2 Implement connections for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1919
diff changeset
69
1928
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
70 for d ∈ 1:ndims(charts(a))
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
71 Is = eachslice(CartesianIndices(charts(a)); dims=d)
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
72 for i ∈ 1:length(Is)-1 # For each interface between slices
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
73 for jk ∈ eachindex(Is[i]) # For each block in slice
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
74 Iᵢⱼₖ = Tuple(Is[i][jk])
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
75 Iᵢ₊₁ⱼₖ = Tuple(Is[i+1][jk])
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
76 push!(c,
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
77 (
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
78 MultiBlockBoundary{Iᵢⱼₖ, CartesianBoundary{d,UpperBoundary}}(),
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
79 MultiBlockBoundary{Iᵢ₊₁ⱼₖ, CartesianBoundary{d,LowerBoundary}}(),
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
80 )
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
81 )
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
82 end
96b3e9c516fe Make cartesian atlas work for any dimension
Jonatan Werpers <jonatan@werpers.com>
parents: 1927
diff changeset
83 end
1920
ae83c91286a2 Implement connections for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1919
diff changeset
84 end
ae83c91286a2 Implement connections for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1919
diff changeset
85
ae83c91286a2 Implement connections for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1919
diff changeset
86 return c
ae83c91286a2 Implement connections for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1919
diff changeset
87 end
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
88
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
89 struct UnstructuredAtlas <: Atlas
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
90 charts::Vector{Chart}
1925
297e3d74f697 Add type to connetions field in UnstructuredAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1920
diff changeset
91 connections::Vector{Tuple{MultiBlockBoundary, MultiBlockBoundary}}
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
92 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
93
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
94 charts(a::UnstructuredAtlas) = a.charts
1931
bb2c708d724d Implement connections for unstructured atlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1929
diff changeset
95 connections(a::UnstructuredAtlas) = a.connections