summaryrefslogtreecommitdiff
path: root/libs/qm-dsp/dsp/rhythm/BeatSpectrum.cpp
blob: 3242dff6def143adf9828f9471ea4b9357b03d8c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */

/*
    QM DSP Library

    Centre for Digital Music, Queen Mary, University of London.
    This file copyright 2008 Kurt Jacobson and QMUL.

    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
    published by the Free Software Foundation; either version 2 of the
    License, or (at your option) any later version.  See the file
    COPYING included with this distribution for more information.
*/

#include "BeatSpectrum.h"

#include "maths/CosineDistance.h"

using std::vector;

vector<double> BeatSpectrum::process(const vector<vector<double> > &m)
{
    int origin = 0;
    int sz = m.size()/2;

    int i, j, k;

    vector<double> v(sz);
    for (i = 0; i < sz; ++i) v[i] = 0.0;

    CosineDistance cd;

    for (i = origin; i < origin + sz; ++i) {

        k = 0;

        for (j = i + 1; j < i + sz + 1; ++j) {

            v[k++] += cd.distance(m[i], m[j]);
        }
    }

    // normalize

    double max = 0.0;

    for (i = 0; i < sz; ++i) {
        if (v[i] > max) max = v[i];
    }

    if (max > 0.0) {
        for (i = 0; i < sz; ++i) {
            v[i] /= max;
        }
    }

    return v;
}