| |
| |
| |
| |
| |
| |
| #pragma once |
| #include <boost/unordered_map.hpp> |
| #include <boost/unordered_set.hpp> |
| #include <deque> |
| #include "../Hypothesis.h" |
| #include "../../TypeDef.h" |
| #include "../../Vector.h" |
| #include "../../MemPool.h" |
| #include "../../Recycler.h" |
| #include "../../legacy/Util2.h" |
|
|
| namespace Moses2 |
| { |
|
|
| class Manager; |
|
|
| namespace NSCubePruningBitmapStack |
| { |
| typedef Vector<const Hypothesis*> Hypotheses; |
|
|
| class MiniStack |
| { |
| public: |
| typedef boost::unordered_set<const Hypothesis*, |
| UnorderedComparer<Hypothesis>, |
| UnorderedComparer<Hypothesis> |
| > _HCType; |
|
|
| MiniStack(const Manager &mgr); |
|
|
| StackAdd Add(const Hypothesis *hypo); |
|
|
| _HCType &GetColl() { |
| return m_coll; |
| } |
|
|
| const _HCType &GetColl() const { |
| return m_coll; |
| } |
|
|
| void Clear(); |
|
|
| Hypotheses &GetSortedAndPruneHypos(const Manager &mgr) const; |
|
|
| protected: |
| _HCType m_coll; |
| mutable Hypotheses *m_sortedHypos; |
|
|
| void SortAndPruneHypos(const Manager &mgr) const; |
|
|
| }; |
|
|
| |
| class Stack |
| { |
| protected: |
|
|
|
|
| public: |
| typedef std::pair<const Bitmap*, size_t> HypoCoverage; |
| |
| typedef boost::unordered_map<HypoCoverage, Hypotheses*> SortedHypos; |
|
|
| typedef const Bitmap* HypoCoverageInternal; |
| typedef boost::unordered_map<HypoCoverageInternal, MiniStack* |
| ,boost::hash<HypoCoverageInternal> |
| ,std::equal_to<HypoCoverageInternal> |
| > Coll; |
|
|
|
|
| Stack(const Manager &mgr); |
| virtual ~Stack(); |
|
|
| size_t GetHypoSize() const; |
|
|
| Coll &GetColl() { |
| return m_coll; |
| } |
| const Coll &GetColl() const { |
| return m_coll; |
| } |
|
|
| void Add(const Hypothesis *hypo, Recycler<Hypothesis*> &hypoRecycle); |
|
|
| MiniStack &GetMiniStack(const HypoCoverageInternal &key); |
|
|
| std::vector<const Hypothesis*> GetBestHypos(size_t num) const; |
| void Clear(); |
|
|
| SortedHypos GetSortedAndPruneHypos(const Manager &mgr) const; |
| void SortAndPruneHypos(const Manager &mgr, Hypotheses &hypos) const; |
|
|
| void DebugCounts(); |
|
|
| protected: |
| const Manager &m_mgr; |
| Coll m_coll; |
|
|
| std::deque<MiniStack*> m_miniStackRecycler; |
|
|
|
|
| }; |
|
|
| } |
|
|
| } |
|
|
|
|
|
|
