import com.decontextualize.a2z.TextFilter;

public class MarkovFilter extends TextFilter {

  public static void main(String[] args) {
    MarkovFilter mf = new MarkovFilter();
    mf.setNgramLength(args[0]);
    mf.setMaxLength(args[1]);
    mf.sendValues();
    mf.run();
  }

  private int en;
  private int el;

  Markov mark = new Markov();

  public void eachLine(String line) {
    mark.feedLine(line);
  }

  public void setNgramLength(String en_) {
    en=Integer.parseInt(en_);
  }

  public void setMaxLength(String el_) {
    el=Integer.parseInt(el_);
  }

  public void sendValues() {
    mark.getValues(en, el);
  }

  public void end() {
    mark.dump();
    println(mark.generateLine());
  }
}