beatanalyser.h

Go to the documentation of this file.

/****************************************************************
 *
 * Copyright (c) 2010
 *
 * Fraunhofer Institute for Manufacturing Engineering   
 * and Automation (IPA)
 *
 * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 *
 * Project name: care-o-bot
 * ROS stack name: cob_driver
 * ROS package name: cob_light
 * Description: Switch robots led color by sending data to
 * the led-µC over serial connection.
 *                              
 * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 *          
 * Author: Benjamin Maidel, email:benjamin.maidel@ipa.fraunhofer.de
 * Supervised by: Benjamin Maidel, email:benjamin.maidel@ipa.fraunhofer.de
 *
 * Date of creation: August 2012
 * ToDo:
 *
 * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Neither the name of the Fraunhofer Institute for Manufacturing 
 *       Engineering and Automation (IPA) nor the names of its
 *       contributors may be used to endorse or promote products derived from
 *       this software without specific prior written permission.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License LGPL as 
 * published by the Free Software Foundation, either version 3 of the 
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License LGPL for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public 
 * License LGPL along with this program. 
 * If not, see <http://www.gnu.org/licenses/>.
 *
 ****************************************************************/

#ifndef BEATANALYSER_H
#define BEATANALYSER_H
#include <vector>
#include "band.h"
#include "inttypes.h"
#include "fft.h"
#include "alsarecorder.h"

namespace mybeat
{
    class BeatAnalyser
    {
    public:
        BeatAnalyser(uint16_t m_bandCount,uint32_t m_sampleRate, uint16_t m_recordSize);
        ~BeatAnalyser();

        void setFFT(FFT* value){m_FFT=value;}

        void processData();

        double getMagSpectrum(uint32_t low, uint32_t high);

        uint16_t getBands(){return m_bandCount;}

        bool getBeat(uint16_t pos);

        bool getBeatFrequency(uint32_t frequency);

        bool getDrumBeat();

        bool getSnareBeat();

        Band* getBand(uint16_t pos);

        double getMaxBandValue() { return m_maxBandValue; }

    private:
        uint16_t m_bandCount;
        uint32_t m_sampleRate;
        uint16_t m_recordSize;
        FFT* m_FFT;
        std::vector<Band*> *m_SubBands;
        std::vector<bool> *m_beats;
        double m_maxBandValue;
    };
}

#endif