| #pragma once |
|
|
| #include <iostream> |
| #include <string> |
| #include <map> |
| #include <vector> |
| #include <boost/shared_ptr.hpp> |
| #include <boost/make_shared.hpp> |
| #include "util/generator.hh" |
| #include "util/exception.hh" |
|
|
| namespace MosesTuning |
| { |
|
|
| class InternalTree; |
| typedef boost::shared_ptr<InternalTree> TreePointer; |
| typedef int NTLabel; |
|
|
| class InternalTree |
| { |
| std::string m_value; |
| std::vector<TreePointer> m_children; |
| bool m_isTerminal; |
| public: |
| InternalTree(const std::string & line, const bool terminal = false); |
| InternalTree(const InternalTree & tree): |
| m_value(tree.m_value), |
| m_isTerminal(tree.m_isTerminal) { |
| const std::vector<TreePointer> & children = tree.m_children; |
| for (std::vector<TreePointer>::const_iterator it = children.begin(); it != children.end(); it++) { |
| m_children.push_back(boost::make_shared<InternalTree>(**it)); |
| } |
| } |
| size_t AddSubTree(const std::string & line, size_t start); |
|
|
| std::string GetString(bool start = true) const; |
| void Combine(const std::vector<TreePointer> &previous); |
| const std::string & GetLabel() const { |
| return m_value; |
| } |
|
|
| size_t GetLength() const { |
| return m_children.size(); |
| } |
| std::vector<TreePointer> & GetChildren() { |
| return m_children; |
| } |
|
|
| bool IsTerminal() const { |
| return m_isTerminal; |
| } |
|
|
| bool IsLeafNT() const { |
| return (!m_isTerminal && m_children.size() == 0); |
| } |
| }; |
|
|
| |
| $generator(leafNT) |
| { |
| std::vector<TreePointer>::iterator it; |
| InternalTree* tree; |
| leafNT(InternalTree* root = 0): tree(root) {} |
| $emit(std::vector<TreePointer>::iterator) |
| for (it = tree->GetChildren().begin(); it !=tree->GetChildren().end(); ++it) { |
| if (!(*it)->IsTerminal() && (*it)->GetLength() == 0) { |
| $yield(it); |
| } else if ((*it)->GetLength() > 0) { |
| if ((*it).get()) { |
| $restart(tree = (*it).get()); |
| } |
| } |
| } |
| $stop; |
| }; |
|
|
| } |
