Custom level generation

libzhl/LuaCore.cpp

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

libzhl/LuaCore.h

@@ -257,6 +257,7 @@ 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;
     }
 
     LIBZHL_API void UnloadMetatables();

libzhl/functions/Level.zhl

@@ -7,6 +7,9 @@ __thiscall void Level::Init(bool unkinitStartRoom);
 "518b89????????e8":
 __thiscall void Level::Update();
 
+"558bec6aff68????????64a1????????5081ecc0000000a1????????33c58945??535657508d45??64a3????????8bc1":
+__thiscall void Level::reset_room_list(bool unkinitStartRoom);
+
 "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/LuaDungeonGenerator.cpp

@@ -0,0 +1,62 @@
+#include "HookSystem.h"
+#include "IsaacRepentance.h"
+#include "LuaCore.h"
+#include "Exception.h"
+#include "Log.h"
+#include "LuaDungeonGenerator.h"
+
+DungeonGenerator* GetDungeonGenerator(lua_State* L) {
+	return *lua::GetRawUserdata<DungeonGenerator**>(L, 1, lua::metatables::DungeonGeneratorMT);
+}
+
+LUA_FUNCTION(place_room) {
+	DungeonGenerator* generator = GetDungeonGenerator(L);
+	RoomConfig_Room* config = lua::GetLuabridgeUserdata<RoomConfig_Room*>(L, 2, lua::Metatables::CONST_ROOM_CONFIG_ROOM, "RoomConfig");
+	uint32_t row = (uint32_t)luaL_checkinteger(L, 3);
+	uint32_t col = (uint32_t)luaL_checkinteger(L, 4);
+	uint32_t seed = (uint32_t)luaL_checkinteger(L, 5);
+
+	DungeonGeneratorRoom* generatorRoom = new DungeonGeneratorRoom(config, row, col, seed);
+	generator->rooms[generator->num_rooms] = *generatorRoom;
+
+	generator->num_rooms++;
+
+	return 1;
+}
+
+LUA_FUNCTION(set_final_boss_room) {
+	DungeonGenerator* generator = GetDungeonGenerator(L);
+	uint32_t row = (uint32_t)luaL_checkinteger(L, 2);
+	uint32_t col = (uint32_t)luaL_checkinteger(L, 3);
+
+	for (size_t i = 0; i < generator->num_rooms; i++) {
+		DungeonGeneratorRoom* generator_room = &(generator->rooms[i]);
+
+		if (generator_room->room != NULL) {
+			if (row == generator_room->row && col == generator_room->col) {
+				generator_room->is_final_boss = true;
+			} else {
+				generator_room->is_final_boss = false;
+			}
+		} else {
+			break;
+		}
+	}
+
+	return 1;
+}
+
+static void RegisterDungeonGenerator(lua_State* L) {
+	luaL_Reg functions[] = {
+		{"PlaceRoom", place_room},
+		{"SetFinalBossRoom", set_final_boss_room},
+		{ NULL, NULL }
+	};
+
+	lua::RegisterNewClass(L, lua::metatables::DungeonGeneratorMT, lua::metatables::DungeonGeneratorMT, functions);
+}
+
+HOOK_METHOD(LuaEngine, RegisterClasses, () -> void) {
+	super();
+	RegisterDungeonGenerator(_state);
+}

repentogon/LuaInterfaces/LuaDungeonGenerator.h

@@ -0,0 +1,34 @@
+#pragma once
+
+struct DungeonGeneratorRoom {
+	RoomConfig_Room* room;
+	uint32_t row;
+	uint32_t col;
+	uint32_t seed;
+	bool is_final_boss = false;
+
+	DungeonGeneratorRoom() {
+		this->room = NULL;
+		this->row = -1;
+		this->col = -1;
+		this->seed = -1;
+	}
+
+	DungeonGeneratorRoom(RoomConfig_Room* room, uint32_t row, uint32_t col, uint32_t seed) {
+		this->room = room;
+		this->row = row;
+		this->col = col;
+		this->seed = seed;
+	}
+};
+
+struct DungeonGenerator {
+	int a;
+	int num_rooms;
+	DungeonGeneratorRoom rooms[169];
+
+	DungeonGenerator() {
+		a = 0;
+		num_rooms = 0;
+	}
+};

repentogon/Patches/ASMPatches/ASMLevel.cpp

@@ -5,6 +5,7 @@
 #include "ASMLevel.h"
 #include "../../LuaInterfaces/Level.h"
 #include "../../LuaInterfaces/Room/Room.h"
+#include "../../LuaInterfaces/LuaDungeonGenerator.h"
 
 std::bitset<36> generateLevels;
 
@@ -82,8 +83,67 @@ void ASMPatchVoidGeneration() {
 	sASMPatcher.PatchAt(addr, &patch);
 }
 
+//MC_PRE_GENERATE_DUNGEON(1340)
+bool ProcessGenerateDungeonCallback(Level* level, RNG* rng, int 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 = new DungeonGenerator();
+	lua::LuaResults results = lua::LuaCaller(L)
+		.push(callbackId)
+		.push(dungeonType)
+		.push(generator, lua::metatables::DungeonGeneratorMT)
+		.push(rng, lua::Metatables::RNG)
+		.call(1);
+
+	if (results || !lua_isboolean(L, -1) || !lua_toboolean(L, -1))
+	{
+		return false;
+	}
+
+	level->reset_room_list(false);
+
+	for (size_t i = 0; i < 507; i++)
+	{
+		g_Game->_roomOffset[i] = -1;
+	}
+
+	g_Game->_nbRooms = 0;
+
+	for (size_t i = 0; i < generator->num_rooms; i++)
+	{
+		DungeonGeneratorRoom generator_room = generator->rooms[i];
+
+		if (generator_room.room != NULL) {
+			LevelGenerator_Room* level_generator_room = new LevelGenerator_Room();
+			level_generator_room->_gridColIdx = generator_room.col;
+			level_generator_room->_gridLineIdx = generator_room.row;
+			level_generator_room->_doors = 15;
+
+			g_Game->PlaceRoom(level_generator_room, generator_room.room, generator_room.seed, 0);
+
+			if (generator_room.is_final_boss) {
+				g_Game->_lastBossRoomListIdx = i;
+			}
+		}
+	}
+
+	return true;
+}
+
 HOOK_METHOD(Level, generate_dungeon, (RNG* rng) -> void)
 {
+	bool skip = ProcessGenerateDungeonCallback(this, rng, 0);
+	if (skip) {
+		return;
+	}
+
 	if (this->_stage == 12)
 	{
 		if (generateLevels.any())
@@ -117,6 +177,51 @@ HOOK_METHOD(Level, generate_dungeon, (RNG* rng) -> void)
 	super(rng);
 }
 
+HOOK_METHOD(Level, generate_blue_womb, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, NULL, 1);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
+HOOK_METHOD(Level, generate_backwards_dungeon, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, NULL, 2);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
+HOOK_METHOD(Level, generate_home_dungeon, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, NULL, 3);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
+HOOK_METHOD(Level, generate_redkey_dungeon, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, NULL, 4);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
+HOOK_METHOD(Level, generate_greed_dungeon, () -> void) {
+	bool skip = ProcessGenerateDungeonCallback(this, NULL, 5);
+	if (skip) {
+		return;
+	}
+
+	super();
+}
+
 bool __stdcall SpawnSpecialQuestDoorValidStageTypeCheck() {
 	// Always return true if we forced it via Room:TrySpawnSpecialQuestDoor
 	bool ret = roomASM.ForceSpecialQuestDoor;

repentogon/resources/scripts/enums_ex.lua

@@ -266,6 +266,8 @@ ModCallbacks.MC_PRE_PICKUP_UPDATE_GHOST_PICKUPS = 1335
 --ModCallbacks.MC_PRE_NPC_GET_LOOT_LIST = 1336 (Reserved for the future)
 --ModCallbacks.MC_PRE_NPC_UPDATE_GHOST_PICKUPS = 1337 (Reserved for the future)
 
+ModCallbacks.MC_PRE_GENERATE_DUNGEON = 1340
+
 ModCallbacks.MC_PRE_PLAYER_ADD_CARD = 1350
 ModCallbacks.MC_POST_PLAYER_ADD_CARD = 1351
 ModCallbacks.MC_PRE_PLAYER_ADD_PILL = 1352
@@ -3080,6 +3082,14 @@ UseActiveItemResultFlag = {
 	REMOVE = 1 << 8,
 }
 
+GenerateDungeonType = {
+	NORMAL = 0,
+	BLUE_WOMB = 1,
+	BACKWARDS = 2,
+	HOME = 3,
+	RED_REDEMPTION = 4,
+	GREED = 5
+}
 
 --deprecated enums