#include "framework.h"
#include "Scripting/Internal/TEN/DisplaySprite/ScriptDisplaySprite.h"

#include "Game/effects/DisplaySprite.h"
#include "Game/Setup.h"
#include "Objects/game_object_ids.h"
#include "Scripting/Internal/LuaHandler.h"
#include "Scripting/Internal/ReservedScriptNames.h"
#include "Scripting/Internal/TEN/Color/Color.h"
#include "Scripting/Internal/TEN/DisplaySprite/AlignModes.h"
#include "Scripting/Internal/TEN/DisplaySprite/ScaleModes.h"
#include "Scripting/Internal/TEN/Vec2/Vec2.h"

using namespace TEN::Effects::DisplaySprite;

namespace TEN::Scripting::DisplaySprite
{
	void ScriptDisplaySprite::Register(sol::table& parent)
	{
		using ctors = sol::constructors<ScriptDisplaySprite(GAME_OBJECT_ID, int, const Vec2&, float, const Vec2&, sol::optional<const ScriptColor&>)>;

		parent.new_usertype<ScriptDisplaySprite>(
			ScriptReserved_DisplaySprite,
			ctors(),
			sol::call_constructor, ctors(),

			ScriptReserved_DisplaySpriteDraw, &ScriptDisplaySprite::Draw,

			/// (Objects.ObjID) ID of the sprite sequence object.
			//@mem ObjectID
			"ObjectID", &ScriptDisplaySprite::ObjectID,

			/// (int) ID of the display sprite in the sprite sequence object.
			//@mem SpriteID
			"SpriteID", &ScriptDisplaySprite::SpriteID,

			/// (Vec2) Display space position of the display sprite in percent. Alignment determined by __DisplaySprite.AlignMode__
			//@mem Position
			"Position", &ScriptDisplaySprite::Position,

			/// (float) Rotation of the display sprite in degrees.
			//@mem Rotation
			"Rotation", &ScriptDisplaySprite::Rotation,

			/// (Vec2) Horizontal and vertical scale of the display sprite in percent. Relative to __DisplaySprite.ScaleMode__.
			//@mem Scale
			"Scale", &ScriptDisplaySprite::Scale,

			/// (Color) Color of the display sprite.
			//@mem Color
			"Color", &ScriptDisplaySprite::Color);
	}

	void ScriptDisplaySprite::RegisterTables(sol::state* state, sol::table& parent)
	{
		auto table = sol::table(state->lua_state(), sol::create);
		parent.set(ScriptReserved_DisplaySprite, table);

		auto handler = LuaHandler(state);
		handler.MakeReadOnlyTable(table, ScriptReserved_DisplaySpriteTableAlignMode, DISPLAY_SPRITE_ALIGN_MODES);
		handler.MakeReadOnlyTable(table, ScriptReserved_DisplaySpriteTableScaleMode, DISPLAY_SPRITE_SCALE_MODES);
	}

	/***
	@int objectID ID of the sprite sequence object.
	@int spriteID ID of the sprite in the sprite sequence object.
	@Vec2 pos Display space position of the display sprite in percent. Alignment determined by __DisplaySprite.AlignMode__
	@float rot Rotation of the display sprite in degrees.
	@Vec2 scale Horizontal and vertical scale of the display sprite in percent. Relative to __DisplaySprite.ScaleMode__.
	@Color color[opt] Color of the display sprite. __Default: Color(255, 255, 255, 255)__
	@treturn DisplaySprite A DisplaySprite object.
	*/
	ScriptDisplaySprite::ScriptDisplaySprite(GAME_OBJECT_ID objectID, int spriteID, const Vec2& pos, float rot, const Vec2& scale,
											 sol::optional<const ScriptColor&> color)
	{
		static const auto DEFAULT_COLOR = ScriptColor(255, 255, 255, 255);

		ObjectID = objectID;
		SpriteID = spriteID;
		Position = pos;
		Rotation = rot;
		Scale = scale;
		Color = color.value_or(DEFAULT_COLOR);
	}

	/*** Draw the display sprite in display space for the current frame.
	@function DisplaySprite:Draw
	@tparam Objects.ObjID[opt] priority Draw priority of the sprite. Can be thought of as a layer, with higher values having higher priority. __Default: 0__
	@tparam DisplaySprite.AlignMode[opt] alignMode Align mode of the sprite. __Default: DisplaySprite.AlignMode.CENTER__
	@tparam DisplaySprite.ScaleMode[opt] scaleMode Scale mode of the sprite. __Default: DisplaySprite.ScaleMode.FIT__
	@tparam Effects.BlendID[opt] blendMode Blend mode of the sprite. __Default: Effects.BlendID.ALPHABLEND__
	*/
	void ScriptDisplaySprite::Draw(sol::optional<int> priority, sol::optional<DisplaySpriteAlignMode> alignMode,
								   sol::optional<DisplaySpriteScaleMode> scaleMode, sol::optional<BLEND_MODES> blendMode)
	{
		// NOTE: Conversion from more intuitive 100x100 screen space resolution to internal 800x600 is required.
		// In a future refactor, everything will use 100x100 natively. -- Sezz 2023.08.31
		constexpr auto POS_CONVERSION_COEFF	  = Vector2(SCREEN_SPACE_RES.x / 100, SCREEN_SPACE_RES.y / 100);
		constexpr auto SCALE_CONVERSION_COEFF = 0.01f;

		constexpr auto DEFAULT_PRIORITY	  = 0;
		constexpr auto DEFAULT_ALIGN_MODE = DisplaySpriteAlignMode::Center;
		constexpr auto DEFAULT_SCALE_MODE = DisplaySpriteScaleMode::Fit;
		constexpr auto DEFAULT_BLEND_MODE = BLENDMODE_ALPHABLEND;

		// Object is not a sprite object; return early.
		if (ObjectID < GAME_OBJECT_ID::ID_HORIZON || ObjectID >= GAME_OBJECT_ID::ID_NUMBER_OBJECTS)
		{
			TENLog("Attempted to draw display sprite from non-sprite object " + std::to_string(ObjectID), LogLevel::Warning);
			return;
		}

		// Sprite missing or sequence not found; return early.
		const auto& object = Objects[ObjectID];
		if (!object.loaded || SpriteID >= abs(object.nmeshes))
		{
			TENLog(
				"Attempted to draw missing display sprite " + std::to_string(SpriteID) +
				" from sprite object " + std::to_string(ObjectID),
				LogLevel::Warning);
			return;
		}

		auto convertedPos = Vector2(Position.x, Position.y) * POS_CONVERSION_COEFF;
		short convertedRot = ANGLE(Rotation);
		auto convertedScale = Vector2(Scale.x, Scale.y) * SCALE_CONVERSION_COEFF;
		auto convertedColor = Vector4(Color.GetR(), Color.GetG(), Color.GetB(), Color.GetA()) / UCHAR_MAX;

		AddDisplaySprite(
			ObjectID, SpriteID,
			convertedPos, convertedRot, convertedScale, convertedColor,
			priority.value_or(DEFAULT_PRIORITY),
			alignMode.value_or(DEFAULT_ALIGN_MODE),
			scaleMode.value_or(DEFAULT_SCALE_MODE),
			blendMode.value_or(DEFAULT_BLEND_MODE));
	}
}