libs/rubberband/src/SpectralDifferenceAudioCurve.cpp

   1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
   2
   3 /*
   4     Rubber Band
   5     An audio time-stretching and pitch-shifting library.
   6     Copyright 2007 Chris Cannam.
   7
   8     This program is free software; you can redistribute it and/or
   9     modify it under the terms of the GNU General Public License as
  10     published by the Free Software Foundation; either version 2 of the
  11     License, or (at your option) any later version.  See the file
  12     COPYING included with this distribution for more information.
  13 */
  14
  15 #include "SpectralDifferenceAudioCurve.h"
  16
  17 namespace RubberBand
  18 {
  19
  20 SpectralDifferenceAudioCurve::SpectralDifferenceAudioCurve(size_t sampleRate, size_t windowSize) :
  21     AudioCurve(sampleRate, windowSize)
  22 {
  23     m_prevMag = new double[m_windowSize/2 + 1];
  24
  25     for (size_t i = 0; i <= m_windowSize/2; ++i) {
  26         m_prevMag[i] = 0.f;
  27     }
  28 }
  29
  30 SpectralDifferenceAudioCurve::~SpectralDifferenceAudioCurve()
  31 {
  32     delete[] m_prevMag;
  33 }
  34
  35 void
  36 SpectralDifferenceAudioCurve::reset()
  37 {
  38     for (size_t i = 0; i <= m_windowSize/2; ++i) {
  39         m_prevMag[i] = 0;
  40     }
  41 }
  42
  43 void
  44 SpectralDifferenceAudioCurve::setWindowSize(size_t newSize)
  45 {
  46     m_windowSize = newSize;
  47 }
  48
  49 float
  50 SpectralDifferenceAudioCurve::process(float *mag, size_t increment)
  51 {
  52     float result = 0.0;
  53
  54     for (size_t n = 0; n <= m_windowSize / 2; ++n) {
  55         result += sqrtf(fabsf((mag[n] * mag[n]) -
  56                               (m_prevMag[n] * m_prevMag[n])));
  57         m_prevMag[n] = mag[n];
  58     }
  59
  60     return result;
  61 }
  62
  63 }
  64