utils-inl.h

Go to the documentation of this file.

#pragma once
#include <algorithm>
#include <iterator>
#include <type_traits>
#include <vector>

namespace tc {
namespace autotune {

namespace detail {
template <typename Vector>
void mergeVectorsImpl(Vector&) {}

template <typename Vector, typename... Vectors>
void mergeVectorsImpl(Vector& sink, Vector&& v, Vectors&&... vs) {
  mergeVectorsImpl(sink, std::forward<Vectors>(vs)...);

  if (std::is_rvalue_reference<decltype(v)>::value) {
    sink.reserve(sink.size() + v.size());
    std::move(v.begin(), v.end(), std::back_inserter(sink));
  } else {
    sink.insert(sink.end(), v.begin(), v.end());
  }
}

} // namespace detail

template <typename Vector, typename... Vectors>
Vector mergeVectors(Vector&& v, Vectors&&... vs) {
  Vector merged;
  detail::mergeVectorsImpl(
      merged, std::forward<Vector>(v), std::forward<Vectors>(vs)...);
  std::sort(merged.begin(), merged.end());
  merged.erase(std::unique(merged.begin(), merged.end()), merged.end());
  return merged;
}

} // namespace autotune
} // namespace tc