afopr_Sign.h

Go to the documentation of this file.

 
#ifndef AFOPR_SIGN_INCLUDED
#define AFOPR_SIGN_INCLUDED
 
#include "Fopr/afopr.h"
 
#include "Field/field_G.h"
#include "Solver/ashiftsolver_CG.h"
#include "Tools/math_Sign_Zolotarev.h"
#include "bridge_complex.h"
#include "complexTraits.h"
 
#include "IO/bridgeIO.h"
using Bridge::vout;
 
 
template<typename AFIELD>
class AFopr_Sign : public AFopr<AFIELD>
{
 public:
  typedef AFopr<AFIELD>                               AFOPR;
  typedef typename AFIELD::real_t                     real_t;
  typedef typename ComplexTraits<real_t>::complex_t   complex_t;
 
  static const std::string class_name;
 
 private:
  // input parameters
  int m_Np;           
  real_t m_x_min;     
  real_t m_x_max;     
  int m_Niter;        
  real_t m_Stop_cond; 
 
  Bridge::VerboseLevel m_vl;
 
  std::string m_mode;
 
  AFOPR *m_fopr;
 
  // local parameters
  int m_Nin, m_Nvol, m_Nex;
 
  std::vector<real_t> m_cl;
  std::vector<real_t> m_bl;
  std::vector<real_t> m_sigma;
 
  //- low-mode subtraction
  int m_Nsbt;
  std::vector<real_t> *m_ev;
  std::vector<AFIELD> *m_vk;
 
  //- shiftsolver
  AShiftsolver_CG<AFIELD, AFOPR> *m_solver;
  std::vector<AFIELD> m_xq;
 
  //- workarea
  AFIELD m_w1;
 
 public:
 
  AFopr_Sign(AFOPR *fopr, const Parameters& params) : m_fopr(fopr)
  { init(params); }
 
  DEPRECATED
  AFopr_Sign(AFOPR *fopr) : m_fopr(fopr) { init(); }
 
  ~AFopr_Sign() { tidyup(); }
 
  void set_parameters(const Parameters& params);
 
  void set_parameters(const int Np, const real_t x_min, const real_t x_max,
                      const int Niter, const real_t Stop_cond);
 
  void get_parameters(Parameters& params) const;
 
  void set_config(Field *U);
 
  void set_mode(const std::string mode);
 
  std::string get_mode() const { return m_mode; }
 
  void set_lowmodes(const int Nsbt, std::vector<real_t> *,
                    std::vector<AFIELD> *);
 
  void mult(AFIELD& v, const AFIELD& w);
 
  void mult_dag(AFIELD& v, const AFIELD& w) { mult(v, w); }
 
  virtual bool needs_convert()
  { return m_fopr->needs_convert(); }
 
  virtual void convert(AFIELD& v, const Field& w)
  { m_fopr->convert(v, w); }
 
  virtual void reverse(Field& v, const AFIELD& w)
  { m_fopr->reverse(v, w); }
 
  int field_nin() { return m_Nin; }
  int field_nvol() { return m_Nvol; }
  int field_nex() { return m_Nex; }
 
  double flop_count();
 
  double flop_count(const std::string mode);
 
 private:
  void init();
 
  void init(const Parameters& params);
 
  void tidyup();
 
  void init_parameters();
 
  void subtract_lowmodes(AFIELD&);
  void evaluate_lowmodes(AFIELD&, const AFIELD&);
 
  real_t sign_zolotarev(const real_t x);
 
#ifdef USE_FACTORY
 private:
  static AFOPR *create_object(AFOPR *fopr)
  {
    return new AFopr_Sign<AFIELD>(fopr);
  }
 
  static AFOPR *create_object_with_params(AFOPR *fopr, const Parameters& params)
  {
    return new AFopr_Sign<AFIELD>(fopr, params);
  }
 
 public:
  static bool register_factory()
  {
    bool init = true;
    init &= AFOPR::Factory_fopr::Register("Sign", create_object);
    init &= AFOPR::Factory_fopr_params::Register("Sign", create_object_with_params);
    return init;
  }
#endif
};
#endif