/*
	ALICE SOFT SYSTEM 3 for Win32

	[ DRI ]
*/

#include "dri.h"
#include <memory>
#include <string.h>
#include "game_id.h"
#include "fileio.h"

void Dri::open(const char* file_name)
{
	link_table.clear();
	fname = file_name;

	auto fio = FILEIO::open(file_name, FILEIO_READ_BINARY);
	if (!fio)
		return;

	int link_sector = fio->getw();
	int data_sector = fio->getw();

	link_table.resize((data_sector - link_sector) * 256);
	fio->seek((link_sector - 1) * 256, SEEK_SET);
	fio->read(link_table.data(), 256 * (data_sector - link_sector));
}

std::vector<uint8> Dri::load(int page)
{
	if (static_cast<size_t>(page) >= link_table.size() / 2)
		return {};

	int disk_index = link_table[(page - 1) * 2];
	int link_index = link_table[(page - 1) * 2 + 1];
	if (disk_index == 0 || disk_index == 0x1a)
		return {};

	fname[0] = 'A' + disk_index - 1;
	auto fio = FILEIO::open(fname.c_str(), FILEIO_READ_BINARY);
	if (!fio)
		return {};

	fio->seek(link_index * 2, SEEK_SET);
	int start_sector = fio->getw();
	int end_sector = fio->getw();
	if (end_sector <= start_sector)
		return {};

	std::vector<uint8_t> buffer((end_sector - start_sector) * 256);
	fio->seek((start_sector - 1) * 256, SEEK_SET);
	fio->read(buffer.data(), 256 * (end_sector - start_sector));

	return buffer;
}

// static
std::vector<uint8> Dri::load_mda(const GameId& game_id, int page)
{
	// データ取得
	const char* name = NULL;

	switch (game_id.game) {
		case GameId::BUNKASAI:		// あぶない文化祭前夜
			name = "AMUS_AB.MDA";
			break;
		case GameId::CRESCENT:		// クレセントムーンがぁる
			name = "AMUS_CRS.MDA";
			break;
		case GameId::DPS:			// D.P.S. - Dream Program System
			name = "AMUS_DPS.MDA";
			break;
		case GameId::DPS_SG_FAHREN:
		case GameId::DPS_SG_KATEI:
		case GameId::DPS_SG_NOBUNAGA:
		case GameId::DPS_SG2_ANTIQUE:
		case GameId::DPS_SG2_IKENAI:
		case GameId::DPS_SG2_AKAI:
		case GameId::DPS_SG3_RABBIT:
		case GameId::DPS_SG3_SHINKON:
		case GameId::DPS_SG3_SOTSUGYOU:
			name = "AMUS_SG.MDA";
			break;
		case GameId::FUKEI:		// 婦警さんＶＸ
			name = "AMUS_VX.MDA";
			break;
		case GameId::INTRUDER:		// Intruder -桜屋敷の探索-
			name = "AMUS_INT";
			break;
		case GameId::TENGU:		// あぶないてんぐ伝説
			name = "AMUS_AT";
			break;
		case GameId::TOUSHIN_HINT:	// 闘神都市 ヒントディスク
			name = "AMUS_T1";
			break;
		case GameId::LITTLE_VAMPIRE:		// Little Vampire
			name = "AMUS_LP2";
			break;
		case GameId::YAKATA:		// ALICEの館
			name = "AMUS_AL1";
			break;

		case GameId::AYUMI_FD:		// あゆみちゃん物語 (FD)
		case GameId::AYUMI_HINT:		// あゆみちゃん物語 ヒントディスク
		case GameId::AYUMI_PROTO:		// あゆみちゃん物語 PROTO
			name = "AMUS_AYM";
			break;
		case GameId::DALK_HINT:		// DALK ヒントディスク
			name = "AMUS_DLK";
			break;
		case GameId::DRSTOP:		// Dr. STOP!
			name = "AMUS_DRS";
			break;
		case GameId::PROG_FD:	// Prostudent -G- (FD)
			name = "AMUS_PSG";
			break;
		case GameId::RANCE3_HINT:		// Rance3 ヒントディスク
			name = "AMUS_R3H";
			break;
		case GameId::SDPS_MARIA:		// Super D.P.S
		case GameId::SDPS_TONO:
		case GameId::SDPS_KAIZOKU:
			name = "AMUS_SDP";
			break;
		case GameId::YAKATA2:		// ALICEの館II
			name = "AMUS_AL2";
			break;

		case GameId::AMBIVALENZ_FD:	// AmbivalenZ (FD)
		case GameId::AMBIVALENZ_CD:	// AmbivalenZ (CD)
			name = "AMUS_AMB.MDA";
			break;
		case GameId::DPS_ALL:		// DPS全部
			name = "AMUS_ALL.MDA";
			break;
		case GameId::FUNNYBEE_CD:		// Funny Bee (CD)
		case GameId::FUNNYBEE_FD:		// Funny Bee (FD)
			name = "AMUS_BEE.MDA";
			break;
		case GameId::ONLYYOU:		// Only You
		case GameId::ONLYYOU_DEMO:	// Only You (DEMO)
			name = "AMUS_OY.MDA";
			break;
		case GameId::PROG_CD:	// Prostudent -G- (CD)
			name = "AMUS_PSG.MDA";
			break;
		case GameId::RANCE41:		// Rance 4.1
			name = "AMUS_R41.MDA";
			break;
		case GameId::RANCE42:		// Rance 4.2
			name = "AMUS_R42.MDA";
			break;
		case GameId::AYUMI_CD:		// あゆみちゃん物語 (CD)
		case GameId::AYUMI_LIVE_256:	// あゆみちゃん物語 実写版
		case GameId::AYUMI_LIVE_FULL:	// あゆみちゃん物語 フルカラー実写版
			name = "AMUS_AYM.MDA";
			break;
		case GameId::YAKATA3_CD:		// アリスの館３ (CD)
		case GameId::YAKATA3_FD:		// アリスの館３ (FD)
			name = "AMUS_AL3.MDA";
			break;
		case GameId::HASHIRIONNA2:	// 走り女２ (Rance 4.x ヒントディスク)
			name = "AMUS_RG2.MDA";
			break;
		case GameId::TOUSHIN2_GD:		// 闘神都市２ グラフィックディスク
			name = "AMUS_T2";
			break;
		case GameId::OTOME:		// 乙女戦記
			name = "AMUS_OTM.MDA";
			break;
		case GameId::MUGEN:		// 夢幻泡影
			name = "AMUS_MGN.MDA";
			break;
	}

	if(name == NULL) {
		return {};
	}

	SDL_RWops* rw = open_resource(name, "mda");
	if (!rw)
		return {};
	uint8 buf[4];

	// ページの位置を取得
	SDL_RWread(rw, buf, 4, 1);
	int link_sector = buf[0] | (buf[1] << 8);
	int data_sector = buf[2] | (buf[3] << 8);

	if(page > (data_sector - link_sector) * 128 - 1) {
		// ページ番号不正
		SDL_RWclose(rw);
		return {};
	}

	SDL_RWseek(rw, (link_sector - 1) * 256 + (page - 1) * 2, RW_SEEK_SET);
	SDL_RWread(rw, buf, 2, 1);

	int disk_index = buf[0];
	int link_index = buf[1];

	if(disk_index == 0 || disk_index == 0x1a) {
		// 欠番
		SDL_RWclose(rw);
		return {};
	}

	// AMUS.MDA以外にリンクされている場合はリソースを開き直す
	if(disk_index == 2) {
		SDL_RWclose(rw);
		switch (game_id.game) {
			case GameId::DPS_SG_FAHREN:	// D.P.S. SG - Fahren Fliegen
				name = "BMUS_FAH";
				break;
			case GameId::DPS_SG_KATEI:	// D.P.S. SG - 家庭教師はステキなお仕事
				name = "BMUS_KAT";
				break;
			case GameId::DPS_SG_NOBUNAGA:	// D.P.S. SG - 信長の淫謀
				name = "BMUS_NOB";
				break;
			case GameId::DPS_SG2_ANTIQUE:	// D.P.S. SG set2 - ANTIQUE HOUSE
				name = "BMUS_ANT";
				break;
			case GameId::DPS_SG2_IKENAI:	// D.P.S. SG set2 - いけない内科検診再び
				name = "BMUS_NAI";
				break;
			case GameId::DPS_SG2_AKAI:	// D.P.S. SG set2 - 朱い夜
				name = "BMUS_AKA";
				break;
			case GameId::DPS_SG3_RABBIT:	// D.P.S. SG set3 - Rabbit P4P
				name = "BMUS_RAB";
				break;
			case GameId::DPS_SG3_SHINKON:	// D.P.S. SG set3 - しんこんさんものがたり
				name = "BMUS_SIN";
				break;
			case GameId::DPS_SG3_SOTSUGYOU:	// D.P.S. SG set3 - 卒業
				name = "BMUS_SOT";
				break;
			case GameId::SDPS_MARIA:		// Super D.P.S - マリアとカンパン
				name = "BMUS_MTK";
				break;
			case GameId::SDPS_TONO:		// Super D.P.S - 遠野の森
				name = "BMUS_TNM";
				break;
			case GameId::SDPS_KAIZOKU:	// Super D.P.S - うれしたのし海賊稼業
				name = "BMUS_KAM";
				break;
			default:
				name = NULL;
				break;
		}
		if(name == NULL) {
			return {};
		}
		if((rw = open_resource(name, "mda")) == NULL) {
			return {};
		}
	} else if(disk_index == 3) {
		SDL_RWclose(rw);
		switch (game_id.game) {
			case GameId::TOUSHIN_HINT:	// 闘神都市 ヒントディスク
				name = "CMUS_T1";
				break;
			default:
				name = NULL;
				break;
		}
		if(name == NULL) {
			return {};
		}
		if((rw = open_resource(name, "mda")) == NULL) {
			return {};
		}
	} else if(disk_index != 1) {
		// AMUS.MDA以外にリンクされている場合は失敗
		SDL_RWclose(rw);
		return {};
	}

	// データ取得
	SDL_RWseek(rw, link_index * 2, RW_SEEK_SET);
	SDL_RWread(rw, buf, 4, 1);
	int start_sector = buf[0] | (buf[1] << 8);
	int end_sector = buf[2] | (buf[3] << 8);

	int size = (end_sector - start_sector) * 256;
	if (size == 0) {
		// サイズ不正
		SDL_RWclose(rw);
		return {};
	}
	std::vector<uint8_t> buffer(size);
	SDL_RWseek(rw, (start_sector - 1) * 256, RW_SEEK_SET);
	SDL_RWread(rw, buffer.data(), size, 1);

	SDL_RWclose(rw);

	return buffer;
}