state.h

Go to the documentation of this file.

#ifndef __ZGEN_SHITREDUCE_TYPES_STATE_H__
#define __ZGEN_SHITREDUCE_TYPES_STATE_H__

#include <iostream>
#include <bitset>
#include <cstring>
#include "settings.h"
#include "shiftreduce/settings.h"
#include "shiftreduce/types/action.h"

namespace ZGen {

namespace ShiftReduce {

class StateItem{
public:
  // The default constructor,
  StateItem();

  // The constructor with sentence reference.
  StateItem(const dependency_t* ref);

  void set_reference(const dependency_t* _ref);

  void clear();

public:
  StateItem & operator = (const StateItem& item);

  bool operator < (const StateItem& other) const;

  bool operator > (const StateItem& other) const;

  friend std::ostream & operator << (std::ostream& os, const StateItem& item);

public:
  int top0;   
  int top1;   
  const dependency_t* ref;    
  const StateItem* previous;  
  floatval_t score;           

  std::vector<int> stack;

  std::vector< word_t >   word_sequence;

  std::vector< postag_t > postag_sequence;

  std::bitset<kMaxNumberOfWords> buffer;

  int rank[kMaxNumberOfWords];

  int postags[kMaxNumberOfWords];

  int heads[kMaxNumberOfWords];

  int deprels[kMaxNumberOfWords];

  int nr_left_children[kMaxNumberOfWords];

  int nr_right_children[kMaxNumberOfWords];

  int nr_left_descendant[kMaxNumberOfWords];

  int nr_right_descendant[kMaxNumberOfWords];

  int left_most_child[kMaxNumberOfWords];

  int right_most_child[kMaxNumberOfWords];

  int left_2nd_most_child[kMaxNumberOfWords];

  int right_2nd_most_child[kMaxNumberOfWords];

  action_t last_action;
public:

  void copy(const StateItem & other);

  bool shift(postag_t postag, word_t word, int word_id);

  bool shift(postag_t postag, word_t word,
      int word_id, StateItem* other) const;

  bool left_arc(deprel_t deprel);

  bool left_arc(deprel_t deprel, StateItem* other) const;

  bool right_arc(deprel_t deprel);

  bool right_arc(deprel_t deprel, StateItem* other) const;
};

} //  end for namespace ShiftReduce
} //  end for namespace ZGen

#endif  //  end for __ZGEN_SHITREDUCE_TYPES_STATE_H__