word_alphabet.h

Go to the documentation of this file.

#ifndef __ZGEN_WORD_ALPHABET_H__
#define __ZGEN_WORD_ALPHABET_H__

#include <iostream>
#include <boost/unordered_map.hpp>
#include <boost/serialization/string.hpp>
#include <boost/serialization/vector.hpp>
#include <boost/serialization/version.hpp>
#include <boost/serialization/split_member.hpp>
#include <boost/serialization/singleton.hpp>
#include "alphabet.h"


namespace ZGen {

namespace Engine {

class WordAlphabet: public MutableAlphabet {
public:
  WordAlphabet();

  ~WordAlphabet();

  int insert(const char* name);

  const char* decode(int id) const;

  int encode(const char* name) const;

  bool save(std::ostream& ofs) const;

  bool load(std::istream& ifs);

  enum SPECIAL_WORD {
    NONE, BEGIN, END
  };

private:
  typedef boost::unordered_map<std::string, int> map_t;

  friend class boost::serialization::access;

  std::vector<std::string> id2name;

  map_t name2id;

  int global_id;

  template<class Archive>
  void serialize(Archive & ar, const unsigned int version) {
    ar & global_id;
    ar & name2id;
    ar & id2name;
  }
};


} //  end for namespace Engine

} //  end for namespace ZGen

#endif    //  end for __ZGEN_WORD_ALPHABET_H__