local rain, snow = false, false

local cw = df.global.current_weather

for i=0,4 do

for j=0,4 do

weather = cw[i][j]

if weather == df.weather_type.Rain then rain = true end

if weather == df.weather_type.Snow then snow = true end

end

end

self.frame.w = (rain and 4 or 0) + (snow and 4 or 0) +

((snow and rain) and 1 or 0)

self.rain, self.snow = rain, snow

local ok, config = pcall(json.decode_file, DWARFMONITOR_CONFIG_FILE)

if not ok or not config.date_format then

return 'Y-M-D'

end

return config.date_format

local year = dfhack.world.ReadCurrentYear()

local month = dfhack.world.ReadCurrentMonth() + 1

local day = dfhack.world.ReadCurrentDay()

local fmt = self.fmt

local datestr = 'Date:'

for i=1,#fmt do

local c = fmt:sub(i, i)

if c == 'y' or c == 'Y' then

datestr = datestr .. year

elseif c == 'm' or c == 'M' then

if c == 'M' and month < 10 then

datestr = datestr .. '0'

end

datestr = datestr .. month

elseif c == 'd' or c == 'D' then

if c == 'D' and day < 10 then

datestr = datestr .. '0'

end

datestr = datestr .. day

else

datestr = datestr .. c

end

end

self.frame.w = #datestr

self.datestr = datestr

self.colors = getStressCategoryColors()

self.stress_category_counts = {}

self.frame.w = 2*#self.colors + 1

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)+1)

counts[stress_category] = (counts[stress_category] or 0) + 1

::continue::

end

local width = 2 + num_colors - 1 -- 'H:' plus the slashes

for i=1,num_colors do

width = width + #tostring(counts[i] or 0)

end

self.stress_category_counts = counts

self.frame.w = width

dc:string('H:', COLOR_WHITE)

local counts = self.stress_category_counts

for i,color in ipairs(self.colors) do

dc:string(tostring(counts[i] or 0), color)

if i < #self.colors then dc:string('/', COLOR_WHITE) end

end

local ok, config = pcall(json.decode_file, DWARFMONITOR_CONFIG_FILE)

if ok then

self.type = tostring(config.coords_type or self.type)

self.short = config.coords_short

else

dfhack.printerr(('Failed to load %s: %s'):format(DWARFMONITOR_CONFIG_FILE, config))

end

local text = {}

for k, v in pairs(CursorWidget.COORD_TYPES) do

-- fall back to 'all' behavior if type isn't recognized

if k == self.type or not CursorWidget.COORD_TYPES[self.type] then

local coords = v.get_coords()

if coords then

local prefix = self.short and '' or (v.description .. ' ')

local coord_str = table.concat({coords.x, coords.y, coords.z}, ',')

table.insert(text, ('%s(%s)'):format(prefix, coord_str))

end

end

end

local width = 0

for i,line in ipairs(text) do

dc:seek(0, i-1):string(line)

width = math.max(width, #line)

end

self.frame.w = width

self.frame.h = #text