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
|
/* -*- c++ -*- */
/*
* Copyright 2004,2010,2012,2018 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 "chunks_to_symbols_impl.h"
#include <gnuradio/io_signature.h>
#include <gnuradio/tag_checker.h>
#include <assert.h>
namespace gr {
namespace digital {
void set_vector_from_pmt(std::vector<gr_complex>& symbol_table,
pmt::pmt_t& symbol_table_pmt)
{
size_t length;
const gr_complex* elements = pmt::c32vector_elements(symbol_table_pmt, length);
symbol_table.assign(elements, elements + length);
}
void set_vector_from_pmt(std::vector<float>& symbol_table, pmt::pmt_t& symbol_table_pmt)
{
size_t length;
const float* elements = pmt::f32vector_elements(symbol_table_pmt, length);
symbol_table.assign(elements, elements + length);
}
template <class IN_T, class OUT_T>
typename chunks_to_symbols<IN_T, OUT_T>::sptr
chunks_to_symbols<IN_T, OUT_T>::make(const std::vector<OUT_T>& symbol_table, const int D)
{
return gnuradio::make_block_sptr<chunks_to_symbols_impl<IN_T, OUT_T>>(symbol_table,
D);
}
template <class IN_T, class OUT_T>
chunks_to_symbols_impl<IN_T, OUT_T>::chunks_to_symbols_impl(
const std::vector<OUT_T>& symbol_table, const int D)
: sync_interpolator("chunks_to_symbols",
io_signature::make(1, -1, sizeof(IN_T)),
io_signature::make(1, -1, sizeof(OUT_T)),
D),
d_D(D),
d_symbol_table(symbol_table)
{
this->message_port_register_in(pmt::mp("set_symbol_table"));
this->set_msg_handler(pmt::mp("set_symbol_table"),
[this](pmt::pmt_t msg) { this->handle_set_symbol_table(msg); });
}
template <class IN_T, class OUT_T>
chunks_to_symbols_impl<IN_T, OUT_T>::~chunks_to_symbols_impl()
{
}
template <class IN_T, class OUT_T>
void chunks_to_symbols_impl<IN_T, OUT_T>::handle_set_symbol_table(
pmt::pmt_t symbol_table_pmt)
{
set_vector_from_pmt(d_symbol_table, symbol_table_pmt);
}
template <class IN_T, class OUT_T>
void chunks_to_symbols_impl<IN_T, OUT_T>::set_symbol_table(
const std::vector<OUT_T>& symbol_table)
{
gr::thread::scoped_lock lock(this->d_setlock);
d_symbol_table = symbol_table;
}
template <class IN_T, class OUT_T>
int chunks_to_symbols_impl<IN_T, OUT_T>::work(int noutput_items,
gr_vector_const_void_star& input_items,
gr_vector_void_star& output_items)
{
gr::thread::scoped_lock lock(this->d_setlock);
assert(noutput_items % d_D == 0);
assert(input_items.size() == output_items.size());
int nstreams = input_items.size();
for (int m = 0; m < nstreams; m++) {
const IN_T* in = (IN_T*)input_items[m];
OUT_T* out = (OUT_T*)output_items[m];
std::vector<tag_t> tags;
this->get_tags_in_range(
tags, m, this->nitems_read(m), this->nitems_read(m) + noutput_items / d_D);
tag_checker tchecker(tags);
// per stream processing
for (int i = 0; i < noutput_items / d_D; i++) {
std::vector<tag_t> tags_now;
tchecker.get_tags(tags_now, i + this->nitems_read(m));
for (unsigned int j = 0; j < tags_now.size(); j++) {
tag_t tag = tags_now[j];
this->dispatch_msg(tag.key, tag.value);
}
assert(((unsigned int)in[i] * d_D + d_D) <= d_symbol_table.size());
memcpy(out, &d_symbol_table[(unsigned int)in[i] * d_D], d_D * sizeof(OUT_T));
out += d_D;
}
}
return noutput_items;
}
template class chunks_to_symbols<std::uint8_t, float>;
template class chunks_to_symbols<std::uint8_t, gr_complex>;
template class chunks_to_symbols<std::int16_t, float>;
template class chunks_to_symbols<std::int16_t, gr_complex>;
template class chunks_to_symbols<std::int32_t, float>;
template class chunks_to_symbols<std::int32_t, gr_complex>;
} /* namespace digital */
} /* namespace gr */
|
