#!/usr/bin/perl
# Takes an analysis file, and extracts aggregate data from it.
# See www.theory.esm.rochester.edu/rock_corpus for more info.
# 
# START END ROMNUM CROOT DROOT KEY AROOT
# 0     1   2      3     4     5   6 
#
# Run it like this:
# ./tally.pl [input-file]


for($c=0; $c<12; $c++) {
    $chord_count[$c] = 0;
    $chord_time[$c] = 0;
    for($c2=0; $c2<12; $c2++) {
	$trans[$c][$c2] = 0;
    }
}

$misc_count = $misc_time = $ma = $mi = $dim = $aug = $inv = 0;

open(INFILE, $ARGV[0]);

while(<INFILE>) {   # or use <STDIN> instead of <INFILE>, then "pipe" in input from another process,
                    # e.g. "... | ./tally.pl"
    $line = $_;
    chomp($line);
    $line =~ s/^[ ]+//;
    @words = split(/[ ]+/, $line);

    if($words[0] eq "%") {
	# printf("%s\n", $words[1]);
	next;
    }

    if($words[0] eq "---") {
	$c=0;
    }

    else {
	if($words[3] != -1) {
	    $chord_count[$words[3]]++;
	    $chord_time[$words[3]] += $words[1] - $words[0];
	}
	else {
	    $misc_count++;
	    $misc_time += $words[1] - $words[0];
	}

        # Count up major / minor / diminished / augmented, and root-pos / inversion

        # If the chord symbol string starts with (an optional b or # and then) an upper-case letter, 
	# it's a major-third chord (aug if it contains #5 or +, ma otherwise);
	# else it's a minor-third chord (dim if it contains iio or h; mi otherwise)
        # For dim: note that "iio" will match "viio" or "iio". I suppose it's possible other dim triads might be used, 
	# but none seem to be at present.
        # We can't just search for "o" because then "dom" will match.

	#printf("%s\n", $words[2]);

	if($words[2] =~ /^[b\#]?[A-Z]/) {
	    if($words[2] =~ /\#5|\+/) {
		$aug++;
	    }
	    else {
		$ma++;
	    }
	}
	else {
	    if($words[2] =~ /iio|h/) {
		$dim++;
	    }
	    else {
		$mi++;
	    }
	}
	if($words[2] =~ /6|42|43/) {
	    $inv++;
	}

	if($c!= 0 && $words[3] != -1 && $prev_croot != -1) {
	    if($words[5] == $prev_key) {
		$trans[$prev_croot][$words[3]]++;
		#printf("Adding for %d %d\n", $prev_croot, $words[3]);
	    }
	}
	$prev_croot = $words[3];
	$prev_key = $words[5];
	$c++;
    }
}

$total_count = $total_time = $total_nontonic_count = 0;
for($c=0; $c<12; $c++) {
    $total_count += $chord_count[$c];
    if($c != 0) {
	$total_nontonic_count += $chord_count[$c];	
    }
    $total_time += $chord_time[$c];
}
$total_count += $misc_count;
$total_time += $misc_time;

printf("\nTotal chord count = %d; total time = %.3f\n\n", $total_count, $total_time);

printf("%d major chords (%5.3f), %d minor chords (%5.3f), %d diminished chords (%5.3f), %d augmented chords (%5.3f)\n", $ma, $ma / $total_count, $mi, $mi / $total_count, $dim, $dim / $total_count, $aug, $aug / $total_count);
printf("%d inverted chords (%5.3f), %d root position (%5.3f)\n\n", $inv, $inv / $total_count, ($total_count - $inv), ($total_count - $inv) / $total_count);

printf("Counts and time for chromatic roots (0=I, 1=bII, etc.)\n");
printf("[Root: count (proportion; proportion excluding tonic), time (proportion)]\n");
for($c=0; $c<12; $c++) {
    if($c==0) {
	printf("%5d: %5d (%5.3f; -----), %8.3f (%5.3f)\n", $c, $chord_count[$c], $chord_count[$c]/$total_count, $chord_time[$c], $chord_time[$c]/$total_time);
    }
    else {
	printf("%5d: %5d (%5.3f; %5.3f), %8.3f (%5.3f)\n", $c, $chord_count[$c], $chord_count[$c]/$total_count, $chord_count[$c]/$total_nontonic_count, $chord_time[$c], $chord_time[$c]/$total_time);
    }
}
printf("\n");

printf("%d misc chords, time = %.3f\n\n", $misc_count, $misc_time);

for($c=0; $c<12; $c++) {
    $cons_total[$c] = $ant_total[$c] = 0;
}

for($c=0; $c<12; $c++) {
    for($c2=0; $c2<12; $c2++) {
	$cons_total[$c] += $trans[$c2][$c];
	$ant_total[$c2] += $trans[$c2][$c];
    }
}

printf("Chromatic root transition counts\n");
printf(" Cons");
for($c2=0; $c2<12; $c2++) {
    printf("%4d ", $c2);
}
printf("\nAnt \n");
for($c=0; $c<12; $c++) {
    printf("%4d ", $c);
    for($c2=0; $c2<12; $c2++) {
	printf("%4d ", $trans[$c][$c2]);
    }
    printf("\n");
}

printf("\nChromatic root transitions as proportion of count for CONSEQUENT chord\n");
printf(" Cons");
for($c2=0; $c2<12; $c2++) {
    printf("%6d", $c2);
}
printf("\nAnt \n");
for($c=0; $c<12; $c++) {
    printf("%4d  ", $c);
    for($c2=0; $c2<12; $c2++) {
	if($cons_total[$c2] > 0) {
	    printf("%.3f ", $trans[$c][$c2] / $cons_total[$c2]);
	}
	else {
	    printf("0.000 ");
	}
    }
    printf("\n");
}

printf("\nChromatic root transitions as proportion of count for ANTECEDENT chord\n");
printf(" Cons");
for($c2=0; $c2<12; $c2++) {
    printf("%6d", $c2);
}
printf("\nAnt \n");
for($c=0; $c<12; $c++) {
    printf("%4d  ", $c);
    for($c2=0; $c2<12; $c2++) {
	if($ant_total[$c] > 0) {
	    printf("%.3f ", $trans[$c][$c2] / $ant_total[$c]);
	}
	else {
	    printf("0.000 ");
	}
    }
    printf("\n");
}

for($i=0; $i<12; $i++) {
    $dia[$i] = 0;
    $chro[$i] = 0;
}

for($c=0; $c<12; $c++) {
    for($c2=0; $c2<12; $c2++) {
	$i = (($c2+12) - $c) % 12;
	$chro[$i] += $trans[$c][$c2];
	if($i == 0) {
	    $dia[0] += $trans[$c][$c2];
	}
	elsif($i < 3) {
	    $dia[1] += $trans[$c][$c2];
	}
	elsif($i < 5) {
	    $dia[2] += $trans[$c][$c2];
	}
	elsif($i < 6) {
	    $dia[3] += $trans[$c][$c2];
	}
	elsif($i < 7) {
	    $dia[7] += $trans[$c][$c2];        # Special case: dia[7] = tritone!
	}
	elsif($i < 8) {
	    $dia[4] += $trans[$c][$c2];
	}
	elsif($i < 10) {
	    $dia[5] += $trans[$c][$c2];
	}
	else {
	    $dia[6] += $trans[$c][$c2];
	}
    }
}

printf("\nChromatic interval counts:\n");
for($i=0; $i<12; $i++) {
    printf("%d: %d\n", $i, $chro[$i]);
}

printf("Diatonic interval counts:\n");
printf("+M/m2: %d\n", $dia[1]);
printf("-M/m2: %d\n", $dia[6]);
printf("+M/m3: %d\n", $dia[2]);
printf("-M/m3: %d\n", $dia[5]);
printf("+P4/-P5: %d\n", $dia[3]);
printf("-P4/+P5: %d\n", $dia[4]);
printf("TT: %d\n", $dia[7]);