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
|
/* -*- c++ -*- */
/*
* Copyright 2016 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 "protocol_formatter_bb_impl.h"
#include <gnuradio/io_signature.h>
#include <volk/volk.h>
#include <cstdio>
namespace gr {
namespace digital {
protocol_formatter_bb::sptr
protocol_formatter_bb::make(const header_format_base::sptr& format,
const std::string& len_tag_key)
{
return gnuradio::make_block_sptr<protocol_formatter_bb_impl>(format, len_tag_key);
}
protocol_formatter_bb_impl::protocol_formatter_bb_impl(
const header_format_base::sptr& format, const std::string& len_tag_key)
: tagged_stream_block("protocol_formatter_bb",
io_signature::make(1, 1, sizeof(char)),
io_signature::make(1, 1, sizeof(char)),
len_tag_key),
d_format(format)
{
set_output_multiple(d_format->header_nbytes());
// This is the worst case rate, because we don't know the true value, of course
set_relative_rate((uint64_t)d_format->header_nbytes(), 1);
set_tag_propagation_policy(TPP_DONT);
}
protocol_formatter_bb_impl::~protocol_formatter_bb_impl() {}
void protocol_formatter_bb_impl::set_header_format(header_format_base::sptr& format)
{
gr::thread::scoped_lock guard(d_setlock);
d_format = format;
}
int protocol_formatter_bb_impl::work(int noutput_items,
gr_vector_int& ninput_items,
gr_vector_const_void_star& input_items,
gr_vector_void_star& output_items)
{
gr::thread::scoped_lock guard(d_setlock);
unsigned char* out = (unsigned char*)output_items[0];
const unsigned char* in = (const unsigned char*)input_items[0];
// Not really sure what to do with these tags; extract them and
// plug them into the info dictionary?
// std::vector<tag_t> tags;
// get_tags_in_window(tags, 0, 0, ninput_items[0]);
pmt::pmt_t pmt_out;
pmt::pmt_t info = pmt::PMT_NIL;
if (!d_format->format(ninput_items[0], in, pmt_out, info)) {
GR_LOG_FATAL(d_logger,
boost::format("header format returned false "
"(this shouldn't happen). Offending "
"header started at %1%") %
nitems_read(0));
throw std::runtime_error("header format returned false.");
}
size_t len;
const uint8_t* data = pmt::u8vector_elements(pmt_out, len);
if (len != d_format->header_nbytes()) {
throw std::runtime_error("Header format got wrong size header");
}
memcpy(out, data, len);
if (pmt::is_dict(info)) {
std::vector<tag_t> mtags;
pmt::pmt_t mkeys = pmt::dict_keys(info);
pmt::pmt_t mvals = pmt::dict_values(info);
for (size_t i = 0; i < pmt::length(mkeys); i++) {
tag_t tag;
tag.offset = nitems_written(0);
tag.key = pmt::nth(i, mkeys);
tag.value = pmt::nth(i, mvals);
tag.srcid = alias_pmt();
add_item_tag(0, tag);
}
}
return len;
}
} /* namespace digital */
} /* namespace gr */
|
