/* -*- c++ -*- */
/*
 * Copyright 2009-2012 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * GNU Radio 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, or (at your option)
 * any later version.
 *
 * GNU Radio 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 GNU Radio; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "pfb_arb_resampler_fff_impl.h"
#include <gnuradio/io_signature.h>
#include <cstdio>

namespace gr {
namespace filter {

pfb_arb_resampler_fff::sptr pfb_arb_resampler_fff::make(float rate,
                                                        const std::vector<float>& taps,
                                                        unsigned int filter_size)
{
    return gnuradio::get_initial_sptr(
        new pfb_arb_resampler_fff_impl(rate, taps, filter_size));
}


pfb_arb_resampler_fff_impl::pfb_arb_resampler_fff_impl(float rate,
                                                       const std::vector<float>& taps,
                                                       unsigned int filter_size)
    : block("pfb_arb_resampler_fff",
            io_signature::make(1, 1, sizeof(float)),
            io_signature::make(1, 1, sizeof(float)))
{
    d_updated = false;

    d_resamp = new kernel::pfb_arb_resampler_fff(rate, taps, filter_size);
    set_history(d_resamp->taps_per_filter());
    set_relative_rate(rate);
}

pfb_arb_resampler_fff_impl::~pfb_arb_resampler_fff_impl() { delete d_resamp; }

void pfb_arb_resampler_fff_impl::forecast(int noutput_items,
                                          gr_vector_int& ninput_items_required)
{
    unsigned ninputs = ninput_items_required.size();
    if (noutput_items / relative_rate() < 1) {
        for (unsigned i = 0; i < ninputs; i++)
            ninput_items_required[i] = relative_rate() + history() - 1;
    } else {
        for (unsigned i = 0; i < ninputs; i++)
            ninput_items_required[i] = noutput_items / relative_rate() + history() - 1;
    }
}

void pfb_arb_resampler_fff_impl::set_taps(const std::vector<float>& taps)
{
    gr::thread::scoped_lock guard(d_mutex);

    d_resamp->set_taps(taps);
    set_history(d_resamp->taps_per_filter());
    d_updated = true;
}

std::vector<std::vector<float>> pfb_arb_resampler_fff_impl::taps() const
{
    return d_resamp->taps();
}

void pfb_arb_resampler_fff_impl::print_taps() { d_resamp->print_taps(); }

void pfb_arb_resampler_fff_impl::set_rate(float rate)
{
    gr::thread::scoped_lock guard(d_mutex);

    d_resamp->set_rate(rate);
    set_relative_rate(rate);
}

void pfb_arb_resampler_fff_impl::set_phase(float ph)
{
    gr::thread::scoped_lock guard(d_mutex);
    d_resamp->set_phase(ph);
}

float pfb_arb_resampler_fff_impl::phase() const { return d_resamp->phase(); }

unsigned int pfb_arb_resampler_fff_impl::interpolation_rate() const
{
    return d_resamp->interpolation_rate();
}

unsigned int pfb_arb_resampler_fff_impl::decimation_rate() const
{
    return d_resamp->decimation_rate();
}

float pfb_arb_resampler_fff_impl::fractional_rate() const
{
    return d_resamp->fractional_rate();
}

unsigned int pfb_arb_resampler_fff_impl::taps_per_filter() const
{
    return d_resamp->taps_per_filter();
}

int pfb_arb_resampler_fff_impl::group_delay() const { return d_resamp->group_delay(); }

float pfb_arb_resampler_fff_impl::phase_offset(float freq, float fs)
{
    return d_resamp->phase_offset(freq, fs);
}

int pfb_arb_resampler_fff_impl::general_work(int noutput_items,
                                             gr_vector_int& ninput_items,
                                             gr_vector_const_void_star& input_items,
                                             gr_vector_void_star& output_items)
{
    gr::thread::scoped_lock guard(d_mutex);

    float* in = (float*)input_items[0];
    float* out = (float*)output_items[0];

    if (d_updated) {
        d_updated = false;
        return 0; // history requirements may have changed.
    }

    int nitems_read;
    int nitems = floorf((float)noutput_items / relative_rate());
    int processed = d_resamp->filter(out, in, nitems, nitems_read);

    consume_each(nitems_read);
    return processed;
}

} /* namespace filter */
} /* namespace gr */
