/* Copyright (C) 2011 by Jacob Alexander
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

// ----- Includes -----

// AVR Includes
#include <util/delay.h>
#include <avr/interrupt.h>

// Project Includes
#include <led.h>
#include <print.h>
#include <scan_loop.h>
#include <usb_com.h>

// Keymaps
#include <keymap.h>
#include <usb_keys.h>

// Local Includes
#include "macro.h"



// ----- Variables -----

// Keeps track of the sequence used to reflash the teensy in software
static uint8_t Bootloader_ConditionSequence[] = {1,16,6,11};
       uint8_t Bootloader_ConditionState      = 0;
       uint8_t Bootloader_NextPositionReady   = 1;
       uint8_t Bootloader_KeyDetected         = 0;



// ----- Functions -----

inline void macro_finishedWithBuffer(void)
{
	/* BudKeypad
	// Boot loader sequence state handler
	switch ( KeyIndex_BufferUsed )
	{
	// The next bootloader key can now be pressed, if there were no keys processed
	case 0:
		Bootloader_NextPositionReady = 1;
		break;
	// If keys were detected, and it wasn't in the sequence (or there was multiple keys detected), start bootloader sequence over
	// This case purposely falls through
	case 1:
		if ( Bootloader_KeyDetected )
			break;
	default:
		Bootloader_ConditionState = 0;
		break;
	}

	Bootloader_KeyDetected = 0;
	*/
}

void jumpToBootloader(void)
{
	cli();
	// disable watchdog, if enabled
	// disable all peripherals
	UDCON = 1;
	USBCON = (1<<FRZCLK);  // disable USB
	UCSR1B = 0;
	_delay_ms(5);

#if defined(__AVR_AT90USB162__)                // Teensy 1.0
	EIMSK = 0; PCICR = 0; SPCR = 0; ACSR = 0; EECR = 0;
	TIMSK0 = 0; TIMSK1 = 0; UCSR1B = 0;
	DDRB = 0; DDRC = 0; DDRD = 0;
	PORTB = 0; PORTC = 0; PORTD = 0;
	asm volatile("jmp 0x3E00");
#elif defined(__AVR_ATmega32U4__)              // Teensy 2.0
	EIMSK = 0; PCICR = 0; SPCR = 0; ACSR = 0; EECR = 0; ADCSRA = 0;
	TIMSK0 = 0; TIMSK1 = 0; TIMSK3 = 0; TIMSK4 = 0; UCSR1B = 0; TWCR = 0;
	DDRB = 0; DDRC = 0; DDRD = 0; DDRE = 0; DDRF = 0; TWCR = 0;
	PORTB = 0; PORTC = 0; PORTD = 0; PORTE = 0; PORTF = 0;
	asm volatile("jmp 0x7E00");
#elif defined(__AVR_AT90USB646__)              // Teensy++ 1.0
	EIMSK = 0; PCICR = 0; SPCR = 0; ACSR = 0; EECR = 0; ADCSRA = 0;
	TIMSK0 = 0; TIMSK1 = 0; TIMSK2 = 0; TIMSK3 = 0; UCSR1B = 0; TWCR = 0;
	DDRA = 0; DDRB = 0; DDRC = 0; DDRD = 0; DDRE = 0; DDRF = 0;
	PORTA = 0; PORTB = 0; PORTC = 0; PORTD = 0; PORTE = 0; PORTF = 0;
	asm volatile("jmp 0xFC00");
#elif defined(__AVR_AT90USB1286__)             // Teensy++ 2.0
	EIMSK = 0; PCICR = 0; SPCR = 0; ACSR = 0; EECR = 0; ADCSRA = 0;
	TIMSK0 = 0; TIMSK1 = 0; TIMSK2 = 0; TIMSK3 = 0; UCSR1B = 0; TWCR = 0;
	DDRA = 0; DDRB = 0; DDRC = 0; DDRD = 0; DDRE = 0; DDRF = 0;
	PORTA = 0; PORTB = 0; PORTC = 0; PORTD = 0; PORTE = 0; PORTF = 0;
	asm volatile("jmp 0x1FC00");
#endif
}

// Given a sampling array, and the current number of detected keypress
// Add as many keypresses from the sampling array to the USB key send array as possible.
/*
inline void keyPressDetection( uint8_t *keys, uint8_t numberOfKeys, uint8_t *modifiers, uint8_t numberOfModifiers, uint8_t *map )
{
	uint8_t Bootloader_KeyDetected = 0;
	uint8_t processed_keys         = 0;

	// Parse the detection array starting from 1 (all keys are purposefully mapped from 1 -> total as per typical PCB labels)
	for ( uint8_t key = 0; key < numberOfKeys + 1; key++ )
	{
		if ( keys[key] & (1 << 7) )
		{
			processed_keys++;

			// Display the detected scancode
			char tmpStr[4];
			int8ToStr( key, tmpStr );
			dPrintStrs( tmpStr, " " );

			// Is this a bootloader sequence key?
			if ( !Bootloader_KeyDetected
			   && Bootloader_NextPositionReady
			   && key == Bootloader_ConditionSequence[Bootloader_ConditionState] )
			{
				Bootloader_KeyDetected = 1;
				Bootloader_NextPositionReady = 0;
				Bootloader_ConditionState++;
			}
			else if ( Bootloader_ConditionState > 0 && key == Bootloader_ConditionSequence[Bootloader_ConditionState - 1] )
			{
				Bootloader_KeyDetected = 1;
			}

			// Determine if the key is a modifier
			uint8_t modFound = 0;
			for ( uint8_t mod = 0; mod < numberOfModifiers; mod++ ) {
				// Modifier found
				if ( modifiers[mod] == key ) {
					USBKeys_Modifiers |= map[key];
					modFound = 1;
					break;
				}
			}

			// Modifier, already done this loop
			if ( modFound )
				continue;

			// Too many keys
			if ( USBKeys_Sent >= USBKeys_MaxSize )
			{
				info_print("USB Key limit reached");
				errorLED( 1 );
				break;
			}

			// Allow ignoring keys with 0's
			if ( map[key] != 0 )
				USBKeys_Array[USBKeys_Sent++] = map[key];
		}
	}

	// Boot loader sequence state handler
	switch ( processed_keys )
	{
	// The next bootloader key can now be pressed, if there were no keys processed
	case 0:
		Bootloader_NextPositionReady = 1;
		break;
	// If keys were detected, and it wasn't in the sequence (or there was multiple keys detected), start bootloader sequence over
	// This case purposely falls through
	case 1:
		if ( Bootloader_KeyDetected )
			break;
	default:
		Bootloader_ConditionState = 0;
		break;
	}

	// Add debug separator if keys sent via USB
	if ( USBKeys_Sent > 0 )
		print("\033[1;32m|\033[0m\n");
}
*/

// Scancode Macro Detection
int scancodeMacro( uint8_t scanCode )
{
	/*
	if ( scanCode == 0x7A )
	{
		scan_resetKeyboard();
	}
	else
	{
		scan_sendData( scanCode );
		_delay_ms( 200 );
		scan_sendData( 0x80 | scanCode );
	}
	return 1;
	*/
	/*
	// BudKeypad
	// Is this a bootloader sequence key?
	if ( !Bootloader_KeyDetected
	   && Bootloader_NextPositionReady
	   && scanCode == Bootloader_ConditionSequence[Bootloader_ConditionState] )
	{
		Bootloader_KeyDetected = 1;
		Bootloader_NextPositionReady = 0;
		Bootloader_ConditionState++;
		erro_dPrint("detect");
	}
	else if ( Bootloader_ConditionState > 0 && scanCode == Bootloader_ConditionSequence[Bootloader_ConditionState - 1] )
	{
		Bootloader_KeyDetected = 0;
		Bootloader_NextPositionReady = 1;
		erro_dPrint("detect-again!");
	}
	// Cancel sequence
	else
	{
		Bootloader_KeyDetected = 0;
		Bootloader_NextPositionReady = 1;
		Bootloader_ConditionState = 0;
		erro_dPrint("Arg");
	}
	*/

	return 0;
}

uint8_t sendCode = 0;

// USBCode Macro Detection
int usbcodeMacro( uint8_t usbCode )
{
	// Keyboard Input Test Macro
	/*
	switch ( usbCode )
	{
	case KEY_F1:
		sendCode--;
		//scan_sendData( 0x90 );
		scan_sendData( sendCode );
		_delay_ms( 200 );
		break;

	case KEY_F2:
		//scan_sendData( 0x90 );
		scan_sendData( sendCode );
		_delay_ms( 200 );
		break;

	case KEY_F3:
		sendCode++;
		//scan_sendData( 0x90 );
		scan_sendData( sendCode );
		_delay_ms( 200 );
		break;

	case KEY_F4:
		sendCode += 0x10;
		//scan_sendData( 0x90 );
		scan_sendData( sendCode );
		_delay_ms( 200 );
		break;

	case KEY_F5:
		// Set 9th bit to 0
		UCSR1B &= ~(1 << 0);
		_delay_ms( 200 );
		break;

	case KEY_F6:
		// Set 9th bit to 1
		UCSR1B |=  (1 << 0);
		_delay_ms( 200 );
		break;

	case KEY_F11:
		// Set click code
		KeyIndex_Add_InputSignal = sendCode;
		_delay_ms( 200 );
		break;

	default:
		return 0;
	}
	
	return 1;
	*/
	return 0;
}


// Given a list of keypresses, translate into the USB key codes
// The buffer is cleared after running
// If the buffer doesn't fit into the USB send array, the extra keys are dropped
void keyPressBufferRead( uint8_t *modifiers, uint8_t numberOfModifiers, uint8_t *map )
{
	// Loop through input buffer
	for ( uint8_t index = 0; index < KeyIndex_BufferUsed; index++ )
	{
		// Get the keycode from the buffer
		uint8_t key = KeyIndex_Buffer[index];

		// Check key for special usages using the scancode
		// If non-zero return, ignore normal processing of the scancode
		if ( scancodeMacro( key ) )
			continue;

		// Check key for special usages using the usbcode
		// If non-zero return, ignore normal processing of the usbcode
		if ( usbcodeMacro( map[key] ) )
			continue;

		// Determine if the key is a modifier
		uint8_t modFound = 0;
		for ( uint8_t mod = 0; mod < numberOfModifiers; mod++ ) {
			// Modifier found
			if ( modifiers[mod] == key ) {
				USBKeys_Modifiers |= map[key];
				modFound = 1;
				break;
			}
		}

		// Modifier, already done this loop
		if ( modFound )
			continue;

		// Too many keys
		if ( USBKeys_Sent >= USBKeys_MaxSize )
		{
			info_print("USB Key limit reached");
			errorLED( 1 );
			break;
		}

		// Allow ignoring keys with 0's
		if ( map[key] != 0 )
		{
			USBKeys_Array[USBKeys_Sent++] = map[key];
		}
		else
		{
			// Key was not mapped
			// TODO Add dead key map
			char tmpStr[6];
			hexToStr( key, tmpStr );
			erro_dPrint( "Key not mapped... - ", tmpStr );
			errorLED( 1 );
		}
	}

	// Signal Macro processor that all of the buffered keys have been processed
	macro_finishedWithBuffer();

	// Signal buffer that we've used it
	scan_finishedWithBuffer();
}

inline void process_macros(void)
{
	// Online process macros once (if some were found), until the next USB send
	if ( USBKeys_Sent != 0 )
		return;

	// Query the input buffer for keypresses
	keyPressBufferRead( MODIFIER_MASK, sizeof(MODIFIER_MASK), KEYINDEX_MASK );

	// Check for bootloader condition
	if ( Bootloader_ConditionState == sizeof( Bootloader_ConditionSequence ) )
		jumpToBootloader();
}