#include "os.h"
#include "charmap.h"
#include "screen.h"
#include "imgdump.h"
#include "browse.h"

typedef struct 
{	unsigned short int y; // horizontal screen offset
	unsigned short int x; // vertical screen offset
	unsigned short int w; // width
	unsigned short int h; // height
	unsigned long int o;  // data offset
} manuf_image;

#define TIME_DELAY	200

#define N_MANUF_IMAGES		12
#define MANUF_IMAGE_OFFSET	0x894
#define MANUF_IMAGE_SIZE	12
#define MANUF_SIZE_OFFSET	0x924
#define MANUF_RAWSIZE_OFFSET	0x928
#define MANUF_DATA_OFFSET	0x930
#define MANUF_SIZE			0x20000

int help=1;
long int rawdata_size;
int imgmanuf_copied=0;
manuf_image imgmanuf_copy;
manuf_image manuf_images[N_MANUF_IMAGES];
unsigned int orderid[N_MANUF_IMAGES];
unsigned int order[N_MANUF_IMAGES]={9,7,7,8,8,5,6,0,1,2,3,4};
unsigned char* infos[N_MANUF_IMAGES]={	"Low Battery",					//10th
										"Boot1 Recoverable Error",		//8th
										"Send Diagnostics Software",	//8th
										"Boot2 Recoverable Error",		//9th
										"Unrecoverable Error",			//9th
										"Progress Bar Background",		//6th
										"Progress Bar",					//7th
										"Background #1",				//1st
										"Background #2",				//2nd	
										"Background #3",				//3rd
										"Background #4",				//4th
										"Background #5"};				//5th
unsigned char buf[MANUF_SIZE];
unsigned char* rawbuf;

unsigned short int* offscreen1;
unsigned short int* offscreen2;
unsigned short int* offscreen3;

int getOrderID(int* k, int* n)
{	int id=-1,i,j=0;
	if(*k>=N_MANUF_IMAGES)
		return id;
	for(i=0;i<N_MANUF_IMAGES && id<0;i++)
	{	if(order[i]==*k && *n==j)
		{	id=i;
			(*n)++;
		}
		else if(order[i]==*k)
			j++;
	}
	if(id<0)
	{	*n=0;
		(*k)++;
		return getOrderID(k,n);
	}
	return id;
}

int getNext(int id)
{	int i;
	for(i=0;i<N_MANUF_IMAGES;i++)
		if(orderid[i]==id)
			break;
	i++;
	if(i>=N_MANUF_IMAGES)
		i=0;
	return orderid[i];
}

void dispLowerImages(int id,unsigned short int* screen)
{	int i,o;
	for(o=0;o<=order[id];o++)
		for(i=0;i<N_MANUF_IMAGES;i++)
			if(order[i]==o && i!=id)	dispImage(i,screen);
}

void dispHigherImages(int id,unsigned short int* screen)
{	int i,o;
	for(o=order[id]+1;o<N_MANUF_IMAGES;o++)
		for(i=0;i<N_MANUF_IMAGES;i++)
			if(order[i]==o)	dispImage(i,screen);
}

void dispImage(int i, unsigned short int* screen)
{	drawBufImg(screen,manuf_images[i].x,manuf_images[i].y,rawbuf+manuf_images[i].o,manuf_images[i].w,manuf_images[i].h);
}

void moveImage(int id)
{	int k=0;
	manuf_image backm;
	drawBufFullBoxRGB(offscreen1,0,0,SCREEN_WIDTH-1,SCREEN_HEIGHT-1,0x18,0x14,0x10);
	dispLowerImages(id,offscreen1);
	if(manuf_images[id].w==0 || manuf_images[id].h==0)
	{	if(manuf_images[id].w==0) manuf_images[id].w=1;
		if(manuf_images[id].h==0) manuf_images[id].h=1;
		manuf_images[id].o=0;
	}
	while(!isKeyPressed(KEY_NSPIRE_ESC) && !isKeyPressed(KEY_NSPIRE_TAB))
	{	memcpy(offscreen2,offscreen1,SCREEN_WIDTH*SCREEN_HEIGHT*2);
		dispImage(id,offscreen2);
		//dispHigherImages(id,offscreen2);
		drawBufLineBoxRGB(offscreen2,manuf_images[id].x,manuf_images[id].y,manuf_images[id].x+manuf_images[id].w-1,manuf_images[id].y+manuf_images[id].h-1,-1,-1,-1,2);
		if(help)
		{	drwBufStrRGB(offscreen2,0,0,            "[] move  [H] help on/off  [esc] quit",0,-1,-1,-1);
			drwBufStrRGB(offscreen2,0,1*CHAR_HEIGHT,"[tab]  next  [I] shift colors",0,-1,-1,-1);
			drwBufStrRGB(offscreen2,0,2*CHAR_HEIGHT,"[ctrl]  [] resize",0,-1,-1,-1);
			drwBufStrRGB(offscreen2,0,3*CHAR_HEIGHT,"[shift] [] shift data",0,-1,-1,-1);
		}
		memcpy(getScreen(),offscreen2,SCREEN_WIDTH*SCREEN_HEIGHT*2);
		if(k)
		{	wait(TIME_DELAY/k);
			while(!any_key_pressed())
				k=0;
		}
		k++;
		memcpy(&backm,&manuf_images[id],sizeof(manuf_image));
		if(isKeyPressed(KEY_NSPIRE_P))
		{	int r,g,b;
			getPixelRGB(manuf_images[id].x,manuf_images[id].y,&r,&g,&b);
		}
		if(isKeyPressed(KEY_NSPIRE_H))
			help=!help;	
		if(isKeyPressed(KEY_NSPIRE_I))
			if(!manuf_images[id].o || manuf_images[id].o%2)
				manuf_images[id].o++;
			else
				manuf_images[id].o--;
		if(isKeyPressed(KEY_NSPIRE_SHIFT))
		{	if(!isKeyUp() && !isKeyDown() && !isKeyRight() && !isKeyLeft())
				k=1;
			if(isKeyUp() && manuf_images[id].o>=manuf_images[id].w*2)
				manuf_images[id].o-=manuf_images[id].w*2;
			if(isKeyDown())
				manuf_images[id].o+=manuf_images[id].w*2;
			if(isKeyLeft() && manuf_images[id].o>=2)
				manuf_images[id].o-=2;
			if(isKeyRight())
				manuf_images[id].o+=2;
		}
		else if(isKeyPressed(KEY_NSPIRE_CTRL))
		{	if(!isKeyUp() && !isKeyDown() && !isKeyRight() && !isKeyLeft())
				k=1;
			if(isKeyUp() && manuf_images[id].h>1)
				manuf_images[id].h--;
			if(isKeyDown())
				manuf_images[id].h++;
			if(isKeyLeft() && manuf_images[id].w>1)
				manuf_images[id].w--;
			if(isKeyRight())
				manuf_images[id].w++;
		}
		else
		{	if(isKeyUp() && manuf_images[id].y>0)
				manuf_images[id].y--;
			if(isKeyDown() && manuf_images[id].y+manuf_images[id].h+(manuf_images[id].x+manuf_images[id].w<=SCREEN_WIDTH?0:1)<SCREEN_HEIGHT)
				manuf_images[id].y++;
			if(isKeyLeft() && (manuf_images[id].x>0 || manuf_images[id].y>0))
			{	if(manuf_images[id].x==0)
				{	if(manuf_images[id].y>0)
					{	manuf_images[id].x=SCREEN_WIDTH-1;
						manuf_images[id].y--;
					}
				}
				else
					manuf_images[id].x--;
			}
			if(isKeyRight())
			{	manuf_images[id].x++;
				if(manuf_images[id].x>=SCREEN_WIDTH)
				{	if(manuf_images[id].y<SCREEN_HEIGHT-1)
					{	manuf_images[id].x=0;
						manuf_images[id].y++;
					}
				}
			}
		}
		// data & screen overflow fix
		if(manuf_images[id].o+manuf_images[id].w*manuf_images[id].h*2>rawdata_size)
			memcpy(&manuf_images[id],&backm,sizeof(manuf_image));
		else
			if((manuf_images[id].y+manuf_images[id].h-1)*SCREEN_WIDTH+manuf_images[id].x+manuf_images[id].w>SCREEN_HEIGHT*SCREEN_WIDTH)
				memcpy(&manuf_images[id],&backm,sizeof(manuf_image));
	}
	if(isKeyPressed(KEY_NSPIRE_TAB))
	{	while(isKeyPressed(KEY_NSPIRE_TAB));
		moveImage(getNext(id));
	}
}

int image_x=0, image_y=0;

int chooseImage(int id)
{	int w=1;
	int i, j, k=0, z, c, sc, d=0, update=1;
	int height = SCREEN_HEIGHT-3*CHAR_HEIGHT;
	int zmax=height/6;
	drawBufFullBoxRGB(offscreen1,0,0,SCREEN_WIDTH-1,SCREEN_HEIGHT-1,0,0,0);
	c=0xFF;
	for(i=0;i<SCREEN_HEIGHT-height;i++)
	{	drawBufFullHorizRGB(offscreen1,SCREEN_HEIGHT-1-i,c,c,c);
		c=c*50/51;
	}
	sc=c;
	for(j=0;j<3;j++)
		for(i=0;i<4;i++)
		{	c = sc;
			for(z=0;z<zmax;z++)
			{	drawBufLineBoxRGB(offscreen1,i*SCREEN_WIDTH/4+z,j*height/3+z,(i+1)*SCREEN_WIDTH/4-1-z,(j+1)*height/3-1-z,c,c,c,1);
				c=c*9/10;
			}
		}
	drwBufStrRGB(offscreen1,0,height+CHAR_HEIGHT,	"[X] Cut   [DEL] Remove  [ENTER] Edit",0,-1,-1,-1);
	drwBufStrRGB(offscreen1,0,height+2*CHAR_HEIGHT,	"[C] Copy  [ V ] Paste   [ ESC ] Exit",0,-1,-1,-1);
	while(!isKeyPressed(KEY_NSPIRE_ESC) && !isKeyPressed(KEY_NSPIRE_ENTER))
	{	if(update)
		{	k=0;
			memcpy(offscreen2,offscreen1,SCREEN_WIDTH*SCREEN_HEIGHT*2);
			for(j=0;j<3;j++)
				for(i=0;i<4;i++)
				{	drawBufImgBox(offscreen2,i*SCREEN_WIDTH/4+1,j*height/3+1,(i+1)*SCREEN_WIDTH/4-1-1,(j+1)*height/3-1-1,rawbuf+manuf_images[orderid[k]].o,manuf_images[orderid[k]].w,manuf_images[orderid[k]].h);
					k++;
				}
			update=0;
		}
		k=0;
		memcpy(offscreen3,offscreen2,SCREEN_WIDTH*SCREEN_HEIGHT*2);
		for(j=0;j<3;j++)
			for(i=0;i<4;i++)
			{	if(j==image_y && i==image_x)
				{	c=0xFF;
					for(z=0;z<zmax;z++)
					{	drawBufLineBoxRGB(offscreen3,i*SCREEN_WIDTH/4+z,j*height/3+z,(i+1)*SCREEN_WIDTH/4-1-z,(j+1)*height/3-1-z,c,c,c,1);
						c=c*9/10;
					}
					z=1;
					drawBufImgBox(offscreen3,i*SCREEN_WIDTH/4+z,j*height/3+z,(i+1)*SCREEN_WIDTH/4-1-z,(j+1)*height/3-1-z,rawbuf+manuf_images[orderid[k]].o,manuf_images[orderid[k]].w,manuf_images[orderid[k]].h);
					drwBufStrRGB(offscreen3,0,height, infos[orderid[k]],-1,-1,-1);
				}
				k++;
			}
		memcpy(getScreen(),offscreen3,SCREEN_WIDTH*SCREEN_HEIGHT*2);
		if(d)
		{	wait(TIME_DELAY/d);
			while(!any_key_pressed())
				d=0;
		}
		d++;
		k=image_y*4+image_x;
		if(w)
		{	while(any_key_pressed());
			w=0;
		}
		if(isKeyLeft())
		{	image_x--;
			if(image_x<0) image_x=3;
		}
		if(isKeyRight())
		{	image_x++;
			if(image_x>3) image_x=0;
		}
		if(isKeyUp())
		{	image_y--;
			if(image_y<0) image_y=2;
		}
		if(isKeyDown())
		{	image_y++;
			if(image_y>2) image_y=0;
		}
		if(isKeyPressed(KEY_NSPIRE_X))
		{	memcpy(&imgmanuf_copy,&manuf_images[orderid[k]],sizeof(manuf_image));
			memset(&manuf_images[orderid[k]],0,sizeof(manuf_image));
			imgmanuf_copied=1;
			update=1;	w=1;
		}
		if(isKeyPressed(KEY_NSPIRE_C))
		{	memcpy(&imgmanuf_copy,&manuf_images[orderid[k]],sizeof(manuf_image));
			imgmanuf_copied=1;
			update=1;
		}
		if(isKeyPressed(KEY_NSPIRE_V) && imgmanuf_copied)
		{	memcpy(&manuf_images[orderid[k]],&imgmanuf_copy,sizeof(manuf_image));
			update=1;
		}
		if(isKeyPressed(KEY_NSPIRE_DEL))
		{	memset(&manuf_images[orderid[k]],0,sizeof(manuf_image));
			update=1;
		}
	}
	if(isKeyPressed(KEY_NSPIRE_ESC))
		return -1;
	return orderid[image_y*4+image_x];
}

void main()
{	int i, manuf_size, z;
	int image_x=0, image_y=0;
	unsigned char* pin;
	unsigned char* pout;
	char path[1024];
	long int data_size;
	int id;
	offscreen1 = malloc(SCREEN_WIDTH*SCREEN_HEIGHT*2);
	offscreen2 = malloc(SCREEN_WIDTH*SCREEN_HEIGHT*2);
	offscreen3 = malloc(SCREEN_WIDTH*SCREEN_HEIGHT*2);
	memset(buf,0,MANUF_SIZE);
	int selected;
	initScreen();
	int k=0,n=0;
	for(i=0;i<N_MANUF_IMAGES;i++)
		orderid[i]=getOrderID(&k,&n);
	while(selected = chooseFile(path, "/documents/ndless/","mySpire 1.01     the Nspire Manuf Editor"))
	{	resetConsole();
		displn("",0,1);
		FILE* h = fopen(path,"rb");
		if(h)
		{	manuf_size = fread(buf,1,MANUF_SIZE,h);
			fclose(h);
			if(manuf_size==MANUF_SIZE)
			{	if(!memcmp(buf+0x890,"\x0C\x01\xA2\x18",4))
				{	memcpy(manuf_images,buf+MANUF_IMAGE_OFFSET,MANUF_IMAGE_SIZE*N_MANUF_IMAGES);
					memcpy(&data_size, buf+MANUF_SIZE_OFFSET,sizeof(long int));
					memcpy(&rawdata_size, buf+MANUF_RAWSIZE_OFFSET,sizeof(long int));
					rawbuf=malloc(rawdata_size);
					pin=buf+MANUF_DATA_OFFSET;
					pout=rawbuf;
					decompress(data_size,&pin,&pout);
					int img_i=0;
					while( (img_i = chooseImage(img_i))>=0)
					{	while(any_key_pressed());
						moveImage(img_i);
						while(any_key_pressed());
					}
					free(rawbuf);
					resetConsole();
					displn("Do you want to save your modifications? (y/n)",1,1);
					waitKey(0);
					if(waitConfirm())
					{	h = fopen(path,"wb");
						memcpy(buf+MANUF_IMAGE_OFFSET,manuf_images,MANUF_IMAGE_SIZE*N_MANUF_IMAGES);
						fwrite(buf,1,MANUF_SIZE,h);
						fclose(h);
					}
				}
				else
				{	displn("ERROR: CX/CM Manuf image doesn't include boot graphics",1,1);
					waitKey(1);
				}
			}
			else
			{	displn("ERROR: Invalid CX/CM Manuf image",1,1);
				waitKey(1);
			}
		}
	}
	free(offscreen1);
	free(offscreen2);
	free(offscreen3);
}