File size: 1,516 Bytes
158b61b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
/*
* Stacks.cpp
*
* Created on: 6 Nov 2015
* Author: hieu
*/
#include "Stacks.h"
#include "../../System.h"
#include "../Manager.h"
using namespace std;
namespace Moses2
{
namespace NSCubePruningPerMiniStack
{
Stacks::Stacks(const Manager &mgr)
:m_mgr(mgr)
{
}
Stacks::~Stacks()
{
}
void Stacks::Init(size_t numStacks)
{
m_stacks.resize(numStacks);
for (size_t i = 0; i < m_stacks.size(); ++i) {
m_stacks[i] = new (m_mgr.GetPool().Allocate<NSCubePruningMiniStack::Stack>()) NSCubePruningMiniStack::Stack(m_mgr);
}
}
std::ostream& operator<<(std::ostream &out, const Stacks &obj)
{
for (size_t i = 0; i < obj.GetSize(); ++i) {
const NSCubePruningMiniStack::Stack &stack = *obj.m_stacks[i];
out << stack.GetHypoSize() << " ";
}
return out;
}
void Stacks::Add(const Hypothesis *hypo, Recycler<Hypothesis*> &hypoRecycle)
{
size_t numWordsCovered = hypo->GetBitmap().GetNumWordsCovered();
//cerr << "numWordsCovered=" << numWordsCovered << endl;
NSCubePruningMiniStack::Stack &stack = *m_stacks[numWordsCovered];
stack.Add(hypo, hypoRecycle);
}
NSCubePruningMiniStack::MiniStack &Stacks::GetMiniStack(const Bitmap &newBitmap, const Range &pathRange)
{
size_t numWordsCovered = newBitmap.GetNumWordsCovered();
//cerr << "numWordsCovered=" << numWordsCovered << endl;
NSCubePruningMiniStack::Stack &stack = *m_stacks[numWordsCovered];
NSCubePruningMiniStack::Stack::HypoCoverage key(&newBitmap, pathRange.GetEndPos());
stack.GetMiniStack(key);
}
}
}
|