fopr_Wilson_General_impl.h

Go to the documentation of this file.

#ifndef FOPR_WILSON_IMPL_INCLUDED
#define FOPR_WILSON_IMPL_INCLUDED

#include "fopr_Wilson_General.h"

#include "gammaMatrixSet.h"
#include "shiftField_lex.h"

#include "channel.h"

#include "mat_SU_N.h"
#include "vec_SU_N.h"

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


class Fopr_Wilson_General::Fopr_Wilson_General_impl
{
 public:
  static const std::string class_name;

 private:

  // lattice parameters
  int m_Nc, m_Nd, m_Nvc, m_Ndf;
  int m_Nx, m_Ny, m_Nz, m_Nt;
  int m_Nvol, m_Ndim;

  // physical parameters
  double              m_kappa_s;    
  double              m_kappa_t;    
  double              m_nu_s;       
  double              m_r_s;        
  std::vector<int>    m_boundary;   
  std::vector<double> m_boundary2;  

  const Field_G            *m_U;    
  std::vector<GammaMatrix> m_GM;    

  Bridge::VerboseLevel m_vl;

  std::string m_mode;
  std::string m_repr;

  Field m_w1, m_w2;   

  double *vcp1_xp, *vcp2_xp, *vcp1_xm, *vcp2_xm;
  double *vcp1_yp, *vcp2_yp, *vcp1_ym, *vcp2_ym;
  double *vcp1_zp, *vcp2_zp, *vcp1_zm, *vcp2_zm;
  double *vcp1_tp, *vcp2_tp, *vcp1_tm, *vcp2_tm;

  // use async data transfer
  std::vector<int>       m_npe;
  std::vector<Channel *> m_fw_send;
  std::vector<Channel *> m_fw_recv;
  std::vector<Channel *> m_bw_send;
  std::vector<Channel *> m_bw_recv;

 public:
  Fopr_Wilson_General_impl() {}

  void init(std::string repr);

  void tidyup();

  void set_parameters(const double kappa_s, const double kappa_t,
                      const double nu_s, const double r_s,
                      const std::vector<int> bc);

  void set_config(Field *U)
  { m_U = (Field_G *)U; }

  void set_mode(std::string mode);

  std::string get_mode() const;

  inline void mult(Field& v, const Field& f)
  { (this->*m_mult)(v, f); }

  inline void mult_dag(Field& v, const Field& f)
  { (this->*m_mult_dag)(v, f); }

  inline void mult_gm5(Field& v, const Field& f)
  { (this->*m_gm5)(v, f); }

  inline void D(Field& v, const Field& f)
  { (this->*m_D)(v, f); }

  inline void Ddag(Field& w, const Field& f)
  {
    mult_gm5(w, f);
    D(m_w1, w);
    mult_gm5(w, m_w1);
  }

  inline void DdagD(Field& w, const Field& f)
  {
    D(m_w1, f);
    mult_gm5(w, m_w1);
    D(m_w1, w);
    mult_gm5(w, m_w1);
  }

  inline void DDdag(Field& w, const Field& f)
  {
    mult_gm5(m_w1, f);
    D(w, m_w1);
    mult_gm5(m_w1, w);
    D(w, m_w1);
  }

  inline void H(Field& w, const Field& f)
  {
    D(m_w1, f);
    mult_gm5(w, m_w1);
  }

  inline void mult_undef(Field&, const Field& f)
  {
    vout.crucial(m_vl, "%s: mode undefined.\n", class_name.c_str());
    exit(EXIT_FAILURE);
  }

  inline void D_ex(Field& v, const int ex1,
                   const Field& f, const int ex2)
  { (this->*m_D_ex)(v, ex1, f, ex2); }

  const Field_F mult_gm5p(int mu, const Field_F& w)
  {
    mult_gm5p(mu, m_w1, w);
    return (Field_F)m_w1;
  }

  void mult_gm5p(int mu, Field&, const Field&);

  void proj_chiral(Field& w, const int ex1,
                   const Field& v, const int ex2, const int ipm);

  void mult_up(int mu, Field&, const Field&);
  void mult_dn(int mu, Field&, const Field&);

  double flop_count();

  // void fprop_normalize(Field& v);
  // void fopr_normalize(Field& v);
  // const double get_fprop_normfactor() { return 2.0 * m_kappa; }
  // const double get_fopr_normfactor() { return 1.0 / (2.0 * m_kappa); }

 private:
  // prohibit copy
  Fopr_Wilson_General_impl(const Fopr_Wilson_General::Fopr_Wilson_General_impl&) {}
  Fopr_Wilson_General_impl& operator=(const Fopr_Wilson_General::Fopr_Wilson_General_impl&);

  void (Fopr_Wilson_General::Fopr_Wilson_General_impl::*m_mult)(Field&, const Field&);
  void (Fopr_Wilson_General::Fopr_Wilson_General_impl::*m_mult_dag)(Field&, const Field&);
  void (Fopr_Wilson_General::Fopr_Wilson_General_impl::*m_D)(Field&, const Field&);
  void (Fopr_Wilson_General::Fopr_Wilson_General_impl::*m_gm5)(Field&, const Field&);
  void (Fopr_Wilson_General::Fopr_Wilson_General_impl::*m_mult_tp)(Field&, const Field&);
  void (Fopr_Wilson_General::Fopr_Wilson_General_impl::*m_mult_tm)(Field&, const Field&);
  void (Fopr_Wilson_General::Fopr_Wilson_General_impl::*m_D_ex)(Field&, const int,
                                                                const Field&, const int);

  void D_chiral(Field&, const Field&);
  void D_dirac(Field&, const Field&);
  void gm5_chiral(Field&, const Field&);
  void gm5_dirac(Field&, const Field&);

  void D_ex_chiral(Field&, const int ex1, const Field&, const int ex2);
  void D_ex_dirac(Field&, const int ex1, const Field&, const int ex2);

  void mult_p(int mu, Field_F&, const Field_F&);
  void mult_m(int mu, Field_F&, const Field_F&);

  void mult_xp(Field&, const Field&);
  void mult_xm(Field&, const Field&);
  void mult_yp(Field&, const Field&);
  void mult_ym(Field&, const Field&);
  void mult_zp(Field&, const Field&);
  void mult_zm(Field&, const Field&);

  void mult_tp_dirac(Field&, const Field&);
  void mult_tm_dirac(Field&, const Field&);
  void mult_tp_chiral(Field&, const Field&);
  void mult_tm_chiral(Field&, const Field&);

  void daypx(Field&, double, const Field&);
  void clear(Field&);

  // member data for threading
  int m_Mz, m_Mt;
  int m_Nthread, m_Ntask;
  int m_Ntask_z, m_Ntask_t;

  struct mult_arg
  {
    int isite;
    int isite_cpx, isite_cpy, isite_cpz, isite_cpt;
    int kz0, kz1, kt0, kt1;
  };
  std::vector<mult_arg> m_arg;

  // member functions for threading
  void setup_thread();
  void mult_xp1_thread(int, double *, const double *);
  void mult_xp2_thread(int, double *, const double *);
  void mult_xpb_thread(int, double *, const double *);
  void mult_xm1_thread(int, double *, const double *);
  void mult_xm2_thread(int, double *, const double *);
  void mult_xmb_thread(int, double *, const double *);
  void mult_yp1_thread(int, double *, const double *);
  void mult_yp2_thread(int, double *, const double *);
  void mult_ypb_thread(int, double *, const double *);
  void mult_ym1_thread(int, double *, const double *);
  void mult_ym2_thread(int, double *, const double *);
  void mult_ymb_thread(int, double *, const double *);
  void mult_zp1_thread(int, double *, const double *);
  void mult_zp2_thread(int, double *, const double *);
  void mult_zpb_thread(int, double *, const double *);
  void mult_zm1_thread(int, double *, const double *);
  void mult_zm2_thread(int, double *, const double *);
  void mult_zmb_thread(int, double *, const double *);
  void mult_tp1_dirac_thread(int, double *, const double *);
  void mult_tp2_dirac_thread(int, double *, const double *);
  void mult_tpb_dirac_thread(int, double *, const double *);
  void mult_tm1_dirac_thread(int, double *, const double *);
  void mult_tm2_dirac_thread(int, double *, const double *);
  void mult_tmb_dirac_thread(int, double *, const double *);
  void mult_tp1_chiral_thread(int, double *, const double *);
  void mult_tp2_chiral_thread(int, double *, const double *);
  void mult_tpb_chiral_thread(int, double *, const double *);
  void mult_tm1_chiral_thread(int, double *, const double *);
  void mult_tm2_chiral_thread(int, double *, const double *);
  void mult_tmb_chiral_thread(int, double *, const double *);
  void daypx_thread(int, double *, double, const double *);
  void clear_thread(int, double *);
  void gm5_dirac_thread(int, double *, const double *);
  void gm5_chiral_thread(int, double *, const double *);
};
#endif