Mirror/aicodix___code
1
0
Fork 0
mirror of https://github.com/aicodix/code.git synced 2026-04-27 22:35:44 +00:00
Code Issues Projects Releases Packages Wiki Activity

added Reed Solomon error correction

Browse source 
lifted from my FEC project:
https://github.com/xdsopl/FEC
This commit is contained in:
Ahmet Inan 2018-09-20 18:04:23 +02:00
commit 01db8f516e
10 changed files with 846 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

88
tests/rs_decoder_test.cc Normal file
View file

@ -0,0 +1,88 @@
/*
Test for the Reed Solomon Decoder
Copyright 2018 Ahmet Inan <inan@aicodix.de>
*/
#include <cassert>
#include <random>
#include <iostream>
#include "galois_field.hh"
#include "reed_solomon_decoder.hh"
int main()
{
std::random_device rd;
std::default_random_engine generator(rd());
if (1) {
// BBC WHP031 RS(15, 11) T=2
typedef CODE::GaloisField<4, 0b10011, uint8_t> GF;
typedef CODE::ReedSolomonDecoder<4, 0, GF> RS;
GF instance;
RS decode;
uint8_t target[15] = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 3, 3, 12, 12 };
uint8_t code[15];
for (int i = 0; i < 15; ++i)
code[i] = target[i];
std::uniform_int_distribution<uint8_t> distribution(0, 15);
auto noise = std::bind(distribution, generator);
for (int i = 0; i < 2; ++i)
code[noise()] = noise();
decode(code);
for (int i = 0; i < 15; ++i)
assert(code[i] == target[i]);
}
if (1) {
// DVB-T RS(255, 239) T=8
typedef CODE::GaloisField<8, 0b100011101, uint8_t> GF;
typedef CODE::ReedSolomonDecoder<16, 0, GF> RS;
GF instance;
RS decode;
uint8_t target[255];
for (int i = 0; i < 239; ++i)
target[i] = i + 1;
uint8_t parity[16] = { 1, 126, 147, 48, 155, 224, 3, 157, 29, 226, 40, 114, 61, 30, 244, 75 };
for (int i = 0; i < 16; ++i)
target[239+i] = parity[i];
uint8_t code[255];
for (int i = 0; i < 255; ++i)
code[i] = target[i];
std::uniform_int_distribution<uint8_t> distribution(0, 255);
auto noise = std::bind(distribution, generator);
for (int i = 0; i < 8; ++i)
code[noise()] = noise();
decode(code);
for (int i = 0; i < 255; ++i)
assert(code[i] == target[i]);
}
if (1) {
// FUN RS(65535, 65471) T=32
typedef CODE::GaloisField<16, 0b10001000000001011, uint16_t> GF;
typedef CODE::ReedSolomonDecoder<64, 1, GF> RS;
GF *instance = new GF();
RS *decode = new RS();
uint16_t *target = new uint16_t[65535];
for (int i = 0; i < 65471; ++i)
target[i] = i + 1;
uint16_t parity[64] = { 25271, 26303, 22052, 31318, 31233, 6076, 40148, 29468, 47507, 32655, 12404, 13265, 23901, 38403, 50967, 50433, 40818, 226, 62296, 23636, 56393, 12952, 11476, 44416, 518, 50014, 10037, 57582, 33421, 42654, 54025, 7157, 4826, 52148, 17167, 23294, 6427, 40953, 11168, 35305, 18209, 1868, 39971, 54928, 27566, 1424, 4846, 25347, 34710, 42190, 56452, 21859, 49805, 28028, 41657, 25756, 22014, 24479, 28758, 17438, 12976, 61743, 46735, 1557 };
for (int i = 0; i < 64; ++i)
target[65471+i] = parity[i];
uint16_t *code = new uint16_t[65535];
for (int i = 0; i < 65535; ++i)
code[i] = target[i];
std::uniform_int_distribution<uint16_t> distribution(0, 65535);
auto noise = std::bind(distribution, generator);
for (int i = 0; i < 32; ++i)
code[noise()] = noise();
(*decode)(code);
for (int i = 0; i < 65535; ++i)
assert(code[i] == target[i]);
delete[] target;
delete[] code;
delete decode;
delete instance;
}
std::cerr << "Reed Solomon Decoder test passed!" << std::endl;
return 0;
}