Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
trixi-framework
GitHub Repository: trixi-framework/Trixi.jl
Path: blob/main/examples/structured_3d_dgsem/elixir_advection_nonperiodic_curved.jl
2055 views
1
using OrdinaryDiffEqLowStorageRK
2
using Trixi
3
4
###############################################################################
5
# semidiscretization of the linear advection equation
6
7
advection_velocity = (0.2, -0.7, 0.5)
8
equations = LinearScalarAdvectionEquation3D(advection_velocity)
9
10
initial_condition = initial_condition_convergence_test
11
boundary_conditions = BoundaryConditionDirichlet(initial_condition)
12
13
solver = DGSEM(polydeg = 3, surface_flux = flux_lax_friedrichs)
14
15
# Mapping as described in https://arxiv.org/abs/2012.12040 but with less warping.
16
function mapping(xi, eta, zeta)
17
# Don't transform input variables between -1 and 1 onto [0,3] to obtain curved boundaries
18
# xi = 1.5 * xi_ + 1.5
19
# eta = 1.5 * eta_ + 1.5
20
# zeta = 1.5 * zeta_ + 1.5
21
22
y = eta +
23
1 / 6 * (cos(1.5 * pi * (2 * xi - 3) / 3) *
24
cos(0.5 * pi * (2 * eta - 3) / 3) *
25
cos(0.5 * pi * (2 * zeta - 3) / 3))
26
27
x = xi +
28
1 / 6 * (cos(0.5 * pi * (2 * xi - 3) / 3) *
29
cos(2 * pi * (2 * y - 3) / 3) *
30
cos(0.5 * pi * (2 * zeta - 3) / 3))
31
32
z = zeta +
33
1 / 6 * (cos(0.5 * pi * (2 * x - 3) / 3) *
34
cos(pi * (2 * y - 3) / 3) *
35
cos(0.5 * pi * (2 * zeta - 3) / 3))
36
37
return SVector(x, y, z)
38
end
39
40
cells_per_dimension = (8, 8, 8)
41
mesh = StructuredMesh(cells_per_dimension, mapping, periodicity = false)
42
43
semi = SemidiscretizationHyperbolic(mesh, equations, initial_condition, solver,
44
boundary_conditions = boundary_conditions)
45
46
###############################################################################
47
# ODE solvers, callbacks etc.
48
49
tspan = (0.0, 1.0)
50
ode = semidiscretize(semi, tspan)
51
52
summary_callback = SummaryCallback()
53
54
analysis_interval = 100
55
analysis_callback = AnalysisCallback(semi, interval = analysis_interval,
56
extra_analysis_integrals = (entropy,))
57
58
alive_callback = AliveCallback(analysis_interval = analysis_interval)
59
60
save_restart = SaveRestartCallback(interval = 100,
61
save_final_restart = true)
62
63
save_solution = SaveSolutionCallback(interval = 100,
64
save_initial_solution = true,
65
save_final_solution = true,
66
solution_variables = cons2prim)
67
68
stepsize_callback = StepsizeCallback(cfl = 1.2)
69
70
callbacks = CallbackSet(summary_callback,
71
analysis_callback,
72
alive_callback,
73
save_restart,
74
save_solution,
75
stepsize_callback)
76
77
###############################################################################
78
# run the simulation
79
80
sol = solve(ode, CarpenterKennedy2N54(williamson_condition = false);
81
dt = 1.0, # solve needs some value here but it will be overwritten by the stepsize_callback
82
ode_default_options()..., callback = callbacks);
83
84