TombEngine/TR5Main/Scripting/GameLogicScript.h

#pragma once

#include <sol.hpp>
#include <vector>
#include <string>
#include <iostream>
#include <fstream>
#include <map>
#include "..\Global\global.h"
#include "..\Specific\IO\ChunkId.h"
#include "..\Specific\IO\ChunkReader.h"
#include "..\Specific\IO\ChunkWriter.h"
#include "..\Specific\IO\LEB128.h"
#include "..\Specific\IO\Streams.h"

using namespace std;

#define ITEM_PARAM_currentAnimState		0
#define ITEM_PARAM_goalAnimState			1
#define ITEM_PARAM_REQUIRED_ANIM_STATE		2
#define ITEM_PARAM_frameNumber				3
#define ITEM_PARAM_animNumber				4
#define ITEM_PARAM_hitPoints				5
#define ITEM_PARAM_HIT_STATUS				6
#define ITEM_PARAM_GRAVITY_STATUS			7
#define ITEM_PARAM_COLLIDABLE				8
#define ITEM_PARAM_POISONED					9
#define ITEM_PARAM_roomNumber				10

#define LUA_VARIABLE_TYPE_INT				0
#define LUA_VARIABLE_TYPE_BOOL				1
#define LUA_VARIABLE_TYPE_FLOAT				2
#define LUA_VARIABLE_TYPE_STRING			3

typedef struct LuaFunction {
	string Name;
	string Code;
	bool Executed;
};

typedef struct GameScriptItemPosition {
	int x;
	int y;
	int z;
	short xRot;
	short yRot;
	short zRot;
	short room;
};

typedef struct GameScriptItem {
	ITEM_INFO*		NativeItem;

	short Get(int param)
	{
		if (NativeItem == NULL)
			return 0;

		switch (param)
		{
		case ITEM_PARAM_currentAnimState:
			return NativeItem->currentAnimState;
		case ITEM_PARAM_REQUIRED_ANIM_STATE:
			return NativeItem->requiredAnimState;
		case ITEM_PARAM_goalAnimState:
			return NativeItem->goalAnimState;
		case ITEM_PARAM_animNumber:
			return NativeItem->animNumber;
		case ITEM_PARAM_frameNumber:
			return NativeItem->frameNumber;
		case ITEM_PARAM_hitPoints:
			return NativeItem->hitPoints;
		case ITEM_PARAM_HIT_STATUS:
			return NativeItem->hitStatus;
		case ITEM_PARAM_GRAVITY_STATUS:
			return NativeItem->gravityStatus;
		case ITEM_PARAM_COLLIDABLE:
			return NativeItem->collidable;
		case ITEM_PARAM_POISONED:
			return NativeItem->poisoned;
		case ITEM_PARAM_roomNumber:
			return NativeItem->roomNumber;
		default:
			return 0;
		}
	}

	void Set(int param, short value)
	{
		if (NativeItem == NULL)
			return;

		switch (param)
		{
		case ITEM_PARAM_currentAnimState:
			NativeItem->currentAnimState = value; break;
		case ITEM_PARAM_REQUIRED_ANIM_STATE:
			NativeItem->requiredAnimState = value; break;
		case ITEM_PARAM_goalAnimState:
			NativeItem->goalAnimState = value; break;
		case ITEM_PARAM_animNumber:
			NativeItem->animNumber = value;
			//NativeItem->frameNumber = Anims[NativeItem->animNumber].frameBase;
			break;
		case ITEM_PARAM_frameNumber:
			NativeItem->frameNumber = value; break;
		case ITEM_PARAM_hitPoints:
			NativeItem->hitPoints = value; break;
		case ITEM_PARAM_HIT_STATUS:
			NativeItem->hitStatus = value; break;
		case ITEM_PARAM_GRAVITY_STATUS:
			NativeItem->gravityStatus = value; break;
		case ITEM_PARAM_COLLIDABLE:
			NativeItem->collidable = value; break;
		case ITEM_PARAM_POISONED:
			NativeItem->poisoned = value; break;
		case ITEM_PARAM_roomNumber:
			NativeItem->roomNumber = value; break;
		default:
			break;
		}
	}

	GameScriptItemPosition GetItemPosition()
	{
		GameScriptItemPosition pos;
		
		//if (m_item == NULL)
		//	return pos;

		pos.x = NativeItem->pos.xPos;
		pos.y = NativeItem->pos.yPos;
		pos.z = NativeItem->pos.zPos;
		pos.xRot = TR_ANGLE_TO_DEGREES(NativeItem->pos.xRot);
		pos.yRot = TR_ANGLE_TO_DEGREES(NativeItem->pos.yRot);
		pos.zRot = TR_ANGLE_TO_DEGREES(NativeItem->pos.zRot);
		pos.room = NativeItem->roomNumber;

		return pos;
	}

	void SetItemPosition(int x, int y, int z)
	{
		if (NativeItem == NULL)
			return;

		NativeItem->pos.xPos = x;
		NativeItem->pos.yPos = y;
		NativeItem->pos.zPos = z;
	}

	void SetItemRotation(int x, int y, int z)
	{
		if (NativeItem == NULL)
			return;

		NativeItem->pos.xRot = ANGLE(x);
		NativeItem->pos.yRot = ANGLE(y);
		NativeItem->pos.zRot = ANGLE(z);
	}
};

typedef struct LuaVariable
{
	bool IsGlobal;
	string Name;
	int Type;
	float FloatValue;
	int IntValue;
	string StringValue;
	bool BoolValue;
};

class GameScript
{
private:
	sol::state*							m_lua;
	sol::table							m_globals;
	sol::table							m_locals;
	map<int, int>				m_itemsMap;
	vector<LuaFunction*>				m_triggers;
	GameScriptItem						m_items[NUM_ITEMS];

	string								loadScriptFromFile(char* luaFilename);

public:	
	GameScript(sol::state* lua);
	~GameScript();
	
	bool								ExecuteScript(char* luaFilename);
	void								FreeLevelScripts();
	void								AddTrigger(LuaFunction* function);
	void								AddLuaId(int luaId, int itemId);
	void								SetItem(int index, ITEM_INFO* item);
	void								AssignItemsAndLara();
	void								ResetVariables();
	void								GetVariables(vector<LuaVariable>* list);
	void								SetVariables(vector<LuaVariable>* list);

	void								EnableItem(short id);
	void								DisableItem(short id);
	void								PlayAudioTrack(short track);
	void								ChangeAmbientSoundTrack(short track);
	bool								ExecuteTrigger(short index);
	void								JumpToLevel(int levelNum);
	int								GetSecretsCount();
	void								SetSecretsCount(int secretsNum);
	void								AddOneSecret();
	void								MakeItemInvisible(short id);
	GameScriptItem						GetItem(short id);
	void								PlaySoundEffectAtPosition(short id, int x, int y, int z, int flags);
	void								PlaySoundEffect(short id, int flags);
};