source: abuse/trunk/src/gamma.cpp @ 39 Tweet

#include <math.h>

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

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