#define _GNU_SOURCE
#include <iostream>
#include <cstdarg>
#include <cstdlib>
#include <cstring>
#include <cassert>
#include <unistd.h>
#include "screenbuffer.h"

using namespace std;


bool operator!=(const ScreenBuffer::Style &a,const ScreenBuffer::Style &b){
	return memcmp(&a,&b,sizeof(ScreenBuffer::Style))!=0;
}

bool operator!=(const ScreenBuffer::Cell &a,const ScreenBuffer::Cell &b){
	return memcmp(&a,&b,sizeof(ScreenBuffer::Cell))!=0;
}


static void initTerminal(){
	cout<<"\x1B[?1049h"  // start alternate screen
	      "\x1B[2J"  // clear screen
	      "\x1B[H"  // move to top left
	    <<flush;
}

static void deinitTerminal(){
	cout<<"\x1B[?1049l"  // end alternate screen
	    <<flush;
}


ScreenBuffer::ScreenBuffer(int W,int H)
		:W(W),H(H){
	screen=new Cell[W*H];
	prevscreen=new Cell[W*H];
	initTerminal();
}

ScreenBuffer::~ScreenBuffer(){
	delete[] screen;
	delete[] prevscreen;
	deinitTerminal();
}

void ScreenBuffer::moveto(int x,int y){
	assert(x>=0&&x<W&&y>=0&&y<H);
	curx=x;
	cury=y;
}

void ScreenBuffer::printstr(const char *buf){
	int basex=curx;
	for(int i=0;buf[i];i++){
		if(buf[i]!='\n'){
			if(curx==W){
				if(cury==H-1){
					curx--;
					break;
				}
				curx=basex;
				cury++;
			}
			screen[W*cury+curx].c=buf[i];
			screen[W*cury+curx].style=curstyle;
			curx++;
		} else {
			if(cury==H-1)break;
			curx=basex;
		}
	}

	if(curx==W){
		if(cury==H-1){
			curx--;
		} else {
			curx=basex;
			cury++;
		}
	}
}

int ScreenBuffer::printf_varargs(const char *format,va_list ap){
	char *buf;
	int ret=vasprintf(&buf,format,ap);
	assert(buf);
	printstr(buf);
	free(buf);
	return ret;
}

__attribute__((format (printf, 2, 3)))
int ScreenBuffer::printf(const char *format,...){
	va_list ap;
	va_start(ap,format);
	int ret=printf_varargs(format,ap);
	va_end(ap);
	return ret;
}

__attribute__((format (printf, 4, 5)))
int ScreenBuffer::mvprintf(int x,int y,const char *format,...){
	moveto(x,y);

	va_list ap;
	va_start(ap,format);
	int ret=printf_varargs(format,ap);
	va_end(ap);
	return ret;
}

void ScreenBuffer::setfg(int fg){
	assert((fg>=0&&fg<=7)||fg==9);
	curstyle.fg=fg;
}

void ScreenBuffer::setbg(int bg){
	assert((bg>=0&&bg<=7)||bg==9);
	curstyle.bg=bg;
}

void ScreenBuffer::draw(){
	bool changed[W*H];
	for(int i=0;i<W*H;i++){
		changed[i]=screen[i]!=prevscreen[i];
	}
	for(int y=0;y<H;y++){
		for(int x=1;x<W-1;x++){
			changed[W*y+x]|=changed[W*y+x-1]&changed[W*y+x+1];
		}
	}

	bool prepped=false;

	Style termstyle;

	for(int y=0;y<H;y++){
		bool needtomove=true;
		for(int x=0;x<W;x++){
			if(!changed[W*y+x]){
				needtomove=true;
				continue;
			}
			if(!prepped){
				cout<<"\x1B[?25l"  // invisible cursor
				      "\x1B[0m";  // reset style
				prepped=true;
			}
			if(needtomove){
				cout<<"\x1B["<<y+1<<';'<<x+1<<'H';
				needtomove=false;
			}
			if(screen[W*y+x].style!=termstyle){
				cout<<"\x1B[0;3"<<screen[W*y+x].style.fg<<";4"<<screen[W*y+x].style.bg<<"m";
				termstyle=screen[W*y+x].style;
			}
			cout.put(screen[W*y+x].c);
		}
	}
	if(prepped){
		cout<<"\x1B[?12;25h"  // visible cursor
		    <<"\x1B["<<H<<';'<<W<<'H'  // move past screen
		    <<flush;
	}

	memcpy(prevscreen,screen,W*H*sizeof(Cell));
}

void ScreenBuffer::emergencyDeinit(){
	static const char *str="\n\x1B[?1049l\x1B[12;25h^C\n";
	write(1,str,strlen(str));
}