solver_CG.h

Go to the documentation of this file.

 
#ifndef SOLVER_CG_INCLUDED
#define SOLVER_CG_INCLUDED
 
#include "solver.h"
 
#include "IO/bridgeIO.h"
using Bridge::vout;
 
 
class Solver_CG : public Solver
{
 public:
  static const std::string class_name;
 
 protected:
  Bridge::VerboseLevel m_vl;
 
 private:
  Fopr *m_fopr;
 
  int m_Niter;
  int m_Nrestart;
  double m_Stop_cond;
  bool m_use_init_guess;
 
  //- working area
  Field m_s, m_r, m_x, m_p;
 
  int m_Nrestart_count;
  int m_Nconv_count;
 
 public:
  Solver_CG(Fopr *fopr)
    : m_vl(CommonParameters::Vlevel()), m_fopr(fopr)
  {
    m_use_init_guess = false;
    m_Nrestart_count = 0;
    m_Nconv_count    = 0;
  }
 
  Solver_CG(Fopr *fopr, const Parameters& params)
    : m_vl(CommonParameters::Vlevel()), m_fopr(fopr)
  {
    m_use_init_guess = false;
    m_Nrestart_count = 0;
    m_Nconv_count    = 0;
 
    set_parameters(params);
  }
 
  ~Solver_CG() {}
 
  void set_parameters(const Parameters& params);
 
  void set_parameters(const int Niter, const int Nrestart, const double Stop_cond);
 
  void set_parameters(const int Niter, const int Nrestart, const double Stop_cond, const bool use_init_guess);
 
  void get_parameters(Parameters& params) const;
 
  void solve(Field& solution, const Field& source, int& Nconv, double& diff);
 
  Fopr *get_fopr() { return m_fopr; }
 
  double flop_count();
 
 private:
  void reset_field(const Field&);
 
  void solve_init(const Field&, double&);
  void solve_step(double&);
 
#ifdef USE_FACTORY
 private:
  static Solver *create_object(Fopr *fopr)
  {
    return new Solver_CG(fopr);
  }
 
  static Solver *create_object_with_params(Fopr *fopr, const Parameters& params)
  {
    return new Solver_CG(fopr, params);
  }
 
 public:
  static bool register_factory()
  {
    bool init = true;
    init &= Solver::Factory::Register("CG", create_object);
    init &= Solver::Factory_params::Register("CG", create_object_with_params);
    return init;
  }
#endif
};
#endif