Custom level generation

libzhl/LuaCore.cpp

@@ -838,6 +838,8 @@ namespace lua {
 		const char* LootListEntryMT = "LootListEntry";
 		const char* MinimapConfigMT = "MinimapConfig";
 		const char* EntityDescMT = "EntityDesc";
+		const char* DungeonGeneratorMT = "DungeonGenerator";
+		const char* DungeonGeneratorRoomMT = "DungeonGeneratorRoom";
 	}
 
 	void TableAssoc(lua_State* L, std::string const& name, int value) {

libzhl/LuaCore.h

@@ -257,6 +257,8 @@ namespace lua {
         extern LIBZHL_API const char* LootListEntryMT;
         extern LIBZHL_API const char* MinimapConfigMT;
         extern LIBZHL_API const char* EntityDescMT;
+        extern LIBZHL_API const char* DungeonGeneratorMT;
+        extern LIBZHL_API const char* DungeonGeneratorRoomMT;
     }
 
     LIBZHL_API void UnloadMetatables();

libzhl/functions/Level.zhl

@@ -2,11 +2,14 @@
 __thiscall void Level::SetStage(int stageid,int alt);
 
 "538bdc83ec0883e4f883c404558b6b??896c24??8bec6aff68????????64a1????????505383ec68a1????????33c58945??5657508d45??64a3????????8bf9897d??8b0d":
-__thiscall void Level::Init(bool unkinitStartRoom);
+__thiscall void Level::Init(bool resetLilPortalRoom);
 
 "518b89????????e8":
 __thiscall void Level::Update();
 
+"558bec6aff68????????64a1????????5081ecc0000000a1????????33c58945??535657508d45??64a3????????8bc1":
+__thiscall void Level::reset_room_list(bool resetLilPortalRoom);
+
 "558bec6aff68????????64a1????????5083ec2c535657a1????????33c5508d45??64a3????????8bd98b45":
 __thiscall void Level::ChangeRoom(int targetRoomIDX, int dimension);
 
@@ -61,6 +64,21 @@ __thiscall int Level::GetRandomRoomIndex(bool IAmErrorRoom, unsigned int Seed);
 "538bdc83ec0883e4f883c404558b6b??896c24??8bec6aff68????????64a1????????505381ec50040000":
 __thiscall void Level::generate_dungeon(RNG * RNG);
 
+"558bec6aff68????????64a1????????5081ecd4030000a1????????33c58945??535657508d45??64a3????????8bd9":
+__thiscall void Level::generate_home_dungeon();
+
+"538bdc83ec0883e4f883c404558b6b??896c24??8bec6aff68????????64a1????????505381ec30040000":
+__thiscall void Level::generate_backwards_dungeon();
+
+"558bec6aff68????????64a1????????5081ecc4030000":
+__thiscall void Level::generate_redkey_dungeon();
+
+"558bec6aff68????????64a1????????5081ecdc030000":
+__thiscall void Level::generate_blue_womb();
+
+"558bec6aff68????????64a1????????5081ecec030000":
+__thiscall void Level::generate_greed_dungeon();
+
 "558bec6aff68????????64a1????????5083ec68a1????????33c58945??535657508d45??64a3????????8bd98b55":
 __thiscall bool Level::place_rooms_backwards(LevelGenerator* levelGen, BackwardsStageDesc* backwardsStage, uint32_t stage, uint32_t minDifficulty, uint32_t maxDifficulty);
 

repentogon/LuaInterfaces/CustomCallbacks.cpp

@@ -15,6 +15,7 @@
 #include "../Patches/MainMenuBlock.h"
 #include "../Patches/XMLData.h"
 #include "../Patches/EntityPlus.h"
+#include "LuaDungeonGenerator.h"
 
 //Callback tracking for optimizations
 std::bitset<500> CallbackState;  // For new REPENTOGON callbacks. I dont think we will add 500 callbacks but lets set it there for now
@@ -5646,4 +5647,90 @@ HOOK_STATIC(LuaEngine, PostEntityKill, (Entity* ent) -> void, __stdcall) {
 			.push(lastSource, lua::Metatables::ENTITY_REF)
 			.call(1);
 	}
+}
+
+
+//MC_PRE_GENERATE_DUNGEON(1340)
+bool ProcessGenerateDungeonCallback(Level* level, RNG& rng, DungeonGenerationType dungeonType) {
+	const int callbackId = 1340;
+	if (!CallbackState.test(callbackId - 1000)) {
+		return false;
+	}
+
+	lua_State* L = g_LuaEngine->_state;
+	lua::LuaStackProtector protector(L);
+	lua_rawgeti(L, LUA_REGISTRYINDEX, g_LuaEngine->runCallbackRegistry->key);
+
+	DungeonGenerator generator(&rng);
+	lua::LuaResults results = lua::LuaCaller(L)
+		.push(callbackId)
+		.push((int)dungeonType)
+		.push(&generator, lua::metatables::DungeonGeneratorMT)
+		.push(&rng, lua::Metatables::RNG)
+		.push((int)dungeonType)
+		.call(1);
+
+	if (results || !lua_isboolean(L, -1) || !lua_toboolean(L, -1))
+	{
+		return false;
+	}
+
+	bool correctGeneration = generator.Generate(level);
+
+	return correctGeneration;
+}
+
+HOOK_METHOD(Level, generate_dungeon, (RNG* rng) -> void)
+{
+	bool skip = ProcessGenerateDungeonCallback(this, *rng, DEFAULT);
+	if (skip) {
+		return;
+	}
+
+	super(rng);
+}
+
+HOOK_METHOD(Level, generate_blue_womb, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, g_Game->_generationRNG, BLUE_WOMB);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
+HOOK_METHOD(Level, generate_backwards_dungeon, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, g_Game->_generationRNG, BACKWARDS);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
+HOOK_METHOD(Level, generate_home_dungeon, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, g_Game->_generationRNG, HOME);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
+HOOK_METHOD(Level, generate_redkey_dungeon, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, g_Game->_generationRNG, RED_REDEMPTION);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
+HOOK_METHOD(Level, generate_greed_dungeon, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, g_Game->_generationRNG, GREED);
+	if (skip) {
+		return;
+	}
+
+	super();
 }