satdump/src-core/common/codings/viterbi/cc_decoder.h
2024-04-09 17:01:25 -04:00

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);
[[deprecated]] void work(uint8_t *inbuffer, uint8_t *outbuffer, int size);
bool set_frame_size(unsigned int frame_size);
double rate();
};
}