Generate.h

#ifndef DASH__ALGORITHM__GENERATE_H__
#define DASH__ALGORITHM__GENERATE_H__

#include <dash/algorithm/LocalRange.h>
#include <dash/algorithm/Operation.h>
#include <dash/iterator/GlobIter.h>

#include <dash/dart/if/dart_communication.h>

#include <algorithm>

namespace dash {

template <typename GlobInputIt, class UnaryFunction>
void generate(
    GlobInputIt first,
    GlobInputIt last,
    UnaryFunction gen)
{
  using iterator_traits = dash::iterator_traits<GlobInputIt>;
  static_assert(
      iterator_traits::is_global_iterator::value,
      "must be a global iterator");
  auto lrange = dash::local_range(first, last);
  auto lfirst = lrange.begin;
  auto llast  = lrange.end;

  std::generate(lfirst, llast, gen);
}

template <typename GlobInputIt, class UnaryFunction>
void generate_with_index(
    GlobInputIt first,
    GlobInputIt last,
    UnaryFunction gen)
{
  using iterator_traits = dash::iterator_traits<GlobInputIt>;
  static_assert(
      iterator_traits::is_global_iterator::value,
      "must be a global iterator");
  auto index_range  = dash::local_index_range(first, last);
  auto lbegin_index = index_range.begin;
  auto lend_index   = index_range.end;

  if (lbegin_index != lend_index) {
    // Pattern from global begin iterator:
    auto& pattern      = first.pattern();
    auto  first_offset = first.pos();
    // Iterate local index range:
    for (auto lindex = lbegin_index; lindex != lend_index; ++lindex) {
      auto gindex     = pattern.global(lindex);
      auto element_it = first + (gindex - first_offset);
      *element_it     = gen(gindex);
    }
  }
}

}  // namespace dash

#endif  // DASH__ALGORITHM__GENERATE_H__