1 files changed, 125 insertions, 0 deletions
diff --git a/gr-fec/lib/ldpc_encoder_impl.cc b/gr-fec/lib/ldpc_encoder_impl.cc
new file mode 100755
index 0000000000..ccc9de8cbd
--- /dev/null
+++ b/gr-fec/lib/ldpc_encoder_impl.cc
@@ -0,0 +1,125 @@
+/* -*- 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 {
+    namespace code{
+
+      generic_encoder::sptr
+      ldpc_encoder::make(std::string alist_file, unsigned int gap)
+      {
+        ldpc_H_matrix::sptr H_obj = ldpc_H_matrix::make(alist_file, gap);
+        return make_H(H_obj);
+      }
+
+      generic_encoder::sptr
+      ldpc_encoder::make_H(const ldpc_H_matrix::sptr H_obj)
+      {
+        return generic_encoder::sptr
+          (new ldpc_encoder_impl(H_obj));
+      }
+
+      ldpc_encoder_impl::ldpc_encoder_impl(const 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();
+        }
+      }
+
+    }
+  }
+}