//
// Programmer:    Craig Stuart Sapp <craig@ccrma.stanford.edu>
// Creation Date: Thu Mar 30 23:37:15 GMTDT 2006
// Last Modified: Thu Mar 30 23:37:21 GMTDT 2006
// Filename:      ...soundfile/examples/monoclick.cpp
// Syntax:        C++ 
//
// Description:   Adds all channels in the input sound file and writes
//                a sound file with the summation of all channels.
//                Amplitude scaling factor can be given to prevent
//                clipping if necessary.
//

#include "soundfile.h"
#include <stdlib.h>

#ifndef OLDCPP
   #include <iostream>
   #include <fstream>
   using namespace std;
#else
   #include <iostream.h>
   #include <fstream.h>
#endif

void getClickTimes(Array<long>& clicktimes, const char* filename);


//////////////////////////////////////////////////////////////////////////

int main(int argc, char** argv) {
   Options options;
   options.define("a|amp=d:1.0",       "amplitude scaling factor");
   options.define("c|click-amp=d:0.15","click track amplitude scaling factor");
   options.process(argc, argv);
   
   double clickamp = options.getDouble("click-amp");
   const char* clicktrack = options.getArg(1);
   const char* outputname = options.getArg(2);

   Array<long> clicktimes;
   getClickTimes(clicktimes, clicktrack);

   SoundHeader    header;
   header.setHighMono();
   SoundFileWrite outsound(outputname, header);

   int i;
   double clicksample;

   for (i=0; i<clicktimes.getSize(); i++) {
      clicktimes[i] = (long)(clicktimes[i] / 1000.0 * header.getSrate() + 0.5);
      cout << clicktimes[i] << endl;
   }

   int clicki = 0;
   int clickstatus = 0;
   int clickcount = clicktimes.getSize();
   int samplecount = clicktimes[clicktimes.getSize()-1] + 100000;

   for (i=0; i<samplecount; i++) {
      if ((clicki < clickcount) && (clicktimes[clicki] == i)) {
         clickstatus = 1;
         clicki++;
      }
      if (clickstatus) {
         clicksample = ((rand() * 2.0)/INT_MAX - 1.0) * clickamp;
         clickstatus++;
         if (clickstatus > 44) {
            clickstatus = 0;
         }
      } else {
         clicksample = 0.0;
      }
      outsound.writeSampleDouble(clicksample);
   }

   return 0;
}



//////////////////////////////
//
// getClickTimes --
//

void getClickTimes(Array<long>& clicktimes, const char* filename) {

   ifstream clickfile(filename);
   if (!clickfile.is_open()) {
      cerr << "Error: cannot open file " << filename << endl;
   }

   long input;
   clicktimes.setSize(100000);
   clicktimes.setGrowth(100000);
   clicktimes.setSize(0);

   clickfile >> input;
   while (!clickfile.eof()) {
      clicktimes.append(input);
      clickfile >> input;
   }
}