ccsds_viterbi.c

/* $Header: /home/karn/ace_rcs/RCS/ccsds_viterbi.c,v 1.3 2000/01/04 07:52:57 karn Exp $ */
#include <stdio.h>
#include <math.h>
#include <memory.h>
#include "viterbi27.h"
/* r=1/2 k=7 Viterbi decoder
 * CCSDS version -- symbols are sent in reversed order relative to the original
 * Viterbi decoder, with the second transmitted symbol inverted to guarantee
 * a channel symbol transition density
 */
int
viterbi27(
long *metric,    	/* Final path metric (returned value) */
unsigned char *data,	/* Decoded output data */
unsigned char *symbols,	/* Raw deinterleaved input symbols */
unsigned int nbits,	/* Number of output bits; 2*(nbits+6) input symbols will be read */
int mettab[2][256],	/* Metric table, [sent sym][rx symbol] */
int startstate,         /* Encoder starting state */
int endstate            /* Encoder ending state */
){
  int bitcnt = -6; /* K-1 */
  unsigned long *pp;
  register unsigned long dec;
  int i,mets[4];
  unsigned long paths[(nbits+6)*2];
  long cmetric[64],nmetric[64];
  
  startstate &= 63;
  endstate &= 63;
  
  /* Starting encoder state is the known ending of the sync word */
  for(i=0;i<64;i++)
    cmetric[i] = -9999999;
  cmetric[startstate] = 0;
   
  pp = paths;
  for(;;){
    /* Read input symbol pair and compute branch metrics */
    mets[0] = mettab[0][symbols[0]] + mettab[1][symbols[1]];
    mets[1] = mettab[1][symbols[0]] + mettab[1][symbols[1]];
    mets[3] = mettab[1][symbols[0]] + mettab[0][symbols[1]];
    mets[2] = mettab[0][symbols[0]] + mettab[0][symbols[1]];
    symbols += 2;
    
    /* These macro calls were generated by genbut.c
     * and rearranged by hand for speed
     */
    dec = 0;
    BUTTERFLY(0,0);
    BUTTERFLY(6,0);
    BUTTERFLY(8,0);
    BUTTERFLY(14,0);
    BUTTERFLY(2,3);
    BUTTERFLY(4,3);
    BUTTERFLY(10,3);
    BUTTERFLY(12,3);
    BUTTERFLY(1,1);
    BUTTERFLY(7,1);
    BUTTERFLY(9,1);
    BUTTERFLY(15,1);
    BUTTERFLY(3,2);
    BUTTERFLY(5,2);
    BUTTERFLY(11,2);
    BUTTERFLY(13,2);
    *pp++ = dec;
    dec = 0;
    BUTTERFLY(19,0);
    BUTTERFLY(21,0);
    BUTTERFLY(27,0);
    BUTTERFLY(29,0);
    BUTTERFLY(17,3);
    BUTTERFLY(23,3);
    BUTTERFLY(25,3);
    BUTTERFLY(31,3);
    BUTTERFLY(18,1);
    BUTTERFLY(20,1);
    BUTTERFLY(26,1);
    BUTTERFLY(28,1);
    BUTTERFLY(16,2);
    BUTTERFLY(22,2);
    BUTTERFLY(24,2);
    BUTTERFLY(30,2);
    *pp++ = dec;
    
    if(++bitcnt == nbits){
      /* Terminal encoder state is start of next sync vector */
      *metric = nmetric[endstate];
      break;
    }
    /* Read input symbol pair and compute branch metrics */
    mets[0] = mettab[0][symbols[0]] + mettab[1][symbols[1]];
    mets[1] = mettab[1][symbols[0]] + mettab[1][symbols[1]];
    mets[3] = mettab[1][symbols[0]] + mettab[0][symbols[1]];
    mets[2] = mettab[0][symbols[0]] + mettab[0][symbols[1]];
    symbols += 2;
    
    /* These macro calls were generated by genbut.c
     * and rearranged by hand for speed
     */
    dec = 0;
    BUTTERFLY2(0,0);
    BUTTERFLY2(6,0);
    BUTTERFLY2(8,0);
    BUTTERFLY2(14,0);
    BUTTERFLY2(2,3);
    BUTTERFLY2(4,3);
    BUTTERFLY2(10,3);
    BUTTERFLY2(12,3);
    BUTTERFLY2(1,1);
    BUTTERFLY2(7,1);
    BUTTERFLY2(9,1);
    BUTTERFLY2(15,1);
    BUTTERFLY2(3,2);
    BUTTERFLY2(5,2);
    BUTTERFLY2(11,2);
    BUTTERFLY2(13,2);
    *pp++ = dec;
    dec = 0;
    BUTTERFLY2(19,0);
    BUTTERFLY2(21,0);
    BUTTERFLY2(27,0);
    BUTTERFLY2(29,0);
    BUTTERFLY2(17,3);
    BUTTERFLY2(23,3);
    BUTTERFLY2(25,3);
    BUTTERFLY2(31,3);
    BUTTERFLY2(18,1);
    BUTTERFLY2(20,1);
    BUTTERFLY2(26,1);
    BUTTERFLY2(28,1);
    BUTTERFLY2(16,2);
    BUTTERFLY2(22,2);
    BUTTERFLY2(24,2);
    BUTTERFLY2(30,2);
    *pp++ = dec;
    
    if(++bitcnt == nbits){
      /* Terminal encoder state is start of next sync vector */
      *metric = cmetric[endstate];
      break;
    }
  }
  pp -= 2;
  /* Chain back from terminal state to produce decoded data */
  if(data == NULL)
    return 0;/* Discard output */
  memset(data,0,(nbits+7)/8); /* round up in case nbits % 8 != 0 */
  for(i=nbits-1;i >= 0;i--){
    if(pp[endstate >> 5] & (1 << (endstate & 31))){
      endstate |= 64;	/* 2^(K-1) */
      data[i>>3] |= 0x80 >> (i&7);
    }
    endstate >>= 1;
    pp -= 2;
  }
  return 0;
}