multi_array/example/for_each.hpp

#ifndef FOR_EACH_HPP
#define FOR_EACH_HPP

//
// for_each.hpp - Writing an algorithm to transform each element of
//   a multi_array
//

#include "boost/type.hpp"

template <typename Array, typename Element, typename Functor>
void for_each (const boost::type<Element>& type_dispatch,
               Array A, Functor& xform) {
  for_each(type_dispatch,A.begin(),A.end(),xform);
}

template <typename Element, typename Functor>
void for_each (const boost::type<Element>&,Element& Val, Functor& xform) {
  Val = xform(Val);
}

template <typename Element, typename Iterator, typename Functor>
void for_each (const boost::type<Element>& type_dispatch,
               Iterator begin, Iterator end,
               Functor& xform) {
  while (begin != end) {
    for_each(type_dispatch,*begin,xform);
    ++begin;
  }
}


template <typename Array, typename Functor>
void for_each (Array& A, Functor xform) {
  // Dispatch to the proper function
  for_each(boost::type<Array::element>(),A.begin(),A.end(),xform);
}


#endif // FOR_EACH_HPP