Skip to content
Normal view History Permalink
Precode_Matrix.hpp 3.03 KiB
Newer Older
Luker's avatar
Initial Commit
Luker committed
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 
/*
 * Copyright (c) 2015, Luca Fulchir<luca@fulchir.it>, All rights reserved.
 *
 * This file is part of "libRaptorQ".
 *
 * libRaptorQ is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3
 * of the License, or (at your option) any later version.
 *
 * libRaptorQ 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
 * and a copy of the GNU Lesser General Public License
 * along with libRaptorQ.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef RAPTORQ_PRECODE_MATRIX_HPP
#define RAPTORQ_PRECODE_MATRIX_HPP

#include "common.hpp"
#include "Parameters.hpp"
#include "Rand.hpp"
#include "multiplication.hpp"
#include <Eigen/Dense>

namespace RaptorQ {
namespace Impl {

// track the bitmask of holes for received symbols.
// also make it easy to know how many non-repair symbols we are missing.
class RAPTORQ_API Bitmask
{
public:
Luker's avatar
Do not add repair symbols twice  
Luker committed
38 
	const uint16_t _max_nonrepair;
Luker's avatar
Initial Commit
Luker committed
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 

	Bitmask (const uint16_t symbols);

	void add (const size_t id);
	bool exists (const size_t id) const;
	uint16_t get_holes () const;
private:
	std::vector<size_t> mask;
	uint16_t holes;
};

using DenseMtx = Eigen::Matrix<Octet, Eigen::Dynamic, Eigen::Dynamic,
															Eigen::RowMajor>;

class RAPTORQ_API Precode_Matrix
{
public:
	const Parameters _params;

	Precode_Matrix(const Parameters &params)
		:_params (params)
	{}
Luker's avatar
seems to be working  
Luker committed
62 
	void gen (const uint16_t repair_overhead);
Luker's avatar
Initial Commit
Luker committed
63
Luker's avatar
seems to be working  
Luker committed
64 
	DenseMtx intermediate_eigen (DenseMtx &D);
Luker's avatar
Initial Commit
Luker committed
65 
	DenseMtx intermediate (DenseMtx &D);
Luker's avatar
Use iterators everywhere  
Luker committed
66 
	DenseMtx intermediate (DenseMtx &D, const Bitmask &mask,
Luker's avatar
Initial Commit
Luker committed
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 
									const std::vector<uint32_t> &repair_esi);
	DenseMtx encode (const DenseMtx &C, const uint32_t iSI) const;

private:
	DenseMtx A;
	uint16_t _repair_overhead;

	// indenting here prepresent which function needs which other.
	// not standard, ask me if I care.
	void init_LDPC1 (const uint16_t S, const uint16_t B);
	void init_LDPC2 (const uint16_t skip, const uint16_t rows,
														const uint16_t cols);
	void add_identity (const uint16_t size, const uint16_t skip_row,
													const uint16_t skip_col);

	void init_HDPC ();
		DenseMtx make_MT() const;		// rfc 6330, pgg 24, used for HDPC
		DenseMtx make_GAMMA() const;	// rfc 6330, pgg 24, used for HDPC
	void add_G_ENC ();


	void decode_phase0 (const Bitmask &mask,
									const std::vector<uint32_t> &repair_esi);
	std::tuple<bool, uint16_t, uint16_t> decode_phase1 (DenseMtx &X,DenseMtx &D,
													std::vector<uint16_t> &c);
	bool decode_phase2 (DenseMtx &D, const uint16_t i,const uint16_t u);
	void decode_phase3 (const DenseMtx &X, DenseMtx &D, const uint16_t i);
	void decode_phase4 (DenseMtx &D, const uint16_t i, const uint16_t u);
	void decode_phase5 (DenseMtx &D, const uint16_t i);
};

}	// namespace Impl
}	// namespace RaptorQ

#endif