C++程序  |  154行  |  3.34 KB

/* Switch to K=9 r=1/3 Viterbi decoder with optional Intel or PowerPC SIMD
 * Copyright Aug 2006, Phil Karn, KA9Q
 */
#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include "fec.h"

/* Create a new instance of a Viterbi decoder */
void *create_viterbi39(int len){
  find_cpu_mode();

  switch(Cpu_mode){
  case PORT:
  default:
    return create_viterbi39_port(len);
#ifdef __VEC__
  case ALTIVEC:
    return create_viterbi39_av(len);
#endif
#ifdef __i386__
  case MMX:
    return create_viterbi39_mmx(len);
  case SSE:
    return create_viterbi39_sse(len);
  case SSE2:
    return create_viterbi39_sse2(len);
#endif
  }
}

void set_viterbi39_polynomial(int polys[3]){
  switch(Cpu_mode){
  case PORT:
  default:
    set_viterbi39_polynomial_port(polys);
    break;
#ifdef __VEC__
  case ALTIVEC:
    set_viterbi39_polynomial_av(polys);
    break;
#endif
#ifdef __i386__
  case MMX:
    set_viterbi39_polynomial_mmx(polys);
    break;
  case SSE:
    set_viterbi39_polynomial_sse(polys);
    break;
  case SSE2:
    set_viterbi39_polynomial_sse2(polys);
    break;
#endif
  }
}


/* Initialize Viterbi decoder for start of new frame */
int init_viterbi39(void *p,int starting_state){
    switch(Cpu_mode){
    case PORT:
    default:
      return init_viterbi39_port(p,starting_state);
#ifdef __VEC__
    case ALTIVEC:
      return init_viterbi39_av(p,starting_state);
#endif
#ifdef __i386__
    case MMX:
      return init_viterbi39_mmx(p,starting_state);
    case SSE:
      return init_viterbi39_sse(p,starting_state);
    case SSE2:
      return init_viterbi39_sse2(p,starting_state);
#endif
    }
}

/* Viterbi chainback */
int chainback_viterbi39(
      void *p,
      unsigned char *data, /* Decoded output data */
      unsigned int nbits, /* Number of data bits */
      unsigned int endstate){ /* Terminal encoder state */

    switch(Cpu_mode){
    case PORT:
    default:
      return chainback_viterbi39_port(p,data,nbits,endstate);
#ifdef __VEC__
    case ALTIVEC:
      return chainback_viterbi39_av(p,data,nbits,endstate);
#endif
#ifdef __i386__
    case MMX:
      return chainback_viterbi39_mmx(p,data,nbits,endstate);
    case SSE:
      return chainback_viterbi39_sse(p,data,nbits,endstate);
    case SSE2:
      return chainback_viterbi39_sse2(p,data,nbits,endstate);
#endif
    }
}

/* Delete instance of a Viterbi decoder */
void delete_viterbi39(void *p){
    switch(Cpu_mode){
    case PORT:
    default:
      delete_viterbi39_port(p);
      break;
#ifdef __VEC__
    case ALTIVEC:
      delete_viterbi39_av(p);
      break;
#endif
#ifdef __i386__
    case MMX:
      delete_viterbi39_mmx(p);
      break;
    case SSE:
      delete_viterbi39_sse(p);
      break;
    case SSE2:
      delete_viterbi39_sse2(p);
      break;
#endif
    }
}

/* Update decoder with a block of demodulated symbols
 * Note that nbits is the number of decoded data bits, not the number
 * of symbols!
 */
int update_viterbi39_blk(void *p,unsigned char syms[],int nbits){
    switch(Cpu_mode){
    case PORT:
    default:
      return update_viterbi39_blk_port(p,syms,nbits);
#ifdef __VEC__
    case ALTIVEC:
      return update_viterbi39_blk_av(p,syms,nbits);
#endif
#ifdef __i386__
    case MMX:
      return update_viterbi39_blk_mmx(p,syms,nbits);
    case SSE:
      return update_viterbi39_blk_sse(p,syms,nbits);
    case SSE2:
      return update_viterbi39_blk_sse2(p,syms,nbits);
#endif
    }
}