comparison testdata/evaluation/run.sh @ 194:72a9696f7caa

Fixes to script

 if ! sonic-annotator -v ; then
     echo "Failed to run sonic-annotator (not in PATH?), giving up"
     exit 1
 fi
 
-rdffile="../silvet.n3"
+rdffile="../../silvet.n3"
 if [ ! -f "$rdffile" ] ; then
-    
+    echo "Failed to find plugin RDF file at $rdffile, giving up"
+    exit 1
+fi
 
 case "$trios_path" in
-*\ *) echo "TRIOS dataset path $trios_path has a space in it, this script won't handle that!"; exit 1;;
+*\ *) echo "TRIOS dataset path $trios_path has a space in it, this script won't handle that"; exit 1;;
 esac
 
 VAMP_PATH=../..
 export VAMP_PATH
 

 infiles=`find "$trios_path" -name \*.wav -print | egrep '(mozart|lussier|take_five)' | grep -v _syn`
 
 grep Piano "$rdffile" | sed 's/^.*( *//' | sed 's/ *).*$//' | sed 's/ "/\n/g' | sed 's/"//g' | tr '[A-Z]' '[a-z]' | tail -n +2 | cat -n > "$instfile"
 
 instrument_for() {
-    filename="$0"
+    filename="$1"
     base=`basename "$filename" .wav`
     instrument_no=`grep "$base" "$instfile" | awk '{ print $1; }'`
-    if [ -z "$instrument_no" ]; 
+    if [ -z "$instrument_no" ] || [ -z "$base" ]; 
     then echo 0
     else echo "$instrument_no"
     fi
 }
+
+echo
+echo "Input files are:"
+echo $infiles | fmt -1
 
 for infile in $infiles; do
 
     echo
     echo "Evaluating for file $infile..."

     instrument=`instrument_for "$infile"`
     case "$instrument" in
 	[0-9]*) ;;
 	*) echo "Instrument extraction failed for infile $infile -- not even default multi-instrument setting returned?"; exit 1;;
     esac
+    
+    piece=`basename \`dirname "$infile" \``
+    arrangement=`basename "$infile" .wav`
 
     echo
-    echo "For file $infile, using instrument setting $instrument..."
+    echo "For piece $piece, arrangement $arrangement, using instrument $instrument..."
 
     sox "$infile" "$tmpwav" gain -n -6.020599913279624
 
     ##!!! todo: actually apply the instrument setting!
 
-    time sonic-annotator \
+    sonic-annotator \
 	--writer csv \
 	--csv-one-file "$outfile" \
 	--csv-force \
 	--default vamp:silvet:silvet:notes \
 	"$tmpwav"

 	sed 's/^[^,]*,//' | \
 	while IFS=, read start duration frequency level label; do
 	    end=`echo "$start $duration + p" | dc`
 	    echo -e "$start\t$end\t$frequency"
     done > "$outfile.lab"
-    
-    piece=`basename \`dirname "$infile" \``
-    arrangement=`basename "$infile" .wav`
 
     for ms in 50 100; do
 	echo
 	echo "Validating against ground truth at $ms ms:"
 	"$yc" ./evaluate_lab.yeti "$ms" "../TRIOS-groundtruth/$piece/$arrangement.lab" "$outfile.lab"