view AutocorrelationProcessor.cpp @ 43:4cf2d163127b

Copyrights and tidying
author Chris Cannam
date Thu, 25 Sep 2014 15:13:45 +0100
parents 1ad47a9afc2e
children a908a5a56267
line wrap: on
line source
/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */

/*
   Vamp Tempogram Plugin
   Carl Bussey, Centre for Digital Music, Queen Mary University of London
   Copyright 2014 Queen Mary University of London.
    
   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 "AutocorrelationProcessor.h"
using namespace std;
#include <iostream>

AutocorrelationProcessor::AutocorrelationProcessor(const size_t &windowLength, const unsigned int &hopSize) :
    m_windowLength(windowLength),
    m_hopSize(hopSize)
{

}

AutocorrelationProcessor::~AutocorrelationProcessor()
{

}

AutoCorrelation AutocorrelationProcessor::process(float * input, const size_t &inputLength) const
{
    int readBlockPointerIndex = 0;
    AutoCorrelation autocorrelation;
    
    while(readBlockPointerIndex <= (int)inputLength) {
        
        vector<float> autocorrelationBlock;
        
        for (int lag = 0; lag < (int)m_windowLength; lag++){
            float sum = 0;
            int readPointer = readBlockPointerIndex - m_windowLength/2;
            
            for (int n = 0; n < (int)m_windowLength; n++){
                if (readPointer+lag >= (int)inputLength) break;
                else if (readPointer >= 0) sum += input[readPointer]*input[readPointer+lag];
                //else cout << readPointer << " : "<< lag << "/" << m_windowLength << endl;
                
                readPointer++;
            }
            autocorrelationBlock.push_back(sum/(2*m_windowLength + 1 - lag));
        }
        
        //autocorrelation.push_back(processBlock());
        autocorrelation.push_back(autocorrelationBlock);
        readBlockPointerIndex += m_hopSize;
    }
    
    return autocorrelation;
}