//
//  TapeUEF.cpp
//  Clock Signal
//
//  Created by Thomas Harte on 18/01/2016.
//  Copyright © 2016 Thomas Harte. All rights reserved.
//

#include "TapeUEF.hpp"
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>

using namespace Storage::Tape;

static float gzgetfloat(gzFile file)
{
	uint8_t bytes[4];
	bytes[0] = (uint8_t)gzgetc(file);
	bytes[1] = (uint8_t)gzgetc(file);
	bytes[2] = (uint8_t)gzgetc(file);
	bytes[3] = (uint8_t)gzgetc(file);

	/* assume a four byte array named Float exists, where Float[0]
	was the first byte read from the UEF, Float[1] the second, etc */

	/* decode mantissa */
	int mantissa;
	mantissa = bytes[0] | (bytes[1] << 8) | ((bytes[2]&0x7f)|0x80) << 16;

	float result = (float)mantissa;
	result = (float)ldexp(result, -23);

	/* decode exponent */
	int exponent;
	exponent = ((bytes[2]&0x80) >> 7) | (bytes[3]&0x7f) << 1;
	exponent -= 127;
	result = (float)ldexp(result, exponent);

	/* flip sign if necessary */
	if(bytes[3]&0x80)
		result = -result;

	return result;
}

static int gzget16(gzFile file)
{
	int result = gzgetc(file);
	result |= (gzgetc(file) << 8);
	return result;
}

static int gzget24(gzFile file)
{
	int result = gzget16(file);
	result |= (gzgetc(file) << 16);
	return result;
}

static int gzget32(gzFile file)
{
	int result = gzget16(file);
	result |= (gzget16(file) << 16);
	return result;
}

UEF::UEF(const char *file_name) :
	_time_base(1200),
	_is_at_end(false),
	_pulse_pointer(0)
{
	_file = gzopen(file_name, "rb");

	char identifier[10];
	int bytes_read = gzread(_file, identifier, 10);
	if(bytes_read < 10 || strcmp(identifier, "UEF File!"))
	{
		throw ErrorNotUEF;
	}

	int minor, major;
	minor = gzgetc(_file);
	major = gzgetc(_file);

	if(major > 0 || minor > 10 || major < 0 || minor < 0)
	{
		throw ErrorNotUEF;
	}

	parse_next_tape_chunk();
}

UEF::~UEF()
{
	gzclose(_file);
}

#pragma mark - Public methods

void UEF::reset()
{
	gzseek(_file, 12, SEEK_SET);
	_is_at_end = false;
	parse_next_tape_chunk();
}

bool UEF::is_at_end()
{
	return _is_at_end;
}

Storage::Tape::Tape::Pulse UEF::get_next_pulse()
{
	Pulse next_pulse;

	if(_is_at_end)
	{
		next_pulse.type = Pulse::Zero;
		next_pulse.length.length = _time_base * 4;
		next_pulse.length.clock_rate = _time_base * 4;
		return next_pulse;
	}

	next_pulse = _queued_pulses[_pulse_pointer];
	_pulse_pointer++;
	if(_pulse_pointer == _queued_pulses.size())
	{
		_queued_pulses.clear();
		_pulse_pointer = 0;
		parse_next_tape_chunk();
	}
	return next_pulse;
}

#pragma mark - Chunk navigator

void UEF::parse_next_tape_chunk()
{
	while(!_queued_pulses.size())
	{
		// read chunk details
		uint16_t chunk_id = (uint16_t)gzget16(_file);
		uint32_t chunk_length = (uint32_t)gzget32(_file);

		// figure out where the next chunk will start
		z_off_t start_of_next_chunk = gztell(_file) + chunk_length;

		if(gzeof(_file))
		{
			_is_at_end = true;
			return;
		}

		switch(chunk_id)
		{
			case 0x0100:	queue_implicit_bit_pattern(chunk_length);	break;
			case 0x0102:	queue_explicit_bit_pattern(chunk_length);	break;
			case 0x0112:	queue_integer_gap();						break;
			case 0x0116:	queue_floating_point_gap();					break;

			case 0x0110:	queue_carrier_tone();						break;
			case 0x0111:	queue_carrier_tone_with_dummy();			break;

			case 0x0114:	queue_security_cycles();					break;
			case 0x0104:	queue_defined_data(chunk_length);			break;

			case 0x113: // change of base rate
			{
				// TODO: something smarter than just converting this to an int
				float new_time_base = gzgetfloat(_file);
				_time_base = (unsigned int)roundf(new_time_base);
			}
			break;

			default:
				printf("!!! Skipping %04x\n", chunk_id);
			break;
		}

		gzseek(_file, start_of_next_chunk, SEEK_SET);
	}
}

#pragma mark - Chunk parsers

void UEF::queue_implicit_bit_pattern(uint32_t length)
{
	while(length--)
	{
		queue_implicit_byte((uint8_t)gzgetc(_file));
	}
}

void UEF::queue_explicit_bit_pattern(uint32_t length)
{
	size_t length_in_bits = (length << 3) - (size_t)gzgetc(_file);
	uint8_t current_byte = 0;
	for(size_t bit = 0; bit < length_in_bits; bit++)
	{
		if(!(bit&7)) current_byte = (uint8_t)gzgetc(_file);
		queue_bit(current_byte&1);
		current_byte >>= 1;
	}
}

void UEF::queue_integer_gap()
{
	Time duration;
	duration.length = (unsigned int)gzget16(_file);
	duration.clock_rate = _time_base;
	_queued_pulses.emplace_back(Pulse::Zero, duration);
}

void UEF::queue_floating_point_gap()
{
	float length = gzgetfloat(_file);
	Time duration;
	duration.length = (unsigned int)(length * 4000000);
	duration.clock_rate = 4000000;
	_queued_pulses.emplace_back(Pulse::Zero, duration);
}

void UEF::queue_carrier_tone()
{
	unsigned int number_of_cycles = (unsigned int)gzget16(_file);
	while(number_of_cycles--) queue_bit(1);
}

void UEF::queue_carrier_tone_with_dummy()
{
	unsigned int pre_cycles = (unsigned int)gzget16(_file);
	unsigned int post_cycles = (unsigned int)gzget16(_file);
	while(pre_cycles--) queue_bit(1);
	queue_implicit_byte(0xaa);
	while(post_cycles--) queue_bit(1);
}

void UEF::queue_security_cycles()
{
	int number_of_cycles = gzget24(_file);
	bool first_is_pulse = gzgetc(_file) == 'P';
	bool last_is_pulse = gzgetc(_file) == 'P';

	uint8_t current_byte = 0;
	for(int cycle = 0; cycle < number_of_cycles; cycle++)
	{
		if(!(cycle&7)) current_byte = (uint8_t)gzgetc(_file);
		int bit = (current_byte >> 7);
		current_byte <<= 1;

		Time duration;
		duration.length = bit ? 1 : 2;
		duration.clock_rate = _time_base * 4;

		if(!cycle && first_is_pulse)
		{
			_queued_pulses.emplace_back(Pulse::High, duration);
		}
		else if(cycle == number_of_cycles-1 && last_is_pulse)
		{
			_queued_pulses.emplace_back(Pulse::Low, duration);
		}
		else
		{
			_queued_pulses.emplace_back(Pulse::Low, duration);
			_queued_pulses.emplace_back(Pulse::High, duration);
		}
	}
}

void UEF::queue_defined_data(uint32_t length)
{
	if(length < 3) return;

	int bits_per_packet = gzgetc(_file);
	char parity_type = (char)gzgetc(_file);
	int number_of_stop_bits = gzgetc(_file);

	bool has_extra_stop_wave = (number_of_stop_bits < 0);
	number_of_stop_bits = abs(number_of_stop_bits);

	length -= 3;
	while(length--)
	{
		uint8_t byte = (uint8_t)gzgetc(_file);

		uint8_t parity_value = byte;
		parity_value ^= (parity_value >> 4);
		parity_value ^= (parity_value >> 2);
		parity_value ^= (parity_value >> 1);

		queue_bit(0);
		int c = bits_per_packet;
		while(c--)
		{
			queue_bit(byte&1);
			byte >>= 1;
		}

		switch(parity_type)
		{
			default: break;
			case 'E': queue_bit(parity_value&1);			break;
			case 'O': queue_bit((parity_value&1) ^ 1);	break;
		}

		int stop_bits = number_of_stop_bits;
		while(stop_bits--) queue_bit(1);
		if(has_extra_stop_wave)
		{
			Time duration;
			duration.length = 1;
			duration.clock_rate = _time_base * 4;
			_queued_pulses.emplace_back(Pulse::Low, duration);
			_queued_pulses.emplace_back(Pulse::High, duration);
		}
	}
}

#pragma mark - Queuing helpers

void UEF::queue_implicit_byte(uint8_t byte)
{
	queue_bit(0);
	int c = 8;
	while(c--)
	{
		queue_bit(byte&1);
		byte >>= 1;
	}
	queue_bit(1);
}

void UEF::queue_bit(int bit)
{
	// TODO: allow for 300-baud encoding
	if(bit)
	{
		// Encode a 1 as four high-frequency waves
		Time duration;
		duration.length = 1;
		duration.clock_rate = _time_base * 4;

		_queued_pulses.emplace_back(Pulse::Low, duration);
		_queued_pulses.emplace_back(Pulse::High, duration);
		_queued_pulses.emplace_back(Pulse::Low, duration);
		_queued_pulses.emplace_back(Pulse::High, duration);
	}
	else
	{
		// Encode a 0 as two low-frequency waves
		Time duration;
		duration.length = 2;
		duration.clock_rate = _time_base * 4;

		_queued_pulses.emplace_back(Pulse::Low, duration);
		_queued_pulses.emplace_back(Pulse::High, duration);
	}
}