//
// Programmer:    Craig Stuart Sapp <craig@ccrma.stanford.edu>
// Creation Date: Tue Jan 16 11:25:31 PST 2001
// Last Modified: Wed Jan 17 10:16:23 PST 2001
// Filename:      ...sig/examples/all/vroot.cpp
// Web Address:   http://sig.sapp.org/examples/museinfo/humdrum/vroot.cpp
// Syntax:        C++; museinfo
//
// Description:   determine the chord at a given metric level
// 

#include "humdrum.h"
#include <string.h>
#include <math.h>

// function declarations
void   checkOptions             (Options& opts, int argc, char* argv[]);
void   example                  (void);
void   usage                    (const char* command);
int    findMinimum              (Array<double>& values);

// global variables
Options      options;            // database for command-line arguments
int          debugQ     = 0;     // used with the --debug option
int          appendQ    = 0;     // used with the -a option
int          compoundQ  = 1;     // used with the -c option
double       rval       = 1.0;   // used with the -r option
int          srhythm    = 4;     // used with the -r option
double       empirical1 = -5.8;  // used with the --e1 option
double       empirical2 = 2.225; // used with the --e2 option
double       sx         = 1.0;   // used with the --sx option
double       sy         = 1.0;   // used with the --sx option
int          octave     = -1;    // used with the -o option


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

int main(int argc, char* argv[]) {
   HumdrumFile infile;
   Array<int>  rhylev;

   // process the command-line options
   checkOptions(options, argc, argv);

   // figure out the number of input files to process
   int numinputs = options.getArgCount();

   int most = 0;
   Array<double> values;
   values.setSize(0);
   char buffer[128] = {0};
   char buffer2[128] = {0};

   Array<double> tonicscores;
   for (int i=0; i<numinputs || i==0; i++) {
      infile.clear();

      // if no command-line arguments read data file from standard input
      if (numinputs < 1) {
         infile.read(cin);
      } else {
         infile.read(options.getArg(i+1));
      }
      infile.analyzeRhythm("4");

      int nextroot = 0;
      int lastroot = 0;
      int line;
      for (line=0; line<infile.getNumLines(); line++) {
         cout << infile.getLine(line);

         if (infile[line].getType() == E_humrec_local_comment) {
            cout << "\t!\n";
            continue;
         }
         if (infile[line].getType() == E_humrec_data_measure) {
            cout << "\t" << infile[line][0] << "\n";
            continue;
         }
         if (infile[line].getType() == E_humrec_interpretation) {
            if (infile[line].equalFieldsQ("**kern")) {
               cout << "\t" << infile[line][0] << "\n";
            } else {
               cout << "\t*\n";
            }
            continue;
         }

         if (infile[line].getType() != E_humrec_data) {
            cout << endl;
            continue;
         }
         if (fabs(rval * infile[line].getBeat() - 
             (int)(rval * infile[line].getBeat())) < 0.001) {
            infile.analyzeMetricLevel(rhylev);
            // find ending point:
            int endline = line+1;
            while (endline < infile.getNumLines()) {
               if (infile[endline].getType() != E_humrec_data) {
                  endline++;
                  continue;
               }
               if (fabs(rval * infile[endline].getAbsBeat() - 
                   (int)(rval * infile[endline].getAbsBeat())) < 0.001) {
                  break;
               }
               endline++;
            }
            if (endline >= infile.getNumLines()) {
               endline = infile.getNumLines() - 1;
            }
            most = infile.analyzeChordProbability(values, line,
                      endline - 1, rhylev, empirical1, empirical2, sx, sy);
            if (octave < 0) {
               nextroot = (most + 2) % 40 + 80;
               if (abs(nextroot + 40 - lastroot) < 13) {
                  nextroot += 40;
               } else if (abs(nextroot - 40 - lastroot) < 13) {
                  nextroot -= 40;
               }
            } else {
               nextroot = (most + 2) % 40 + 40 * octave;
            }
            Convert::base40ToKern(buffer, nextroot);
            lastroot = nextroot;
            cout << "\t";
            if (infile[line].getDuration() > 1.0/rval) {
               cout << Convert::durationToKernRhythm(buffer2,
                  infile[line].getDuration());
            } else {
               cout << srhythm;
            }
            if (values[most] == 0) {
               cout << "r";
            } else {
               cout << buffer;
            }
            // cout << " [" << line << ", " << endline-1 << "]";
            cout << endl;
         } else {
            cout << "\t." << endl;
         }
      }

   }

   return 0;
}


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



//////////////////////////////
//
// findMinimum --
//

int findMinimum(Array<double>& values) {
   int i;
   int min = 0;
   for (i=1; i<values.getSize(); i++) {
      if (values[i] < values[min]) {
         min = i;
      }
   }
   return min;
}



//////////////////////////////
//
// checkOptions -- validate and process command-line options.
//

void checkOptions(Options& opts, int argc, char* argv[]) {
   opts.define("a|append=b",    "append analysis to data in output");   
   opts.define("C|compound=b",  "don't try to use compound meters");   
   opts.define("r|rhythm=i:4",  "rhythm to measure root at");   
   opts.define("e1|empirical1=d:-5.8",  "empirical rhythm value 1");
   opts.define("e2|empirical2=d:2.225",  "empirical rhythm value 2");
   opts.define("sx=d:1.0", "scale factor for the x-pitch space axis");
   opts.define("sy=d:1.0", "scale factor for the y-pitch space axis");
   opts.define("o|octave=i:-1",  "octave number of bass line");

   opts.define("debug=b",  "trace input parsing");   
   opts.define("author=b",  "author of the program");   
   opts.define("version=b", "compilation information"); 
   opts.define("example=b", "example usage"); 
   opts.define("h|help=b",  "short description"); 
   opts.process(argc, argv);
   
   // handle basic options:
   if (opts.getBoolean("author")) {
      cout << "Written by Craig Stuart Sapp, "
           << "craig@ccrma.stanford.edu, Dec 2000" << endl;
      exit(0);
   } else if (opts.getBoolean("version")) {
      cout << argv[0] << ", version: 6 Dec 2000" << endl;
      cout << "compiled: " << __DATE__ << endl;
      cout << MUSEINFO_VERSION << endl;
      exit(0);
   } else if (opts.getBoolean("help")) {
      usage(opts.getCommand());
      exit(0);
   } else if (opts.getBoolean("example")) {
      example();
      exit(0);
   }

   debugQ = opts.getBoolean("debug");
   appendQ = opts.getBoolean("append");

   if (opts.getBoolean("compound")) {
      compoundQ = 0;
   } else {
      compoundQ = 0;
   }

   rval = opts.getInteger("rhythm")/4.0;
   srhythm = (int)(rval * 4.0);

   empirical1 = opts.getDouble("empirical1");
   empirical2 = opts.getDouble("empirical2");
   sx = opts.getDouble("sx");
   sy = opts.getDouble("sy");

   octave = opts.getInteger("octave");
}



//////////////////////////////
//
// example -- example usage of the maxent program
//

void example(void) {
   cout <<
   "                                                                        \n"
   << endl;
}



//////////////////////////////
//
// usage -- gives the usage statement for the quality program
//

void usage(const char* command) {
   cout <<
   "                                                                        \n"
   << endl;
}