summaryrefslogtreecommitdiff
path: root/gr-channels/lib/sro_model_impl.h
blob: 420ec0ea719804d2caa54a255a043d3d99827942 (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
/* -*- c++ -*- */
/*
 * Copyright 2004,2007,2012 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 *
 */

#ifndef INCLUDED_CHANNELS_SRO_MODEL_CC_IMPL_H
#define INCLUDED_CHANNELS_SRO_MODEL_CC_IMPL_H

#include <gnuradio/analog/fastnoise_source.h>
#include <gnuradio/channels/sro_model.h>
#include <gnuradio/filter/mmse_fir_interpolator_cc.h>
#include <gnuradio/filter/mmse_interpolator_cc.h>

namespace gr {
namespace channels {

class CHANNELS_API sro_model_impl : public sro_model
{
private:
    float d_mu;
    float d_mu_inc;
    float d_sro;
    float d_samp_rate;
    float d_max_dev_hz;
    float d_std_dev_hz;
    const gr::filter::mmse_fir_interpolator_cc d_interp;
    gr::analog::fastnoise_source_f::sptr d_noise;
    double d_noise_seed;

public:
    sro_model_impl(double sample_rate_hz,
                   double std_dev_hz,
                   double max_dev_hz,
                   double noise_seed = 0);
    ~sro_model_impl();

    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);

    float mu() const;
    float interp_ratio() const;
    void set_mu(float mu);
    void set_interp_ratio(float interp_ratio);
    void setup_rpc();

    void set_std_dev(double _dev)
    {
        d_std_dev_hz = _dev;
        d_noise =
            gr::analog::fastnoise_source_f::make(analog::GR_GAUSSIAN, _dev, d_noise_seed);
    }
    void set_max_dev(double _dev) { d_max_dev_hz = _dev; }
    void set_samp_rate(double _rate) { d_samp_rate = _rate; }

    double std_dev() const { return d_std_dev_hz; }
    double max_dev() const { return d_max_dev_hz; }
    double samp_rate() const { return d_samp_rate; }
};

} // namespace channels
} /* namespace gr */

#endif