//
// Programmer:    Craig Stuart Sapp <craig@ccrma.stanford.edu>
// Creation Date: Wed Nov 22 18:42:05 PST 2000
// Last Modified: Wed Nov 22 18:42:09 PST 2000
// Filename:      ...sig/doc/examples/all/arto/arto.cpp
// Web Address:   http://sig/doc/examples/all/arto/arto.cpp
// Syntax:        C++ 
//
// Description:   reduces chordal textures to their unembellished forms.
//

#include "humdrum.h"
#include <string.h>
#include <ctype.h>
#include <stdio.h>

#define NOTEDISSONANT_UNKNOWN  -1
#define NOTEDISSONANT_YES       0
#define NOTEDISSONANT_NO        1

typedef Array<int> notearray;

// function declarations
int    displayChord(HumdrumFile& infile, Array<int>& chord, int line);
void   generateReduction(HumdrumFile& infile, Array<int>& chord);
double getChordDuration(HumdrumFile& infile, Array<int>& chord, int line);
void   printNewDuration(const char* kern, double duration);

void   checkOptions             (Options& opts, int argc, char* argv[]);
void   example                  (void);
void   displayNoteDiss          (Array<notearray>& notediss);
void   generateAnalysis         (HumdrumFile& infile, 
                                      Array<int>& soncondis,
                                      Array<int>& tertian,
                                      Array<int>& terdis,
                                      Array<int>& accent,
                                      Array<int>& dissic,
                                      Array<notearray>& notediss,
                                      Array<int>& chord,
                                      Array<ChordQuality>& cq);
int    measureAccentuation      (HumdrumFile& infile, int line);
int    measureTertian           (HumdrumFile& infile, int line);
int    measureTertianDissonance (HumdrumFile& infile, int line);
int    measureVerticalConsonance(HumdrumFile& infile, int line);
int    measureDissic            (int line, Array<int>& soncondis, 
                                      Array<int>& tertian, Array<int>& terdis, 
                                      Array<int>& accent);
int    measureNoteDissonance    (int line, HumdrumFile& score, int note,
                                      Array<int>& soncondis, Array<int>& accent,
                                      Array<int>& dissic);
void   measureChordQuality      (int line, HumdrumFile&score, 
                                      Array<int>& notediss,
                                      Array<ChordQuality>& cq);
int    measureChordFunction1    (int line, HumdrumFile& score, 
                                      Array<int>& soncondis, 
                                      Array<int>& tertian, Array<int>& terdis, 
                                      Array<int>& accent, Array<int>& dissic,
                                      Array<ChordQuality>& cq);
int    measureChordFunction2    (int line, Array<int>& soncondis, 
                                      Array<int>& tertian, Array<int>& terdis, 
                                      Array<int>& accent, Array<int>& dissic,
                                      Array<ChordQuality>& cq);
void   printAnalysis            (HumdrumFile& infile, 
                                      Array<int>& soncondis,
                                      Array<int>& tertian,
                                      Array<int>& terdis,
                                      Array<int>& accent,
                                      Array<int>& dissic,
                                      Array<notearray>& notediss,
                                      Array<int>& chord,
                                      Array<ChordQuality>& cq);
void   rotateNotes              (Array<int>& notes);
void   usage                    (const char* command);


// 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


// variables for keeping track of the change in meter during the score:
int    metercount = 4;
int    meterbase = 4;
double beatduration = 1.0;

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

int main(int argc, char* argv[]) {
   HumdrumFile infile;
   Array<int>  soncondis;
   Array<int>  tertian;
   Array<int>  terdis;
   Array<int>  accent;
   Array<int>  dissic;
   Array<notearray> notediss;
   Array<int>  chord;
   Array<ChordQuality>  cq;

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

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

   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));
      }

      generateAnalysis(infile, soncondis, tertian, terdis, accent,
            dissic, notediss, chord, cq);
//      printAnalysis(infile, soncondis, tertian, terdis, accent,
//            dissic, notediss, chord, cq);
      generateReduction(infile, chord);
   }

   return 0;
}


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

//////////////////////////////
//
// generateReduction -- remove chordal embellishments.
//

void generateReduction(HumdrumFile& infile, Array<int>& chord) {
   int i;
   for (i=0; i<infile.getNumLines(); i++) {
      switch (infile[i].getType()) {
      case E_humrec_global_comment:
      case E_humrec_bibliography:
      case E_humrec_none:
      case E_humrec_empty:
      case E_humrec_data_comment:
      case E_humrec_data_interpretation:
      case E_humrec_data_measure:
         cout << infile[i].getLine() << "\n";
         break;
      case E_humrec_data:
         i = displayChord(infile, chord, i);
         break;
      }
   }
}



//////////////////////////////
// 
// displayChord -- display only the true chords
//

int displayChord(HumdrumFile& infile, Array<int>& chord, int line) {
   double duration = 0.0;
   int i;
   if (chord[line] == 'c') {
      duration = getChordDuration(infile, chord, line);
      for (i=0; i<infile[line].getFieldCount(); i++) {
         if (strcmp(infile[line][i], ".") == 0) {
            printNewDuration(infile.getDotValue(line, i), duration);
         } else {
            printNewDuration(infile[line][i], duration);
         }
         if (i < infile[line].getFieldCount() - 1) {
            cout << "\t";
         }
      }
      cout << "\n";
   }
   return line;
}

//////////////////////////////
//
// printNewDuration -- 
//

void printNewDuration(const char* kern, double duration) {
   char buffer[128] = {0};
   Convert::durationToKernRhythm(buffer, duration);
   int length = strlen(kern);
   int notprinted = 1;

   for (int i=0; i<length; i++) {
      if (isdigit(kern[i]) && notprinted) {
         cout << buffer;
         notprinted = 0;
         continue;
      }

      if (isdigit(kern[i]) || kern[i] == '.' || kern[i] == '[' ||
          kern[i] == '_' || kern[i] == ']') {
         continue;
      }

      if (tolower(kern[i]) == 'a' || tolower(kern[i]) == 'b' ||
          tolower(kern[i]) == 'c' || tolower(kern[i]) == 'd' ||
          tolower(kern[i]) == 'e' || tolower(kern[i]) == 'f' ||
          tolower(kern[i]) == 'g' || kern[i] == '-' ||
          kern[i] == '#' || kern[i] == '-' || kern[i] == ';') {
         cout << kern[i];
      }

   }

}



//////////////////////////////
//
// getChordDuration -- 
//

double getChordDuration(HumdrumFile& infile, Array<int>& chord, int line) {
   double output = 0;

   output += infile[line].getDuration();

   int i;
   for (i=line+1; i<chord.getSize(); i++) {
      if (chord[i] == 'c') {
         break;
      }
      if (chord[i] == 'p') {
         output += infile[i].getDuration();
      }
      if (chord[i] == 'g') {
         output += infile[i].getDuration();
      }
      if (chord[i] == 'n') {
         break;
      }
   }

   for (i=line-1; i>= 0; i--) {
      if (chord[i] == 'c') {
         break;
      }
      if (chord[i] == 'n') {
         output += infile[i].getDuration();
      }
      if (chord[i] == 'g') {
         break;
      }
      if (chord[i] == 'p') {
         break;
      }
   }

   return output;
}


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


//////////////////////////////
//
// 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("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, Nov 2000" << endl;
      exit(0);
   } else if (opts.getBoolean("version")) {
      cout << argv[0] << ", version: Nov 2000" << endl;
      cout << "compiled: " << __DATE__ << endl;
      cout << HUMDRUM_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;
   }

}



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

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



//////////////////////////////
//
// generateAnalysis -- 
//

void generateAnalysis(HumdrumFile& infile, Array<int>& soncondis,
      Array<int>& tertian, Array<int>& terdis, Array<int>& accent, 
      Array<int>& dissic, Array<notearray>& notediss, Array<int>& chord,
      Array<ChordQuality>& cq) {
   soncondis.setSize(infile.getNumLines());
   tertian.setSize(infile.getNumLines());
   terdis.setSize(infile.getNumLines());
   accent.setSize(infile.getNumLines());
   dissic.setSize(infile.getNumLines());
   notediss.setSize(infile.getNumLines());
   cq.setSize(infile.getNumLines());
   chord.setSize(infile.getNumLines());
   int i, j;

   for (i=0; i<notediss.getSize(); i++) {
      notediss[i].setSize(0); 
   }

   infile.analyzeRhythm();

   for (i=0; i<infile.getNumLines(); i++) {
      if (debugQ) {
         cout << "processing line " << (i+1) << " of input ..." << endl;
      }

      if (infile[i].getType() != E_humrec_data) {
         soncondis[i] = -1;
         tertian[i]   = -1;
         terdis[i]    = -1;
         accent[i]    = -1;
      }

      if (infile[i].getType() == E_humrec_interpretation) {
         for (j=0; j<infile[i].getFieldCount(); j++) {
            if (infile[i].getExInterpNum(j) != E_KERN_EXINT) {
               continue;
            }
            if ((strncmp(infile[i][j], "*M", 2) == 0) && 
                  isdigit(infile[i][j][2]) && 
                  (strchr(infile[i][j], '/') != NULL)) {
               sscanf(infile[i][j], "*M%d/%d", &metercount, &meterbase);
               if (compoundQ && (metercount % 3 == 0)) {
                  beatduration = 4.0/meterbase * 3;
               } else {
                  beatduration = 4.0/meterbase;
               }
               break;
            }
         }
      }

      if (infile[i].getType() != E_humrec_data) {
         continue;
      }

      soncondis[i] = measureVerticalConsonance(infile, i);
      tertian[i]   = measureTertian(infile, i);
      terdis[i]    = measureTertianDissonance(infile, i);
      accent[i]    = measureAccentuation(infile, i);
      dissic[i]    = measureDissic(i, soncondis, tertian, terdis, accent);

      
      Array<int> currentnotes;
      infile.getNoteList(currentnotes, i, NL_NOFILL | NL_NOSORT |
            NL_NOUNIQ | NL_NORESTS);
      int m;
      for (m=0; m<currentnotes.getSize(); m++) {
         currentnotes[m] = measureNoteDissonance(i, infile, 
               currentnotes[m], soncondis, accent, dissic);
      }

      notediss[i] = currentnotes;

      measureChordQuality(i, infile, notediss[i], cq);
      chord[i] = measureChordFunction1(i, infile, soncondis, tertian, terdis, 
            accent, dissic, cq);
   }

}


//////////////////////////////
//
// measureChordQuality -- try to find the root and inversion of the
//   given note set.
//

void measureChordQuality(int line, HumdrumFile& score, 
      Array<int>& notediss, Array<ChordQuality>& cq) {

   Array<int> notes;
   score.getNoteList(notes, line, NL_NOPC | NL_FILL | NL_NOSORT | NL_NOUNIQ | 
         NL_NORESTS);
   cq[line] = Convert::noteSetToChordQuality(notes);

   if (cq[line].getType() != E_unknown) {
      return;
   }

   // try getting rid of dissonant notes to see if a chord can be found:

   Array<int> testnotes;
   testnotes.setSize(0);
   score.getNoteList(testnotes, line, NL_NOPC | NL_NOFILL | NL_NOSORT | 
         NL_NOUNIQ | NL_RESTS);

   if (testnotes.getSize() == 0) {
      return;
   } 

   int restcount = 0;
   int i, j;
   for (i=0; i<notes.getSize(); i++) {
      if (notes[i] < 0) {
         restcount++;
      }
   }

   ChordQuality testcq;
   for (i=0; i<testnotes.getSize(); i++) {
      if (notediss[i] == NOTEDISSONANT_YES) {
         for (j=0; j<notes.getSize(); j++) {
            if (notes[j] == testnotes[i]) {
               notes[j] = E_base40_rest;
               restcount++;
            }
         } 
         if (restcount == notes.getSize()) {
            return;
         }
         testcq = Convert::noteSetToChordQuality(notes);
         if (testcq.getType() != E_chord_unknown) {
            cq[line] = testcq;
            return;
         }
      }
   }   

}



//////////////////////////////
//
// measureVerticalConsonance -- apply the rules 1 & 2 from page 337:
//
// Rule 1 & Rule 2.  IF a sonority consists of only one note, OR it
// consists of two notes that form a consonant interval other than a
// perfect fourth, OR it consists of three of more notes forming only
// consonant intervals, THEN it is a consonant vertical, OTHERWISE,
// it is a dissonant vertical.
//

#define CONSONANT_VERTICAL 0
#define DISSONANT_VERTICAL 1
#define UNDEFINED_VERTICAL  -1

int measureVerticalConsonance(HumdrumFile& infile, int line) {
   Array<int> notes;

   if (infile[line].getType() != E_humrec_data) {
      return UNDEFINED_VERTICAL;
   }

   infile.getNoteList(notes, line, NL_FILL | NL_SORT | NL_UNIQ | NL_NORESTS);
   
   // IF a sonority consists of only one note, THEN it is a consonant vertical.
   if (notes.getSize() <= 1) {
      return CONSONANT_VERTICAL;
   }

   // IF a sonority consists of two notes that form a consonant interval
   // other than a perfect fourth, THEN it is a consonant vertical.
   if (notes.getSize() == 2) {
      int interval = notes[1] - notes[0];
      if (interval < 0) {
         cout << "Error on line " << line+1 << " of file: " 
              << "problem determing interval" << endl;
         exit(1);
      }

      interval = interval % 40;
      if (interval == E_base40_per4) {
         return DISSONANT_VERTICAL;
      } else if (interval == E_base40_per1 ||
                 interval == E_base40_maj3 ||
                 interval == E_base40_min3 ||
                 interval == E_base40_min6 ||
                 interval == E_base40_maj6 ||
                 interval == E_base40_per5) {
         return CONSONANT_VERTICAL;
      } else {
         return DISSONANT_VERTICAL;
      }
   }

   // IF a sonority consists of three or more notes forming only 
   // consonant intervals, THEN it is a consonant vertical.
   int i, j;
   int interval;
   for (i=0; i<notes.getSize()-1; i++) {
      for (j=i+1; j<notes.getSize(); j++) {
         interval = notes[j] - notes[i];
         interval = interval % 40;
         if (interval == E_base40_per1 ||
             interval == E_base40_maj3 ||
             interval == E_base40_min3 ||
             interval == E_base40_min6 ||
             interval == E_base40_maj6 ||
             interval == E_base40_per4 ||
             interval == E_base40_per5) {
            // do nothing
         } else {
            return DISSONANT_VERTICAL;
         }
      }
   }

   // the chord notes contain only consonant intervals.
   return CONSONANT_VERTICAL;
}



//////////////////////////////
//
// measureTertian -- apply the rule 5 on page 337:
//
// Rule 5: IF the pitch classes of a vertical sonority can be arranged
// so that each note is separated from its neighbor(s) by a third or fifth,
// and the pitches all lie within a single octave, THEN it is tertian.
//

#define TERTIAN_YES       0
#define TERTIAN_NO        1
#define TERTIAN_UNKNOWN  -1

int measureTertian(HumdrumFile& infile, int line) {
   Array<int> notes;

   if (infile[line].getType() != E_humrec_data) {
      return TERTIAN_UNKNOWN;
   }

   infile.getNoteList(notes, line, NL_PC | NL_FILL | NL_SORT | NL_UNIQ | 
         NL_NORESTS);
   
   // if there is one or fewer notes, then cannot be tertian
   if (notes.getSize() <= 1) {
      return TERTIAN_NO;
   }

   int i, j;
   int foundTertianQ = 0;
   int interval;
   for (i=0; i<notes.getSize(); i++) {
      foundTertianQ = 1;
      for (j=1; j<notes.getSize(); j++) {
         interval = notes[j] - notes[j-1];
         if (interval == E_base40_maj3 ||
             interval == E_base40_min3 ||
             interval == E_base40_dim3 ||
             interval == E_base40_aug3 ||
             interval == E_base40_per5 ||
             interval == E_base40_dim5 ||
             interval == E_base40_aug5) {
            // do nothing
         } else {
            // invalid interval
            foundTertianQ = 0;
            break;
         }
      }
      if (foundTertianQ == 1) {
         return TERTIAN_YES;
      }
      rotateNotes(notes);
   }

   return TERTIAN_NO;
}



//////////////////////////////
//
// measureTertianDissonance -- based on the table on page 338.
//
// Table 1: Levels of Tertian Dissonance, page 338
// 
// Dissonance Level	Chord Quality
// ==============================================
// 	   1		consonant intervals
//      		major triads
//			minor triads
//	   2 		tritone
//			diminished triad
//			minor-minor seventh
//			major-minor seventh
//	   3		augmented fifth
//	   		augmented triad
//	   4		half-diminished seventh
//			fully-diminished sevenths
//			augmented-sixth sonorities
//	   5 		verticals with major sevenths
//			anything else
// 
// 

#define TERTIAN_UNKNOWN  -1
#define TERTIAN_0         0   /* i.e., all rests or single notes */
#define TERTIAN_1         1
#define TERTIAN_2         2
#define TERTIAN_3         3
#define TERTIAN_4         4
#define TERTIAN_5         5

int measureTertianDissonance(HumdrumFile& infile, int line) {
   Array<int> notes;

   if (infile[line].getType() != E_humrec_data) {
      return TERTIAN_UNKNOWN;
   }

   infile.getNoteList(notes, line, NL_PC | NL_FILL | NL_SORT | NL_UNIQ | 
         NL_NORESTS);
   
   if (notes.getSize() == 0) {
      return TERTIAN_0;
   }

   if (notes.getSize() == 1) {
      return TERTIAN_0;
   }

   int i, j;
   int foundTertianQ = 0;
   int interval;
   for (i=0; i<notes.getSize(); i++) {
      foundTertianQ = 1;
      for (j=1; j<notes.getSize(); j++) {
         interval = notes[j] - notes[j-1];
         if (interval == E_base40_maj3 ||
             interval == E_base40_min3 ||
             interval == E_base40_dim3 ||
             interval == E_base40_aug3 ||
             interval == E_base40_per5 ||
             interval == E_base40_dim5 ||
             interval == E_base40_aug5) {
            // do nothing
         } else {
            // invalid interval
            foundTertianQ = 0;
            break;
         }
      }
      if (foundTertianQ == 1) {
         break;
      }
      rotateNotes(notes);
   }

   // filter augmented sixth chords:
   for (i=1; i<notes.getSize(); i++) {
      interval = notes[i] - notes[i-1];
      if (interval == E_base40_aug6 || interval == E_base40_dim3) {
         return TERTIAN_4;
      }
   }

   // filter out non tertian sonorities:
   if (!foundTertianQ) {
      return TERTIAN_5;
   }

   // filter tertian chords with major sevenths:
   interval = notes[notes.getSize() - 1] - notes[0];
   if (interval == E_base40_maj7) {
      return TERTIAN_5;
   }

   // filter out augmented fifth chords:
   if (notes[notes.getSize() - 1] - notes[0] < E_base40_min6) {
      for (i=0; i<notes.getSize() - 1; i++) {
         for (j=i+1; j<notes.getSize(); j++) {
            interval = notes[j] - notes[j-1];
            if (interval == E_base40_aug5) {
               return TERTIAN_3;
            }
         }
      }
   }

   if (notes.getSize() == 2) {
      if (notes[1] - notes[0] == E_base40_dim5) {
         return TERTIAN_2;
      } else {
         return TERTIAN_1;
      }
   }

   int interval1;
   int interval2;
   interval1 = notes[1] - notes[0];
   interval2 = notes[2] - notes[0];
   if (notes.getSize() == 3) {
      if (interval2 == E_base40_per5) {
         if (interval1 == E_base40_maj3 || interval1 == E_base40_min3) {
            return TERTIAN_1;
         } else {
            return TERTIAN_5;
         }
      } else if (interval2 == E_base40_dim5) {
         if (interval1 == E_base40_min3) {
            return TERTIAN_2;
         } else {
            return TERTIAN_5;
         }
      } else if (interval2 == E_base40_min7) {
         if (interval1 == E_base40_min3 ||
             interval1 == E_base40_maj3 ||
             interval1 == E_base40_per5) {
            return TERTIAN_2;
         } else {
            return TERTIAN_5;
         }
      } else if (interval2 == E_base40_dim7) {
         if (interval1 == E_base40_min3 ||
             interval1 == E_base40_dim5 ||
             interval1 == E_base40_per5) {
            return TERTIAN_4;
         } else {
            return TERTIAN_5;
         }
      } else {
         return TERTIAN_5;
      }
   }

   int interval3;
   interval1 = notes[1] - notes[0];
   interval2 = notes[2] - notes[0];
   interval3 = notes[3] - notes[0];
   if (notes.getSize() == 4) {
      if (interval3 == E_base40_maj7) {
         return TERTIAN_5;
      } else if (interval3 == E_base40_min7) {
         if (interval2 == E_base40_per5) {
            if (interval1 == E_base40_min3 || interval1 == E_base40_maj3) {
               return TERTIAN_2;
            } 
         }
      } else if (interval3 == E_base40_dim7) {
         if (interval2 == E_base40_per5) {
            if (interval1 == E_base40_min3) {
               return TERTIAN_4;
            } 
         } else if (interval2 == E_base40_dim5) {
            if (interval1 == E_base40_min3) {
               return TERTIAN_4;
            } 
         } 
      } 
   }

   // couldn't identify tertian chord
   return TERTIAN_5;
}



//////////////////////////////
//
// measureAccentuation -- apply the rule 6 on page 337:
//
// Rule 6.  IF a musical event falls on the "primary beat unit" of
//    the meter (as defined by a table relating meters to beat units)
//    THEN it is accented.
//

#define ACCENT_YES       0
#define ACCENT_NO        1
#define ACCENT_UNKNOWN  -1

int measureAccentuation(HumdrumFile& infile, int line) {
   if (infile[line].getType() != E_humrec_data) {
      return ACCENT_UNKNOWN;
   }

   double fraction;
   fraction = infile[line].getBeat();
   if (compoundQ && (metercount % 3 == 0)) {
      fraction = fraction/3;
   }
   int beat = (int)fraction;
   fraction = fraction - beat;
   if (fraction < 0.001) {
      fraction = 0;
   } else if (fraction > 0.999) {
      fraction = 0;
      beat++;
   }

   if (fraction == 0) {
      return ACCENT_YES;
   } else {
      return ACCENT_NO;
   }
}



//////////////////////////////
//
// measureDissic -- determine dissonance in context value using
//     definition by John Maxwell.
//
// Rule 7: IF [a sonority is not tertian] OR [it is accented AND dissonant AND
// the next sonority is tertian AND the next sonority has a lower 
// tertian-dissonance level] OR [it is unaccented AND dissonant AND the last 
// sonority is tertian AND the last sonority has a lower tertian-dissonance 
// level], THEN the sonority is dissonant in context.
//

#define DISSIC_UNDEFINED -1
#define DISSIC_YES        0
#define DISSIC_NO         1

int measureDissic(int line, Array<int>& soncondis, Array<int>& tertian, 
      Array<int>& terdis, Array<int>& accent) {
   int i;


   // case 1: IF a sonority is not tertian, THEN the sonority is dissonant
   // in context.
   if (tertian[line] == TERTIAN_NO) {
      return DISSIC_YES;
   }


   // case 2: IF a sonority is accented AND dissonant AND the next 
   // sonority is tertian AND the next sonority has a lower tertian-dissonance
   // level, THEN the sonority is dissonant in context.
   int nextdislevel = -1;
   int nexttertian = -1;
   for (i=line+1; i<tertian.getSize(); i++) {
      if (tertian[i] != TERTIAN_UNKNOWN) {
         nextdislevel = terdis[i];
         nexttertian  = tertian[i];
         break;
      } 
   }
   if ((accent[line] == ACCENT_YES) && 
         (soncondis[line] == DISSONANT_VERTICAL) &&
         (nexttertian == TERTIAN_YES) && (nextdislevel < terdis[line])) {
      return DISSIC_YES;
   }


   // case 3: IF a sonority is unaccented AND dissonant AND the last sonority
   // is tertian AND the last sonority has a lower tertian-dissonance level,
   // THEN the sonority is dissonant in context.
   int lasttertian = -1;
   int lastterdis = -1;
   for (i=line-1; i>0; i--) {
      if (tertian[i] != TERTIAN_UNKNOWN) {
         lastterdis = terdis[i];
         lasttertian  = tertian[i];
         break;
      } 
   }
   if ((accent[line] == ACCENT_NO) && 
       (soncondis[line] == DISSONANT_VERTICAL) &&
       (lasttertian == TERTIAN_YES) && (lastterdis < terdis[line])) {
      return DISSIC_YES;
   }

   return DISSIC_NO;
}
     


//////////////////////////////
//
// measureNoteDissonance -- determine if note is dissonant using
//     rules by John Maxwell.
//
// Rule 8: IF a note is in a vertical that is both "dissonant" and 
// "dissonant in context," AND the note formas a dissonant interval with 
// another note in the vertical, AND the vertical is accented, AND the
// note is note continued or repeated in the next vertical, THEN the note
// is a dissonant note.
//
// Rule 9: IF a note is in a vertical that is both "dissonant" and
// "dissonant in context", AND the note forms a dissonant interval
// with another note in the vertical, AND the vertical is unaccented, AND
// the note is not continued or repeated from the previous vertical, THEN
// the note is a dissonant note.
//

int measureNoteDissonance(int line, HumdrumFile& score, int note,
   Array<int>& soncondis, Array<int>& accent, Array<int>& dissic) {

   if (score[line].getType() != E_humrec_data) {
      return NOTEDISSONANT_UNKNOWN;
   }

   if (note == E_base40_rest) {
      return NOTEDISSONANT_UNKNOWN;
   }

   // remove octave information from note:
   note = note % 40;

   int dissintervalQ = 0;
   int interval;
   Array<int> notes;
   score.getNoteList(notes, line, NL_PC | NL_FILL | NL_SORT | NL_UNIQ | 
         NL_NORESTS);
   int i, j;
   for (i=0; i<notes.getSize()-1; i++) {
      for (j=i+1; j<notes.getSize(); j++) {
         interval = notes[j] - notes[i];
         if (interval == E_base40_per1 ||
             interval == E_base40_min3 ||
             interval == E_base40_maj3 ||
             interval == E_base40_per4 ||
             interval == E_base40_per5 ||
             interval == E_base40_min6 ||
             interval == E_base40_maj6) {
            // do nothing
         } else {
            dissintervalQ = 1;
            break;
         }
      }
   }

   Array<int> nextnotes;
   nextnotes.setSize(0);
   int nextline = line+1;
   while ((nextline < score.getNumLines()) && 
         (score[nextline].getType() != E_humrec_data)) {
      nextline++;
   }
   int noteinnextQ = 0;
   if (nextline < score.getNumLines()) {
      score.getNoteList(nextnotes, nextline, NL_PC | NL_FILL | NL_SORT | 
            NL_UNIQ | NL_NORESTS);
      noteinnextQ = 0;
      for (i=0; i<nextnotes.getSize(); i++) {
         if (note == nextnotes[i]) {
            noteinnextQ = 1;
            break;
         }
      }
   } else {
      noteinnextQ = 0;
   }

   // Rule 8:
   if ((soncondis[line] == DISSONANT_VERTICAL) &&
       (dissic[line] == DISSIC_YES) && dissintervalQ &&
       (accent[line] == ACCENT_YES) && !noteinnextQ) {
      return NOTEDISSONANT_YES;
   }

   int noteinlastQ = 0;
   Array<int> lastnotes;
   lastnotes.setSize(0);
   int lastline = line-1;
   while ((lastline > 0) && (score[lastline].getType() != E_humrec_data)) {
      lastline--;
   }
   if (lastline > 0) {
      score.getNoteList(lastnotes, lastline, NL_PC | NL_FILL | NL_SORT | 
            NL_UNIQ | NL_NORESTS);
      noteinlastQ = 0;
      for (i=0; i<lastnotes.getSize(); i++) {
         if (note == lastnotes[i]) {
            noteinlastQ = 1;
            break;
         }
      }
   } else {
      noteinlastQ = 0;
   }

   // Rule 9:
   if ((soncondis[line] == DISSONANT_VERTICAL) &&
       (dissic[line] == DISSIC_YES) && dissintervalQ &&
       (accent[line] == ACCENT_NO) && !noteinlastQ) {
      return NOTEDISSONANT_YES;
   }

   return NOTEDISSONANT_NO;
}



//////////////////////////////
//
// measureChordFunction1 -- apply rules 20 through 28 to identify
//    chord and non-chord sonorities.
//

#define CHORD_UNKNOWN 'X'
#define CHORD         'c'
#define CHORD_PREVSUB 'p'
#define CHORD_NEXTSUB 'n'
#define CHORD_PASSING 'g'
#define CHORD_SUSPEND 's'

int measureChordFunction1(int line, HumdrumFile& score, Array<int>& soncondis, 
      Array<int>& tertian, Array<int>& terdis, Array<int>& accent,
      Array<int>& dissic, Array<ChordQuality>& cq) {

   //
   // preliminaries 
   //

   int lastline = line - 1;
   while (lastline > 0 && score[lastline].getType() != E_humrec_data) {
      lastline--;
   }

   int nextline = line + 1;
   while (nextline < score.getNumLines() && 
         score[nextline].getType() != E_humrec_data) {
      nextline++;
   }
   if (nextline >= score.getNumLines()) {
      nextline = -1;
   }

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

   // Rule 20: IF a vertical is tertian AND its duration is at least as long
   // as the primary beat of the meter, THEN it is a chord.

   if ((tertian[line] == TERTIAN_YES) && 
       (score[line].getDuration() >= beatduration)) {
      return CHORD;
   }

   // Rule 21: IF a vertical is tertian AND it is accented AND it is not
   // "dissonant in context," THEN it is a chord.

   if ((tertian[line] == TERTIAN_YES) && (accent[line] == ACCENT_YES) &&
       (dissic[line] == DISSIC_NO)) {
      return CHORD;
   }

   // Rule 22: IF a vertical is unaccented AND it is tertian AND the previous
   // vertical is tertian AND they both have the same root, THEN the vertical
   // is subordinate to the previous vertical.

   if ((accent[line] == ACCENT_NO) && (tertian[line] == TERTIAN_YES) &&
       (lastline > 0) && (tertian[lastline] == TERTIAN_YES) &&
       (cq[line].getRoot() >= 0) &&
       (cq[line].getRoot() == cq[lastline].getRoot())) {
      return CHORD_PREVSUB;
   }

   // Rule 23: IF a vertical is accented AND it is dissonant in context AND
   // the next vertical is tertian, THEN it is subordinate to the next vertical,
   // and is not an independent chord.

   if ((accent[line] == ACCENT_YES) && (dissic[line] == DISSIC_YES) &&
       (nextline > 0) && (tertian[nextline] == TERTIAN_YES)) {
      return CHORD_NEXTSUB;
   }

   // Rule 24: IF a vertical is unaccented AND it is not dissonant in context
   // AND the last vertical is dissonant in context, THEN the vertical is a 
   // chord.

   if ((accent[line] == ACCENT_NO) && (dissic[line] == DISSIC_NO) &&
      (lastline > 0) && (dissic[lastline] == DISSIC_YES)) {
      return CHORD;
   }

   // Rule 25: IF a vertical is tertian AND the previous vertical is
   // tertian AND the previous vertical is not dissonant in context AND the
   // two verticals do not have the same root, THEN the vertical is a chord,
   // but is marked passing.

   if ((tertian[line] == TERTIAN_YES) && (lastline > 0) &&
       (tertian[lastline] == TERTIAN_YES) && 
       (dissic[lastline] == TERTIAN_NO) && 
       (cq[line].getRoot() >= 0) && (cq[lastline].getRoot() >= 0) &&
       (cq[line].getRoot() != cq[lastline].getRoot())) {
      return CHORD_PASSING;
   }

   // Rule 26: IF a vertical is not tertian AND it is accented, THEN
   // it is subordinate to whatever chord follows.

   if ((tertian[line] == TERTIAN_NO) && (accent[line] == ACCENT_YES)) {
      return CHORD_NEXTSUB;
   }

   // Rule 27: IF a vertical is not tertian AND it is unaccented, THEN
   // it is subordinate to whatever chord precedes it.

   if ((tertian[line] == TERTIAN_NO) && (accent[line] == ACCENT_NO)) {
      return CHORD_PREVSUB;
   }

   return CHORD_UNKNOWN;
}



//////////////////////////////
//
// measureChordFunction2 -- apply rules 28 through 36 to identify
//    chord and non-chord sonorities.
//

int measureChordFunction2(int line, Array<int>& soncondis, Array<int>& tertian, 
      Array<int>& terdis, Array<int>& accent, Array<int>& dissic,
      Array<ChordQuality>& cq) {

   // Rule 28: IF a vertical contains a suspension AND the resolution
   // note of the suspension can be substituted for the suspension note to
   // form an implied tertian vertical AND the implied tertian vertical is
   // more consonant than the original vertical, THEN the implied vertical
   // should be substituted for the suspension vertical, and is made a
   // chord.


   // Rule 29: IF a vertical is unaccented AND it is not dissonant in
   // context AND it is a two-voice sonority AND the previous vertical is a
   // two-voice sonority AND the notes of the vertical can be combined with
   // the notes of the previous vertical producing an implied tertian
   // sonority with the same root as the previous vertical, THEN the
   // unaccented vertical should be subordinate to the previous vertical,
   // and its notes can be included with those of the previous vertical.

   // Rule 30: IF a vertical consists of a single note from a single
   // voice, THEN the vertical is subordinate to the previous chord.

   // Rule 31: IF a chord is in the same key as the previous chord AND
   // the chord has the same harmonic function as the previous chord AND
   // they have the same inversion, THEN the chord can be subordinated to
   // the previous chord.

   // Rule 32: IF a chord is accented AND it is followed by a chord
   // with a weaker harmonic strength, which is followed by a chord with the
   // same function as the first chord, THEN the second two chrods can be
   // subordinated to the first chord.

   // Rule 33: IF a chord is accented AND it is followed by a chord
   // that is unaccented AND the second chord has a weaker harmonic function
   // than the first, THEN the second chord is subordinate to the first.

   // Rule 34: IF a chord contains a suspension AND the resolution of
   // the suspension note can be substituted for the suspension note to
   // produce a chord with a greater function strength that the original
   // chord, THEN the new function should be substituted for the old
   // function. (This rule does note combine chords, but can cause other
   // rules to do so.).

   // Rule 35: IF a chord is accented AND it is on a strong beat AND
   // it is a six-four chord AND the next chord has the same bass note AND
   // the next chord is in root position, THEN the first chord is
   // subordinate to the second.

   // Rule 36: IF a chord is a two-voice chrod AND the previous chord
   // is accented AND the two chords are in the same measure AND the notes
   // of the chord can be combined with those of the previous chord without
   // changing the root of the previous chord, THEN the chord is subordinate
   // to the previous chord.

   return CHORD_UNKNOWN;
}



//////////////////////////////
//
// displayNoteDiss -- display the note dissonance determinations
//

void displayNoteDiss(Array<int>& notediss) {
   int i;
   for (i=0; i<notediss.getSize(); i++) {
      if (notediss[i] == NOTEDISSONANT_YES) {
         cout << 'd';
      } else if (notediss[i] == NOTEDISSONANT_NO) {
         cout << 'c';
      } else {
         cout << 'r';
      }
      if (i < notediss.getSize() - 1) {
         cout << ' ';
      }
   }
}



//////////////////////////////
//
// printAnalysis -- 
//

void printAnalysis(HumdrumFile& infile, Array<int>& soncondis,
      Array<int>& tertian, Array<int>& terdis, Array<int>& accent, 
      Array<int>& dissic, Array<notearray>& notediss, 
      Array<int>& chord, Array<ChordQuality>& cq) {
   int i;
   if (appendQ) {
      for (i=0; i<infile.getNumLines(); i++) {
         switch (infile[i].getType()) {
         case E_humrec_global_comment:
         case E_humrec_bibliography:
         case E_humrec_none:
         case E_humrec_empty:
            cout << infile[i].getLine() << "\n";
            break;
         case E_humrec_data:
            cout << infile[i].getLine() << "\t";
            cout << (char)chord[i];
            cout << "\n";
            break;
         case E_humrec_data_comment:
            if (infile[i].equalFieldsQ("**kern")) {
               cout << infile[i].getLine() << "\t"
                    << infile[i][0] << "\n";
            } else {
               cout << infile[i].getLine() << "\t!\n";
            }
            break;
         case E_humrec_data_measure:
            if (infile[i].equalFieldsQ("**kern")) {
               cout << infile[i].getLine() << "\t"
                    << infile[i][0] << "\n";
            } else {
               cout << infile[i].getLine() << "\t=\n";
            }
            break;
         case E_humrec_data_interpretation:
            if (strncmp(infile[i][0], "**", 2) == 0) {
               cout << infile[i].getLine() << "\t";
               cout << "**chord" << "\n";
            } else if (infile[i].equalFieldsQ("**kern")) {
               cout << infile[i].getLine() << "\t"
                    << infile[i][0] << "\n";
            } else {
               cout << infile[i].getLine() << "\t*\n";
            }
            break;
         }
      }

   } else {

      for (i=0; i<infile.getNumLines(); i++) {
         switch (infile[i].getType()) {
         case E_humrec_global_comment:
         case E_humrec_bibliography:
         case E_humrec_none:
         case E_humrec_empty:
            cout << infile[i].getLine() << "\n";
            break;
         case E_humrec_data:
            cout << (char)chord[i];
            cout << "\n";
            break;
         case E_humrec_data_comment:
            if (infile[i].equalFieldsQ("**kern")) {
               cout << infile[i][0] << "\n";
            } else {
               // do nothing
            }
            break;
         case E_humrec_data_measure:
            if (infile[i].equalFieldsQ("**kern")) {
               cout << infile[i][0] << "\n";
            } else {
               cout << "\t=\n";
            }
            break;
         case E_humrec_data_interpretation:
            if (strncmp(infile[i][0], "**", 2) == 0) {
               cout << "**chord" << "\n";
            } else if (infile[i].equalFieldsQ("**kern")) {
               cout << infile[i][0] << "\n";
            } else {
               // do nothing
            }
            break;
         }
      }
   }
}



//////////////////////////////
//
// rotateNotes -- 
// 

void rotateNotes(Array<int>& notes) {
   if (notes.getSize() < 2) {
      return;
   }

   int note = notes[0];
   int i;
   for (i=0; i<notes.getSize() - 1; i++) {
      notes[i] = notes[i+1];
   }
   notes[notes.getSize()-1] = note + 40;
}



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

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