ashiftsolver_CG.h

Go to the documentation of this file.

 
#ifndef ASHIFTSOLVER_CG_INCLUDED
#define ASHIFTSOLVER_CG_INCLUDED
 
#include "Solver/ashiftsolver.h"
 
#include "IO/bridgeIO.h"
using Bridge::vout;
 
 
 
template<typename FIELD, typename FOPR>
class AShiftsolver_CG : public AShiftsolver<FIELD>
{
 public:
  typedef typename FIELD::real_t real_t;
 
  static const std::string class_name;
 
 private:
  Bridge::VerboseLevel m_vl;
 
  FOPR *m_fopr;
 
  int m_Niter;
  double m_Stop_cond;
 
  std::vector<FIELD> m_x, m_p;
  FIELD m_r, m_s;
  std::vector<double> m_zeta1, m_zeta2, m_csh2, m_pp;
 
  double m_snorm, m_alpha_p, m_beta_p;
  int m_Nshift2;
 
  double m_sigma0;
 
 public:
 
  AShiftsolver_CG(FOPR *fopr)
    : m_vl(CommonParameters::Vlevel()),
    m_fopr(fopr) {}
 
  AShiftsolver_CG(FOPR *fopr, int niter, double stop_cond)
    : m_vl(CommonParameters::Vlevel()),
    m_fopr(fopr)
  { set_parameters(niter, stop_cond); }
 
  ~AShiftsolver_CG() {}
 
  void set_parameters(const Parameters& params);
 
  void set_parameters(const int niter, const double stop_cond);
 
  void get_parameters(Parameters& params) const;
 
  void solve(
    std::vector<FIELD>& solution,
    const std::vector<double>& shift,
    const FIELD& source,
    int& Nconv,
    double& diff);
 
  double flop_count();
 
 private:
 
  void solve_init(double&);
 
  void solve_step(double&);
 
  void reset_field(const FIELD& b,
                   const std::vector<double>& sigma,
                   const int Nshift);
};
#endif