/*
 *   This file is part of fastboot 3DS
 *   Copyright (C) 2017 derrek, profi200
 *
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   This program 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 General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "types.h"
#include "arm11/menu/splash.h"
#include "arm11/lz11.h"
#include "hardware/gfx.h"



void getSplashDimensions(const void *const data, u32 *const width, u32 *const height)
{
	const SplashHeader *const header = (const SplashHeader *const)data;

	if(width) *width = header->width;
	if(height) *height = header->height;
}

static bool validateSplashHeader(const SplashHeader *const header, u8 screen)
{
	if(memcmp(&header->magic, "SPLA", 4)) return false;

	const u16 width = header->width, height = header->height;
	if(width == 0 || height == 0) return false;
	if(width > (screen ? SCREEN_WIDTH_BOT : SCREEN_WIDTH_TOP) || height > SCREEN_HEIGHT_TOP)
		return false;

	const u32 flags = header->flags;
	if((flags & FORMAT_INVALID) != FORMAT_RGB565) return false;
	if(!(flags & FLAG_ROTATED) || flags & FLAG_SWAPPED) return false;

	return true;
}

bool drawSplashscreen(const void *const data, u16 *const tmpBuf, s32 startX, s32 startY, u8 screen)
{
	if(!data || !tmpBuf) return false;

	const SplashHeader *const header = (const SplashHeader *const)data;
	if(!validateSplashHeader(header, screen)) return false;

	const u16 width = header->width, height = header->height;
	const u32 flags = header->flags;
	const bool isCompressed = flags & FLAG_COMPRESSED;

	u16 *imgData;
	if(isCompressed)
	{
		imgData = tmpBuf;
		lz11Decompress(data + sizeof(SplashHeader), imgData, width * height * 2);
	}
	else imgData = (u16*)(data + sizeof(SplashHeader));

	u32 screenWidth, screenHeight, xx, yy;
	if(!screen)	/* SCREEN_TOP */
	{
		screenWidth = SCREEN_WIDTH_TOP;
		screenHeight = SCREEN_HEIGHT_TOP;
	}
	else	/* SCREEN_BOT */
	{
		screenWidth = SCREEN_WIDTH_BOT;
		screenHeight = SCREEN_HEIGHT_BOT;
	}
	if(startX < 0 || (u32)startX > screenWidth - width) xx = (screenWidth - width) / 2;
	else xx = (u32)startX;
	if(startY < 0 || (u32)startY > screenHeight - height) yy = (screenHeight - height) / 2;
	else yy = (u32)startY;

	u16 *fb = (screen ? (u16*)RENDERBUF_BOT : (u16*)RENDERBUF_TOP);
	for(u32 x = 0; x < width; x++)
	{
		for(u32 y = 0; y < height; y++)
		{
			fb[(x + xx) * screenHeight + y + yy] = imgData[x * height + y];
		}
	}

	return true;
}