#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <math.h>
#include <limits.h>

#include "bitmap.h" 

//#define DEB

#define FOREGROUND_BLUE	1
#define FOREGROUND_GREEN	2
#define FOREGROUND_RED	4
#define FOREGROUND_INTENSITY	8
#define BACKGROUND_BLUE	16
#define BACKGROUND_GREEN	32
#define BACKGROUND_RED	64
#define BACKGROUND_INTENSITY	128

////////////////////////////////////

#define PS_FILL_EMPTY  0x20
#define PS_FILL_FULL   0xDB
#define PS_FILL_DOWN   0xDC
#define PS_FILL_UP     0xDF

/////////////////////////

#define FILL_0   0x20
#define FILL_25  0xB0
#define FILL_50  0xB1
#define FILL_75  0xB2
#define FILL_100 0xDB

#define MODE_GRAY   1
#define MODE_COLOR1 2
#define MODE_COLOR2 3

double value(int r,int g,int b){
        return (0.299*(r&0xFF) + 0.587*(g&0xFF) + 0.114*(b&0xFF)) / 255.0;
}

double distance(int r1,int g1,int b1,int r2,int g2,int b2){
	double dr = 0.299 * (r1 - r2);
	double dg = 0.587 * (g1 - g2);
	double db = 0.114 * (b1 - b2);
	return sqrt(dr * dr + dg * dg + db * db);
}

void printColor(int r,int g,int b){
      #ifdef DEB
       printf("(%02X,%02X,%02X)",r,g,b);
      #endif
}


void drawColor2(int w,int h,int linesize,byte *buf,int inverse,HANDLE hConsole){
       fprintf(stderr,"\n::drawColor2(..) not implemented yet.\n");
       exit(-5);
}

// TODO: Warning! The height should be divided by 2
void drawColor1(int w,int h,int linesize,byte *buf,int inverse,HANDLE hConsole){
    if (h%2==1){
       fprintf(stderr,"\nThe height should be divided by 2\n");
       exit(-3);
    }
    int x;
    int y;
    for(y=0;y<h;y+=2){
       for(x=0;x<w;x++){
           
           byte *p1 = buf + x * 3 + (h - y - 1) * linesize;
           int b1 = *p1++;
           int g1 = *p1++;
           int r1 = *p1++;

           byte *p2 = buf + x * 3 + (h - y) * linesize;
           int b2 = *p2++;
           int g2 = *p2++;
           int r2 = *p2++;

           #ifdef DEB
           printf("(%d,%d) ",x,y);
           printColor(r1,g1,b1);
           printf("\n");
           #endif

           double v1min = 255.0; 
           double v2min = 255.0;

           int k;
           int re,ge,be;
           int rr1,gr1,br1,ir1;
           int rr2,gr2,br2,ir2;
           int ir;
           // TODO: This loop should be refactored logic or optimized 
           for(k=0;k<16;k++){
               re = 255 * ((k&0x04)?1:0);
               ge = 255 * ((k&0x02)?1:0);
               be = 255 * ((k&0x01)?1:0);
               // TODO: The grey level may be 192,192,192 or 128,128,128. I dont know.
               if (!(k&&0x08)){
                  re/=2;
                  ge/=2;
                  be/=2;
                  ir=0;
               } else {
                  ir=1;
               }
               double v1 = distance(r1,g1,b1,re,ge,be);
               double v2 = distance(r2,g2,b2,re,ge,be);
               if (v1<v1min){
                  v1min=v1;
                  #ifdef DEB
                  printf(" v1 = %f ",v1);
                  printColor(re,ge,be);
                  printf("\n");
                  #endif
                  rr1=re; // 0..255
                  gr1=ge; // 0..255
                  br1=be; // 0..255
                  ir1=ir; // 0..1
                  
               }
               if (v2<v2min){
                  v2min=v2;
                  #ifdef DEB
                  printf(" v2 = %f ",v2);
                  printColor(re,ge,be);
                  printf("\n");
                  #endif
                  rr2=re;
                  gr2=ge;
                  br2=be;
                  ir2=ir;
               }
           }
           #ifdef DEB
           printf(" [1]-> ");
           printColor(rr1,gr1,br1);
           printf("\n");
           printf(" [2]-> ");
           printColor(rr2,gr2,br2);
           printf("\n");
           #endif
           // TODO: To much complex. Should be simplifyed.
           SetConsoleTextAttribute(hConsole, 
                  (rr1?BACKGROUND_RED:0) |
                  (gr1?BACKGROUND_GREEN:0) |
                  (br1?BACKGROUND_BLUE:0) |
                  (ir1?BACKGROUND_INTENSITY:0) |
                  (rr2?FOREGROUND_RED:0) |
                  (gr2?FOREGROUND_GREEN:0) |
                  (br2?FOREGROUND_BLUE:0) |
                  (ir2?FOREGROUND_INTENSITY:0) 
           );
	   printf("%c",PS_FILL_UP);
       }
       printf("\n");
    }
}


void drawGray(int w,int h,int linesize,byte *buf,int inverse,HANDLE hConsole){
    int x;
    int y;
    for(y=0;y<h;y++){
       for(x=0;x<w;x++){
           byte *p = buf + x * 3 + (h - y - 1) * linesize;
           int r = *p++;
           int g = *p++;
           int b = *p++;
           double v = value(r,g,b);
           if (inverse){
              v = 1.0 - v;
           }
	   if (v<0.2){
              printf("%c",FILL_0);
           } else if (v<0.4){
              printf("%c",FILL_25);
           } else if (v<0.6){
              printf("%c",FILL_50);
           } else if (v<0.8){
              printf("%c",FILL_75);
           } else {
              printf("%c",FILL_100);
           }
       }
       printf("\n");
    }
}

 
main(int argc, char *argv[])
{
        int mode = 0;

	if (argc==1) {
		fprintf(stderr,"\nTextModeArt demo 1.0 (c) Mayton and SQL.RU. Written in 'C' :)\n");
		fprintf(stderr,"\nUsage: tma [inputfile.bmp] [-i] [ { -gray | -color1 | color2 } ]");
		fprintf(stderr,"\n");
		return -1;
	} else {

          HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);    
          int  n, w, h, fd, size;
          unsigned char *data;
          BITMAPFILEHEADER bmp;
          BITMAPINFOHEADER inf;
          char* buf;
          int inverse = 0;
          if ((fd = open(argv[1], O_RDONLY)) == -1) {
              fprintf(stderr,"Error open bitmap\n");
              return -1;
          }

          read(fd, &bmp, sizeof(BITMAPFILEHEADER));

          if (bmp.bfType!=0x4d42){
             fprintf(stderr,"This is not a Windows bitmap file!\n");
      	     return -2;
          }

          read(fd, &inf, sizeof(BITMAPINFOHEADER));

          //printf("inf.biBitCount = %d\n",inf.biBitCount);
          if (inf.biBitCount!=24) {
             fprintf(stderr,"The only 24-bit (TrueColor) images is supported! (Sorry...:()\n");
      	     return -6;
          }
       
          w = inf.biWidth;
          h = inf.biHeight;

          int linesize = w * 3 + w % 4;

          if(inf.biSizeImage == 0)  {
            size = linesize * h;
          } else {
            size = inf.biSizeImage;
           }
       
          buf = malloc(size);

          lseek(fd, bmp.bfOffBits, SEEK_SET);
       
          n = read(fd, buf, size);

          int cnt = 2;
          while(cnt<argc){
             char *arg = argv[cnt];
             if (!stricmp("-i",arg)){
                inverse = 1;
             } else if (!stricmp("-gray",  arg)){
                mode = MODE_GRAY;
             } else if (!stricmp("-color1",arg)){
                mode = MODE_COLOR1;
             } else if (!stricmp("-color2",arg)){
                mode = MODE_COLOR2;
             }
             cnt++;
          }
          
          switch(mode){
               case MODE_COLOR1:
                    drawColor1(w,h,linesize,buf,inverse,hConsole);
                    break;
               case MODE_COLOR2:       
                    drawColor2(w,h,linesize,buf,inverse,hConsole);
                    break;
               case MODE_GRAY:
                    drawGray(w,h,linesize,buf,inverse,hConsole);
               default:
                    fprintf(stderr,"Please select mode (gray, color...) !\n");
      	            return -4; 
          }
       
          free(buf);
       }
}