fftadaptertest.cpp

/*************************************************************************

  Copyright 2011-2015 Ibrahim Sha'ath

  This file is part of LibKeyFinder.

  LibKeyFinder 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 3 of the License, or
  (at your option) any later version.

  LibKeyFinder is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with LibKeyFinder.  If not, see <http://www.gnu.org/licenses/>.

*************************************************************************/

#include "_testhelpers.h"

TEST (FftAdapterTest, ForwardAndBackward) {

  unsigned int frameSize = 4096;
  std::vector<double> original(frameSize);
  KeyFinder::FftAdapter forwards(frameSize);

  for (unsigned int i = 0; i < frameSize; i++) {
    float sample = 0.0;
    sample += sine_wave(i,  2, frameSize, 10000);
    sample += sine_wave(i,  4, frameSize,  8000);
    sample += sine_wave(i,  5, frameSize,  6000);
    sample += sine_wave(i,  7, frameSize,  4000);
    sample += sine_wave(i, 13, frameSize,  2000);
    sample += sine_wave(i, 20, frameSize,   500);
    forwards.setInput(i, sample);
    original[i] = sample;
  }

  forwards.execute();

  for (unsigned int i = 0; i < frameSize; i++) {
    float out = forwards.getOutputMagnitude(i);
    if (i == 2) {
      ASSERT_FLOAT_EQ(10000 / 2 * frameSize, out);
    } else if (i == 4) {
      ASSERT_FLOAT_EQ(8000 / 2 * frameSize, out);
    } else if (i == 5) {
      ASSERT_FLOAT_EQ(6000 / 2 * frameSize, out);
    } else if (i == 7) {
      ASSERT_FLOAT_EQ(4000 / 2 * frameSize, out);
    } else if (i == 13) {
      ASSERT_FLOAT_EQ(2000 / 2 * frameSize, out);
    } else if (i == 20) {
      ASSERT_NEAR(500 / 2 * frameSize, out, 0.1);
    } else {
      ASSERT_GT(5, out);
    }
  }

  KeyFinder::InverseFftAdapter backwards(frameSize);

  for (unsigned int i = 0; i < frameSize; i++) {
    backwards.setInput(i, forwards.getOutputReal(i), forwards.getOutputImaginary(i));
  }

  backwards.execute();

  for (unsigned int i = 0; i < frameSize; i++) {
    ASSERT_NEAR(original[i], backwards.getOutput(i), 0.001);
  }
}