Real random(const Teuchos::RCP<Epetra_Comm> &comm) {
Real val = 0.0;
if ( comm->MyPID()==0 ) {
val = (Real)rand()/(Real)RAND_MAX;
}
comm->Broadcast(&val,1,0);
return val;
}
int main(int argc, char *argv[])
{
// Initialize MPI
Teuchos::GlobalMPISession mpiSession(&argc,&argv);
int ierr = 0;
int MyPID = 0;
double alpha = 1.00; // stable steady state
double beta = BETA_INIT;
double D1 = 1.0/40.0;
double D2 = 1.0/40.0;
int NumGlobalNodes = 10; // default
int spatialProcs = 1; // default
int numTimeSteps = 100; // default
try {
// Check for verbose output
bool verbose = false;
if (argc>1)
if (argv[1][0]=='-' && argv[1][1]=='v')
verbose = true;
// Get the number of elements from the command line
if ((argc > 2) && (verbose))
NumGlobalNodes = atoi(argv[2]) + 1;
else if ((argc > 1) && (!verbose))
NumGlobalNodes = atoi(argv[1]) + 1;
// Get the number of processors to use for the spatial problem
if ((argc > 3) && (verbose))
spatialProcs = atoi(argv[3]);
else if ((argc > 2) && (!verbose))
spatialProcs = atoi(argv[2]);
// Get the number of processors to use for the spatial problem
if ((argc > 4) && (verbose))
numTimeSteps = atoi(argv[4]);
else if ((argc > 3) && (!verbose))
numTimeSteps = atoi(argv[3]);
// MPI MANIPULATION FOR XYZT PROBLEMS
#ifdef HAVE_MPI
Teuchos::RCP<EpetraExt::MultiMpiComm> globalComm =
Teuchos::rcp(new EpetraExt::MultiMpiComm(MPI_COMM_WORLD,
spatialProcs, numTimeSteps));
#else
Teuchos::RCP<EpetraExt::MultiSerialComm> globalComm =
Teuchos::rcp(new EpetraExt::MultiSerialComm(numTimeSteps));
#endif
Epetra_Comm& Comm = globalComm->SubDomainComm();
MyPID = globalComm->MyPID();
// Get the total number of processors
int NumProc = Comm.NumProc();
// The number of unknowns must be at least equal to the
// number of processors.
if (NumGlobalNodes < NumProc) {
std::cout << "numGlobalNodes = " << NumGlobalNodes
<< " cannot be < number of processors = " << NumProc
<< std::endl;
exit(1);
}
// Create the Brusselator problem class. This creates all required
// Epetra objects for the problem and allows calls to the
// function (F) and Jacobian evaluation routines.
Brusselator::OverlapType OType = Brusselator::ELEMENTS;
Brusselator Problem(NumGlobalNodes, Comm, OType);
// Get the vector from the Problem
Epetra_Vector& soln = Problem.getSolution();
// Begin Nonlinear Solver ************************************
// Create the top level parameter list
Teuchos::RCP<Teuchos::ParameterList> paramList =
Teuchos::rcp(new Teuchos::ParameterList);
getParamList(&(*paramList), MyPID);
// Sublist for "Linear Solver"
Teuchos::ParameterList& lsParams =
paramList->sublist("NOX").sublist("Direction").sublist("Newton").sublist("Linear Solver");
// Sublist for "Printing"
Teuchos::ParameterList& printParams =
paramList->sublist("NOX").sublist("Printing");
// Change NOX priting settings if "-verbose" not requested
if (!verbose)
printParams.set("Output Information", NOX::Utils::Error);
// Create the interface between the test problem and the nonlinear solver
Teuchos::RCP<Problem_Interface> interface =
Teuchos::rcp(new Problem_Interface(Problem));
// Create the Epetra_RowMatrixfor the Jacobian/Preconditioner
Teuchos::RCP<Epetra_RowMatrix> A =
Teuchos::rcp(&Problem.getJacobian(),false);
// Create initial guess
Epetra_MultiVector initGuess(soln.Map(),
globalComm->NumTimeStepsOnDomain());
for (int i=0; i<globalComm->NumTimeStepsOnDomain(); i++)
*(initGuess(i)) = soln;
// Get XYZT preconditioner linear solver parameters
Teuchos::RCP<Teuchos::ParameterList> precLSParams =
Teuchos::rcp(new Teuchos::ParameterList);
getPrecLSParams(&(*precLSParams), MyPID);
// Get XYZT preconditioner print parameters
Teuchos::RCP<Teuchos::ParameterList> precPrintParams =
Teuchos::rcp(new Teuchos::ParameterList);
getPrecPrintParams(&(*precPrintParams), MyPID);
// Change xyztPrec priting settings if "-verbose" not requested
if (!verbose)
precPrintParams->set("Output Information", NOX::Utils::Error);
// Create the XYZT object
Teuchos::RCP<LOCA::Epetra::Interface::xyzt> ixyzt =
Teuchos::rcp(new LOCA::Epetra::Interface::xyzt(interface,
initGuess, A,
globalComm,
soln, 0.5,
precPrintParams.get(),
precLSParams.get()));
// Create the XYZT operator, solution, and preconditioner
Teuchos::RCP<Epetra_RowMatrix> Axyzt =
Teuchos::rcp(&(ixyzt->getJacobian()),false);
Epetra_Vector& solnxyzt = ixyzt->getSolution();
Teuchos::RCP<Epetra_Operator> Mxyzt =
Teuchos::rcp(&(ixyzt->getPreconditioner()),false);
// Create the Linear System
Teuchos::RCP<LOCA::Epetra::Interface::Required> iReq = ixyzt;
Teuchos::RCP<NOX::Epetra::Interface::Jacobian> iJac = ixyzt;
Teuchos::RCP<NOX::Epetra::Interface::Preconditioner> iPrec =
Teuchos::rcp(&(ixyzt->getPreconditioner()),false);
Teuchos::RCP<NOX::Epetra::LinearSystemAztecOO> linSys =
Teuchos::rcp(new NOX::Epetra::LinearSystemAztecOO(printParams, lsParams,
iJac, Axyzt,
iPrec, Mxyzt,
solnxyzt));
// Create full xyzt initial guess
NOX::Epetra::Vector initialGuess(solnxyzt);
// Create and initialize the parameter vector
LOCA::ParameterVector pVector;
pVector.addParameter("alpha",alpha);
pVector.addParameter("beta",beta);
pVector.addParameter("D1",D1);
pVector.addParameter("D2",D2);
// Create Epetra factory
Teuchos::RCP<LOCA::Abstract::Factory> epetraFactory =
Teuchos::rcp(new LOCA::Epetra::Factory);
// Create global data object
Teuchos::RCP<LOCA::GlobalData> globalData =
LOCA::createGlobalData(paramList, epetraFactory);
Teuchos::RCP<LOCA::Epetra::Group> grp =
Teuchos::rcp(new LOCA::Epetra::Group(globalData, printParams,
iReq, initialGuess, linSys,
pVector));
grp->computeF();
// Create the convergence tests
Teuchos::RCP<NOX::StatusTest::NormF> absresid =
Teuchos::rcp(new NOX::StatusTest::NormF(1.0e-8,
NOX::StatusTest::NormF::Unscaled));
Teuchos::RCP<NOX::StatusTest::MaxIters> maxiters =
Teuchos::rcp(new NOX::StatusTest::MaxIters(MAX_NEWTON_ITERS));
Teuchos::RCP<NOX::StatusTest::Combo> combo =
Teuchos::rcp(new NOX::StatusTest::Combo(NOX::StatusTest::Combo::OR));
combo->addStatusTest(absresid);
combo->addStatusTest(maxiters);
// Create stepper
LOCA::Stepper stepper(globalData, grp, combo, paramList);
LOCA::Abstract::Iterator::IteratorStatus status = stepper.run();
if (status != LOCA::Abstract::Iterator::Finished) {
ierr =1;
if (globalData->locaUtils->isPrintType(NOX::Utils::Error))
globalData->locaUtils->out()
<< "Stepper failed to converge!" << std::endl;
}
// Get the final solution from the stepper
Teuchos::RCP<const LOCA::Epetra::Group> finalGroup =
Teuchos::rcp_dynamic_cast<const LOCA::Epetra::Group>(stepper.getSolutionGroup());
const NOX::Epetra::Vector& finalSolution =
dynamic_cast<const NOX::Epetra::Vector&>(finalGroup->getX());
// Output the parameter list
if (globalData->locaUtils->isPrintType(NOX::Utils::StepperParameters)) {
globalData->locaUtils->out()
<< std::endl << "Final Parameters" << std::endl
<< "****************" << std::endl;
stepper.getList()->print(globalData->locaUtils->out());
globalData->locaUtils->out() << std::endl;
}
// Check some statistics on the solution
NOX::TestCompare testCompare(globalData->locaUtils->out(),
*(globalData->locaUtils));
if (globalData->locaUtils->isPrintType(NOX::Utils::TestDetails))
globalData->locaUtils->out()
<< std::endl
<< "***** Checking solution statistics *****"
<< std::endl;
// Check number of continuation steps
int numSteps = stepper.getStepNumber();
int numSteps_expected = 3;
ierr += testCompare.testValue(numSteps, numSteps_expected, 0.0,
"number of continuation steps",
NOX::TestCompare::Absolute);
// Check number of failed steps
int numFailedSteps = stepper.getNumFailedSteps();
int numFailedSteps_expected = 0;
ierr += testCompare.testValue(numFailedSteps, numFailedSteps_expected, 0.0,
"number of failed continuation steps",
NOX::TestCompare::Absolute);
// Check final value of continuation parameter
double beta_final = finalGroup->getParam("beta");
double beta_expected = 2.0;
ierr += testCompare.testValue(beta_final, beta_expected, 1.0e-14,
"final value of continuation parameter",
NOX::TestCompare::Relative);
// Check final value of ||F||
double Fnorm_final = (const_cast<Teuchos::ParameterList&>(*stepper.getList()).sublist("NOX").sublist("Output").get("2-Norm of Residual",1.0e+10));
double Fnorm_expected = 0.0;
ierr += testCompare.testValue(Fnorm_final, Fnorm_expected, 1.0e-8,
"final value of ||F||",
NOX::TestCompare::Absolute);
// Check number of nonlinear iterations on last continuation step
int nonlin_final = (const_cast<Teuchos::ParameterList&>(*stepper.getList()).sublist("NOX").
sublist("Output").get("Nonlinear Iterations",MAX_NEWTON_ITERS));
int nonlin_expected = 4;
ierr += testCompare.testValue(nonlin_final, nonlin_expected, 0.0,
"number of nonlinear iterations on last continuation step",
NOX::TestCompare::Absolute);
// initialize solution comparison norm on all procs
double solution_diff_norm = 0.0;
// Compute norm of difference of computed solution - expected solution
if (globalComm->MyPID() == (globalComm->NumProc()-1)) {
// Get solution at last time step
ixyzt->getSolution().ExtractBlockValues(soln,
(globalComm->NumTimeStepsOnDomain() +
globalComm->FirstTimeStepOnDomain() - 1));
// Check final solution at final time step
NOX::Epetra::Vector final_solution(soln);
NOX::Epetra::Vector final_x_expected(final_solution);
for (int i=0; i<NumGlobalNodes; i++) {
final_x_expected.getEpetraVector()[2*i] = alpha;
final_x_expected.getEpetraVector()[2*i+1] = beta_final/alpha;
}
solution_diff_norm = testCompare.computeVectorNorm(final_solution,
final_x_expected,
1.0e-4,
1.0e-4);
}
// Check final solution at final time step on all procs
globalComm->Broadcast(&solution_diff_norm, 1, globalComm->NumProc()-1);
double solution_diff_norm_expected = 0.0;
ierr += testCompare.testValue(solution_diff_norm, solution_diff_norm_expected, 1.0e-4,
"inf norm of (solution_final - solution_expected) at last time step",
NOX::TestCompare::Absolute);
LOCA::destroyGlobalData(globalData);
}
catch (std::exception& e) {
std::cout << e.what() << std::endl;
ierr = 1;
}
catch (const char *s) {
std::cout << s << std::endl;
ierr = 1;
}
catch (...) {
std::cout << "Caught unknown exception!" << std::endl;
ierr = 1;
}
if (MyPID == 0) {
if (ierr == 0)
std::cout << "All tests passed!" << std::endl;
else
std::cout << ierr << " test(s) failed!" << std::endl;
}
return ierr;
}