@@ -15,39 +15,39 @@ println("Loading Chmy-GMG tools")
15
15
16
16
Creates a GMG `CartGrid` data structure from a `Chmy` grid object
17
17
"""
18
- function create_CartGrid (grid:: StructuredGrid ; ylevel= 0.0 )
18
+ function create_CartGrid (grid:: StructuredGrid ; ylevel = 0.0 )
19
19
20
- coord1D = Vector .(coords (grid, Vertex ()))
20
+ coord1D = Vector .(coords (grid, Vertex ()))
21
21
coord1D_cen = Vector .(coords (grid, Center ()))
22
- N = length .(coord1D)
23
- L = extent (grid, Vertex ())
24
- X₁ = origin (grid, Vertex ())
25
- Δ = spacing (grid)
26
- ConstantΔ = false ;
22
+ N = length .(coord1D)
23
+ L = extent (grid, Vertex ())
24
+ X₁ = origin (grid, Vertex ())
25
+ Δ = spacing (grid)
26
+ ConstantΔ = false
27
27
if isa (grid, UniformGrid)
28
28
ConstantΔ = true
29
29
end
30
- if ndims (grid)== 2
30
+ if ndims (grid) == 2
31
31
# we need a special treatment of this, as all GMG routines work with 3D coordinates
32
- X₁ = (X₁[1 ], ylevel, X₁[2 ])
33
- L = (L[1 ], 0.0 , L[2 ])
34
- Δ = (Δ[1 ], 0.0 , Δ[2 ])
35
- N = (N[1 ],1 , N[2 ])
32
+ X₁ = (X₁[1 ], ylevel, X₁[2 ])
33
+ L = (L[1 ], 0.0 , L[2 ])
34
+ Δ = (Δ[1 ], 0.0 , Δ[2 ])
35
+ N = (N[1 ], 1 , N[2 ])
36
36
coord1D = (coord1D[1 ], [0.0 ], coord1D[2 ])
37
37
coord1D_cen = (coord1D_cen[1 ], [0.0 ], coord1D_cen[2 ])
38
38
end
39
- Xₙ = X₁ .+ L
40
-
39
+ Xₙ = X₁ .+ L
41
40
42
- return CartGrid (ConstantΔ,N,Δ,L,X₁,Xₙ,coord1D, coord1D_cen)
41
+
42
+ return CartGrid (ConstantΔ, N, Δ, L, X₁, Xₙ, coord1D, coord1D_cen)
43
43
end
44
44
45
45
# all functions to be ported
46
46
function_names = (:add_box! , :add_sphere! , :add_ellipsoid! , :add_cylinder! , :add_layer! , :add_polygon! , :add_slab! , :add_stripes! , :add_volcano! )
47
47
48
48
for fn in function_names
49
49
50
- @eval begin
50
+ @eval begin
51
51
"""
52
52
$($ fn) ( Phase::Field,
53
53
Temp::Field,
@@ -56,22 +56,24 @@ for fn in function_names
56
56
57
57
Sets `$($ fn) ` function for `Chmy` fields and grids.
58
58
"""
59
- function $fn ( Phase:: Field ,
60
- Temp:: Field ,
61
- Grid:: StructuredGrid ; # required input
62
- kwargs... )
59
+ function $fn (
60
+ Phase:: Field ,
61
+ Temp:: Field ,
62
+ Grid:: StructuredGrid ; # required input
63
+ kwargs...
64
+ )
63
65
64
66
CartGrid = create_CartGrid (Grid)
65
67
66
68
cell = false
67
- if all (location (Phase).== Center ())
69
+ if all (location (Phase) .== Center ())
68
70
cell = true
69
71
end
70
-
71
- return ($ fn)(Phase, Temp, CartGrid; cell= cell, kwargs... )
72
+
73
+ return ($ fn)(Phase, Temp, CartGrid; cell = cell, kwargs... )
72
74
end
73
75
end
74
-
76
+
75
77
end
76
78
77
79
@@ -80,28 +82,30 @@ function_names = (:above_surface, :below_surface)
80
82
81
83
for fn in function_names
82
84
83
- @eval begin
85
+ @eval begin
84
86
"""
85
87
$($ fn) ( Grid::StructuredGrid, field::Field, DataSurface_Cart::CartData; kwargs...)
86
88
87
89
Sets `$($ fn) ` function for `Chmy` grids and the field `field` which can be either on vertices or centers
88
90
"""
89
- function $fn ( Grid:: StructuredGrid ,
90
- field:: Field ,
91
- DataSurface_Cart:: CartData ;
92
- kwargs... )
91
+ function $fn (
92
+ Grid:: StructuredGrid ,
93
+ field:: Field ,
94
+ DataSurface_Cart:: CartData ;
95
+ kwargs...
96
+ )
93
97
94
98
CartGrid = create_CartGrid (Grid)
95
-
99
+
96
100
cell = false
97
- if all (location (field).== Center ())
101
+ if all (location (field) .== Center ())
98
102
cell = true
99
103
end
100
104
101
- return ($ fn)(CartGrid, DataSurface_Cart; cell= cell, kwargs... )
105
+ return ($ fn)(CartGrid, DataSurface_Cart; cell = cell, kwargs... )
102
106
end
103
107
end
104
-
108
+
105
109
end
106
110
107
111
0 commit comments