import fractions as F

import xml2b40

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(n / 40) for n in noten])

            x = F.Fraction (tikken, tikkenPerKwart)

            n, d = x.numerator, x.denominator

            t = n / d

            r = x - t

            if r == 0: r = ''

            regels.append ('%4d %3s %s' % (t, r, noten))

    return regels    

if __name__ == '__main__':

    fnm = 'mids/bwv539p_arr.xml'

    noten = xml2b40.xml2b40 (fnm)

    regels = mkAna (noten, fnm)

    print '\n'.join (regels)