Make Full Random work for real world maps

Author: goto-bus-stop

aoc-builtin-rms.c

@@ -73,6 +73,9 @@ typedef int __thiscall (*fn_ai_define_const)(void*, char*, int);
 static const size_t offs_vtbl_map_generate = 0x638114;
 static const size_t offs_map_generate = 0x45EE10;
 static const size_t offs_load_scx = 0x40DF00;
+static const size_t offs_vtbl_launch_game = 0x63E0B0;
+static const size_t offs_launch_game = 0x4FF390;
+typedef void __thiscall (*fn_launch_game)(void*, int, int, int, int, int, int, int, int, int);
 typedef int __thiscall (*fn_map_generate)(void*, int, int, char*, void*, int);
 typedef int __thiscall (*fn_load_scx)(void*, char*, int, void*);
 
@@ -90,6 +93,7 @@ static fn_text_get_value aoc_text_get_value = 0;
 static fn_text_set_rollover_id aoc_text_set_rollover_id = 0;
 static fn_ai_define_symbol aoc_ai_define_symbol = 0;
 static fn_ai_define_const aoc_ai_define_const = 0;
+static fn_launch_game aoc_launch_game = 0;
 static fn_map_generate aoc_map_generate = 0;
 static fn_load_scx aoc_load_scx = 0;
 static fn_texture_create aoc_texture_create = 0;
@@ -100,6 +104,11 @@ static int get_map_type() {
   return *(int*)(base_offset + offs_map_type);
 }
 
+static void set_map_type(int map_type) {
+  int base_offset = *(int*)offs_game_instance;
+  *(int*)(base_offset + offs_map_type) = map_type;
+}
+
 static void* get_world() {
   int base_offset = *(int*)offs_game_instance;
   return *(void**)(base_offset + offs_world);
@@ -189,6 +198,41 @@ static void apply_terrain_overrides(terrain_overrides_t* overrides) {
   }
 }
 
+static int builtin_real_world_map_ids[] = { 34, 35, 36, 37, 38, 39, 40, 41, 42, 43 };
+static int num_builtin_real_world_maps = sizeof(builtin_real_world_map_ids) / sizeof(int);
+static int count_real_world_maps() {
+  int result = num_builtin_real_world_maps;
+  for (int i = 0; i < num_custom_maps; i++) {
+    if (custom_maps[i].type == RMS_REALWORLD) {
+      result++;
+    }
+  }
+  return result;
+}
+
+// only has this many "parameters" in userpatch, i think
+// maybe can tweak this depending on the game version
+static void __thiscall launch_game_hook(void* screen, int a, int b, int c, int d, int e, int f, int g, int h, int i) {
+  if (get_map_type() == 47) {
+    debug("[aoc-builtin-rms] picking full random real world map...\n");
+    int max = count_real_world_maps();
+    int pick = (rand() % max);
+    if (pick < num_builtin_real_world_maps) {
+      set_map_type(builtin_real_world_map_ids[pick]);
+    } else {
+      pick -= num_builtin_real_world_maps;
+      for (int i = 0; i < num_custom_maps; i++) {
+        if (custom_maps[i].type == RMS_REALWORLD && --pick == 0) {
+          set_map_type(custom_maps[i].id);
+          break;
+        }
+      }
+    }
+  }
+
+  return aoc_launch_game(screen, a, b, c, d, e, f, g, h, i);
+}
+
 static void* current_game_info;
 static void __thiscall map_generate_hook(void* map, int size_x, int size_y, char* name, void* game_info, int num_players) {
   debug("[aoc-builtin-rms] called hooked map_generate %s %p\n", name, game_info);
@@ -345,7 +389,9 @@ void aoc_builtin_rms_init(custom_map_t* new_custom_maps, size_t new_num_custom_m
   /* Stuff to make real world maps work */
   aoc_map_generate = (fn_map_generate) offs_map_generate;
   aoc_load_scx = (fn_load_scx) offs_load_scx;
+  aoc_launch_game = (fn_launch_game) offs_launch_game;
   install_vtblhook((void*) offs_vtbl_map_generate, map_generate_hook);
+  install_vtblhook((void*) offs_vtbl_launch_game, launch_game_hook);
 }
 
 void deinit() {