samplebrain/brain/src/mfcc.h

/**
 * @file Mfcc.h
 *
 * Calculation of MFCC signal features.
 *
 * This file is part of the Aquila DSP library.
 * Aquila is free software, licensed under the MIT/X11 License. A copy of
 * the license is provided with the library in the LICENSE file.
 *
 * @package Aquila
 * @version 3.0.0-dev
 * @author Zbigniew Siciarz
 * @date 2007-2014
 * @license http://www.opensource.org/licenses/mit-license.php MIT
 * @since 3.0.0
 */

#ifndef MFCC_H
#define MFCC_H

#include "aquila/global.h"
//#include "aquila/transform/FftFactory.h"
#include <spiralcore/sample.h>
#include <cstddef>
#include <vector>

namespace Aquila
{

    /**
     * The Mfcc class implements calculation of MFCC features from input signal.
     *
     * MFCC coefficients are commonly used in speech recognition. The common
     * workflow is to split input signal in frames of equal length and apply
     * MFCC calculation to each frame individually. Hence a few assumptions
     * were made here:
     *
     * - a single Mfcc instance can be used only to process signals of equal
     *   length, for example consecutive frames
     * - if you need to handle signals of various lengths, just create new
     *   Mfcc object per each signal source
     *
     * The code below is a simplest possible example of how to calculate MFCC
     * for each frame of input signal.
     *
     * FramesCollection frames(data, FRAME_SIZE);
     * Mfcc mfcc(FRAME_SIZE);
     * for (Frame& frame : frames) {
     *    auto mfccValues = mfcc.calculate(frame);
     *    // do something with the calculated values
     * }
     *
     */
    class AQUILA_EXPORT Mfcc
    {
    public:
        /**
         * Constructor creates the FFT object to reuse between calculations.
         *
         * @param inputSize input length (common to all inputs)
         */
        Mfcc(std::size_t inputSize, unsigned int sampleRate):
        m_inputSize(inputSize), // m_fft(FftFactory::getFft(inputSize))
        m_sampleRate(sampleRate)
        {
        }

        std::vector<double> calculate(SpectrumType s, std::size_t numFeatures = 12);

    private:
        /**
         * Number of samples in each processed input.
         */
        const std::size_t m_inputSize;
        const unsigned int m_sampleRate;
        
        /**
         * FFT calculator.
         */
        //std::shared_ptr<Fft> m_fft;
    };
}

#endif // MFCC_H