From 3e1e1245cee19bf54b3cc3655ea20f975da97eed Mon Sep 17 00:00:00 2001
From: hayaunderscore <stophayapls123@proton.me>
Date: Tue, 4 Jun 2024 17:13:35 +0800
Subject: [PATCH] Expose `V_SetClipRect` and `V_ClearClipRect` to lua

Via the `v.setClipRect` and `v.clearClipRect` functions respectively.
---
 src/lua_hudlib.c          | 41 +++++++++++++++++++++++++++++++++++++++
 src/lua_hudlib_drawlist.c | 36 ++++++++++++++++++++++++++++++++++
 src/lua_hudlib_drawlist.h | 11 +++++++++++
 3 files changed, 88 insertions(+)

diff --git a/src/lua_hudlib.c b/src/lua_hudlib.c
index 46d5b8a02..e3cf0db32 100644
--- a/src/lua_hudlib.c
+++ b/src/lua_hudlib.c
@@ -972,6 +972,45 @@ static int libd_drawKartString(lua_State *L)
 	return 0;
 }
 
+static int libd_setClipRect(lua_State *L)
+{
+	fixed_t x = luaL_checkinteger(L, 1);
+	fixed_t y = luaL_checkinteger(L, 2);
+	fixed_t w = luaL_checkinteger(L, 3);
+	fixed_t h = luaL_checkinteger(L, 4);
+	INT32 flags = luaL_optinteger(L, 5, 0);
+	huddrawlist_h list;
+
+	flags &= ~V_PARAMMASK; // Don't let crashes happen.
+
+	HUDONLY
+	lua_getfield(L, LUA_REGISTRYINDEX, "HUD_DRAW_LIST");
+	list = (huddrawlist_h) lua_touserdata(L, -1);
+	lua_pop(L, 1);
+
+	if (LUA_HUD_IsDrawListValid(list))
+		LUA_HUD_AddSetClipRect(list, x, y, w, h, flags);
+	else
+		V_SetClipRect(x, y, w, h, flags);
+	return 0;
+}
+
+static int libd_clearClipRect(lua_State *L)
+{
+	huddrawlist_h list;
+
+	HUDONLY
+	lua_getfield(L, LUA_REGISTRYINDEX, "HUD_DRAW_LIST");
+	list = (huddrawlist_h) lua_touserdata(L, -1);
+	lua_pop(L, 1);
+
+	if (LUA_HUD_IsDrawListValid(list))
+		LUA_HUD_AddClearClipRect(list);
+	else
+		V_ClearClipRect();
+	return 0;
+}
+
 static int libd_titleCardStringWidth(lua_State *L)
 {
 	const char *str = luaL_checkstring(L, 1);
@@ -1263,6 +1302,8 @@ static luaL_Reg lib_draw[] = {
 	{"drawString", libd_drawString},
 	{"drawTitleCardString", libd_drawTitleCardString},
 	{"drawKartString", libd_drawKartString},
+	{"setClipRect", libd_setClipRect},
+	{"clearClipRect", libd_clearClipRect},
 	// misc
 	{"stringWidth", libd_stringWidth},
 	{"titleCardStringWidth", libd_titleCardStringWidth},
diff --git a/src/lua_hudlib_drawlist.c b/src/lua_hudlib_drawlist.c
index 9a12c7b99..ae62b629b 100644
--- a/src/lua_hudlib_drawlist.c
+++ b/src/lua_hudlib_drawlist.c
@@ -37,6 +37,8 @@ enum drawitem_e {
 	DI_DrawTitleCardString,
 	DI_DrawTitleCardStringBoss,
 	DI_DrawKartString,
+	DI_SetClipRect,
+	DI_ClearClipRect,
 	DI_MAX,
 };
 
@@ -555,6 +557,34 @@ void LUA_HUD_AddDrawKartString(
 	// CalcStringCoords(item, str);
 }
 
+void LUA_HUD_AddSetClipRect(
+	huddrawlist_h list,
+	fixed_t x,
+	fixed_t y,
+	fixed_t w,
+	fixed_t h,
+	INT32 flags
+)
+{
+	size_t i = AllocateDrawItem(list);
+	drawitem_t *item = &list->items[i];
+	item->type = DI_SetClipRect;
+	item->x = x;
+	item->y = y;
+	item->w = w;
+	item->h = h;
+	item->flags = flags;
+}
+
+void LUA_HUD_AddClearClipRect(
+	huddrawlist_h list
+)
+{
+	size_t i = AllocateDrawItem(list);
+	drawitem_t *item = &list->items[i];
+	item->type = DI_ClearClipRect;
+}
+
 void LUA_HUD_DrawList(huddrawlist_h list)
 {
 	size_t i;
@@ -692,6 +722,12 @@ void LUA_HUD_DrawList(huddrawlist_h list)
 			case DI_DrawKartString:
 				V_DrawKartString(LERPS(x), LERPS(y), item->flags, itemstr);
 				break;
+			case DI_SetClipRect:
+				V_SetClipRect(item->x, item->y, item->w, item->h, item->flags);
+				break;
+			case DI_ClearClipRect:
+				V_ClearClipRect();
+				break;
 			default:
 				I_Error("can't draw draw list item: invalid draw list item type");
 				continue;
diff --git a/src/lua_hudlib_drawlist.h b/src/lua_hudlib_drawlist.h
index 15249e4f9..1e6bb97d9 100644
--- a/src/lua_hudlib_drawlist.h
+++ b/src/lua_hudlib_drawlist.h
@@ -120,6 +120,17 @@ void LUA_HUD_AddDrawKartString(
 	const char *str,
 	INT32 flags
 );
+void LUA_HUD_AddSetClipRect(
+	huddrawlist_h list,
+	fixed_t x,
+	fixed_t y,
+	fixed_t w,
+	fixed_t h,
+	INT32 flags
+);
+void LUA_HUD_AddClearClipRect(
+	huddrawlist_h list
+);
 
 // Draws the given draw list
 void LUA_HUD_DrawList(huddrawlist_h list);