/*
 * dirichlet_fas.cpp
 *
 *  Created on: 25.01.2011
 *      Author: Julian Iseringhausen
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include "base/factory.hpp"
#include "base/vector.hpp"
#include "comm/comm_serial.hpp"
#include "level/level_operator_fas.hpp"
#include "level/level_operator.hpp"
#include "samples/discretization_poisson_fd.hpp"
#include "samples/stencils.hpp"
#include "samples/techniques.hpp"
#include "smoother/gsrb.hpp"
#ifdef HAVE_LAPACK
#include "solver/dgesv.hpp"
#endif
#include "solver/givens.hpp"
#include "solver/solver_regular.hpp"
#include "mg.hpp"

#include "interface_sinus.hpp"
#include "dirichlet_fas.hpp"

using namespace VMG;
using VMGTests::DirichletFASTestSuite;

CPPUNIT_TEST_SUITE_REGISTRATION(DirichletFASTestSuite);

void DirichletFASTestSuite::setUp()
{
  Factory& factory = MG::GetFactory();

  Comm *comm = new CommSerial(Boundary(Dirichlet, Dirichlet, Dirichlet));
  comm->Register("COMM");

  Interface* interface = new VMGInterfaces::InterfaceSinus(comm->BoundaryConditions(), 2, 6, 0.0, 1.0);
  MG::SetInterface(interface, comm);

  Discretization* discretization = new DiscretizationPoissonFD();
  discretization->Register("DISCRETIZATION");

  LevelOperator* lop = new LevelOperatorFAS(Stencils::RestrictionFullWeight, Stencils::Injection, Stencils::InterpolationTrilinear);
  lop->Register("LEVEL_OPERATOR");

  Smoother* smoother = new GaussSeidelRB();
  smoother->Register("SMOOTHER");

#ifdef HAVE_LAPACK
  Solver* solver = new DGESV<SolverRegular>();
#else
  Solver* solver = new Givens<SolverRegular>();
#endif

  solver->Register("SOLVER");

  Techniques::SetFullApproximationSchemeDirichlet(interface->MinLevel(), interface->MaxLevel(), 2);

  factory.RegisterObjectStorage("PRESMOOTHSTEPS", 3);
  factory.RegisterObjectStorage("POSTSMOOTHSTEPS", 3);
  factory.RegisterObjectStorage("PRECISION", 1e-10);
  factory.RegisterObjectStorage("MAX_ITERATION", 20);

  MG::IsInitialized();
}

void DirichletFASTestSuite::tearDown()
{
  MG::Destroy();
}

void DirichletFASTestSuite::DirichletFASTest()
{
  MG::Solve();

  double res_init = MG::GetFactory().Get("INITIAL_RESIDUAL")->Cast< ObjectStorage<double> >()->Val();
  double res = MG::GetComm()->ComputeResidualNorm(*MG::GetSol(), *MG::GetRhs());

  CPPUNIT_ASSERT_DOUBLES_EQUAL(0.0, fabs(res/res_init), 1e-10);
}