/* -*- c++ -*- */
/*
 * Copyright 2013-2014 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * GNU Radio is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3, or (at your option)
 * any later version.
 *
 * GNU Radio is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GNU Radio; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#ifndef INCLUDED_FEC_CC_DECODER_IMPL_H
#define INCLUDED_FEC_CC_DECODER_IMPL_H

#include <gnuradio/fec/cc_decoder.h>
#include <map>
#include <string>

namespace gr {
namespace fec {
namespace code {

class FEC_API cc_decoder_impl : public cc_decoder
{
private:
    // plug into the generic fec api
    int get_output_size();
    int get_input_size();
    int get_history();
    float get_shift();
    int get_input_item_size();
    const char* get_input_conversion();
    // const char* get_output_conversion();

    // 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();

    unsigned char* 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;
    cc_mode_t d_mode;
    int d_padding;

    struct v* d_vp;
    unsigned char* d_managed_in;
    unsigned int d_managed_in_size;
    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:
    cc_decoder_impl(int frame_size,
                    int k,
                    int rate,
                    std::vector<int> polys,
                    int start_state = 0,
                    int end_state = -1,
                    cc_mode_t mode = CC_STREAMING,
                    bool padded = false);
    ~cc_decoder_impl();

    void generic_work(void* inbuffer, void* outbuffer);
    bool set_frame_size(unsigned int frame_size);
    double rate();
};

} /* namespace code */
} /* namespace fec */
} /* namespace gr */

#endif /* INCLUDED_FEC_CC_DECODER_IMPL_H */
