/*
	ALICE SOFT SYSTEM 3 for Win32

	[ AGS - pms loader ]
*/

#include "ags.h"
#include "game_id.h"
#include <string.h>

void AGS::load_pms(uint8* data, bool set_palette, int transparent)
{
	// ヘッダ取得
	int sx = data[0x0] | (data[0x1] << 8);
	int sy = data[0x2] | (data[0x3] << 8);
	int ex = data[0x4] | (data[0x5] << 8);
	int ey = data[0x6] | (data[0x7] << 8);
	int width = ex - sx + 1;
	int height = ey - sy + 1;
	uint8 mask = data[0xa];

	// Jコマンドの処理
	if(set_cg_dest) {
		sx = cg_dest_x;
		sy = cg_dest_y;
		ex = sx + width - 1;
		ey = sy + height - 1;
		set_cg_dest = false;
	}

	if(get_palette) {
		SDL_Color colors[256];
		int p = 0x20;
		for(int i = 0; i < 16; i++) {
			for(int j = 0; j < 16; j++) {
				uint8_t r = data[p++];
				uint8_t g = data[p++];
				uint8_t b = data[p++];
				colors[i * 16 + j] = {r, g, b, 255};
			}
		}
		if (game_id.is_rance4x() || game_id.is(GameId::HASHIRIONNA2)) {
			// 上下16色は取得しない
			for(int i = 1; i < 15; i++) {
				if (!(mask & 1 << i)) {
					SDL_SetPaletteColors(program_palette, colors + i * 16, i * 16, 16);
				}
			}
		} else if (game_id.is(GameId::MUGEN) && transparent != -1) {
			// Uコマンドでは上下32色は取得しない
			for(int i = 2; i < 14; i++) {
				if (!(mask & 1 << i)) {
					SDL_SetPaletteColors(program_palette, colors + i * 16, i * 16, 16);
				}
			}
		} else {
			for(int i = 0; i < 16; i++) {
				if (!(mask & 1 << i)) {
					SDL_SetPaletteColors(program_palette, colors + i * 16, i * 16, 16);
				}
			}
		}
	}

	// パレット展開
	if (set_palette) {
		if (game_id.is_rance4x() || game_id.is(GameId::HASHIRIONNA2)) {
			// 上下16色は展開しない
			for(int i = 1; i < 15; i++) {
				if (!(mask & 1 << i)) {
					SDL_SetPaletteColors(screen_palette, program_palette->colors + i * 16, i * 16, 16);
				}
			}
		} else if (game_id.is(GameId::MUGEN) && transparent != -1) {
			// Uコマンドでは上下32色は展開しない
			for(int i = 2; i < 14; i++) {
				if (!(mask & 1 << i)) {
					SDL_SetPaletteColors(screen_palette, program_palette->colors + i * 16, i * 16, 16);
				}
			}
		} else {
			for(int i = 0; i < 16; i++) {
				if (!(mask & 1 << i)) {
					SDL_SetPaletteColors(screen_palette, program_palette->colors + i * 16, i * 16, 16);
				}
			}
		}
	}

	// PMS展開
	uint8 cgdata[3][640];
	int p = 0x320;
	memset(cgdata, 0, sizeof(cgdata));

	for(int y = 0; y < height; y++) {
		int x = 0;
		while(x < width) {
			uint8 d1 = data[p++];
			if(d1 == 0xff) {
				int length = data[p++] + 3;
				memcpy(cgdata[0] + x, cgdata[1] + x, length);
				x += length;
			} else if(d1 == 0xfe) {
				int length = data[p++] + 3;
				memcpy(cgdata[0] + x, cgdata[2] + x, length);
				x += length;
			} else if(d1 == 0xfd) {
				int length = data[p++] + 4;
				uint8 d2 = data[p++];
				memset(cgdata[0] + x, d2, length);
				x += length;
			} else if(d1 == 0xfc) {
				int length = data[p++] + 3;
				uint8 d2 = data[p++];
				uint8 d3 = data[p++];
				for(int i = 0; i < length; i++) {
					cgdata[0][x++] = d2;
					cgdata[0][x++] = d3;
				}
			} else if(d1 == 0xfb || d1 == 0xfa || d1 == 0xf9 || d1 == 0xf8) {
				cgdata[0][x++] = data[p++];
			} else {
				cgdata[0][x++] = d1;
			}
		}

		// VRAMに転送
		if(extract_cg) {
			uint8_t* dest = &vram[dest_screen][y + sy][sx];
			if(transparent == -1) {
				for(int x = 0; x < width; x++) {
					cgdata[2][x] = cgdata[1][x];
					cgdata[1][x] = cgdata[0][x];
					dest[x] = cgdata[0][x];
				}
			} else {
				for(int x = 0; x < width; x++) {
					cgdata[2][x] = cgdata[1][x];
					cgdata[1][x] = cgdata[0][x];
					if(cgdata[0][x] != transparent) {
						dest[x] = cgdata[0][x];
					}
				}
			}
		}
	}

	// 画面更新
	if(dest_screen == 0 && extract_cg) {
		draw_screen(sx, sy, width, height);
	}
}