#include <avr/pgmspace.h>
#include "main.h"
#include "display.h"
#include "xprintf.h"
#include "working_modes.h"
#include "timec.h"

#ifndef LCD_GRAPHIC
__flash const unsigned char battery_states[][8] = {
	{
		0b01110,
		0b11111,
		0b11111,
		0b11111,
		0b11111,
		0b11111,
		0b11111,
		0b11111,
	},
	{
		0b01110,
		0b11111,
		0b10001,
		0b11111,
		0b11111,
		0b11111,
		0b11111,
		0b11111,
	},
	{
		0b01110,
		0b11111,
		0b10001,
		0b10001,
		0b10001,
		0b11111,
		0b11111,
		0b11111,
	},
	{
		0b01110,
		0b11111,
		0b10001,
		0b10001,
		0b10001,
		0b10001,
		0b10001,
		0b11111,
	},
};

__flash const unsigned char custom_chars[] = {
	0b00000, /* 0x01 down arrow */
	0b00100,
	0b00100,
	0b00100,
	0b10101,
	0b01110,
	0b00100,
	0b00000,
	
	0b00000, /* 0x02 up arrow */
	0b00100,
	0b01110,
	0b10101,
	0b00100,
	0b00100,
	0b00100,
	0b00000,
};
#endif /* LCD_GRAPHIC */

struct disp_s disp;

void disp_init(void) { /* send custom characters starting with 0x01 */
#ifndef LCD_GRAPHIC
	unsigned char i;
	LCD_WriteCommand(0x40 + 8); // 0x01
	for(i=0; i<sizeof(custom_chars); i++){
		LCD_WriteData(custom_chars[i]);
	};
#endif /* LCD_GRAPHIC */
}

void battery_state_display(void) {
	unsigned char index;
	if (System.bat_volt > 4.0)
		index = 0;
	else if (System.bat_volt > 3.7)
		index = 1;
	else if (System.bat_volt > 3.4)
		index = 2;
	else
		index = 3;

#ifdef LCD_GRAPHIC
	LCD_GoTo(15,0);
	LCD_WriteChar(0x82 + index);
#else
	unsigned char i;
	LCD_WriteCommand(0x40 + 0); // 0x00

	for(i=0; i<8; i++){
		LCD_WriteData(battery_states[index][i]);
	};
#endif /* LCD_GRAPHIC */
}

static inline void disp_line(unsigned char i, __flash const char *text) {
	switch(i) {
		case 2: strcpy_P(disp.line2, text); break;
		case 1: strcpy_P(disp.line1, text); break;
	}
}

void disp_line1(__flash const char *text) {
	disp_line(1, text);
}

void disp_line2(__flash const char *text) {
	disp_line(2, text);
}

__flash const char _gps_wait[] =		"Czekam na GPS...";
__flash const char _gps_ok[] =			"GPS OK!";
__flash const char _card_ok[] =			"Karta OK!";
__flash const char _logging_active[] =	"Zapis aktywny";
__flash const char _logging_paused[] =	"Zapis wstrzymany";
__flash const char _starting[] =		"Uruchamianie...";
__flash const char _battery_low[] =		"Bateria slaba!";
__flash const char _shutting_down[] =	"Wylaczanie...";
__flash const char _no_card[] =			"Brak karty!";
__flash const char _card_error[] =		"Blad karty!";
__flash const char _write_error[] =		"Blad zapisu!";
__flash const char _fat_write_error[] =	"Blad zapisu FAT!";
__flash const char _file_close_error[] =	"Blad zamk.pliku!";
__flash const char _file_open_error[] =	"Blad otw. pliku!";
__flash const char _files_closed[] =	"Pliki zamkniete";
__flash const char _files_open[] =		"Pliki otwarte";
__flash const char _tracking_paused[] =	"Wstrzymano!";
__flash const char _tracking_resumed[] =	"Wznowiono!";
__flash const char _point_saved[] =		"Zapisano!";
__flash const char _point_not_saved[] =	"Nie zapisano!";
__flash const char _logging_auto_paused[] = "Autom. wstrzym.";
__flash const char _sat_count_low[] = "Za malo satelit";

void display_event(unsigned char event) { /* overrides display with current messages */
	switch (event) {
		case DISPLAY_EVENT_STARTUP:
			disp_line1(_starting);
			break;
		case DISPLAY_EVENT_LOW_BATTERY:
			disp_line2(_battery_low);
		/* fall through */
		case DISPLAY_EVENT_POWEROFF:
			disp_line1(_shutting_down);
			break;
		case DISPLAY_EVENT_INITIALIZED:
			disp_line1(PSTR("Start"));
			switch(System.status){
				case STATUS_NO_POWER: case STATUS_OK: case STATUS_NO_GPS: disp.line2[0] = '\0'; break;
				case STATUS_NO_DISK:			disp_line2(_no_card); break;
				case STATUS_DISK_ERROR:			disp_line2(_card_error); break;
				case STATUS_FILE_WRITE_ERROR:	disp_line2(_write_error); break;
				case STATUS_FILE_SYNC_ERROR:	disp_line2(_fat_write_error); break;
				case STATUS_FILE_CLOSE_ERROR:	disp_line2(_file_close_error); break;
				case STATUS_FILE_OPEN_ERROR:	disp_line2(_file_open_error); break;
			}
			break;
		case DISPLAY_EVENT_CARD_INITIALIZED:
			disp_line1(_card_ok);
			disp_line2(_gps_wait);
			break;
		case DISPLAY_EVENT_FILE_CLOSED:
			disp_line2(_files_closed);
			break;
		case DISPLAY_EVENT_FILE_OPEN:
			disp_line2(_files_open);
			break;
		case DISPLAY_EVENT_TRACKING_PAUSED:
			disp_line2(_tracking_paused);
			break;
		case DISPLAY_EVENT_TRACKING_RESUMED:
			disp_line2(_tracking_resumed);
			break;
		case DISPLAY_EVENT_POINT_SAVED:
			disp_line2(_point_saved);
			break;
		case DISPLAY_EVENT_POINT_NOT_SAVED:
			disp_line2(_point_not_saved);
			break;
	}
	display_refresh(1);
}

void disp_func_main_default(void) {
	if (FLAGS & F_FILEOPEN) {
		if (System.tracking_paused) {
			disp_line1(_logging_paused);
		} else if (System.tracking_auto_paused) {
			disp_line1(_logging_auto_paused);
		} else {
			disp_line1(_logging_active);
		}
	} else
		disp_line1(_card_ok);

	if (FLAGS & F_GPSOK) {
		if (System.sat_count_low)
			disp_line2(_sat_count_low);
		else
			disp_line2(_gps_ok);
	} else {
		disp_line2(_gps_wait);
	}
}

void disp_func_coord(void) {
	if (System.location_valid == LOC_INVALID) {
		disp_line1(PSTR("??? N/S"));
		disp_line2(PSTR("??? E/W"));
		return;
	}
	xsprintf(disp.line1, PSTR("%2.6f%c"), (location.lat < 0)?(-location.lat):location.lat, (location.lat < 0)?'S':'N');
	xsprintf(disp.line2, PSTR("%3.6f%c"), (location.lon < 0)?(-location.lon):location.lon, (location.lon < 0)?'W':'E');
}

void disp_func_ele_sat(void) {
	if (System.location_valid == LOC_INVALID) {
		disp_line1(PSTR("ele = ???"));
	} else {
		xsprintf(disp.line1, PSTR("ele = %.1fm"), location.alt);
	}
	xsprintf(disp.line2, PSTR("%2d satelit"), System.satellites_used);
	if (System.sbas)
		strcat_P(disp.line2, PSTR(", DGPS"));
}

void disp_distance_and_time(void) {
	xsprintf(disp.line1, PSTR("%.2f km"), (float)System.distance / 100000.0);
	if (utc > 0 && System.time_start > 0) {
		xsprintf(disp.line2, PSTR("t=%u s"), (unsigned int)(utc - System.time_start));
	} else {
		disp_line2(PSTR("Czas nieznany"));
	}
}

void disp_speed(void) {
	disp_line1(PSTR("Predkosc:"));
	if (utc > 0 && System.time_start > 0) {
		xsprintf(disp.line2, PSTR("%.2f km/h"), (float)System.distance / (float)(utc - System.time_start) * 0.036);
	} else {
		disp_line2(PSTR("nieznana"));
	}
}

void disp_time(void) {
	if (utc == 0) {
		disp_line1(PSTR("?"));
		disp_line2(PSTR("?"));
		return;
	}
	time_t time = utc;
	time += local_time_diff(time) * (signed int)3600;
	struct tm *ct = gmtime(&time);

	xsprintf(disp.line1, PSTR("%d.%02d.%4d"), ct->tm_mday, ct->tm_mon+1, ct->tm_year+1900);
	xsprintf(disp.line2, PSTR("%d:%02d:%02d"), ct->tm_hour, ct->tm_min, ct->tm_sec);
}

void disp_func_temperature(void) {
	disp_line1(PSTR("Temperatura"));
	if (System.temperature_ok) {
		xsprintf(disp.line2, PSTR("%.1f stC"), System.temperature);
	} else {
		disp_line2(PSTR("Blad!"));
	}
}

void display_refresh(unsigned char changed) {
	if (timer_expired(lcd)) {
		changed = 1;
	}

	/* write to LCD */
	if (changed) {
		set_timer(lcd, 1000);
		battery_state_display();
		unsigned char len;
		LCD_GoTo(0,0);
		len = strlen(disp.line1);
		LCD_WriteText(disp.line1);
		while (len<15) {
			len++;
			LCD_WriteData(' ');
		}
#ifndef LCD_GRAPHIC
		LCD_WriteData(0); /* battery symbol */
#endif
		LCD_GoTo(0,1);
		len = strlen(disp.line2);
		LCD_WriteText(disp.line2);
		while (len<16) {
			len++;
			LCD_WriteData(' ');
		}
	}
}