/*
 * Copyright (C) 2013 Bastian Bloessl <bloessl@ccs-labs.org>
 *
 * This program 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 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#include <ieee802-11/ofdm_sync_short.h>
#include <gnuradio/io_signature.h>
#include "utils.h"

#include <iostream>

using namespace gr::ieee802_11;

static const int MIN_GAP = 480;

class ofdm_sync_short_impl : public ofdm_sync_short {

public:
ofdm_sync_short_impl(double threshold, unsigned int max_samples,
		unsigned int min_plateau, bool log, bool debug) : block("ofdm_sync_short",
			gr::io_signature::make2(2, 2, sizeof(gr_complex), sizeof(float)),
			gr::io_signature::make(1, 1, sizeof(gr_complex))),
			d_log(log),
			d_debug(debug),
			d_state(SEARCH),
			d_plateau(0),
			MAX_SAMPLES(max_samples),
			MIN_PLATEAU(min_plateau),
			d_threshold(threshold) {

	set_tag_propagation_policy(block::TPP_DONT);
}

~ofdm_sync_short_impl(){
}

int general_work (int noutput_items, gr_vector_int& ninput_items,
		gr_vector_const_void_star& input_items,
		gr_vector_void_star& output_items) {

	const gr_complex *in = (const gr_complex*)input_items[0];
	const float *in2 = (const float*)input_items[1];
	gr_complex *out = (gr_complex*)output_items[0];

	int noutput = noutput_items;
	int ninput = std::min(ninput_items[0], ninput_items[1]);

	// dout << "SHORT noutput : " << noutput << " ninput: " << ninput_items[0] << std::endl;

	switch(d_state) {

	case SEARCH: {
		int i;

		for(i = 0; i < ninput; i++) {
			if(in2[i] > d_threshold) {
				if(d_plateau < MIN_PLATEAU) {
					d_plateau++;

				} else {
					d_state = COPY;
					d_copy_left = MAX_SAMPLES;
					d_plateau = 0;
					insert_tag(nitems_written(0));
					dout << "SHORT Frame!" << std::endl;
					break;
				}
			} else {
				d_plateau = 0;
			}
		}

		consume(0, i);
		consume(1, i);
		return 0;
	}

	case COPY: {

		int o = 0;
		while( (o < ninput) && (o < noutput) && d_copy_left) {
			if(in2[o] > d_threshold) {
				if(d_plateau < MIN_PLATEAU) {
					d_plateau++;

				// there's another frame
				} else if((MAX_SAMPLES - d_copy_left) > MIN_GAP) {
					d_copy_left = MAX_SAMPLES;
					d_plateau = 0;
					insert_tag(nitems_written(0) + o);
					dout << "SHORT Frame!" << std::endl;
					break;
				}

			} else {
				d_plateau = 0;
			}

			out[o] = in[o];
			o++;
			d_copy_left--;
		}

		if(!d_copy_left) {
			d_state = SEARCH;
		}

		dout << "SHORT copied " << o << std::endl;

		consume(0, o);
		consume(1, o);
		return o;
	}
	}

	throw std::runtime_error("sync short: unknown state");
	return 0;
}

void insert_tag(uint64_t item) {
	mylog(boost::format("frame start at %1%") % item);

	const pmt::pmt_t key = pmt::string_to_symbol("ofdm_start");
	const pmt::pmt_t value = pmt::PMT_T;
	const pmt::pmt_t srcid = pmt::string_to_symbol(name());
	add_item_tag(0, item, key, value, srcid);
}

private:
	enum {SEARCH, COPY} d_state;
	int d_copy_left;
	int d_plateau;
	const double d_threshold;
	const bool d_log;
	const bool d_debug;
	const unsigned int MAX_SAMPLES;
	const unsigned int MIN_PLATEAU;
};

ofdm_sync_short::sptr
ofdm_sync_short::make(double threshold, unsigned int max_samples,
		unsigned int min_plateau, bool log, bool debug) {
	return gnuradio::get_initial_sptr(new ofdm_sync_short_impl(threshold,
			max_samples, min_plateau, log, debug));
}