qdp_subset.cc

Go to the documentation of this file.

 
#include "qdp.h"
#include "qdp_util.h"
 
namespace QDP 
{
 
  //---------------------------------------------------------------------
  Set set_all;

  Subset all;

  Set rb3;

  Set rb;

  Set mcb;

  Subset even;

  Subset odd;
 

  class SetAllFunc : public SetFunc
  {
  public:
    int operator() (const multi1d<int>& coordinate) const {return 0;}
    int numSubsets() const {return 1;}
  };
 
  
  class SetRBFunc : public SetFunc
  {
  public:
    int operator() (const multi1d<int>& coordinate) const
      {
        int sum = 0;
        for(int m=0; m < coordinate.size(); ++m)
          sum += coordinate[m];
 
        return sum & 1;
      }
 
    int numSubsets() const {return 2;}
  };

  class SetRB3Func : public SetFunc
  {
  public:
    int operator() (const multi1d<int>& coordinate) const
      {
        if (coordinate.size() < 3) { 
          QDPIO::cerr << "Need at least 3d for 3d checkerboarding" << std::endl;
          QDP_abort(1);
        }
        int sum = 0;
        for(int m=0; m < 3; ++m)
          sum += coordinate[m];
 
        return sum & 1;
      }
 
    int numSubsets() const {return 2;}
  };
 
  
  class Set32CBFunc : public SetFunc
  {
  public:
    int operator() (const multi1d<int>& coordinate) const
      {
        int initial_color = 0;
        for(int m=Nd-1; m >= 0; --m)
          initial_color = (initial_color << 1) + (coordinate[m] & 1);
 
        int cb = 0;
        for(int m=0; m < Nd; ++m)
          cb += coordinate[m] >> 1;
 
        cb &= 1;
        return initial_color + (cb << Nd);
      }
 
    int numSubsets() const {return 1 << (Nd+1);}
  };
 

  void initDefaultSets()
  {
    // Initialize the red/black checkerboard
    rb.make(SetRBFunc());
 
    // Initialize the 3d red/black checkerboard.
    rb3.make(SetRB3Func());
 
    // Initialize the 32-style checkerboard
    mcb.make(Set32CBFunc());
 
    // The all set
    set_all.make(SetAllFunc());
 
    // The all subset
    all.make(set_all[0]);
 
    // COPY the rb[0] to the even subset
    even = rb[0];
 
    // COPY the rb[1] to the odd subset
    odd = rb[1];
  }
 
          
  //-----------------------------------------------------------------------------
  void Subset::make(bool _rep, int _start, int _end, multi1d<int>* ind, int cb, Set* _set)
  {
    ordRep    = _rep;
    startSite = _start;
    endSite   = _end;
    sub_index = cb;
    sitetable = ind;
    set       = _set;
  }

  void Subset::make(const Subset& s)
  {
    ordRep    = s.ordRep;
    startSite = s.startSite;
    endSite   = s.endSite;
    sub_index = s.sub_index;
    sitetable = s.sitetable;
    set       = s.set;
  }

  Subset& Subset::operator=(const Subset& s)
  {
    make(s);
    return *this;
  }
 
  //-----------------------------------------------------------------------------
  // = operator
  Set& Set::operator=(const Set& s)
  {
    sub = s.sub;
    lat_color = s.lat_color;
    sitetables = s.sitetables;
    return *this;
  }
 
  //-----------------------------------------------------------------------------
 
} // namespace QDP;