#include "UC1601S-I2C.h"
#include <avr/pgmspace.h>

__flash const unsigned char FontLookup [][5] = {
	{ 0x00, 0x00, 0x00, 0x00, 0x00 },  // sp
	{ 0x00, 0x00, 0x2f, 0x00, 0x00 },   // !
	{ 0x00, 0x07, 0x00, 0x07, 0x00 },   // "
	{ 0x14, 0x7f, 0x14, 0x7f, 0x14 },   // #
	{ 0x24, 0x2a, 0x7f, 0x2a, 0x12 },   // $
	{ 0xc4, 0xc8, 0x10, 0x26, 0x46 },   // %
	{ 0x36, 0x49, 0x55, 0x22, 0x50 },   // &
	{ 0x00, 0x05, 0x03, 0x00, 0x00 },   // '
	{ 0x00, 0x1c, 0x22, 0x41, 0x00 },   // (
	{ 0x00, 0x41, 0x22, 0x1c, 0x00 },   // )
	{ 0x14, 0x08, 0x3E, 0x08, 0x14 },   // *
	{ 0x08, 0x08, 0x3E, 0x08, 0x08 },   // +
	{ 0x00, 0x00, 0x50, 0x30, 0x00 },   // ,
	{ 0x10, 0x10, 0x10, 0x10, 0x10 },   // -
	{ 0x00, 0x60, 0x60, 0x00, 0x00 },   // .
	{ 0x20, 0x10, 0x08, 0x04, 0x02 },   // /
	{ 0x3E, 0x51, 0x49, 0x45, 0x3E },   // 0
	{ 0x00, 0x42, 0x7F, 0x40, 0x00 },   // 1
	{ 0x42, 0x61, 0x51, 0x49, 0x46 },   // 2
	{ 0x21, 0x41, 0x45, 0x4B, 0x31 },   // 3
	{ 0x18, 0x14, 0x12, 0x7F, 0x10 },   // 4
	{ 0x27, 0x45, 0x45, 0x45, 0x39 },   // 5
	{ 0x3C, 0x4A, 0x49, 0x49, 0x30 },   // 6
	{ 0x01, 0x71, 0x09, 0x05, 0x03 },   // 7
	{ 0x36, 0x49, 0x49, 0x49, 0x36 },   // 8
	{ 0x06, 0x49, 0x49, 0x29, 0x1E },   // 9
	{ 0x00, 0x36, 0x36, 0x00, 0x00 },   // :
	{ 0x00, 0x56, 0x36, 0x00, 0x00 },   // ;
	{ 0x08, 0x14, 0x22, 0x41, 0x00 },   // <
	{ 0x14, 0x14, 0x14, 0x14, 0x14 },   // =
	{ 0x00, 0x41, 0x22, 0x14, 0x08 },   // >
	{ 0x02, 0x01, 0x51, 0x09, 0x06 },   // ?
	{ 0x32, 0x49, 0x59, 0x51, 0x3E },   // @
	{ 0x7E, 0x11, 0x11, 0x11, 0x7E },   // A
	{ 0x7F, 0x49, 0x49, 0x49, 0x36 },   // B
	{ 0x3E, 0x41, 0x41, 0x41, 0x22 },   // C
	{ 0x7F, 0x41, 0x41, 0x22, 0x1C },   // D
	{ 0x7F, 0x49, 0x49, 0x49, 0x41 },   // E
	{ 0x7F, 0x09, 0x09, 0x09, 0x01 },   // F
	{ 0x3E, 0x41, 0x49, 0x49, 0x7A },   // G
	{ 0x7F, 0x08, 0x08, 0x08, 0x7F },   // H
	{ 0x00, 0x41, 0x7F, 0x41, 0x00 },   // I
	{ 0x20, 0x40, 0x41, 0x3F, 0x01 },   // J
	{ 0x7F, 0x08, 0x14, 0x22, 0x41 },   // K
	{ 0x7F, 0x40, 0x40, 0x40, 0x40 },   // L
	{ 0x7F, 0x02, 0x0C, 0x02, 0x7F },   // M
	{ 0x7F, 0x04, 0x08, 0x10, 0x7F },   // N
	{ 0x3E, 0x41, 0x41, 0x41, 0x3E },   // O
	{ 0x7F, 0x09, 0x09, 0x09, 0x06 },   // P
	{ 0x3E, 0x41, 0x51, 0x21, 0x5E },   // Q
	{ 0x7F, 0x09, 0x19, 0x29, 0x46 },   // R
	{ 0x46, 0x49, 0x49, 0x49, 0x31 },   // S
	{ 0x01, 0x01, 0x7F, 0x01, 0x01 },   // T
	{ 0x3F, 0x40, 0x40, 0x40, 0x3F },   // U
	{ 0x1F, 0x20, 0x40, 0x20, 0x1F },   // V
	{ 0x3F, 0x40, 0x38, 0x40, 0x3F },   // W
	{ 0x63, 0x14, 0x08, 0x14, 0x63 },   // X
	{ 0x07, 0x08, 0x70, 0x08, 0x07 },   // Y
	{ 0x61, 0x51, 0x49, 0x45, 0x43 },   // Z
	{ 0x00, 0x7F, 0x41, 0x41, 0x00 },   // [
	{ 0x55, 0x2A, 0x55, 0x2A, 0x55 },   // 55
	{ 0x00, 0x41, 0x41, 0x7F, 0x00 },   // ]
	{ 0x04, 0x02, 0x01, 0x02, 0x04 },   // ^
	{ 0x40, 0x40, 0x40, 0x40, 0x40 },   // _
	{ 0x00, 0x01, 0x02, 0x04, 0x00 },   // '
	{ 0x20, 0x54, 0x54, 0x54, 0x78 },   // a
	{ 0x7F, 0x48, 0x44, 0x44, 0x38 },   // b
	{ 0x38, 0x44, 0x44, 0x44, 0x28 },   // c
	{ 0x38, 0x44, 0x44, 0x48, 0x7F },   // d
	{ 0x38, 0x54, 0x54, 0x54, 0x18 },   // e
	{ 0x08, 0x7E, 0x09, 0x01, 0x02 },   // f
	{ 0x0C, 0x52, 0x52, 0x52, 0x3E },   // g
	{ 0x7F, 0x08, 0x04, 0x04, 0x78 },   // h
	{ 0x00, 0x44, 0x7D, 0x40, 0x00 },   // i
	{ 0x20, 0x40, 0x44, 0x3D, 0x00 },   // j
	{ 0x7F, 0x10, 0x28, 0x44, 0x00 },   // k
	{ 0x00, 0x41, 0x7F, 0x40, 0x00 },   // l
	{ 0x7C, 0x04, 0x18, 0x04, 0x78 },   // m
	{ 0x7C, 0x08, 0x04, 0x04, 0x78 },   // n
	{ 0x38, 0x44, 0x44, 0x44, 0x38 },   // o
	{ 0x7C, 0x14, 0x14, 0x14, 0x08 },   // p
	{ 0x08, 0x14, 0x14, 0x18, 0x7C },   // q
	{ 0x7C, 0x08, 0x04, 0x04, 0x08 },   // r
	{ 0x48, 0x54, 0x54, 0x54, 0x20 },   // s
	{ 0x04, 0x3F, 0x44, 0x40, 0x20 },   // t
	{ 0x3C, 0x40, 0x40, 0x20, 0x7C },   // u
	{ 0x1C, 0x20, 0x40, 0x20, 0x1C },   // v
	{ 0x3C, 0x40, 0x30, 0x40, 0x3C },   // w
	{ 0x44, 0x28, 0x10, 0x28, 0x44 },   // x
	{ 0x0C, 0x50, 0x50, 0x50, 0x3C },   // y
	{ 0x44, 0x64, 0x54, 0x4C, 0x44 },   // z
	{ 0x7F, 0x50, 0x48, 0x44, 0x40 },   // Ł \x7b
	{ 0x44, 0x64, 0x56, 0x4D, 0x44 },   // ź \x7c
	{ 0x38, 0x54, 0xd4, 0x54, 0x18 },   // ę \x7d
	{ 0x48, 0x54, 0x56, 0x55, 0x20 },   // ś \x7e
	{ 0x20, 0x54, 0x54, 0xd4, 0x78 },   // ą \x7f
	{ 0x38, 0x44, 0x46, 0x45, 0x28 },   // ć \x80
	{ 0x00, 0x49, 0x7F, 0x44, 0x00 },   // ł \x81
	{ 0xfe, 0xff, 0xff, 0xff, 0xfe },	// 3/3 \x82
	{ 0xfe, 0xfb, 0xfb, 0xfb, 0xfe },	// 2/3 \x83
	{ 0xfe, 0xe3, 0xe3, 0xe3, 0xfe },	// 1/3 \x84
	{ 0xfe, 0x83, 0x83, 0x83, 0xfe },	// 0/3 \x85
	{ 0x10, 0x20, 0x7e, 0x20, 0x10 },	// ↓ \x86
	{ 0x08, 0x04, 0x7e, 0x04, 0x08 },	// ↑ \x87
};

static inline void uc1601s_set_column_address(unsigned char addr) {
	uc1601s_write_command(UC1601S_CA30 | (addr&0x0f));
	uc1601s_write_command(UC1601S_CA74 | (addr>>4));
}

static inline void uc1601s_set_temp_compensation(unsigned char tc) {
	tc &= 0x3;
	uc1601s_write_command(UC1601S_TC | tc);
}

static inline void uc1601s_set_power_control(unsigned char pc) {
	pc &= 0x7;
	uc1601s_write_command(UC1601S_PC | pc);
}

static inline void uc1601s_set_scroll_line(unsigned char sl) {
	sl &= 0x3f;
	uc1601s_write_command(UC1601S_SL | sl);
}

static inline void uc1601s_set_page_address(unsigned char pa) {
	pa &= 0x0f;
	uc1601s_write_command(UC1601S_PA | pa);
}

static inline void uc1601s_set_vbias(unsigned char pm) {
	uc1601s_write_command(UC1601S_PM);
	uc1601s_write_command(pm);
}

static inline void uc1601s_set_partial_display(unsigned char lc) {
	lc &= 0x1;
	uc1601s_write_command(UC1601S_LC4 | lc);
}

static inline void uc1601s_set_ram_address(unsigned char ac) {
	ac &= 0x3;
	uc1601s_write_command(UC1601S_AC | ac);
}

static inline void uc1601_set_frame_rate(unsigned char lc) {
	lc &= 0x1;
	uc1601s_write_command(UC1601S_LC3 | lc);
}

static inline void uc1601s_set_all_pixel_on(unsigned char dc) {
	dc &= 0x1;
	uc1601s_write_command(UC1601S_DC1 | dc);
}

static inline void uc1601s_set_inverse_display(unsigned char dc) {
	dc &= 0x1;
	uc1601s_write_command(UC1601S_DC0 | dc);
}

static inline void uc1601s_set_display_enable(unsigned char dc) {
	dc &= 0x1;
	uc1601s_write_command(UC1601S_DC2 | dc);
}

static inline void uc1601s_set_lcd_mapping(unsigned char lc) {
	lc &= 0x6;
	uc1601s_write_command(UC1601S_LC21 | lc);
}

static inline void uc1601s_system_reset(void) {
	uc1601s_write_command(UC1601S_SYSTEMRESET);
}

static inline void uc1601s_set_lcd_bias_ratio(unsigned char br) {
	br &= 0x3;
	uc1601s_write_command(UC1601S_BR | br);
}

static inline void uc1601s_set_com_end(unsigned char cen) {
	uc1601s_write_command(UC1601S_CEN);
	uc1601s_write_command(cen);
}

static inline void uc1601s_set_partial_display_start(unsigned char dst) {
	uc1601s_write_command(UC1601S_DST);
	uc1601s_write_command(dst);
}

static inline void uc1601s_set_partial_display_end(unsigned char den) {
	uc1601s_write_command(UC1601S_DEN);
	uc1601s_write_command(den);
}

void uc1601s_write_command(unsigned char cmd) {
	expander_set_bit_no_send(UC1601S_CD_PORT, UC1601S_CD, 0);
	exp_output[0] = cmd;
	expander_write(0);
	expander_set_bit(UC1601S_WR_PORT, UC1601S_WR, 0);
	expander_set_bit_no_send(UC1601S_WR_PORT, UC1601S_WR, 1);
	expander_set_bit(UC1601S_CD_PORT, UC1601S_CD, 1);
}

void uc1601s_write_data(unsigned char data) {
	exp_output[0] = data;
	expander_write(0);
	expander_set_bit(UC1601S_WR_PORT, UC1601S_WR, 0);
	expander_set_bit(UC1601S_WR_PORT, UC1601S_WR, 1);
}

unsigned char uc1601s_read_data(void) {
	unsigned char out;
	UC1601S_DATA_INPUT();
	expander_set_bit(UC1601S_RD_PORT, UC1601S_RD, 0);
	out = expander_read_byte(0, 0);
	expander_set_bit(UC1601S_RD_PORT, UC1601S_RD, 1);
	UC1601S_DATA_OUTPUT();
	return out;
}

#define LCD_COM_COUNT	32
#define LCD_ROW_COUNT	(LCD_COM_COUNT/8)
#define LCD_COL_COUNT	128

#define UC1601S_MY		0x4
#define UC1601S_MX		0x2

void LCD_Initialize(void) {
	expander_set_bit_no_send(UC1601S_CS_PORT, UC1601S_CS, 0);
	expander_set_bit(UC1601S_RST_PORT, UC1601S_RST, 1);
	_delay_ms(5);
	expander_set_bit_no_send(UC1601S_RD_PORT, UC1601S_RD, 1);
	expander_set_bit(UC1601S_WR_PORT, UC1601S_WR, 1);
	UC1601S_DATA_OUTPUT();
	uc1601s_set_power_control(UC1601S_INTERNAL_VLCD);
	uc1601s_set_lcd_bias_ratio(UC1601S_BIAS_RATIO_7);
	uc1601s_set_com_end(LCD_COM_COUNT-1);
	uc1601s_set_lcd_mapping(UC1601S_MY);
	uc1601s_set_vbias(230);
	uc1601s_set_display_enable(1);
}

void LCD_Clear(void) {
	unsigned int col, row;
	
	for (row=0; row<LCD_ROW_COUNT; row++) {
		uc1601s_set_page_address(row);
		uc1601s_set_column_address(0);
		for (col=0; col<LCD_COL_COUNT; col++) {
			uc1601s_write_data(0);
		}
	}
}

void LCD_GoTo(unsigned char x, unsigned char y) {
	uc1601s_set_page_address(y);
	uc1601s_set_column_address(x*6);
}


void LCD_WriteTextP(__flash const char *t) {
	char c;
	while ((c = pgm_read_byte(t))) {
		LCD_WriteChar(c);
		t++;
	}
}

void LCD_WriteText(const char *t) {
	while (*t) {
		LCD_WriteChar(*t);
		t++;
	}
}

void LCD_WriteChar(char c) {
	unsigned char col;
	
	c -= ' ';
	for (col = 0; col < 5; col++) {
		uc1601s_write_data(FontLookup[(int)c][col]);
	}
	uc1601s_write_data(0);
}