File size: 1,077 Bytes
fd49381 |
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 |
#ifndef SEARCH_CONTEXT__
#define SEARCH_CONTEXT__
#include "search/config.hh"
#include "search/vertex.hh"
#include <boost/pool/object_pool.hpp>
namespace search {
class ContextBase {
public:
explicit ContextBase(const Config &config) : config_(config) {}
VertexNode *NewVertexNode() {
VertexNode *ret = vertex_node_pool_.construct();
assert(ret);
return ret;
}
void DeleteVertexNode(VertexNode *node) {
vertex_node_pool_.destroy(node);
}
unsigned int PopLimit() const { return config_.PopLimit(); }
Score LMWeight() const { return config_.LMWeight(); }
const Config &GetConfig() const { return config_; }
private:
boost::object_pool<VertexNode> vertex_node_pool_;
Config config_;
};
template <class Model> class Context : public ContextBase {
public:
Context(const Config &config, const Model &model) : ContextBase(config), model_(model) {}
const Model &LanguageModel() const { return model_; }
private:
const Model &model_;
};
} // namespace search
#endif // SEARCH_CONTEXT__
|