allocunorderedset.h

Go to the documentation of this file.

#ifndef ION_BASE_STLALLOC_ALLOCUNORDEREDSET_H_
#define ION_BASE_STLALLOC_ALLOCUNORDEREDSET_H_

#include <unordered_set>

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

namespace ion {
namespace base {


template <typename Value,
          typename Hash = ::std::hash<Value>,
          typename Pred = ::std::equal_to<Value> >
class AllocUnorderedSet
    : public std::unordered_set<Value, Hash, Pred, StlAllocator<Value> > {
 public:
  typedef StlAllocator<Value> AllocType;
  typedef std::unordered_set<Value, Hash, Pred, AllocType> SetType;

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

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

 private:
  enum { kBucketCountHint = 10 };
};

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_STLALLOC_ALLOCUNORDEREDSET_H_