#!/bin/bash

check_wer () {
    dir=$1
    if [ -d $dir ]; then
        seen_dir=false
        for ddir in $dir/decode*; do
            if [ -d $ddir ]; then
                seen_dir=true
                line=`grep Sum $ddir/score_*/*.sys 2>/dev/null | $char_command | utils/best_wer.sh`
                if [ ! -z "$line" ]; then
                    printf " % -40s " $ddir
                    echo $line | cut -c 1-65
                fi
            fi
        done
        ! $seen_dir
    fi
}
check_wer_split () {
    dir=$1
    if [ -d $dir ]; then
        seen_dir=false
        for ddir in $dir/decode*wb*; do
            if [ -d $ddir ]; then
                seen_dir=true
                printf " % -40s " $ddir
                line=`grep Sum $ddir/score_*/*.sys 2>/dev/null | $char_command | utils/best_wer.sh`
                if [ ! -z "$line" ]; then
                    printf " % -40s " $ddir
                    echo
                    grep -E '(NB|TB)' $(echo $line | cut -c 65-)
                fi
            fi
        done
        ! $seen_dir
    fi
}
split=false
final=false
char_command="grep -v char"

for n in `seq 10`; do
    if [ "$1" == "--final" ]; then
        final=true
        shift
    fi
    if [ "$1" == "--char" ]; then
        char_command="grep char"
        shift
    fi
    if [ "$1" == "--split" ]; then
        split=true
        shift
    fi
done

if [ $# != 0 ]; then
    echo "Usage: local/check_wers.sh [--final] [--char]"
    exit 1;
fi

if $split; then check_cmd=check_wer_split
else check_cmd=check_wer
fi
for dir in exp/tri5 exp/{nnet3,chain}/* exp/chain/semisupervised/* exp/nnet3/semisupervised/*; do
    $check_cmd $dir
    done