ensure face textures get reloaded on map load

plugins/lua/sort.lua

@@ -6,6 +6,8 @@ local setbelief = reqscript('modtools/set-belief')
 local utils = require('utils')
 local widgets = require('gui.widgets')
 
+local GLOBAL_KEY = 'sort'
+
 local CH_UP = string.char(30)
 local CH_DN = string.char(31)
 
@@ -1196,7 +1198,8 @@ local to_pen = dfhack.pen.parse
 local DASH_PEN = to_pen{ch='-', fg=COLOR_WHITE, keep_lower=true}
 
 local FACE_TILES = {}
-for idx=0,6 do
+local function init_face_tiles()
+    for idx=0,6 do
         FACE_TILES[idx] = {}
         local face_off = (6 - idx) * 2
         for y=0,1 do
@@ -1205,15 +1208,16 @@ for idx=0,6 do
                 ensure_key(FACE_TILES[idx], y)[x] = tile
             end
         end
-end
+    end
 
-for idx,color in ipairs{COLOR_RED, COLOR_LIGHTRED, COLOR_YELLOW, COLOR_WHITE, COLOR_GREEN, COLOR_LIGHTGREEN, COLOR_LIGHTCYAN} do
+    for idx,color in ipairs{COLOR_RED, COLOR_LIGHTRED, COLOR_YELLOW, COLOR_WHITE, COLOR_GREEN, COLOR_LIGHTGREEN, COLOR_LIGHTCYAN} do
         local face = {}
         ensure_key(face, 0)[0] = to_pen{tile=FACE_TILES[idx-1][0][0], ch=1, fg=color}
         ensure_key(face, 0)[1] = to_pen{tile=FACE_TILES[idx-1][0][1], ch='\\', fg=color}
         ensure_key(face, 1)[0] = to_pen{tile=FACE_TILES[idx-1][1][0], ch='\\', fg=color}
         ensure_key(face, 1)[1] = to_pen{tile=FACE_TILES[idx-1][1][1], ch='/', fg=color}
         FACE_TILES[idx-1] = face
+    end
 end
 
 function get_stress_face_tile(idx, x, y)
@@ -1261,6 +1265,14 @@ OVERLAY_WIDGETS = {
     squad_annotation=SquadAnnotationOverlay,
 }
 
+dfhack.onStateChange[GLOBAL_KEY] = function(sc)
+    if sc ~= SC_MAP_LOADED or df.global.gamemode ~= df.game_mode.DWARF then
+        return
+    end
+
+    init_face_tiles()
+end
+
 --[[
 local utils = require('utils')
 local units = require('plugins.sort.units')