makeValueIndex.h

Go to the documentation of this file.

#ifndef LARCOREALG_COREUTILS_MAKEVALUEINDEX_H
#define LARCOREALG_COREUTILS_MAKEVALUEINDEX_H

// LArSoft libraries
#include "larcorealg/CoreUtils/DebugUtils.h"
#include "larcorealg/CoreUtils/fromFutureImport.h" // util::pre_std::identity<>

// External libraries
#include "range/v3/view/enumerate.hpp"

// C/C++ standard library
#include <cstddef> // std::size_t
#include <map>
#include <stdexcept>   // std::runtime_error
#include <string>      // std::to_string()
#include <type_traits> // std::invoke_result_t, std::remove_reference_t

namespace util {

  template <typename Coll, typename Extractor>
  decltype(auto) makeValueIndex(Coll const& coll, Extractor getter);

  template <typename Coll>
  auto makeValueIndex(Coll const& coll)
  {
    return makeValueIndex(coll, util::pre_std::identity());
  }

} // namespace util

//------------------------------------------------------------------------------
//--- template implementation
//------------------------------------------------------------------------------
template <typename Coll, typename Extractor>
decltype(auto) util::makeValueIndex(Coll const& coll, Extractor getter)
{
  using Value_t = typename Coll::value_type;
  using Key_t = std::remove_reference_t<std::invoke_result_t<Extractor, Value_t>>;

  using Map_t = std::map<Key_t, std::size_t>;

  Map_t index;
  for (auto&& [iValue, collValue] : coll | ranges::views::enumerate) {

    Key_t const& key = getter(collValue);
    auto const iKey = index.lower_bound(key);
    if ((iKey != index.end()) && (iKey->first == key)) {
      // no guarantee that `key` supports `std::to_string()`: print only indices
      throw std::runtime_error(std::string(__func__) + ": element #" + std::to_string(iValue) +
                               " has the same key as #" + std::to_string(iKey->second));
    }
    index.emplace_hint(iKey, key, iValue);
  }

  return index;
}

//------------------------------------------------------------------------------

#endif // LARCOREALG_COREUTILS_MAKEVALUEINDEX_H