#! /usr/bin/env python3

import sys
import random
import argparse

import Config
from Transition import Transition

################################################################################
def applyTransition(ts, strat, config, name) :
  transition = [trans for trans in ts if trans.name == name][0]
  movement = strat[transition.name]
  transition.apply(config)
  config.moveWordIndex(movement)
################################################################################

################################################################################
def randomDecode(ts, strat, config) :
  EOS = Transition("EOS")
  config.moveWordIndex(0)
  while config.wordIndex < len(config.lines) - 1 :
    candidates = [trans for trans in transitionSet if trans.appliable(config)]
    candidate = candidates[random.randint(0, 100) % len(candidates)]
    applyTransition(transitionSet, strategy, config, candidate.name)
    if args.debug :
      print(candidate.name, file=sys.stderr)
      config.printForDebug(sys.stderr)
  EOS.apply(config)
################################################################################

################################################################################
if __name__ == "__main__" :
  parser = argparse.ArgumentParser()
  parser.add_argument("trainCorpus", type=str,
    help="Name of the CoNLL-U training file.")
  parser.add_argument("--debug", "-d", default=False, action="store_true",
    help="Print debug infos on stderr.")
  args = parser.parse_args()

  transitionSet = [Transition(elem) for elem in ["RIGHT", "LEFT", "SHIFT", "REDUCE"]]
  strategy = {"RIGHT" : 1, "SHIFT" : 1, "LEFT" : 0, "REDUCE" : 0}

  sentences = Config.readConllu(sys.argv[1])

  first = True
  for config in sentences :
    randomDecode(transitionSet, strategy, config)
    config.print(sys.stdout, header=first)
    first = False
################################################################################