summaryrefslogtreecommitdiff
path: root/gr-qtgui/lib/waterfall_sink_c_impl.h
blob: 3b04a5032267237c6114eb2b32e06e4f9c2eae7a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
/* -*- c++ -*- */
/*
 * Copyright 2012,2015,2019 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 */

#ifndef INCLUDED_QTGUI_WATERFALL_SINK_C_IMPL_H
#define INCLUDED_QTGUI_WATERFALL_SINK_C_IMPL_H

#include <gnuradio/qtgui/waterfall_sink_c.h>

#include <gnuradio/fft/fft.h>
#include <gnuradio/fft/fft_shift.h>
#include <gnuradio/filter/firdes.h>
#include <gnuradio/high_res_timer.h>
#include <gnuradio/qtgui/waterfalldisplayform.h>

namespace gr {
namespace qtgui {

class QTGUI_API waterfall_sink_c_impl : public waterfall_sink_c
{
private:
    void forecast(int noutput_items, gr_vector_int& ninput_items_required);

    void initialize();

    int d_fftsize;
    fft::fft_shift<float> d_fft_shift;
    float d_fftavg;
    filter::firdes::win_type d_wintype;
    std::vector<float> d_window;
    double d_center_freq;
    double d_bandwidth;
    const std::string d_name;
    int d_nconnections;
    int d_nrows;

    const pmt::pmt_t d_port;
    const pmt::pmt_t d_port_bw;

    bool d_shift;
    fft::fft_complex* d_fft;

    int d_index;
    std::vector<gr_complex*> d_residbufs;
    std::vector<double*> d_magbufs;
    double* d_pdu_magbuf;
    float* d_fbuf;

    int d_argc;
    char* d_argv;
    QWidget* d_parent;
    WaterfallDisplayForm* d_main_gui;

    gr::high_res_timer_type d_update_time;
    gr::high_res_timer_type d_last_time;

    void windowreset();
    void buildwindow();
    void fftresize();
    void check_clicked();
    void fft(float* data_out, const gr_complex* data_in, int size);

    // Handles message input port for setting new bandwidth
    // The message is a PMT pair (intern('bw'), double(bw))
    void handle_set_bw(pmt::pmt_t msg);

    // Handles message input port for setting new center frequency.
    // The message is a PMT pair (intern('freq'), double(frequency)).
    void handle_set_freq(pmt::pmt_t msg);

    // Handles message input port for displaying PDU samples.
    void handle_pdus(pmt::pmt_t msg);

public:
    waterfall_sink_c_impl(int size,
                          int wintype,
                          double fc,
                          double bw,
                          const std::string& name,
                          int nconnections,
                          QWidget* parent = NULL);
    ~waterfall_sink_c_impl();

    bool check_topology(int ninputs, int noutputs);

    void exec_();
    QWidget* qwidget();

#ifdef ENABLE_PYTHON
    PyObject* pyqwidget();
#else
    void* pyqwidget();
#endif

    void clear_data();

    void set_fft_size(const int fftsize);
    int fft_size() const;
    void set_fft_average(const float fftavg);
    float fft_average() const;
    void set_fft_window(const gr::filter::firdes::win_type win);
    gr::filter::firdes::win_type fft_window();

    void set_frequency_range(const double centerfreq, const double bandwidth);
    void set_intensity_range(const double min, const double max);

    void set_update_time(double t);
    void set_time_per_fft(double t);
    void set_title(const std::string& title);
    void set_time_title(const std::string& title);
    void set_line_label(unsigned int which, const std::string& label);
    void set_line_alpha(unsigned int which, double alpha);
    void set_color_map(unsigned int which, const int color);

    std::string title();
    std::string line_label(unsigned int which);
    double line_alpha(unsigned int which);
    int color_map(unsigned int which);

    void set_size(int width, int height);

    void auto_scale();
    double min_intensity(unsigned int which);
    double max_intensity(unsigned int which);

    void enable_menu(bool en);
    void enable_grid(bool en);
    void disable_legend();
    void enable_axis_labels(bool en);

    int work(int noutput_items,
             gr_vector_const_void_star& input_items,
             gr_vector_void_star& output_items);
};

} /* namespace qtgui */
} /* namespace gr */

#endif /* INCLUDED_QTGUI_WATERFALL_SINK_C_IMPL_H */