#!/usr/bin/env python3

import os, xml2b40
import fractions as F
import segLibB40 as S

def keySeg (iseg,jseg):
    hist = S.getSegment (iseg, jseg)
    ks = S.keyCor (hist)
    c1, key, m = ks [0] # correlation coeff, key, major/minor
    c2, _, _ = ks [1]   # second best key
    conf = 300 * (c1 - c2) / c2
    if conf > 100: conf = 10
    return conf, S.b40nm (key) + (m and 'm' or ' ')

def mkAna (events, fnm):
    kgroep = S.readEvents (events)
    tikkenPerKwart = 384
    S.mkWeights (kgroep)
    conf, key = keySeg (0, len (kgroep)-1)
    tonalcentre = key[:2] if key[1] in 'b#' else key[:1]  # cut minor from key
    S.mkTemplates ()
    segs, kgroep = S.analyseK (kgroep, debug=0)
    regels = []
    regels.append ('File: %s, ticks per quarternote: %d' % (fnm, tikkenPerKwart))
    regels.append ('%d note groups, %d chord segments' % (len (kgroep), len (segs)))
    regels.append ('Key: %s, confidence: %.0f%%' % (key, conf))
    for (kb, ke, score, chord, _) in segs:
        grpsInSeg = kgroep [kb:ke]
        totNotes = sum ([len (ns) for ts, ns in grpsInSeg])
        percnt = 100. * score / totNotes
        regels.append ('---- chord %s, score: %.0f%%' % (chord, percnt))
        for tikken, noten in grpsInSeg:
            noten = ', '.join ([S.b40nm (n) + str(int (n / 40)) for n in noten])
            x = F.Fraction (tikken, tikkenPerKwart)
            n, d = x.numerator, x.denominator
            t = int (n / d)
            r = x - t
            if r == 0: r = ''
            regels.append ('%4d %3s %s' % (t, r, noten))
    return regels

if __name__ == '__main__':
    from optparse import OptionParser
    parser = OptionParser (usage='%prog [-h] [-g TPQ] [--C5] <file1>')
    parser.add_option ("-t", action="store", type="int", help="ticks per quarternote", default=384, metavar='TPQ')
    parser.add_option ("--C5", action="store_true", help="central C is C5", default=False)
    options, args = parser.parse_args ()
    if len (args) < 1: parser.error ('file argument needed')
    fnmext = args [0]
    fnm, ext = os.path.splitext (fnmext)
    if ext != '.xml': parser.error ('.xml file needed file needed')
    if not os.path.exists (fnmext): parser.error ('%s does not exist' % fnmext)

    noten = xml2b40.xml2b40 (fnmext, options.t, options.C5)
    regels = mkAna (noten, fnmext)
    print ('\n'.join (regels))