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

added metric() helper

Browse source
This commit is contained in:
Ahmet Inan 2020-06-09 13:33:42 +02:00
commit a33c2082d9
1 changed files with 18 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

44
short_bch_code_decoder.hh
View file

@ -26,6 +26,14 @@ class ShortBCHCodeDecoder
}
return inp;
}
static int metric(const int8_t *soft, int hard)
{
int lol = 8 * sizeof(hard) - 1;
int sum = 0;
for (int i = 0; i < N; ++i)
sum += ((hard << (lol-i)) >> lol) * soft[i];
return sum;
}
public:
ShortBCHCodeDecoder(int generator, int T)
{
@ -56,13 +64,10 @@ public:
if (0) {
int word = 0, best = 0;
for (int msg = 0; msg < W; ++msg) {
int sum = 0;
int enc = (msg << P) | par[msg];
int lol = 8 * sizeof(enc) - 1;
for (int i = 0; i < N; ++i)
sum += ((enc << (lol-i)) >> lol) * code[i];
if (sum > best) {
best = sum;
int met = metric(code, enc);
if (met > best) {
best = met;
word = enc;
}
}
@ -75,26 +80,16 @@ public:
// hard decision
if (0)
return word;
int best = 0;
// metric of hard decision
if (1) {
int lol = 8 * sizeof(word) - 1;
int sum = 0;
for (int i = 0; i < N; ++i)
sum += ((word << (lol-i)) >> lol) * code[i];
best = sum;
}
int best = metric(code, word);
// flip each bit and see ..
if (0) {
for (int j = 0; j < N; ++j) {
int tmp = 1 << j;
int dec = (*this)(cw ^ tmp);
int lol = 8 * sizeof(dec) - 1;
int sum = 0;
for (int i = 0; i < N; ++i)
sum += ((dec << (lol-i)) >> lol) * code[i];
if (sum > best) {
best = sum;
int met = metric(code, dec);
if (met > best) {
best = met;
word = dec;
}
}
@ -125,12 +120,9 @@ public:
for (int i = 0; i < num; ++i)
tmp |= ((j>>i)&1) << worst[i];
int dec = (*this)(cw ^ tmp);
int lol = 8 * sizeof(dec) - 1;
int sum = 0;
for (int i = 0; i < N; ++i)
sum += ((dec << (lol-i)) >> lol) * code[i];
if (sum > best) {
best = sum;
int met = metric(code, dec);
if (met > best) {
best = met;
word = dec;
}
}