//
// Programmer:    Craig Stuart Sapp <craig@ccrma.stanford.edu>
// Creation Date: Thu Feb 14 07:20:44 PST 2002
// Last Modified: Thu Jan 16 22:47:38 PST 2003
// Filename:      ...soundfile/examples/sndpower.cpp
// Syntax:        C++ 
//
// Description:   Calculate the average signal power for given durations
//

#include "soundfile.h"
#include <stdlib.h>
#include <iostream.h>
#include <iomanip.h>
#include <math.h>

int main(int argc, char** argv) {
   Options options;
   options.define("a|average=i:10000", "Number of samples to average over");
   options.define("s|start=i:0",       "Starting sample");
   options.define("n|count=i:-1",      "Number of analyses to make");
   options.process(argc, argv);
   if (options.getArgCount() == 0) {
      cout << "Usage: " << options.getCommand() 
           << " filename" 
           << endl;
      exit(1);
   }

   SoundFileRead soundfile(options.getArg(1));

   int chunk = options.getInteger("average");
   int start = options.getInteger("start");
   int totalframes = options.getInteger("count");
   if (totalframes <= 0) {
      totalframes = (soundfile.getSamples() - start)/chunk + 1;
   } 

   // print the header:
   cout << "; Filename        = " << options.getArg(1) << "\n";
   cout << "; Samples         = " << soundfile.getSamples() << "\n";
   if (start != 0) {
      cout << "; Starting sample = " << start << "\n";
   }
   cout << "; Averaging size  = " << chunk << " samples" << "\n";
   cout << "; Total frames    = " << totalframes << "\n";
   cout << "; start-sample	RMS		Db\n";

   double sum, value, db, rms;
   int i, j, channel;
   for (i=0; i<totalframes; i++) {
      sum = 0.0;
      cout << i * chunk << ":\t";
      for (j=0; j<chunk; j++) {
         for (channel=0; channel < soundfile.getChannels(); channel++) {
            value = soundfile.getCurrentSampleDouble(channel);
            sum += value * value;
         }
         soundfile.incrementSample(); 
      }
      if (sum == 0.0) {
         db = -1000;
      } else {
         db = 10.0 * log10(sum/chunk/soundfile.getChannels());
      }
      rms = sqrt(sum)/chunk/soundfile.getChannels();
      cout << rms << '\t' << db << "\n";
   }
   
   return 0;
}