partial_decoder.h

Go to the documentation of this file.

#ifndef __ZGEN_SHIFTREDUCE_PARTIAL_DECODER_H__
#define __ZGEN_SHIFTREDUCE_PARTIAL_DECODER_H__

#include "decoder.h"
#include "types/tree.h"

namespace ZGen {
namespace ShiftReduce {

class PartialDecoder: public Decoder {
public:
  PartialDecoder(const option_t& opts,
      DecodeContext* ctx,
      Model* model,
      const Knowledge::PostagConstrain* constrain);

private:
  void config_input(const dependency_t* input);

  void get_possible_actions(const StateItem& item);

  bool legal_shift(const StateItem& item, int i);

  bool legal_left_arc(const StateItem& item);

  bool legal_right_arc(const StateItem& item);

  DependencyForest forest;
};

} //  end for namespace Shiftreduce
} //  end for namespace ZGen

#endif  //  end for __ZGEN_SHIFTREDUCE_PARTIAL_DECODER_H__