1) Minor correction in Viterbi. 2) Checked in interleaver structure
[debian/gnuradio] / gr-trellis / src / lib / trellis_permutation.cc
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 #ifdef HAVE_CONFIG_H
24 #include "config.h"
25 #endif
26
27 #include <trellis_permutation.h>
28 #include <gr_io_signature.h>
29 #include <iostream>
30
31 trellis_permutation_sptr 
32 trellis_make_permutation (const int K, const std::vector<int> &TABLE, const size_t NBYTES)
33 {
34   return trellis_permutation_sptr (new trellis_permutation (K,TABLE,NBYTES));
35 }
36
37 trellis_permutation::trellis_permutation (const int K, const std::vector<int> &TABLE, const size_t NBYTES)
38   : gr_sync_block ("permutation",
39                    gr_make_io_signature (1, -1, NBYTES),
40                    gr_make_io_signature (1, -1, NBYTES)),
41     d_K (K),
42     d_TABLE (TABLE),
43     d_NBYTES (NBYTES)
44 {
45     set_output_multiple (d_K);
46     //std::cout << d_K << "\n";
47 }
48
49
50
51 int 
52 trellis_permutation::work (int noutput_items,
53                         gr_vector_const_void_star &input_items,
54                         gr_vector_void_star &output_items)
55 {
56   int nstreams = input_items.size();
57   assert (input_items.size() == output_items.size());
58   assert (noutput_items % d_K ==0);
59   //std::cout << noutput_items << "\n";
60
61   for (int m=0;m<nstreams;m++) {
62     const char *in = (const char *) input_items[m];
63     char *out = (char *) output_items[m];
64
65     // per stream processing
66     for (unsigned int i = 0; i < noutput_items; i++){
67       //std::cout << i << " " << i*d_NBYTES << " " << (d_K*(i/d_K)+d_TABLE[i%d_K])*d_NBYTES  << "\n";
68       memcpy(&(out[i*d_NBYTES]), &(in[(d_K*(i/d_K)+d_TABLE[i%d_K])*d_NBYTES]), d_NBYTES);
69     }
70     // end per stream processing
71   }
72   return noutput_items;
73 }