source: src/solver/solver.hpp@ 66f24d

/**
 * @file   solver.hpp
 * @author Julian Iseringhausen <isering@ins.uni-bonn.de>
 * @date   Mon Apr 18 13:11:14 2011
 *
 * @brief  VMG::Solver
 *
 */

#ifndef SOLVER_HPP_
#define SOLVER_HPP_

#include "base/discretization.hpp"
#include "base/object.hpp"
#include "base/stencil.hpp"
#include "comm/comm.hpp"
#include "grid/grid.hpp"
#include "mg.hpp"

namespace VMGTests
{
class SolverTestSuite;
}

namespace VMG
{

class Solver : public Object
{
public:
  friend class VMGTests::SolverTestSuite;

  Solver()
  {
    max_size = 0;
    vec_size = 0;
    b = NULL;
    x = NULL;
    A = NULL;
  }

  virtual ~Solver()
  {
    delete [] b;
    delete [] x;
    for (int i=0; i<max_size; i++)
      delete [] A[i];
    delete [] A;
  }

  void Run(Grid& sol, Grid& rhs)
  {
#ifdef DEBUG
    sol.IsConsistent();
    rhs.IsConsistent();
    sol.IsCompatible(rhs);
#endif

    Grid *rhsGlobal = MG::GetComm()->CommMergeGrids(rhs);

    this->Realloc(*rhsGlobal);
    this->AssembleMatrix(*rhsGlobal);
    this->Compute();
    this->ExportSol(sol, rhs);
  }

private:

  virtual void Compute() = 0; ///< Solves the system of equations
  virtual void AssembleMatrix(const Grid& rhs) = 0; ///< Assembles all matrices and vectors.
  virtual void ExportSol(Grid& sol, Grid& rhs) = 0; ///< Exports the solution back to a given mesh.

  vmg_float **A, *b, *x;
  int max_size, vec_size;

protected:
  void Realloc(int n);
  void Realloc(Grid& x);

  vmg_float& Mat(int i, int j)
  {
    return A[i][j];
  }

  vmg_float& Rhs(int i)
  {
    return b[i];
  }

  vmg_float& Sol(int i)
  {
    return x[i];
  }

  const int& Size() const {return vec_size;}
  const int& MemSize()  const {return max_size;}
};

}

#endif