diff options
Diffstat (limited to 'phrase-extract/postprocess-egret-forests/PostprocessEgretForests.h')
-rw-r--r-- | phrase-extract/postprocess-egret-forests/PostprocessEgretForests.h | 18 |
1 files changed, 5 insertions, 13 deletions
diff --git a/phrase-extract/postprocess-egret-forests/PostprocessEgretForests.h b/phrase-extract/postprocess-egret-forests/PostprocessEgretForests.h index 95da24c71..51970084e 100644 --- a/phrase-extract/postprocess-egret-forests/PostprocessEgretForests.h +++ b/phrase-extract/postprocess-egret-forests/PostprocessEgretForests.h @@ -4,6 +4,8 @@ #include <ostream> #include <string> +#include "syntax-common/tool.h" + namespace MosesTraining { namespace Syntax @@ -14,31 +16,21 @@ namespace PostprocessEgretForests struct Options; class SplitPointFileParser; -class PostprocessEgretForests +class PostprocessEgretForests : public Tool { public: - PostprocessEgretForests() : m_name("postprocess-egret-forests") {} - - void Error(const std::string &) const; - - const std::string &GetName() const { - return m_name; - } + PostprocessEgretForests() : Tool("postprocess-egret-forests") {} - int Main(int argc, char *argv[]); + virtual int Main(int argc, char *argv[]); private: void OneBestTree(std::istream &, std::ostream &, SplitPointFileParser *, const Options &); - void OpenInputFileOrDie(const std::string &, std::ifstream &); - void ProcessForest(std::istream &, std::ostream &, SplitPointFileParser *, const Options &); void ProcessOptions(int, char *[], Options &) const; - - std::string m_name; }; } // namespace PostprocessEgretForests |