BeatTracker.cpp

/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */

/*
  Vamp feature extraction plugin for the BeatRoot beat tracker.

  Centre for Digital Music, Queen Mary, University of London.
  This file copyright 2011 Simon Dixon, Chris Cannam 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 "BeatTracker.h"

EventList BeatTracker::beatTrack(AgentParameters params,
                                 EventList events, EventList beats,
                                 EventList *unfilledReturn)
{
    AgentList agents;
    int count = 0;
    double beatTime = -1;
    if (!beats.empty()) {
	count = beats.size() - 1;
	EventList::iterator itr = beats.end();
	--itr;
	beatTime = itr->time;
    }
    if (count > 0) { // tempo given by mean of initial beats
	double ioi = (beatTime - beats.begin()->time) / count;
	agents.push_back(new Agent(params, ioi));
    } else // tempo not given; use tempo induction
	agents = Induction::beatInduction(params, events);
    if (!beats.empty())
	for (AgentList::iterator itr = agents.begin(); itr != agents.end();
	     ++itr) {
	    (*itr)->beatTime = beatTime;
	    (*itr)->beatCount = count;
	    (*itr)->events = beats;
	}
    agents.beatTrack(events, params, -1);
    Agent *best = agents.bestAgent();
    EventList results;
    if (best) {
        if (unfilledReturn) *unfilledReturn = best->events;
	best->fillBeats(beatTime);
	results = best->events;
    }
    for (AgentList::iterator ai = agents.begin(); ai != agents.end(); ++ai) {
	delete *ai;
    }
    return results;
} // beatTrack()/1