state_analyzer.h

Go to the documentation of this file.

#ifndef META_PARSER_STATE_ANALYZER_H_
#define META_PARSER_STATE_ANALYZER_H_

#include "parser/sr_parser.h"
#include "parser/state.h"

namespace meta
{
namespace parser
{

class sr_parser::state_analyzer
{
  public:
    feature_vector featurize(const state& state) const;

  private:
    void unigram_featurize(const state& state, feature_vector& feats) const;

    void bigram_featurize(const state& state, feature_vector& feats) const;

    void trigram_featurize(const state& state, feature_vector& feats) const;

    void children_featurize(const state& state, feature_vector& feats) const;

    void dependents_featurize(const state& state, feature_vector& feats) const;

    void unigram_stack_feats(const node* n, std::string prefix,
                             feature_vector& feats) const;

    void bigram_features(const node* n1, std::string name1, const node* n2,
                         std::string name2, feature_vector& feats) const;

    void child_feats(const node* n, std::string prefix, feature_vector& feats,
                     bool doubs) const;
};
}
}

#endif