470 lines
16 KiB
C++
470 lines
16 KiB
C++
// MFEM Euler Equation examples
|
|
//
|
|
// Compile with: make euler
|
|
//
|
|
// Sample runs:
|
|
//
|
|
// euler -p 1 -r 2 -o 1 -s 3
|
|
// euler -p 1 -r 1 -o 3 -s 4
|
|
// euler -p 1 -r 0 -o 5 -s 6
|
|
// euler -p 2 -r 1 -o 1 -s 3
|
|
// euler -p 2 -r 0 -o 3 -s 3
|
|
//
|
|
// Description: This example code solves the compressible Euler system of
|
|
// equations, a model nonlinear hyperbolic PDE, with a
|
|
// discontinuous Galerkin (DG) formulation.
|
|
//
|
|
// Specifically, it solves for an exact solution of the equations
|
|
// whereby a vortex is transported by a uniform flow. Since all
|
|
// boundaries are periodic here, the method's accuracy can be
|
|
// assessed by measuring the difference between the solution and
|
|
// the initial condition at a later time when the vortex returns
|
|
// to its initial location.
|
|
//
|
|
// Note that as the order of the spatial discretization increases,
|
|
// the timestep must become smaller. This example currently uses a
|
|
// simple estimate derived by Cockburn and Shu for the 1D RKDG
|
|
// method. An additional factor can be tuned by passing the --cfl
|
|
// (or -c shorter) flag.
|
|
//
|
|
// The example demonstrates user-defined bilinear and nonlinear
|
|
// form integrators for systems of equations that are defined with
|
|
// block vectors, and how these are used with an operator for
|
|
// explicit time integrators. In this case the system also
|
|
// involves an external approximate Riemann solver for the DG
|
|
// interface flux. It also demonstrates how to use GLVis for
|
|
// in-situ visualization of vector grid functions.
|
|
//
|
|
// We recommend viewing examples 9, 14 and 17 before viewing this
|
|
// example.
|
|
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
|
|
#include "mfem.hpp"
|
|
|
|
// Classes HyperbolicConservationLaws, NumericalFlux, and FaceIntegrator
|
|
// shared between the serial and parallel version of the example.
|
|
#include "hyperbolic_conservation_laws.hpp"
|
|
|
|
// Choice for the problem setup. See InitialCondition in ex18.hpp.
|
|
int problem;
|
|
void EulerInitialCondition(const Vector &x, Vector &y);
|
|
|
|
void UpdateSystem(FiniteElementSpace &fes, FiniteElementSpace &dfes,
|
|
FiniteElementSpace &vfes, DGHyperbolicConservationLaws &euler,
|
|
GridFunction &sol, ODESolver *ode_solver);
|
|
|
|
void EulerInitialCondition(const Vector &x, Vector &y);
|
|
|
|
int main(int argc, char *argv[]) {
|
|
Mpi::Init(argc, argv);
|
|
Hypre::Init();
|
|
// 1. Parse command-line options.
|
|
problem = 1;
|
|
const double specific_heat_ratio = 1.4;
|
|
const double gas_constant = 1.0;
|
|
|
|
const char *mesh_file = "../data/periodic-square-4x4.mesh";
|
|
int IntOrderOffset = 3;
|
|
int ser_ref_levels = 0;
|
|
int par_ref_levels = 4;
|
|
int order = 3;
|
|
int ode_solver_type = 4;
|
|
double t_final = 2.0;
|
|
double dt = -0.01;
|
|
double cfl = 0.3;
|
|
bool visualization = true;
|
|
int vis_steps = 50;
|
|
|
|
int precision = 8;
|
|
cout.precision(precision);
|
|
|
|
OptionsParser args(argc, argv);
|
|
args.AddOption(&mesh_file, "-m", "--mesh", "Mesh file to use.");
|
|
args.AddOption(&problem, "-p", "--problem",
|
|
"Problem setup to use. See options in velocity_function().");
|
|
args.AddOption(&ser_ref_levels, "-rs", "--refine-serial",
|
|
"Number of times to refine the mesh uniformly before parallel"
|
|
" partitioning, -1 for auto.");
|
|
args.AddOption(&par_ref_levels, "-rp", "--refine-parallel",
|
|
"Number of times to refine the mesh uniformly after parallel"
|
|
" partitioning.");
|
|
args.AddOption(&order, "-o", "--order",
|
|
"Order (degree) of the finite elements.");
|
|
args.AddOption(&ode_solver_type, "-s", "--ode-solver",
|
|
"ODE solver: 1 - Forward Euler,\n\t"
|
|
" 2 - RK2 SSP, 3 - RK3 SSP, 4 - RK4, 6 - RK6.");
|
|
args.AddOption(&t_final, "-tf", "--t-final", "Final time; start time is 0.");
|
|
args.AddOption(&dt, "-dt", "--time-step",
|
|
"Time step. Positive number skips CFL timestep calculation.");
|
|
args.AddOption(&cfl, "-c", "--cfl-number",
|
|
"CFL number for timestep calculation.");
|
|
args.AddOption(&visualization, "-vis", "--visualization", "-no-vis",
|
|
"--no-visualization",
|
|
"Enable or disable GLVis visualization.");
|
|
args.AddOption(&vis_steps, "-vs", "--visualization-steps",
|
|
"Visualize every n-th timestep.");
|
|
|
|
args.Parse();
|
|
if (!args.Good()) {
|
|
if (Mpi::Root()) {
|
|
args.PrintUsage(cout);
|
|
}
|
|
return 1;
|
|
}
|
|
if (Mpi::Root()) {
|
|
args.PrintOptions(cout);
|
|
}
|
|
|
|
// 2. Read the mesh from the given mesh file. This example requires a 2D
|
|
// periodic mesh, such as ../data/periodic-square.mesh.
|
|
Mesh mesh(mesh_file, 1, 1);
|
|
const int dim = mesh.Dimension();
|
|
const int num_equations = dim + 2;
|
|
|
|
// 3. Define the ODE solver used for time integration. Several explicit
|
|
// Runge-Kutta methods are available.
|
|
ODESolver *ode_solver = NULL;
|
|
switch (ode_solver_type) {
|
|
case 1:
|
|
ode_solver = new ForwardEulerSolver;
|
|
break;
|
|
case 2:
|
|
ode_solver = new RK2Solver(1.0);
|
|
break;
|
|
case 3:
|
|
ode_solver = new RK3SSPSolver;
|
|
break;
|
|
case 4:
|
|
ode_solver = new RK4Solver;
|
|
break;
|
|
case 6:
|
|
ode_solver = new RK6Solver;
|
|
break;
|
|
default:
|
|
cout << "Unknown ODE solver type: " << ode_solver_type << '\n';
|
|
return 3;
|
|
}
|
|
|
|
// 4. Refine the mesh to increase the resolution. In this example we do
|
|
// 'ref_levels' of uniform refinement, where 'ref_levels' is a
|
|
// command-line parameter.
|
|
mesh.EnsureNCMesh();
|
|
for (int lev = 0; lev < ser_ref_levels; lev++) {
|
|
mesh.UniformRefinement();
|
|
}
|
|
|
|
ParMesh pmesh(MPI_COMM_WORLD, mesh);
|
|
mesh.Clear();
|
|
for (int lev = 0; lev < par_ref_levels; lev++)
|
|
{
|
|
pmesh.UniformRefinement();
|
|
}
|
|
|
|
// 5. Define the discontinuous DG finite element space of the given
|
|
// polynomial order on the refined mesh.
|
|
DG_FECollection fec(order, dim);
|
|
// Finite element space for a scalar (thermodynamic quantity)
|
|
ParFiniteElementSpace fes(&pmesh, &fec);
|
|
// Finite element space for a mesh-dim vector quantity (momentum)
|
|
ParFiniteElementSpace dfes(&pmesh, &fec, dim, Ordering::byNODES);
|
|
// Finite element space for all variables together (total thermodynamic state)
|
|
ParFiniteElementSpace vfes(&pmesh, &fec, num_equations, Ordering::byNODES);
|
|
|
|
// This example depends on this ordering of the space.
|
|
MFEM_ASSERT(fes.GetOrdering() == Ordering::byNODES, "");
|
|
|
|
HYPRE_BigInt glob_size = vfes.GlobalTrueVSize();
|
|
if (Mpi::Root()) {
|
|
cout << "Number of unknowns: " << glob_size << endl;
|
|
}
|
|
// 6. Define the initial conditions, save the corresponding mesh and grid
|
|
// functions to a file. This can be opened with GLVis with the -gc option.
|
|
|
|
// The solution u has components {density, x-momentum, y-momentum, energy}.
|
|
// These are stored contiguously in the BlockVector u_block.
|
|
|
|
Array<int> offsets(num_equations + 1);
|
|
for (int k = 0; k <= num_equations; k++) {
|
|
offsets[k] = k * vfes.GetNDofs();
|
|
}
|
|
BlockVector u_block(offsets);
|
|
|
|
// Momentum grid function on dfes for visualization.
|
|
ParGridFunction mom(&dfes, u_block.GetData() + offsets[1]);
|
|
// Initialize the state.
|
|
VectorFunctionCoefficient u0(num_equations, EulerInitialCondition);
|
|
ParGridFunction sol(&vfes, u_block.GetData());
|
|
sol.ProjectCoefficient(u0);
|
|
|
|
// Output the initial solution.
|
|
{
|
|
ostringstream mesh_name;
|
|
mesh_name << "vortex-mesh." << setfill('0')
|
|
<< setw(6) << Mpi::WorldRank();
|
|
ofstream mesh_ofs(mesh_name.str().c_str());
|
|
mesh_ofs.precision(precision);
|
|
mesh_ofs << pmesh;
|
|
|
|
for (int k = 0; k < num_equations; k++)
|
|
{
|
|
ParGridFunction uk(&fes, u_block.GetBlock(k));
|
|
ostringstream sol_name;
|
|
sol_name << "vortex-" << k << "-init."
|
|
<< setfill('0') << setw(6) << Mpi::WorldRank();
|
|
ofstream sol_ofs(sol_name.str().c_str());
|
|
sol_ofs.precision(precision);
|
|
sol_ofs << uk;
|
|
}
|
|
}
|
|
|
|
// 7. Set up the nonlinear form corresponding to the DG discretization of the
|
|
// flux divergence, and assemble the corresponding mass matrix.
|
|
EulerElementFormIntegrator *eulerElementFormIntegrator =
|
|
new EulerElementFormIntegrator(dim, specific_heat_ratio, gas_constant,
|
|
IntOrderOffset);
|
|
|
|
EulerFaceFormIntegrator *eulerFaceFormIntegrator =
|
|
new EulerFaceFormIntegrator(new RusanovFlux(), dim, specific_heat_ratio,
|
|
gas_constant, IntOrderOffset);
|
|
ParNonlinearForm nonlinForm(&vfes);
|
|
|
|
// 8. Define the time-dependent evolution operator describing the ODE
|
|
// right-hand side, and perform time-integration (looping over the time
|
|
// iterations, ti, with a time-step dt).
|
|
DGHyperbolicConservationLaws euler(vfes, nonlinForm,
|
|
*eulerElementFormIntegrator,
|
|
*eulerFaceFormIntegrator, num_equations);
|
|
|
|
// Visualize the density
|
|
socketstream sout;
|
|
if (visualization) {
|
|
char vishost[] = "localhost";
|
|
int visport = 19916;
|
|
|
|
MPI_Barrier(pmesh.GetComm());
|
|
sout.open(vishost, visport);
|
|
if (!sout) {
|
|
if (Mpi::Root()) {
|
|
cout << "Unable to connect to GLVis server at " << vishost << ':'
|
|
<< visport << endl;
|
|
}
|
|
visualization = false;
|
|
if (Mpi::Root()) {
|
|
cout << "GLVis visualization disabled.\n";
|
|
}
|
|
} else {
|
|
sout << "parallel " << Mpi::WorldSize() << " " << Mpi::WorldRank()
|
|
<< "\n";
|
|
sout.precision(precision);
|
|
sout << "solution\n" << pmesh << mom;
|
|
sout << "pause\n";
|
|
sout << flush;
|
|
if (Mpi::Root()) {
|
|
cout << "GLVis visualization paused."
|
|
<< " Press space (in the GLVis window) to resume it.\n";
|
|
}
|
|
}
|
|
}
|
|
|
|
// Determine the minimum element size.
|
|
double hmin;
|
|
if (cfl > 0) {
|
|
double my_hmin = pmesh.GetElementSize(0, 1);
|
|
for (int i = 1; i < pmesh.GetNE(); i++) {
|
|
my_hmin = min(pmesh.GetElementSize(i, 1), my_hmin);
|
|
}
|
|
// Reduce to find the global minimum element size
|
|
MPI_Allreduce(&my_hmin, &hmin, 1, MPI_DOUBLE, MPI_MIN, pmesh.GetComm());
|
|
}
|
|
|
|
// Start the timer.
|
|
tic_toc.Clear();
|
|
tic_toc.Start();
|
|
|
|
double t = 0.0;
|
|
euler.SetTime(t);
|
|
ode_solver->Init(euler);
|
|
|
|
if (cfl > 0) {
|
|
// Find a safe dt, using a temporary vector. Calling Mult() computes the
|
|
// maximum char speed at all quadrature points on all faces.
|
|
Vector z(vfes.GetNDofs() * num_equations);
|
|
euler.Mult(sol, z);
|
|
double my_maxcharspeed = euler.getMaxCharSpeed();
|
|
double max_char_speed;
|
|
MPI_Allreduce(&my_maxcharspeed, &max_char_speed, 1, MPI_DOUBLE, MPI_MAX,
|
|
pmesh.GetComm());
|
|
if (cfl > 0) {
|
|
dt = cfl * hmin / max_char_speed / (2 * order + 1);
|
|
}
|
|
}
|
|
|
|
// Integrate in time.
|
|
bool done = false;
|
|
for (int ti = 0; !done;) {
|
|
double dt_real = min(dt, t_final - t);
|
|
|
|
ode_solver->Step(sol, t, dt_real);
|
|
double my_maxcharspeed = euler.getMaxCharSpeed();
|
|
double max_char_speed;
|
|
MPI_Allreduce(&my_maxcharspeed, &max_char_speed, 1, MPI_DOUBLE, MPI_MAX,
|
|
pmesh.GetComm());
|
|
if (cfl > 0) {
|
|
dt = cfl * hmin / max_char_speed / (2 * order + 1);
|
|
}
|
|
ti++;
|
|
|
|
done = (t >= t_final - 1e-8 * dt);
|
|
if (done || ti % vis_steps == 0) {
|
|
if (Mpi::Root()) {
|
|
cout << "time step: " << ti << ", time: " << t << endl;
|
|
}
|
|
if (visualization) {
|
|
MPI_Barrier(pmesh.GetComm());
|
|
sout << "parallel " << Mpi::WorldSize() << " " << Mpi::WorldRank()
|
|
<< "\n";
|
|
sout << "solution\n" << pmesh << mom << flush;
|
|
}
|
|
}
|
|
}
|
|
|
|
tic_toc.Stop();
|
|
if (Mpi::Root()) {
|
|
cout << " done, " << tic_toc.RealTime() << "s." << endl;
|
|
}
|
|
|
|
// 9. Save the final solution. This output c
|
|
for (int k = 0; k < num_equations; k++) {
|
|
ParGridFunction uk(&fes, u_block.GetBlock(k));
|
|
ostringstream sol_name;
|
|
sol_name << "vortex-" << k << "-final." << setfill('0') << setw(6)
|
|
<< Mpi::WorldRank();
|
|
ofstream sol_ofs(sol_name.str().c_str());
|
|
sol_ofs.precision(precision);
|
|
sol_ofs << uk;
|
|
}
|
|
|
|
// 12. Compute the L2 solution error summed for all components.
|
|
if (t_final == 2.0) {
|
|
const double error = sol.ComputeLpError(2, u0);
|
|
if (Mpi::Root()) {
|
|
cout << "Solution error: " << error << endl;
|
|
}
|
|
}
|
|
|
|
// Free the used memory.
|
|
delete ode_solver;
|
|
|
|
return 0;
|
|
}
|
|
|
|
void UpdateSystem(FiniteElementSpace &fes, FiniteElementSpace &dfes,
|
|
FiniteElementSpace &vfes, DGHyperbolicConservationLaws &euler,
|
|
GridFunction &sol, ODESolver *ode_solver) {
|
|
fes.Update();
|
|
dfes.Update();
|
|
vfes.Update();
|
|
sol.Update();
|
|
euler.Update();
|
|
ode_solver->Init(euler);
|
|
fes.UpdatesFinished();
|
|
dfes.UpdatesFinished();
|
|
vfes.UpdatesFinished();
|
|
}
|
|
|
|
// Initial condition
|
|
void EulerInitialCondition(const Vector &x, Vector &y) {
|
|
if (problem < 3) {
|
|
MFEM_ASSERT(x.Size() == 2, "");
|
|
const double specific_heat_ratio = 1.4;
|
|
const double gas_constant = 1.0;
|
|
|
|
double radius = 0, Minf = 0, beta = 0;
|
|
if (problem == 1) {
|
|
// "Fast vortex"
|
|
radius = 0.2;
|
|
Minf = 0.5;
|
|
beta = 1. / 5.;
|
|
} else if (problem == 2) {
|
|
// "Slow vortex"
|
|
radius = 0.2;
|
|
Minf = 0.05;
|
|
beta = 1. / 50.;
|
|
} else {
|
|
mfem_error(
|
|
"Cannot recognize problem."
|
|
"Options are: 1 - fast vortex, 2 - slow vortex");
|
|
}
|
|
|
|
const double xc = 0.0, yc = 0.0;
|
|
|
|
// Nice units
|
|
const double vel_inf = 1.;
|
|
const double den_inf = 1.;
|
|
|
|
// Derive remainder of background state from this and Minf
|
|
const double pres_inf =
|
|
(den_inf / specific_heat_ratio) * (vel_inf / Minf) * (vel_inf / Minf);
|
|
const double temp_inf = pres_inf / (den_inf * gas_constant);
|
|
|
|
double r2rad = 0.0;
|
|
r2rad += (x(0) - xc) * (x(0) - xc);
|
|
r2rad += (x(1) - yc) * (x(1) - yc);
|
|
r2rad /= (radius * radius);
|
|
|
|
const double shrinv1 = 1.0 / (specific_heat_ratio - 1.);
|
|
|
|
const double velX =
|
|
vel_inf * (1 - beta * (x(1) - yc) / radius * exp(-0.5 * r2rad));
|
|
const double velY =
|
|
vel_inf * beta * (x(0) - xc) / radius * exp(-0.5 * r2rad);
|
|
const double vel2 = velX * velX + velY * velY;
|
|
|
|
const double specific_heat = gas_constant * specific_heat_ratio * shrinv1;
|
|
const double temp = temp_inf - 0.5 * (vel_inf * beta) * (vel_inf * beta) /
|
|
specific_heat * exp(-r2rad);
|
|
|
|
const double den = den_inf * pow(temp / temp_inf, shrinv1);
|
|
const double pres = den * gas_constant * temp;
|
|
const double energy = shrinv1 * pres / den + 0.5 * vel2;
|
|
|
|
y(0) = den;
|
|
y(1) = den * velX;
|
|
y(2) = den * velY;
|
|
y(3) = den * energy;
|
|
} else if (problem == 3) {
|
|
MFEM_ASSERT(x.Size() == 2, "");
|
|
// std::cout << "2D Accuracy Test." << std::endl;
|
|
// std::cout << "domain = (-1, 1) x (-1, 1)" << std::endl;
|
|
const double density = 1.0 + 0.2 * __sinpi(x(0) + x(1));
|
|
const double velocity_x = 0.7;
|
|
const double velocity_y = 0.3;
|
|
const double pressure = 1.0;
|
|
const double energy =
|
|
pressure / (1.4 - 1.0) +
|
|
density * 0.5 * (velocity_x * velocity_x + velocity_y * velocity_y);
|
|
|
|
y(0) = density;
|
|
y(1) = density * velocity_x;
|
|
y(2) = density * velocity_y;
|
|
y(3) = energy;
|
|
} else if (problem == 4) {
|
|
MFEM_ASSERT(x.Size() == 1, "");
|
|
// std::cout << "2D Accuracy Test." << std::endl;
|
|
// std::cout << "domain = (-1, 1) x (-1, 1)" << std::endl;
|
|
const double density = 1.0 + 0.2 * __sinpi(x(0));
|
|
const double velocity_x = 1.0;
|
|
const double pressure = 1.0;
|
|
const double energy =
|
|
pressure / (1.4 - 1.0) + density * 0.5 * (velocity_x * velocity_x);
|
|
|
|
y(0) = density;
|
|
y(1) = density * velocity_x;
|
|
y(2) = energy;
|
|
} else {
|
|
mfem_error("Invalid problem.");
|
|
}
|
|
} |