#include <conio.h>
#include <graphics.h>
#include <stdlib.h>
#include <time.h>
#include "mouse.c"

#define YES 1
#define NO  0

#define XPX 15		/* X pixels by square */
#define YPX 15		/* Y pixels by square */

#define DEFCX 10	/* Default number of squares */
#define DEFCY 10

#define MINE 255

#define STSQUARE struct stsquare

STSQUARE {
   unsigned char value;		/* Number of mines in the surround squares */
   unsigned char sqopen;	/* Square is open */
   unsigned char sqpress;	/* Square is pressed */
   unsigned char sqmark;	/* Square is marked */
} *psquare;

#define value(x,y)   (psquare+(x)*noy+(y))->value
#define sqopen(x,y)  (psquare+(x)*noy+(y))->sqopen
#define sqpress(x,y) (psquare+(x)*noy+(y))->sqpress
#define sqmark(x,y)  (psquare+(x)*noy+(y))->sqmark

int XST,	/* Offset of first pixel X */
    YST,
    nox,	/* Number of squares in X */
    noy,
    dmines,	/* Mines discovered */
    initmines,	/* Number of initial mines */
    sqclosed,	/* Squares still closed */
    maxy;	/* Max number of y pixels of the screen */

void Graph_init();
void Init_param();
void Set_mines(int );
void Set_square(int, int, int);
void Mouse_set();
void Draw_squares();
int  Do_all();
void Blow_up();
void Open_square(int ,int );

/************************************************************************/

void main()
{
   Graph_init();
   Init_param();
   Mouse_set();
   do {
      randomize();
      cleardevice();
      Set_mines(dmines=initmines);
      mouse_enable();
      Draw_squares();
   }
   while (Do_all() != '\x1b');
   closegraph();
}

/*************************************************************************
*                                                                        *
*                           F U N C T I O N S                            *
*                                                                        *
*************************************************************************/

/*----------------------------------------------------------------------*/

void Graph_init(void)
{
   int graphdriver=DETECT, graphmode, errorcode;
   initgraph(&graphdriver, &graphmode, "c:\\tc\\bgi");
   maxy=getmaxy();
}

/*----------------------------------------------------------------------*/

void Init_param()
{

   XST=100;
   nox=DEFCX;
   YST=30;
   noy=DEFCY;

   initmines= nox*noy*4/25;
   (void*)psquare=calloc(nox*noy, sizeof(STSQUARE));
}

/*----------------------------------------------------------------------*/

void Set_mines(int nminas)
{
   STSQUARE *p;
   int i, x, y, a, b;

   sqclosed=nox*noy-nminas;
   p=psquare;
   for (i=nox*noy; i>0; i--, p++)
      p->value = p->sqopen = p->sqpress = p->sqmark = NO;
   for (i=nminas; i>0; i--) {
      x=random(nox);
      y=random(noy);
      value(x,y)=MINE;
   }
   for (x=nox-1; x>=0; x--) {
      for (y=noy-1; y>=0; y--) {
	 if (value(x,y) == MINE)
	    continue;
	 a=x-1;
	 b=y-1;
	 if (a>=0 && b>=0 && a<nox && b<noy)
	    value(x,y)+=(value(a,b) == MINE);
	 a++;
	 if (a>=0 && b>=0 && a<nox && b<noy)
	    value(x,y)+=(value(a,b) == MINE);
	 a++;
	 if (a>=0 && b>=0 && a<nox && b<noy)
	    value(x,y)+=(value(a,b) == MINE);
	 b++;
	 if (a>=0 && b>=0 && a<nox && b<noy)
	    value(x,y)+=(value(a,b) == MINE);
	 a-=2;
	 if (a>=0 && b>=0 && a<nox && b<noy)
	    value(x,y)+=(value(a,b) == MINE);
	 b++;
	 if (a>=0 && b>=0 && a<nox && b<noy)
	    value(x,y)+=(value(a,b) == MINE);
	 a++;
	 if (a>=0 && b>=0 && a<nox && b<noy)
	    value(x,y)+=(value(a,b) == MINE);
	 a++;
	 if (a>=0 && b>=0 && a<nox && b<noy)
	    value(x,y)+=(value(a,b) == MINE);
      }
   }
}

/*----------------------------------------------------------------------*/

/*
   If status=1 then draw the square up, if it's 0 then draw it down
*/
void Set_square(int x, int y, int status)
{
   int xl, xr, yt, yb;

   mouse_disable();
   xl=XST+XPX*x;
   xr=xl+XPX-1;
   yt=YST+YPX*y;
   yb=yt+YPX-1;
   setfillstyle(SOLID_FILL, LIGHTGRAY);
   bar(xl, yt, xr, yb);
   if (status) {
      setcolor(WHITE);
      line(xl,yt,xl,yb-1);
      line(xl,yt,xr-1,yt);
      setcolor(DARKGRAY);
   //  setcolor(BLUE);
      line(xr,yb,xl,yb);
      line(xr,yb,xr,yt);
      switch (sqmark(x,y)) {
	 case 2 : setfillstyle(SOLID_FILL, LIGHTBLUE);
		  setcolor(LIGHTBLUE);
		  break;
	 case 1 : setfillstyle(SOLID_FILL, LIGHTRED);
		  setcolor(LIGHTRED);
		  break;
	 default: setcolor(LIGHTGRAY);
      }
      fillellipse(XST+XPX*x+XPX/2, YST+YPX*y+YPX/2, 3, 3);
      switch (sqmark(x,y)) {
	 case 2 :
	 case 1 : setfillstyle(SOLID_FILL, YELLOW);
		  setcolor(YELLOW);
		  fillellipse(XST+XPX*x+XPX/2, YST+YPX*y+YPX/2, 1, 1);
      }
   }
   else {
      sqclosed--;
      setcolor(DARKGRAY);
      line(xl,yt,xl,yb);
      line(xl,yt,xr,yt);
   }
   mouse_enable();
}

/*----------------------------------------------------------------------*/

void Mouse_set(void)
{
   mouse_horizontal_range(XST, XST+XPX*nox-3);
   mouse_vertical_range(YST, YST+YPX*noy-3);
}

/*----------------------------------------------------------------------*/

void Draw_squares(void)
{
   int x, y;

   mouse_disable();
   for (x=0; x<nox; x++)
      for (y=0; y<noy; y++)
	 Set_square(x,y,1);
   mouse_enable();
}

/*----------------------------------------------------------------------*/

int Do_all(void)
{
   int x, y, xant=-1, yant, mlstant, cminasant=0, ttrans, tant=0;
   long tst, tact;
   char str[] ="xxxxx: 0000",
	blk[] ="ллллллллллл";

   mouse_enable();
   time(&tst);
   while (!kbhit()) {
      if (!sqclosed && !mouse_status) {		/* All mines discovered */
	 settextjustify(LEFT_TEXT, CENTER_TEXT);
	 setcolor(BLACK);
	 outtextxy(0, maxy/2-10, blk);
	 sprintf(str,"Mines:    0");
	 setcolor(LIGHTRED);
	 outtextxy(0, maxy/2-10, str);
	 settextjustify(LEFT_TEXT, CENTER_TEXT);
	 setcolor(BLACK);
	 outtextxy(0, maxy/2+10, blk);
	 sprintf(str," Time: %4d",ttrans);
	 setcolor(LIGHTGREEN);
	 outtextxy(0, maxy/2+10, str);
	 for (x=nox-1; x>=0; x--)
	    for (y=noy-1; y>=0; y--)
	       if (value(x,y)==MINE) {
		  sqmark(x,y)=1;
		  Set_square(x,y,1);
	       }
	 break;
      }
      if (cminasant != dmines) {		/* Update Mines: */
	 cminasant=dmines;
	 settextjustify(LEFT_TEXT, CENTER_TEXT);
	 setcolor(BLACK);
	 outtextxy(0, maxy/2-10, blk);
	 sprintf(str,"Mines: %4d",dmines);
	 setcolor(LIGHTRED);
	 outtextxy(0, maxy/2-10, str);
      }
      if (tant != (ttrans=(int)(time(&tact)-tst))) {	/* Update Time: */
	 tant=ttrans;
	 settextjustify(LEFT_TEXT, CENTER_TEXT);
	 setcolor(BLACK);
	 outtextxy(0, maxy/2+10, blk);
	 sprintf(str," Time: %4d",ttrans);
	 setcolor(LIGHTRED);
	 outtextxy(0, maxy/2+10, str);
      }
      mouse_read_cursor();
      x=(mouse_x-XST)/XPX;
      y=(mouse_y-YST)/YPX;
      if (xant == -1) {
	 xant=x;
	 yant=y;
      }
      if (x!=xant || y!=yant) {				/* Position change */
	 if (sqpress(xant,yant) && !sqopen(xant,yant)) {
	    sqpress(xant,yant)=NO;
	    Set_square(xant,yant,1);
	 }
	 xant=x;
	 yant=y;
      }
      if (!mouse_status)
	 mlstant=0;
      if ((mouse_status == 2) &&
	   !sqopen(x,y)       &&
	   mlstant!=mouse_status) {	/* Right button pressed */
	 mlstant=mouse_status;
	 mouse_disable();
	 switch (sqmark(x,y)=(sqmark(x,y)+1) % 3) {
	    case 1 : dmines--; break;
	    case 2 : dmines++;
	 }
	 Set_square(x,y,1);
	 mouse_enable();
      }
      else if (mouse_status==1 &&	/* Left button pressed */
	       !sqopen(x,y)    &&
	       !sqpress(x,y)   &&	/* And square not pressed */
	       sqmark(x,y)!=1) {
	 sqpress(x,y)=YES;
	 Set_square(x,y,0);
      }
      else if (!mouse_status &&		/* Left button released */
	       !sqopen(x,y)  &&
	       sqpress(x,y)  &&
	       sqmark(x,y)!=1) {    	/* Open the square */
	 if (value(x,y) == MINE) {
	    Blow_up();
	    break;
	 }
	 else
	    Open_square(x,y);
      }
   }
   mouse_disable();
   return (getch());
}

/*----------------------------------------------------------------------*/

void Blow_up(void)
{
   int x, y, xl, yt, xr, yb;

   mouse_disable();
   for (x=0; x<nox; x++) {
      for (y=0; y<noy; y++) {
	 if (value(x,y) == MINE) {
	    settextjustify(CENTER_TEXT, CENTER_TEXT);
	    setcolor(LIGHTRED);
	    outtextxy(XST+XPX*x+XPX/2-1, YST+YPX*y+YPX/2, "л");
	    setcolor(BLACK);
	    outtextxy(XST+XPX*x+XPX/2, YST+YPX*y+YPX/2+1, "*");
	 }
	 else if (sqmark(x,y)==1) {
	    setcolor(YELLOW);
	    xl=XST+XPX*x;
	    xr=xl+XPX-1;
	    yt=YST+YPX*y;
	    yb=yt+YPX-1;
	    line(xl,yt,xr,yb);
	    line(xl,yb,xr,yt);
	 }
      }
   }
}	/* Blow_up */

/*----------------------------------------------------------------------*/

void Open_square(int x, int y)
{
   char num[2]="0";
   int a, b, c;

   sqopen(x,y)=YES;
   if ((num[0]=value(x,y)) != 0) {
      num[0]+='0';
      switch(num[0]) {
	 case '1' : c=LIGHTBLUE; break;
	 case '2' : c=LIGHTGREEN; break;
	 case '3' : c=YELLOW; break;
	 case '4' : c=LIGHTRED; break;
	 default  : c=LIGHTMAGENTA;
      }
      mouse_disable();
      setcolor(c);
      settextjustify(CENTER_TEXT, CENTER_TEXT);
      outtextxy(XST+XPX*x+XPX/2, YST+YPX*y+YPX/2+1, num);
      mouse_enable();
   }
   else {		/* Open near squares if the current number is 0 */
      a=x-1;
      b=y-1;
      if (a>=0 && b>=0 && a<nox && b<noy && !sqopen(a,b) && sqmark(a,b)!=1) {
	 Set_square(a,b,0);
	 Open_square(a,b);
      }
      a++;
      if (a>=0 && b>=0 && a<nox && b<noy && !sqopen(a,b) && sqmark(a,b)!=1) {
	 Set_square(a,b,0);
	 Open_square(a,b);
      }
      a++;
      if (a>=0 && b>=0 && a<nox && b<noy && !sqopen(a,b) && sqmark(a,b)!=1) {
	 Set_square(a,b,0);
	 Open_square(a,b);
      }
      b++;
      if (a>=0 && b>=0 && a<nox && b<noy && !sqopen(a,b) && sqmark(a,b)!=1) {
	 Set_square(a,b,0);
	 Open_square(a,b);
      }
      a-=2;
      if (a>=0 && b>=0 && a<nox && b<noy && !sqopen(a,b) && sqmark(a,b)!=1) {
	 Set_square(a,b,0);
	 Open_square(a,b);
      }
      b++;
      if (a>=0 && b>=0 && a<nox && b<noy && !sqopen(a,b) && sqmark(a,b)!=1) {
	 Set_square(a,b,0);
	 Open_square(a,b);
      }
      a++;
      if (a>=0 && b>=0 && a<nox && b<noy && !sqopen(a,b) && sqmark(a,b)!=1) {
	 Set_square(a,b,0);
	 Open_square(a,b);
      }
      a++;
      if (a>=0 && b>=0 && a<nox && b<noy && !sqopen(a,b) && sqmark(a,b)!=1) {
	 Set_square(a,b,0);
	 Open_square(a,b);
      }
   }
}

/*----------------------------------------------------------------------*/