1 files changed, 40 insertions, 0 deletions
diff --git a/contrib/moses2/legacy/Bitmaps.h b/contrib/moses2/legacy/Bitmaps.h
new file mode 100644
index 000000000..d8207b59e
--- /dev/null
+++ b/contrib/moses2/legacy/Bitmaps.h
@@ -0,0 +1,40 @@
+#pragma once
+
+#include <boost/unordered_set.hpp>
+#include <boost/unordered_map.hpp>
+#include <set>
+#include <stack>
+#include "Bitmap.h"
+#include "Util2.h"
+
+namespace Moses2
+{
+class MemPool;
+
+class Bitmaps
+{
+  typedef boost::unordered_map<const Range*, const Bitmap*> NextBitmaps;
+  typedef boost::unordered_map<const Bitmap*, NextBitmaps,
+      UnorderedComparer<Bitmap>, UnorderedComparer<Bitmap> > Coll;
+  //typedef std::set<const Bitmap*, OrderedComparer<Bitmap> > Coll;
+  Coll m_coll;
+  Bitmap *m_initBitmap;
+
+  MemPool &m_pool;
+  std::stack<Bitmap*> m_recycler;
+
+  const Bitmap &GetNextBitmap(const Bitmap &bm, const Range &range);
+public:
+  Bitmaps(MemPool &pool);
+  virtual ~Bitmaps();
+  void Init(size_t inputSize, const std::vector<bool> &initSourceCompleted);
+
+  const Bitmap &GetInitialBitmap() const
+  {
+    return *m_initBitmap;
+  }
+  const Bitmap &GetBitmap(const Bitmap &bm, const Range &range);
+};
+
+}
+