mpi/distribution/hp_adaptive_driven_cavity/hp_adaptive_driven_cavity.cc File Reference

#include "generic.h"
#include "navier_stokes.h"
#include "meshes/simple_rectangular_quadmesh.h"

Classes
class	SimpleRefineableRectangularQuadMesh< ELEMENT >
class	PRefineableDrivenCavityProblem< ELEMENT >

Namespaces
	Global_Physical_Variables
	Global variables.

Functions
int	main (int argc, char **argv)
	Driver for RefineableDrivenCavity test problem. More...

Function Documentation

main()

int main	(	int argc	,
		char **	argv
	)

Driver for RefineableDrivenCavity test problem.

{
 
#ifdef OOMPH_HAS_MPI
 
 // Initialise MPI
 MPI_Helpers::init(argc,argv);
 
#endif
 
 // Store command line arguments
 CommandLineArgs::setup(argc,argv);
 
 // Set output directory
 DocInfo doc_info;
 doc_info.set_directory("RESLT");
 
 
 
 // Solve problem with hp-refineable Crouzeix Raviart elements
 //-----------------------------------------------------------
 // Build problem
 PRefineableDrivenCavityProblem<PRefineableQCrouzeixRaviartElement<2> > problem;
 //Are there command-line arguments?
 if(CommandLineArgs::Argc==1)
  {
 
#ifdef OOMPH_HAS_MPI
 
   // Provide storage for each element's partition number
   const unsigned n_element=problem.mesh_pt()->nelement();
   Vector<unsigned> out_element_partition(n_element);
   
   // Distribute the problem
   bool report_stats=true;
   out_element_partition=problem.distribute(report_stats);
   
   // Write partition to disk
   std::ofstream output_file;
   char filename[100];
   sprintf(filename,"out_hp_adaptive_cavity_partition.dat");
   output_file.open(filename);
   for (unsigned e=0;e<n_element;e++)
    {
     output_file << out_element_partition[e] << std::endl;
    }
   
   // Check halo schemes (optional)
   problem.check_halo_schemes(doc_info);
   
#endif
   
   oomph_info << "\n\n\nProblem self-test ";
   if (problem.self_test()==0) 
    {
     oomph_info << "passed: Problem can be solved." << std::endl;
    }
   else 
    {
     throw OomphLibError("Self test failed",
                         OOMPH_CURRENT_FUNCTION,
                         OOMPH_EXCEPTION_LOCATION);
    }
   
   problem.p_adapt();
   problem.newton_solve();
   doc_info.number()=1;
   problem.doc_solution(doc_info);
   
   problem.p_adapt();
   problem.newton_solve();
   doc_info.number()=3;
   problem.doc_solution(doc_info);
   
   problem.adapt();
   problem.newton_solve();
   doc_info.number()=4;
   problem.doc_solution(doc_info);
 
   problem.p_adapt();
   problem.newton_solve();
   doc_info.number()=5;
   problem.doc_solution(doc_info);
 
   problem.adapt();
   problem.newton_solve();
   doc_info.number()=6;
   problem.doc_solution(doc_info);
 
   problem.p_adapt();
   problem.newton_solve();
   doc_info.number()=7;
   problem.doc_solution(doc_info);
 
   problem.adapt();
   problem.newton_solve();
   doc_info.number()=8;
   problem.doc_solution(doc_info);
 
  } // end of no command-line arguments
 else // Validation run - read in partition from file
  {
 
#ifdef OOMPH_HAS_MPI
 
    DocInfo mesh_doc_info;
    mesh_doc_info.set_directory("RESLT_MESH");
    mesh_doc_info.number()=0;
    std::ifstream input_file;
    char filename[100];
 
    // Get the partition to be used from file
    const unsigned n_element=problem.mesh_pt()->nelement();
    Vector<unsigned> element_partition(n_element);
    sprintf(filename,"hp_adaptive_cavity_partition.dat");
    input_file.open(filename);
    std::string input_string;
    for (unsigned e=0;e<n_element;e++)
     {
      getline(input_file,input_string,'\n');
      element_partition[e]=atoi(input_string.c_str());
     }
 
    // Now perform the distribution
    bool report_stats=true;
    problem.distribute(element_partition,mesh_doc_info,report_stats);
 
#endif
   
   problem.p_adapt();
   problem.newton_solve();
   doc_info.number()=1;
   problem.doc_solution(doc_info);
   
   problem.p_adapt();
   problem.newton_solve();
   doc_info.number()=3;
   problem.doc_solution(doc_info);
   
   problem.adapt();
   problem.newton_solve();
   doc_info.number()=4;
   problem.doc_solution(doc_info);
 
   problem.p_adapt();
   problem.newton_solve();
   doc_info.number()=5;
   problem.doc_solution(doc_info);
 
   problem.adapt();
   problem.newton_solve();
   doc_info.number()=6;
   problem.doc_solution(doc_info);
   
  } // end of validation run
 
 // Count hanging nodes
 //cout << "Hanging nodes:" << endl;
 unsigned num_hang=0;
 for (unsigned n=0; n<problem.mesh_pt()->nnode(); n++)
  {
   if (problem.mesh_pt()->node_pt(n)->is_hanging())
    {
     /*
     cout << "  node " << n << " is hanging... at  ("
          << problem.mesh_pt()->node_pt(n)->x(0) << ", "
          << problem.mesh_pt()->node_pt(n)->x(1) << ")" << endl;
     HangInfo* hang_pt = problem.mesh_pt()->node_pt(n)->hanging_pt();
     cout << "    Nmaster = " << hang_pt->nmaster() << endl;
     double totweight = 0.0;
     for (unsigned nm=0; nm<hang_pt->nmaster(); nm++)
      {
       cout << "    master node:  x = (" << hang_pt->master_node_pt(nm)->x(0)
            << ", " << hang_pt->master_node_pt(nm)->x(1) << ")  w = "
            << hang_pt->master_weight(nm) << endl;
       totweight += hang_pt->master_weight(nm);
      }
     cout << "    Total weights = " << totweight << endl;
     */
     num_hang++;
    }
  }
 oomph_info << "There were "<<num_hang<<" hanging nodes." << endl;
 
 // Step number
 doc_info.number()=0;
 
 //Output solution
 problem.doc_solution(doc_info);
 
 
// Finalise MPI
#ifdef OOMPH_HAS_MPI
 
 MPI_Helpers::finalize();
 
#endif
 
} // end_of_main

References oomph::CommandLineArgs::Argc, e(), MergeRestartFiles::filename, n, oomph::DocInfo::number(), OOMPH_CURRENT_FUNCTION, OOMPH_EXCEPTION_LOCATION, oomph::oomph_info, problem, oomph::DocInfo::set_directory(), Flag_definition::setup(), and oomph::Global_string_for_annotation::string().