/*

    SDL - Simple DirectMedia Layer

    Copyright (C) 1997-2006 Sam Lantinga

    This library is free software; you can redistribute it and/or

    modify it under the terms of the GNU Lesser General Public

    License as published by the Free Software Foundation; either

    version 2.1 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,

    but WITHOUT ANY WARRANTY; without even the implied warranty of

    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public

    License along with this library; if not, write to the Free Software

    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

    Sam Lantinga

    slouken@libsdl.org

*/

#include "SDL_config.h"

/* AAlib based SDL video driver implementation.

*/

#include <unistd.h>

#include <sys/stat.h>

#include "SDL_video.h"

#include "SDL_mouse.h"

#include "../SDL_sysvideo.h"

#include "../SDL_pixels_c.h"

#include "../../events/SDL_events_c.h"

#include "SDL_aavideo.h"

#include "SDL_aaevents_c.h"

#include "SDL_aamouse_c.h"

#include <aalib.h>

/* Initialization/Query functions */

static int AA_VideoInit(_THIS, SDL_PixelFormat *vformat);

static SDL_Rect **AA_ListModes(_THIS, SDL_PixelFormat *format, Uint32 flags);

static SDL_Surface *AA_SetVideoMode(_THIS, SDL_Surface *current, int width, int height, int bpp, Uint32 flags);

static int AA_SetColors(_THIS, int firstcolor, int ncolors, SDL_Color *colors);

static void AA_VideoQuit(_THIS);

/* Hardware surface functions */

static int AA_AllocHWSurface(_THIS, SDL_Surface *surface);

static int AA_LockHWSurface(_THIS, SDL_Surface *surface);

static int AA_FlipHWSurface(_THIS, SDL_Surface *surface);

static void AA_UnlockHWSurface(_THIS, SDL_Surface *surface);

static void AA_FreeHWSurface(_THIS, SDL_Surface *surface);

/* Cache the VideoDevice struct */

static struct SDL_VideoDevice *local_this;

/* AAlib driver bootstrap functions */

static int AA_Available(void)

{

	return 1; /* Always available ! */

}

static void AA_DeleteDevice(SDL_VideoDevice *device)

{

	SDL_free(device->hidden);

	SDL_free(device);

}

static SDL_VideoDevice *AA_CreateDevice(int devindex)

{

	SDL_VideoDevice *device;

	/* Initialize all variables that we clean on shutdown */

	device = (SDL_VideoDevice *)SDL_malloc(sizeof(SDL_VideoDevice));

	if ( device ) {

		SDL_memset(device, 0, (sizeof *device));

		device->hidden = (struct SDL_PrivateVideoData *)

				SDL_malloc((sizeof *device->hidden));

	}

	if ( (device == NULL) || (device->hidden == NULL) ) {

		SDL_OutOfMemory();

		if ( device ) {

			SDL_free(device);

		}

		return(0);

	}

	SDL_memset(device->hidden, 0, (sizeof *device->hidden));

	/* Set the function pointers */

	device->VideoInit = AA_VideoInit;

	device->ListModes = AA_ListModes;

	device->SetVideoMode = AA_SetVideoMode;

	device->CreateYUVOverlay = NULL;

	device->SetColors = AA_SetColors;

	device->UpdateRects = NULL;

	device->VideoQuit = AA_VideoQuit;

	device->AllocHWSurface = AA_AllocHWSurface;

	device->CheckHWBlit = NULL;

	device->FillHWRect = NULL;

	device->SetHWColorKey = NULL;

	device->SetHWAlpha = NULL;

	device->LockHWSurface = AA_LockHWSurface;

	device->UnlockHWSurface = AA_UnlockHWSurface;

	device->FlipHWSurface = NULL;

	device->FreeHWSurface = AA_FreeHWSurface;

	device->SetCaption = NULL;

	device->SetIcon = NULL;

	device->IconifyWindow = NULL;

	device->GrabInput = NULL;

	device->GetWMInfo = NULL;

	device->InitOSKeymap = AA_InitOSKeymap;

	device->PumpEvents = AA_PumpEvents;

	device->free = AA_DeleteDevice;

	return device;

}

VideoBootStrap AALIB_bootstrap = {

	"aalib", "ASCII Art Library",

	AA_Available, AA_CreateDevice

};

static void AA_ResizeHandler(aa_context *);

int AA_VideoInit(_THIS, SDL_PixelFormat *vformat)

{

	int keyboard;

	int i;

	/* Initialize all variables that we clean on shutdown */

	for ( i=0; i<SDL_NUMMODES; ++i ) {

		SDL_modelist[i] = SDL_malloc(sizeof(SDL_Rect));

		SDL_modelist[i]->x = SDL_modelist[i]->y = 0;

	}

	/* Modes sorted largest to smallest */

	SDL_modelist[0]->w = 1024; SDL_modelist[0]->h = 768;

	SDL_modelist[1]->w = 800; SDL_modelist[1]->h = 600;

	SDL_modelist[2]->w = 640; SDL_modelist[2]->h = 480;

	SDL_modelist[3]->w = 320; SDL_modelist[3]->h = 400;

	SDL_modelist[4]->w = 320; SDL_modelist[4]->h = 240;

	SDL_modelist[5]->w = 320; SDL_modelist[5]->h = 200;

	SDL_modelist[6] = NULL;

	/* Initialize the library */

	AA_mutex = SDL_CreateMutex();

	aa_parseoptions (NULL, NULL, NULL, NULL);

	AA_context = aa_autoinit(&aa_defparams);

	if ( ! AA_context ) {

		SDL_SetError("Unable to initialize AAlib");

		return(-1);

	}

	/* Enable mouse and keyboard support */

	if ( ! aa_autoinitkbd (AA_context, AA_SENDRELEASE) ) {

		SDL_SetError("Unable to initialize AAlib keyboard");

		return(-1);

	}

	if ( ! aa_autoinitmouse (AA_context, AA_SENDRELEASE) ) {

		fprintf(stderr,"Warning: Unable to initialize AAlib mouse");

	}

	AA_rparams = aa_getrenderparams();

	local_this = this;

	aa_resizehandler(AA_context, AA_ResizeHandler);

	fprintf(stderr,"Using AAlib driver: %s (%s)\n", AA_context->driver->name, AA_context->driver->shortname);

	AA_in_x11 = (SDL_strcmp(AA_context->driver->shortname,"X11") == 0);

	/* Determine the screen depth (use default 8-bit depth) */

	vformat->BitsPerPixel = 8;

	vformat->BytesPerPixel = 1;

	/* We're done! */

	return(0);

}

SDL_Rect **AA_ListModes(_THIS, SDL_PixelFormat *format, Uint32 flags)

{

     if(format->BitsPerPixel != 8)

 		return NULL;

	 if ( flags & SDL_FULLSCREEN ) {

		 return SDL_modelist;

	 } else {

		 return (SDL_Rect **) -1;

	 }

}

/* From aavga.c

   AAlib does not give us the choice of the actual resolution, thus we have to simulate additional

   resolution by scaling down manually each frame

*/

static void fastscale (register char *b1, register char *b2, int x1, int x2, int y1, int y2)

{

	register int ex, spx = 0, ddx, ddx1;

	int ddy1, ddy, spy = 0, ey;

	int x;

	char *bb1 = b1;

	if (!x1 || !x2 || !y1 || !y2)

		return;

	ddx = x1 + x1;

	ddx1 = x2 + x2;

	if (ddx1 < ddx)

		spx = ddx / ddx1, ddx %= ddx1;

	ddy = y1 + y1;

	ddy1 = y2 + y2;

	if (ddy1 < ddy)

		spy = (ddy / ddy1) * x1, ddy %= ddy1;

	ey = -ddy1;

	for (; y2; y2--) {

		ex = -ddx1;

		for (x = x2; x; x--) {

			*b2 = *b1;

			b2++;

			b1 += spx;

			ex += ddx;

			if (ex > 0) {

				b1++;

				ex -= ddx1;

			}

		}

		bb1 += spy;

		ey += ddy;

		if (ey > 0) {

			bb1 += x1;

			ey -= ddy1;

		}

		b1 = bb1;

	}

}

/* Various screen update functions available */

static void AA_DirectUpdate(_THIS, int numrects, SDL_Rect *rects);

SDL_Surface *AA_SetVideoMode(_THIS, SDL_Surface *current,

				int width, int height, int bpp, Uint32 flags)

{

	int mode;

	if ( AA_buffer ) {

		SDL_free( AA_buffer );

	}

	AA_buffer = SDL_malloc(width * height);

	if ( ! AA_buffer ) {

		SDL_SetError("Couldn't allocate buffer for requested mode");

		return(NULL);

	}

/* 	printf("Setting mode %dx%d\n", width, height); */

	SDL_memset(aa_image(AA_context), 0, aa_imgwidth(AA_context) * aa_imgheight(AA_context));

	SDL_memset(AA_buffer, 0, width * height);

	/* Allocate the new pixel format for the screen */

	if ( ! SDL_ReallocFormat(current, 8, 0, 0, 0, 0) ) {

		return(NULL);

	}

	/* Set up the new mode framebuffer */

	current->flags = SDL_FULLSCREEN;

	AA_w = current->w = width;

	AA_h = current->h = height;

	current->pitch = current->w;

	current->pixels = AA_buffer;

	AA_x_ratio = ((double)aa_imgwidth(AA_context)) / ((double)width);

	AA_y_ratio = ((double)aa_imgheight(AA_context)) / ((double)height);

	/* Set the blit function */

	this->UpdateRects = AA_DirectUpdate;

	/* We're done */

	return(current);

}

static void AA_ResizeHandler(aa_context *context)

{

	aa_resize(context);

	local_this->hidden->x_ratio = ((double)aa_imgwidth(context)) / ((double)local_this->screen->w);

	local_this->hidden->y_ratio = ((double)aa_imgheight(context)) / ((double)local_this->screen->h);

	fastscale (local_this->hidden->buffer, aa_image(context), local_this->hidden->w, aa_imgwidth (context), local_this->hidden->h, aa_imgheight (context));

	aa_renderpalette(context, local_this->hidden->palette, local_this->hidden->rparams, 0, 0, aa_scrwidth(context), aa_scrheight(context));

	aa_flush(context);

}

/* We don't actually allow hardware surfaces other than the main one */

static int AA_AllocHWSurface(_THIS, SDL_Surface *surface)

{

	return(-1);

}

static void AA_FreeHWSurface(_THIS, SDL_Surface *surface)

{

	return;

}

/* We need to wait for vertical retrace on page flipped displays */

static int AA_LockHWSurface(_THIS, SDL_Surface *surface)

{

	/* TODO ? */

	return(0);

}

static void AA_UnlockHWSurface(_THIS, SDL_Surface *surface)

{

	return;

}

/* FIXME: How is this done with AAlib? */

static int AA_FlipHWSurface(_THIS, SDL_Surface *surface)

{

	SDL_mutexP(AA_mutex);

	aa_flush(AA_context);

	SDL_mutexV(AA_mutex);

	return(0);

}

static void AA_DirectUpdate(_THIS, int numrects, SDL_Rect *rects)

{

	int i;

	SDL_Rect *rect;

	fastscale (AA_buffer, aa_image(AA_context), AA_w, aa_imgwidth (AA_context), AA_h, aa_imgheight (AA_context));

#if 1

	aa_renderpalette(AA_context, AA_palette, AA_rparams, 0, 0, aa_scrwidth(AA_context), aa_scrheight(AA_context));

#else

	/* Render only the rectangles in the list */

	printf("Update rects : ");

	for ( i=0; i < numrects; ++i ) {

		rect = &rects[i];

		printf("(%d,%d-%d,%d)", rect->x, rect->y, rect->w, rect->h);

		aa_renderpalette(AA_context, AA_palette, AA_rparams, rect->x * AA_x_ratio, rect->y * AA_y_ratio, rect->w * AA_x_ratio, rect->h * AA_y_ratio);

	}

	printf("\n");

#endif

	SDL_mutexP(AA_mutex);

	aa_flush(AA_context);

	SDL_mutexV(AA_mutex);

	return;

}

int AA_SetColors(_THIS, int firstcolor, int ncolors, SDL_Color *colors)

{

	int i;

	for ( i=0; i < ncolors; i++ ) {

	        aa_setpalette(AA_palette, firstcolor + i,

			      colors[i].r>>2,

			      colors[i].g>>2,

			      colors[i].b>>2);

	}

	return(1);

}

/* Note:  If we are terminated, this could be called in the middle of

   another SDL video routine -- notably UpdateRects.

*/

void AA_VideoQuit(_THIS)

{

	int i;

	aa_uninitkbd(AA_context);

	aa_uninitmouse(AA_context);

	/* Free video mode lists */

	for ( i=0; i<SDL_NUMMODES; ++i ) {

		if ( SDL_modelist[i] != NULL ) {

			SDL_free(SDL_modelist[i]);

			SDL_modelist[i] = NULL;

		}

	}

	aa_close(AA_context);

	SDL_DestroyMutex(AA_mutex);

	this->screen->pixels = NULL;	

}