afopr_Clover_eo.h

Go to the documentation of this file.

 
#ifndef ACCEL_AFOPR_CLOVER_EO_H
#define ACCEL_AFOPR_CLOVER_EO_H
 
#include <cstdio>
#include <cstdlib>
#include <string>
#include <vector>
 
#include "lib/Fopr/afopr.h"
#include "lib/Parameters/commonParameters.h"
#include "lib/Communicator/communicator.h"
#include "lib/Communicator/communicator_impl.h"
#include "lib/Fopr/fopr_CloverTerm_eo.h"
#include "lib/Tools/timer.h"
#include "lib/IO/bridgeIO.h"
using Bridge::vout;
 
#include "lib_alt_Accel/Field/aindex_eo.h"
#include "lib_alt_Accel/Fopr/afopr_CloverTerm.h"
 
class Field;
 
template<typename AFIELD>
class AFopr_Clover_eo : public AFopr<AFIELD>
{
 public:
  typedef typename AFIELD::real_t real_t;
  static const std::string class_name;
 
 protected:
  enum Gamma_repr {DIRAC, CHIRAL} m_repr;  
 
  int m_Nc, m_Nd;
  int m_Ndf, m_Ncol, m_Nvc;
  int m_Nst, m_Nx, m_Ny, m_Nz, m_Nt;
  int m_Ndim;
  int m_Nx2, m_Nst2;
  int m_Nx2v, m_Nst2v;
 
  real_t  m_CKs;                
  real_t  m_cSW;                
  std::vector<int> m_boundary;  
  // std::string  m_repr;       //!< gamma matrix representation
  Bridge::VerboseLevel m_vl;    
 
  int m_Ieo_origin;
 
  Field *m_conf;        
  AFIELD m_Ueo; 
 
  AFIELD m_fee_inv, m_foo_inv;
  AFIELD m_Tinv;
 
  std::string m_mode;  
 
  AIndex_eo<real_t,AFIELD::IMPL> m_index_eo;
 
  AFIELD m_v1;
  AFIELD m_v2;
 
  int do_comm[4];  // switchs of communication (4=Ndim): (0: n, 1: y).
  int do_comm_any; // switchs of communication (if any): (0: n, 1: y).
 
  int m_Nsize[4];
  int m_bc[4];
  int m_bc2[4];
 
  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;
 
  Fopr_CloverTerm_eo* m_fopr;
 
  AFopr_CloverTerm<AFIELD>  *m_fopr_ct;
 
  Timer m_timer;
 
public:
  AFopr_Clover_eo(const Parameters& params){
    init(params);
  }
 
  ~AFopr_Clover_eo(){ tidyup(); }
 
  void set_parameters(const Parameters& params);
 
  void set_parameters(real_t CKs, real_t cSW, std::vector<int> bc);
 
  void get_parameters(Parameters& params) const;
 
  void set_config(Field* u);
    
  void set_boundary_config(AFIELD&, int mu);
 
  inline Field* get_conf() { return m_conf; };
 
  void set_mode(std::string mode);
 
  std::string get_mode() const { return m_mode; }
 
  void mult(AFIELD&, const AFIELD&);
  void mult_dag(AFIELD&, const AFIELD&);
  void mult_gm5(AFIELD&, const AFIELD&);
 
  void mult_gm4(AFIELD&, const AFIELD&);
 
  void mult(AFIELD&, const AFIELD&,
            const std::string mode);
 
  void DdagD(AFIELD&, const AFIELD&);
  void Ddag(AFIELD&, const AFIELD&);
  void D(AFIELD&, const AFIELD&);
  void gm5(AFIELD&, const AFIELD&);
 
  //  void aypx(real_t, AFIELD&, const AFIELD&);
 
  void mult_test(AFIELD&, const AFIELD&);
 
  void Meo(AFIELD&, const AFIELD&, const int ieo);
 
  void Meo(AFIELD&, const AFIELD&, const AFIELD&,
           const int ieo, const int iflag);
 
  void Meo_alt(AFIELD&, const AFIELD&, const int ieo);
 
  int field_nin() { return 2 * m_Nc * m_Nd; }
 
  int field_nvol() { return m_Nst2; }
 
  int field_nex() { return 1; }
 
  double flop_count();
 
 private:
  void init(const Parameters& params);
 
  void tidyup();
 
  void setup_channels();
 
  void set_config_omp(Field* u);
  void set_config_impl(Field* u);
 
  void mult_xp(real_t*, real_t*, const int);
  void mult_xm(real_t*, real_t*, const int);
  void mult_yp(real_t*, real_t*, const int);
  void mult_ym(real_t*, real_t*, const int);
  void mult_zp(real_t*, real_t*, const int);
  void mult_zm(real_t*, real_t*, const int);
  void mult_tp(real_t*, real_t*, const int);
  void mult_tm(real_t*, real_t*, const int);
 
  void set_csw(int ieo);
 
  void mult_cswinv(AFIELD&, const AFIELD&, int ieo);
 
  void mult_cswinv(real_t *v2, real_t *v1, int ieo);
 
  void mult_cswinv_dirac(real_t *v2, real_t *v1,
                         real_t *csw_inv, int site);
 
  void mult_gm5(real_t*, real_t*);
 
#ifdef USE_FACTORY
 private:
  static AFopr<AFIELD> *create_object_with_params(const Parameters& params)
  { return new AFopr_Clover_eo(params); }
 
 public:
  static bool register_factory()
  {
    bool init1 = AFopr<AFIELD>::Factory_params::Register("Clover_eo",
                                               create_object_with_params);
    return init1;
  }
#endif
 
};
 
#endif  // ACCEL_AFOPR_CLOVER_EO_H