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:

            r = x - t

            if r == 0: r = ''

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

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)