allocunorderedmap.h

Go to the documentation of this file.

#ifndef ION_BASE_STLALLOC_ALLOCUNORDEREDMAP_H_
#define ION_BASE_STLALLOC_ALLOCUNORDEREDMAP_H_

#include <unordered_map>

#include "ion/base/allocatable.h"
#include "ion/base/allocationmanager.h"
#include "ion/base/stlalloc/stlallocator.h"

namespace ion {
namespace base {


template <typename Key,
          typename Value,
          typename Hash = ::std::hash<Key>,
          typename Pred = ::std::equal_to<Key> >
class AllocUnorderedMap : public std::unordered_map<
  Key, Value, Hash, Pred, StlAllocator<std::pair<const Key, Value> > > {
 public:
  typedef std::pair<const Key, Value> PairType;
  typedef StlAllocator<PairType> AllocType;
  typedef std::unordered_map<Key, Value, Hash, Pred, AllocType> MapType;

  explicit AllocUnorderedMap(const AllocatorPtr& alloc)
      : MapType(kBucketCountHint, Hash(), Pred(),
                AllocType(AllocationManager::GetNonNullAllocator(alloc))) {}
  explicit AllocUnorderedMap(const Allocatable& owner)
      : MapType(kBucketCountHint, Hash(), Pred(),
                AllocType(owner.GetNonNullAllocator())) {}

  template <typename ContainerT>
  AllocUnorderedMap(const AllocatorPtr& alloc, const ContainerT& from)
      : MapType(from.begin(), from.end(), kBucketCountHint, Hash(), Pred(),
                AllocType(AllocationManager::GetNonNullAllocator(alloc))) {}
  template <typename ContainerT>
  AllocUnorderedMap(const Allocatable& owner, const ContainerT& from)
      : MapType(from.begin(), from.end(), kBucketCountHint, Hash(), Pred(),
                AllocType(owner.GetNonNullAllocator())) {}

 private:
  enum { kBucketCountHint = 10 };
};

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_STLALLOC_ALLOCUNORDEREDMAP_H_