/*
 *   This file is part of Universal-Core
 *   Copyright (C) 2021 Universal-Team
 *
 *   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/>.
 *
 *   Additional Terms 7.b and 7.c of GPLv3 apply to this file:
 *       * Requiring preservation of specified reasonable legal notices or
 *         author attributions in that material or in the Appropriate Legal
 *         Notices displayed by works containing it.
 *       * Prohibiting misrepresentation of the origin of that material,
 *         or requiring that modified versions of such material be marked in
 *         reasonable ways as different from the original version.
 */

#include "image.hpp"

#include "tonccpy.h"

Image::Image(const std::vector<std::string> &paths) {
	// Attempt to load from the given paths
	FILE *file = nullptr;
	for(const auto &path : paths) {
		file = fopen(path.c_str(), "rb");
		if(file)
			break;
	}

	// Return if no file found
	if(!file)
		return;

	// Return if not in the correct format
	char magic[5] = {0};
	fread(magic, 1, 4, file);
	if(strcmp(magic, ".GFX"))
		return;

	// Read image
	fread(&_width, 1, 2, file);
	fread(&_height, 1, 2, file);
	_bitmap = std::vector<u8>(_width * _height);
	fread(_bitmap.data(), 1, _bitmap.size(), file);
	u16 palCount;
	fread(&palCount, 1, 2, file);
	fread(&_palOfs, 1, 2, file);
	_palette = std::vector<u16>(palCount);
	fread(_palette.data(), 2, palCount, file);
	fclose(file);
}

Image::Image(FILE *file) {
	// Return if no file
	if(!file)
		return;

	// Return if not in the correct format
	char magic[5];
	fread(magic, 1, 4, file);
	if(strcmp(magic, ".GFX"))
		return;

	// Read image
	fread(&_width, 1, 2, file);
	fread(&_height, 1, 2, file);
	_bitmap = std::vector<u8>(_width * _height);
	fread(_bitmap.data(), 1, _bitmap.size(), file);
	u16 palCount;
	fread(&palCount, 1, 2, file);
	fread(&_palOfs, 1, 2, file);
	_palette = std::vector<u16>(palCount);
	fread(_palette.data(), 2, palCount, file);
	fclose(file);
}

void Image::draw(int x, int y, bool top, int layer, int channel, bool copyPal) {
	if(copyPal)
		tonccpy((top ? BG_PALETTE : BG_PALETTE_SUB) + _palOfs, _palette.data(), _palette.size() * 2);

	u8 *dst   = (u8 *)bgGetGfxPtr(top ? layer : layer + 4);
	int width = 256;

	// If full width and X is 0, copy it all in one go
	if(_width == width && x == 0) {
		dmaCopyHalfWords(channel, _bitmap.data(), dst, _width * _height);
	} else {
		for(int i = 0; i < _height; i++) {
			dmaCopyHalfWords(channel, _bitmap.data() + (i * _width), dst + (y + i) * 256 + x, _width);
		}
	}
}

void Image::drawSpecial(int x, int y, bool top, int layer, float scaleX, float scaleY, int paletteOffset,
						bool copyPal) {
	if(copyPal)
		tonccpy((top ? BG_PALETTE : BG_PALETTE_SUB) + _palOfs + paletteOffset, _palette.data(), _palette.size() * 2);

	u8 *dst = (u8 *)bgGetGfxPtr(top ? layer : layer + 4);

	// If the scale is 1 use faster integer math
	if(scaleX == 1.0f && scaleY == 1.0f) {
		for(int i = 0; i < _height; i++) {
			for(float j = 0; j < _width; j++) {
				u8 px = _bitmap[i * _width + j];
				if(_palette[px - _palOfs] & 0x8000)
					toncset(dst + int((y + i) * 256 + x + j), px + paletteOffset, 1);
			}
		}
	} else {
		for(float i = 0.0f; i < _height; i += 1 / scaleY) {
			for(float j = 0.0f; j < _width; j += 1 / scaleX) {
				u8 px = _bitmap[int((i * _width) + j)];
				if(_palette[px - _palOfs] & 0x8000)
					toncset(dst + int((y + i) * 256 + x + j), px + paletteOffset, 1);
			}
		}
	}
}

void Image::drawSegment(int x, int y, int imageX, int imageY, int w, int h, bool top, int layer, int channel,
						bool copyPal) {
	if(copyPal)
		tonccpy((top ? BG_PALETTE : BG_PALETTE_SUB) + _palOfs, _palette.data(), _palette.size() * 2);

	for(int i = 0; i < h; i++) {
		dmaCopyHalfWords(channel, _bitmap.data() + ((imageY + i) * _width) + imageX,
						 (u8 *)bgGetGfxPtr(top ? layer : layer + 4) + ((y + i) * 256) + x, w);
	}
}

void Image::drawSegmentSpecial(int x, int y, int imageX, int imageY, int w, int h, bool top, int layer, float scaleX,
							   float scaleY, int paletteOffset, bool copyPal) {
	if(copyPal)
		tonccpy((top ? BG_PALETTE : BG_PALETTE_SUB) + _palOfs + paletteOffset, _palette.data(), _palette.size() * 2);

	u8 *dst = (u8 *)bgGetGfxPtr(top ? layer : layer + 4);

	// If the scale is 1 use faster integer math
	if(scaleX == 1.0f && scaleY == 1.0f) {
		for(int i = 0; i < h; i++) {
			for(int j = 0; j < w; j++) {
				u8 px = _bitmap[i * _width + j];
				if(_palette[px - _palOfs] & 0x8000)
					toncset(dst + ((y + i) * 256 + x + j), px + paletteOffset, 1);
			}
		}
	} else {
		for(float i = 0; i < h; i += 1 / scaleX) {
			for(float j = 0; j < _width; j += 1 / scaleY) {
				u8 px = _bitmap[i * _width + j];
				if(_palette[px - _palOfs] & 0x8000)
					toncset(dst + int((y + i) * 256 + x + j), px + paletteOffset, 1);
			}
		}
	}
}