#!/bin/sh
#####################################################-*-mode:shell-script-*-
###                                                                       ##
###                    Alan W Black and Kevin Lenzo                       ##
###                         Copyright (c) 1998                            ##
###                        All Rights Reserved.                           ##
###                                                                       ##
###  Permission to use, copy, modify,  and licence this software and its  ##
###  documentation for any purpose, is hereby granted without fee,        ##
###  subject to the following conditions:                                 ##
###   1. The code must retain the above copyright notice, this list of    ##
###      conditions and the following disclaimer.                         ##
###   2. Any modifications must be clearly marked as such.                ##
###   3. Original authors' names are not deleted.                         ##
###                                                                       ##
###  THE AUTHORS OF THIS WORK DISCLAIM ALL WARRANTIES WITH REGARD TO      ##
###  THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY   ##
###  AND FITNESS, IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY         ##
###  SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES            ##
###  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN   ##
###  AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,          ##
###  ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF       ##
###  THIS SOFTWARE.                                                       ##
###                                                                       ##
###  This file is part "Building Voices in the Festival Speech            ##
###  Synthesis System" by Alan W Black and Kevin Lenzo written at         ##
###  Robotics Institute, Carnegie Mellon University, fall 98              ##
############################################################################
###                                                                       ##
###  Generate cepstrum and autolabel waveforms from prompts               ##
###                                                                       ##
############################################################################

if [ $# = 0 ]
then
   echo align nonsense word phone labels from prompts
   echo "Usage: make_labs prompt-wav/*.wav"
   echo expects the following to exist:
   echo "prompt-wav/*.wav   The waveform prompts"
   echo "prompt-lab/*.lab   The labels for the prompts"
   echo "wav/*.wav          The spoken waveforms"
   echo "cep/               Will fill this directory with cepstrum files"
   echo "prompt-cep/        Will fill this directory with cepstrum files"
   echo "lab/               Will fill this directory with label files"
   exit 1
fi

if [ ! "$ESTDIR" ]
then
   echo "environment variable ESTDIR is unset"
   echo "set it to your local speech tools directory e.g."
   echo '   bash$ export ESTDIR=/home/awb/projects/speech_tools/'
   echo or
   echo '   csh% setenv ESTDIR /home/awb/projects/speech_tools/'
   exit 1
fi
SIG2FV=$ESTDIR/bin/sig2fv
# These parameters are chosen by becuase they seems vaguely right
# and do produce reasonable results, but if you think you know better
# you probably do, feel free to experiment
SIG2FVPARAMS='-coefs melcep -delta melcep -melcep_order 12 -fbank_order 24 -shift 0.005 -factor 5.0 -preemph 0.97'

if [ ! "$FESTVOXDIR" ]
then
   echo "environment variable FESTVOXDIR is unset"
   echo "set it to your local festvox directory e.g."
   echo '   bash$ export FESTVOXDIR=/home/awb/projects/festvox/'
   echo or
   echo '   csh% setenv FESTVOXDIR /home/awb/projects/festvox/'
   exit 1
fi

PHONEALIGN=$FESTVOXDIR/src/diphones/phonealign

# Because I've made this mistake *every* time I've run this 
if [ `dirname $1` = "wav" ]
then
   echo 
   echo "ERROR"
   echo "aligning wav/ files to wav/ files which is not what you want"
   echo "to do you probably wanted to type "
   echo "      bin/make_labs prompt-wav/*.wav"
   echo "not"
   echo "      bin/make_labs wav/*.wav"
   echo
   exit
fi

#
#  You may want to comment out some of the following parts if
#  you have already run them before
#
for i in $*
do
   fname=`basename $i .wav`
   echo $fname
   # Construct cepstrum files
   $SIG2FV $SIG2FVPARAMS -otype est $i -o prompt-cep/$fname.cep
   $SIG2FV $SIG2FVPARAMS -otype est_binary wav/$fname.wav -o cep/$fname.cep

   # Because of a bug in the synthesis process the prompt cepstrum files
   # may be slightly longer than necessary and have 0s in the signal
   # This can causes NaN in the cepstrum files.  The following
   # clips the prompt-cepstrum files with respect to the time of the
   # last label in the prompt label file.
   $ESTDIR/bin/ch_track -end `tail -1 prompt-lab/$fname.lab | awk '{print $1}'` prompt-cep/$fname.cep -otype est_binary -o prompt-cep/$fname.cep
   
   # Now the actual alignment   
   $PHONEALIGN -itrack prompt-cep/$fname.cep -ilabel prompt-lab/$fname.lab -otrack cep/$fname.cep -olabel lab/$fname.lab

done