#!/usr/bin/perl

use strict;
use Getopt::Std;

our $opt_c; # DON'T normalize the file
our $opt_j; # jitter the pitch of each subsample by N 1/100th semitones rather than repitching big chunks
getopts('cj:');

print "<<<< DATASET >>>>\n";

my $fmt = <<FMT;
Input File     : '01_snapping_clean.wav'
Channels       : 1
Sample Rate    : 44100
Precision      : 16-bit
Duration       : 00:03:09.35 = 8350438 samples = 14201.4 CDDA sectors
File Size      : 16.7M
Bit Rate       : 706k
Sample Encoding: 16-bit Signed Integer PCM
FMT

sub process($) {
  my $filename = shift or die "Usage: $0 [...filenames]\n";
  my ($name, $ext) = split(/\./, $filename, 2);
  if ($ext !~ /(wav|mp3|aiff?|flac)/) {
    print "not a valid file extension: $ext\n";
    return;
  }

  if ($ext eq 'mp3') {
    print "Converting mp3 to wav\n";
    system('ffmpeg', '-y', '-hide_banner', '-loglevel', 'error', '-i', $filename, $name . '.wav');
    $filename = $name . '.wav';
  }
  my $soxi = `soxi $filename`;
  my @lines = split("\n", $soxi);

  print $soxi;

  my $seconds;
  for my $line (@lines) {
    if ($line =~ /Duration       : (\d\d):(\d\d):(\d\d)\./) {
      my $h = $1;
      my $m = $2;
      my $s = $3;
      $seconds = (((($h * 60) + $m) * 60) + $s) + 0;
    }
  }

  print "Seconds: $seconds\n";

  my $a_tmp = "a_" . $filename;
  my $b_tmp = "b_" . $filename;
  if (!$opt_c) {
    print "Normalizing...";
    system("sox", "-v", 0.945, $filename, $a_tmp);
  } else {
    $a_tmp = $filename;
  }

  # jitter mide:  make copies of the original file and join together,
  #   split into chunks, and then pitchshift the chunks.
  if ($opt_j) {
    my $copies = 4080 / $seconds;
    print "Copies to make: $copies\n";
    print "\n";
    for (my $i = 0; $i < $copies; $i++) {
     system('cp', $a_tmp, sprintf("tmp_%d.wav", $i));
    }
    system 'sox', glob('tmp_*'), $b_tmp;
    system 'rm', glob 'tmp_*';
    system './split44k.sh', $b_tmp, 8, $name;
    print "pitchshifting...\n";
    my $i = 0;
    opendir my $dh, $name or die "cant opendir $name: $!";
    while (readdir $dh) {
      next if /^\./;
      print "$i...\n" if ++$i % 100 == 0;
      my $fn = $_;
      my $in_fn = $name . '/' . $fn;
      my $out_fn = $name . '/z_' . $fn;
      my $n = int rand( 2 * $opt_j ) - $opt_j;
      system 'sox', $in_fn, $out_fn, 'pitch', $n;
      my $rep = `soxi $name/z_$fn`;
      if ($rep =~ /352800/) {
        unlink $in_fn;
      } else {
        unlink $out_fn;
      }
    }
    closedir $dh;
  }

  # standard mode: make a bunch of copies and then join them together,
  #   then subdivide into 8s chunks
  else {
    my $scale = sprintf("%.09f", 5e-7 * $seconds);
    print "Scale factor: $scale\n";
    print "\n";
    system("./spread.sh", $a_tmp, $b_tmp, 0.999, $scale, 1.001);
    system("./split44k.sh", $b_tmp, 8, $name);
  }
  if (!$opt_c) {
    system("/bin/rm", $a_tmp);
  }
  system("/bin/rm", $b_tmp);
}

foreach my $file (@ARGV) {
  print $file . "\n";
  if ( -e $file ) {
    process($file);
  }
}