summaryrefslogtreecommitdiff
path: root/gr-fec/lib/ldpc_encoder_impl.cc
blob: 95879ea4b79503c6fdf47d552d25619dc0377bb9 (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 2015 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.
 */

#include "ldpc_encoder_impl.h"
#include <math.h>
#include <boost/assign/list_of.hpp>
#include <volk/volk.h>
#include <sstream>
#include <stdio.h>
#include <vector>
#include <algorithm>            // for std::reverse
#include <string.h>             // for memcpy
#include <gsl/gsl_matrix.h>

namespace gr {
  namespace fec {

    generic_encoder::sptr
    ldpc_encoder::make(std::string alist_file, unsigned int gap)
    {
      code::ldpc_H_matrix::sptr H_obj = code::ldpc_H_matrix::make(alist_file, gap);
      return make_H(H_obj);
    }

    generic_encoder::sptr
    ldpc_encoder::make_H(const code::ldpc_H_matrix::sptr H_obj)
    {
      return generic_encoder::sptr
        (new ldpc_encoder_impl(H_obj));
    }

    ldpc_encoder_impl::ldpc_encoder_impl(const code::ldpc_H_matrix::sptr H_obj)
      : generic_encoder("ldpc_encoder")
    {
      // LDPC parity check matrix to use for encoding
      d_H = H_obj;

      d_rate = static_cast<double>(d_H->n())/static_cast<double>(d_H->k());

      // Set frame size to k, the # of bits in the information word
      // All buffers and settings will be based on this value.
      set_frame_size(d_H->k());
    }

    ldpc_encoder_impl::~ldpc_encoder_impl()
    {
    }

    int
    ldpc_encoder_impl::get_output_size()
    {
      //return outputSize;
      return d_output_size;
    }

    int
    ldpc_encoder_impl::get_input_size()
    {
      //return inputSize;
      return d_frame_size;
    }

    bool
    ldpc_encoder_impl::set_frame_size(unsigned int frame_size)
    {
      bool ret = true;

      if(frame_size % d_H->k() != 0) {
        GR_LOG_ERROR(d_logger, boost::format("Frame size (%1% bits) must be a "
                                             "multiple of the information word "
                                             "size of the LDPC matrix (%2%).") \
                     % frame_size % (d_H->k()));
        throw std::runtime_error("ldpc_encoder: cannot use frame size.");
      }

      d_frame_size = frame_size;

      d_output_size = static_cast<int>(d_rate * d_frame_size);

      return ret;
    }

    double
    ldpc_encoder_impl::rate()
    {
      return d_rate;
    }

    void
    ldpc_encoder_impl::generic_work(void *inbuffer, void *outbuffer)
    {
      // Populate the information word
      const unsigned char *in = (const unsigned char *)inbuffer;
      unsigned char *out = (unsigned char*)outbuffer;

      int j = 0;
      for(int i = 0; i < get_input_size(); i+=d_H->k()) {
        d_H->encode(&out[j], &in[i]);
        j += d_H->n();
      }
    }

  }
}