/* -*- c++ -*- */
/*
 * Copyright 2007,2010,2011,2013 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.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "cvsd_encode_sb_impl.h"
#include <gnuradio/io_signature.h>
#include <limits.h>

namespace gr {
namespace vocoder {

cvsd_encode_sb::sptr cvsd_encode_sb::make(short min_step,
                                          short max_step,
                                          double step_decay,
                                          double accum_decay,
                                          int K,
                                          int J,
                                          short pos_accum_max,
                                          short neg_accum_max)
{
    return gnuradio::get_initial_sptr(new cvsd_encode_sb_impl(
        min_step, max_step, step_decay, accum_decay, K, J, pos_accum_max, neg_accum_max));
}

cvsd_encode_sb_impl::cvsd_encode_sb_impl(short min_step,
                                         short max_step,
                                         double step_decay,
                                         double accum_decay,
                                         int K,
                                         int J,
                                         short pos_accum_max,
                                         short neg_accum_max)
    : sync_decimator("vocoder_cvsd_encode_sb",
                     io_signature::make(1, 1, sizeof(short)),
                     io_signature::make(1, 1, sizeof(unsigned char)),
                     8),
      d_min_step(min_step),
      d_max_step(max_step),
      d_step_decay(step_decay),
      d_accum_decay(accum_decay),
      d_K(K),
      d_J(J),
      d_pos_accum_max(pos_accum_max),
      d_neg_accum_max(neg_accum_max),
      d_accum(0),
      d_loop_counter(1),
      d_runner(0),
      d_stepsize(min_step)
{
    if (d_K > 32)
        throw std::runtime_error("cvsd_decode_bs_impl: K must be <= 32\n");

    if (d_J > d_K)
        throw std::runtime_error("cvsd_decode_bs_impl: J must be <= K\n");
}

cvsd_encode_sb_impl::~cvsd_encode_sb_impl()
{
    // nothing else required in this example
}

unsigned char cvsd_encode_sb_impl::cvsd_bitwise_sum(unsigned int input)
{
    unsigned int temp = input;
    unsigned char bits = 0;

    while (temp) {
        temp = temp & (temp - 1);
        bits++;
    }
    return bits;
}

int cvsd_encode_sb_impl::cvsd_round(double input)
{
    double temp;
    temp = input + 0.5;
    temp = floor(temp);

    return (int)temp;
}

unsigned int cvsd_encode_sb_impl::cvsd_pow(short radix, short power)
{
    double d_radix = (double)radix;
    int i_power = (int)power;
    double output;

    output = pow(d_radix, i_power);
    return ((unsigned int)cvsd_round(output));
}

int cvsd_encode_sb_impl::work(int noutput_items,
                              gr_vector_const_void_star& input_items,
                              gr_vector_void_star& output_items)
{
    const short* in = (const short*)input_items[0];
    unsigned char* out = (unsigned char*)output_items[0];

    unsigned short i = 0;          // 2 bytes, 0 .. 65,535
    unsigned char output_bit = 0;  // 1 byte, 0 .. 255
    unsigned char output_byte = 0; // 1 bytes 0.255
    unsigned char bit_count = 0;   // 1 byte, 0 .. 255
    unsigned int mask = 0;         // 4 bytes, 0 .. 4,294,967,295

    // Loop through each input data point
    for (i = 0; i < noutput_items * 8; i++) {
        if ((int)in[i] >= d_accum) { // Note:  sign((data(n)-accum))
            output_bit = 1;
        } else {
            output_bit = 0;
        }

        // Update Accum (i.e. the reference value)
        if (output_bit) {
            d_accum = d_accum + d_stepsize;
            // printf("Adding %d to the accum; the result is: %d.\n", d_stepsize,
            // d_accum);
        } else {
            d_accum = d_accum - d_stepsize;
            // printf("Subtracting %d to the accum; the result is: %d.\n", d_stepsize,
            // d_accum);
        }

        // Multiply by Accum_Decay
        d_accum = (cvsd_round(d_accum * d_accum_decay));

        // Check for overflow
        if (d_accum >= ((int)d_pos_accum_max)) {
            d_accum = (int)d_pos_accum_max;
        } else if (d_accum <= ((int)d_neg_accum_max)) {
            d_accum = (int)d_neg_accum_max;
        }

        // Update runner with the last output bit
        // Update Step Size
        if (d_loop_counter >=
            d_J) { // Run this only if you have >= J bits in your shift register
            mask = (cvsd_pow(2, d_J) - 1);
            if ((cvsd_bitwise_sum(d_runner & mask) >= d_J) ||
                (cvsd_bitwise_sum((~d_runner) & mask) >= d_J)) {
                // Runs of 1s and 0s
                d_stepsize = std::min((short)(d_stepsize + d_min_step), d_max_step);
            } else {
                // No runs of 1s and 0s
                d_stepsize =
                    std::max((short)cvsd_round(d_stepsize * d_step_decay), d_min_step);
            }
        }

        // Runner is a shift-register; shift left, add on newest output bit
        d_runner = (d_runner << 1) | ((unsigned int)output_bit);

        // Update the output type; shift left, add on newest output bit
        // If you have put in 8 bits, output it as a byte
        output_byte = (output_byte << 1) | output_bit;
        bit_count++;

        if (d_loop_counter <= d_K) {
            d_loop_counter++;
        }

        // If you have put 8 bits, output and clear.
        if (bit_count == 8) {
            // Read in short from the file
            *(out++) = output_byte;

            // Reset the bit_count
            bit_count = 0;
            output_byte = 0;
        }
    } // While

    return noutput_items;
}

} /* namespace vocoder */
} /* namespace gr */
