Several enhancements to gr-trellis and gnuradio-examples/python/channel-coding:
[debian/gnuradio] / gr-trellis / src / lib / trellis_siso_f.h
1 /* -*- c++ -*- */
2 /*
3  * Copyright 2004 Free Software Foundation, Inc.
4  * 
5  * This file is part of GNU Radio
6  * 
7  * GNU Radio is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2, or (at your option)
10  * any later version.
11  * 
12  * GNU Radio is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  * 
17  * You should have received a copy of the GNU General Public License
18  * along with GNU Radio; see the file COPYING.  If not, write to
19  * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
20  * Boston, MA 02111-1307, USA.
21  */
22
23 #ifndef INCLUDED_TRELLIS_SISO_F_H
24 #define INCLUDED_TRELLIS_SISO_F_H
25
26 #include "fsm.h"
27 #include "trellis_siso_type.h"
28 #include <gr_block.h>
29
30 class trellis_siso_f;
31 typedef boost::shared_ptr<trellis_siso_f> trellis_siso_f_sptr;
32
33 trellis_siso_f_sptr trellis_make_siso_f (
34     const fsm &FSM, 
35     int K,
36     int S0,
37     int SK,
38     bool POSTI,
39     bool POSTO,
40     trellis_siso_type_t d_SISO_TYPE);
41
42
43
44 class trellis_siso_f : public gr_block
45 {
46   fsm d_FSM;
47   int d_K;
48   int d_S0;
49   int d_SK;
50   bool d_POSTI;
51   bool d_POSTO;
52   trellis_siso_type_t d_SISO_TYPE;
53   std::vector<float> d_alpha;
54   std::vector<float> d_beta;
55
56   friend trellis_siso_f_sptr trellis_make_siso_f (
57     const fsm &FSM,
58     int K,
59     int S0,
60     int SK,
61     bool POSTI,
62     bool POSTO,
63     trellis_siso_type_t d_SISO_TYPE);
64
65
66   trellis_siso_f (
67     const fsm &FSM,
68     int K,
69     int S0,
70     int SK,
71     bool POSTI,
72     bool POSTO,
73     trellis_siso_type_t d_SISO_TYPE);
74
75
76 public:
77   fsm FSM () const { return d_FSM; }
78   int K () const { return d_K; }
79   int S0 () const { return d_S0; }
80   int SK () const { return d_SK; }
81   bool POSTI () const { return d_POSTI; }
82   bool POSTO () const { return d_POSTO; }
83   trellis_siso_type_t SISO_TYPE () const { return d_SISO_TYPE; }
84   void forecast (int noutput_items,
85                  gr_vector_int &ninput_items_required);
86   int general_work (int noutput_items,
87                     gr_vector_int &ninput_items,
88                     gr_vector_const_void_star &input_items,
89                     gr_vector_void_star &output_items);
90 };
91
92 #endif