keymap.h

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-3-Clause
#ifndef TTG_BASE_KEYMAP_H
#define TTG_BASE_KEYMAP_H
 
#include <type_traits>
#include "ttg/util/meta.h"
#include "ttg/util/hash.h"
 
namespace ttg {
  namespace detail {
 
    template <typename keyT, typename Enabler = void>
    struct default_keymap_impl;
    template <typename keyT>
    struct default_keymap_impl<
        keyT, std::enable_if_t<meta::has_ttg_hash_specialization_v<keyT> || meta::is_void_v<keyT>>> {
      default_keymap_impl() = default;
      default_keymap_impl(int world_size) : world_size(world_size) {}
 
      template <typename Key = keyT>
      std::enable_if_t<!meta::is_void_v<Key>,int>
      operator()(const Key &key) const {
        if (world_size == 1) {
          return 0;
        } else {
          return ttg::hash<keyT>{}(key) % world_size;
        }
      }
      template <typename Key = keyT>
      std::enable_if_t<meta::is_void_v<Key>,int>
      operator()() const { return 0; }
 
     private:
      int world_size;
    };
 
 
    template <typename keyT>
    struct default_priomap_impl {
      default_priomap_impl() = default;
 
      template <typename Key = keyT>
      std::enable_if_t<!meta::is_void_v<Key>,int>
      operator()(const Key &key) const { return 0; }
      template <typename Key = keyT>
      std::enable_if_t<meta::is_void_v<Key>,int>
      operator()() const { return 0; }
    };
 
  }  // namespace detail
 
} // namespace ttg
 
#endif // TTG_BASE_KEYMAP_H