aindex_lex.h

Go to the documentation of this file.

 
#ifndef ACCEL_AINDEX_LEX_INCLUDED
#define ACCEL_AINDEX_LEX_INCLUDED
 
#include "lib_alt/Field/aindex_lex_base.h"
 
#include "lib/Parameters/commonParameters.h"
 
#include "lib_alt_Accel/inline/define_params.h"
#include "lib_alt_Accel/inline/define_index.h"
 
 
template<typename REALTYPE>
class AIndex_lex<REALTYPE, ACCEL> {
 protected:
  int m_Nc, m_Nd, m_Ndf, m_Nvcd;
  int m_Nx, m_Ny, m_Nz, m_Nt;
  int m_Nvol;        
  int m_Nvol_pad;    
 
 public:
  AIndex_lex() {
    m_Nc = CommonParameters::Nc();
    m_Nd = CommonParameters::Nd();
    m_Ndf  = 2 * m_Nc * m_Nc;
    m_Nvcd = 2 * m_Nc * m_Nd;
    m_Nx = CommonParameters::Nx();
    m_Ny = CommonParameters::Ny();
    m_Nz = CommonParameters::Nz();
    m_Nt = CommonParameters::Nt();
    m_Nvol = CommonParameters::Nvol();
    m_Nvol_pad = ceil_nwp(m_Nvol);
   }
 
  AIndex_lex(int Nx, int Ny, int Nz, int Nt)
  {
    m_Nc = CommonParameters::Nc();
    m_Nd = CommonParameters::Nd();
    m_Ndf  = 2 * m_Nc * m_Nc;
    m_Nvcd = 2 * m_Nc * m_Nd;
    m_Nx = Nx;
    m_Ny = Ny;
    m_Nz = Nz;
    m_Nt = Nt;
    m_Nvol = Nx * Ny * Nz * Nt;
    m_Nvol_pad = ceil_nwp(m_Nvol);
  }
 
  int site(const int x, const int y, const int z, const int t) const
  {  return m_Nx * (m_Ny * (m_Nz * t + z) + y) + x;  }
 
  int idx(const int in, const int Nin, const int ist, const int ex) const
  {
    int istx = ist + m_Nvol_pad * ex;
    return IDX2(Nin, in, istx);
  }
 
  int idx_G(const int idf, const int ist, const int ex) const
  {  return idx(idf, NDF, ist, ex);  }
 
  int idx_Gr(const int ic1, const int ic2, const int ist, const int ex) const
  { int idf = 2*(ic1 + NC *ic2);
    return idx(idf, NDF, ist, ex); }
 
  int idx_Gi(const int ic1, const int ic2, const int ist, const int ex) const
  { int idf = 1 + 2*(ic1 + NC *ic2);
    return idx(idf, NDF, ist, ex); }
 
  int idx_SP(const int in, const int ist, const int ex) const
  {  return idx(in, NVCD, ist, ex);  }
 
  int idx_SPr(const int ic, const int id, const int ist, const int ex) const
  { int in = 2*(ic + NC *id);
    return idx(in, NVCD, ist, ex); }
 
  int idx_SPi(const int ic, const int id, const int ist, const int ex) const
  { int in = 1 + 2*(ic + NC *id);
    return idx(in, NVCD, ist, ex); }
 
  int nvol_pad() const
  { return m_Nvol_pad; }
 
};
 
#endif