/*
 *  Abuse - dark 2D side-scrolling platform game
 *  Copyright (c) 1995 Crack dot Com
 *
 *  This software was released into the Public Domain. As with most public
 *  domain software, no warranty is made or implied by Crack dot Com or
 *  Jonathan Clark.
 */

#include "config.h"

#include <math.h>

#include "game.hpp"

#include "jwindow.hpp"
#include "lisp.hpp"
#include "scroller.hpp"
#include "id.hpp"
#include "cache.hpp"
#include "language.hpp"
#include "dprint.hpp"
#include "loader2.hpp"

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, 255 );
		}
	}
	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;
	}
};


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

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

	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 ) );
		}
		// load in a fine gray palette they can chose from
		palette *gray_pal = pal->copy();
		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 - 190, 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 )
		{
			char *gammapath;
			FILE *fp;

			gammapath = (char *)jmalloc( strlen( get_save_filename_prefix() ) + 10, "gammapath" );
			sprintf( gammapath, "%sgamma.lsp", get_save_filename_prefix() );
			fp = open_FILE( gammapath, "wb" );
			if( fp )
			{
				fprintf( fp, "(setq darkest_gray %ld)\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" );
			}
			jfree( gammapath);
		}
	}

	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++ )
	{
		uint8_t oldr, oldg, oldb;
		old_pal->get( i, oldr, oldg, oldb );
		pal->set( i, (int)( pow( oldr / 255.0, gamma ) * 255 ),
			(int)( pow( oldg / 255.0, gamma ) * 255 ),
			(int)( pow( oldb / 255.0, gamma ) * 255 ) );
	}

	pal->load();
}