Functions
int	main (int argc, char **argv)
Function Documentation

◆ main()

int main	(	int	argc,
		char **	argv
	)
Definition at line 61 of file fem_system_ex1.C.
 {
   // Initialize libMesh.
   LibMeshInit init (argc, argv);
 
   // This example requires a linear solver package.
   libmesh_example_requires(libMesh::default_solver_package() != INVALID_SOLVER_PACKAGE,
                            "--enable-petsc, --enable-trilinos, or --enable-eigen");
 
   // This example fails without at least double precision FP
 #ifdef LIBMESH_DEFAULT_SINGLE_PRECISION
   libmesh_example_requires(false, "--disable-singleprecision");
 #endif
 
 #ifndef LIBMESH_ENABLE_AMR
   libmesh_example_requires(false, "--enable-amr");
 #else
 
   // We use Dirichlet boundary conditions here
 #ifndef LIBMESH_ENABLE_DIRICHLET
   libmesh_example_requires(false, "--enable-dirichlet");
 #endif
 
   // Trilinos and Eigen solvers NaN by default here.
   // We'll skip this example for now.
   libmesh_example_requires(libMesh::default_solver_package() != TRILINOS_SOLVERS, "--enable-petsc");
 
   // Parse the input file
   GetPot infile("fem_system_ex1.in");
 
   // Override input file arguments from the command line
   infile.parse_command_line(argc, argv);
 
   // Read in parameters from the input file
   const Real global_tolerance          = infile("global_tolerance", 0.);
   const unsigned int nelem_target      = infile("n_elements", 400);
   const bool transient                 = infile("transient", true);
   const Real deltat                    = infile("deltat", 0.005);
   unsigned int n_timesteps             = infile("n_timesteps", 20);
   const unsigned int coarsegridsize    = infile("coarsegridsize", 1);
   const unsigned int coarserefinements = infile("coarserefinements", 0);
   const unsigned int max_adaptivesteps = infile("max_adaptivesteps", 10);
   const unsigned int dim               = infile("dimension", 2);
   const std::string slvr_type          = infile("solver_type", "newton");
   const std::string mesh_type          = infile("mesh_type"  , "replicated");
   const bool constrain_in_solver       = infile("constrain_in_solver", true);
 
   // More desperate debugging options
   const bool print_solutions           = infile("print_solutions", false);
   const bool print_residuals           = infile("print_residuals", false);
   const bool print_jacobians           = infile("print_jacobians", false);
 
 #ifdef LIBMESH_HAVE_EXODUS_API
   const unsigned int write_interval    = infile("write_interval", 5);
 #endif
 
   // Skip higher-dimensional examples on a lower-dimensional libMesh build
   libmesh_example_requires(dim <= LIBMESH_DIM, "2D/3D support");
 
   // We have only defined 2 and 3 dimensional problems
   libmesh_assert (dim == 2 || dim == 3);
 
   // Create a mesh, with dimension to be overridden later, distributed
   // across the default MPI communicator.
   std::shared_ptr<UnstructuredMesh> mesh;
 
   if (mesh_type == "distributed")
     mesh = std::make_shared<DistributedMesh>(init.comm());
   else if (mesh_type == "replicated")
     mesh = std::make_shared<ReplicatedMesh>(init.comm());
   else
     libmesh_error_msg("Error: specified mesh_type not understood");
 
   // And an object to refine it
   MeshRefinement mesh_refinement(*mesh);
   mesh_refinement.coarsen_by_parents() = true;
   mesh_refinement.absolute_global_tolerance() = global_tolerance;
   mesh_refinement.nelem_target() = nelem_target;
   mesh_refinement.refine_fraction() = 0.3;
   mesh_refinement.coarsen_fraction() = 0.3;
   mesh_refinement.coarsen_threshold() = 0.1;
 
   // Use the MeshTools::Generation mesh generator to create a uniform
   // grid on the square [-1,1]^D.  We instruct the mesh generator
   // to build a mesh of 8x8 Quad9 elements in 2D, or Hex27
   // elements in 3D.  Building these higher-order elements allows
   // us to use higher-order approximation, as in example 3.
   if (dim == 2)
     MeshTools::Generation::build_square (*mesh,
                                          coarsegridsize,
                                          coarsegridsize,
                                          0., 1.,
                                          0., 1.,
                                          QUAD9);
   else if (dim == 3)
     MeshTools::Generation::build_cube (*mesh,
                                        coarsegridsize,
                                        coarsegridsize,
                                        coarsegridsize,
                                        0., 1.,
                                        0., 1.,
                                        0., 1.,
                                        HEX27);
 
   mesh_refinement.uniformly_refine(coarserefinements);
 
   // Print information about the mesh to the screen.
   mesh->print_info();
 
   // Create an equation systems object.
   EquationSystems equation_systems (*mesh);
 
   // Declare the system "Navier-Stokes" and its variables.
   NavierSystem & system =
     equation_systems.add_system<NavierSystem> ("Navier-Stokes");
 
   // Debug it if requested
   system.print_solutions = print_solutions;
   system.print_solution_norms = print_solutions;
   system.print_residuals = print_residuals;
   system.print_residual_norms = print_residuals;
   system.print_jacobians = print_jacobians;
   system.print_jacobian_norms = print_jacobians;
 
   // Solve this as a time-dependent or steady system
   if (transient)
     system.time_solver = std::make_unique<EulerSolver>(system);
   else
     {
       system.time_solver = std::make_unique<SteadySolver>(system);
       libmesh_assert_equal_to (n_timesteps, 1);
     }
 
   // Initialize the system
   equation_systems.init ();
 
   // Set the time stepping options
   system.deltat = deltat;
 
   // And the nonlinear solver options
   if (slvr_type == "newton")
     system.time_solver->diff_solver() = std::make_unique<NewtonSolver>(system);
   else if (slvr_type == "petscdiff")
 #if defined(LIBMESH_HAVE_PETSC) && defined(LIBMESH_HAVE_METAPHYSICL)
     system.time_solver->diff_solver() = std::make_unique<PetscDiffSolver>(system);
 #else
   libmesh_example_requires(false, "--enable-petsc --enable-metaphysicl-required");
 #endif
   else
     libmesh_error_msg("Error: specified solver_type not understood");
 
   DiffSolver & solver = *(system.time_solver->diff_solver().get());
   solver.init();
 
   solver.quiet = infile("solver_quiet", true);
   solver.verbose = !solver.quiet;
   solver.max_nonlinear_iterations =
     infile("max_nonlinear_iterations", 15);
   solver.relative_step_tolerance =
     infile("relative_step_tolerance", 1.e-3);
   solver.relative_residual_tolerance =
     infile("relative_residual_tolerance", 0.0);
   solver.absolute_residual_tolerance =
     infile("absolute_residual_tolerance", 0.0);
 
   system.set_constrain_in_solver(constrain_in_solver);
 
   // And the linear solver options
   solver.max_linear_iterations =
     infile("max_linear_iterations", 50000);
   solver.initial_linear_tolerance =
     infile("initial_linear_tolerance", 1.e-3);
 
   // Print information about the system to the screen.
   equation_systems.print_info();
 
   // Now we begin the timestep loop to compute the time-accurate
   // solution of the equations.
   for (unsigned int t_step=0; t_step != n_timesteps; ++t_step)
     {
       // A pretty update message
       libMesh::out << "\n\nSolving time step "
                    << t_step
                    << ", time = "
                    << system.time
                    << std::endl;
 
       // Adaptively solve the timestep
       unsigned int a_step = 0;
       for (; a_step != max_adaptivesteps; ++a_step)
         {
           system.solve();
 
           system.postprocess();
 
           ErrorVector error;
 
           std::unique_ptr<ErrorEstimator> error_estimator;
 
           // To solve to a tolerance in this problem we
           // need a better estimator than Kelly
           if (global_tolerance != 0.)
             {
               // We can't adapt to both a tolerance and a mesh
               // size at once
               libmesh_assert_equal_to (nelem_target, 0);
 
               error_estimator = std::make_unique<UniformRefinementEstimator>();
 
               // The lid-driven cavity problem isn't in H1, so
               // lets estimate L2 error
               error_estimator->error_norm = L2;
             }
           else
             {
               // If we aren't adapting to a tolerance we need a
               // target mesh size
               libmesh_assert_greater (nelem_target, 0);
 
               // Kelly is a lousy estimator to use for a problem
               // not in H1 - if we were doing more than a few
               // timesteps we'd need to turn off or limit the
               // maximum level of our adaptivity eventually
               error_estimator = std::make_unique<KellyErrorEstimator>();
             }
 
           // Calculate error based on u and v (and w?) but not p
           std::vector<Real> weights(2,1.0);  // u, v
           if (dim == 3)
             weights.push_back(1.0);          // w
           weights.push_back(0.0);            // p
           // Keep the same default norm type.
           std::vector<FEMNormType>
             norms(1, error_estimator->error_norm.type(0));
           error_estimator->error_norm = SystemNorm(norms, weights);
 
           error_estimator->estimate_error(system, error);
 
           // Print out status at each adaptive step.
           Real global_error = error.l2_norm();
           libMesh::out << "Adaptive step "
                        << a_step
                        << ": "
                        << std::endl;
 
           if (global_tolerance != 0.)
             libMesh::out << "Global_error = "
                          << global_error
                          << std::endl;
 
           if (global_tolerance != 0.)
             libMesh::out << "Worst element error = "
                          << error.maximum()
                          << ", mean = "
                          << error.mean()
                          << std::endl;
 
           if (global_tolerance != 0.)
             {
               // If we've reached our desired tolerance, we
               // don't need any more adaptive steps
               if (global_error < global_tolerance)
                 break;
               mesh_refinement.flag_elements_by_error_tolerance(error);
             }
           else
             {
               // If flag_elements_by_nelem_target returns true, this
               // should be our last adaptive step.
               if (mesh_refinement.flag_elements_by_nelem_target(error))
                 {
                   mesh_refinement.refine_and_coarsen_elements();
                   equation_systems.reinit();
                   a_step = max_adaptivesteps;
                   break;
                 }
             }
 
           // Carry out the adaptive mesh refinement/coarsening
           mesh_refinement.refine_and_coarsen_elements();
           equation_systems.reinit();
 
           libMesh::out << "Refined mesh to "
                        << mesh->n_active_elem()
                        << " active elements and "
                        << equation_systems.n_active_dofs()
                        << " active dofs."
                        << std::endl;
         }
       // Do one last solve if necessary
       if (a_step == max_adaptivesteps)
         {
           system.solve();
 
           system.postprocess();
         }
 
       // Advance to the next timestep in a transient problem
       system.time_solver->advance_timestep();
 
 #ifdef LIBMESH_HAVE_EXODUS_API
       // Write out this timestep if we're requested to
       if ((t_step+1)%write_interval == 0)
         {
           std::ostringstream file_name;
 
           // We write the file in the ExodusII format.
           file_name << "out_"
                     << std::setw(3)
                     << std::setfill('0')
                     << std::right
                     << t_step+1
                     << ".e";
 
           ExodusII_IO(*mesh).write_timestep(file_name.str(),
                                            equation_systems,
                                            1, // This number indicates how many time steps
                                               // are being written to the file
                                            system.time);
         }
 #endif // #ifdef LIBMESH_HAVE_EXODUS_API
     }
 #endif // #ifndef LIBMESH_ENABLE_AMR
 
   // All done.
   return 0;
 }
Functions

Function Documentation

◆ main()
