blob: e68a88cd4ef826b06d3998f9c71b44472a7ee9b0 (
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
|
/* -*- 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.
*/
/* -----------------------------------------------------------------
*
* This class handles sparse matrices specified in alist-format.
* For details about alist format please visit the link below.
* - http://www.inference.phy.cam.ac.uk/mackay/codes/alist.html
*
* Alist class is an efficient way of representing a sparse matrix
* the parity check matrix H of an LDPC code for instance.
*
*/
#ifndef ALIST_H
#define ALIST_H
#include <iostream>
#include <fstream>
#include <sstream>
#include <vector>
#include <stdlib.h>
#include <gnuradio/fec/api.h>
class FEC_API alist
{
public:
//! Default Constructor
alist() : data_ok(false) {}
//! Constructor which loads alist class from an alist-file
alist(const char * fname);
//! Read alist data from a file
void read(const char * fname);
//! Write alist data to a file
void write(const char * fname) const;
//! Returns N, the number of variable nodes
int get_N();
//! Return M, the number of check nodes
int get_M();
//! Return the m_list variable
std::vector< std::vector<int> > get_mlist();
//! Returns the n_list variable
std::vector< std::vector<int> > get_nlist();
//! Returns the num_mlist variable
std::vector<int> get_num_mlist();
//! Returns the num_nlist variable
std::vector<int> get_num_nlist();
//! Returns the max_num_nlist variable
int get_max_num_nlist();
//! Returns the max_num_mlist variable
int get_max_num_mlist();
//! Prints the nlist[i] variable
void print_nlist_i(int i);
//! Prints the mlist[i] variable
void print_mlist_i(int i);
//! Returns the corresponding H matrix
std::vector<std::vector<char> > get_matrix();
protected:
//! A variable indicating if data has been read from alist-file
bool data_ok;
//! Number of variable nodes
int N;
//! Number of check nodes
int M;
//! Maximum weight of rows
int max_num_mlist;
//! Maximum weight of columns
int max_num_nlist;
//! Weight of each column n
std::vector<int> num_nlist;
//! Weight of each row m
std::vector<int> num_mlist;
//! List of integer coordinates along each rows with non-zero entries
std::vector< std::vector<int> > mlist;
//! List of integer coordinates along each column with non-zero entries
std::vector< std::vector<int> > nlist;
};
#endif // ifndef ALIST_H
|
