FFTransform_UnitTest.cc File Reference

#include <Nsound/Buffer.h>
#include <Nsound/Cosine.h>
#include <Nsound/FFTransform.h>
#include <Nsound/Plotter.h>
#include <Nsound/Sine.h>
#include <Nsound/Wavefile.h>
#include "UnitTest.h"
#include <stdlib.h>
#include <iostream>

Include dependency graph for FFTransform_UnitTest.cc:

Go to the source code of this file.

Functions
void	FFTransform_UnitTest ()

Variables
static const char *	THIS_FILE = "FFTransform_UnitTest.cc"
static const float64	GAMMA = 1.0e-11

Function Documentation

void FFTransform_UnitTest

(

)

Definition at line 53 of file FFTransform_UnitTest.cc.

References Nsound::Buffer::abs(), Nsound::Generator::buzz(), Nsound::FFTransform::fft(), GAMMA, Nsound::Buffer::getAbs(), Nsound::FFTChunk::getImaginary(), Nsound::Buffer::getLength(), Nsound::Buffer::getMax(), Nsound::FFTChunk::getReal(), Nsound::FFTransform::ifft(), M_PI, Nsound::Buffer::plot(), Nsound::FFTChunk::setCartesian(), Nsound::Wavefile::setDefaultSampleSize(), Nsound::Wavefile::setIEEEFloat(), Nsound::FFTChunk::setPolar(), Nsound::Plotter::show(), SUCCESS, TEST_ERROR_HEADER, TEST_HEADER, and THIS_FILE.

Referenced by main().

{
    cout << endl << THIS_FILE;

    Wavefile::setDefaultSampleSize(64);
    Wavefile::setIEEEFloat(true);

    Cosine cos(512);

    Buffer input = cos.buzz(1.0, 8.0, 2.0, 0.0);

    FFTransform fft(512);

    FFTChunkVector vec;

    Buffer gold;

    cout << TEST_HEADER << "Testing FFTransform::fft(), ifft() ..." << flush;

    vec = fft.fft(input, 128, 0);

    if(vec.size() != 512 / 128)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match!"
             << endl
             << "vec.size() = " << vec.size() << " != " << 512 / 128
             << endl;

        exit(1);
    }

    Buffer data = fft.ifft(vec);

//~    input.plot("input");
//~    data.plot("fft,ifft");

    if(data.getLength() != input.getLength())
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match!"
             << endl
             << "result.getLength() = " << data.getLength() << " != "
             << input.getLength()
             << endl;

        exit(1);
    }

    // Create gold file
//~    data >> "gold/FFTransform_out1.wav";

    gold = Buffer("gold/FFTransform_out1.wav");

    Buffer diff = data - gold;

    if(diff.getAbs().getMax() > GAMMA)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("input - gold");
        data.plot("data");
        gold.plot("gold");

        Plotter::show();
        exit(1);
    }

    cout << SUCCESS;

    cout << TEST_HEADER << "Testing FFTChunk::getReal(), getImaginary() ..." << flush;

    vec = fft.fft(input, 128, 0);

    Buffer real = vec[0].getReal();
    Buffer img  = vec[0].getImaginary();

    // Create gold files.
//~    real >> "gold/FFTransform_out2.wav";
//~    img  >> "gold/FFTransform_out3.wav";

    gold = Buffer("gold/FFTransform_out2.wav");

    diff = real - gold;
    diff.abs();
    if(diff.getMax() > GAMMA)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("real - gold");
        real.plot("real");
        gold.plot("gold");

        Plotter::show();
        exit(1);
    }

    gold = Buffer("gold/FFTransform_out3.wav");

    diff = img - gold;
    diff.abs();

    if(diff.getMax() > GAMMA)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("img - gold");
        img.plot("img");
        gold.plot("gold");

        Plotter::show();
        exit(1);
    }

    cout << SUCCESS;

    cout << TEST_HEADER << "Testing FFTChunk::getMagnitude(), getPhase() ..." << flush;

    vec = fft.fft(input, 128, 0);

    Buffer mag   = vec[0].getMagnitude();
    Buffer phase = vec[0].getPhase();

    // Create gold files.
//~    mag >> "gold/FFTransform_out4.wav";
//~    phase >> "gold/FFTransform_out5.wav";

    gold = Buffer("gold/FFTransform_out4.wav");

    diff = mag - gold;

    if(diff.getAbs().getMax() > GAMMA)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("data - gold");
        mag.plot("data");
        gold.plot("gold");

        Plotter::show();
        exit(1);
    }

    gold = Buffer("gold/FFTransform_out5.wav");

    diff = phase - gold;

    diff(diff > 2.0 * M_PI - 0.01) = 0.0;
    diff(diff < 2.0 * M_PI - 0.01) = 0.0;

    if(diff.getAbs().getMax() > GAMMA)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("phase - gold");
        phase.plot("phase");
        gold.plot("gold");

        Plotter::show();
        exit(1);
    }

    cout << SUCCESS;

    cout << TEST_HEADER << "Testing FFTChunk::toCartesian(), toPolar() ..." << flush;

    vec[0].toPolar();

    mag   = vec[0].getMagnitude();
    phase = vec[0].getPhase();

    gold = Buffer("gold/FFTransform_out4.wav");

    diff = mag - gold;

    if(diff.getAbs().getMax() > GAMMA)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("data - gold");
        mag.plot("data");
        gold.plot("gold");

        Plotter::show();
        exit(1);
    }

    gold = Buffer("gold/FFTransform_out5.wav");

    diff = phase - gold;
    diff.abs();
    if(diff.getMax() > 2*M_PI)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("phase - gold");
        phase.plot("phase");
        gold.plot("gold");

        Plotter::show();
        exit(1);
    }

    cout << SUCCESS;

    cout << TEST_HEADER << "Testing FFTChunk::setCartesian(), setPolar() ..." << flush;

    vec = fft.fft(input, 512, 0);

    mag = vec[0].getMagnitude();
    phase = vec[0].getPhase();

    FFTChunk orig(vec[0]);
    FFTChunk chunk(512, 512);

    chunk.setPolar(mag, phase);

    vec[0] = chunk;

    data = fft.ifft(vec);

    diff = data - input;

    if(diff.getAbs().getMax() > 0.005)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("data - gold");
        data.plot("data");
        input.plot("gold");

        Plotter::show();
        exit(1);
    }

    chunk.setCartesian(orig.getReal(), orig.getImaginary());

    vec[0] = chunk;

    data = fft.ifft(vec);

    diff = data - input;

    if(diff.getAbs().getMax() > 2*GAMMA)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("data - gold");
        data.plot("data");
        input.plot("gold");

        Plotter::show();
        exit(1);
    }

    cout << SUCCESS;

    cout << TEST_HEADER << "Testing FFTChunk::getFrequencyAxis() ..." << flush;

    data = vec[0].getFrequencyAxis();

//~    // Create gold files.
//~    data >> "gold/FFTransform_out6.wav";

    gold = Buffer("gold/FFTransform_out6.wav");

    diff = data - gold;

    if(diff.getAbs().getMax() > 1e-16)
    {
        cerr << TEST_ERROR_HEADER
             << "Output did not match gold file!"
             << endl;

        diff.plot("data - gold");
        data.plot("data");
        gold.plot("gold");

        Plotter::show();
        exit(1);
    }

    cout << SUCCESS << endl;
}

Variable Documentation

const char* THIS_FILE = "FFTransform_UnitTest.cc"

static

Definition at line 48 of file FFTransform_UnitTest.cc.

Referenced by FFTransform_UnitTest().

const float64 GAMMA = 1.0e-11

static

Definition at line 50 of file FFTransform_UnitTest.cc.

Referenced by FFTransform_UnitTest().