don't overcount units for misery widget

myk002 · web-flow · commit ff447d46bd2f · 2022-11-20T09:45:23.000-08:00
addendum to DFHack#2366
diff --git a/plugins/lua/dwarfmonitor.lua b/plugins/lua/dwarfmonitor.lua
@@ -116,9 +116,11 @@ end
 function MiseryWidget:overlay_onupdate()
     local counts, num_colors = {}, #self.colors
     for _,unit in ipairs(df.global.world.units.active) do
+        if not dfhack.units.isCitizen(unit, true) then goto continue end
         local stress_category = math.min(num_colors,
                                          dfhack.units.getStressCategory(unit))
         counts[stress_category] = (counts[stress_category] or 0) + 1
+        ::continue::
     end
 
     local width = 2 + num_colors - 1 -- 'H:' plus the slashes
