afopr_Domainwall_5din.h

Go to the documentation of this file.

 
#ifndef ACCEL_AFOPR_DOMAINWALL_5DIN_INCLUDED
#define ACCEL_AFOPR_DOMAINWALL_5DIN_INCLUDED
 
#include <vector>
#include <string>
 
#include "lib/Fopr/afopr.h"
#include "lib/Parameters/commonParameters.h"
#include "lib/Communicator/communicator.h"
#include "lib/Communicator/communicator_impl.h"
#include "lib/IO/bridgeIO.h"
using Bridge::vout;
 
class Field;
class Field_G;
 
 
template<typename AFIELD>
class AFopr_Domainwall_5din : public AFopr<AFIELD>
{
 public:
  typedef typename AFIELD::real_t  real_t;
  static const std::string class_name;
 
 private:
  // parameters common to overlap fermion
  real_t m_mq;                 
  real_t m_M0;                 
  int m_Ns;                    
  std::vector<int> m_boundary; 
  std::vector<real_t> m_b;     
  std::vector<real_t> m_c;     
  real_t m_alpha;              
  Bridge::VerboseLevel m_vl;   
  std::string m_kernel_type;   
  std::string m_repr;          
  std::string m_impl;          
 
  int m_Nx, m_Ny, m_Nz, m_Nt;
  int m_Nvol, m_Ndim;
  int m_NinF, m_Nvcd, m_Ndf;
 
  // for preconditioning
  std::vector<real_t> m_dp;
  std::vector<real_t> m_dpinv;
  std::vector<real_t> m_dm;
  std::vector<real_t> m_e;
  std::vector<real_t> m_f;
  real_t m_g;
 
  int m_Nsize[4];  // lattice sizes in units of SIMD variable
  int m_bc[4];
  int m_bc2[4];
 
  int do_comm[4];  
  int do_comm_any; 
 
  std::vector<int> m_Nbdsize;
  using Channel = Channel_impl<std::allocator<real_t> >;
  std::vector<Channel> chsend_up, chrecv_up, chsend_dn, chrecv_dn;
  ChannelSet chset_send, chset_recv;
 
  std::string m_mode;
 
  AFIELD m_U;
 
  //  AFIELD m_w4, m_v4, m_t4, m_y4; //!< working 4d vectors.
  AFIELD m_w1, m_v1, m_v2;       
 
  // for convert and reverse
  Field m_w4lex;
  AFIELD m_v4lex;
 
 
 public:
  AFopr_Domainwall_5din(const Parameters& params) : AFopr<AFIELD>()
  { init(params); }
 
  ~AFopr_Domainwall_5din() { tidyup(); }
 
  void set_parameters(const Parameters& params);
 
  void set_parameters(const real_t mq, const real_t M0,
                      const int Ns, const std::vector<int> bc,
                      const real_t b, const real_t c,
                      const real_t alpha);
 
  /*
  void set_parameters(const real_t mq, const real_t M0,
                      const int Ns, const std::vector<int> bc,
                      const std::vector<real_t> vec_b,
                      const std::vector<real_t> vec_c,
                      const real_t alpha);
  */
 
  void get_parameters(Parameters& params) const;
 
  void setup_channels();
 
  void set_precond_parameters();
 
  void set_coefficients(const std::vector<real_t> b,
                        const std::vector<real_t> c);
 
  bool needs_convert() { return true; }
 
  void convert(AFIELD&, const Field&);
 
  void reverse(Field&, const AFIELD&);
 
  void set_config(Field *U);
 
  void set_config_omp(Field *U);
 
  void set_config_impl(Field *U);
 
  void set_mode(std::string mode);
 
  std::string get_mode() const { return m_mode; }
 
  void mult(AFIELD& v, const AFIELD& w);
 
  void mult_dag(AFIELD& v, const AFIELD& w);
 
  void mult(AFIELD& v, const AFIELD& w, std::string mode);
 
  void mult_dag(AFIELD& v, const AFIELD& w, std::string mode);
 
  void mult_gm5(AFIELD&, const AFIELD&);
 
  void mult_chproj_4d(AFIELD&, const AFIELD&, const int ipm);
 
  int field_nin() { return m_NinF; }
  int field_nvol() { return m_Nvol; }
  int field_nex() { return 1; }
 
  double flop_count() { return flop_count(m_mode); }
 
  double flop_count(std::string mode);
 
  void DdagD(AFIELD&, const AFIELD&);
  void DDdag(AFIELD&, const AFIELD&);
  void D(AFIELD&, const AFIELD&);
  void Ddag(AFIELD&, const AFIELD&);
  void H(AFIELD&, const AFIELD&);
  void Hdag(AFIELD&, const AFIELD&);
  void mult_gm5R(AFIELD&, const AFIELD&);
  void mult_R(AFIELD&, const AFIELD&);
 
  //  preconditioner
  void DdagD_prec(AFIELD&, const AFIELD&);
  void D_prec(AFIELD&, const AFIELD&);
  void Ddag_prec(AFIELD&, const AFIELD&);
  void Prec(AFIELD&, const AFIELD&);
  void Precdag(AFIELD&, const AFIELD&);
 
 private:
 
  void init(const Parameters& params);
 
  void tidyup();
 
  void LU_inv(AFIELD&, const AFIELD&);
  void LUdag_inv(AFIELD&, const AFIELD&);
 
  void mult_gm5_4d(AFIELD&, const AFIELD&);
 
#ifdef USE_FACTORY
 private:
  static AFopr<AFIELD> *create_object_with_params(const Parameters& params)
  { return new AFopr_Domainwall_5din(params); }
 
 public:
  static bool register_factory()
  {
    bool init = true;
    init &= AFopr<AFIELD>::Factory_params::Register(
                     "Domainwall_5din", create_object_with_params);
    return init;
  }
#endif
};
 
#endif