viterbi decoder: separate viterbi implementations
reorganizes viterbi decoder source so that each implementation resides in a separate file.
This commit is contained in:
Stephen Wolfe
@@ -0,0 +1,512 @@
|
||||
/*
|
||||
* Copyright 1995 Phil Karn, KA9Q
|
||||
* Copyright 2008 Free Software Foundation, Inc.
|
||||
* 2014 Added SSE2 implementation Bogdan Diaconescu
|
||||
*
|
||||
* 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.
|
||||
*/
|
||||
|
||||
/*
|
||||
* Viterbi decoder for K=7 rate=1/2 convolutional code
|
||||
* Some modifications from original Karn code by Matt Ettus
|
||||
* Major modifications by adding SSE2 code by Bogdan Diaconescu
|
||||
*/
|
||||
#include "viterbi_decoder.h"
|
||||
#include <cstring>
|
||||
#include <iostream>
|
||||
|
||||
/* The basic Viterbi decoder operation, called a "butterfly"
|
||||
* operation because of the way it looks on a trellis diagram. Each
|
||||
* butterfly involves an Add-Compare-Select (ACS) operation on the two nodes
|
||||
* where the 0 and 1 paths from the current node merge at the next step of
|
||||
* the trellis.
|
||||
*
|
||||
* The code polynomials are assumed to have 1's on both ends. Given a
|
||||
* function encode_state() that returns the two symbols for a given
|
||||
* encoder state in the low two bits, such a code will have the following
|
||||
* identities for even 'n' < 64:
|
||||
*
|
||||
* encode_state(n) = encode_state(n+65)
|
||||
* encode_state(n+1) = encode_state(n+64) = (3 ^ encode_state(n))
|
||||
*
|
||||
* Any convolutional code you would actually want to use will have
|
||||
* these properties, so these assumptions aren't too limiting.
|
||||
*
|
||||
* Doing this as a macro lets the compiler evaluate at compile time the
|
||||
* many expressions that depend on the loop index and encoder state and
|
||||
* emit them as immediate arguments.
|
||||
* This makes an enormous difference on register-starved machines such
|
||||
* as the Intel x86 family where evaluating these expressions at runtime
|
||||
* would spill over into memory.
|
||||
*/
|
||||
|
||||
#define BUTTERFLY(i,sym) { \
|
||||
int m0,m1,m2,m3; \
|
||||
/* ACS for 0 branch */ \
|
||||
m0 = state[i].metric + mets[sym]; /* 2*i */ \
|
||||
m1 = state[i+32].metric + mets[3 ^ sym]; /* 2*i + 64 */ \
|
||||
if(m0 > m1){ \
|
||||
next[2*i].metric = m0; \
|
||||
next[2*i].path = state[i].path << 1; \
|
||||
} else { \
|
||||
next[2*i].metric = m1; \
|
||||
next[2*i].path = (state[i+32].path << 1)|1; \
|
||||
} \
|
||||
/* ACS for 1 branch */ \
|
||||
m2 = state[i].metric + mets[3 ^ sym]; /* 2*i + 1 */ \
|
||||
m3 = state[i+32].metric + mets[sym]; /* 2*i + 65 */ \
|
||||
if(m2 > m3){ \
|
||||
next[2*i+1].metric = m2; \
|
||||
next[2*i+1].path = state[i].path << 1; \
|
||||
} else { \
|
||||
next[2*i+1].metric = m3; \
|
||||
next[2*i+1].path = (state[i+32].path << 1)|1; \
|
||||
} \
|
||||
}
|
||||
|
||||
using namespace gr::ieee802_11;
|
||||
|
||||
|
||||
viterbi_decoder::viterbi_decoder() :
|
||||
d_store_pos(0) {
|
||||
}
|
||||
|
||||
viterbi_decoder::~viterbi_decoder() {
|
||||
}
|
||||
|
||||
void
|
||||
viterbi_decoder::viterbi_butterfly2_generic(unsigned char *symbols,
|
||||
unsigned char *mm0, unsigned char *mm1, unsigned char *pp0,
|
||||
unsigned char *pp1)
|
||||
{
|
||||
int i, j, k;
|
||||
|
||||
unsigned char *metric0, *metric1;
|
||||
unsigned char *path0, *path1;
|
||||
|
||||
metric0 = mm0;
|
||||
path0 = pp0;
|
||||
metric1 = mm1;
|
||||
path1 = pp1;
|
||||
|
||||
// Operate on 4 symbols (2 bits) at a time
|
||||
|
||||
unsigned char m0[16], m1[16], m2[16], m3[16], decision0[16], decision1[16], survivor0[16], survivor1[16];
|
||||
unsigned char metsv[16], metsvm[16];
|
||||
unsigned char shift0[16], shift1[16];
|
||||
unsigned char tmp0[16], tmp1[16];
|
||||
unsigned char sym0v[16], sym1v[16];
|
||||
unsigned short simd_epi16;
|
||||
|
||||
for (j = 0; j < 16; j++) {
|
||||
sym0v[j] = symbols[0];
|
||||
sym1v[j] = symbols[1];
|
||||
}
|
||||
|
||||
for (i = 0; i < 2; i++) {
|
||||
if (symbols[0] == 2) {
|
||||
for (j = 0; j < 16; j++) {
|
||||
metsvm[j] = d_branchtab27_generic[1].c[(i*16) + j] ^ sym1v[j];
|
||||
metsv[j] = 1 - metsvm[j];
|
||||
}
|
||||
}
|
||||
else if (symbols[1] == 2) {
|
||||
for (j = 0; j < 16; j++) {
|
||||
metsvm[j] = d_branchtab27_generic[0].c[(i*16) + j] ^ sym0v[j];
|
||||
metsv[j] = 1 - metsvm[j];
|
||||
}
|
||||
}
|
||||
else {
|
||||
for (j = 0; j < 16; j++) {
|
||||
metsvm[j] = (d_branchtab27_generic[0].c[(i*16) + j] ^ sym0v[j]) + (d_branchtab27_generic[1].c[(i*16) + j] ^ sym1v[j]);
|
||||
metsv[j] = 2 - metsvm[j];
|
||||
}
|
||||
}
|
||||
|
||||
for (j = 0; j < 16; j++) {
|
||||
m0[j] = metric0[(i*16) + j] + metsv[j];
|
||||
m1[j] = metric0[((i+2)*16) + j] + metsvm[j];
|
||||
m2[j] = metric0[(i*16) + j] + metsvm[j];
|
||||
m3[j] = metric0[((i+2)*16) + j] + metsv[j];
|
||||
}
|
||||
|
||||
for (j = 0; j < 16; j++) {
|
||||
decision0[j] = ((m0[j] - m1[j]) > 0) ? 0xff : 0x0;
|
||||
decision1[j] = ((m2[j] - m3[j]) > 0) ? 0xff : 0x0;
|
||||
survivor0[j] = (decision0[j] & m0[j]) | ((~decision0[j]) & m1[j]);
|
||||
survivor1[j] = (decision1[j] & m2[j]) | ((~decision1[j]) & m3[j]);
|
||||
}
|
||||
|
||||
for (j = 0; j < 16; j += 2) {
|
||||
simd_epi16 = path0[(i*16) + j];
|
||||
simd_epi16 |= path0[(i*16) + (j+1)] << 8;
|
||||
simd_epi16 <<= 1;
|
||||
shift0[j] = simd_epi16;
|
||||
shift0[j+1] = simd_epi16 >> 8;
|
||||
|
||||
simd_epi16 = path0[((i+2)*16) + j];
|
||||
simd_epi16 |= path0[((i+2)*16) + (j+1)] << 8;
|
||||
simd_epi16 <<= 1;
|
||||
shift1[j] = simd_epi16;
|
||||
shift1[j+1] = simd_epi16 >> 8;
|
||||
}
|
||||
for (j = 0; j < 16; j++) {
|
||||
shift1[j] = shift1[j] + 1;
|
||||
}
|
||||
|
||||
for (j = 0, k = 0; j < 16; j += 2, k++) {
|
||||
metric1[(2*i*16) + j] = survivor0[k];
|
||||
metric1[(2*i*16) + (j+1)] = survivor1[k];
|
||||
}
|
||||
for (j = 0; j < 16; j++) {
|
||||
tmp0[j] = (decision0[j] & shift0[j]) | ((~decision0[j]) & shift1[j]);
|
||||
}
|
||||
|
||||
for (j = 0, k = 8; j < 16; j += 2, k++) {
|
||||
metric1[((2*i+1)*16) + j] = survivor0[k];
|
||||
metric1[((2*i+1)*16) + (j+1)] = survivor1[k];
|
||||
}
|
||||
for (j = 0; j < 16; j++) {
|
||||
tmp1[j] = (decision1[j] & shift0[j]) | ((~decision1[j]) & shift1[j]);
|
||||
}
|
||||
|
||||
for (j = 0, k = 0; j < 16; j += 2, k++) {
|
||||
path1[(2*i*16) + j] = tmp0[k];
|
||||
path1[(2*i*16) + (j+1)] = tmp1[k];
|
||||
}
|
||||
for (j = 0, k = 8; j < 16; j += 2, k++) {
|
||||
path1[((2*i+1)*16) + j] = tmp0[k];
|
||||
path1[((2*i+1)*16) + (j+1)] = tmp1[k];
|
||||
}
|
||||
}
|
||||
|
||||
metric0 = mm1;
|
||||
path0 = pp1;
|
||||
metric1 = mm0;
|
||||
path1 = pp0;
|
||||
|
||||
for (j = 0; j < 16; j++) {
|
||||
sym0v[j] = symbols[2];
|
||||
sym1v[j] = symbols[3];
|
||||
}
|
||||
|
||||
for (i = 0; i < 2; i++) {
|
||||
if (symbols[2] == 2) {
|
||||
for (j = 0; j < 16; j++) {
|
||||
metsvm[j] = d_branchtab27_generic[1].c[(i*16) + j] ^ sym1v[j];
|
||||
metsv[j] = 1 - metsvm[j];
|
||||
}
|
||||
}
|
||||
else if (symbols[3] == 2) {
|
||||
for (j = 0; j < 16; j++) {
|
||||
metsvm[j] = d_branchtab27_generic[0].c[(i*16) + j] ^ sym0v[j];
|
||||
metsv[j] = 1 - metsvm[j];
|
||||
}
|
||||
}
|
||||
else {
|
||||
for (j = 0; j < 16; j++) {
|
||||
metsvm[j] = (d_branchtab27_generic[0].c[(i*16) + j] ^ sym0v[j]) + (d_branchtab27_generic[1].c[(i*16) + j] ^ sym1v[j]);
|
||||
metsv[j] = 2 - metsvm[j];
|
||||
}
|
||||
}
|
||||
|
||||
for (j = 0; j < 16; j++) {
|
||||
m0[j] = metric0[(i*16) + j] + metsv[j];
|
||||
m1[j] = metric0[((i+2)*16) + j] + metsvm[j];
|
||||
m2[j] = metric0[(i*16) + j] + metsvm[j];
|
||||
m3[j] = metric0[((i+2)*16) + j] + metsv[j];
|
||||
}
|
||||
|
||||
for (j = 0; j < 16; j++) {
|
||||
decision0[j] = ((m0[j] - m1[j]) > 0) ? 0xff : 0x0;
|
||||
decision1[j] = ((m2[j] - m3[j]) > 0) ? 0xff : 0x0;
|
||||
survivor0[j] = (decision0[j] & m0[j]) | ((~decision0[j]) & m1[j]);
|
||||
survivor1[j] = (decision1[j] & m2[j]) | ((~decision1[j]) & m3[j]);
|
||||
}
|
||||
|
||||
for (j = 0; j < 16; j += 2) {
|
||||
simd_epi16 = path0[(i*16) + j];
|
||||
simd_epi16 |= path0[(i*16) + (j+1)] << 8;
|
||||
simd_epi16 <<= 1;
|
||||
shift0[j] = simd_epi16;
|
||||
shift0[j+1] = simd_epi16 >> 8;
|
||||
|
||||
simd_epi16 = path0[((i+2)*16) + j];
|
||||
simd_epi16 |= path0[((i+2)*16) + (j+1)] << 8;
|
||||
simd_epi16 <<= 1;
|
||||
shift1[j] = simd_epi16;
|
||||
shift1[j+1] = simd_epi16 >> 8;
|
||||
}
|
||||
for (j = 0; j < 16; j++) {
|
||||
shift1[j] = shift1[j] + 1;
|
||||
}
|
||||
|
||||
for (j = 0, k = 0; j < 16; j += 2, k++) {
|
||||
metric1[(2*i*16) + j] = survivor0[k];
|
||||
metric1[(2*i*16) + (j+1)] = survivor1[k];
|
||||
}
|
||||
for (j = 0; j < 16; j++) {
|
||||
tmp0[j] = (decision0[j] & shift0[j]) | ((~decision0[j]) & shift1[j]);
|
||||
}
|
||||
|
||||
for (j = 0, k = 8; j < 16; j += 2, k++) {
|
||||
metric1[((2*i+1)*16) + j] = survivor0[k];
|
||||
metric1[((2*i+1)*16) + (j+1)] = survivor1[k];
|
||||
}
|
||||
for (j = 0; j < 16; j++) {
|
||||
tmp1[j] = (decision1[j] & shift0[j]) | ((~decision1[j]) & shift1[j]);
|
||||
}
|
||||
|
||||
for (j = 0, k = 0; j < 16; j += 2, k++) {
|
||||
path1[(2*i*16) + j] = tmp0[k];
|
||||
path1[(2*i*16) + (j+1)] = tmp1[k];
|
||||
}
|
||||
for (j = 0, k = 8; j < 16; j += 2, k++) {
|
||||
path1[((2*i+1)*16) + j] = tmp0[k];
|
||||
path1[((2*i+1)*16) + (j+1)] = tmp1[k];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Find current best path
|
||||
unsigned char
|
||||
viterbi_decoder::viterbi_get_output_generic(unsigned char *mm0,
|
||||
unsigned char *pp0, int ntraceback, unsigned char *outbuf) {
|
||||
int i;
|
||||
int bestmetric, minmetric;
|
||||
int beststate = 0;
|
||||
int pos = 0;
|
||||
#ifndef IEEE80211_MSSE2
|
||||
int j;
|
||||
#endif
|
||||
|
||||
// circular buffer with the last ntraceback paths
|
||||
d_store_pos = (d_store_pos + 1) % ntraceback;
|
||||
|
||||
for (i = 0; i < 4; i++) {
|
||||
for (j = 0; j < 16; j++) {
|
||||
d_mmresult[(i*16) + j] = mm0[(i*16) + j];
|
||||
d_ppresult[d_store_pos][(i*16) + j] = pp0[(i*16) + j];
|
||||
}
|
||||
}
|
||||
|
||||
// Find out the best final state
|
||||
bestmetric = d_mmresult[beststate];
|
||||
minmetric = d_mmresult[beststate];
|
||||
|
||||
for (i = 1; i < 64; i++) {
|
||||
if (d_mmresult[i] > bestmetric) {
|
||||
bestmetric = d_mmresult[i];
|
||||
beststate = i;
|
||||
}
|
||||
if (d_mmresult[i] < minmetric) {
|
||||
minmetric = d_mmresult[i];
|
||||
}
|
||||
}
|
||||
|
||||
// Trace back
|
||||
for (i = 0, pos = d_store_pos; i < (ntraceback - 1); i++) {
|
||||
// Obtain the state from the output bits
|
||||
// by clocking in the output bits in reverse order.
|
||||
// The state has only 6 bits
|
||||
beststate = d_ppresult[pos][beststate] >> 2;
|
||||
pos = (pos - 1 + ntraceback) % ntraceback;
|
||||
}
|
||||
|
||||
// Store output byte
|
||||
*outbuf = d_ppresult[pos][beststate];
|
||||
|
||||
for (i = 0; i < 4; i++) {
|
||||
for (j = 0; j < 16; j++) {
|
||||
pp0[(i*16) + j] = 0;
|
||||
mm0[(i*16) + j] = mm0[(i*16) + j] - minmetric;
|
||||
}
|
||||
}
|
||||
|
||||
return bestmetric;
|
||||
}
|
||||
|
||||
uint8_t*
|
||||
viterbi_decoder::depuncture(uint8_t *in) {
|
||||
|
||||
int count;
|
||||
int n_cbps = d_ofdm->n_cbps;
|
||||
uint8_t *depunctured;
|
||||
|
||||
if (d_ntraceback == 5) {
|
||||
count = d_frame->n_sym * n_cbps;
|
||||
depunctured = in;
|
||||
|
||||
} else {
|
||||
depunctured = d_depunctured;
|
||||
count = 0;
|
||||
for(int i = 0; i < d_frame->n_sym; i++) {
|
||||
for(int k = 0; k < n_cbps; k++) {
|
||||
while (d_depuncture_pattern[count % (2 * d_k)] == 0) {
|
||||
depunctured[count] = 2;
|
||||
count++;
|
||||
}
|
||||
|
||||
// Insert received bits
|
||||
depunctured[count] = in[i * n_cbps + k];
|
||||
count++;
|
||||
|
||||
while (d_depuncture_pattern[count % (2 * d_k)] == 0) {
|
||||
depunctured[count] = 2;
|
||||
count++;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return depunctured;
|
||||
}
|
||||
|
||||
uint8_t*
|
||||
viterbi_decoder::decode(ofdm_param *ofdm, frame_param *frame, uint8_t *in) {
|
||||
|
||||
d_ofdm = ofdm;
|
||||
d_frame = frame;
|
||||
|
||||
reset();
|
||||
uint8_t *depunctured = depuncture(in);
|
||||
|
||||
int in_count = 0;
|
||||
int out_count = 0;
|
||||
int n_decoded = 0;
|
||||
|
||||
while(n_decoded < d_frame->n_data_bits) {
|
||||
|
||||
if ((in_count % 4) == 0) { //0 or 3
|
||||
viterbi_butterfly2_generic(&depunctured[in_count & 0xfffffffc], d_metric0_generic, d_metric1_generic,
|
||||
d_path0_generic, d_path1_generic);
|
||||
|
||||
if ((in_count > 0) && (in_count % 16) == 8) { // 8 or 11
|
||||
unsigned char c;
|
||||
|
||||
viterbi_get_output_generic(d_metric0_generic, d_path0_generic, d_ntraceback, &c);
|
||||
|
||||
if (out_count >= d_ntraceback) {
|
||||
for (int i= 0; i < 8; i++) {
|
||||
d_decoded[(out_count - d_ntraceback) * 8 + i] = (c >> (7 - i)) & 0x1;
|
||||
n_decoded++;
|
||||
}
|
||||
}
|
||||
out_count++;
|
||||
}
|
||||
}
|
||||
in_count++;
|
||||
}
|
||||
|
||||
return d_decoded;
|
||||
}
|
||||
|
||||
void
|
||||
viterbi_decoder::reset() {
|
||||
|
||||
viterbi_chunks_init_generic();
|
||||
|
||||
switch(d_ofdm->encoding) {
|
||||
case BPSK_1_2:
|
||||
case QPSK_1_2:
|
||||
case QAM16_1_2:
|
||||
d_ntraceback = 5;
|
||||
d_depuncture_pattern = PUNCTURE_1_2;
|
||||
d_k = 1;
|
||||
break;
|
||||
case QAM64_2_3:
|
||||
d_ntraceback = 9;
|
||||
d_depuncture_pattern = PUNCTURE_2_3;
|
||||
d_k = 2;
|
||||
break;
|
||||
case BPSK_3_4:
|
||||
case QPSK_3_4:
|
||||
case QAM16_3_4:
|
||||
case QAM64_3_4:
|
||||
d_ntraceback = 10;
|
||||
d_depuncture_pattern = PUNCTURE_3_4;
|
||||
d_k = 3;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// Initialize starting metrics to prefer 0 state
|
||||
void
|
||||
viterbi_decoder::viterbi_chunks_init_generic() {
|
||||
int i, j;
|
||||
|
||||
for (i = 0; i < 4; i++) {
|
||||
d_metric0_generic[i] = 0;
|
||||
d_path0_generic[i] = 0;
|
||||
}
|
||||
|
||||
int polys[2] = { 0x6d, 0x4f };
|
||||
for(i=0; i < 32; i++) {
|
||||
d_branchtab27_generic[0].c[i] = (polys[0] < 0) ^ PARTAB[(2*i) & abs(polys[0])] ? 1 : 0;
|
||||
d_branchtab27_generic[1].c[i] = (polys[1] < 0) ^ PARTAB[(2*i) & abs(polys[1])] ? 1 : 0;
|
||||
}
|
||||
|
||||
for (i = 0; i < 64; i++) {
|
||||
d_mmresult[i] = 0;
|
||||
for (j = 0; j < TRACEBACK_MAX; j++) {
|
||||
d_ppresult[j][i] = 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/* Parity lookup table */
|
||||
const unsigned char viterbi_decoder::PARTAB[256] = {
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
0, 1, 1, 0, 1, 0, 0, 1,
|
||||
1, 0, 0, 1, 0, 1, 1, 0,
|
||||
};
|
||||
|
||||
const unsigned char viterbi_decoder::PUNCTURE_1_2[2] = {1, 1};
|
||||
const unsigned char viterbi_decoder::PUNCTURE_2_3[4] = {1, 1, 1, 0};
|
||||
const unsigned char viterbi_decoder::PUNCTURE_3_4[6] = {1, 1, 1, 0, 0, 1};
|
||||
Reference in New Issue
Block a user