mpi/solvers/direct_solver_test.cc File Reference

#include "generic.h"
#include "poisson.h"
#include "meshes/one_d_mesh.h"

Classes
class	OneDPoissonProblem< ELEMENT >
	1D Poisson problem in unit interval. More...

Namespaces
	FishSolnOneDPoisson
	Namespace for fish-shaped solution of 1D Poisson equation.

Functions
void	FishSolnOneDPoisson::get_exact_u (const Vector< double > &x, Vector< double > &u)
	Exact, fish-shaped solution as a 1D vector. More...
void	FishSolnOneDPoisson::source_function (const Vector< double > &x, double &source)
	Source function required to make the fish shape an exact solution. More...
int	main (int argc, char **argv)
	Driver for 1D Poisson problem. More...

Function Documentation

main()

int main	(	int argc	,
		char **	argv
	)

Driver for 1D Poisson problem.

//////////////////////////////////////////////////////////////////////////// //////////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////////

{
#ifdef OOMPH_HAS_MPI
 // Setup mpi but don't make a copy of mpi_comm_world because
 // mumps wants to work with the real thing.
 bool make_copy_of_mpi_comm_world=false;
 MPI_Helpers::init(argc,argv,make_copy_of_mpi_comm_world);
#endif
 
 //Number of elements
 unsigned n_element=40;
 
 // global distribution
 // LinearAlgebraDistribution 
 //  distributed_dist(problem.communicator_pt(),ndof,true);
 
 // the doc info
 DocInfo doc_info;
 doc_info.set_directory("RESLT");
 
#ifdef OOMPH_HAS_MPI
 // SuperLU_dist Test
 
 // MATRIX BASED SOLVE / CRDoubleMatrix (global)
 {
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl;
  oomph_info 
   << "TESTING: SuperLU_dist matrix based solve w/ global CRDoubleMatrix"
   << std::endl;
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl << std::endl;
  OneDPoissonProblem<QPoissonElement<1,4> >
    problem(n_element,FishSolnOneDPoisson::source_function);
  unsigned ndof = problem.ndof();
  LinearAlgebraDistribution global_dist(problem.communicator_pt(),ndof,false);
  CRDoubleMatrix A(&global_dist);
  DoubleVector b(&global_dist,0.0);
  DoubleVector x(&global_dist,0.0);
  SuperLUSolver solver;
  solver.set_solver_type(SuperLUSolver::Distributed);
  problem.get_jacobian(b,A);
  solver.enable_doc_time();
  solver.solve(&A,b,x);
  x.output("RESLT/SuperLU_dist_CRDoubleMatrix_global.dat");
 }
 
 // MATRIX BASED SOLVE / CRDoubleMatrix (distributed)
 {
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl;
  oomph_info 
   << "TESTING: SuperLU_dist matrix based solve w/ dist CRDoubleMatrix"
   << std::endl;
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl << std::endl;
  OneDPoissonProblem<QPoissonElement<1,4> >
    problem(n_element,FishSolnOneDPoisson::source_function);
  unsigned ndof = problem.ndof();
  LinearAlgebraDistribution distributed_dist(problem.communicator_pt(),
                         ndof,true);
  CRDoubleMatrix A(&distributed_dist);
  DoubleVector b(&distributed_dist,0.0);
  DoubleVector x(&distributed_dist,0.0);
  SuperLUSolver solver;
  solver.set_solver_type(SuperLUSolver::Distributed);
  problem.get_jacobian(b,A);
  solver.enable_doc_time();
  solver.solve(&A,b,x);
  x.output("RESLT/SuperLU_dist_CRDoubleMatrix_distributed.dat");
 }
 
 // PROBLEM BASED SOLVE (global)
 {
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl;
  oomph_info << "TESTING: SuperLU_dist global problem based solve"
             << std::endl;
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl << std::endl;
  OneDPoissonProblem<QPoissonElement<1,4> >
    problem(n_element,FishSolnOneDPoisson::source_function);
  DoubleVector x;
  SuperLUSolver solver;
  solver.set_solver_type(SuperLUSolver::Distributed);
  solver.use_global_solve_in_superlu_dist();
  problem.linear_solver_pt() = &solver;
  problem.newton_solve();
  doc_info.number()++;
  problem.doc_solution(doc_info);
  problem.zero_dofs();
 }
 
 // PROBLEM BASED SOLVE (distributed)
 {
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl;
  oomph_info << "TESTING: SuperLU_dist distributed solve"
             << std::endl;
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl << std::endl;
  OneDPoissonProblem<QPoissonElement<1,4> >
    problem(n_element,FishSolnOneDPoisson::source_function);
  DoubleVector x;
  SuperLUSolver solver;
  solver.set_solver_type(SuperLUSolver::Distributed);
  solver.use_distributed_solve_in_superlu_dist();
  problem.linear_solver_pt() = &solver;
  problem.newton_solve();
  doc_info.number()++;
  problem.doc_solution(doc_info);
  problem.zero_dofs();
 }
#endif
 
 
#ifdef OOMPH_HAS_MUMPS
 
 // PROBLEM BASED SOLVE (distributed, using mumps)
 {
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl;
  oomph_info << "TESTING: MUMPS global problem based solve"
             << std::endl;
  oomph_info 
   << "///////////////////////////////////////////////////////////////////////"
   << std::endl << std::endl;
  OneDPoissonProblem<QPoissonElement<1,4> >
    problem(n_element,FishSolnOneDPoisson::source_function);
  DoubleVector x;
  MumpsSolver solver;
  problem.linear_solver_pt() = &solver;
  problem.newton_solve();
  doc_info.number()++;
  problem.doc_solution(doc_info);
  problem.zero_dofs();
 }
 
#else
 
 ofstream some_file;
 char filename[100];
 
 // Output solution with specified number of plot points per element
 sprintf(filename,"RESLT/dummy_mumps.dat");
 some_file.open(filename);
 some_file << "dummy data for missing mumps\n";
 some_file.close();
 
#endif
 
 
#ifdef OOMPH_HAS_MPI
 MPI_Helpers::finalize();
#endif
} // end of main

References b, MergeRestartFiles::filename, oomph::DocInfo::number(), oomph::oomph_info, problem, oomph::DocInfo::set_directory(), solver, FishSolnOneDPoisson::source_function(), and plotDoE::x.