// SONIC ROBO BLAST 2
//-----------------------------------------------------------------------------
// Copyright (C) 1993-1996 by id Software, Inc.
// Copyright (C) 1998-2000 by DooM Legacy Team.
// Copyright (C) 2013-2016 by Matthew "Inuyasha" Walsh.
// Copyright (C) 1999-2018 by Sonic Team Junior.
//
// This program is free software distributed under the
// terms of the GNU General Public License, version 2.
// See the 'LICENSE' file for more details.
//-----------------------------------------------------------------------------
/// \file  f_wipe.c
/// \brief SRB2 2.1 custom fade mask "wipe" behavior.

#include "f_finale.h"
#include "i_video.h"
#include "v_video.h"

#include "r_data.h" // NearestColor
#include "r_draw.h" // transtable
#include "p_pspr.h" // tr_transxxx

#include "w_wad.h"
#include "z_zone.h"

#include "i_time.h"
#include "i_system.h"
#include "i_threads.h"
#include "console.h"
#include "d_main.h"
#include "m_misc.h" // movie mode
#include "d_clisrv.h" // So the network state can be updated during the wipe

#include "g_game.h"
#include "st_stuff.h"

#ifdef HWRENDER
#include "hardware/hw_main.h"
#endif

#if NUMSCREENS < 5
#define NOWIPE // do not enable wipe image post processing for ARM, SH and MIPS CPUs
#endif

// SRB2Kart
#include "k_menu.h"

typedef struct fademask_s {
	UINT8* mask;
	UINT16 width, height;
	size_t size;
	fixed_t xscale, yscale;
} fademask_t;

UINT8 wipedefs[NUMWIPEDEFS] = {
	99, // wipe_credits_intermediate (0)

	0,  // wipe_level_toblack
	0,  // wipe_intermission_toblack
	0,  // wipe_voting_toblack,
	0,  // wipe_continuing_toblack
	0,  // wipe_titlescreen_toblack
	1,  // wipe_menu_toblack
	99, // wipe_credits_toblack
	0,  // wipe_evaluation_toblack
	0,  // wipe_gameend_toblack
	0,  // wipe_ceremony_toblack
	UINT8_MAX, // wipe_intro_toblack (hardcoded)
	99, // wipe_ending_toblack (hardcoded)
	99, // wipe_cutscene_toblack (hardcoded)

	72, // wipe_encore_toinvert
	99, // wipe_encore_towhite

	UINT8_MAX, // wipe_level_final
	0,  // wipe_intermission_final
	0,  // wipe_voting_final
	0,  // wipe_continuing_final
	0,  // wipe_titlescreen_final
	1,  // wipe_menu_final
	99, // wipe_credits_final
	0,  // wipe_evaluation_final
	0,  // wipe_gameend_final
	0,  // wipe_ceremony_final
	99, // wipe_intro_final (hardcoded)
	99, // wipe_ending_final (hardcoded)
	99  // wipe_cutscene_final (hardcoded)
};

static boolean g_wipedef_toblack[NUMWIPEDEFS] = {
	true, // wipe_credits_intermediate (0)

	true, // wipe_level_toblack
	true, // wipe_intermission_toblack
	true, // wipe_voting_toblack,
	true, // wipe_continuing_toblack
	true, // wipe_titlescreen_toblack
	true, // wipe_menu_toblack
	true, // wipe_credits_toblack
	true, // wipe_evaluation_toblack
	true, // wipe_gameend_toblack
	true, // wipe_ceremony_toblack
	true, // wipe_intro_toblack (hardcoded)
	true, // wipe_ending_toblack (hardcoded)
	true, // wipe_cutscene_toblack (hardcoded)

	false, // wipe_encore_toinvert
	false, // wipe_encore_towhite

	true, // wipe_level_final
	true, // wipe_intermission_final
	true, // wipe_voting_final
	true, // wipe_continuing_final
	true, // wipe_titlescreen_final
	true, // wipe_menu_final
	true, // wipe_credits_final
	true, // wipe_evaluation_final
	true, // wipe_gameend_final
	true, // wipe_ceremony_final
	true, // wipe_intro_final (hardcoded)
	true, // wipe_ending_final (hardcoded)
	true  // wipe_cutscene_final (hardcoded)
};

static boolean g_wipedef_toinvert[NUMWIPEDEFS] = {
	false, // wipe_credits_intermediate (0)

	false, // wipe_level_toblack
	false, // wipe_intermission_toblack
	false, // wipe_voting_toblack,
	false, // wipe_continuing_toblack
	false, // wipe_titlescreen_toblack
	false, // wipe_menu_toblack
	false, // wipe_credits_toblack
	false, // wipe_evaluation_toblack
	false, // wipe_gameend_toblack
	false, // wipe_ceremony_toblack
	false, // wipe_intro_toblack (hardcoded)
	false, // wipe_ending_toblack (hardcoded)
	false, // wipe_cutscene_toblack (hardcoded)

	true, // wipe_encore_toinvert
	false, // wipe_encore_towhite

	false, // wipe_level_final
	false, // wipe_intermission_final
	false, // wipe_voting_final
	false, // wipe_continuing_final
	false, // wipe_titlescreen_final
	false, // wipe_menu_final
	false, // wipe_credits_final
	false, // wipe_evaluation_final
	false, // wipe_gameend_final
	false, // wipe_ceremony_final
	false, // wipe_intro_final (hardcoded)
	false, // wipe_ending_final (hardcoded)
	false  // wipe_cutscene_final (hardcoded)
};

static boolean g_wipedef_towhite[NUMWIPEDEFS] = {
	false, // wipe_credits_intermediate (0)

	false, // wipe_level_toblack
	false, // wipe_intermission_toblack
	false, // wipe_voting_toblack,
	false, // wipe_continuing_toblack
	false, // wipe_titlescreen_toblack
	false, // wipe_menu_toblack
	false, // wipe_credits_toblack
	false, // wipe_evaluation_toblack
	false, // wipe_gameend_toblack
	false, // wipe_ceremony_toblack
	false, // wipe_intro_toblack (hardcoded)
	false, // wipe_ending_toblack (hardcoded)
	false, // wipe_cutscene_toblack (hardcoded)

	false, // wipe_encore_toinvert
	true, // wipe_encore_towhite

	false, // wipe_level_final
	false, // wipe_intermission_final
	false, // wipe_voting_final
	false, // wipe_continuing_final
	false, // wipe_titlescreen_final
	false, // wipe_menu_final
	false, // wipe_credits_final
	false, // wipe_evaluation_final
	false, // wipe_gameend_final
	false, // wipe_ceremony_final
	false, // wipe_intro_final (hardcoded)
	false, // wipe_ending_final (hardcoded)
	false  // wipe_cutscene_final (hardcoded)
};

static boolean g_wipedef_crossfade[NUMWIPEDEFS] = {
	false, // wipe_credits_intermediate (0)

	false, // wipe_level_toblack
	false, // wipe_intermission_toblack
	false, // wipe_voting_toblack,
	false, // wipe_continuing_toblack
	false, // wipe_titlescreen_toblack
	false, // wipe_menu_toblack
	false, // wipe_credits_toblack
	false, // wipe_evaluation_toblack
	false, // wipe_gameend_toblack
	false, // wipe_ceremony_toblack
	false, // wipe_intro_toblack (hardcoded)
	false, // wipe_ending_toblack (hardcoded)
	false, // wipe_cutscene_toblack (hardcoded)

	false, // wipe_encore_toinvert
	false, // wipe_encore_towhite

	true, // wipe_level_final
	true, // wipe_intermission_final
	true, // wipe_voting_final
	true, // wipe_continuing_final
	true, // wipe_titlescreen_final
	true, // wipe_menu_final
	true, // wipe_credits_final
	true, // wipe_evaluation_final
	true, // wipe_gameend_final
	true, // wipe_ceremony_final
	true, // wipe_intro_final (hardcoded)
	true, // wipe_ending_final (hardcoded)
	true  // wipe_cutscene_final (hardcoded)
};

//--------------------------------------------------------------------------
//                        SCREEN WIPE PACKAGE
//--------------------------------------------------------------------------

boolean WipeInAction = false;
UINT8 g_wipemode = 0;
UINT8 g_wipetype = 0;
UINT8 g_wipeframe = 0;
boolean g_wipereverse = false;
boolean g_wipeskiprender = false;
boolean g_wipeencorewiggle = false;
boolean WipeStageTitle = false;
INT32 lastwipetic = 0;

#ifndef NOWIPE

#define GENLEN 31

static UINT8 *wipe_scr_start; //screen 3
static UINT8 *wipe_scr_end; //screen 4
static UINT8 *wipe_scr; //screen 0 (main drawing)
static UINT8 pallen;
static fixed_t paldiv;

/** Create fademask_t from lump
  *
  * \param	lump	Lump name to get data from
  * \return	fademask_t for lump
  */
static fademask_t *F_GetFadeMask(UINT8 masknum, UINT8 scrnnum) {
	static char lumpname[9] = "FADEmmss";
	static fademask_t fm = {NULL,0,0,0,0,0};
	lumpnum_t lumpnum;
	UINT8 *lump, *mask;
	size_t lsize;
	RGBA_t *pcolor;

	if (masknum > 99 || scrnnum > 99)
		goto freemask;

	// SRB2Kart: This suddenly triggers ERRORMODE now
	//sprintf(&lumpname[4], "%.2hu%.2hu", (UINT16)masknum, (UINT16)scrnnum);

	lumpname[4] = '0'+(masknum/10);
	lumpname[5] = '0'+(masknum%10);

	lumpname[6] = '0'+(scrnnum/10);
	lumpname[7] = '0'+(scrnnum%10);

	lumpnum = W_CheckNumForName(lumpname);
	if (lumpnum == LUMPERROR)
		goto freemask;

	lump = W_CacheLumpNum(lumpnum, PU_CACHE);
	lsize = W_LumpLength(lumpnum);
	switch (lsize)
	{
		case 256000: // 640x400
			fm.width = 640;
			fm.height = 400;
			break;
		case 64000: // 320x200
			fm.width = 320;
			fm.height = 200;
			break;
		case 16000: // 160x100
			fm.width = 160;
			fm.height = 100;
			break;
		case 4000: // 80x50 (minimum)
			fm.width = 80;
			fm.height = 50;
			break;

		default: // bad lump
			CONS_Alert(CONS_WARNING, "Fade mask lump %s of incorrect size, ignored\n", lumpname);
		case 0: // end marker (not bad!, but still need clearing)
			goto freemask;
	}
	if (lsize != fm.size)
		fm.mask = Z_Realloc(fm.mask, lsize, PU_STATIC, NULL);
	fm.size = lsize;

	mask = fm.mask;

	while (lsize--)
	{
		// Determine pixel to use from fademask
		pcolor = &pMasterPalette[*lump++];
		*mask++ = FixedDiv((pcolor->s.red+1)<<FRACBITS, paldiv)>>FRACBITS;
	}

	fm.xscale = FixedDiv(vid.width<<FRACBITS, fm.width<<FRACBITS);
	fm.yscale = FixedDiv(vid.height<<FRACBITS, fm.height<<FRACBITS);
	return &fm;

	// Landing point for freeing data -- do this instead of just returning NULL
	// this ensures the fade data isn't remaining in memory, unused
	// (could be up to 256,000 bytes if it's a HQ fade!)
	freemask:
	if (fm.mask)
	{
		Z_Free(fm.mask);
		fm.mask = NULL;
		fm.size = 0;
	}

	return NULL;
}

#endif

/** Save the "before" screen of a wipe.
  */
void F_WipeStartScreen(void)
{
#ifndef NOWIPE
#ifdef HWRENDER
	if(rendermode != render_soft)
	{
		HWR_StartScreenWipe();
		return;
	}
#endif
	wipe_scr_start = screens[3];
	I_ReadScreen(wipe_scr_start);
	I_FinishUpdateWipeStartScreen();
#endif
}

/** Save the "after" screen of a wipe.
  */
void F_WipeEndScreen(void)
{
#ifndef NOWIPE
#ifdef HWRENDER
	if(rendermode != render_soft)
	{
		HWR_EndScreenWipe();
		return;
	}
#endif
	wipe_scr_end = screens[4];
	I_ReadScreen(wipe_scr_end);
	V_DrawBlock(0, 0, 0, vid.width, vid.height, wipe_scr_start);
	I_FinishUpdateWipeEndScreen();
#endif
}

/** Draw the stage title.
  */
void F_WipeStageTitle(void)
{
	// draw level title
	if ((WipeStageTitle) && G_IsTitleCardAvailable())
	{
		ST_runTitleCard();
		ST_drawTitleCard();
	}
}

/** After setting up the screens you want to wipe,
  * calling this will do a 'typical' wipe.
  */
void F_RunWipe(UINT8 wipemode, UINT8 wipetype, boolean drawMenu, const char *colormap, boolean reverse, boolean encorewiggle)
{
#ifdef NOWIPE
	(void)wipemode;
	(void)wipetype;
	(void)drawMenu;
	(void)colormap;
	(void)reverse;
	(void)encorewiggle;
#else
	tic_t nowtime;
	UINT8 wipeframe = 0;
	fademask_t *fmask;

	lumpnum_t clump = LUMPERROR;
	lighttable_t *fcolor = NULL;

	if (colormap != NULL)
		clump = W_GetNumForName(colormap);

	if (clump != LUMPERROR && wipetype != UINT8_MAX)
	{
		pallen = 32;
		fcolor = Z_MallocAlign((256 * pallen), PU_STATIC, NULL, 8);
		W_ReadLump(clump, fcolor);
	}
	else
	{
		pallen = 11;
		reverse = false;
	}

	paldiv = FixedDiv(257<<FRACBITS, pallen<<FRACBITS);

	// Init the wipe
	WipeInAction = true;
	g_wipeskiprender = false;
	wipe_scr = screens[0];

	// lastwipetic should either be 0 or the tic we last wiped
	// on for fade-to-black
	for (;;)
	{
		// get fademask first so we can tell if it exists or not
		fmask = F_GetFadeMask(wipetype, wipeframe++);
		if (!fmask)
			break;

		// wait loop
		while (!((nowtime = I_GetTime()) - lastwipetic))
		{
			I_Sleep(cv_sleep.value);
			I_UpdateTime(cv_timescale.value);
		}
		lastwipetic = nowtime;

#ifdef HWRENDER
		if (rendermode == render_opengl)
			HWR_DoWipe(wipetype, wipeframe-1); // send in the wipe type and wipeframe because we need to cache the graphic
		else
#endif

		if (rendermode != render_none) //this allows F_RunWipe to be called in dedicated servers
		{
			// F_DoWipe(fmask, fcolor, reverse);
			g_wipemode = wipemode;
			g_wipetype = wipetype;
			g_wipeframe = wipeframe - 1;
			g_wipereverse = reverse;

			if (encorewiggle)
			{
				g_wipeencorewiggle = wipeframe - 1;
			}
			else
			{
				g_wipeencorewiggle = 0;
			}
		}

		I_OsPolling();
		I_UpdateNoBlit();

		if (drawMenu)
		{
#ifdef HAVE_THREADS
			I_lock_mutex(&k_menu_mutex);
#endif
			M_Drawer(); // menu is drawn even on top of wipes
#ifdef HAVE_THREADS
			I_unlock_mutex(k_menu_mutex);
#endif
		}

		I_FinishUpdateWipe(); // page flip or blit buffer

		if (rendermode != render_none)
		{
			// Skip subsequent renders until the end of the wipe to preserve the current frame.
			g_wipeskiprender = true;
		}

#ifdef HWRENDER
		if (moviemode && rendermode == render_opengl)
			M_LegacySaveFrame();
#endif

		NetKeepAlive(); // Update the network so we don't cause timeouts
	}

	WipeInAction = false;
	g_wipeskiprender = false;

	if (fcolor)
	{
		Z_Free(fcolor);
		fcolor = NULL;
	}
#endif
}

/** Returns tic length of wipe
  * One lump equals one tic
  */
tic_t F_GetWipeLength(UINT8 wipetype)
{
#ifdef NOWIPE
	(void)wipetype;
	return 0;
#else
	static char lumpname[10] = "FADEmmss";
	lumpnum_t lumpnum;
	UINT8 wipeframe;

	if (wipetype > 99)
		return 0;

	for (wipeframe = 0; wipeframe < 100; wipeframe++)
	{
		sprintf(&lumpname[4], "%.2hu%.2hu", (UINT16)wipetype, (UINT16)wipeframe);

		lumpnum = W_CheckNumForName(lumpname);
		if (lumpnum == LUMPERROR)
			return --wipeframe;
	}
	return --wipeframe;
#endif
}

/** Does the specified wipe exist?
  */
boolean F_WipeExists(UINT8 wipetype)
{
#ifdef NOWIPE
	(void)wipetype;
	return false;
#else
	static char lumpname[10] = "FADEmm00";
	lumpnum_t lumpnum;

	if (wipetype > 99)
		return false;

	sprintf(&lumpname[4], "%.2hu00", (UINT16)wipetype);

	lumpnum = W_CheckNumForName(lumpname);
	return !(lumpnum == LUMPERROR);
#endif
}

boolean F_WipeIsToBlack(UINT8 wipemode)
{
	return g_wipedef_toblack[wipemode];
}

boolean F_WipeIsToWhite(UINT8 wipemode)
{
	return g_wipedef_towhite[wipemode];
}

boolean F_WipeIsToInvert(UINT8 wipemode)
{
	return g_wipedef_toinvert[wipemode];
}

boolean F_WipeIsCrossfade(UINT8 wipemode)
{
	return g_wipedef_crossfade[wipemode];
}