postag_constrain.h

Go to the documentation of this file.

#ifndef __ZGEN_KNOWLEDGE_POSTAG_CONSTRAIN_H__
#define __ZGEN_KNOWLEDGE_POSTAG_CONSTRAIN_H__

#include <iostream>
#include <vector>
#include <boost/unordered_map.hpp>
#include "settings.h"
#include "types/instance.h"

namespace ZGen {

namespace Knowledge {

class PostagConstrain {
public:
  PostagConstrain(const char* postag_dict_path);

  void get_possible_tags(const char* word,
      std::vector<postag_t>& possible_tags) const;

  void get_possible_tags(const word_t word,
      std::vector<postag_t>& possible_tags) const;

private:
  void init_postag_constraints_from_file(std::istream& ifs);

  void init_default_postag_constraints();

  typedef std::bitset<kNumberOfPostags> postag_mask_t;
  typedef boost::unordered_map<word_t, postag_mask_t> map_t;

  map_t postag_constraints;
};

}

} //  end for namespace

#endif  //  end for __ZGEN_KNOWLEDGE_POSTAG_CONSTRAIN_H__