parallel/RCB.hpp

// -*- C++ -*-
//
//  ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
//                                   Fehmi Cirak
//                        California Institute of Technology
//                           (C) 2004 All Rights Reserved
//
//  <LicenseText>
//
//  ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
// EDITED BY MATT GALLOWAY FOR USE WITH DAPTA

#ifndef __RCB_HPP__
#define __RCB_HPP__

#include "parallel/DomainDecomposer.hpp"
#include "tools/Tools.hpp"

#include <vector>
#include <algorithm>

namespace DAPTA { // Define namespace DAPTA


template <typename V>
class RCBDecomposer : public DomainDecomposer<V> {
   public:
      typedef V* VertexHandle;
      
      RCBDecomposer(std::vector<VertexHandle> _vertices) : DomainDecomposer<V>(_vertices) {;}
      RCBDecomposer() {;}
      ~RCBDecomposer() {;}
      
      std::vector<int> decompose(unsigned numberOfPartitions, unsigned myNumber); 
   private:
      typedef typename std::vector<VertexHandle> PointCont;
      typedef typename PointCont::iterator       PointIt;
      
      struct RCBTree {
         RCBTree(PointIt dataBegin, PointIt dataEnd, int level) : 
               _dataBegin(dataBegin), _dataEnd(dataEnd), _level(level) {
            _childL=NULL;
            _childR=NULL;
         }
         
         ~RCBTree() {
            if (_childL) delete _childL;
            if (_childR) delete _childR;
         }
         
         // data stored at the nodes
         PointIt     _dataBegin;
         PointIt     _dataEnd;
         
         // internal rcb data
         int    _level; 
         struct RCBTree     *_childL;
         struct RCBTree     *_childR;
      };
      
      int  maxBoxInDimension   (PointIt begin, PointIt end);                                              
      void buildRCBTree        (RCBTree *tree, const int& maxLevel);                                      
      bool leaf                (RCBTree *tree) { return (!tree->_childL); }                               
      void collectLeafNodeData (RCBTree *tree, std::vector<PointIt>& begins, std::vector<PointIt>& ends); 
};

} // namespace DAPTA

#include "RCB.cpp"

#endif

---