#include <algorithm>
#include <iostream>
#include <string>

#include "ctcdecode/scorer.h"
#include "fst/fstlib.h"
#include "alphabet.h"

using namespace std;

int generate_trie(const char* alphabet_path, const char* kenlm_path, const char* trie_path) {
  Alphabet alphabet(alphabet_path);
  Scorer scorer(0.0, 0.0, kenlm_path, "", alphabet);
  scorer.save_dictionary(trie_path);
  return 0;
}

int main(int argc, char** argv) {
  if (argc != 4) {
    std::cerr << "Usage: " << argv[0] << " <alphabet> <lm_model> <trie_path>" << std::endl;
    return -1;
  }

  return generate_trie(argv[1], argv[2], argv[3]);
}