summaryrefslogtreecommitdiff
path: root/gr-error-correcting-codes/src/lib/libecc/encoder.cc
blob: 5bc97e7fc734796775b5914252c2582a0117b360 (plain)
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
/* -*- c++ -*- */
/*
 * Copyright 2006 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 2, 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., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <encoder.h>
#include <iostream>

#define DO_PRINT_DEBUG 0

/*
 * encode a certain number of output bits
 *
 * the 'in_buf' and 'out_buf' must have enough memory to handle the
 *     number of input and output bits; no error checking is done!
 *
 * n_bits_to_output: the number of bits per output stream to encode
 *
 * returns the actual number of bits used per input stream
 */

size_t
encoder::encode
(const char** in_buf,
 char** out_buf,
 size_t n_bits_to_output)
{
  if (DO_PRINT_DEBUG) {
    std::cout << "encode{out}(): Starting:";
  }

  // set the class-internal number of input bits and
  // output bits left to encode

  size_t saved_n_input_bits;
  saved_n_input_bits = d_n_input_bits_left =
    compute_n_input_bits (n_bits_to_output);
  d_n_output_bits_left = n_bits_to_output;

  if (DO_PRINT_DEBUG) {
    std::cout <<
      "# output bits provided = " << d_n_output_bits_left << "\n"
      "# input bits computed  = " << d_n_input_bits_left << "\n";
  }

  // call the private encode function

  encode_private (in_buf, out_buf);

  if (DO_PRINT_DEBUG) {
    std::cout <<
      "n_input_bits_used  = " <<
      (saved_n_input_bits - d_n_input_bits_left) << "\n"
      "n_output_bits_used = " <<
      (n_bits_to_output - d_n_output_bits_left) << '\n';
  }

  // return the actual number of input bits used

  return (saved_n_input_bits - d_n_input_bits_left);
}

/*
 * encode a certain number of input bits
 *
 * the 'in_buf' and 'out_buf' must have enough memory to handle the
 *     number of input and output bits; no error checking is done!
 *
 * n_bits_to_input: the number of bits per input stream to encode
 *
 * returns the actual number of bits written per output stream
 */

size_t
encoder::encode
(const char** in_buf,	
 size_t n_bits_to_input,
 char** out_buf)
{
  // set the class-internal number of input and
  // output bits left to encode

  size_t saved_n_output_bits;
  saved_n_output_bits = d_n_output_bits_left =
    compute_n_output_bits (n_bits_to_input);
  d_n_input_bits_left = n_bits_to_input;

  // call the private encode function

  encode_private (in_buf, out_buf);

  if (DO_PRINT_DEBUG) {
    std::cout << "n_input_bits_used = " <<
      (n_bits_to_input - d_n_input_bits_left) << '\n';
    std::cout << "n_output_bits_used = " <<
      (saved_n_output_bits - d_n_output_bits_left) << '\n';
  }

  // return the actual number of output bits written

  return (saved_n_output_bits - d_n_output_bits_left);
}