1 files changed, 90 insertions, 0 deletions
diff --git a/contrib/moses2/MemPoolAllocator.h b/contrib/moses2/MemPoolAllocator.h
new file mode 100644
index 000000000..6cc699893
--- /dev/null
+++ b/contrib/moses2/MemPoolAllocator.h
@@ -0,0 +1,90 @@
+#pragma once
+#include "MemPool.h"
+
+namespace Moses2
+{
+
+template<typename T>
+class MemPoolAllocator
+{
+public:
+  typedef T value_type;
+  typedef T* pointer;
+  typedef const T* const_pointer;
+  typedef T& reference;
+  typedef const T& const_reference;
+  typedef std::size_t size_type;
+  typedef std::ptrdiff_t difference_type;
+
+  template<class U>
+  struct rebind
+  {
+    typedef MemPoolAllocator<U> other;
+  };
+
+  MemPoolAllocator(Moses2::MemPool &pool) :
+      m_pool(pool)
+  {
+  }
+  MemPoolAllocator(const MemPoolAllocator &other) :
+      m_pool(other.m_pool)
+  {
+  }
+
+  template<class U>
+  MemPoolAllocator(const MemPoolAllocator<U>& other) :
+      m_pool(other.m_pool)
+  {
+  }
+
+  size_type max_size() const
+  {
+    return std::numeric_limits<size_type>::max();
+  }
+
+  void deallocate(pointer p, size_type n)
+  {
+    //std::cerr << "deallocate " << p << " " << n << std::endl;
+  }
+
+  pointer allocate(size_type n, std::allocator<void>::const_pointer hint = 0)
+  {
+    //std::cerr << "allocate " << n << " " << hint << std::endl;
+    pointer ret = m_pool.Allocate<T>(n);
+    return ret;
+  }
+
+  void construct(pointer p, const_reference val)
+  {
+    //std::cerr << "construct " << p << " " << n << std::endl;
+    new ((void *) p) T(val);
+  }
+
+  void destroy(pointer p)
+  {
+    //std::cerr << "destroy " << p << " " << n << std::endl;
+  }
+
+  // return address of values
+  pointer address (reference value) const {
+    return &value;
+  }
+  const_pointer address (const_reference value) const {
+    return &value;
+  }
+
+  bool operator==(const MemPoolAllocator<T> &allocator) const {
+    return true;
+  }
+
+  bool operator!=(const MemPoolAllocator<T> &allocator) const {
+    return false;
+  }
+
+  MemPool &m_pool;
+protected:
+};
+
+}
+
+