signal_separator_c_impl.h

Go to the documentation of this file.

/* -*- c++ -*- */
/*
 * Copyright 2019 Free Software Foundation, Inc.
 *
 * This 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.
 *
 * This software 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 this software; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#ifndef INCLUDED_INSPECTOR_SIGNAL_SEPARATOR_C_IMPL_H
#define INCLUDED_INSPECTOR_SIGNAL_SEPARATOR_C_IMPL_H

#include <gnuradio/blocks/rotator.h>
#include <gnuradio/filter/fir_filter.h>
#include <gnuradio/filter/firdes.h>
#include <gnuradio/messages/msg_queue.h>
#include <gnuradio/thread/thread.h>
#include <inspector/signal_separator_c.h>

namespace gr {
namespace inspector {

class signal_separator_c_impl : public signal_separator_c
{
private:
    bool d_use_file;
    int d_buffer_len;
    unsigned int d_buffer_stage;
    float d_trans_width, d_oversampling;
    double d_samp_rate;
    gr_complex* d_temp_buffer;

    filter::firdes::win_type d_window;

    std::vector<filter::kernel::fir_filter_ccf*> d_filterbank;
    std::vector<std::vector<float>> d_rf_map;

    std::vector<float> build_taps(double cutoff);
    std::vector<int> d_decimations;
    std::vector<int> d_ntaps;
    std::vector<blocks::rotator> d_rotators;

    std::vector<gr_complex*> d_history_buffer;
    std::vector<std::vector<gr_complex>> d_result_vector;
    gr::thread::mutex d_mutex;

    std::map<float, std::vector<float>> d_precalc;
    std::vector<float> d_taps;


public:
    signal_separator_c_impl(double samp_rate,
                            int window,
                            float trans_width,
                            float oversampling,
                            bool taps_file,
                            std::map<float, std::vector<float>>& file_path);

    ~signal_separator_c_impl();

    // free allocated buffer arrays and FIR filters
    void free_allocation();
    // build new filter for signal
    void build_filter(unsigned int signal);
    // build new filters for all signals
    void rebuild_all_filters();
    // apply filter no i to history buffer i
    void apply_filter(int i);
    // trigger rebuild of filters and unpack message
    void handle_msg(pmt::pmt_t msg);
    // create pmt from filtered samples
    pmt::pmt_t pack_message();
    // create RF map from received message
    void unpack_message(pmt::pmt_t msg);

    // Where all the action really happens
    void forecast(int noutput_items, gr_vector_int& ninput_items_required);

    int general_work(int noutput_items,
                     gr_vector_int& ninput_items,
                     gr_vector_const_void_star& input_items,
                     gr_vector_void_star& output_items);

    //<editor-fold desc="Getter and Setter">

    double samp_rate() const { return d_samp_rate; }

    void set_samp_rate(double d_samp_rate)
    {
        signal_separator_c_impl::d_samp_rate = d_samp_rate;
        rebuild_all_filters();
    }

    filter::firdes::win_type window() const { return d_window; }

    void set_window(int d_window)
    {
        signal_separator_c_impl::d_window =
            static_cast<filter::firdes::win_type>(d_window);
        rebuild_all_filters();
    }

    float trans_width() const { return d_trans_width; }

    void set_trans_width(float d_trans_width)
    {
        signal_separator_c_impl::d_trans_width = d_trans_width;
    }

    int oversampling() const { return d_oversampling; }

    void set_oversampling(float d_oversampling)
    {
        signal_separator_c_impl::d_oversampling = d_oversampling;
        rebuild_all_filters();
    }

    //</editor-fold>
};

} // namespace inspector
} // namespace gr

#endif /* INCLUDED_INSPECTOR_SIGNAL_SEPARATOR_C_IMPL_H */