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-2008 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 float[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     delete[] m_prevMag;
  47     m_windowSize = newSize;
  48
  49     m_prevMag = new float[m_windowSize/2 + 1];
  50
  51     reset();
  52 }
  53
  54 float
  55 SpectralDifferenceAudioCurve::process(const float *R__ mag, size_t increment)
  56 {
  57     float result = 0.0;
  58
  59     for (size_t n = 0; n <= m_windowSize / 2; ++n) {
  60         result += sqrtf(fabsf((mag[n] * mag[n]) -
  61                               (m_prevMag[n] * m_prevMag[n])));
  62         m_prevMag[n] = mag[n];
  63     }
  64
  65     return result;
  66 }
  67
  68 float
  69 SpectralDifferenceAudioCurve::processDouble(const double *R__ mag, size_t increment)
  70 {
  71     float result = 0.0;
  72
  73     for (size_t n = 0; n <= m_windowSize / 2; ++n) {
  74         result += sqrtf(fabsf((mag[n] * mag[n]) -
  75                               (m_prevMag[n] * m_prevMag[n])));
  76         m_prevMag[n] = (float)mag[n];
  77     }
  78
  79     return result;
  80 }
  81
  82 }
  83