BeatRoot Vamp Plugin

/* -*- 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.

*/

#ifndef _AGENT_LIST_H_

#define _AGENT_LIST_H_

#include "Agent.h"

#include "Event.h"

#include <vector>

#include <algorithm>

#ifdef DEBUG_BEATROOT

#include <iostream>

#endif

/** Class for maintaining the set of all Agents involved in beat tracking a piece of music.

 */

class AgentList

{

public:

    typedef std::vector<Agent *> Container;

    typedef Container::iterator iterator;

protected:

    Container list;

    static bool agentComparator(const Agent *a, const Agent *b) {

        if (a->beatInterval == b->beatInterval) {

            return a->idNumber < b->idNumber; // ensure stable ordering

        } else {

            return a->beatInterval < b->beatInterval;

        }

    }

public:

    // expose some vector methods

    //!!! can we remove these again once the rest of AgentList is implemented?

    bool empty() const { return list.empty(); }

    Container::iterator begin() { return list.begin(); }

    Container::iterator end() { return list.end(); }

    size_t size() { return list.size(); }

    void push_back(Agent *a) {

        list.push_back(a);

#ifdef DEBUG_BEATROOT

        std::cerr << "  Added Ag#" << a->idNumber << ", have " << list.size() << " agent(s)" << std::endl;

#endif

    }

    /** Flag for choice between sum and average beat salience values for Agent scores.

     *  The use of summed saliences favours faster tempi or lower metrical levels. */

    static bool useAverageSalience;

    /** For the purpose of removing duplicate agents, the default JND of IBI */

    static const double DEFAULT_BI;

    /** For the purpose of removing duplicate agents, the default JND of phase */

    static const double DEFAULT_BT;

    /** Inserts newAgent into the list in ascending order of beatInterval */

    void add(Agent *a) {

        add(a, true);

    } // add()/1

    /** Appends newAgent to list (sort==false), or inserts newAgent into the list

     *  in ascending order of beatInterval

     *  @param newAgent The agent to be added to the list

     *  @param sort Flag indicating whether the list is sorted or not

     */

    void add(Agent *newAgent, bool sort){

        push_back(newAgent);

        if (sort) this->sort();

    } // add()/2

    /** Sorts the AgentList by increasing beatInterval. */

    void sort() {

#ifdef DEBUG_BEATROOT

        std::cerr << "sort: before: ";

        for (iterator i = list.begin(); i != list.end(); ++i) {

            std::cerr << (*i)->idNumber << " ";

        }

        std::cerr << std::endl;

#endif

        std::sort(list.begin(), list.end(), agentComparator);

#ifdef DEBUG_BEATROOT

        std::cerr << "sort: after: ";

        for (iterator i = list.begin(); i != list.end(); ++i) {

            std::cerr << (*i)->idNumber << " ";

        }

        std::cerr << std::endl;

#endif

    } // sort()

    /** Removes the given item from the list.

     *  @param ptr Points to the Agent which is removed from the list

     */

    void remove(iterator itr) {

        list.erase(itr);

    } // remove()

protected:

    /** Removes Agents from the list which are duplicates of other Agents.

     *  A duplicate is defined by the tempo and phase thresholds

     *  thresholdBI and thresholdBT respectively.

     */

    void removeDuplicates();

public:

    /** Perform beat tracking on a list of events (onsets).

     *  @param el The list of onsets (or events or peaks) to beat track

     */

    void beatTrack(EventList el, AgentParameters params) {

        beatTrack(el, params, -1.0);

    } // beatTrack()/1

    /** Perform beat tracking on a list of events (onsets).

     *  @param el The list of onsets (or events or peaks) to beat track.

     *  @param stop Do not find beats after <code>stop</code> seconds.

     */

    void beatTrack(EventList el, AgentParameters params, double stop);

    /** Finds the Agent with the highest score in the list, or NULL if beat tracking has failed.

     *  @return The Agent with the highest score

     */

    Agent *bestAgent();

}; // class AgentList

#endif