FilterBandRejectIIR.cc

Go to the documentation of this file.

//-----------------------------------------------------------------------------
//
//  $Id: FilterBandRejectIIR.cc 874 2014-09-08 02:21:29Z weegreenblobbie $
//
//  Copyright (c) 2006 Nick Hilton
//
//  weegreenblobbie_yahoo_com (replace '_' with '@' and '.')
//
//-----------------------------------------------------------------------------

//-----------------------------------------------------------------------------
//
//  This program 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 2 of the License, or
//  (at your option) any later version.
//
//  This program 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 Library General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with this program; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
//-----------------------------------------------------------------------------

#include <Nsound/AudioStream.h>
#include <Nsound/Buffer.h>
#include <Nsound/FFTransform.h>
#include <Nsound/FilterLowPassIIR.h>
#include <Nsound/FilterHighPassIIR.h>
#include <Nsound/FilterBandRejectIIR.h>
#include <Nsound/Plotter.h>

#include <cmath>
#include <cstdio>
#include <iostream>

using namespace Nsound;

using std::cerr;
using std::endl;

#define CERR_HEADER __FILE__ << ":" << __LINE__ << ": "

FilterBandRejectIIR::
FilterBandRejectIIR(
    const float64 & sample_rate,
    uint32 n_poles,
    const float64 & frequency_Hz_low,
    const float64 & frequency_Hz_high,
    const float64 & percent_ripple)
    :
    Filter(sample_rate),
    low_(new  FilterLowPassIIR(
        sample_rate,
        n_poles,
        frequency_Hz_low,
        percent_ripple)),
    high_(new FilterHighPassIIR(
        sample_rate,
        n_poles,
        frequency_Hz_high,
        percent_ripple))
{
    reset();

    kernel_size_ = n_poles;
}

FilterBandRejectIIR::
~FilterBandRejectIIR()
{
    delete low_;
    delete high_;
}

float64
FilterBandRejectIIR::
getFrequencyLow() const
{
    return low_->getFrequency();
}

float64
FilterBandRejectIIR::
getFrequencyHigh() const
{
    return high_->getFrequency();
}

AudioStream
FilterBandRejectIIR::
filter(const AudioStream & x)
{
    return Filter::filter(x);
}

AudioStream
FilterBandRejectIIR::
filter(const AudioStream & x, const float64 & f)
{
    return Filter::filter(x);
}

AudioStream
FilterBandRejectIIR::
filter(const AudioStream & x, const Buffer & frequencies)
{
    return Filter::filter(x);
}

AudioStream
FilterBandRejectIIR::
filter(
    const AudioStream & x,
    const float64 & f_low,
    const float64 & f_high)
{
    if(!is_realtime_) reset();

    return low_->filter(x, f_low)
         + high_->filter(x, f_high);
}

AudioStream
FilterBandRejectIIR::
filter(
    const AudioStream & x,
    const Buffer & frequencies_Hz_low,
    const Buffer & frequencies_Hz_high)
{
    if(!is_realtime_) reset();

    return low_->filter(x, frequencies_Hz_low)
         + high_->filter(x, frequencies_Hz_high);
}

Buffer
FilterBandRejectIIR::
filter(const Buffer & x)
{
    return Filter::filter(x);
}

Buffer
FilterBandRejectIIR::
filter(const Buffer & x, const float64 & f)
{
    return Filter::filter(x);
}

Buffer
FilterBandRejectIIR::
filter(const Buffer & x, const Buffer & frequencies)
{
    return Filter::filter(x);
}

Buffer
FilterBandRejectIIR::
filter(
    const Buffer & x,
    const float64 & f_low,
    const float64 & f_high)
{
    if(!is_realtime_) reset();

    return low_->filter(x, f_low)
         + high_->filter(x, f_high);
}

Buffer
FilterBandRejectIIR::
filter(
    const Buffer & x,
    const Buffer & frequencies_Hz_low,
    const Buffer & frequencies_Hz_high)
{
    if(!is_realtime_) reset();

    uint32 n_high_freqs = frequencies_Hz_high.getLength();
    uint32 n_low_freqs  = frequencies_Hz_low.getLength();

    Buffer y;

    uint32 x_samples = x.getLength();

    for(uint32 i = 0; i < x_samples; ++i)
    {
        y << FilterBandRejectIIR::filter(
            x[i],
            frequencies_Hz_low [i % n_low_freqs],
            frequencies_Hz_high[i % n_high_freqs]);
    }

    return y;
}

float64
FilterBandRejectIIR::
filter(const float64 & x)
{
    return low_->filter(x) + high_->filter(x);
};

float64
FilterBandRejectIIR::
filter(const float64 & x, const float64 & frequency)
{
    return FilterBandRejectIIR::filter(x);
}

float64
FilterBandRejectIIR::
filter(
    const float64 & x,
    const float64 & frequencies_Hz_low,
    const float64 & frequencies_Hz_high)
{
    return low_->filter(x, frequencies_Hz_low)
        + high_->filter(x, frequencies_Hz_high);
}

void
FilterBandRejectIIR::
plot(boolean show_fc, boolean show_phase)
{
    char title[128];
    sprintf(title,
        "Band Reject IIR Frequency Response\n"
        "order = %d, fl = %0.1f Hz, fl = %0.1f Hz, sr = %0.1f Hz",
        low_->getKernelSize() - 1,
        low_->getFrequency(),
        high_->getFrequency(),
        sample_rate_);

    Filter::plot(show_phase);

    Plotter pylab;

    uint32 n_rows = 1;

    if(show_phase)
    {
        n_rows = 2;
    }

    if(show_fc)
    {
        pylab.subplot(n_rows, 1, 1);

        pylab.axvline(low_->getFrequency(),"color='red'");
        pylab.axvline(high_->getFrequency(),"color='red'");

        pylab.title(title);
    }
}

void
FilterBandRejectIIR::
reset()
{
    low_->reset();
    high_->reset();
}