diff AutocorrelationProcessor.cpp @ 25:fe23998968b4

* Added tempogram via autocorrelation feature, using AutocorrelationProcessor * Moved calculateMax() from NoveltyCurveProcessor to SpectrogramProcessor
author Carl Bussey <c.bussey@se10.qmul.ac.uk>
date Wed, 20 Aug 2014 16:00:37 +0100
parents
children ff6110f1144b
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/AutocorrelationProcessor.cpp	Wed Aug 20 16:00:37 2014 +0100
@@ -0,0 +1,52 @@
+//
+//  AutocorrelationProcessor.cpp
+//  Tempogram
+//
+//  Created by Carl Bussey on 20/08/2014.
+//  Copyright (c) 2014 Carl Bussey. All rights reserved.
+//
+
+#include "AutocorrelationProcessor.h"
+using namespace std;
+
+AutocorrelationProcessor::AutocorrelationProcessor(const size_t &windowLength, const unsigned int &hopSize, const unsigned int &lagIncrement) :
+    m_windowLength(windowLength),
+    m_hopSize(hopSize),
+    m_lagIncrement(lagIncrement)
+{
+    //Nothing to do here
+}
+
+AutoCorrelation AutocorrelationProcessor::process(float * input, const size_t &inputLength) const
+{
+    int readBlockPointerIndex = 0;
+    AutoCorrelation autocorrelation;
+    
+    while(readBlockPointerIndex <= (int)inputLength) {
+        int readPointer = readBlockPointerIndex - m_windowLength/2;
+        
+        autocorrelation.push_back(processBlock(&input[readPointer], min(inputLength-readPointer, m_windowLength)));
+        readBlockPointerIndex += m_hopSize;
+    }
+    
+    return autocorrelation;
+}
+
+vector<float> AutocorrelationProcessor::processBlock(float * blockInput, const size_t &blockLength) const
+{
+    vector<float> autocorrelation;
+    
+    int N = m_windowLength/m_lagIncrement;
+    
+    for (int lag = 0; lag < N; lag++){
+        float sum = 0;
+        int sampleLag = m_lagIncrement*lag;
+        
+        for (int n = sampleLag; n < (int)blockLength; n++){
+            sum += blockInput[n-sampleLag]*blockInput[n];
+        }
+        autocorrelation.push_back(sum/(2*N + 1 - lag));
+    }
+    
+    return autocorrelation;
+}
\ No newline at end of file