neighborhood_util_impl.hpp

Go to the documentation of this file.

 
#include "neighborhood_util.h"
#include "grid.h"
#include "lib_grid/selector.h"
#include "lib_grid/algorithms/selection_util.h"
 
namespace ug {
  template <typename TElem>
  void GetNeighborhood
  (
    Grid& grid,
    size_t extSize,
    TElem* elem,
    typename geometry_traits<TElem>::const_iterator& begin,
    typename geometry_traits<TElem>::const_iterator& end
  ) {
    Selector sel(grid);
    typedef typename geometry_traits<TElem>::const_iterator Iter;
    sel.template select<TElem>(elem);
    grid.begin_marking();
    for (size_t extIters = 0; extIters < extSize; ++extIters)
    {
      SelectAssociatedGridObjects(sel, 1);
      for(size_t lvl = 0; lvl < sel.num_levels(); ++lvl){
        for(Iter iter = sel.template begin<TElem>(lvl);
            iter != sel.template end<TElem>(lvl); ++iter)
        {
          TElem* el = *iter;
          if(!grid.is_marked(el)){
            grid.mark(el);
            typename Grid::traits<TElem>::secure_container elemsOut;
            grid.template associated_elements<TElem>(elemsOut, el);
            size_t size = elemsOut.size();
            for (size_t i = 0; i < size; i++) {
              sel.template select<TElem>(elemsOut[i], 1);
            }
          }
        }
      }
    }
    grid.end_marking();
    begin = sel.template begin<TElem>();
    end = sel.template end<TElem>();
  }
} // namespace ug