// DR. ROBOTNIK'S RING RACERS
//-----------------------------------------------------------------------------
// Copyright (C) 2025 by James Robert Roman
// Copyright (C) 2025 by Kart Krew
//
// This program is free software distributed under the
// terms of the GNU General Public License, version 2.
// See the 'LICENSE' file for more details.
//-----------------------------------------------------------------------------

#ifndef __V_DRAW_HPP__
#define __V_DRAW_HPP__

#include <string_view>
#include <optional>
#include <utility>

#include <fmt/core.h>

#include "core/hash_map.hpp"
#include "core/string.h"
#include "doomdef.h" // skincolornum_t
#include "doomtype.h"
#include "screen.h" // BASEVIDWIDTH
#include "typedef.h"
#include "v_video.h"
#include "g_input.h"

// Helpers for setting up pad napping nonsense
typedef enum
{
	gb_mask = 0xF0,
	gb_a = 0xF0,
	gb_b,
	gb_x,
	gb_y,
	gb_lb,
	gb_rb,
	gb_lt,
	gb_rt,
	gb_start,
	gb_back,
	gb_ls,
	gb_rs,
	gb_dpad
} generic_buttons_e;

typedef enum
{
	sb_up = 0x00,
	sb_down,
	sb_right,
	sb_left,
	sb_lua1,
	sb_lua2,
	sb_lua3,
	sb_r,
	sb_l,
	sb_start,
	sb_a,
	sb_b,
	sb_c,
	sb_x,
	sb_y,
	sb_z
} saturn_buttons_e;

// Garden-variety standard gamepad
static const srb2::HashMap<INT32, char> standardpad = {
	{nc_a, gb_a},
	{nc_b, gb_b},
	{nc_x, gb_x},
	{nc_y, gb_y},
	{nc_lb, gb_lb},
	{nc_rb, gb_rb},
	{nc_lt, gb_lt},
	{nc_rt, gb_rt},
	{nc_start, gb_start},
	{nc_back, gb_back},
	{nc_ls, gb_ls},
	{nc_rs, gb_rs},
};

// Standard gamepad, but evil Nintendo layout flip was applied by your
// controller firmware or Steam Input—swap B/A and X/Y
static const srb2::HashMap<INT32, char> flippedpad = {
	{nc_a, gb_b},
	{nc_b, gb_a},
	{nc_x, gb_y},
	{nc_y, gb_x},
	{nc_lb, gb_lb},
	{nc_rb, gb_rb},
	{nc_lt, gb_lt},
	{nc_rt, gb_rt},
	{nc_start, gb_start},
	{nc_back, gb_back},
	{nc_ls, gb_ls},
	{nc_rs, gb_rs},
};

// Saturn Type A - Retrobit Wired Dinput, RB RT LB LT (CZLR)
static const srb2::HashMap<INT32, char> saturntypeA = {
	{nc_a, sb_a},
	{nc_b, sb_b},
	{nc_x, sb_x},
	{nc_y, sb_y},
	{nc_rb, sb_c},
	{nc_rt, sb_z},
	{nc_lb, sb_l},
	{nc_lt, sb_r},
	{nc_start, gb_start},
	{nc_back, gb_back},
	{nc_ls, gb_ls},
	{nc_rs, gb_rs},
};

// Saturn Type B - Retrobit Wireless Dinput, LB RB LT RT (CZLR)
static const srb2::HashMap<INT32, char> saturntypeB = {
	{nc_a, sb_a},
	{nc_b, sb_b},
	{nc_x, sb_x},
	{nc_y, sb_y},
	{nc_lb, sb_c},
	{nc_rb, sb_z},
	{nc_lt, sb_l},
	{nc_rt, sb_r},
	{nc_start, gb_start},
	{nc_back, gb_back},
	{nc_ls, gb_ls},
	{nc_rs, gb_rs},
};

// Saturn Type C - Retrobit Xinput, RT LT LB RB (CZLR)
static const srb2::HashMap<INT32, char> saturntypeC = {
	{nc_a, sb_a},
	{nc_b, sb_b},
	{nc_x, sb_x},
	{nc_y, sb_y},
	{nc_rt, sb_c},
	{nc_lt, sb_z},
	{nc_lb, sb_l},
	{nc_rb, sb_r},
	{nc_start, gb_start},
	{nc_back, gb_back},
	{nc_ls, gb_ls},
	{nc_rs, gb_rs},
};

// Saturn Type D - 8BitDo M30 / arcade, RT RB LB LT (CZLR)
// This cannot be disambiguated (shares L/R with type 1)
// but is more spatially correct w/r/t SDL expectations
// and standard arcade mapping (Z on top, C on bottom)
static const srb2::HashMap<INT32, char> saturntypeD = {
	{nc_a, sb_a},
	{nc_b, sb_b},
	{nc_x, sb_x},
	{nc_y, sb_y},
	{nc_rt, sb_c},
	{nc_rb, sb_z},
	{nc_lb, sb_l},
	{nc_lt, sb_r},
	{nc_start, gb_start},
	{nc_back, gb_back},
	{nc_ls, gb_ls},
	{nc_rs, gb_rs},
};

// Saturn Type E - Hori, RT RB (CZ) with some fucked up bumpers/triggers
// The Hori layout is, to my knowledge, the only 6bt one that has fully
// unique buttons in every slot while having both bumpers and triggers,
// so there's no way to accurately portray it without using generics.
static const srb2::HashMap<INT32, char> saturntypeE = {
	{nc_a, sb_a},
	{nc_b, sb_b},
	{nc_x, sb_x},
	{nc_y, sb_y},
	{nc_rt, sb_c},
	{nc_rb, sb_z},
	{nc_lb, gb_rb},
	{nc_lt, gb_rt},
	{nc_ls, gb_lb},
	{nc_rs, gb_lt},
	{nc_start, gb_start},
	{nc_back, gb_back},
};

// Saturn Type F - Mayflash XInput Saturn adapter, CZ = RS LS
// Cannot be disambiguated. GEE I SURE WISH THERE WERE A STANDARD
static const srb2::HashMap<INT32, char> saturntypeF = {
	{nc_a, sb_a},
	{nc_b, sb_b},
	{nc_x, sb_x},
	{nc_y, sb_y},
	{nc_rt, gb_rt},
	{nc_rb, gb_rb},
	{nc_lb, gb_lb},
	{nc_lt, gb_lt},
	{nc_ls, sb_z},
	{nc_rs, sb_c},
	{nc_start, gb_start},
	{nc_back, gb_back},
};

namespace srb2
{

class Draw
{
public:
	enum class Font
	{
		kThin,
		kGamemode,
		kGenesis,
		kConsole,
		kFreeplay,
		kZVote,
		kPing,
		kTimer,
		kThinTimer,
		kMenu,
		kMedium,
		kRollingNum,
		kRollingNum4P,
	};
	Font fontno_to_font(int font);

	enum class Align
	{
		kLeft,
		kCenter,
		kRight,
	};

	enum class Stretch
	{
		kNone,
		kWidth,
		kHeight,
		kBoth,
	};

	enum class Button
	{
		a,
		b,
		c,
		x,
		y,
		z,
		start,
		l,
		r,
		up,
		down,
		right,
		left,
		lua1,
		lua2,
		lua3,
	};

	enum class GenericButton
	{
		a,
		b,
		x,
		y,
		lb,
		rb,
		lt,
		rt,
		start,
		back,
		ls,
		rs,
		dpad
	};

	class TextElement
	{
	public:
		explicit TextElement() {}

		explicit TextElement(const srb2::String& string) : string_(string) {}

		template <class... Args>
		explicit TextElement(fmt::format_string<Args...> format, Args&&... args) :
			TextElement(fmt::format(format, args...))
		{
		}

		const srb2::String& string() const { return string_; }
		std::optional<Font> font() const { return font_; }
		std::optional<INT32> flags() const { return flags_; }
		std::optional<UINT8> as() const { return as_; }

		int width() const;

		TextElement& string(srb2::String string)
		{
			string_ = string;
			return *this;
		}

		TextElement& parse(std::string_view string);

		template <class... Args>
		TextElement& parse(fmt::format_string<Args...> format, Args&&... args)
		{
			return parse(fmt::format(format, args...));
		}

		TextElement& font(Font font)
		{
			font_ = font;
			return *this;
		}

		TextElement& flags(INT32 flags)
		{
			flags_ = flags;
			return *this;
		}

		TextElement& as(UINT8 as)
		{
			as_ = as;
			return *this;
		}
	private:
		srb2::String string_;
		std::optional<Font> font_;
		std::optional<INT32> flags_;
		std::optional<UINT8> as_;
	};

	class Chain
	{
	public:
		float x() const { return x_; }
		float y() const { return y_; }
		INT32 flags() const { return flags_; }

		// Methods add relative to the current state
		Chain& x(float x);
		Chain& y(float y);
		Chain& xy(float x, float y);
		Chain& flags(INT32 flags);

		// Methods overwrite the current state
		Chain& width(float width);
		Chain& height(float height);
		Chain& size(float width, float height);
		Chain& scale(float scale);
		Chain& font(Font font);
		Chain& align(Align align);
		Chain& stretch(Stretch stretch);

		// Absolute screen coordinates
		Chain& clipx(float left, float right); // 0 to BASEVIDWIDTH
		Chain& clipy(float top, float bottom); // 0 to BASEVIDHEIGHT

		Chain& clipx() { return clipx(x_, x_ + width_); }
		Chain& clipy() { return clipy(y_, y_ + height_); }

		// True to use internal clipping state
		// False to use global state (default)
		// Changing the clipping dimensions implicitly sets
		// this to true
		Chain& clip(bool yes);

		Chain& colormap(const UINT8* colormap);
		Chain& colormap(UINT16 color);
		Chain& colormap(INT32 skin, UINT16 color);
		Chain& colorize(UINT16 color);

		void text(const char* str) const { string(str, flags_, font_); }
		void text(const srb2::String& str) const { text(str.c_str()); }
		void text(const TextElement& elm) const
		{
			string(elm.string().c_str(), elm.flags().value_or(flags_), elm.font().value_or(font_));
		}

		template <class... Args>
		void text(fmt::format_string<Args...> format, Args&&... args) const { text(fmt::format(format, args...)); }

		TextElement text() const { return TextElement().font(font_).flags(flags_); }

		void patch(patch_t* patch) const;
		void patch(const char* name) const { patch(Draw::cache_patch(name)); }
		void patch(const srb2::String& name) const { patch(name.c_str()); }

		void thumbnail(UINT16 mapnum) const;

		void fill(UINT8 color) const;

		void button(Button type, std::optional<bool> press = {}) const { button_(type, 0, press); }
		void small_button(Button type, std::optional<bool> press = {}) const { button_(type, 1, press); }

		void generic_button(GenericButton type, std::optional<bool> press = {}) const { generic_button_(type, 0, press); }
		void generic_small_button(GenericButton type, std::optional<bool> press = {}) const { generic_button_(type, 1, press); }

		void sticker(patch_t* end_graphic, UINT8 color) const;
		void sticker() const { sticker(Draw::cache_patch("K_STIKEN"), 24); }
		void small_sticker() const { sticker(Draw::cache_patch("K_STIKE2"), 24); }

	private:
		constexpr Chain() {}
		explicit Chain(float x, float y) : x_(x), y_(y) {}
		Chain(const Chain&) = default;
		Chain& operator=(const Chain&) = default;

		struct Clipper
		{
			explicit Clipper(const Chain& chain);
			~Clipper();

		private:
			cliprect_t save_;
		};

		float x_ = 0.f;
		float y_ = 0.f;
		float width_ = 0.f;
		float height_ = 0.f;
		float scale_ = 1.f;

		float clipx1_ = 0.f;
		float clipx2_ = BASEVIDWIDTH;
		float clipy1_ = 0.f;
		float clipy2_ = BASEVIDHEIGHT;
		bool clip_ = false;

		INT32 flags_ = 0;

		Font font_ = Font::kThin;
		Align align_ = Align::kLeft;
		Stretch stretch_ = Stretch::kNone;

		const UINT8* colormap_ = nullptr;

		void string(const char* str, INT32 flags, Font font) const;
		void button_(Button type, int ver, std::optional<bool> press = {}) const;
		void generic_button_(GenericButton type, int ver, std::optional<bool> press = {}) const;

		friend Draw;
	};

	static patch_t* cache_patch(const char* name);

	constexpr Draw() {}
	explicit Draw(float x, float y) : chain_(x, y) {}
	Draw(const Chain& chain) : chain_(chain) {}

	// See class Chain for documentation

	float x() const { return chain_.x(); }
	float y() const { return chain_.y(); }
	INT32 flags() const { return chain_.flags(); }

#define METHOD(Name) \
	template <typename... Args>\
	Chain Name (Args&&... args) const { return Chain(chain_).Name(std::forward<Args>(args)...); }

	METHOD(x);
	METHOD(y);
	METHOD(xy);
	METHOD(flags);
	METHOD(width);
	METHOD(height);
	METHOD(size);
	METHOD(scale);
	METHOD(font);
	METHOD(align);
	METHOD(stretch);
	METHOD(clipx);
	METHOD(clipy);
	METHOD(clip);
	METHOD(colormap);
	METHOD(colorize);

#undef METHOD

#define VOID_METHOD(Name) \
	template <typename... Args>\
	auto Name (Args&&... args) const { return chain_.Name(std::forward<Args>(args)...); }

	VOID_METHOD(text);
	VOID_METHOD(patch);
	VOID_METHOD(thumbnail);
	VOID_METHOD(fill);
	VOID_METHOD(button);
	VOID_METHOD(small_button);
	VOID_METHOD(generic_button);
	VOID_METHOD(generic_small_button);
	VOID_METHOD(sticker);
	VOID_METHOD(small_sticker);

#undef VOID_METHOD

private:
	Chain chain_;

	static int font_to_fontno(Font font);
	static INT32 default_font_flags(Font font);
	static fixed_t font_width(Font font, INT32 flags, const char* string);
};

}; // namespace srb2

#include "v_draw_setter.hpp"

#endif // __V_DRAW_HPP__