#include <stdio.h>
#include <iostream.h>
#include <graphics.h>
#include <dos.h>
#include <conio.h>
#include <stdlib.h>
#include <math.h>

//VIEWPORT dimensions
#define XLPOS 0
#define YLPOS 20
#define XRPOS maxx-50
#define YRPOS maxy-50

//COLOR Window dimensions
#define COLOR_XLPOS 50
#define COLOR_YLPOS maxy-40
#define COLOR_XRPOS 400
#define COLOR_YRPOS maxy
#define COLOR_XSIZE 50
#define COLOR_YSIZE 20

//FUNCTION Window dimensions
#define FUNC_XLPOS maxx-40
#define FUNC_YLPOS 50
#define FUNC_XRPOS maxx
#define FUNC_YRPOS 300
#define FUNC_XSIZE 20
#define FUNC_YSIZE 50

#define DEFAULT_FUNCTION POINT
#define ERASE_SIZE 5

int sqr(int n) { return n*n; }

enum bool{FALSE,TRUE};
enum Button{NONE,LEFT,RIGHT,BOTH};
enum Function{POINT,PENCIL,LINE,POLYLINE,CIRCLE,RECTANGLE,FILL,ERASE,SAVE,OPEN};
enum Area{VIEWPORT,COLOR,FUNCTION,NON};
char* functions[]={"Point","Pen","Line","Poly","Circle","Rect","Fill","Erase","Save","Open"};

int selectedColor=WHITE,borderColor=WHITE;
int maxx,maxy;

union REGS p,o;

initmouse()
{
	p.x.ax=0;
	int86(0x33,&p,&o);
	return(o.x.ax);
}

void showmouseptr()
{
	p.x.ax=1;
	int86(0x33,&p,&o);
}

void restrictmouseptr(int x1,int y1,int x2,int y2)
{
	p.x.ax=7;
	p.x.ax=x1;
	p.x.ax=x2;
	int86(0x33,&p,&o);
	p.x.ax=8;
	p.x.ax=y1;
	p.x.ax=y2;
	int86(0x33,&p,&o);
}

void getmousepos(int* button,int* x,int* y)
{
	p.x.ax=3;
	int86(0x33,&p,&o);
	*button=o.x.bx;
	*x=o.x.cx;
	*y=o.x.dx;
}

void hidemouseptr()
{
	p.x.ax=2;
	int86(0x33,&p,&o);
}

Area getClickedArea(int x,int y)
{
	if(x>XLPOS && x<XRPOS && y<YRPOS & y>YLPOS) return VIEWPORT;
	else if(x>=FUNC_XLPOS && x<FUNC_XRPOS && y>=FUNC_YLPOS && y<=FUNC_YRPOS) return FUNCTION;
	else if(x>=COLOR_XLPOS && x<COLOR_XRPOS && y>COLOR_YLPOS && y<COLOR_YRPOS) return COLOR;
	else return NON;
}

Button pressed(int button)
{
	switch(button)
	{
		case 0: return NONE;break;
		case 1: return LEFT;break;
		case 2: return RIGHT;break;
		case 3: return BOTH;break;
	}
}

void drawBorder()
{
	setcolor(WHITE);
	rectangle(XLPOS,YLPOS,XRPOS,YRPOS);
	setcolor(selectedColor);
}

void getCords(int& x,int& y,Button& b)
{
	int button;
	getmousepos(&button,&x,&y);
	b=pressed(button);
}

void drawFunctions(Function selectedFunction)
{
	int z=0;
	for(int i=FUNC_XLPOS;i<FUNC_XRPOS;i+=FUNC_XSIZE)
	{
		for(int j=FUNC_YLPOS;j<FUNC_YRPOS;j+=FUNC_YSIZE)
		{
			setfillstyle(SOLID_FILL,z++);
			bar(i,j,i+FUNC_XSIZE,j+FUNC_YSIZE);
			setcolor(WHITE);
			setfillstyle(SOLID_FILL,WHITE);
			outtextxy(i,j+(FUNC_YSIZE)/2,functions[z-1]);
		}
	}

	setcolor(WHITE);
	outtextxy(FUNC_XLPOS,FUNC_YRPOS+44,"ллллллллллллл");
	outtextxy(FUNC_XLPOS,FUNC_YRPOS+50,"ллллллллллллл");
	outtextxy(FUNC_XLPOS,FUNC_YRPOS+56,"ллллллллллллл");
	setcolor(RED);
	outtextxy(FUNC_XLPOS,FUNC_YRPOS+50,functions[selectedFunction]);
	setcolor(selectedColor);
	setfillstyle(SOLID_FILL,BLACK);
}

void drawColors(int selectedColor,int borderColor)
{
	int z=WHITE;
	for(int i=COLOR_XLPOS;i<COLOR_XRPOS;i+=COLOR_XSIZE)
	{
		for(int j=COLOR_YLPOS;j<COLOR_YRPOS;j+=COLOR_YSIZE)
		{
			setfillstyle(SOLID_FILL,z%16);
			z++;
			bar(i,j,i+COLOR_XSIZE,j+COLOR_YSIZE);
		}
	}
	setfillstyle(SOLID_FILL,selectedColor);
	bar(COLOR_XRPOS+COLOR_XSIZE,COLOR_YLPOS,COLOR_XRPOS+2*(COLOR_XSIZE),COLOR_YRPOS-COLOR_YSIZE);

	setfillstyle(SOLID_FILL,borderColor);
	bar(COLOR_XRPOS+COLOR_XSIZE,COLOR_YRPOS-COLOR_YSIZE,COLOR_XRPOS+2*(COLOR_XSIZE),COLOR_YRPOS);

	setfillstyle(SOLID_FILL,BLACK);
}

void setImage(char* file)
{
	hidemouseptr();

	clrscr();
	cleardevice();
	drawBorder();
	drawColors(selectedColor,borderColor);
	drawFunctions(DEFAULT_FUNCTION);

	FILE* fp=fopen(file,"rb");
	for(int i=XLPOS;i<XRPOS;i++)
	{
		for(int j=YLPOS;j<YRPOS;j++)
		{
			putpixel(i,j,getc(fp));
		}
	}
	fclose(fp);
	showmouseptr();
}

void getImage(char* file)
{
	hidemouseptr();

	FILE* fp=fopen(file,"w");
	for(int i=XLPOS;i<XRPOS;i++)
	{
		for(int j=YLPOS;j<YRPOS;j++)
		{
			fputc(getpixel(i,j),fp);
		}
	}
	fclose(fp);

	setImage(file);

	showmouseptr();
}


void funcPoint(int x,int y,Button b)
{
	if(b==LEFT)
	{
		hidemouseptr();
		putpixel(x,y,getcolor());
		showmouseptr();
	}
}

void funcLine(int x,int y,Button b)
{
	int prevx=x;
	int prevy=y;

	while(b==LEFT)
	{
		bool isOut=FALSE;
		if(getClickedArea(x,y)!=VIEWPORT) isOut=TRUE;
		if(isOut) break;
		getCords(x,y,b);
	}

	hidemouseptr();
	if(x!=prevx &&  y!=prevy) line(x,y,prevx,prevy);
	showmouseptr();
}

void funcPolyLine(int x,int y,Button b)
{
	bool isOut;
	while(b!=RIGHT)
	{
		int prevx=x;
		int prevy=y;

		do
		{
			getCords(x,y,b);
		}
		while(b!=LEFT && b!=RIGHT);

		isOut=FALSE;
		if(getClickedArea(x,y)!=VIEWPORT) isOut=TRUE;
		if(isOut) break;

		hidemouseptr();
		line(prevx,prevy,x,y);
		showmouseptr();
	}
}

void funcPencil(int x,int y,Button b)
{
	bool isOut;
	if(b==LEFT)
	{
		int prevx=x;
		int prevy=y;
		hidemouseptr();
		while(b==LEFT)
		{
			isOut=FALSE;
			if(getClickedArea(x,y)!=VIEWPORT) isOut=TRUE;
			if(isOut) break;

			line(x,y,prevx,prevy);
			prevx=x;prevy=y;
			getCords(x,y,b);
		}
		showmouseptr();
	}
}

void funcFill(int x,int y,Button b)
{
	if(b==LEFT)
	{
		hidemouseptr();
		setfillstyle(SOLID_FILL,selectedColor);
		floodfill(x,y,borderColor);
		showmouseptr();
	}
}

void funcCircle(int x,int y,Button b)
{
	if(b==LEFT)
	{
		int prevx=x;
		int prevy=y;

		bool isOut;
		while(b==LEFT)
		{
			isOut=FALSE;
			if(getClickedArea(x,y)!=VIEWPORT) isOut=TRUE;
			if(isOut) break;
			getCords(x,y,b);
		}
		hidemouseptr();
		if(!isOut)
			ellipse((int)fabs(prevx+x)/2,(int)fabs(prevy+y)/2,0,360,(int)fabs(prevx-x)/2,(int)fabs(prevy-y)/2);
		showmouseptr();
	}
}

void funcRectangle(int x,int y,Button b)
{
	if(b==LEFT)
	{
		int prevx=x;
		int prevy=y;

		bool isOut;
		while(b==LEFT)
		{
			isOut=FALSE;
			if(getClickedArea(x,y)!=VIEWPORT) isOut=TRUE;
			if(isOut) break;
			getCords(x,y,b);
		}
		hidemouseptr();
		if(!isOut) rectangle(prevx,prevy,x,y);
		showmouseptr();
	}
}

void funcErase(int x,int y,Button b)
{
	if(b==LEFT)
	{
		int prevx=x;
		int prevy=y;

		bool isOut;
		while(b==LEFT)
		{
			isOut=FALSE;
			if(getClickedArea(x,y)!=VIEWPORT) isOut=TRUE;
			if(isOut) break;

			hidemouseptr();
			setfillstyle(SOLID_FILL,BLACK);
			bar(prevx-ERASE_SIZE,prevy-ERASE_SIZE,x+ERASE_SIZE,y+ERASE_SIZE);
			showmouseptr();
			prevx=x;
			prevy=y;
			getCords(x,y,b);
		}
	}
}

void main()
{
	clrscr();
	getch();
	int gd=DETECT,gm;
	initgraph(&gd,&gm,"");
	setcolor(selectedColor);
	int button,x,y;
	maxx=getmaxx();
	maxy=getmaxy();
	setviewport(0,0,maxx,maxy,1);
	drawBorder();
	drawColors(selectedColor,borderColor);
	drawFunctions(DEFAULT_FUNCTION);

	if(initmouse()==0)
	{
		closegraph();
		restorecrtmode();
		cout<<"Mouse driver not loaded";
		exit(1);
	}
	showmouseptr();

	while(!kbhit())
	{
		restrictmouseptr(0,0,maxx,maxy);
		Button b;
		Area a;
		Function f;

		getCords(x,y,b);
		if(b==LEFT || b==RIGHT) a=getClickedArea(x,y);

		if(a==VIEWPORT)
		{
			restrictmouseptr(XLPOS,YLPOS,XRPOS,YRPOS);
			switch(f)
			{
				case POINT:
					funcPoint(x,y,b);break;
				case LINE:
					funcLine(x,y,b);a=NON;break;
				case POLYLINE:
					funcPolyLine(x,y,b);a=NON;break;
				case PENCIL:
					funcPencil(x,y,b);break;
				case FILL:
					funcFill(x,y,b);break;
				case CIRCLE:
					funcCircle(x,y,b);break;
				case RECTANGLE:
					funcRectangle(x,y,b);break;
				case ERASE:
					funcErase(x,y,b);drawBorder();break;
			}
		}

		else if(a==COLOR)
		{
			hidemouseptr();
			if(b==LEFT)
			{
				selectedColor=getpixel(x,y);
				setcolor(selectedColor);
				setfillstyle(SOLID_FILL,selectedColor);
			}
			else if(b==RIGHT)
			{
				borderColor=getpixel(x,y);
			}
			b=NONE;
			drawColors(selectedColor,borderColor);
			a=NON;
			showmouseptr();
		 }

		 else if(a==FUNCTION)
		 {
			hidemouseptr();
			f=getpixel(x,y);
			drawFunctions(f);
			switch(f)
			{
				case SAVE:
					cout<<"Enter file to save: ";
					char file[20];
					gets(file);
					getImage(file);break;
				case OPEN:
					cout<<"Enter file to open: ";
					gets(file);
					setImage(file);break;
			}
			a=NON;
			showmouseptr();
		 }
	}
	getch();
}