#!/bin/bash
#

if [ -f path.sh ]; then . path.sh; fi

lm_dir=$1
src_lang=$2
tgt_lang=$3

arpa_lm=${lm_dir}/lm.arpa
[ ! -f $arpa_lm ] && echo No such file $arpa_lm && exit 1;

rm -rf $tgt_lang
cp -r $src_lang $tgt_lang

# Build G.fst for rescoring
#cat ${lm_dir}/lm_orig.arpa | \
#   grep -v '<s> <s>' | \
#   grep -v '</s> <s>' | \
#   grep -v '</s> </s>' | \
#   grep -v -i '<unk>' | \
#   grep -v -i '<spoken_noise>' | \
#   arpa2fst --read-symbol-table=$tgt_lang/words.txt --keep-symbols=true - | fstprint | \
#   tools/fst/eps2disambig.pl | tools/fst/s2eps.pl | fstcompile --isymbols=$tgt_lang/words.txt \
#     --osymbols=$tgt_lang/words.txt  --keep_isymbols=false --keep_osymbols=false | \
#    fstrmepsilon | fstarcsort --sort_type=ilabel > $tgt_lang/G_no_prune.fst

# Compose the language model to FST
cat $arpa_lm | \
   grep -v '<s> <s>' | \
   grep -v '</s> <s>' | \
   grep -v '</s> </s>' | \
   grep -v -i '<unk>' | \
   grep -v -i '<spoken_noise>' | \
   arpa2fst --read-symbol-table=$tgt_lang/words.txt --keep-symbols=true - | fstprint | \
   tools/fst/eps2disambig.pl | tools/fst/s2eps.pl | fstcompile --isymbols=$tgt_lang/words.txt \
     --osymbols=$tgt_lang/words.txt  --keep_isymbols=false --keep_osymbols=false | \
    fstrmepsilon | fstarcsort --sort_type=ilabel > $tgt_lang/G.fst

echo  "Checking how stochastic G is (the first of these numbers should be small):"
fstisstochastic $tgt_lang/G.fst

# Compose the token, lexicon and language-model FST into the final decoding graph
fsttablecompose $tgt_lang/L.fst $tgt_lang/G.fst | fstdeterminizestar --use-log=true | \
    fstminimizeencoded | fstarcsort --sort_type=ilabel > $tgt_lang/LG.fst || exit 1;
fsttablecompose $tgt_lang/T.fst $tgt_lang/LG.fst > $tgt_lang/TLG.fst || exit 1;

echo "Composing decoding graph TLG.fst succeeded"
#rm -r $tgt_lang/LG.fst   # We don't need to keep this intermediate FST