#!/usr/bin/perl

# Anvil Concatenation 
#  
# Concatenates two non-overlapping .anvil files that share the same timeline.
#
# Usage: anvilcat.pl file_one.anvil file_two.anvil output.anvil
# 
# Chris Osborn 
# cosborn@ldc.upenn.edu 
# 
# FORM Project, Linguistic Data Consortium 
# University of Pennsylvania 
# 
# http://www.ldc.upenn.edu/Projects/FORM/ 
# 

print("\n");

open(ONE, "./$ARGV[0]");
open(TWO, "./$ARGV[1]");
open(OUT, ">./$ARGV[2]");
open(TEMP, ">./temp");

#extract lines from tracks of second file
while(<TWO>){
  if(/track name/){
    ($trackName) = ($_ =~ /track name="(.*?)"/);   #print "$lastIndex{$trackName}:$index\n";
    $trackName =~ s/ //g;
    $num = 0;
  }elsif($_ =~ /<el/){
    ($temp) = ($_ =~ /(.*?)>/);
    $trackHash{$trackName}[$num] = $temp;
    $num++;
  }elsif($_ =~ /<\/el>/){
    ($temp) = ($_ =~ /^(.*?)>/);
    $trackHash{$trackName}[$num] = $temp;
    $num++;
  }elsif($_ =~ /<attribute/){
    ($temp) = ($_ =~ /^(.*?>.*?)>/);
    $trackHash{$trackName}[$num] = $temp;
    $num++;
  }
}
close(TWO);

#insert saved lines into first file, copy to new output
while(<ONE>){
  if(/track name/){
    ($trackName) = ($_ =~ /track name="(.*?)"/);
    $trackName =~ s/ //g;
    $lastIndex{$trackName} = 0;
    $elements{$trackName} = 0;
  }elsif(/<el/){
    ($index) = ($_ =~ /<el index="(\d*?)"/);
     $elements{$trackName}++;
  }
  print TEMP $_;
}
close(ONE);
close(TEMP);

#replace indexes
foreach $track (keys %trackHash){
  $elindex = $elements{$track};
  for($i = 0; $i<@{$trackHash{$track}}; $i++){
    if($trackHash{$track}[$i] =~ /<el/){
      $trackHash{$track}[$i] =~ s/index="\d*?"/index="$elindex"/;
      $elindex++;
    }
  }
}

#insert newlines into output file
open(TEMP2, "./temp");
while(<TEMP2>){
  if(/track name/){
    ($trackName) = ($_ =~ /track name="(.*?)"/);
    $trackName =~ s/ //g;
  }elsif(/<\/track/){
    for($i=(@{$trackHash{$trackName}}-1); $i >= 0; $i--){
      $_ =~ s/^/$trackHash{$trackName}[$i]>\n/g;
    }
  }
  print OUT $_;
}

close(TEMP2);
close(OUT);

unlink("./temp");