1
0
mirror of https://github.com/f4exb/sdrangel.git synced 2024-11-23 08:28:36 -05:00
sdrangel/plugins/channelrx/demoddatv/leansdr/dvb.h

1709 lines
51 KiB
C
Raw Normal View History

2018-02-22 16:52:49 -05:00
#ifndef LEANSDR_DVB_H
#define LEANSDR_DVB_H
#include <stdint.h>
#include "leansdr/viterbi.h"
#include "leansdr/convolutional.h"
#include "leansdr/sdr.h"
#include "leansdr/rs.h"
#include "leansdr/incrementalarray.h"
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
namespace leansdr
{
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
static const int SIZE_RSPACKET = 204;
static const int MPEG_SYNC = 0x47;
static const int MPEG_SYNC_INV = (MPEG_SYNC ^ 0xff);
static const int MPEG_SYNC_CORRUPTED = 0x55;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// Generic deconvolution
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
enum code_rate
{
2018-02-22 16:52:49 -05:00
FEC12, FEC23, FEC46, FEC34, FEC56, FEC78, // DVB-S
2018-02-24 18:07:08 -05:00
FEC45,
FEC89,
FEC910, // DVB-S2
2018-02-22 16:52:49 -05:00
FEC_MAX
2018-02-24 18:07:08 -05:00
};
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// Customize APSK radii according to code rate
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
inline cstln_lut<256> * make_dvbs2_constellation(cstln_lut<256>::predef c,
code_rate r)
{
float gamma1 = 1, gamma2 = 1, gamma3 = 1;
switch (c)
{
2018-02-22 16:52:49 -05:00
case cstln_lut<256>::APSK16:
2018-02-24 18:07:08 -05:00
// EN 302 307, section 5.4.3, Table 9
switch (r)
{
case FEC23:
case FEC46:
gamma1 = 3.15;
break;
case FEC34:
gamma1 = 2.85;
break;
case FEC45:
gamma1 = 2.75;
break;
case FEC56:
gamma1 = 2.70;
break;
case FEC89:
gamma1 = 2.60;
break;
case FEC910:
gamma1 = 2.57;
break;
default:
fail("cstln_lut<256>::make_dvbs2_constellation", "Code rate not supported with APSK16");
return 0;
2018-02-24 18:07:08 -05:00
}
break;
2018-02-22 16:52:49 -05:00
case cstln_lut<256>::APSK32:
2018-02-24 18:07:08 -05:00
// EN 302 307, section 5.4.4, Table 10
switch (r)
{
case FEC34:
gamma1 = 2.84;
gamma2 = 5.27;
break;
case FEC45:
gamma1 = 2.72;
gamma2 = 4.87;
break;
case FEC56:
gamma1 = 2.64;
gamma2 = 4.64;
break;
case FEC89:
gamma1 = 2.54;
gamma2 = 4.33;
break;
case FEC910:
gamma1 = 2.53;
gamma2 = 4.30;
break;
default:
fail("cstln_lut<256>::make_dvbs2_constellation", "Code rate not supported with APSK32");
return 0;
2018-02-24 18:07:08 -05:00
}
break;
2018-02-22 16:52:49 -05:00
case cstln_lut<256>::APSK64E:
2018-02-24 18:07:08 -05:00
// EN 302 307-2, section 5.4.5, Table 13f
gamma1 = 2.4;
gamma2 = 4.3;
gamma3 = 7;
break;
2018-02-22 16:52:49 -05:00
default:
2018-02-24 18:07:08 -05:00
break;
2018-02-22 16:52:49 -05:00
}
return new cstln_lut<256>(c, gamma1, gamma2, gamma3);
2018-02-24 18:07:08 -05:00
}
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// EN 300 421, section 4.4.3, table 2 Punctured code, G1=0171, G2=0133
static const int DVBS_G1 = 121;
static const int DVBS_G2 = 91;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// G1 = 0b1111001
2018-02-22 16:52:49 -05:00
// G2 = 0b1011011
//
// G1 = [ 1 1 1 1 0 0 1 ]
// G2 = [ 1 0 1 1 0 1 1 ]
//
// C = [ G2 ;
// G1 ;
// 0 G2 ;
// 0 G1 ;
// 0 0 G2 ;
// 0 0 G1 ]
//
// C = [ 1 0 1 1 0 1 1 0 0 0 0 0 0 ;
// 1 1 1 1 0 0 1 0 0 0 0 0 0 ;
// 0 1 0 1 1 0 1 1 0 0 0 0 0 ;
// 0 1 1 1 1 0 0 1 0 0 0 0 0 ;
// 0 0 1 0 1 1 0 1 1 0 0 0 0 ;
// 0 0 1 1 1 1 0 0 1 0 0 0 0 ;
// 0 0 0 1 0 1 1 0 1 1 0 0 0 ;
// 0 0 0 1 1 1 1 0 0 1 0 0 0 ;
// 0 0 0 0 1 0 1 1 0 1 1 0 0 ;
// 0 0 0 0 1 1 1 1 0 0 1 0 0 ;
// 0 0 0 0 0 1 0 1 1 0 1 1 0 ;
// 0 0 0 0 0 1 1 1 1 0 0 1 0 ;
// 0 0 0 0 0 0 1 0 1 1 0 1 1 ;
// 0 0 0 0 0 0 1 1 1 1 0 0 1 ]
//
// IQ = [ Q1; I1; ... Q10; I10 ] = C * S
//
// D * C == [ 1 0 0 0 0 0 0 0 0 0 0 0 0 0 ]
2018-02-24 18:07:08 -05:00
//
2018-02-22 16:52:49 -05:00
// D = [ 0 1 0 1 1 1 0 1 1 1 0 0 0 0]
2018-02-24 18:07:08 -05:00
// D = 0x3ba
template<typename Tbyte, Tbyte BYTE_ERASED>
struct deconvol_sync: runnable
{
deconvol_sync(scheduler *sch, pipebuf<softsymbol> &_in,
pipebuf<Tbyte> &_out, uint32_t gX, uint32_t gY, uint32_t pX,
uint32_t pY) :
runnable(sch, "deconvol_sync"), fastlock(false), in(_in), out(_out,
SIZE_RSPACKET), skip(0)
{
conv = new uint32_t[2];
conv[0] = gX;
conv[1] = gY;
nG = 2;
punct = new uint32_t[2];
punct[0] = pX;
punct[1] = pY;
punctperiod = 0;
punctweight = 0;
for (int i = 0; i < 2; ++i)
{
int nbits = log2(punct[i]) + 1;
if (nbits > punctperiod)
punctperiod = nbits;
punctweight += hamming_weight(punct[i]);
}
if (sch->verbose)
fprintf(stderr, "puncturing %d/%d\n", punctperiod, punctweight);
deconv = new iq_t[punctperiod];
deconv2 = new iq_t[punctperiod];
inverse_convolution();
init_syncs();
locked = &syncs[0];
2018-02-22 16:52:49 -05:00
}
typedef uint64_t signal_t;
typedef uint64_t iq_t;
2018-02-24 18:07:08 -05:00
static int log2(uint64_t x)
{
int n = -1;
for (; x; ++n, x >>= 1)
;
return n;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
iq_t convolve(signal_t s)
{
int sbits = log2(s) + 1;
iq_t iq = 0;
unsigned char state = 0;
for (int b = sbits - 1; b >= 0; --b)
{ // Feed into convolver, MSB first
unsigned char bit = (s >> b) & 1;
state = (state >> 1) | (bit << 6); // Shift register
for (int j = 0; j < nG; ++j)
{
unsigned char xy = parity(state & conv[j]); // Taps
if (punct[j] & (1 << (b % punctperiod)))
iq = (iq << 1) | xy;
}
}
return iq;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
run_decoding();
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void next_sync()
{
if (fastlock)
{
fail("deconvol_sync::next_sync", "Bug: next_sync() called with fastlock");
return;
}
2018-02-24 18:07:08 -05:00
++locked;
if (locked == &syncs[NSYNCS])
{
locked = &syncs[0];
// Try next symbol alignment (for FEC other than 1/2)
skip = 1;
}
2018-02-22 16:52:49 -05:00
}
bool fastlock;
2018-02-24 18:07:08 -05:00
private:
2018-02-22 16:52:49 -05:00
static const int maxsbits = 64;
iq_t response[maxsbits];
//static const int traceback = 48; // For code rate 7/8
static const int traceback = 64; // For code rate 7/8 with fastlock
2018-02-24 18:07:08 -05:00
void solve_rec(iq_t prefix, unsigned int nprefix, signal_t exp, iq_t *best)
{
if (prefix > *best)
return;
if (nprefix > sizeof(prefix) * 8)
return;
int solved = 1;
for (int b = 0; b < maxsbits; ++b)
{
if (parity(prefix & response[b]) != ((exp >> b) & 1))
{
// Current candidate does not solve this column.
if ((response[b] >> nprefix) == 0)
// No more bits to trace back.
return;
solved = 0;
}
}
if (solved)
{
*best = prefix;
return;
}
solve_rec(prefix, nprefix + 1, exp, best);
solve_rec(prefix | ((iq_t) 1 << nprefix), nprefix + 1, exp, best);
2018-02-22 16:52:49 -05:00
}
static const int LATENCY = 0;
2018-02-24 18:07:08 -05:00
void inverse_convolution()
{
for (int sbit = 0; sbit < maxsbits; ++sbit)
{
response[sbit] = convolve((iq_t) 1 << sbit);
//fprintf(stderr, "response %d = %x\n", sbit, response[sbit]);
}
for (int b = 0; b < punctperiod; ++b)
{
deconv[b] = -(iq_t) 1;
solve_rec(0, 0, 1 << (LATENCY + b), &deconv[b]);
}
// Alternate polynomials for fastlock
for (int b = 0; b < punctperiod; ++b)
{
uint64_t d = deconv[b], d2 = d;
// 1/2
if (d == 0x00000000000003baLL)
d2 = 0x0000000000038ccaLL;
// 2/3
if (d == 0x0000000000000f29LL)
d2 = 0x000000003c569329LL;
if (d == 0x000000000003c552LL)
d2 = 0x00000000001dee1cLL;
if (d == 0x0000000000007948LL)
d2 = 0x00000001e2b49948LL;
if (d == 0x00000000000001deLL)
d2 = 0x00000000001e2a90LL;
// 3/4
if (d == 0x000000000000f247LL)
d2 = 0x000000000fd6383bLL;
if (d == 0x00000000000fd9eeLL)
d2 = 0x000000000fd91392LL;
if (d == 0x0000000000f248d8LL)
d2 = 0x00000000fd9eef18LL;
// 5/6
if (d == 0x0000000000f5727fLL)
d2 = 0x000003d5c909758fLL;
if (d == 0x000000003d5c90aaLL)
d2 = 0x0f5727f0229c90aaLL;
if (d == 0x000000003daa371cLL)
d2 = 0x000003d5f45630ecLL;
if (d == 0x0000000f5727ff48LL)
d2 = 0x0000f57d28260348LL;
if (d == 0x0000000f57d28260LL)
d2 = 0xf5727ff48128260LL;
// 7/8
if (d == 0x0000fbeac76c454fLL)
d2 = 0x00fb11d6ba045a8fLL;
if (d == 0x00000000fb11d6baLL)
d2 = 0xfbea3c7d930e16baLL;
if (d == 0x0000fb112d5038dcLL)
d2 = 0x00fb112d5038271cLL;
if (d == 0x000000fbea3c7d68LL)
d2 = 0x00fbeac7975462a8LL;
if (d == 0x00000000fb112d50LL)
d2 = 0x00fbea3c86793290LL;
if (d == 0x0000fb112dabd2e0LL)
d2 = 0x00fb112d50c3cd20LL;
if (d == 0x00000000fb11d640LL)
d2 = 0x00fbea3c8679c980LL;
if (d2 == d)
{
fail("deconvol_sync::inverse_convolution", "Alt polynomial not provided");
return;
}
2018-02-24 18:07:08 -05:00
deconv2[b] = d2;
}
if (sch->debug)
{
for (int b = 0; b < punctperiod; ++b)
fprintf(stderr, "deconv[%d]=0x%016llx %d taps / %d bits\n", b,
(unsigned long long) deconv[b],
hamming_weight(deconv[b]), log2(deconv[b]) + 1);
}
// Sanity check
for (int b = 0; b < punctperiod; ++b)
{
for (int i = 0; i < maxsbits; ++i)
{
iq_t iq = convolve((iq_t) 1 << (LATENCY + i));
int expect = (b == i) ? 1 : 0;
int d = parity(iq & deconv[b]);
if (d != expect)
{
fail("deconvol_sync::inverse_convolution", "Failed to inverse convolutional coding");
}
2018-02-24 18:07:08 -05:00
int d2 = parity(iq & deconv2[b]);
if (d2 != expect)
{
fail("deconvol_sync::inverse_convolution", "Failed to inverse convolutional coding (alt)");
}
2018-02-24 18:07:08 -05:00
}
if (traceback > sizeof(iq_t) * 8)
{
fail("deconvol_sync::inverse_convolution", "Bug: traceback exceeds register size");
}
2018-02-24 18:07:08 -05:00
if (log2(deconv[b]) + 1 > traceback)
{
fail("deconvol_sync::inverse_convolution", "traceback insufficient for deconvolution");
}
2018-02-24 18:07:08 -05:00
if (log2(deconv2[b]) + 1 > traceback)
{
fail("deconvol_sync::inverse_convolution", "traceback insufficient for deconvolution (alt)");
}
2018-02-24 18:07:08 -05:00
}
2018-02-22 16:52:49 -05:00
}
static const int NSYNCS = 4;
2018-02-24 18:07:08 -05:00
struct sync_t
{
u8 lut[2][2]; // lut[(re>0)?1:0][(im>0)?1:0] = 0b000000IQ
iq_t in;
int n_in;
signal_t out;
int n_out;
// Auxiliary shift register for fastlock
iq_t in2;
int n_in2, n_out2;
2018-02-22 16:52:49 -05:00
} syncs[NSYNCS];
2018-02-24 18:07:08 -05:00
void init_syncs()
{
// EN 300 421, section 4.5, Figure 5 QPSK constellation
// Four rotations * two conjugations.
// 180° rotation is detected as polarity inversion in mpeg_sync.
for (int sync_id = 0; sync_id < NSYNCS; ++sync_id)
{
for (int re_pos = 0; re_pos <= 1; ++re_pos)
for (int im_pos = 0; im_pos <= 1; ++im_pos)
{
int re_neg = !re_pos;
int im_neg __attribute__((unused)) = !im_pos;
int I, Q;
switch (sync_id)
{
case 0: // Direct 0°
I = re_pos ? 0 : 1;
Q = im_pos ? 0 : 1;
break;
case 1: // Direct 90°
I = im_pos ? 0 : 1;
Q = re_neg ? 0 : 1;
break;
case 2: // Conj 0°
I = re_pos ? 0 : 1;
Q = im_pos ? 1 : 0;
break;
case 3: // Conj 90°
I = im_pos ? 1 : 0;
Q = re_neg ? 0 : 1;
break;
2018-02-22 16:52:49 -05:00
#if 0
2018-02-24 18:07:08 -05:00
case 4: // Direct 180°
I = re_neg ? 0 : 1;
Q = im_neg ? 0 : 1;
break;
case 5:// Direct 270°
I = im_neg ? 0 : 1;
Q = re_pos ? 0 : 1;
break;
case 6:// Conj 180°
I = re_neg ? 0 : 1;
Q = im_neg ? 1 : 0;
break;
case 7:// Conj 270°
I = im_neg ? 1 : 0;
Q = re_pos ? 0 : 1;
break;
2018-02-22 16:52:49 -05:00
#endif
2018-02-24 18:07:08 -05:00
}
syncs[sync_id].lut[re_pos][im_pos] = (I << 1) | Q;
}
syncs[sync_id].n_in = 0;
syncs[sync_id].n_out = 0;
syncs[sync_id].n_in2 = 0;
syncs[sync_id].n_out2 = 0;
}
2018-02-22 16:52:49 -05:00
}
// TODO: Unroll for each code rate setting.
// 1/2: 8 symbols -> 1 byte
// 2/3 12 symbols -> 2 bytes
// 3/4 16 symbols -> 3 bytes
// 5/6 24 symbols -> 5 bytes
// 7/8 32 symbols -> 7 bytes
2018-02-24 18:07:08 -05:00
inline Tbyte readbyte(sync_t *s, softsymbol *&p)
{
while (s->n_out < 8)
{
iq_t iq = s->in;
while (s->n_in < traceback)
{
u8 iqbits = s->lut[(p->symbol & 2) ? 1 : 0][p->symbol & 1];
++p;
iq = (iq << 2) | iqbits;
s->n_in += 2;
}
s->in = iq;
for (int b = punctperiod - 1; b >= 0; --b)
{
u8 bit = parity(iq & deconv[b]);
s->out = (s->out << 1) | bit;
}
s->n_out += punctperiod;
s->n_in -= punctweight;
}
Tbyte res = (s->out >> (s->n_out - 8)) & 255;
s->n_out -= 8;
return res;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
inline unsigned long readerrors(sync_t *s, softsymbol *&p)
{
unsigned long res = 0;
while (s->n_out2 < 8)
{
iq_t iq = s->in2;
while (s->n_in2 < traceback)
{
u8 iqbits = s->lut[(p->symbol & 2) ? 1 : 0][p->symbol & 1];
++p;
iq = (iq << 2) | iqbits;
s->n_in2 += 2;
}
s->in2 = iq;
for (int b = punctperiod - 1; b >= 0; --b)
{
u8 bit = parity(iq & deconv[b]);
u8 bit2 = parity(iq & deconv2[b]);
if (bit2 != bit)
++res;
}
s->n_out2 += punctperiod;
s->n_in2 -= punctweight;
}
s->n_out2 -= 8;
return res;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run_decoding()
{
in.read(skip);
skip = 0;
// 8 byte margin to fill the deconvolver
if (in.readable() < 64)
return;
int maxrd = (in.readable() - 64) / (punctweight / 2) * punctperiod / 8;
int maxwr = out.writable();
unsigned int n = (maxrd < maxwr) ? maxrd : maxwr;
if (!n)
return;
// Require enough symbols to discriminate in fastlock mode
// (threshold must be less than size of rspacket)
if (n < 32)
return;
if (fastlock)
{
// Try all sync alignments
unsigned long errors_best = 1 << 30;
sync_t *best = &syncs[0];
for (sync_t *s = syncs; s < syncs + NSYNCS; ++s)
{
softsymbol *pin = in.rd();
unsigned long errors = 0;
for (int c = n; c--;)
errors += readerrors(s, pin);
if (errors < errors_best)
{
errors_best = errors;
best = s;
}
}
if (best != locked)
{
// Another alignment produces fewer bit errors
if (sch->debug)
fprintf(stderr, "{%d->%d}\n", (int) (locked - syncs),
(int) (best - syncs));
locked = best;
}
// If deconvolution bit error rate > 33%, try next sample alignment
if (errors_best > n * 8 / 3)
{
// fprintf(stderr, ">");
skip = 1;
}
}
softsymbol *pin = in.rd(), *pin0 = pin;
Tbyte *pout = out.wr(), *pout0 = pout;
while (n--)
*pout++ = readbyte(locked, pin);
in.read(pin - pin0);
out.written(pout - pout0);
}
2018-02-22 16:52:49 -05:00
pipereader<softsymbol> in;
pipewriter<Tbyte> out;
// DECONVOL
int nG;
uint32_t *conv; // [nG] Convolution polynomials; MSB is newest
uint32_t *punct; // [nG] Puncturing pattern
int punctperiod, punctweight;
iq_t *deconv; // [punctperiod] Deconvolution polynomials
iq_t *deconv2; // [punctperiod] Alternate polynomials (for fastlock)
sync_t *locked;
int skip;
2018-02-24 18:07:08 -05:00
};
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
typedef deconvol_sync<u8, 0> deconvol_sync_simple;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
inline deconvol_sync_simple *make_deconvol_sync_simple(scheduler *sch,
pipebuf<softsymbol> &_in, pipebuf<u8> &_out, enum code_rate rate)
{
2018-02-22 16:52:49 -05:00
// EN 300 421, section 4.4.3 Inner coding
uint32_t pX, pY;
2018-02-24 18:07:08 -05:00
switch (rate)
{
2018-02-22 16:52:49 -05:00
case FEC12:
2018-02-24 18:07:08 -05:00
pX = 0x1; // 1
pY = 0x1; // 1
break;
2018-02-22 16:52:49 -05:00
case FEC23:
case FEC46:
2018-02-24 18:07:08 -05:00
pX = 0xa; // 1010 (Handle as FEC4/6, no half-symbols)
pY = 0xf; // 1111
break;
2018-02-22 16:52:49 -05:00
case FEC34:
2018-02-24 18:07:08 -05:00
pX = 0x5; // 101
pY = 0x6; // 110
break;
2018-02-22 16:52:49 -05:00
case FEC56:
2018-02-24 18:07:08 -05:00
pX = 0x15; // 10101
pY = 0x1a; // 11010
break;
2018-02-22 16:52:49 -05:00
case FEC78:
2018-02-24 18:07:08 -05:00
pX = 0x45; // 1000101
pY = 0x7a; // 1111010
break;
2018-02-22 16:52:49 -05:00
default:
2018-02-24 18:07:08 -05:00
//fail("Code rate not implemented");
// For testing DVB-S2 constellations.
fprintf(stderr, "Code rate not implemented; proceeding anyway\n");
pX = pY = 1;
2018-02-22 16:52:49 -05:00
}
return new deconvol_sync_simple(sch, _in, _out, DVBS_G1, DVBS_G2, pX, pY);
2018-02-24 18:07:08 -05:00
}
// CONVOLUTIONAL ENCODER
static const uint16_t polys_fec12[] =
{ DVBS_G1, DVBS_G2 // X1Y1
};
static const uint16_t polys_fec23[] =
{ DVBS_G1, DVBS_G2, DVBS_G2 << 1 // X1Y1Y2
};
// Same code rate as 2/3, usable with QPSK
static const uint16_t polys_fec46[] =
{ DVBS_G1, DVBS_G2, DVBS_G2 << 1, // X1Y1Y2
DVBS_G1 << 2, DVBS_G2 << 2, DVBS_G2 << 3 // X3Y3Y4
};
static const uint16_t polys_fec34[] =
{ DVBS_G1, DVBS_G2, // X1Y1
DVBS_G2 << 1, DVBS_G1 << 2 // Y2X3
};
static const uint16_t polys_fec45[] =
{ // Non standard
DVBS_G1, DVBS_G2, // X1Y1
DVBS_G2 << 1, DVBS_G1 << 2, // Y2X3
DVBS_G1 << 3 // X4
};
static const uint16_t polys_fec56[] =
{ DVBS_G1, DVBS_G2, // X1Y1
DVBS_G2 << 1, DVBS_G1 << 2, // Y2X3
DVBS_G2 << 3, DVBS_G1 << 4 // Y4X5
};
static const uint16_t polys_fec78[] =
{ DVBS_G1, DVBS_G2, // X1Y1
DVBS_G2 << 1, DVBS_G2 << 2, // Y2Y3
DVBS_G2 << 3, DVBS_G1 << 4, // Y4X5
DVBS_G2 << 5, DVBS_G1 << 6 // Y6X7
};
// FEC parameters, for convolutional coding only (not S2).
static struct fec_spec
{
2018-02-22 16:52:49 -05:00
int bits_in; // Entering the convolutional coder
int bits_out; // Exiting the convolutional coder
const uint16_t *polys; // [bits_out]
2018-02-24 18:07:08 -05:00
} fec_specs[FEC_MAX] =
{ [FEC12] =
{ 1, 2, polys_fec12 }, [FEC23] =
{ 2, 3, polys_fec23 }, [FEC46] =
{ 4, 6, polys_fec46 }, [FEC34] =
{ 3, 4, polys_fec34 }, [FEC56] =
{ 5, 6, polys_fec56 }, [FEC78] =
{ 7, 8, polys_fec78 }, [FEC45] =
{ 4, 5, polys_fec45 }, // Non-standard
};
struct dvb_convol: runnable
{
2018-02-22 16:52:49 -05:00
typedef u8 uncoded_byte;
typedef u8 hardsymbol;
2018-02-24 18:07:08 -05:00
dvb_convol(scheduler *sch, pipebuf<uncoded_byte> &_in,
pipebuf<hardsymbol> &_out, code_rate fec, int bits_per_symbol) :
runnable(sch, "dvb_convol"), in(_in), out(_out, 64) // BPSK 7/8: 7 bytes in, 64 symbols out
2018-02-22 16:52:49 -05:00
{
2018-02-24 18:07:08 -05:00
fec_spec *fs = &fec_specs[fec];
if (!fs->bits_in)
{
fail("dvb_convol::dvb_convol", "Unexpected FEC");
return;
}
2018-02-24 18:07:08 -05:00
convol.bits_in = fs->bits_in;
convol.bits_out = fs->bits_out;
convol.polys = fs->polys;
convol.bps = bits_per_symbol;
// FEC must output a whole number of IQ symbols
if (convol.bits_out % convol.bps)
{
fail("dvb_convol::dvb_convol", "Code rate not suitable for this constellation");
return;
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
int count = min(in.readable(),
out.writable() * convol.bps / convol.bits_out * convol.bits_in
/ 8);
// Process in multiples of the puncturing period and of 8 bits.
int chunk = convol.bits_in;
count = (count / chunk) * chunk;
convol.encode(in.rd(), out.wr(), count);
in.read(count);
int nout = count * 8 / convol.bits_in * convol.bits_out / convol.bps;
out.written(nout);
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
private:
2018-02-22 16:52:49 -05:00
pipereader<uncoded_byte> in;
pipewriter<hardsymbol> out;
convol_multipoly<uint16_t, 16> convol;
2018-02-24 18:07:08 -05:00
};
// dvb_convol
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// NEW ALGEBRAIC DECONVOLUTION
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// QPSK 1/2 only;
// With DVB-S polynomials hardcoded.
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
template<typename Tin>
struct dvb_deconvol_sync: runnable
{
2018-02-22 16:52:49 -05:00
typedef u8 decoded_byte;
int resync_period;
static const int chunk_size = 64; // At least 2*sizeof(Thist)/8
2018-02-24 18:07:08 -05:00
dvb_deconvol_sync(scheduler *sch, pipebuf<Tin> &_in,
pipebuf<decoded_byte> &_out) :
runnable(sch, "deconvol_sync_multipoly"), resync_period(32), in(
_in), out(_out, chunk_size), resync_phase(0)
2018-02-22 16:52:49 -05:00
{
2018-02-24 18:07:08 -05:00
init_syncs();
locked = &syncs[0];
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
while (in.readable() >= chunk_size * 8 && out.writable() >= chunk_size)
{
int errors_best = 1 << 30;
sync_t *best = NULL;
for (sync_t *s = syncs; s < syncs + NSYNCS; ++s)
{
if (resync_phase != 0 && s != locked)
// Decode only the currently-locked alignment
continue;
Tin *pin = in.rd();
static decoded_byte dummy[chunk_size];
decoded_byte *pout = (s == locked) ? out.wr() : dummy;
int nerrors = s->deconv.run(pin, s->lut, pout, chunk_size);
if (nerrors < errors_best)
{
errors_best = nerrors;
best = s;
}
}
in.read(chunk_size * 8);
out.written(chunk_size);
if (best != locked)
{
if (sch->debug)
fprintf(stderr, "%%%d", (int) (best - syncs));
locked = best;
}
if (++resync_phase >= resync_period)
resync_phase = 0;
} // Work to do
2018-02-22 16:52:49 -05:00
} // run()
2018-02-24 18:07:08 -05:00
private:
2018-02-22 16:52:49 -05:00
pipereader<Tin> in;
pipewriter<decoded_byte> out;
int resync_phase;
2018-02-24 18:07:08 -05:00
2018-02-22 16:52:49 -05:00
static const int NSYNCS = 4;
2018-02-24 18:07:08 -05:00
struct sync_t
{
deconvol_poly2<Tin, uint32_t, uint64_t, 0x3ba, 0x38f70> deconv;
u8 lut[4]; // TBD Swap and flip bits in the polynomials instead.
2018-02-22 16:52:49 -05:00
} syncs[NSYNCS];
sync_t *locked;
2018-02-24 18:07:08 -05:00
void init_syncs()
{
for (int s = 0; s < NSYNCS; ++s)
// EN 300 421, section 4.5, Figure 5 QPSK constellation
// Four rotations * two conjugations.
// 180° rotation is detected as polarity inversion in mpeg_sync.
// 2 | 0
// --+--
// 3 | 1
// 0°
syncs[0].lut[0] = 0;
syncs[0].lut[1] = 1;
syncs[0].lut[2] = 2;
syncs[0].lut[3] = 3;
// 90°
syncs[1].lut[0] = 2;
syncs[1].lut[1] = 0;
syncs[1].lut[2] = 3;
syncs[1].lut[3] = 1;
// 0° conjugated
syncs[2].lut[0] = 1;
syncs[2].lut[1] = 0;
syncs[2].lut[2] = 3;
syncs[2].lut[3] = 2;
// 90° conjugated
syncs[3].lut[0] = 0;
syncs[3].lut[1] = 2;
syncs[3].lut[2] = 1;
syncs[3].lut[3] = 3;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
};
// dvb_deconvol_sync
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
typedef dvb_deconvol_sync<softsymbol> dvb_deconvol_sync_soft;
typedef dvb_deconvol_sync<u8> dvb_deconvol_sync_hard;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// BIT ALIGNMENT AND MPEG SYNC DETECTION
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
template<typename Tbyte, Tbyte BYTE_ERASED>
struct mpeg_sync: runnable
{
2018-02-22 16:52:49 -05:00
int scan_syncs, want_syncs;
unsigned long lock_timeout;
bool fastlock;
int resync_period;
2018-02-24 18:07:08 -05:00
mpeg_sync(
scheduler *sch,
pipebuf<Tbyte> &_in,
pipebuf<Tbyte> &_out,
deconvol_sync<Tbyte, 0> *_deconv,
pipebuf<int> *_state_out = NULL,
pipebuf<unsigned long> *_locktime_out = NULL) :
runnable(sch, "sync_detect"),
scan_syncs(8),
want_syncs(4),
lock_timeout(4),
fastlock(false),
resync_period(1),
in(_in),
out(_out, SIZE_RSPACKET * (scan_syncs + 1)),
deconv(_deconv),
polarity(0),
resync_phase(0),
bitphase(0),
synchronized(false),
next_sync_count(0),
phase8(-1),
lock_timeleft(0),
locktime(0),
report_state(true)
{
state_out = _state_out ? new pipewriter<int>(*_state_out) : NULL;
locktime_out = locktime_out ? new pipewriter<unsigned long>(*_locktime_out) : NULL;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
if (report_state && state_out && state_out->writable() >= 1)
{
// Report unlocked state on first invocation.
state_out->write(0);
report_state = false;
}
if (synchronized)
run_decoding();
else
{
if (fastlock)
run_searching_fast();
else
run_searching();
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run_searching()
{
bool next_sync = false;
unsigned int chunk = SIZE_RSPACKET * scan_syncs;
while (in.readable() >= chunk + 1 && // Need 1 ahead for bit shifting
out.writable() >= chunk && // Use as temp buffer
(!state_out || state_out->writable() >= 1))
{
if (search_sync())
return;
in.read(chunk);
// Switch to next bit alignment
++bitphase;
if (bitphase == 8)
{
bitphase = 0;
next_sync = true;
}
}
if (next_sync)
{
// No lock this time
++next_sync_count;
if (next_sync_count >= 3)
{
// After a few cycles without a lock, resync the deconvolver.
next_sync_count = 0;
if (deconv)
deconv->next_sync();
}
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run_searching_fast()
{
unsigned int chunk = SIZE_RSPACKET * scan_syncs;
while (in.readable() >= chunk + 1 && // Need 1 ahead for bit shifting
out.writable() >= chunk && // Use as temp buffer
(!state_out || state_out->writable() >= 1))
{
if (resync_phase == 0)
{
// Try all bit alighments
for (bitphase = 0; bitphase <= 7; ++bitphase)
{
if (search_sync())
return;
}
}
in.read(SIZE_RSPACKET);
if (++resync_phase >= resync_period)
resync_phase = 0;
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
bool search_sync()
{
int chunk = SIZE_RSPACKET * scan_syncs;
// Bit-shift [scan_sync] packets according to current [bitphase]
Tbyte *pin = in.rd(), *pend = pin + chunk;
Tbyte *pout = out.wr();
unsigned short w = *pin++;
for (; pin <= pend; ++pin, ++pout)
{
w = (w << 8) | *pin;
*pout = w >> bitphase;
}
// Search for [want_sync] start codes at all 204 offsets
for (int i = 0; i < SIZE_RSPACKET; ++i)
{
int nsyncs_p = 0, nsyncs_n = 0; // # start codes assuming pos/neg polarity
int phase8_p = -1, phase8_n = -1; // Position in sequence of 8 packets
Tbyte *p = &out.wr()[i];
for (int j = 0; j < scan_syncs; ++j, p += SIZE_RSPACKET)
{
Tbyte b = *p;
if (b == MPEG_SYNC)
{
++nsyncs_p;
phase8_n = (8 - j) & 7;
}
if (b == MPEG_SYNC_INV)
{
++nsyncs_n;
phase8_p = (8 - j) & 7;
}
}
// Detect most likely polarity
int nsyncs;
if (nsyncs_p > nsyncs_n)
{
polarity = 0;
nsyncs = nsyncs_p;
phase8 = phase8_p;
}
else
{
polarity = -1;
nsyncs = nsyncs_n;
phase8 = phase8_n;
}
if (nsyncs >= want_syncs && phase8 >= 0)
{
if (sch->debug)
fprintf(stderr, "Locked\n");
if (!i)
{ // Avoid fixpoint detection in scheduler
i = SIZE_RSPACKET;
phase8 = (phase8 + 1) & 7;
}
in.read(i); // Skip to first start code
synchronized = true;
lock_timeleft = lock_timeout;
locktime = 0;
if (state_out)
state_out->write(1);
return true;
}
}
return false;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run_decoding()
{
while (in.readable() >= SIZE_RSPACKET + 1
&& // +1 for bit shifting
out.writable() >= SIZE_RSPACKET
&& (!state_out || state_out->writable() >= 1)
&& (!locktime_out || locktime_out->writable() >= 1))
{
Tbyte *pin = in.rd(), *pend = pin + SIZE_RSPACKET;
Tbyte *pout = out.wr();
unsigned short w = *pin++;
for (; pin <= pend; ++pin, ++pout)
{
w = (w << 8) | *pin;
*pout = (w >> bitphase) ^ polarity;
}
in.read(SIZE_RSPACKET);
Tbyte syncbyte = *out.wr();
out.written(SIZE_RSPACKET);
++locktime;
if (locktime_out)
locktime_out->write(locktime);
// Reset timer if sync byte is correct
Tbyte expected = phase8 ? MPEG_SYNC : MPEG_SYNC_INV;
if (syncbyte == expected)
lock_timeleft = lock_timeout;
phase8 = (phase8 + 1) & 7;
--lock_timeleft;
if (!lock_timeleft)
{
if (sch->debug)
fprintf(stderr, "Unlocked\n");
synchronized = false;
next_sync_count = 0;
if (state_out)
state_out->write(0);
return;
}
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
private:
2018-02-22 16:52:49 -05:00
pipereader<Tbyte> in;
pipewriter<Tbyte> out;
2018-02-24 18:07:08 -05:00
deconvol_sync<Tbyte, 0> *deconv;
2018-02-22 16:52:49 -05:00
unsigned char polarity; // XOR mask, 0 or 0xff
int resync_phase;
int bitphase;
bool synchronized;
int next_sync_count;
int phase8; // Position in 8-packet cycle, -1 if not synchronized
unsigned long lock_timeleft;
unsigned long locktime;
pipewriter<int> *state_out;
pipewriter<unsigned long> *locktime_out;
bool report_state;
2018-02-24 18:07:08 -05:00
};
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
template<typename Tbyte>
struct rspacket
{
Tbyte data[SIZE_RSPACKET];
};
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// INTERLEAVER
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
struct interleaver: runnable
{
interleaver(scheduler *sch, pipebuf<rspacket<u8> > &_in, pipebuf<u8> &_out) :
runnable(sch, "interleaver"), in(_in), out(_out, SIZE_RSPACKET)
{
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
while (in.readable() >= 12 && out.writable() >= SIZE_RSPACKET)
{
rspacket<u8> *pin = in.rd();
u8 *pout = out.wr();
int delay = 0;
for (int i = 0; i < SIZE_RSPACKET;
++i, ++pout, delay = (delay + 1) % 12)
*pout = pin[11 - delay].data[i];
in.read(1);
out.written(SIZE_RSPACKET);
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
private:
pipereader<rspacket<u8> > in;
2018-02-22 16:52:49 -05:00
pipewriter<u8> out;
2018-02-24 18:07:08 -05:00
};
// interleaver
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// DEINTERLEAVER
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
template<typename Tbyte>
struct deinterleaver: runnable
{
2018-02-22 16:52:49 -05:00
deinterleaver(scheduler *sch, pipebuf<Tbyte> &_in,
2018-02-24 18:07:08 -05:00
pipebuf<rspacket<Tbyte> > &_out) :
runnable(sch, "deinterleaver"), in(_in), out(_out)
{
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
while (in.readable() >= 17 * 11 * 12 + SIZE_RSPACKET
&& out.writable() >= 1)
{
Tbyte *pin = in.rd() + 17 * 11 * 12, *pend = pin + SIZE_RSPACKET;
Tbyte *pout = out.wr()->data;
for (int delay = 17 * 11; pin < pend;
++pin, ++pout, delay = (delay - 17 + 17 * 12) % (17 * 12))
*pout = pin[-delay * 12];
in.read(SIZE_RSPACKET);
out.written(1);
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
private:
2018-02-22 16:52:49 -05:00
pipereader<Tbyte> in;
2018-02-24 18:07:08 -05:00
pipewriter<rspacket<Tbyte> > out;
};
// deinterleaver
static const int SIZE_TSPACKET = 188;
struct tspacket
{
u8 data[SIZE_TSPACKET];
};
// RS ENCODER
struct rs_encoder: runnable
{
rs_encoder(scheduler *sch, pipebuf<tspacket> &_in,
pipebuf<rspacket<u8> > &_out) :
runnable(sch, "RS encoder"), in(_in), out(_out)
{
}
void run()
{
while (in.readable() >= 1 && out.writable() >= 1)
{
u8 *pin = in.rd()->data;
u8 *pout = out.wr()->data;
// The first 188 bytes are the uncoded message P(X)
memcpy(pout, pin, SIZE_TSPACKET);
// Append 16 RS parity bytes R(X) = - (P(X)*X^16 mod G(X))
// so that G(X) divides the coded message S(X) = P(X)*X^16 - R(X).
rs.encode(pout);
in.read(1);
out.written(1);
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
private:
2018-02-22 16:52:49 -05:00
rs_engine rs;
pipereader<tspacket> in;
2018-02-24 18:07:08 -05:00
pipewriter<rspacket<u8> > out;
};
// rs_encoder
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// RS DECODER
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
template<typename Tbyte, int BYTE_ERASED>
struct rs_decoder: runnable
{
2018-02-22 16:52:49 -05:00
rs_engine rs;
2018-02-24 18:07:08 -05:00
rs_decoder(scheduler *sch, pipebuf<rspacket<Tbyte> > &_in,
pipebuf<tspacket> &_out, pipebuf<int> *_bitcount = NULL,
pipebuf<int> *_errcount = NULL) :
runnable(sch, "RS decoder"), in(_in), out(_out)
{
bitcount = _bitcount ? new pipewriter<int>(*_bitcount) : NULL;
errcount = _errcount ? new pipewriter<int>(*_errcount) : NULL;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
if (bitcount && bitcount->writable() < 1)
return;
if (errcount && errcount->writable() < 1)
return;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
int nbits = 0, nerrs = 0;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
while (in.readable() >= 1 && out.writable() >= 1)
{
Tbyte *pin = in.rd()->data;
u8 *pout = out.wr()->data;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
nbits += SIZE_RSPACKET * 8;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// The message is the first 188 bytes.
if (sizeof(Tbyte) == 1)
memcpy(pout, pin, SIZE_TSPACKET);
else
{
fail("rs_decoder::run", "Erasures not implemented");
return;
}
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
u8 synd[16];
bool corrupted = rs.syndromes(pin, synd);
2018-02-22 16:52:49 -05:00
#if 0
2018-02-24 18:07:08 -05:00
if ( ! corrupted )
{
// Test BM
fprintf(stderr, "Simulating errors\n");
pin[203] ^= 42;
pin[202] ^= 99;
corrupted = rs.syndromes(pin, synd);
}
2018-02-22 16:52:49 -05:00
#endif
2018-02-24 18:07:08 -05:00
if (!corrupted)
{
if (sch->debug)
fprintf(stderr, "_"); // Packet received without errors.
}
else
{
corrupted = rs.correct(synd, pout, pin, &nerrs);
if (sch->debug)
{
if (!corrupted)
fprintf(stderr, "."); // Errors were corrected.
else
fprintf(stderr, "!"); // Packet still corrupted.
}
}
in.read(1);
// Output corrupted packets (with a special mark)
// otherwise the derandomizer will lose synchronization.
if (corrupted)
pout[0] ^= MPEG_SYNC_CORRUPTED;
out.written(1);
}
if (nbits)
{
if (bitcount)
bitcount->write(nbits);
if (errcount)
errcount->write(nerrs);
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
private:
pipereader<rspacket<Tbyte> > in;
2018-02-22 16:52:49 -05:00
pipewriter<tspacket> out;
pipewriter<int> *bitcount, *errcount;
2018-02-24 18:07:08 -05:00
};
// rs_decoder
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// RANDOMIZER
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
struct randomizer: runnable
{
randomizer(scheduler *sch, pipebuf<tspacket> &_in, pipebuf<tspacket> &_out) :
runnable(sch, "derandomizer"), in(_in), out(_out)
{
precompute_pattern();
pos = pattern;
pattern_end = pattern + sizeof(pattern) / sizeof(pattern[0]);
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void precompute_pattern()
{
// EN 300 421, section 4.4.1 Transport multiplex adaptation
pattern[0] = 0xff; // Invert one in eight sync bytes
unsigned short st = 169; // 0b 000 000 010 101 001 (Fig 2 reversed)
2018-02-24 18:07:08 -05:00
for (int i = 1; i < 188 * 8; ++i)
{
u8 out = 0;
for (int n = 8; n--;)
{
int bit = ((st >> 13) ^ (st >> 14)) & 1; // Taps
out = (out << 1) | bit; // MSB first
st = (st << 1) | bit; // Feedback
}
pattern[i] = (i % 188) ? out : 0; // Inhibit on sync bytes
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
while (in.readable() >= 1 && out.writable() >= 1)
{
u8 *pin = in.rd()->data, *pend = pin + SIZE_TSPACKET;
u8 *pout = out.wr()->data;
if (pin[0] != MPEG_SYNC)
fprintf(stderr, "randomizer: bad MPEG sync %02x\n", pin[0]);
for (; pin < pend; ++pin, ++pout, ++pos)
*pout = *pin ^ *pos;
if (pos == pattern_end)
pos = pattern;
in.read(1);
out.written(1);
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
private:
u8 pattern[188 * 8], *pattern_end, *pos;
2018-02-22 16:52:49 -05:00
pipereader<tspacket> in;
pipewriter<tspacket> out;
2018-02-24 18:07:08 -05:00
};
// randomizer
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// DERANDOMIZER
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
struct derandomizer: runnable
{
derandomizer(scheduler *sch, pipebuf<tspacket> &_in,
pipebuf<tspacket> &_out) :
runnable(sch, "derandomizer"), in(_in), out(_out)
{
precompute_pattern();
pos = pattern;
pattern_end = pattern + sizeof(pattern) / sizeof(pattern[0]);
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void precompute_pattern()
{
// EN 300 421, section 4.4.1 Transport multiplex adaptation
pattern[0] = 0xff; // Restore the inverted sync byte
unsigned short st = 169; // 0b 000 000 010 101 001 (Fig 2 reversed)
2018-02-24 18:07:08 -05:00
for (int i = 1; i < 188 * 8; ++i)
{
u8 out = 0;
for (int n = 8; n--;)
{
int bit = ((st >> 13) ^ (st >> 14)) & 1; // Taps
out = (out << 1) | bit; // MSB first
st = (st << 1) | bit; // Feedback
}
pattern[i] = (i % 188) ? out : 0; // Inhibit on sync bytes
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
while (in.readable() >= 1 && out.writable() >= 1)
{
u8 *pin = in.rd()->data, *pend = pin + SIZE_TSPACKET;
u8 *pout = out.wr()->data;
if (pin[0] == MPEG_SYNC_INV
|| pin[0] == (MPEG_SYNC_INV ^ MPEG_SYNC_CORRUPTED))
{
if (pos != pattern)
{
if (sch->debug)
fprintf(stderr, "derandomizer: resynchronizing\n");
pos = pattern;
}
}
for (; pin < pend; ++pin, ++pout, ++pos)
*pout = *pin ^ *pos;
if (pos == pattern_end)
pos = pattern;
in.read(1);
u8 sync = out.wr()->data[0];
if (sync == MPEG_SYNC)
{
out.written(1);
}
else
{
if (sync != (MPEG_SYNC ^ MPEG_SYNC_CORRUPTED))
if (sch->debug)
fprintf(stderr, "(%02x)", sync);
out.wr()->data[1] |= 0x80; // Set the Transport Error Indicator bit
// We could output corrupted packets here, in case the
// MPEG decoder can use them somehow.
//out.written(1);
}
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
private:
u8 pattern[188 * 8], *pattern_end, *pos;
2018-02-22 16:52:49 -05:00
pipereader<tspacket> in;
pipewriter<tspacket> out;
2018-02-24 18:07:08 -05:00
};
// derandomizer
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// VITERBI DECODING
// Supports all code rates and constellations
// Simplified metric to support large constellations.
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
// This version implements puncturing by expanding the trellis.
// TBD Compare performance vs skipping updates in a 1/2 trellis.
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
struct viterbi_sync: runnable
{
2018-02-22 16:52:49 -05:00
typedef uint8_t TS, TCS, TUS;
typedef int32_t TBM; // Only 16 bits per IQ, but several IQ per Viterbi CS
typedef int32_t TPM;
2018-02-24 18:07:08 -05:00
typedef viterbi_dec_interface<TUS, TCS, TBM, TPM> dvb_dec_interface;
2018-02-22 16:52:49 -05:00
// 1/2: 6 bits of state, 1 bit in, 2 bits out
2018-02-24 18:07:08 -05:00
typedef bitpath<uint32_t, TUS, 1, 32> path_12;
typedef trellis<TS, 64, TUS, 2, 4> trellis_12;
typedef viterbi_dec<TS, 64, TUS, 2, TCS, 4, TBM, TPM, path_12> dvb_dec_12;
2018-02-22 16:52:49 -05:00
// 2/3: 6 bits of state, 2 bits in, 3 bits out
2018-02-24 18:07:08 -05:00
typedef bitpath<uint64_t, TUS, 3, 21> path_23;
typedef trellis<TS, 64, TUS, 4, 8> trellis_23;
typedef viterbi_dec<TS, 64, TUS, 4, TCS, 8, TBM, TPM, path_23> dvb_dec_23;
2018-02-22 16:52:49 -05:00
// 4/6: 6 bits of state, 4 bits in, 6 bits out
2018-02-24 18:07:08 -05:00
typedef bitpath<uint64_t, TUS, 4, 16> path_46;
typedef trellis<TS, 64, TUS, 16, 64> trellis_46;
typedef viterbi_dec<TS, 64, TUS, 16, TCS, 64, TBM, TPM, path_46> dvb_dec_46;
2018-02-22 16:52:49 -05:00
// 3/4: 6 bits of state, 3 bits in, 4 bits out
2018-02-24 18:07:08 -05:00
typedef bitpath<uint64_t, TUS, 3, 21> path_34;
typedef trellis<TS, 64, TUS, 8, 16> trellis_34;
typedef viterbi_dec<TS, 64, TUS, 8, TCS, 16, TBM, TPM, path_34> dvb_dec_34;
2018-02-22 16:52:49 -05:00
// 4/5: 6 bits of state, 4 bits in, 5 bits out (non-standard)
2018-02-24 18:07:08 -05:00
typedef bitpath<uint64_t, TUS, 4, 16> path_45;
typedef trellis<TS, 64, TUS, 16, 32> trellis_45;
typedef viterbi_dec<TS, 64, TUS, 16, TCS, 32, TBM, TPM, path_45> dvb_dec_45;
2018-02-22 16:52:49 -05:00
// 5/6: 6 bits of state, 5 bits in, 6 bits out
2018-02-24 18:07:08 -05:00
typedef bitpath<uint64_t, TUS, 5, 12> path_56;
typedef trellis<TS, 64, TUS, 32, 64> trellis_56;
typedef viterbi_dec<TS, 64, TUS, 32, TCS, 64, TBM, TPM, path_56> dvb_dec_56;
2018-02-22 16:52:49 -05:00
// QPSK 7/8: 6 bits of state, 7 bits in, 8 bits out
2018-02-24 18:07:08 -05:00
typedef bitpath<uint64_t, TUS, 7, 9> path_78;
typedef trellis<TS, 64, TUS, 128, 256> trellis_78;
typedef viterbi_dec<TS, 64, TUS, 128, TCS, 256, TBM, TPM, path_78> dvb_dec_78;
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
private:
2018-02-22 16:52:49 -05:00
pipereader<softsymbol> in;
pipewriter<unsigned char> out;
cstln_lut<256> *cstln;
fec_spec *fec;
int bits_per_symbol; // Bits per IQ symbol (not per coded symbol)
int nsyncs;
int nshifts;
2018-02-24 18:07:08 -05:00
struct sync
{
int shift;
dvb_dec_interface *dec;
TCS *map; // [nsymbols]
}*syncs; // [nsyncs]
IncrementalArray<TPM> m_totaldiscr;
2018-02-22 16:52:49 -05:00
int current_sync;
static const int chunk_size = 128;
int resync_phase;
2018-02-24 18:07:08 -05:00
public:
2018-02-22 16:52:49 -05:00
int resync_period;
2018-02-24 18:07:08 -05:00
viterbi_sync(scheduler *sch, pipebuf<softsymbol> &_in,
pipebuf<unsigned char> &_out, cstln_lut<256> *_cstln, code_rate cr) :
runnable(sch, "viterbi_sync"), in(_in), out(_out, chunk_size), cstln(
_cstln), current_sync(0), resync_phase(0), resync_period(32) // 1/32 = 9% synchronization overhead TBD
2018-02-22 16:52:49 -05:00
{
2018-02-24 18:07:08 -05:00
bits_per_symbol = log2i(cstln->nsymbols);
fec = &fec_specs[cr];
{ // Sanity check: FEC block size must be a multiple of label size.
int symbols_per_block = fec->bits_out / bits_per_symbol;
if (bits_per_symbol * symbols_per_block != fec->bits_out)
{
fail("viterbi_sync::viterbi_sync", "Code rate not suitable for this constellation");
return;
}
2018-02-24 18:07:08 -05:00
}
int nconj;
switch (cstln->nsymbols)
{
case 2:
nconj = 1;
break; // Conjugation is not relevant for BPSK
default:
nconj = 2;
break;
}
int nrotations;
switch (cstln->nsymbols)
{
case 2:
case 4:
// For BPSK and QPSK, 180° rotation is handled as
// polarity inversion in mpeg_sync.
nrotations = cstln->nrotations / 2;
break;
default:
nrotations = cstln->nrotations;
break;
}
nshifts = fec->bits_out / bits_per_symbol;
nsyncs = nconj * nrotations * nshifts;
// TBD Many HOM constellations are labelled in such a way
// that certain rot/conj combinations are equivalent to
// polarity inversion. We could reduce nsyncs.
syncs = new sync[nsyncs];
m_totaldiscr.allocate(nsyncs);
2018-02-24 18:07:08 -05:00
for (int s = 0; s < nsyncs; ++s)
{
// Bit pattern [shift|conj|rot]
int rot = s % nrotations;
int conj = (s / nrotations) % nconj;
int shift = s / nrotations / nconj;
syncs[s].shift = shift;
if (shift) // Reuse identical map
syncs[s].map = syncs[conj * nrotations + rot].map;
else
syncs[s].map = init_map(conj,
2 * M_PI * rot / cstln->nrotations);
2018-02-22 16:52:49 -05:00
#if 0
2018-02-24 18:07:08 -05:00
fprintf(stderr, "sync %3d: conj%d offs%d rot%d/%d map:",
s, conj, syncs[s].shift, rot, cstln->nrotations);
for ( int i=0; i<cstln->nsymbols; ++i )
fprintf(stderr, " %2d", syncs[s].map[i]);
fprintf(stderr, "\n");
2018-02-22 16:52:49 -05:00
#endif
2018-02-24 18:07:08 -05:00
}
if (cr == FEC12)
{
trellis_12 *trell = new trellis_12();
trell->init_convolutional(fec->polys);
for (int s = 0; s < nsyncs; ++s)
syncs[s].dec = new dvb_dec_12(trell);
}
else if (cr == FEC23)
{
trellis_23 *trell = new trellis_23();
trell->init_convolutional(fec->polys);
for (int s = 0; s < nsyncs; ++s)
syncs[s].dec = new dvb_dec_23(trell);
}
else if (cr == FEC46)
{
trellis_46 *trell = new trellis_46();
trell->init_convolutional(fec->polys);
for (int s = 0; s < nsyncs; ++s)
syncs[s].dec = new dvb_dec_46(trell);
}
else if (cr == FEC34)
{
trellis_34 *trell = new trellis_34();
trell->init_convolutional(fec->polys);
for (int s = 0; s < nsyncs; ++s)
syncs[s].dec = new dvb_dec_34(trell);
}
else if (cr == FEC45)
{
trellis_45 *trell = new trellis_45();
trell->init_convolutional(fec->polys);
for (int s = 0; s < nsyncs; ++s)
syncs[s].dec = new dvb_dec_45(trell);
}
else if (cr == FEC56)
{
trellis_56 *trell = new trellis_56();
trell->init_convolutional(fec->polys);
for (int s = 0; s < nsyncs; ++s)
syncs[s].dec = new dvb_dec_56(trell);
}
else if (cr == FEC78)
{
trellis_78 *trell = new trellis_78();
trell->init_convolutional(fec->polys);
for (int s = 0; s < nsyncs; ++s)
syncs[s].dec = new dvb_dec_78(trell);
}
else
{
fail("viterbi_sync::viterbi_sync", "CR not supported");
return;
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
TCS *init_map(bool conj, float angle)
{
// Each constellation has its own pattern for labels.
// Here we simply tabulate systematically.
TCS *map = new TCS[cstln->nsymbols];
float ca = cosf(angle), sa = sinf(angle);
for (int i = 0; i < cstln->nsymbols; ++i)
{
int8_t I = cstln->symbols[i].re;
int8_t Q = cstln->symbols[i].im;
if (conj)
Q = -Q;
int8_t RI = I * ca - Q * sa;
int8_t RQ = I * sa + Q * ca;
cstln_lut<256>::result *pr = cstln->lookup(RI, RQ);
map[i] = pr->ss.symbol;
}
return map;
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
inline TUS update_sync(int s, softsymbol *pin, TPM *discr)
{
// Read one FEC ouput block
pin += syncs[s].shift;
TCS cs = 0;
TBM cost = 0;
for (int i = 0; i < nshifts; ++i, ++pin)
{
cs = (cs << bits_per_symbol) | syncs[s].map[pin->symbol];
cost += pin->cost;
}
return syncs[s].dec->update(cs, cost, discr);
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
void run()
{
// Number of FEC blocks to fill the bitpath depth.
// Before that we cannot discriminate between synchronizers
int discr_delay = 64 / fec->bits_in;
// Process [chunk_size] FEC blocks at a time
while ((long) in.readable() >= nshifts * chunk_size + (nshifts - 1)
&& ((long) out.writable() * 8) >= fec->bits_in * chunk_size)
{
//TPM totaldiscr[nsyncs];
TPM *totaldiscr = m_totaldiscr.m_array;
2018-02-24 18:07:08 -05:00
for (int s = 0; s < nsyncs; ++s)
totaldiscr[s] = 0;
uint64_t outstream = 0;
int nout = 0;
softsymbol *pin = in.rd();
for (int blocknum = 0; blocknum < chunk_size; ++blocknum, pin +=
nshifts)
{
TPM discr;
TUS result = update_sync(current_sync, pin, &discr);
outstream = (outstream << fec->bits_in) | result;
nout += fec->bits_in;
if (blocknum >= discr_delay)
totaldiscr[current_sync] += discr;
if (!resync_phase)
{
// Every [resync_period] chunks, also run the other decoders.
for (int s = 0; s < nsyncs; ++s)
{
if (s == current_sync)
continue;
TPM discr;
(void) update_sync(s, pin, &discr);
if (blocknum >= discr_delay)
totaldiscr[s] += discr;
}
}
while (nout >= 8)
{
out.write(outstream >> (nout - 8));
nout -= 8;
}
} // chunk_size
in.read(chunk_size * nshifts);
if (nout)
{
fail("viterbi_sync::run", "overlapping out");
return;
}
2018-02-24 18:07:08 -05:00
if (!resync_phase)
{
// Switch to another decoder ?
int best = current_sync;
for (int s = 0; s < nsyncs; ++s)
if (totaldiscr[s] > totaldiscr[best])
best = s;
if (best != current_sync)
{
if (sch->debug)
fprintf(stderr, "{%d->%d}", current_sync, best);
current_sync = best;
}
}
if (++resync_phase >= resync_period)
resync_phase = 0;
}
2018-02-22 16:52:49 -05:00
}
2018-02-24 18:07:08 -05:00
};
// viterbi_sync
2018-02-22 16:52:49 -05:00
2018-02-24 18:07:08 -05:00
}// namespace
2018-02-22 16:52:49 -05:00
#endif // LEANSDR_DVB_H