review

library/Core.cpp

@@ -1692,6 +1692,7 @@ bool Core::InitSimulationThread()
     }
     std::cerr << "Initializing textures.\n";
     Textures::init(con);
+    Textures::initDynamic(con);
     // create mutex for syncing with interactive tasks
     std::cerr << "Initializing plugins.\n";
     // create plugin manager

library/include/modules/Textures.h

@@ -1,13 +1,14 @@
 #pragma once
 
-#include <optional>
+#include <vector>
+#include <string>
 
 #include "Export.h"
 #include "ColorText.h"
 
 #include <SDL_surface.h>
 
-using TexposHandle = uintptr_t;
+typedef typename void* TexposHandle;
 
 namespace DFHack {
 
@@ -24,12 +25,18 @@ namespace Textures {
  */
 DFHACK_EXPORT TexposHandle loadTexture(SDL_Surface* surface);
 
+/**
+ * Load tileset from image file.
+ * Return vector of handles to obtain valid texposes.
+ */
+DFHACK_EXPORT std::vector<TexposHandle> loadTileset(const std::string& file, int tile_px_w, int tile_px_h);
+
 /**
  * Get texpos by handle.
  * Always use this function, if you need to get valid texpos for your texure.
  * Texpos can change on game textures reset, but handle will be the same.
  */
-DFHACK_EXPORT std::optional<long> getTexposByHandle(TexposHandle handle);
+DFHACK_EXPORT long getTexposByHandle(TexposHandle handle);
 
 /**
  * Call this on DFHack init and on every viewscreen change so we can reload
@@ -37,6 +44,11 @@ DFHACK_EXPORT std::optional<long> getTexposByHandle(TexposHandle handle);
  */
 void init(DFHack::color_ostream& out);
 
+/**
+ * Call this on DFHack init just once to setup interposed handlers.
+ */
+void initDynamic(DFHack::color_ostream& out);
+
 /**
  * Call this when DFHack is being unloaded.
  *

library/modules/Textures.cpp

@@ -25,7 +25,6 @@ namespace DFHack {
 }
 
 static bool g_loaded = false;
-static bool g_dynamic_loaded = false;
 static long g_num_dfhack_textures = 0;
 static long g_dfhack_logo_texpos_start = -1;
 static long g_green_pin_texpos_start = -1;
@@ -45,6 +44,9 @@ static std::unordered_map<TexposHandle, long> g_handle_to_texpos;
 static std::unordered_map<TexposHandle, SDL_Surface*> g_handle_to_surface;
 static std::mutex g_adding_mutex;
 
+const uint32_t TILE_WIDTH_PX = 8;
+const uint32_t TILE_HEIGHT_PX = 12;
+
 // Converts an arbitrary Surface to something like the display format
 // (32-bit RGBA), and converts magenta to transparency if convert_magenta is set
 // and the source surface didn't already have an alpha channel.
@@ -95,7 +97,7 @@ TexposHandle Textures::loadTexture(SDL_Surface* surface) {
     if (!surface)
         return 0; // should be some error, i guess
 
-    auto handle = reinterpret_cast<uintptr_t>(surface); // not the best way?
+    auto handle = reinterpret_cast<TexposHandle>(surface);
     g_handle_to_surface.emplace(handle, surface);
     surface->refcount++; // prevent destruct on next FreeSurface by game
     auto texpos = add_texture(surface);
@@ -103,21 +105,53 @@ TexposHandle Textures::loadTexture(SDL_Surface* surface) {
     return handle;
 }
 
-std::optional<long> Textures::getTexposByHandle(TexposHandle handle) {
+std::vector<TexposHandle> Textures::loadTileset(const std::string& file,
+                                                int tile_px_w = TILE_WIDTH_PX,
+                                                int tile_px_h = TILE_HEIGHT_PX) {
+    std::vector<TexposHandle> handles{};
+
+    SDL_Surface* surface = DFIMG_Load(file.c_str());
+    if (!surface) {
+        ERR(textures).printerr("unable to load textures from '%s'\n", file.c_str());
+        return handles;
+    }
+
+    surface = canonicalize_format(surface);
+    int dimx = surface->w / tile_px_w;
+    int dimy = surface->h / tile_px_h;
+    for (int y = 0; y < dimy; y++) {
+        for (int x = 0; x < dimx; x++) {
+            SDL_Surface* tile = DFSDL_CreateRGBSurface(0, tile_px_w, tile_px_h, 32,
+                                                        surface->format->Rmask, surface->format->Gmask,
+                                                        surface->format->Bmask, surface->format->Amask);
+            SDL_Rect vp{ tile_px_w * x, tile_px_h * y, tile_px_w, tile_px_h };
+            DFSDL_UpperBlit(surface, &vp, tile, NULL);
+            auto handle = Textures::loadTexture(tile);
+            handles.push_back(handle);
+        }
+    }
+
+    DFSDL_FreeSurface(surface);
+    DEBUG(textures).print("loaded %ld textures from '%s'\n", handles.size(), file.c_str());
+
+    return handles;
+}
+
+long Textures::getTexposByHandle(TexposHandle handle) {
     if (!handle)
-        return std::nullopt;
+        return -1;
 
-    if (auto it = g_handle_to_texpos.find(handle); it != g_handle_to_texpos.end())
+    if (g_handle_to_texpos.contains(handle))
-        return it->second;
+        return g_handle_to_texpos[handle];
 
-    if (auto it = g_handle_to_surface.find(handle); it != g_handle_to_surface.end()) {
+    if (g_handle_to_surface.contains(handle)) {
-      it->second->refcount++; // prevent destruct on next FreeSurface by game
+        g_handle_to_surface[handle]->refcount++; // prevent destruct on next FreeSurface by game
-      auto texpos = add_texture(it->second);
+        auto texpos = add_texture(g_handle_to_surface[handle]);
         g_handle_to_texpos.emplace(handle, texpos);
         return texpos;
     }
 
-    return std::nullopt;
+    return -1;
 }
 
 static void reset_texpos() {
@@ -203,8 +237,7 @@ struct tracking_stage_new_arena : df::viewscreen_new_arenast {
 };
 IMPLEMENT_VMETHOD_INTERPOSE(tracking_stage_new_arena, logic);
 
-static void
+static void install_reset_point() {
-  install_reset_point() {
     INTERPOSE_HOOK(tracking_stage_new_region, logic).apply();
     INTERPOSE_HOOK(tracking_stage_adopt_region, logic).apply();
     INTERPOSE_HOOK(tracking_stage_load_region, logic).apply();
@@ -218,9 +251,6 @@ static void uninstall_reset_point() {
     INTERPOSE_HOOK(tracking_stage_new_arena, logic).remove();
 }
 
-const uint32_t TILE_WIDTH_PX = 8;
-const uint32_t TILE_HEIGHT_PX = 12;
-
 static size_t load_tiles_from_image(color_ostream& out, const char* fname,
                                     long* texpos_start,
                                     int tile_w = TILE_WIDTH_PX,
@@ -258,6 +288,11 @@ static size_t load_tiles_from_image(color_ostream& out, const char* fname,
     return count;
 }
 
+void Textures::initDynamic(color_ostream& out) {
+    install_reset_point();
+    DEBUG(textures, out).print("dynamic texture reset points installed");
+}
+
 // DFHack could conceivably be loaded at any time, so we need to be able to
 // handle loading textures before or after a world is loaded.
 // If a world is already loaded, then append our textures to the raws. they'll
@@ -267,11 +302,6 @@ static size_t load_tiles_from_image(color_ostream& out, const char* fname,
 // unloaded.
 //
 void Textures::init(color_ostream& out) {
-    if (!g_dynamic_loaded) {
-        g_dynamic_loaded = true;
-        install_reset_point();
-    }
-
     if (!enabler)
         return;
 
@@ -324,6 +354,10 @@ void Textures::init(color_ostream& out) {
 
 // It's ok to leave NULLs in the raws list (according to usage in g_src)
 void Textures::cleanup() {
+    reset_texpos();
+    reset_surface();
+    uninstall_reset_point();
+
     if (!g_loaded)
         return;
 
@@ -341,13 +375,6 @@ void Textures::cleanup() {
     g_loaded = false;
     g_num_dfhack_textures = 0;
     g_dfhack_logo_texpos_start = -1;
-
-    if (g_dynamic_loaded) {
-        g_dynamic_loaded = false;
-        reset_texpos();
-        reset_surface();
-        uninstall_reset_point();
-    }
 }
 
 long Textures::getDfhackLogoTexposStart() {