Add trackstop and rollers overlays

changelog.txt

@@ -28,6 +28,7 @@ Template for new versions:
 
 ## New Tools
 - `sync-windmills`: synchronize or randomize movement of active windmills
+- `trackstop`: new overlay to allow changing track stop dump direction and friction and roller direction and speed after construction
 
 ## New Features
 - `gui/design`: show selected dimensions next to the mouse cursor when designating with vanilla tools, for example when painting a burrow or designating digging

docs/trackstop.rst

@@ -0,0 +1,10 @@
+trackstop
+=========
+
+.. dfhack-tool::
+    :summary: Overlay to allow changing track stop friction and dump direction and roller direction and speed after construction.
+    :tags: fort gameplay buildings interface
+
+This script provides 2 overlays that are managed by the `overlay` framework.
+The trackstop overlay allows the player to change the friction and dump direction of a selected track stop after it has been constructed.
+The rollers overlay allows the player to change the roller direction and speed of a selected track stop after it has been constructed.

trackstop.lua

@@ -0,0 +1,259 @@
+-- Overlay to allow changing track stop friction and dump direction after construction
+--@ module = true
+local gui = require('gui')
+local widgets = require('gui.widgets')
+local overlay = require('plugins.overlay')
+
+local NORTH = 'North'
+local EAST = 'East'
+local SOUTH = 'South'
+local WEST = 'West'
+
+local LOW = 'Low'
+local MEDIUM = 'Medium'
+local HIGH = 'High'
+local HIGHER = 'Higher'
+local MAX = 'Max'
+
+local NONE = 'None'
+
+local FRICTION_MAP = {
+  [NONE] = 10,
+  [LOW] = 50,
+  [MEDIUM] = 500,
+  [HIGH] = 10000,
+  [MAX] = 50000,
+}
+
+local FRICTION_MAP_REVERSE = {}
+for k, v in pairs(FRICTION_MAP) do
+  FRICTION_MAP_REVERSE[v] = k
+end
+
+local SPEED_MAP = {
+  [LOW] = 10000,
+  [MEDIUM] = 20000,
+  [HIGH] = 30000,
+  [HIGHER] = 40000,
+  [MAX] = 50000,
+}
+
+local SPEED_MAP_REVERSE = {}
+for k, v in pairs(SPEED_MAP) do
+  SPEED_MAP_REVERSE[v] = k
+end
+
+local DIRECTION_MAP = {
+  [NORTH] = df.screw_pump_direction.FromSouth,
+  [EAST] = df.screw_pump_direction.FromWest,
+  [SOUTH] = df.screw_pump_direction.FromNorth,
+  [WEST] = df.screw_pump_direction.FromEast,
+}
+
+local DIRECTION_MAP_REVERSE = {}
+for k, v in pairs(DIRECTION_MAP) do
+  DIRECTION_MAP_REVERSE[v] = k
+end
+
+TrackStopOverlay = defclass(TrackStopOverlay, overlay.OverlayWidget)
+TrackStopOverlay.ATTRS{
+  default_pos={x=-71, y=29},
+  default_enabled=true,
+  viewscreens='dwarfmode/ViewSheets/BUILDING/Trap',
+  frame={w=27, h=4},
+  frame_style=gui.MEDIUM_FRAME,
+  frame_background=gui.CLEAR_PEN,
+}
+
+function TrackStopOverlay:getFriction()
+  return dfhack.gui.getSelectedBuilding().friction
+end
+
+function TrackStopOverlay:setFriction(friction)
+  local building = dfhack.gui.getSelectedBuilding()
+
+  building.friction = FRICTION_MAP[friction]
+end
+
+function TrackStopOverlay:getDumpDirection()
+  local building = dfhack.gui.getSelectedBuilding()
+  local use_dump = building.use_dump
+  local dump_x_shift = building.dump_x_shift
+  local dump_y_shift = building.dump_y_shift
+
+  if use_dump == 0 then
+    return NONE
+  else
+    if dump_x_shift == 0 and dump_y_shift == -1 then
+      return NORTH
+    elseif dump_x_shift == 1 and dump_y_shift == 0 then
+      return EAST
+    elseif dump_x_shift == 0 and dump_y_shift == 1 then
+      return SOUTH
+    elseif dump_x_shift == -1 and dump_y_shift == 0 then
+      return WEST
+    end
+  end
+end
+
+function TrackStopOverlay:setDumpDirection(direction)
+  local building = dfhack.gui.getSelectedBuilding()
+
+  if direction == NONE then
+    building.use_dump = 0
+    building.dump_x_shift = 0
+    building.dump_y_shift = 0
+  elseif direction == NORTH then
+    building.use_dump = 1
+    building.dump_x_shift = 0
+    building.dump_y_shift = -1
+  elseif direction == EAST then
+    building.use_dump = 1
+    building.dump_x_shift = 1
+    building.dump_y_shift = 0
+  elseif direction == SOUTH then
+    building.use_dump = 1
+    building.dump_x_shift = 0
+    building.dump_y_shift = 1
+  elseif direction == WEST then
+    building.use_dump = 1
+    building.dump_x_shift = -1
+    building.dump_y_shift = 0
+  end
+end
+
+function TrackStopOverlay:render(dc)
+  if not self:shouldRender() then
+    return
+  end
+
+  local building = dfhack.gui.getSelectedBuilding()
+  local friction = building.friction
+  local friction_cycle = self.subviews.friction
+
+  friction_cycle:setOption(FRICTION_MAP_REVERSE[friction])
+
+  self.subviews.dump_direction:setOption(self:getDumpDirection())
+
+  TrackStopOverlay.super.render(self, dc)
+end
+
+function TrackStopOverlay:shouldRender()
+  local building = dfhack.gui.getSelectedBuilding()
+  return building and building.trap_type == df.trap_type.TrackStop
+end
+
+function TrackStopOverlay:onInput(keys)
+  if not self:shouldRender() then
+    return
+  end
+  TrackStopOverlay.super.onInput(self, keys)
+end
+
+function TrackStopOverlay:init()
+  self:addviews{
+    widgets.Label{
+      frame={t=0, l=0},
+      text='Dump',
+    },
+    widgets.CycleHotkeyLabel{
+      frame={t=0, l=9},
+      key='CUSTOM_CTRL_X',
+      options={NONE, NORTH, EAST, SOUTH, WEST},
+      view_id='dump_direction',
+      on_change=function(val) self:setDumpDirection(val) end,
+    },
+    widgets.Label{
+      frame={t=1, l=0},
+      text='Friction',
+    },
+    widgets.CycleHotkeyLabel{
+      frame={t=1, l=9},
+      key='CUSTOM_CTRL_F',
+      options={NONE, LOW, MEDIUM, HIGH, MAX},
+      view_id='friction',
+      on_change=function(val) self:setFriction(val) end,
+    },
+  }
+end
+
+RollerOverlay = defclass(RollerOverlay, overlay.OverlayWidget)
+RollerOverlay.ATTRS{
+  default_pos={x=-71, y=29},
+  default_enabled=true,
+  viewscreens='dwarfmode/ViewSheets/BUILDING/Rollers',
+  frame={w=27, h=4},
+  frame_style=gui.MEDIUM_FRAME,
+  frame_background=gui.CLEAR_PEN,
+}
+
+function RollerOverlay:getDirection()
+  local building = dfhack.gui.getSelectedBuilding()
+  local direction = building.direction
+
+  return DIRECTION_MAP_REVERSE[direction]
+end
+
+function RollerOverlay:setDirection(direction)
+  local building = dfhack.gui.getSelectedBuilding()
+
+  building.direction = DIRECTION_MAP[direction]
+end
+
+function RollerOverlay:getSpeed()
+  local building = dfhack.gui.getSelectedBuilding()
+  local speed = building.speed
+
+  return SPEED_MAP_REVERSE[speed]
+end
+
+function RollerOverlay:setSpeed(speed)
+  local building = dfhack.gui.getSelectedBuilding()
+
+  building.speed = SPEED_MAP[speed]
+end
+
+function RollerOverlay:render(dc)
+  local building = dfhack.gui.getSelectedBuilding()
+
+  self.subviews.direction:setOption(DIRECTION_MAP_REVERSE[building.direction])
+  self.subviews.speed:setOption(SPEED_MAP_REVERSE[building.speed])
+
+  TrackStopOverlay.super.render(self, dc)
+end
+
+function RollerOverlay:init()
+  self:addviews{
+    widgets.Label{
+      frame={t=0, l=0},
+      text='Direction',
+    },
+    widgets.CycleHotkeyLabel{
+      frame={t=0, l=10},
+      key='CUSTOM_CTRL_X',
+      options={NORTH, EAST, SOUTH, WEST},
+      view_id='direction',
+      on_change=function(val) self:setDirection(val) end,
+    },
+    widgets.Label{
+      frame={t=1, l=0},
+      text='Speed',
+    },
+    widgets.CycleHotkeyLabel{
+      frame={t=1, l=10},
+      key='CUSTOM_CTRL_F',
+      options={LOW, MEDIUM, HIGH, HIGHER, MAX},
+      view_id='speed',
+      on_change=function(val) self:setSpeed(val) end,
+    },
+  }
+end
+
+OVERLAY_WIDGETS = {
+  trackstop=TrackStopOverlay,
+  rollers=RollerOverlay,
+}
+
+if not dfhack_flags.module then
+  main{...}
+end