/* ScummVM - Graphic Adventure Engine
*
* ScummVM is the legal property of its developers, whose names
* are too numerous to list here. Please refer to the COPYRIGHT
* file distributed with this source distribution.
*
* 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 2
* 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, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
*/

/*
* Based on the Reverse Engineering work of Christophe Fontanel,
* maintainer of the Dungeon Master Encyclopaedia (http://dmweb.free.fr/)
*/
#include "common/system.h"
#include "common/savefile.h"
#include "graphics/thumbnail.h"

#include "dm.h"
#include "dungeonman.h"
#include "timeline.h"
#include "group.h"
#include "champion.h"
#include "menus.h"
#include "eventman.h"
#include "projexpl.h"
#include "dialog.h"
#include <gui/saveload.h>
#include <common/translation.h>


namespace DM {
#define C2_FORMAT_DM_AMIGA_2X_PC98_X68000_FM_TOWNS_CSB_ATARI_ST 2
#define C3_PLATFORM_AMIGA 3
#define C10_DUNGEON_DM 10

LoadgameResponse DMEngine::f435_loadgame(int16 slot) {
	if (slot == -1 && _g298_newGame == k0_modeLoadSavedGame)
		return kM1_LoadgameFailure;

	bool L1366_B_FadePalette = true;
	Common::String fileName;
	Common::SaveFileManager *saveFileManager = nullptr;
	Common::InSaveFile *file = nullptr;

	struct {
		int16 _saveFormat;
		int16 _saveAndPlayChoice;
		int32 _gameId;
		int16 _platform;
		uint16 _dungeonId;
	} dmSaveHeader;

	if (!_g298_newGame) {
		fileName = getSavefileName(slot);
		saveFileManager = _system->getSavefileManager();
		file = saveFileManager->openForLoading(fileName);
	}

	_g528_saveFormat = C2_FORMAT_DM_AMIGA_2X_PC98_X68000_FM_TOWNS_CSB_ATARI_ST;
	_g527_platform = C3_PLATFORM_AMIGA;
	_g526_dungeonId = C10_DUNGEON_DM;
	if (_g298_newGame) {
		//L1366_B_FadePalette = !F0428_DIALOG_RequireGameDiskInDrive_NoDialogDrawn(C0_DO_NOT_FORCE_DIALOG_DM_CSB, true);
		_g524_restartGameAllowed = false;
		_championMan->_g305_partyChampionCount = 0;
		_championMan->_g414_leaderHandObject = Thing::_none;
		_g525_gameId = ((int32)getRandomNumber(65536)) * getRandomNumber(65536);
	} else {
		warning(false, "MISSING CODE: missing check for matching _g525_gameId in f435_loadgame");
		/*if (_vm->_g523_restartGameRequest && (L1372_s_SaveHeader.GameID != _vm->_g525_gameId)) {
			L1367_pc_Message = G0546_pc_THATSNOTTHESAMEGAME;
			goto T0435004;
		}*/


		SaveGameHeader header;
		readSaveGameHeader(file, &header);

		warning(false, "MISSING CODE: missing check for matching format and platform in save in f435_loadgame");


		dmSaveHeader._saveFormat = file->readSint16BE();
		dmSaveHeader._saveAndPlayChoice = file->readSint16BE();
		dmSaveHeader._gameId = file->readSint32BE();
		dmSaveHeader._platform = file->readSint16BE();
		dmSaveHeader._dungeonId = file->readUint16BE();

		_g525_gameId = dmSaveHeader._gameId;

		_g313_gameTime = file->readSint32BE();
		// G0349_ul_LastRandomNumber = L1371_s_GlobalData.LastRandomNumber;
		_championMan->_g305_partyChampionCount = file->readUint16BE();
		_dungeonMan->_g306_partyMapX = file->readSint16BE();
		_dungeonMan->_g307_partyMapY = file->readSint16BE();
		_dungeonMan->_g308_partyDir = (Direction)file->readUint16BE();
		_dungeonMan->_g309_partyMapIndex = file->readByte();
		_championMan->_g411_leaderIndex = (ChampionIndex)file->readSint16BE();
		_championMan->_g514_magicCasterChampionIndex = (ChampionIndex)file->readSint16BE();
		_timeline->_g372_eventCount = file->readUint16BE();
		_timeline->_g373_firstUnusedEventIndex = file->readUint16BE();
		_timeline->_g369_eventMaxCount = file->readUint16BE();
		_groupMan->_g377_currActiveGroupCount = file->readUint16BE();
		_projexpl->_g361_lastCreatureAttackTime = file->readSint32BE();
		_projexpl->_g362_lastPartyMovementTime = file->readSint32BE();
		_g310_disabledMovementTicks = file->readSint16BE();
		_g311_projectileDisableMovementTicks = file->readSint16BE();
		_g312_lastProjectileDisabledMovementDirection = file->readSint16BE();
		_championMan->_g414_leaderHandObject = Thing(file->readUint16BE());
		_groupMan->_g376_maxActiveGroupCount = file->readUint16BE();
		if (!_g523_restartGameRequest) {
			_timeline->f233_initTimeline();
			_groupMan->f196_initActiveGroups();
		}

		_groupMan->load1_ActiveGroupPart(file);
		_championMan->load2_PartyPart(file);
		_timeline->load3_eventsPart(file);
		_timeline->load4_timelinePart(file);

		// read sentinel
		uint32 sentinel = file->readUint32BE();
		assert(sentinel == 0x6f85e3d3);
	}

	_dungeonMan->f434_loadDungeonFile(file);
	delete file;

	if (_g298_newGame) {
		_timeline->f233_initTimeline();
		_groupMan->f196_initActiveGroups();

		if (L1366_B_FadePalette) {
			_displayMan->f436_STARTEND_FadeToPalette(_displayMan->_g345_aui_BlankBuffer);
			f22_delay(1);
			_displayMan->fillScreen(k0_ColorBlack);
			_displayMan->f436_STARTEND_FadeToPalette(_displayMan->_g347_paletteTopAndBottomScreen);
		}
	} else {
		_g528_saveFormat = dmSaveHeader._saveFormat;
		_g527_platform = dmSaveHeader._platform;
		_g526_dungeonId = dmSaveHeader._dungeonId;

		_g524_restartGameAllowed = true;

		switch (getGameLanguage()) { // localized
		default:
		case Common::EN_ANY:
			_dialog->f427_dialogDraw(nullptr, "LOADING GAME . . .", nullptr, nullptr, nullptr, nullptr, true, true, true);
			break;
		case Common::DE_DEU:
			_dialog->f427_dialogDraw(nullptr, "SPIEL WIRD GELADEN . . .", nullptr, nullptr, nullptr, nullptr, true, true, true);
			break;
		case Common::FR_FRA:
			_dialog->f427_dialogDraw(nullptr, "CHARGEMENT DU JEU . . .", nullptr, nullptr, nullptr, nullptr, true, true, true);
			break;
		}

	}
	_championMan->_g303_partyDead = false;

	return k1_LoadgameSuccess;
}


void DMEngine::f433_processCommand140_saveGame() {
	_menuMan->f456_drawDisabledMenu();
	_eventMan->f78_showMouse();

	switch (getGameLanguage()) { // localized
	default:
	case Common::EN_ANY:
		_dialog->f427_dialogDraw(nullptr, nullptr, "SAVE AND PLAY", "SAVE AND QUIT", "CANCEL", "LOAD", false, false, false);
		break;
	case Common::DE_DEU:
		_dialog->f427_dialogDraw(nullptr, nullptr, "SICHERN/SPIEL", "SICHERN/ENDEN", "WIDERRUFEN", "LOAD", false, false, false);
		break;
	case Common::FR_FRA:
		_dialog->f427_dialogDraw(nullptr, nullptr, "GARDER/JOUER", "GARDER/SORTIR", "ANNULLER", "LOAD", false, false, false);
		break;
	}

	enum SaveAndPlayChoice {
		kSaveAndPlay = 1,
		kSaveAndQuit = 2,
		kCancel = 3,
		kLoad = 4
	};

	SaveAndPlayChoice saveAndPlayChoice = (SaveAndPlayChoice)_dialog->f424_dialogGetChoice(4, k0_DIALOG_SET_VIEWPORT, 0, k0_DIALOG_CHOICE_NONE);

	if (saveAndPlayChoice == kLoad) {
		GUI::SaveLoadChooser *dialog = new GUI::SaveLoadChooser(_("Restore game:"), _("Restore"), false);
		int loadSlot = dialog->runModalWithCurrentTarget();
		if (loadSlot >= 0) {
			_loadSaveSlotAtRuntime = loadSlot;
			return;
		}

		saveAndPlayChoice = kCancel;
	}

	if (saveAndPlayChoice == kSaveAndQuit || saveAndPlayChoice == kSaveAndPlay) {
		GUI::SaveLoadChooser *dialog = new GUI::SaveLoadChooser(_("Save game:"), _("Save"), true);
		int16 saveSlot = dialog->runModalWithCurrentTarget();
		Common::String saveDescription = dialog->getResultString();
		if (saveDescription.empty())
			saveDescription = "Nice save ^^";
		delete dialog;

		if (saveSlot >= 0) {
			switch (getGameLanguage()) { // localized
			default:
			case Common::EN_ANY:
				_dialog->f427_dialogDraw(nullptr, "SAVING GAME . . .", nullptr, nullptr, nullptr, nullptr, false, false, false);
				break;
			case Common::DE_DEU:
				_dialog->f427_dialogDraw(nullptr, "SPIEL WIRD GESICHERT . . .", nullptr, nullptr, nullptr, nullptr, false, false, false);
				break;
			case Common::FR_FRA:
				_dialog->f427_dialogDraw(nullptr, "UN MOMENT A SAUVEGARDER DU JEU...", nullptr, nullptr, nullptr, nullptr, false, false, false);
				break;
			}

			uint16 champHandObjWeight = 0;
			if (!_championMan->_g415_leaderEmptyHanded) {
				champHandObjWeight = _dungeonMan->f140_getObjectWeight(_championMan->_g414_leaderHandObject);
				_championMan->_gK71_champions[_championMan->_g411_leaderIndex]._load -= champHandObjWeight;
			}

			if (!writeCompleteSaveFile(saveSlot, saveDescription, saveAndPlayChoice)) {
				_dialog->f427_dialogDraw(nullptr, "Unable to open file for saving", "OK", nullptr, nullptr, nullptr, false, false, false);
				_dialog->f424_dialogGetChoice(1, k0_DIALOG_SET_VIEWPORT, 0, k0_DIALOG_CHOICE_NONE);
			}

			if (!_championMan->_g415_leaderEmptyHanded) {
				_championMan->_gK71_champions[_championMan->_g411_leaderIndex]._load += champHandObjWeight;
			}
		} else
			saveAndPlayChoice = kCancel;
	}


	if (saveAndPlayChoice == kSaveAndQuit) {
		_eventMan->f77_hideMouse();
		f444_endGame(false);
	}

	_g524_restartGameAllowed = true;
	_menuMan->f457_drawEnabledMenus();
	_eventMan->f77_hideMouse();
}

Common::String DMEngine::getSavefileName(uint16 slot) {
	return Common::String::format("%s.%03u", _targetName.c_str(), slot);
}

#define SAVEGAME_ID       MKTAG('D', 'M', 'D', 'M')
#define SAVEGAME_VERSION  1

void DMEngine::writeSaveGameHeader(Common::OutSaveFile* out, const Common::String& saveName) {
	out->writeUint32BE(SAVEGAME_ID);

	// Write version
	out->writeByte(SAVEGAME_VERSION);

	// Write savegame name
	out->writeString(saveName);
	out->writeByte(0);

	// Save the game thumbnail
	if (_saveThumbnail) {
		out->write(_saveThumbnail->getData(), _saveThumbnail->size());
	} else
		Graphics::saveThumbnail(*out);

	// Creation date/time
	TimeDate curTime;
	_system->getTimeAndDate(curTime);

	uint32 saveDate = ((curTime.tm_mday & 0xFF) << 24) | (((curTime.tm_mon + 1) & 0xFF) << 16) | ((curTime.tm_year + 1900) & 0xFFFF);
	uint16 saveTime = ((curTime.tm_hour & 0xFF) << 8) | ((curTime.tm_min) & 0xFF);
	uint32 playTime = getTotalPlayTime() / 1000;

	out->writeUint32BE(saveDate);
	out->writeUint16BE(saveTime);
	out->writeUint32BE(playTime);
}

bool DMEngine::writeCompleteSaveFile(int16 saveSlot, Common::String& saveDescription, int16 saveAndPlayChoice) {
	Common::String savefileName = getSavefileName(saveSlot);
	Common::SaveFileManager *saveFileManager = _system->getSavefileManager();
	Common::OutSaveFile *file = saveFileManager->openForSaving(savefileName);

	if (!file) {
		return false;
	}

	writeSaveGameHeader(file, saveDescription);

	file->writeSint16BE(_g528_saveFormat);
	file->writeSint16BE(saveAndPlayChoice);
	file->writeSint32BE(_g525_gameId);
	file->writeSint16BE(_g527_platform);
	file->writeUint16BE(_g526_dungeonId);

	// write C0_SAVE_PART_GLOBAL_DATA part
	file->writeSint32BE(_g313_gameTime);
	//L1348_s_GlobalData.LastRandomNumber = G0349_ul_LastRandomNumber;
	file->writeUint16BE(_championMan->_g305_partyChampionCount);
	file->writeSint16BE(_dungeonMan->_g306_partyMapX);
	file->writeSint16BE(_dungeonMan->_g307_partyMapY);
	file->writeUint16BE(_dungeonMan->_g308_partyDir);
	file->writeByte(_dungeonMan->_g309_partyMapIndex);
	file->writeSint16BE(_championMan->_g411_leaderIndex);
	file->writeSint16BE(_championMan->_g514_magicCasterChampionIndex);
	file->writeUint16BE(_timeline->_g372_eventCount);
	file->writeUint16BE(_timeline->_g373_firstUnusedEventIndex);
	file->writeUint16BE(_timeline->_g369_eventMaxCount);
	file->writeUint16BE(_groupMan->_g377_currActiveGroupCount);
	file->writeSint32BE(_projexpl->_g361_lastCreatureAttackTime);
	file->writeSint32BE(_projexpl->_g362_lastPartyMovementTime);
	file->writeSint16BE(_g310_disabledMovementTicks);
	file->writeSint16BE(_g311_projectileDisableMovementTicks);
	file->writeSint16BE(_g312_lastProjectileDisabledMovementDirection);
	file->writeUint16BE(_championMan->_g414_leaderHandObject.toUint16());
	file->writeUint16BE(_groupMan->_g376_maxActiveGroupCount);

	// write C1_SAVE_PART_ACTIVE_GROUP part
	_groupMan->save1_ActiveGroupPart(file);
	// write C2_SAVE_PART_PARTY part
	_championMan->save2_PartyPart(file);
	// write C3_SAVE_PART_EVENTS part
	_timeline->save3_eventsPart(file);
	// write C4_SAVE_PART_TIMELINE part
	_timeline->save4_timelinePart(file);

	// write sentinel
	file->writeUint32BE(0x6f85e3d3);

	// save _g278_dungeonFileHeader
	{
		DungeonFileHeader &header = _dungeonMan->_g278_dungeonFileHeader;
		file->writeUint16BE(header._ornamentRandomSeed);
		file->writeUint16BE(header._rawMapDataSize);
		file->writeByte(header._mapCount);
		file->writeByte(0); // to match the structure of dungeon.dat, will be discarded
		file->writeUint16BE(header._textDataWordCount);
		file->writeUint16BE(header._partyStartLocation);
		file->writeUint16BE(header._squareFirstThingCount);
		for (uint16 i = 0; i < 16; ++i)
			file->writeUint16BE(header._thingCounts[i]);
	}

	// save _g277_dungeonMaps
	for (uint16 i = 0; i < _dungeonMan->_g278_dungeonFileHeader._mapCount; ++i) {
		Map &map = _dungeonMan->_g277_dungeonMaps[i];
		uint16 tmp;

		file->writeUint16BE(map._rawDunDataOffset);
		file->writeUint32BE(0); // to match the structure of dungeon.dat, will be discarded
		file->writeByte(map._offsetMapX);
		file->writeByte(map._offsetMapY);

		tmp = ((map._height & 0x1F) << 11) | ((map._width & 0x1F) << 6) | (map._level & 0x3F);
		file->writeUint16BE(tmp);

		tmp = ((map._randFloorOrnCount & 0xF) << 12) | ((map._floorOrnCount & 0xF) << 8)
			| ((map._randWallOrnCount & 0xF) << 4) | (map._wallOrnCount & 0xF);
		file->writeUint16BE(tmp);

		tmp = ((map._difficulty & 0xF) << 12) | ((map._creatureTypeCount & 0xF) << 4) | (map._doorOrnCount & 0xF);
		file->writeUint16BE(tmp);

		tmp = ((map._doorSet1 & 0xF) << 12) | ((map._doorSet0 & 0xF) << 8)
			| ((map._wallSet & 0xF) << 4) | (map._floorSet & 0xF);
		file->writeUint16BE(tmp);
	}

	// save _g280_dungeonColumnsCumulativeSquareThingCount
	for (uint16 i = 0; i < _dungeonMan->_g282_dungeonColumCount; ++i)
		file->writeUint16BE(_dungeonMan->_g280_dungeonColumnsCumulativeSquareThingCount[i]);

	// save _g283_squareFirstThings
	for (uint16 i = 0; i < _dungeonMan->_g278_dungeonFileHeader._squareFirstThingCount; ++i)
		file->writeUint16BE(_dungeonMan->_g283_squareFirstThings[i].toUint16());

	// save _g260_dungeonTextData
	for (uint16 i = 0; i < _dungeonMan->_g278_dungeonFileHeader._textDataWordCount; ++i)
		file->writeUint16BE(_dungeonMan->_g260_dungeonTextData[i]);

	// save _g284_thingData
	for (uint16 thingIndex = 0; thingIndex < 16; ++thingIndex)
		for (uint16 i = 0; i < g235_ThingDataWordCount[thingIndex] * _dungeonMan->_g278_dungeonFileHeader._thingCounts[thingIndex]; ++i)
			file->writeUint16BE(_dungeonMan->_g284_thingData[thingIndex][i]);

	// save _g276_dungeonRawMapData
	for (uint32 i = 0; i < _dungeonMan->_g278_dungeonFileHeader._rawMapDataSize; ++i)
		file->writeByte(_dungeonMan->_g276_dungeonRawMapData[i]);

	file->flush();
	file->finalize();
	delete file;
	
	return true;
}

bool readSaveGameHeader(Common::InSaveFile* in, SaveGameHeader* header) {
	uint32 id = in->readUint32BE();

	// Check if it's a valid ScummVM savegame
	if (id != SAVEGAME_ID)
		return false;

	// Read in the version
	header->_version = in->readByte();

	// Check that the save version isn't newer than this binary
	if (header->_version > SAVEGAME_VERSION)
		return false;

	// Read in the save name
	Common::String saveName;
	char ch;
	while ((ch = (char)in->readByte()) != '\0')
		saveName += ch;
	header->_descr.setDescription(saveName);

	// Get the thumbnail
	header->_descr.setThumbnail(Graphics::loadThumbnail(*in));

	uint32 saveDate = in->readUint32BE();
	uint16 saveTime = in->readUint16BE();
	uint32 playTime = in->readUint32BE();

	int day = (saveDate >> 24) & 0xFF;
	int month = (saveDate >> 16) & 0xFF;
	int year = saveDate & 0xFFFF;
	header->_descr.setSaveDate(year, month, day);

	int hour = (saveTime >> 8) & 0xFF;
	int minutes = saveTime & 0xFF;
	header->_descr.setSaveTime(hour, minutes);

	header->_descr.setPlayTime(playTime * 1000);
	if (g_engine)
		g_engine->setTotalPlayTime(playTime * 1000);

	return true;
}

}