SetIterator.hpp

Go to the documentation of this file.

#ifndef MB_SETITERATOR_HPP
#define MB_SETITERATOR_HPP

#include "moab/Interface.hpp"

namespace moab {

class Core;
    
class SetIterator
{
public:

  friend class Core;
  
  virtual ~SetIterator();

  inline EntityHandle ent_set() const {return entSet;};
  
  inline unsigned int chunk_size() const {return chunkSize;};

  inline EntityType ent_type() const {return entType;};

  inline int ent_dimension() const {return entDimension;};

  virtual ErrorCode get_next_arr(std::vector<EntityHandle> &arr,
                                 bool &atend) = 0;
  
  virtual ErrorCode reset() = 0;
  
protected:

  inline SetIterator(Core *core, EntityHandle eset, unsigned int chunk_sz, 
                     EntityType ent_tp, int ent_dim, bool check_valid = false) 
          : myCore(core), entSet(eset), chunkSize(chunk_sz), 
            entType(ent_tp), entDimension(ent_dim), checkValid(check_valid) {};
  
  Core *myCore;
  
  EntityHandle entSet;
  
  unsigned int chunkSize;
  
  EntityType entType;
  
  int entDimension;

  bool checkValid;

};

class RangeSetIterator : public SetIterator
{
public:
  friend class Core;
  
  virtual ~RangeSetIterator();
  
  virtual ErrorCode get_next_arr(std::vector<EntityHandle> &arr,
                                 bool &atend);
  
  virtual ErrorCode reset();
  
protected:
  RangeSetIterator(Core *core, EntityHandle ent_set, int chunk_size, 
                   EntityType ent_type, int ent_dimension, bool check_valid = false);
  
private:
  ErrorCode get_next_by_type(const EntityHandle *&ptr, int count,
                             std::vector<EntityHandle> &arr, bool &atend);
  
  ErrorCode get_next_by_dimension(const EntityHandle *&ptr, int count,
                                  std::vector<EntityHandle> &arr, bool &atend);
  
  ErrorCode build_pair_vec();
  
  EntityHandle iterPos;

  EntityHandle *pairPtr;

  int numPairs;
};
    
class VectorSetIterator : public SetIterator
{
public:
  friend class Core;

  virtual ErrorCode get_next_arr(std::vector<EntityHandle> &arr,
                                 bool &atend);
  
  virtual ErrorCode reset();
  
  inline ErrorCode decrement(int num);
  
protected:
  inline VectorSetIterator(Core *core, EntityHandle eset, int chunk_sz, 
                           EntityType ent_tp, int ent_dim, bool check_valid = false)
          : SetIterator(core, eset, chunk_sz, ent_tp, ent_dim, check_valid),
            iterPos(0)
      {}
  
private:
  int iterPos;
  
};

inline ErrorCode VectorSetIterator::decrement(int num) 
{
  iterPos -= num; 
  return (iterPos < 0 ? MB_FAILURE : MB_SUCCESS);
}
    
} // namespace moab

#endif