mirror of
https://github.com/SatDump/SatDump
synced 2026-08-13 17:47:30 -04:00
78 lines
2.3 KiB
C++
78 lines
2.3 KiB
C++
#pragma once
|
|
|
|
#include "cc_common.h"
|
|
#include <volk/volk_alloc.hh>
|
|
#include <map>
|
|
|
|
namespace viterbi
|
|
{
|
|
typedef void (*conv_kernel)(unsigned char *Y,
|
|
unsigned char *X,
|
|
unsigned char *syms,
|
|
unsigned char *dec,
|
|
unsigned int framebits,
|
|
unsigned int excess,
|
|
unsigned char *Branchtab);
|
|
|
|
class CCDecoder
|
|
{
|
|
private:
|
|
// everything else...
|
|
void create_viterbi();
|
|
int init_viterbi(struct v *vp, int starting_state);
|
|
int init_viterbi_unbiased(struct v *vp);
|
|
int update_viterbi_blk(unsigned char *syms, int nbits);
|
|
int chainback_viterbi(unsigned char *data,
|
|
unsigned int nbits,
|
|
unsigned int endstate,
|
|
unsigned int tailsize);
|
|
int find_endstate();
|
|
|
|
volk::vector<unsigned char> d_branchtab;
|
|
unsigned char Partab[256];
|
|
|
|
int d_ADDSHIFT;
|
|
int d_SUBSHIFT;
|
|
conv_kernel d_kernel;
|
|
unsigned int d_max_frame_size;
|
|
unsigned int d_frame_size;
|
|
unsigned int d_k;
|
|
unsigned int d_rate;
|
|
std::vector<int> d_polys;
|
|
|
|
struct v d_vp;
|
|
volk::vector<unsigned char> d_managed_in;
|
|
int d_numstates;
|
|
int d_decision_t_size;
|
|
int *d_start_state;
|
|
int d_start_state_chaining;
|
|
int d_start_state_nonchaining;
|
|
int *d_end_state;
|
|
int d_end_state_chaining;
|
|
int d_end_state_nonchaining;
|
|
unsigned int d_veclen;
|
|
|
|
int parity(int x);
|
|
int parityb(unsigned char x);
|
|
void partab_init(void);
|
|
|
|
public:
|
|
CCDecoder(int frame_size,
|
|
int k,
|
|
int rate,
|
|
std::vector<int> polys,
|
|
int start_state = 0,
|
|
int end_state = -1);
|
|
~CCDecoder();
|
|
|
|
// Disable copy because of the raw pointers.
|
|
CCDecoder(const CCDecoder &) = delete;
|
|
CCDecoder &operator=(const CCDecoder &) = delete;
|
|
|
|
void work(uint8_t *inbuffer, uint8_t *outbuffer);
|
|
void work(uint8_t *inbuffer, uint8_t *outbuffer, int size);
|
|
|
|
bool set_frame_size(unsigned int frame_size);
|
|
double rate();
|
|
};
|
|
}
|