add hypsm

Author: MarcoArtiano

examples/tree_1d_dgsem/elixir_hyperbolic_sainte_marie_ec.jl

@@ -0,0 +1,73 @@
+using OrdinaryDiffEqSSPRK, OrdinaryDiffEqLowStorageRK
+using Trixi
+using TrixiShallowWater
+
+###############################################################################
+# semidiscretization of the shallow water equations with a discontinuous
+# bottom topography function for a fully wet configuration
+
+equations = HyperbolicSainteMarieEquations1D(gravity=1.0, b0=0.1)
+
+function initial_condition_periodic(x, t, equations::HyperbolicSainteMarieEquations1D)
+    h = 1 + exp(sinpi(2 * x[1]))
+    v = 1
+    w = 1
+    p = 10
+    b = 0.1*exp(sinpi(2 * x[1]))
+    H = h+b
+    return prim2cons(SVector(H, v, w, p, b), equations)
+end
+
+initial_condition = initial_condition_periodic
+###############################################################################
+# Get the DG approximation space
+alpha_coefficients = (1/2, 1.0, 2/3)
+volume_flux = (flux_conservative_artiano_ranocha(alpha_coefficients...), flux_nonconservative_artiano_ranocha(alpha_coefficients...))
+surface_flux = volume_flux
+polydeg = 3
+    solver = DGSEM(polydeg=polydeg, surface_flux=surface_flux,
+        volume_integral=VolumeIntegralFluxDifferencing(volume_flux))
+    boundary_condition = BoundaryConditionDirichlet(initial_condition)
+    ###############################################################################
+    # Get the TreeMesh and setup a periodic mesh
+    coordinates_min = 0.0
+    coordinates_max = 1.0
+    mesh = TreeMesh(coordinates_min, coordinates_max,
+        initial_refinement_level=7,
+        n_cells_max=10_000,
+        periodicity=true)
+
+    # Create the semi discretization object
+    semi = SemidiscretizationHyperbolic(mesh, equations, initial_condition, solver, source_terms = source_term_hyperbolic_sainte_marie,
+                                    boundary_conditions = boundary_condition_periodic)
+
+    ###############################################################################
+    # ODE solver
+    tspan = (0.0, 0.2)
+    ode = semidiscretize(semi, tspan)
+
+    ###############################################################################
+    # Callbacks
+
+    summary_callback = SummaryCallback()
+
+    analysis_interval = 1000
+    analysis_callback = AnalysisCallback(semi, interval=analysis_interval, extra_analysis_integrals = (entropy, ))
+
+    alive_callback = AliveCallback(analysis_interval=analysis_interval)
+
+    stepsize_callback = StepsizeCallback(cfl=0.1)
+
+    callbacks = CallbackSet(
+        summary_callback,
+        analysis_callback,
+        alive_callback
+    )
+
+    ###############################################################################
+    # run the simulation
+
+    sol = solve(ode,
+        SSPRK43(thread=Trixi.True());
+        dt=1.0, # solve needs some value here but it will be overwritten by the stepsize_callback
+        ode_default_options()..., callback=callbacks);

src/TrixiShallowWater.jl

@@ -28,7 +28,8 @@ export ShallowWaterEquations1D, ShallowWaterEquations2D,
        ShallowWaterTwoLayerEquations1D, ShallowWaterTwoLayerEquations2D,
        ShallowWaterMultiLayerEquations1D, ShallowWaterMultiLayerEquations2D,
        ShallowWaterEquationsQuasi1D,
-       ShallowWaterMomentEquations1D, ShallowWaterLinearizedMomentEquations1D
+       ShallowWaterMomentEquations1D, ShallowWaterLinearizedMomentEquations1D,
+       HyperbolicSainteMarieEquations1D
 
 export hydrostatic_reconstruction_chen_noelle, flux_nonconservative_chen_noelle,
        min_max_speed_chen_noelle, flux_hll_chen_noelle,
@@ -39,7 +40,8 @@ export hydrostatic_reconstruction_chen_noelle, flux_nonconservative_chen_noelle,
        flux_nonconservative_audusse_etal, hydrostatic_reconstruction_audusse_etal,
        FluxHydrostaticReconstruction,
        source_term_manning_friction, source_term_newtonian_slip_friction,
-       source_term_bottom_friction
+       source_term_bottom_friction, source_term_hyperbolic_sainte_marie,
+       flux_conservative_artiano_ranocha, flux_nonconservative_artiano_ranocha
 
 export ManningFriction, MeyerPeterMueller, GrassModel, ShieldsStressModel,
        dissipation_roe, water_sediment_height

src/equations/equations.jl

@@ -88,6 +88,8 @@ Retrieve the number of moments from an equation instance of the `AbstractMomentE
     NMOMENTS
 end
 
+include("hyperbolic_sainte_marie_1d.jl")
+
 # TODO: Add suitable default thresholds for Float32
 # Provide default thresholds dependent on number format (Currently default thresholds are only provided
 # for Float64)