ThreadGroups.h

Go to the documentation of this file.

// $Id$
//==============================================================================
//==============================================================================
 
#ifndef _THREAD_GROUPS_H
#define _THREAD_GROUPS_H
 
#include <vector>
#include <cstddef>
 
 
class ThreadGroups
{
  typedef std::vector<bool>   BoolVec; 
  typedef std::vector<int>    IntVec;  
  typedef std::vector<IntVec> IntMat;  
 
public:
  enum StripDirection { NONE, U, V, W, ANY };
 
  explicit ThreadGroups(StripDirection dir = ANY) : stripDir(dir) {}
 
  void calcGroups(const BoolVec& el1, const BoolVec& el2,
                  int p1, int p2);
  void calcGroups(int nel1, int nel2, int minsize);
 
  void calcGroups(const BoolVec& el1, const BoolVec& el2, const BoolVec& el3,
                  int p1, int p2, int p3);
  void calcGroups(int nel1, int nel2, int nel3, int minsize);
  void oneGroup(size_t nel);
  void oneGroup(size_t nel, const IntVec& elms);
  void oneStripe(size_t nel);
  void oneStripe(size_t nel, const IntVec& elms);
 
  void applyMap(const IntVec& map);
 
  size_t size() const { return tg[1].empty() ? 1 : 2; }
  bool empty() const { return tg[0].empty() && tg[1].empty(); }
  const IntMat& operator[](int i) const { return tg[i]; }
  IntMat& operator[](int i) { return tg[i]; }
 
  ThreadGroups filter(const IntVec& elmList) const;
 
  void analyzeUnstruct(bool listAllSizes = false) const;
 
protected:
  static StripDirection getStripDirection(int nel1, int nel2,
                                          int parts);
  static StripDirection getStripDirection(int nel1, int nel2, int nel3,
                                          int parts);
 
public:
  StripDirection stripDir; 
 
private:
  IntMat tg[2]; 
};
 
#endif