qdp_subset.h

Go to the documentation of this file.

// -*- C++ -*-
 
#ifndef QDP_SUBSET_H
#define QDP_SUBSET_H
 
namespace QDP {

 
//-----------------------------------------------------------------------
class SetFunc
{
public:
  // Virtual destructor to stop compiler warnings - no cleanup needed
  virtual ~SetFunc() {}
  virtual int operator() (const multi1d<int>& coordinate) const = 0;
  virtual int numSubsets() const = 0;
};
 
//-----------------------------------------------------------------------
// Forward declaration
class Set;
 
//-----------------------------------------------------------------------
class Subset 
{
public:
  Subset() {}

  Subset(const Subset& s):
    ordRep(s.ordRep), startSite(s.startSite), endSite(s.endSite), 
    sub_index(s.sub_index), sitetable(s.sitetable), set(s.set)
    {}
 
  // Simple constructor
  void make(const Subset& s);

  virtual ~Subset() {}

  Subset& operator=(const Subset& s);

  int color() const {return sub_index;}
 
protected:
  // Simple constructor
  void make(bool rep, int start, int end, multi1d<int>* ind, int cb, Set* set);
 
private:
  bool ordRep;
  int startSite;
  int endSite;
  int sub_index;

  multi1d<int>* sitetable;

  Set *set;
 
public:
  inline bool hasOrderedRep() const {return ordRep;}
  inline int start() const {return startSite;}
  inline int end() const {return endSite;}
 
  const multi1d<int>& siteTable() const {return *sitetable;}
  inline int numSiteTable() const {return sitetable->size();}

  const Set& getSet() const { return *set; }
 
  friend class Set;
};
 
 
//-----------------------------------------------------------------------
class Set 
{
public:
  Set() {}

  Set(const SetFunc& fn) {make(fn);}

  void make(const SetFunc& fn);

  const Subset& operator[](int subset_index) const {return sub[subset_index];}

  int numSubsets() const {return sub.size();}

  virtual ~Set() {}

  Set& operator=(const Set& s);
 
protected:
  multi1d<Subset> sub;

  multi1d<int> lat_color;

  multi1d<multi1d<int> > sitetables;
 
public:
  const multi1d<int>& latticeColoring() const {return lat_color;}
};
 
 
 
//-----------------------------------------------------------------------
extern Subset all;
 

extern Set rb3;

extern Set rb;

extern Set mcb;

extern Subset even;

extern Subset odd;
 
 // end of group subsetss
 
} // namespace QDP
 
#endif