/*
 * main.c
 *
 * Created: 14.11.2016 07:26:13
 * Author : Dennis
 *
 * Description:
 * This is the demo code for the "LED Cube Pendant" project on instructables.com:
 * instructables.com/tag/type-id/user-nqtronix/keyword-XG7D
 */ 

#define F_CPU 8000000UL

#include <avr/io.h>
#include <avr/interrupt.h>
#include <avr/sleep.h>
#include <util/delay.h>

#include "HAL/rtc.h"
#include "HAL/cplex.h"
#include "Data/img_rgb24b_4x4x4.h"


uint8_t display_data[CPLEX_LEDS] = {0};

volatile uint8_t frameStart = 0;
volatile uint8_t enable_sleep = 0;
volatile uint8_t adc_release = 0;

// Uncomment line to set B3 (D- line) whenever processor is in active mode
//#define DEBUG_ON_TIME

// Uncomment for long term power down
//#define FORCE_OFF

// Uncomment to display all white instead of color test pattern
//#define DEBUG_LEDS_ALL_ON

// Uncomment to enable test animation (overrides static image)
//#define DEBUG_LEDS_CYCLE


int main(void)
{
	#ifdef FORCE_OFF
		set_sleep_mode(SLEEP_MODE_PWR_DOWN);
		sleep_mode();
	#endif
	
	
	// disable unused modules
	PRR0	= (1<<PRTWI0)|(1<<PRTIM0)|(1<<PRUSART1)|(1<<PRTIM1)|(1<<PRSPI0)|(1<<PRUSART0);
	PRR1	= (1<<PRTIM3)|(1<<PRSPI1)|(1<<PRTIM4)|(1<<PRPTC)|(1<<PRTWI1);
	
	//Configure outputs
	DDRB	= (1<<5);
	
	#ifdef DEBUG_ON_TIME
		DDRB	|= (1<<3);
	#endif
	
	// Over-clock controller
	OSCCAL	= 180;
	
	// configure on/off button
	PCMSK3	= (1<<PCINT26);
	PCICR	= (1<<PCIE3);

	// set up ADC for button low detection in active mode
	ADMUX	= (0b11<<REFS0)|(1<<ADLAR)|(0b0110<<MUX0);		// MUX -> ADC6/ PE2
	ADCSRA	= (0b111<<ADPS0);								// /128x prescaler
	
	
	#ifdef DEBUG_LEDS_ALL_ON 
		// turn on all LEDs
		for (uint8_t i=0; i<CPLEX_LEDS; i++)
		{
			display_data[i] = 0xFF;
		} 
	#else
		// load image from flash
		for (uint8_t i=0; i<CPLEX_LEDS; i++)
		{
			display_data[i] = pgm_read_byte(&(img_color_demo[i]));
		}
	#endif
	
	
	
	
	// Starts RTC with UNIX time "0"
	rtc_start(0);
		
	// generate frame-tick
	TCCR2A	= 0;
	TIMSK2	|= (1<<OCIE2A);

	sei();
	
    while (1)
    {	
		#ifdef DEBUG_ON_TIME
			PORTB	|= (1<<3);
		#endif
		
		// start conversion ADC:	in high power mode the input w/ the button has an
		//							analog voltage of 0.015 (low) or ~1V (high)
		PRR0	&= ~(1<<PRADC);
		ADCSRA	|= (1<<ADEN)|(1<<ADSC);
		
		// forces frame-rate of currently 128hz
		if(frameStart)
		{
			#ifdef DEBUG_LEDS_CYCLE
			
				static uint8_t dbg_i = 0;
				static uint8_t dbg_cycles = 0;
				
				// wait 16 frames (about 0.25 sec)
				if (dbg_cycles < 16)
				{
					dbg_cycles ++;
				}
				else
				{
					dbg_cycles = 0;
					
					// clear last LED
					display_data[dbg_i] = 0;
					
					// increase LED counter
					dbg_i ++;
					if (dbg_i >= CPLEX_LEDS)
					{
						dbg_i = 0;
					}
					
					// turn on LED at fulll brightness
					display_data[dbg_i] = 0xFF;
				}
				
			#endif

			
			// load the next block of data
			cplex_nextFrame(display_data);
			
			// display each pixel after another
			uint8_t l_LEDs_left;
			do{
				 l_LEDs_left = cplex_nextLED();
			}while (l_LEDs_left);
			
			frameStart = 0;
		}
		
		// Wait for the result of the ADC
		while (ADCSRA&(1<<ADSC))
		{
			// force "empty" loop
			asm volatile("");
		}
		
		// button has to be released between wakeup and power-save
		if (ADCH > 200)
		{
			adc_release = 1;
		}
		
		// go into power-save
		if (adc_release && (ADCH < 10))
		{
			// disable step-up + LED timer
			// disabling interrupt effectively disables LEDs
			TIMSK2	&= ~(1<<OCIE2A);
			PORTB	&= ~(1<<5);
			
			// disable internal modules
			PRR0	|= (1<<PRADC);
			
			// enable button interrupt
			PCICR	= (1<<PCIE3);
		}
		
		#ifdef DEBUG_ON_TIME
			PORTB	&= ~(1<<3);
		#endif
		
		// Timer2 used for frame timing -> PWR_SAVE mode reduces current consumption
		set_sleep_mode(SLEEP_MODE_PWR_SAVE);
		sleep_mode();
    }
}

ISR(PCINT3_vect)
{
	// my type of button does not bounce! (withing a 100 cycle experiment period)
	if (!(PINE&(1<<PINE2)))
	{
		// disable button interrupt (in active mode: polling)
		PCICR	= 0;
		
		// reset ADC button thing
		adc_release = 0;
		
		// enable step-up + LED timer
		PORTB	|= (1<<5);
		TIMSK2	|= (1<<OCIE2A);
	}
}

ISR(TIMER2_COMPA_vect)
{
	frameStart	= 1;
}