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
|
/* -*- c++ -*- */
/*
* Copyright 2012-2013 Free Software Foundation, Inc.
*
* This file is part of GNU Radio
*
* SPDX-License-Identifier: GPL-3.0-or-later
*
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "message_strobe_random_impl.h"
#include <gnuradio/io_signature.h>
namespace gr {
namespace blocks {
message_strobe_random::sptr
message_strobe_random::make(pmt::pmt_t msg,
message_strobe_random_distribution_t dist,
float mean_ms,
float std_ms)
{
return gnuradio::make_block_sptr<message_strobe_random_impl>(
msg, dist, mean_ms, std_ms);
}
message_strobe_random_impl::message_strobe_random_impl(
pmt::pmt_t msg,
message_strobe_random_distribution_t dist,
float mean_ms,
float std_ms)
: block("message_strobe_random",
io_signature::make(0, 0, 0),
io_signature::make(0, 0, 0)),
d_rng(),
d_mean_ms(mean_ms),
d_std_ms(std_ms),
d_dist(dist),
pd(d_mean_ms),
nd(d_mean_ms, d_std_ms),
ud(d_mean_ms - d_std_ms, d_mean_ms + d_std_ms),
d_finished(false),
d_msg(msg),
d_port(pmt::mp("strobe"))
{
// set up ports
message_port_register_out(d_port);
d_thread = gr::thread::thread([this] { run(); });
message_port_register_in(pmt::mp("set_msg"));
set_msg_handler(pmt::mp("set_msg"), [this](pmt::pmt_t msg) { this->set_msg(msg); });
}
void message_strobe_random_impl::set_mean(float mean_ms)
{
d_mean_ms = mean_ms;
pd = std::poisson_distribution<>(d_mean_ms);
nd = std::normal_distribution<>(d_mean_ms, d_std_ms);
ud = std::uniform_real_distribution<>(d_mean_ms - d_std_ms, d_mean_ms + d_std_ms);
}
float message_strobe_random_impl::mean() const { return d_mean_ms; }
void message_strobe_random_impl::set_std(float std_ms)
{
d_std_ms = std_ms;
nd = std::normal_distribution<>(d_mean_ms, d_std_ms);
ud = std::uniform_real_distribution<>(d_mean_ms - d_std_ms, d_mean_ms + d_std_ms);
}
long message_strobe_random_impl::next_delay()
{
switch (d_dist) {
case STROBE_POISSON:
// return d_variate_poisson->operator()();
return static_cast<long>(pd(d_rng));
case STROBE_GAUSSIAN:
return static_cast<long>(nd(d_rng));
case STROBE_UNIFORM:
return static_cast<long>(ud(d_rng));
default:
throw std::runtime_error(
"message_strobe_random_impl::d_distribution is very unhappy with you");
}
}
message_strobe_random_impl::~message_strobe_random_impl()
{
d_finished = true;
d_thread.interrupt();
d_thread.join();
}
void message_strobe_random_impl::run()
{
while (!d_finished) {
boost::this_thread::sleep(
boost::posix_time::milliseconds(std::max(0L, next_delay())));
if (d_finished) {
return;
}
message_port_pub(d_port, d_msg);
}
}
} /* namespace blocks */
} /* namespace gr */
|
