#include "jwindow.hpp"
#include "lisp.hpp"
#include "scroller.hpp"
#include "id.hpp"
#include "cache.hpp"
#include "language.hpp"
#include "dprint.hpp"
#include "loader2.hpp"
#include <math.h>

extern window_manager *eh;
extern int dev_ok;
palette *old_pal=NULL;

class gray_picker : public spicker
{
  public :
  int sc;
  virtual void draw_item(window_manager *wm, image *screen, int x, int y, int num, int active)
  { 
    long x2=x+item_width(wm)-1;
    long y2=y+item_height(wm)-1;
    screen->bar(x,y,x2,y2,0);
    screen->bar(x,y,x2-3,y2,sc+num);
    if (active) screen->rectangle(x,y,x2,y2,31);
  }
  void set_pos(int x) { cur_sel=x; }
  virtual int total() { return 32; }
  virtual int item_width(window_manager *wm) { return 12; }
  virtual int item_height(window_manager *wm) { return 20; }
  virtual int activate_on_mouse_move() { return 0; }
  gray_picker(int X, int Y, int ID, int start, int current, ifield *Next) : spicker(X,Y,ID,1,20,0,0,Next)
  { cur_sel=current; sc=start; reconfigure(); cur_sel=current; }    
} ;


FILE *open_FILE(char *filename, char *mode);

void gamma_correct(palette *&pal, int force_menu)
{
  long dr=0,dg=7*4,db=0,old_dg=0;
  int abort=0;

  Cell *gs=find_symbol("darkest_gray");                   // see if user has already done this routine

  if (old_pal)
  { delete pal; pal=old_pal; old_pal=NULL; }

  if (gs && DEFINEDP(symbol_value(gs)) && !force_menu)
    dg=lnumber_value(symbol_value(gs));
  else
  {    
    if (gs && DEFINEDP(symbol_value(gs)))
      dg=old_dg=lnumber_value(symbol_value(gs));


    palette *gray_pal=pal->copy();            // load in a fine gray palette they can chose from
    int i=0;
    int tc=32;

    for (;i<tc;i++)
      gray_pal->set(i,i*4,i*4,i*4);

    gray_pal->load();


    int wm_bc=eh->bright_color(),wm_mc=eh->medium_color(),wm_dc=eh->dark_color();

    int br_r=pal->red(wm_bc)+20;       if (br_r>255) br_r=255;
    int br_g=pal->green(wm_bc)+20;     if (br_g>255) br_g=255;
    int br_b=pal->blue(wm_bc)+20;      if (br_b>255) br_b=255;

    int md_r=pal->red(wm_mc)-20;       if (md_r<0) md_r=0;
    int md_g=pal->green(wm_mc)-20;     if (md_g<0) md_g=0;
    int md_b=pal->blue(wm_mc)-20;      if (md_b<0) md_b=0;

    int dr_r=pal->red(wm_dc)-40;       if (dr_r<0) dr_r=0;
    int dr_g=pal->green(wm_dc)-40;     if (dr_g<0) dr_g=0;
    int dr_b=pal->blue(wm_dc)-40;      if (dr_b<0) dr_b=0;

    eh->set_colors(gray_pal->find_closest(br_r,br_g,br_b),
		   gray_pal->find_closest(md_r,md_g,md_b),
		   gray_pal->find_closest(dr_r,dr_g,dr_b));
		   

    int wl=WINDOW_FRAME_LEFT,wh=WINDOW_FRAME_TOP;
    int sh=eh->font()->height()+35;
    button *but=new button(wl+5,wh+5+sh*3,ID_GAMMA_OK,cash.img(ok_button),
		new info_field(wl+35,wh+10+sh*3,ID_NULL,lang_string("gamma_msg"),0));

    gray_picker *gp=new gray_picker(wl+2,wh+5+sh*1,ID_GREEN_PICKER,0,dg/4,but);    gp->set_pos(dg/4);
    
    jwindow *gw=eh->new_window(xres/2-180,yres/2-90,-1,-1,gp);

    event ev;
    eh->flush_screen();
    do
    {
      do { eh->get_event(ev); } while (ev.type==EV_MOUSE_MOVE && eh->event_waiting()); 
      eh->flush_screen();
      if (ev.type==EV_CLOSE_WINDOW) abort=1;
      if (ev.type==EV_KEY && ev.key==JK_ESC) abort=1;
    } while (!abort && (ev.type!=EV_MESSAGE || ev.message.id!=ID_GAMMA_OK));

    dg=((spicker *)gw->inm->get(ID_GREEN_PICKER))->first_selected()*4;

    eh->close_window(gw);
    eh->flush_screen();

    eh->set_colors(wm_bc,wm_mc,wm_dc);
    delete gray_pal;

    if (!abort)
    {
      FILE *fp=open_FILE("gamma.lsp","wb");
      if (fp)
      {
				fprintf(fp,"(setq darkest_gray %d)\n",dg);
				fclose(fp);
				int sp=current_space;
				current_space=PERM_SPACE;
				set_symbol_value(make_find_symbol("darkest_gray"),new_lisp_number(dg));
			
				current_space=sp;
      } else dprintf("Unable to write to file gamma.lsp\n");
    }
  }

  if (abort) dg=old_dg;

  if (dg<1) dg=1; else if (dg>128) dg=128;

  double gamma=log(dg/255.0)/log(16.0/255.0);

  old_pal=pal;
  pal=new palette;

  for (int i=0;i<256;i++)
  {
    double r,g,b;
    uchar or,og,ob;
    old_pal->get(i,or,og,ob);
    pal->set(i,(int)(pow(or/255.0,gamma)*255),
	     (int)(pow(og/255.0,gamma)*255),
	     (int)(pow(ob/255.0,gamma)*255));
  }

  pal->load();
}