tree_analyzer.h

Go to the documentation of this file.

#ifndef META_TREE_ANALYZER_H_
#define META_TREE_ANALYZER_H_

#include "corpus/document.h"
#include "analyzers/analyzer.h"
#include "analyzers/analyzer_factory.h"
#include "parser/analyzers/featurizers/tree_featurizer.h"
#include "parser/sr_parser.h"
#include "sequence/perceptron.h"
#include "util/clonable.h"

namespace meta
{
namespace analyzers
{

class tree_analyzer : public util::clonable<analyzer, tree_analyzer>
{
  public:
    tree_analyzer(std::unique_ptr<token_stream> stream,
                  const std::string& tagger_prefix,
                  const std::string& parser_prefix);

    tree_analyzer(const tree_analyzer& other);

    void tokenize(corpus::document& doc) override;

    void add(std::unique_ptr<const tree_featurizer> featurizer);

    const static std::string id;

  private:
    std::shared_ptr<std::vector<std::unique_ptr<const tree_featurizer>>>
        featurizers_;

    std::unique_ptr<token_stream> stream_;

    std::shared_ptr<const sequence::perceptron> tagger_;

    std::shared_ptr<const parser::sr_parser> parser_;
};

template <>
std::unique_ptr<analyzer> make_analyzer<tree_analyzer>(const cpptoml::table&,
                                                       const cpptoml::table&);
}

namespace parser
{
void register_analyzers();
}
}
#endif