allocmap.h

Go to the documentation of this file.

#ifndef ION_BASE_STLALLOC_ALLOCMAP_H_
#define ION_BASE_STLALLOC_ALLOCMAP_H_

#include <map>

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

namespace ion {
namespace base {


template <typename K, typename V, typename Compare = std::less<K> >
class AllocMap : public std::map<K, V, Compare,
                                 StlAllocator<std::pair<const K, V> > > {
 public:
  typedef std::pair<const K, V> PairType;
  typedef StlAllocator<PairType> AllocType;
  typedef std::map<K, V, Compare, AllocType> MapType;
  explicit AllocMap(const AllocatorPtr& alloc)
      : MapType(Compare(),
                AllocType(AllocationManager::GetNonNullAllocator(alloc))) {}
  explicit AllocMap(const Allocatable& owner)
      : MapType(Compare(), AllocType(owner.GetNonNullAllocator())) {}

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

  AllocMap(const AllocatorPtr& alloc, std::initializer_list<PairType> init)
      : MapType(init, Compare(),
                AllocType(AllocationManager::GetNonNullAllocator(alloc))) {}
  AllocMap(const Allocatable& owner, std::initializer_list<PairType> init)
      : MapType(init, Compare(),
                AllocType(owner.GetNonNullAllocator())) {}
};

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_STLALLOC_ALLOCMAP_H_