From 3674aa1fc8efc9beb1b02fda872ed98f770be620 Mon Sep 17 00:00:00 2001
From: Starkku <starkku.ppmod@gmail.com>
Date: Fri, 6 Oct 2023 13:25:58 +0300
Subject: [PATCH] Allow wall overlays to use Palette

---
 FA2sp/Ext/CLoading/Body.h               |  2 +-
 FA2sp/Ext/CLoading/Hooks.cpp            | 25 +++++++++++++++++++++++++
 FA2sp/Ext/CMapData/Body.h               |  1 +
 FA2sp/Ext/CMapData/Hooks.cpp            |  1 +
 FA2sp/Ext/CTileSetBrowserView/Hooks.cpp | 18 ++++++++++++++++++
 5 files changed, 46 insertions(+), 1 deletion(-)

diff --git a/FA2sp/Ext/CLoading/Body.h b/FA2sp/Ext/CLoading/Body.h
index 28f6fcb..0ddf450 100644
--- a/FA2sp/Ext/CLoading/Body.h
+++ b/FA2sp/Ext/CLoading/Body.h
@@ -34,8 +34,8 @@ class NOVTABLE CLoadingExt : public CLoading
 	static ppmfc::CString GetBuildingImageName(ppmfc::CString ID, int nFacing, int state);
 	
 	static void ClearItemTypes();
-private:
 	void GetFullPaletteName(ppmfc::CString& PaletteName);
+private:
 	static ppmfc::CString* __cdecl GetDictName(ppmfc::CString* ret, const char* ID, int nFacing) { JMP_STD(0x475450); }
 	static ppmfc::CString GetDictName(ppmfc::CString ID, int nFacing)
 	{
diff --git a/FA2sp/Ext/CLoading/Hooks.cpp b/FA2sp/Ext/CLoading/Hooks.cpp
index 54d092a..172683d 100644
--- a/FA2sp/Ext/CLoading/Hooks.cpp
+++ b/FA2sp/Ext/CLoading/Hooks.cpp
@@ -1,9 +1,12 @@
 #include "Body.h"
 
+#include <Drawing.h>
 #include <CFinalSunApp.h>
 #include <CMixFile.h>
 #include <CLoading.h>
 #include <CINI.h>
+#include "..\..\Miscs\Palettes.h"
+#include "..\CMapData\Body.h"
 
 DEFINE_HOOK(486B00, CLoading_InitMixFiles, 7)
 {
@@ -82,3 +85,25 @@ DEFINE_HOOK(4B8CFC, CMapData_CreateMap_InitMixFiles_Removal, 5)
 	reinterpret_cast<CLoadingHelper*>(CLoading::Instance())->DTOR();
 	return 0x4B8D0C;
 }
+
+
+DEFINE_HOOK(4903F3, CLoading_DrawOverlay_Palette, 7) 
+{
+	GET(CLoadingExt*, pThis, EDI);
+	REF_STACK(ImageDataClass, pDrawData, STACK_OFFS(0x1C8, 0xC8));
+	GET_STACK(int, nOverlayIndex, STACK_OFFS(0x1C8, -0x8));
+
+	if (nOverlayIndex >= 0 && nOverlayIndex < 255)
+	{
+		auto const& typeData = CMapDataExt::OverlayTypeDatas[nOverlayIndex];
+
+		if (typeData.Wall)
+		{
+			auto palName = typeData.PaletteName;
+			pThis->GetFullPaletteName(palName);
+			pDrawData.pPalette = PalettesManager::LoadPalette(palName);
+		}
+	}
+
+	return 0;
+}
diff --git a/FA2sp/Ext/CMapData/Body.h b/FA2sp/Ext/CMapData/Body.h
index 762e9be..8f6df8f 100644
--- a/FA2sp/Ext/CMapData/Body.h
+++ b/FA2sp/Ext/CMapData/Body.h
@@ -8,6 +8,7 @@
 struct OverlayTypeData
 {
     bool Rock;
+    ppmfc::CString PaletteName;
 };
 
 struct BuildingDataExt
diff --git a/FA2sp/Ext/CMapData/Hooks.cpp b/FA2sp/Ext/CMapData/Hooks.cpp
index 0a48a5a..1bc7ceb 100644
--- a/FA2sp/Ext/CMapData/Hooks.cpp
+++ b/FA2sp/Ext/CMapData/Hooks.cpp
@@ -80,6 +80,7 @@ DEFINE_HOOK(47AB50, CLoading_InitPics_InitOverlayTypeDatas, 7)
     {
         auto& item = CMapDataExt::OverlayTypeDatas.emplace_back();
         item.Rock = Variables::Rules.GetBool(id, "IsARock");
+        item.PaletteName = CINI::Art->GetString(id, "Palette", "unit");
     }
 
     return 0;
diff --git a/FA2sp/Ext/CTileSetBrowserView/Hooks.cpp b/FA2sp/Ext/CTileSetBrowserView/Hooks.cpp
index 046a933..eb9d79e 100644
--- a/FA2sp/Ext/CTileSetBrowserView/Hooks.cpp
+++ b/FA2sp/Ext/CTileSetBrowserView/Hooks.cpp
@@ -1,3 +1,4 @@
+#include <CPalette.h>
 #include <CTileSetBrowserView.h>
 #include <Helpers/Macro.h>
 
@@ -20,6 +21,23 @@ DEFINE_HOOK(4F258B, CTileSetBrowserView_OnDraw_SetOverlayFrameToDisplay, 7)
     return i < nDisplayLimit ? 0x4F2230 : 0x4F2598;
 }
 
+DEFINE_HOOK(4F22F7, CTileSetBrowserView_OnDraw_OverlayPalette, 5)
+{
+    GET(Palette*, pPalette, EAX);
+    GET_STACK(BytePalette*, pBytePalette, STACK_OFFS(0xDC, 0xBC));
+
+    if (pPalette == Palette::PALETTE_ISO || pPalette == Palette::PALETTE_THEATER || pPalette == Palette::PALETTE_UNIT)
+        return 0;
+
+    *pBytePalette = BytePalette();
+
+    for (int i = 0; i < 256; i++) 
+    {
+        pBytePalette->Data[i] = pPalette->GetByteColor(i);
+    }
+
+    return 0x4F2315;
+}
 
 //DEFINE_HOOK(4F1D70, ASDG, 6)
 //{