TombEngine/TR5Main/Scripting/GameLogicScript.h

#pragma once
#include "ChunkId.h"
#include "ChunkReader.h"
#include "ChunkWriter.h"
#include "LEB128.h"
#include "Streams.h"
#include "items.h"

#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 {
	std::string Name;
	std::string Code;
	bool Executed;
};

class GameScriptPosition {
private:
	float								xPos;
	float								yPos;
	float								zPos;
	std::function<float()>					readXPos;
	std::function<void(float)>				writeXPos;
	std::function<float()>					readYPos;
	std::function<void(float)>				writeYPos;
	std::function<float()>					readZPos;
	std::function<void(float)>				writeZPos;

public:
	GameScriptPosition(float x, float y, float z);
	GameScriptPosition(std::function<float()> readX, std::function<void(float)> writeX, std::function<float()> readY, std::function<void(float)> writeY, std::function<float()> readZ, std::function<void(float)> writeZ);

	float								GetXPos();
	void								SetXPos(float x);
	float								GetYPos();
	void								SetYPos(float y);
	float								GetZPos();
	void								SetZPos(float z);
};

class GameScriptRotation {
private:
	float								xRot;
	float								yRot;
	float								zRot;
	std::function<float()>					readXRot;
	std::function<void(float)>				writeXRot;
	std::function<float()>					readYRot;
	std::function<void(float)>				writeYRot;
	std::function<float()>					readZRot;
	std::function<void(float)>				writeZRot;

public:
	GameScriptRotation(float x, float y, float z);
	GameScriptRotation(std::function<float()> readX, std::function<void(float)> writeX, std::function<float()> readY, std::function<void(float)> writeY, std::function<float()> readZ, std::function<void(float)> writeZ);

	float								GetXRot();
	void								SetXRot(float x);
	float								GetYRot();
	void								SetYRot(float y);
	float								GetZRot();
	void								SetZRot(float z);
};

class GameScriptItem {
private:
	short								NativeItemNumber;
	ITEM_INFO*							NativeItem;

public:
	GameScriptItem(short itemNumber);

	GameScriptPosition					GetPosition();
	GameScriptRotation					GetRotation();
	short								GetHP();
	void								SetHP(short hp);
	short								GetRoom();
	void								SetRoom(short room);
	short								GetCurrentState();
	void								SetCurrentState(short state);
	short								GetGoalState();
	void								SetGoalState(short state);
	short								GetRequiredState();
	void								SetRequiredState(short state);
	void								EnableItem();
	void								DisableItem();
};

class LuaVariables
{
public:
	std::map<std::string, sol::object>			variables;

	sol::object							GetVariable(std::string key);
	void								SetVariable(std::string key, sol::object value);
};

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

class GameScript
{
private:
	sol::state*							m_lua;
	LuaVariables						m_globals;
	LuaVariables						m_locals;
	std::map<int, short>						m_itemsMapId;
	std::map<std::string, short>					m_itemsMapName;
	std::vector<LuaFunction*>				m_triggers;

public:	
	GameScript(sol::state* lua);

	bool								ExecuteScript(const std::string& luaFilename, std::string& message);
	bool								ExecuteString(const std::string& command, std::string& message);
	void								FreeLevelScripts();
	void								AddTrigger(LuaFunction* function);
	void								AddLuaId(int luaId, short itemNumber);
	void								AddLuaName(std::string luaName, short itemNumber);
	void								AssignItemsAndLara();
	void								ResetVariables();

	template <typename T>
	void								GetVariables(std::map<std::string, T>& locals, std::map<std::string, T>& globals);
	template <typename T>
	void								SetVariables(std::map<std::string, T>& locals, std::map<std::string, T>& globals);
	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);
	std::unique_ptr<GameScriptItem>			GetItemById(int id);
	std::unique_ptr<GameScriptItem>			GetItemByName(std::string name);
	void								PlaySoundEffectAtPosition(short id, int x, int y, int z, int flags);
	void								PlaySoundEffect(short id, int flags);
	GameScriptPosition					CreatePosition(float x, float y, float z);
	GameScriptPosition					CreateSectorPosition(float x, float y, float z);
	GameScriptRotation					CreateRotation(float x, float y, float z);
	float								CalculateDistance(GameScriptPosition pos1, GameScriptPosition pos2);
	float								CalculateHorizontalDistance(GameScriptPosition pos1, GameScriptPosition pos2);
};