allocset.h

Go to the documentation of this file.

#ifndef ION_BASE_STLALLOC_ALLOCSET_H_
#define ION_BASE_STLALLOC_ALLOCSET_H_

#include <set>

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

namespace ion {
namespace base {


template <typename T, typename Compare = std::less<T> >
class AllocSet : public std::set<T, Compare, StlAllocator<T> > {
 public:
  typedef StlAllocator<T> AllocType;
  typedef std::set<T, Compare, AllocType> SetType;
  explicit AllocSet(const AllocatorPtr& alloc)
      : SetType(Compare(),
                AllocType(AllocationManager::GetNonNullAllocator(alloc))) {}
  explicit AllocSet(const Allocatable& owner)
      : SetType(Compare(), AllocType(owner.GetNonNullAllocator())) {}

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

  AllocSet(const AllocatorPtr& alloc, std::initializer_list<T> init)
      : SetType(init, Compare(),
                AllocType(AllocationManager::GetNonNullAllocator(alloc))) {}
  AllocSet(const Allocatable& owner, std::initializer_list<T> init)
      : SetType(init, Compare(),
                AllocType(owner.GetNonNullAllocator())) {}
};

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_STLALLOC_ALLOCSET_H_