annotate src/Grids/manifolds.jl @ 1869:20bd8887db0d feature/grids/manifolds

Fix Aqua and JET fails
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 24 Jan 2025 23:40:28 +0100
parents de4b4f2aee4f
children 04c251bccbd4
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 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
2 ParameterSpace{D}
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 space of parameters of dimension `D`. Used with `Chart` to indicate which
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
5 parameters are valid for that chart.
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
6
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
7 Common parameter spaces are created using the functions unit sized spaces
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
8 * `unitinterval`
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
9 * `unitrectangle`
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
10 * `unitbox`
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
11 * `unittriangle`
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
12 * `unittetrahedron`
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
13 * `unithyperbox`
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
14 * `unitsimplex`
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
15
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
16 See also: [`Interval`](@ref), [`Rectangle`](@ref), [`Box`](@ref),
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
17 [`Triangle`](@ref), [`Tetrahedron`](@ref), [`HyperBox`](@ref),
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18 [`Simplex`](@ref),
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
19 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
20 abstract type ParameterSpace{D} end
1580
fdee60ab8c4e Add ndims method for ParameterSpace
Jonatan Werpers <jonatan@werpers.com>
parents: 1579
diff changeset
21 Base.ndims(::ParameterSpace{D}) where D = D
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
22
1779
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
23 struct Interval{T} <: ParameterSpace{1}
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
24 a::T
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
25 b::T
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
26
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
27 function Interval(a,b)
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
28 a, b = promote(a, b)
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
29 new{typeof(a)}(a,b)
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
30 end
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
31 end
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
32
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
33 limits(i::Interval) = (i.a, i.b)
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
34
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
35 unitinterval(T=Float64) = Interval(zero(T), one(T))
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
36
2a8a2b52a112 Refactor Interval to its own type. This allows the user better control of the coordinate type of 1d grids
Jonatan Werpers <jonatan@werpers.com>
parents: 1644
diff changeset
37
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
38 struct HyperBox{T,D} <: ParameterSpace{D}
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
39 a::SVector{D,T}
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
40 b::SVector{D,T}
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
41 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
42
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
43 function HyperBox(a,b)
1780
8ecdc5bb46be Allow mixed types in constructors for HyperBox and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1779
diff changeset
44 ET = promote_type(eltype(a),eltype(b))
8ecdc5bb46be Allow mixed types in constructors for HyperBox and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1779
diff changeset
45 T = SVector{length(a),ET}
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
46 HyperBox(convert(T,a), convert(T,b))
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
47 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
48
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
49 Rectangle{T} = HyperBox{T,2}
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
50 Box{T} = HyperBox{T,3}
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 limits(box::HyperBox, d) = (box.a[d], box.b[d])
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
53 limits(box::HyperBox) = (box.a, box.b)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
54
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
55 unitsquare(T=Float64) = unithyperbox(T,2)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
56 unitcube(T=Float64) = unithyperbox(T,3)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
57 unithyperbox(T, D) = HyperBox((@SVector zeros(T,D)), (@SVector ones(T,D)))
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
58 unithyperbox(D) = unithyperbox(Float64,D)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
59
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
60
1579
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
61 struct Simplex{T,D,NV} <: ParameterSpace{D}
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
62 verticies::NTuple{NV,SVector{D,T}}
1869
20bd8887db0d Fix Aqua and JET fails
Jonatan Werpers <jonatan@werpers.com>
parents: 1867
diff changeset
63
20bd8887db0d Fix Aqua and JET fails
Jonatan Werpers <jonatan@werpers.com>
parents: 1867
diff changeset
64 Simplex(verticies::Tuple{SVector{D,T}, Vararg{SVector{D,T},N}}) where {T,D,N} = new{T,D,N+1}(verticies)
20bd8887db0d Fix Aqua and JET fails
Jonatan Werpers <jonatan@werpers.com>
parents: 1867
diff changeset
65 Simplex(::Tuple{}) = throw(ArgumentError("Must provide at least one vertex."))
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
66 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
67
1780
8ecdc5bb46be Allow mixed types in constructors for HyperBox and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1779
diff changeset
68 function Simplex(verticies::Vararg{AbstractArray})
8ecdc5bb46be Allow mixed types in constructors for HyperBox and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1779
diff changeset
69 ET = mapreduce(eltype,promote_type,verticies)
8ecdc5bb46be Allow mixed types in constructors for HyperBox and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1779
diff changeset
70 T = SVector{length(verticies[1]),ET}
8ecdc5bb46be Allow mixed types in constructors for HyperBox and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1779
diff changeset
71
8ecdc5bb46be Allow mixed types in constructors for HyperBox and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1779
diff changeset
72 return Simplex(Tuple(convert(T,v) for v ∈ verticies))
8ecdc5bb46be Allow mixed types in constructors for HyperBox and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1779
diff changeset
73 end
1564
35fe4375b35f Export things and fix ConcreteChart and Simplex
Jonatan Werpers <jonatan@werpers.com>
parents: 1558
diff changeset
74
1579
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
75 verticies(s::Simplex) = s.verticies
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
76
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
77 Triangle{T} = Simplex{T,2}
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
78 Tetrahedron{T} = Simplex{T,3}
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
79
1579
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
80 unittriangle(T=Float64) = unitsimplex(T,2)
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
81 unittetrahedron(T=Float64) = unitsimplex(T,3)
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
82 function unitsimplex(T,D)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
83 z = @SVector zeros(T,D)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
84 unitelement = one(eltype(z))
1579
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
85 verticies = ntuple(i->setindex(z, unitelement, i), D)
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
86 return Simplex((z,verticies...))
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
87 end
1579
14d79b13b54f Add tests, fix bugs, add exports, for Simplex and friends
Jonatan Werpers <jonatan@werpers.com>
parents: 1578
diff changeset
88 unitsimplex(D) = unitsimplex(Float64, D)
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
89
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
90 """
1581
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
91 Chart{D}
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
92
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
93 A parametrized description of a manifold or part of a manifold.
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
94 """
1581
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
95 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
96 mapping::MT
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
97 parameterspace::PST
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
98 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
99
1781
a73838c9ef94 Let Chart implement Base.ndims instead of domain_dim
Jonatan Werpers <jonatan@werpers.com>
parents: 1780
diff changeset
100 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
101 (c::Chart)(ξ) = c.mapping(ξ)
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
102 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
103
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
104 """
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
105 jacobian(c::Chart, ξ)
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
106
1581
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
107 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
108 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
109 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
110 chart itself.
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
111 ```julia
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
112 c = Chart(f, ps)
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
113 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
114 ```
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
115 or
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
116 ```julia
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
117 c = Chart(f, ps)
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
118 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
119 ```
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
120 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
121 """
f77c5309dd2b Rename ConcreteChart to Chart and remove the abstarct chart type
Jonatan Werpers <jonatan@werpers.com>
parents: 1580
diff changeset
122 jacobian(c::Chart, ξ) = jacobian(c.mapping, ξ)
1644
e213bd857f3f Add some todos and tbds
Jonatan Werpers <jonatan@werpers.com>
parents: 1625
diff changeset
123 # 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
124
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
125
1867
de4b4f2aee4f Add some tests for CartesianAtlas
Jonatan Werpers <jonatan@werpers.com>
parents: 1844
diff changeset
126 # TBD: Should Charts, parameterspaces, Atlases, have boundary names?
1644
e213bd857f3f Add some todos and tbds
Jonatan Werpers <jonatan@werpers.com>
parents: 1625
diff changeset
127
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
128 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
129 Atlas
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
130
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
131 A collection of charts and their connections.
1844
1987347752ef Add stub for connections(::Atlas)
Jonatan Werpers <jonatan@werpers.com>
parents: 1782
diff changeset
132 Should implement methods for `charts` and `connections`.
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
133 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
134 abstract type Atlas end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
135
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
136 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
137 charts(::Atlas)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
138
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
139 The colloction of charts in the atlas.
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
140 """
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
141 function charts end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
142
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
143 """
1844
1987347752ef Add stub for connections(::Atlas)
Jonatan Werpers <jonatan@werpers.com>
parents: 1782
diff changeset
144 connections(::Atlas)
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
145
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
146 TBD: What exactly should this return?
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
147 """
1844
1987347752ef Add stub for connections(::Atlas)
Jonatan Werpers <jonatan@werpers.com>
parents: 1782
diff changeset
148 function connections end
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
149
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
150 struct CartesianAtlas <: Atlas
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
151 charts::Matrix{Chart}
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
152 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
153
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
154 charts(a::CartesianAtlas) = a.charts
1782
614f731af685 Add stubs for connections method
Jonatan Werpers <jonatan@werpers.com>
parents: 1781
diff changeset
155 connections(a::CartesianAtlas) = nothing
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
156
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
157 struct UnstructuredAtlas <: Atlas
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
158 charts::Vector{Chart}
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
159 connections
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
160 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
161
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
162 charts(a::UnstructuredAtlas) = a.charts
1782
614f731af685 Add stubs for connections method
Jonatan Werpers <jonatan@werpers.com>
parents: 1781
diff changeset
163 connections(a::UnstructuredAtlas) = nothing
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
164
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
165
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
166 ###
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
167 # Geometry
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
168 ###
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
169
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
170 abstract type Curve end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
171 abstract type Surface end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
172
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
173
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
174 struct Line{PT} <: Curve
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
175 p::PT
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
176 tangent::PT
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
177 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
178
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
179 (c::Line)(s) = c.p + s*c.tangent
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
180
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
181
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
182 struct LineSegment{PT} <: Curve
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
183 a::PT
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
184 b::PT
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
185 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
186
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
187 (c::LineSegment)(s) = (1-s)*c.a + s*c.b
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
188
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
189
1625
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
190 function linesegments(ps...)
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
191 return [LineSegment(ps[i], ps[i+1]) for i ∈ 1:length(ps)-1]
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
192 end
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
193
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
194
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
195 function polygon_edges(ps...)
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
196 n = length(ps)
1869
20bd8887db0d Fix Aqua and JET fails
Jonatan Werpers <jonatan@werpers.com>
parents: 1867
diff changeset
197 return [LineSegment(ps[i], ps[mod1(i+1,n)]) for i ∈ eachindex(ps)]
1625
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
198 end
84c3b9d71218 Add linesegments function
Jonatan Werpers <jonatan@werpers.com>
parents: 1581
diff changeset
199
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
200 struct Circle{T,PT} <: Curve
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
201 c::PT
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
202 r::T
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
203 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
204
1869
20bd8887db0d Fix Aqua and JET fails
Jonatan Werpers <jonatan@werpers.com>
parents: 1867
diff changeset
205 function (C::Circle)(θ)
20bd8887db0d Fix Aqua and JET fails
Jonatan Werpers <jonatan@werpers.com>
parents: 1867
diff changeset
206 (;c, r) = C
20bd8887db0d Fix Aqua and JET fails
Jonatan Werpers <jonatan@werpers.com>
parents: 1867
diff changeset
207 c + r*@SVector[cos(θ), sin(θ)]
20bd8887db0d Fix Aqua and JET fails
Jonatan Werpers <jonatan@werpers.com>
parents: 1867
diff changeset
208 end
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
209
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
210 struct TransfiniteInterpolationSurface{T1,T2,T3,T4} <: Surface
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
211 c₁::T1
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
212 c₂::T2
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
213 c₃::T3
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
214 c₄::T4
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
215 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
216
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
217 function (s::TransfiniteInterpolationSurface)(u,v)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
218 c₁, c₂, c₃, c₄ = s.c₁, s.c₂, s.c₃, s.c₄
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
219 P₀₀ = c₁(0)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
220 P₁₀ = c₂(0)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
221 P₁₁ = c₃(0)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
222 P₀₁ = c₄(0)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
223 return (1-v)*c₁(u) + u*c₂(v) + v*c₃(1-u) + (1-u)*c₄(1-v) - (
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
224 (1-u)*(1-v)*P₀₀ + u*(1-v)*P₁₀ + u*v*P₁₁ + (1-u)*v*P₀₁
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
225 )
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
226 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
227
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
228 function (s::TransfiniteInterpolationSurface)(ξ̄::AbstractArray)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
229 s(ξ̄...)
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
230 end
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
231
1644
e213bd857f3f Add some todos and tbds
Jonatan Werpers <jonatan@werpers.com>
parents: 1625
diff changeset
232 # TODO: Implement jacobian() for the different mapping helpers
1558
81e97d3bec8c Start adding manifolds
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
233