UT_ConcurrentHashMap.h

Go to the documentation of this file.

/*
 * PROPRIETARY INFORMATION.  This software is proprietary to
 * Side Effects Software Inc., and is not to be reproduced,
 * transmitted, or disclosed in any way without written permission.
 *
 * NAME:        UT_ConcurrentHashMap.h (UT Library, C++)
 *
 * COMMENTS:    Wrapper around Intel TBB's concurrent_hash_map container.
 *              Microsoft will probably have an implementation of this in their
 *              Concurrent Runtime Parallel Patterns Library (PPL) since they
 *              already have concurrent_vector and concurrent_queue.
 *
 */

#ifndef __UT_CONCURRENTHASHMAP_H_INCLUDED__
#define __UT_CONCURRENTHASHMAP_H_INCLUDED__

#include "UT_API.h"
#include <SYS/SYS_Types.h>
// IWYU pragma: begin_exports
#include <hboost/container_hash/hash.hpp>
#include <tbb/concurrent_hash_map.h>
// IWYU pragma: end_exports
#include <utility>

namespace UT::detail
{
template <typename Key>
class HashCompare
{
public:
    std::size_t hash(const Key &a) const
    {
        return myHasher(a);
    }
    bool equal(const Key &a, const Key &b) const
    {
        return myEqual(a, b);
    }
private:
    hboost::hash<Key> myHasher;
    std::equal_to<Key> myEqual;
};
}

template
<
    typename K,
    typename T,
    typename H = UT::detail::HashCompare<K>,
    typename A = tbb::tbb_allocator<std::pair<const K, T>>
>
using UT_ConcurrentHashMap = tbb::concurrent_hash_map<K, T, H, A>;

template<typename K, typename V, typename H, typename A>
int64
UTgetMemoryUsage(const UT_ConcurrentHashMap<K, V, H, A> &map, const bool inclusive)
{
    int64 mem = 0;
    if (inclusive)
        mem += sizeof(map);

    // The size of the buckets (a mutex and a pointer to the first node)
    mem += map.bucket_count()*(sizeof(intptr_t) + sizeof(void *));
    // The size of the nodes (a mutex, a pointer to the next node,
    // and an std::pair<K,V> forthe key and value)
    mem += map.size()*(sizeof(intptr_t) + sizeof(void *) + sizeof(std::pair<K,V>));

    return mem;
}

#endif // __UT_CONCURRENTHASHMAP_H_INCLUDED__