viterbi.h

Go to the documentation of this file.

#ifndef VITERBI_H
#define VITERBI_H

#pragma once

#include <vector>
#include <stdlib.h>
#include <math.h>
#include <memory.h>

#include <pmmintrin.h>
#include <emmintrin.h>
#include <xmmintrin.h>
#include <mmintrin.h>

#define K 7
#define RATE 2
#define POLYS { 121, 91 }
#define NUMSTATES 64
#define DECISIONTYPE unsigned char
#define DECISIONTYPE_BITSIZE 8
#define COMPUTETYPE unsigned char

namespace fun
{
    //decision_t is a BIT vector

    typedef union {
        DECISIONTYPE   t[NUMSTATES/DECISIONTYPE_BITSIZE];
        unsigned int   w[NUMSTATES/32];
        unsigned short s[NUMSTATES/16];
        unsigned char  c[NUMSTATES/8];
    } decision_t __attribute__ ((aligned (16)));

    typedef union {
        COMPUTETYPE t[NUMSTATES];
    } metric_t __attribute__ ((aligned (16)));

    struct v {
        __attribute__ ((aligned (16))) metric_t metrics1; /* path metric buffer 1 */
        __attribute__ ((aligned (16))) metric_t metrics2; /* path metric buffer 2 */
        metric_t *old_metrics,*new_metrics; /* Pointers to path metrics, swapped on every bit */
        decision_t *decisions;   /* decisions */
        void (*update_blk)(struct v *vp, const COMPUTETYPE *syms, int nbits);
};

    class viterbi
    {
    private:

        COMPUTETYPE Branchtab[NUMSTATES/2*RATE] __attribute__ ((aligned (16)));

        void viterbi_chainback(struct v *vp,
              unsigned char *data, /* Decoded output data */
              unsigned int nbits, /* Number of data bits */
              unsigned int endstate) ;

        void FULL_SPIRAL(int nbits, unsigned char *Y, unsigned char *X, const unsigned char *syms, unsigned char *dec, unsigned char *Branchtab);

        struct v *viterbi_alloc(int len);

        void viterbi_free(struct v *vp);

        void viterbi_init(struct v *vp, int starting_state);

        void viterbi_decode(struct v *vp, const COMPUTETYPE *symbols, unsigned char *data, int nbits);

        void viterbi_update_blk_SPIRAL(struct v *vp, const COMPUTETYPE *syms, int nbits);
        //void viterbi_spiral(struct v *vp);

    public:

        void conv_decode(unsigned char * symbols, unsigned char * data, int data_bits);

        void conv_encode(unsigned char * data, unsigned char * symbols, int data_bits);
    };

}


#endif // VITERBI_H