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
|
/* -*- c++ -*- */
/*
* Copyright 2004,2010,2012 Free Software Foundation, Inc.
*
* This file is part of GNU Radio
*
* GNU Radio is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3, or (at your option)
* any later version.
*
* GNU Radio is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GNU Radio; see the file COPYING. If not, write to
* the Free Software Foundation, Inc., 51 Franklin Street,
* Boston, MA 02110-1301, USA.
*/
// @WARNING@
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "@NAME@.h"
#include <gr_io_signature.h>
#include <iostream>
namespace gr {
namespace trellis {
static const float INF = 1.0e9;
@BASE_NAME@::sptr
@BASE_NAME@::make(const fsm &FSM, int K,
int S0, int SK, int D,
const std::vector<@I_TYPE@> &TABLE,
digital::trellis_metric_type_t TYPE)
{
return gnuradio::get_initial_sptr
(new @NAME@(FSM, K, S0, SK, D, TABLE,TYPE));
}
@IMPL_NAME@::@IMPL_NAME@(const fsm &FSM, int K,
int S0, int SK, int D,
const std::vector<@I_TYPE@> &TABLE,
digital::trellis_metric_type_t TYPE)
: gr_block("@BASE_NAME@",
gr_make_io_signature(1, -1, sizeof(@I_TYPE@)),
gr_make_io_signature(1, -1, sizeof(@O_TYPE@))),
d_FSM(FSM), d_K(K), d_S0(S0), d_SK(SK), d_D(D),
d_TABLE(TABLE), d_TYPE(TYPE)//,
//d_trace(FSM.S()*K)
{
set_relative_rate(1.0 / ((double)d_D));
set_output_multiple(d_K);
}
@IMPL_NAME@::~@IMPL_NAME@()
{
}
void
@IMPL_NAME@::set_TABLE(const std::vector<@I_TYPE@> &table)
{
d_TABLE = table;
}
void
@IMPL_NAME@::forecast(int noutput_items,
gr_vector_int &ninput_items_required)
{
int input_required = d_D * noutput_items;
unsigned ninputs = ninput_items_required.size();
for(unsigned int i = 0; i < ninputs; i++) {
ninput_items_required[i] = input_required;
}
}
int
@IMPL_NAME@::general_work(int noutput_items,
gr_vector_int &ninput_items,
gr_vector_const_void_star &input_items,
gr_vector_void_star &output_items)
{
int nstreams = input_items.size();
int nblocks = noutput_items / d_K;
for(int m=0;m<nstreams;m++) {
const @I_TYPE@ *in = (const @I_TYPE@*)input_items[m];
@O_TYPE@ *out = (@O_TYPE@*)output_items[m];
for(int n=0;n<nblocks;n++) {
viterbi_algorithm_combined(d_FSM.I(), d_FSM.S(), d_FSM.O(),
d_FSM.NS(), d_FSM.OS(), d_FSM.PS(),
d_FSM.PI(), d_K,d_S0, d_SK, d_D,
d_TABLE, d_TYPE,
&(in[n*d_K*d_D]), &(out[n*d_K]));
}
}
consume_each(d_D * noutput_items);
return noutput_items;
}
} /* namespace trellis */
} /* namespace gr */
|
