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
|
/* -*- c++ -*- */
/*
* Copyright 2014 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 "puncture_bb_impl.h"
#include <gnuradio/io_signature.h>
#include <pmt/pmt.h>
#include <volk/volk.h>
#include <cstdio>
#include <string>
namespace gr {
namespace fec {
puncture_bb::sptr puncture_bb::make(int puncsize, int puncpat, int delay)
{
return gnuradio::make_block_sptr<puncture_bb_impl>(puncsize, puncpat, delay);
}
puncture_bb_impl::puncture_bb_impl(int puncsize, int puncpat, int delay)
: block("puncture_bb",
io_signature::make(1, 1, sizeof(char)),
io_signature::make(1, 1, sizeof(char))),
d_puncsize(puncsize),
d_delay(delay)
{
// Create a mask of all 1's of puncsize length
int mask = 0;
for (int i = 0; i < d_puncsize; i++)
mask |= 1 << i;
// Rotate the pattern for the delay value; then mask it if there
// are any excess 1's in the pattern.
for (int i = 0; i < d_delay; ++i) {
puncpat = ((puncpat & 1) << (d_puncsize - 1)) + (puncpat >> 1);
}
d_puncpat = puncpat & mask;
// Calculate the number of holes in the pattern. The mask is all
// 1's given puncsize and puncpat is a pattern with >= puncsize
// 0's (masked to ensure this). The difference between the
// number of 1's in the mask and the puncpat is the number of
// holes.
uint32_t count_mask = 0, count_pat = 0;
volk_32u_popcnt(&count_mask, static_cast<uint32_t>(mask));
volk_32u_popcnt(&count_pat, static_cast<uint32_t>(d_puncpat));
d_puncholes = count_mask - count_pat;
set_fixed_rate(true);
set_relative_rate((uint64_t)(d_puncsize - d_puncholes), (uint64_t)d_puncsize);
set_output_multiple(d_puncsize - d_puncholes);
// set_msg_handler(<portname>, [this](pmt::pmt_t msg) { this->catch_msg(msg); });
}
puncture_bb_impl::~puncture_bb_impl() {}
int puncture_bb_impl::fixed_rate_ninput_to_noutput(int ninput)
{
return std::lround(((d_puncsize - d_puncholes) / (double)(d_puncsize)) * ninput);
}
int puncture_bb_impl::fixed_rate_noutput_to_ninput(int noutput)
{
return std::lround((d_puncsize / (double)(d_puncsize - d_puncholes)) * noutput);
}
void puncture_bb_impl::forecast(int noutput_items, gr_vector_int& ninput_items_required)
{
ninput_items_required[0] =
std::lround((d_puncsize / (double)(d_puncsize - d_puncholes)) * noutput_items);
}
/*
void
puncture_bb_impl::catch_msg(pmt::pmt_t msg)
{
long mlong = pmt::pmt_to_long(msg);
for(int i = 0; i < mlong; ++i) {
d_puncholes = (d_puncholes >> 1) | ((d_puncholes & 1) << (d_puncsize - 1));
}
}
*/
int puncture_bb_impl::general_work(int noutput_items,
gr_vector_int& ninput_items,
gr_vector_const_void_star& input_items,
gr_vector_void_star& output_items)
{
const char* in = (const char*)input_items[0];
char* out = (char*)output_items[0];
for (int i = 0, k = 0; i < noutput_items / output_multiple(); ++i) {
for (int j = 0; j < d_puncsize; ++j) {
if ((d_puncpat >> (d_puncsize - 1 - j)) & 1) {
out[k++] = in[i * d_puncsize + j];
}
}
}
/*
GR_LOG_DEBUG(d_debug_logger, ">>>>>> start");
for(int i = 0, k=0; i < noutput_items; ++i) {
if((d_puncpat >> (d_puncsize - 1 - (i % d_puncsize))) & 1) {
GR_LOG_DEBUG(d_debug_logger, boost::format("%1%...%2%") \
% out[k++] % in[i]);
}
else {
GR_LOG_DEBUG(d_debug_logger, boost::format("snit %1%") % in[i]);
}
}
GR_LOG_DEBUG(d_debug_logger, boost::format("comp: %1%, %2%\n") \
% noutput_items % ninput_items[0]);
GR_LOG_DEBUG(d_debug_logger, boost::format("consuming %1%") \
% ((int)(((1.0/relative_rate()) * noutput_items) + .5)));
*/
consume_each(std::lround((1.0 / relative_rate()) * noutput_items));
return noutput_items;
}
} /* namespace fec */
} /* namespace gr */
|