/*
    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"

#include <nds.h>
#include <nds/registers_alt.h>
#include "SDL.h"
#include "SDL_error.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_ndsvideo.h"
#include "SDL_ndsevents_c.h"
#include "SDL_ndsmouse_c.h"

#define NDSVID_DRIVER_NAME "nds"

/* Initialization/Query functions */
static int NDS_VideoInit(_THIS, SDL_PixelFormat *vformat);
static SDL_Rect **NDS_ListModes(_THIS, SDL_PixelFormat *format, Uint32 flags);
static SDL_Surface *NDS_SetVideoMode(_THIS, SDL_Surface *current, int width, int height, int bpp, Uint32 flags);
static int NDS_SetColors(_THIS, int firstcolor, int ncolors, SDL_Color *colors);
static void NDS_VideoQuit(_THIS);

/* Hardware surface functions */
static int NDS_AllocHWSurface(_THIS, SDL_Surface *surface);
static int NDS_LockHWSurface(_THIS, SDL_Surface *surface);
static int NDS_FlipHWSurface(_THIS, SDL_Surface *surface);
static void NDS_UnlockHWSurface(_THIS, SDL_Surface *surface);
static void NDS_FreeHWSurface(_THIS, SDL_Surface *surface);

/* etc. */
static void NDS_UpdateRects(_THIS, int numrects, SDL_Rect *rects);

/* NDS driver bootstrap functions */

static int NDS_Available(void)
{
	return(1);
}

static void NDS_DeleteDevice(SDL_VideoDevice *device)
{
	SDL_free(device->hidden);
	SDL_free(device);
}

void on_irq_vblank() 
{	
  // Disable interrupts
  //REG_IME = 0;
  scanKeys();

  //  VBLANK_INTR_WAIT_FLAGS |= IRQ_VBLANK; 
  //  REG_IF |= IRQ_VBLANK; 
  //REG_IF = REG_IF;

  // Enable interrupts
  //REG_IME = 1;
}

static int HWAccelBlit(SDL_Surface *src, SDL_Rect *srcrect,
                        SDL_Surface *dst, SDL_Rect *dstrect)
 {
	return 0;
 }
 
static int CheckHWBlit(_THIS, SDL_Surface *src, SDL_Surface *dst)
{
 	if (src->flags & SDL_SRCALPHA) return false;
 	if (src->flags & SDL_SRCCOLORKEY) return false;
 	if (src->flags & SDL_HWPALETTE ) return false;
 	if (dst->flags & SDL_SRCALPHA) return false;
 	if (dst->flags & SDL_SRCCOLORKEY) return false;
 	if (dst->flags & SDL_HWPALETTE ) return false;

 	if (src->format->BitsPerPixel != dst->format->BitsPerPixel) return false;
 	if (src->format->BytesPerPixel != dst->format->BytesPerPixel) return false;
 		
        src->map->hw_blit = HWAccelBlit;
        return true;
}

static SDL_VideoDevice *NDS_CreateDevice(int devindex)
{
	SDL_VideoDevice *device=0;


	/* 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 = NDS_VideoInit;
	device->ListModes = NDS_ListModes;
	device->SetVideoMode = NDS_SetVideoMode;
	device->CreateYUVOverlay = NULL;
	device->SetColors = NDS_SetColors;
	device->UpdateRects = NDS_UpdateRects;
	device->VideoQuit = NDS_VideoQuit;
	device->AllocHWSurface = NDS_AllocHWSurface;
	device->CheckHWBlit = CheckHWBlit;
	device->FillHWRect = NULL;
	device->SetHWColorKey = NULL;
	device->SetHWAlpha = NULL;
	device->LockHWSurface = NDS_LockHWSurface;
	device->UnlockHWSurface = NDS_UnlockHWSurface;
	device->FlipHWSurface = NDS_FlipHWSurface;
	device->FreeHWSurface = NDS_FreeHWSurface;
	device->SetCaption = NULL;
	device->SetIcon = NULL;
	device->IconifyWindow = NULL;
	device->GrabInput = NULL;
	device->GetWMInfo = NULL;
	device->InitOSKeymap = NDS_InitOSKeymap;
	device->PumpEvents = NDS_PumpEvents;
	device->info.blit_hw=1;

	device->free = NDS_DeleteDevice;
	return device;
}

VideoBootStrap NDS_bootstrap = {
	NDSVID_DRIVER_NAME, "SDL NDS video driver",
	NDS_Available, NDS_CreateDevice
};

	u16* frontBuffer;// = (u16*)(0x06000000);
	u16* backBuffer;// =  (u16*)(0x06000000 + 256 * 256 * 2); 
int NDS_VideoInit(_THIS, SDL_PixelFormat *vformat)
{
	//printf("WARNING: You are using the SDL NDS video driver!\n");

	/* Determine the screen depth (use default 8-bit depth) */
	/* we change this during the SDL_SetVideoMode implementation... */
	vformat->BitsPerPixel = 16;	// mode 3
	vformat->BytesPerPixel = 2;
	vformat->Rmask = 0x0000f800;
	vformat->Gmask = 0x000007e0;
	vformat->Bmask = 0x0000001f; 
    powerON(POWER_ALL);
	irqInit();
	irqSet(IRQ_VBLANK, on_irq_vblank); 
	irqEnable(IRQ_VBLANK);

    //set the mode for 2 text layers and two extended background layers
	//videoSetMode(MODE_5_2D | DISPLAY_BG3_ACTIVE); 
	videoSetMode(MODE_6_2D| DISPLAY_BG2_ACTIVE); 
	
	//set the sub background up for text display (we could just print to one
	//of the main display text backgrounds just as easily
	videoSetModeSub(MODE_0_2D | DISPLAY_BG0_ACTIVE); //sub bg 0 will be used to print text
	
    //set the first two banks as background memory and the third as sub background memory
    //D is not used..if you need a bigger background then you will need to map
    //more vram banks consecutivly (VRAM A-D are all 0x20000 bytes in size)
    //vramSetMainBanks(VRAM_A_MAIN_BG_0x6000000, VRAM_B_MAIN_BG_0x6020000,VRAM_C_SUB_BG , VRAM_D_LCD); 
	vramSetMainBanks(VRAM_A_MAIN_BG,VRAM_B_MAIN_BG,VRAM_C_MAIN_BG,VRAM_D_MAIN_BG);
	//vramSetBankA(VRAM_A_MAIN_BG);
	//vramSetBankB(VRAM_B_MAIN_BG);
	//vramSetBankC(VRAM_C_MAIN_BG);
	//vramSetBankD(VRAM_D_MAIN_BG);
	//vramSetBankE(VRAM_E_MAIN_BG);
	//vramSetBankF(VRAM_F_MAIN_BG);
	//vramSetBankG(VRAM_G_MAIN_BG);
	vramSetBankH(VRAM_H_SUB_BG);
	vramSetBankI(VRAM_I_LCD);
    
	////////////////set up text background for text/////////////////////
    SUB_BG0_CR = BG_MAP_BASE(8);
	
	BG_PALETTE_SUB[255] = RGB15(31,31,31);//by default font will be rendered with color 255
	///////////////set up our bitmap background///////////////////////

	//BG3_CR = BG_BMP16_512x512;
	
	//these are rotation backgrounds so you must set the rotation attributes:
    //these are fixed point numbers with the low 8 bits the fractional part
    //this basicaly gives it a 1:1 translation in x and y so you get a nice flat bitmap
      /*  BG3_XDX = 1<<8;
        BG3_XDY = 0; 
        BG3_YDX = 0;
        BG3_YDY = 1<<8;
    //our bitmap looks a bit better if we center it so scroll down (256 - 192) / 2 
        BG3_CX = 0;
        BG3_CY = 0; 	
		*/
	//consoleInit() is a lot more flexible but this gets you up and running quick
	consoleInitDefault((u16*)SCREEN_BASE_BLOCK_SUB(8), (u16*)CHAR_BASE_BLOCK_SUB(0), 16); 
	

	frontBuffer =(u16*)(0x06000000);
	//backBuffer  =(u16*)(0x06000000 + 1024 * 512*2); 

	//lcdSwap();
	/* We're done! */
	return(0); 
}

SDL_Rect **NDS_ListModes(_THIS, SDL_PixelFormat *format, Uint32 flags)
{
	return (SDL_Rect **) -1;
}

SDL_Surface *NDS_SetVideoMode(_THIS, SDL_Surface *current,
				int width, int height, int bpp, Uint32 flags)
{
	Uint32 Rmask, Gmask, Bmask, Amask; 

	//if(width > 1024 || height > 512 || bpp > 16)
	//	return(NULL);

	if(bpp >8) {
		bpp=16;
 		Rmask = 0x0000001F;
		Gmask = 0x000003E0;
		Bmask = 0x00007C00;
		Amask = 0x00008000;

		videoSetMode(MODE_5_2D| DISPLAY_BG2_ACTIVE); 

		vramSetMainBanks(VRAM_A_MAIN_BG,VRAM_B_MAIN_BG,VRAM_C_MAIN_BG,VRAM_D_MAIN_BG);

		BG2_CR = BG_BMP16_512x512;
	    BG2_XDX = ((width / 256) << 8) | (width % 256) ; 
        BG2_XDY = 0; 
        BG2_YDX = 0;	
        BG2_YDY = ((height / 192) << 8) | ((height % 192) + (height % 192) / 3) ;
        BG2_CX = 0;
        BG2_CY = 0; 
//        for (i=0;i<256*192;i++)
//	        frontBuffer[i] = RGB15(31,0,0)|BIT(15);
	}
	else
	if(bpp <= 8) {
		bpp=8;
		Rmask = 0x00000000;
		Gmask = 0x00000000; 
		Bmask = 0x00000000;
		BG2_CR = BG_BMP8_1024x512;
        BG2_XDX = ((width / 256) << 8) | (width % 256) ;
        BG2_XDY = 0; 
        BG2_YDX = 0;
        BG2_YDY = ((height / 192) << 8) | ((height % 192) + (height % 192) / 3) ;

	}
	else
		if(bpp < 15) bpp=15;
	if(width<=256) width=256;
	else
		if(width<256) width=256;
	if(height<=192) height=192;
	else
		if(height<192) height=192;
	
	if(bpp==8)
	{
		if(width<256) width=256;
		if(height<192) height=192;
		this->hidden->ndsmode=4;
	}
	
	if(bpp==15)
	{
		if(width<256) this->hidden->ndsmode=5;
		else this->hidden->ndsmode=3; 
	}

	this->hidden->buffer= frontBuffer;//NDS_VRAM_BASE;
	
	//NDS_DISPCNT = NDS_DISP_MODE(this->hidden->ndsmode)|NDS_DISP_BG2;
	
 	//fprintf(stderr,"Setting mode %dx%d (ndsmode %d)\n", width, height,this->hidden->ndsmode);

	// FIXME: How do I tell that 15 bits mode is 555?

	SDL_memset(this->hidden->buffer, 0, 1024 * 512* ((this->hidden->ndsmode==4 || this->hidden->ndsmode==5) ? 2 : 1 ) * ((bpp+7) / 8));

	/* Allocate the new pixel format for the screen */
	if ( ! SDL_ReallocFormat(current, bpp, Rmask, Gmask, Bmask, Amask) ) {
		this->hidden->buffer = NULL;
		SDL_SetError("Couldn't allocate new pixel format for requested mode");
		return(NULL);
	}

	/* Set up the new mode framebuffer */
	current->flags = flags | SDL_FULLSCREEN | SDL_HWSURFACE | (this->hidden->ndsmode > 0 ? SDL_DOUBLEBUF : 0);
	this->hidden->w = current->w = width;
	this->hidden->h = current->h = height;
	current->pixels = frontBuffer;

	if (flags & SDL_DOUBLEBUF) { 
		this->hidden->secondbufferallocd=1;
		backBuffer=(u16*)SDL_malloc(1024*512*2);
		current->pixels = backBuffer; 
	}
	if(bpp==8)
		current->pitch =1024;
	else
		current->pitch =1024*2;

	/* We're done */
	return(current);
}

static int NDS_AllocHWSurface(_THIS, SDL_Surface *surface)
{
	if(this->hidden->secondbufferallocd) {
		//printf("double double buffer alloc\n");
		return -1;
	}
	//if(this->hidden->ndsmode==3)
	//{
	//	printf("no 2nd buffer in mode3\n");
	//	return -1;
	//}
	//printf("second buffer\n");
	//this->hidden->secondbufferallocd=1;
	//backBuffer=(u16*)malloc(1024*512*2);
	//surface->pixels = backBuffer; 

	return(0);
}
static void NDS_FreeHWSurface(_THIS, SDL_Surface *surface)
{
	//free(backBuffer);
	this->hidden->secondbufferallocd=0;
}
int z=0;
/* We need to wait for vertical retrace on page flipped displays */
static int NDS_LockHWSurface(_THIS, SDL_Surface *surface)
{
/*
	uint8* a = surface->pixels;
  int i,j;
  a += 5 * SCREEN_WIDTH + 5;
  for( i = 0; i < 195; ++i) {
    uint16* line = a + (SCREEN_WIDTH * i);
    for( j = 0; j < 158; ++j) {
      *line++ = RGB15(155,155,25);
    }
  }
*/
	//if (z <256)
	// BG_PALETTE[z++]=RGB15(255-z,z,255-z);

 
	return(0);
}

static void NDS_UnlockHWSurface(_THIS, SDL_Surface *surface)
{
	return;
}

static int NDS_FlipHWSurface(_THIS, SDL_Surface *surface)
{
	if(this->hidden->secondbufferallocd){
		while(DISP_Y!=192);
	    while(DISP_Y==192); 
		//printf("flip");

		dmaCopyAsynch(backBuffer,frontBuffer,1024*512);
	}
		//printf("flip\n");
        //u16* temp = surface->pixels;
        //surface->pixels = frontBuffer;
        //frontBuffer = temp;
	/*	u8* vram=BG_GFX;
	int x,y;
	for(y = 0; y < 512; y++)
		dmaCopy(&frontBuffer[y*rects->w], &vram[y*512],512);
	//unsigned char buf;
	
	//printf("NDS_FlipHWSurface\n");
	//printf("ptr now: 0x%x\n",surface->pixels);

	    while(DISP_Y!=192);
	    while(DISP_Y==192); 
        //swap
        u16* temp = frontBuffer;
        frontBuffer = backBuffer;
        backBuffer = temp;
        
        //flip 
        //base is 16KB and screen size is 256x256x2 (128KB)
        BG2_CR ^= BG_BMP_BASE( 512 / 16 ); */
/*
	if(surface->pixels == frontBuffer)//NDS_VRAM_BASE)
	{
			while(DISP_Y!=192);
	while(DISP_Y==192); 
        //swap
        u16* temp = backBuffer;
        backBuffer = frontBuffer;
        frontBuffer = temp;
        
        //flip 
        //base is 16KB and screen size is 256x256x2 (128KB)
        BG3_CR ^= BG_BMP_BASE( 128 / 16 ); 
	}
	else
	{

		while(DISP_Y!=192);
	while(DISP_Y==192); 
        //swap
        u16* temp = frontBuffer;
        frontBuffer = backBuffer;
        backBuffer = temp;
        
        //flip 
        //base is 16KB and screen size is 256x256x2 (128KB)
        BG3_CR ^= BG_BMP_BASE( 128 / 16 ); 
		
	}
	*/
	//printf("ptr then: 0x%x\n",surface->pixels);

	//printf("setting dispcnt to 0x%x\n",NDS_DISPCNT = NDS_DISP_MODE(this->hidden->ndsmode)|NDS_DISP_BG2| buf);
	return(0);
}

static void NDS_UpdateRects(_THIS, int numrects, SDL_Rect *rects)
{
	//fprintf(stderr,"update\n");
	/* do nothing. */
	//dmaCopy(frontBuffer,BG_GFX,512*512);
	 /*
	u8* vram=(u8*)BG_GFX;
	int x,y;
	for(y = 0; y < 512; y++)
		dmaCopy(&frontBuffer[y*rects->w], &vram[y*512],512);
	 */

}

int NDS_SetColors(_THIS, int firstcolor, int ncolors, SDL_Color *colors)
{
	//printf("SetColors\n");
	short r,g,b;
	
	if(this->hidden->ndsmode != 4)
	{
		printf("This is not a palettized mode\n");
		return -1;
	}

	int i,j=firstcolor+ncolors;
	for(i=firstcolor;i<j;i++)
	{
		r=colors[i].r>>3;
		g=colors[i].g>>3;
		b=colors[i].b>>3;
		BG_PALETTE[i]=RGB15(r, g, b);
	} 

	return(0);
}

/* Note:  If we are terminated, this could be called in the middle of
   another SDL video routine -- notably UpdateRects.
*/
void NDS_VideoQuit(_THIS)
{
}