shiftsolver_CG.h

Go to the documentation of this file.

#ifndef SHIFTSOLVER_CG_INCLUDED
#define SHIFTSOLVER_CG_INCLUDED

#include "shiftsolver.h"

#include "bridgeIO.h"
using Bridge::vout;

//- parameters class
class Parameters_Shiftsolver_CG : virtual public Parameters
{
 public:
  Parameters_Shiftsolver_CG();
};
//- end


class Shiftsolver_CG : public Shiftsolver
{
 public:
  static const std::string class_name;

 private:

  Fopr   *m_fopr;
  int    m_Niter;
  double m_Stop_cond;

  std::vector<Field>  x, p;
  Field               r, s;
  std::vector<double> zeta1, zeta2, csh2, pp;
  double              snorm, alpha_p, beta_p;
  int                 Nshift2;

  double m_sigma0;

 public:
  Shiftsolver_CG(Fopr *fopr)
    : Shiftsolver(), m_fopr(fopr) {}

  Shiftsolver_CG(Fopr *fopr, int niter, double stop_cond)
    : Shiftsolver(), m_fopr(fopr)
  {
    set_parameters(niter, stop_cond);
  }

  Shiftsolver_CG(unique_ptr<Fopr>& fopr)
    : Shiftsolver(), m_fopr(fopr.get()) {}

  Shiftsolver_CG(unique_ptr<Fopr>& fopr, int niter, double stop_cond)
    : Shiftsolver(), m_fopr(fopr.get())
  {
    set_parameters(niter, stop_cond);
  }

  ~Shiftsolver_CG() {}

  void set_parameters(const Parameters& params);
  void set_parameters(const int niter, const double stop_cond);

  void solve(
    std::vector<Field>& solution,
    const std::vector<double>& shift,
    const Field& source,
    int& Nconv, double& diff);

 private:

  void solve_init(double&);
  void solve_step(double&);

  void reset_field(const Field& b, const std::vector<double>& sigma, const int Nshift);
};
#endif