FilterLeastSquaresFIR_UnitTest.cc File Reference

#include <Nsound/AudioStream.h>
#include <Nsound/Buffer.h>
#include <Nsound/FilterLeastSquaresFIR.h>
#include <Nsound/Plotter.h>
#include <Nsound/Sine.h>
#include <Nsound/Wavefile.h>
#include "UnitTest.h"
#include <cmath>
#include <stdlib.h>
#include <iostream>

Include dependency graph for FilterLeastSquaresFIR_UnitTest.cc:

Go to the source code of this file.

Macros
#define	THIS_FILE   "FilterLeastSquaresFIR_UnitTest.cc"
#define	GAMMA   1.5e-14

Functions
void	FilterLeastSquaresFIR_UnitTest ()

Macro Definition Documentation

#define THIS_FILE   "FilterLeastSquaresFIR_UnitTest.cc"

Definition at line 50 of file FilterLeastSquaresFIR_UnitTest.cc.

Referenced by FilterLeastSquaresFIR_UnitTest().

#define GAMMA   1.5e-14

Definition at line 52 of file FilterLeastSquaresFIR_UnitTest.cc.

Referenced by FilterLeastSquaresFIR_UnitTest().

Function Documentation

void FilterLeastSquaresFIR_UnitTest

(

)

Definition at line 54 of file FilterLeastSquaresFIR_UnitTest.cc.

References Nsound::Buffer::abs(), Nsound::FilterLeastSquaresFIR::filter(), GAMMA, Nsound::Buffer::getAbs(), Nsound::Buffer::getdB(), Nsound::Filter::getFrequencyResponse(), Nsound::Buffer::getMax(), Nsound::FilterLeastSquaresFIR::makeKernel(), Nsound::Buffer::plot(), Nsound::RECTANGULAR, Nsound::Wavefile::setDefaultSampleSize(), Nsound::Wavefile::setIEEEFloat(), Nsound::FilterLeastSquaresFIR::setWindow(), Nsound::Plotter::show(), sr, SUCCESS, TEST_ERROR_HEADER, TEST_HEADER, and THIS_FILE.

Referenced by main().

{
    cout << endl << THIS_FILE;

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

    float64 sr = 1000.0;

    // Create a low pass least squares FIR filter

    Buffer freqs;
    Buffer amplitude;

    freqs     << 0.0 << 100.0 << 100.0 << 500.0;
    amplitude << 1.0 << 1.0   << 0.0   << 0.0;

    FilterLeastSquaresFIR f(sr, 1025, freqs, amplitude);

    Buffer noise("gold/Filter_noise.wav");

    cout << TEST_HEADER << "Testing FilterLeastSquaresFIR::filter(input) ...";

    Buffer data;

    data  = f.filter(noise);

    // Create the gold file
    //~    data >> "gold/FilterLeastSquaresFIR_out1.wav";

    Buffer gold("gold/FilterLeastSquaresFIR_out1.wav");

    Buffer diff = data - gold;

    diff.abs();

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

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

        Plotter::show();

        exit(1);
    }

    cout << SUCCESS;

    // Repeated to test that reset() is being called.
    cout << TEST_HEADER << "Testing FilterLeastSquaresFIR::filter(input) ..."
         << flush;

    data = f.filter(noise);

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

    if(diff.getMax() > GAMMA)
    {
        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;

    // Create a high pass least squares FIR filter

    freqs = Buffer(4);
    amplitude = Buffer(4);

    freqs     << 0.0 << 100.0 << 100.0 << 500.0;
    amplitude << 0.0 << 0.0   << 1.0   << 1.0;

    f.makeKernel(freqs, amplitude);

    cout << TEST_HEADER << "Testing FilterLeastSquaresFIR::filter(input) ...";

    data = f.filter(0.9 * noise);

    // Create the gold file
    //~    data >> "gold/FilterLeastSquaresFIR_out2.wav";

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

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

    if(diff.getMax() > GAMMA)
    {
        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;

    // Repeated to test that reset() is being called.
    cout << TEST_HEADER << "Testing FilterLeastSquaresFIR::filter(input) ...";

    data = f.filter(0.9 * noise);

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

    if(diff.getMax() > GAMMA)
    {
        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;

    // Repeated to test that reset() is being called.
    cout << TEST_HEADER << "Testing FilterLeastSquaresFIR::setWindow() ...";

    freqs     = Buffer(4);
    amplitude = Buffer(4);

    freqs     << 0.0 << 30.0 << 30.0 << 500.0;
    amplitude << 1.0 << 1.0   << 0.0   << 0.0;

    f = FilterLeastSquaresFIR(sr, 32, freqs, amplitude);

    f.setWindow(RECTANGULAR);

    data = f.getFrequencyResponse().getdB();

//~    // Create gold file
//~    cout << "SAVING GOLD FILE: gold/FilterLeastSquaresFIR_out3.wav" << endl;
//~    data >> "gold/FilterLeastSquaresFIR_out3.wav";

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

    diff = data - gold;

    float64 max_diff = diff.getAbs().getMax();

    if(max_diff > 1e-9) // gamma is changed here
    {
        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;
}