FilesExpand file tree

 remove-modules

/

slab.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

115 lines (101 loc) · 3.68 KB

 remove-modules

/

slab.lua

Copy path

Top

File metadata and controls

• Code
• Blame

115 lines (101 loc) · 3.68 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

local _ENV = mkmodule('plugins.sort.slab')

local gui = require('gui')

local sortoverlay = require('plugins.sort.sortoverlay')

local widgets = require('gui.widgets')

local building = df.global.game.main_interface.building

-- ----------------------

-- SlabOverlay

--

SlabOverlay = defclass(SlabOverlay, sortoverlay.SortOverlay)

SlabOverlay.ATTRS{

desc='Adds search and filter functionality to the slab engraving panel.',

default_pos={x=-40, y=12},

viewscreens='dwarfmode/ViewSheets/BUILDING/Workshop',

frame={w=57, h=3},

}

function SlabOverlay:init()

local panel = widgets.Panel{

frame_background=gui.CLEAR_PEN,

visible=self:callback('get_key'),

}

panel:addviews{

widgets.BannerPanel{

frame={l=0, t=0, r=0, h=1},

subviews={

widgets.EditField{

view_id='search',

frame={l=1, t=0, r=1},

label_text="Search: ",

key='CUSTOM_ALT_S',

on_change=function(text) self:do_search(text) end,

},

},

},

widgets.BannerPanel{

frame={l=0, t=2, r=0, h=1},

subviews={

widgets.ToggleHotkeyLabel{

view_id='only_needs_slab',

frame={l=1, t=0, r=1},

label="Show only citizens who need a slab:",

key='CUSTOM_SHIFT_E',

initial_option=false,

on_change=function() self:do_search(self.subviews.search.text, true) end,

},

},

},

}

self:addviews{panel}

self:register_handler('SLAB', building.filtered_button,

curry(sortoverlay.single_vector_search,

{

get_search_key_fn=self:callback('get_search_key'),

matches_filters_fn=self:callback('matches_filters'),

}))

end

function SlabOverlay:onInput(keys)

if SlabOverlay.super.onInput(self, keys) then return true end

if keys._MOUSE_L and self:get_key() and self:getMousePos() then

return true

end

end

function SlabOverlay:get_key()

-- DF fails to set building.category back to NONE if there are no units that

-- can be memorialized, so we have to manually check for a populated button vector

if #building.button > 0 and

building.category == df.interface_category_building.SELECT_MEMORIAL_UNIT

then

return 'SLAB'

else

if safe_index(self.state, 'SLAB', 'saved_original') then

-- elements get freed as soon as the screen changes

self.state.SLAB.saved_original = nil

end

end

end

function SlabOverlay:reset()

SlabOverlay.super.reset(self)

self.subviews.only_needs_slab:setOption(false, false)

end

local function get_unit(if_button)

local hf = df.historical_figure.find(if_button.specdata.hist_figure_id)

return hf and df.unit.find(hf.unit_id) or nil

end

function SlabOverlay:get_search_key(if_button)

local unit = get_unit(if_button)

if not unit then return if_button.filter_str end

return sortoverlay.get_unit_search_key(unit)

end

local function needs_slab(if_button)

local unit = get_unit(if_button)

if not unit then return false end

if not dfhack.units.isOwnGroup(unit) then return false end

local info = dfhack.toSearchNormalized(if_button.info)

if not info:find('no slabs engraved', 1, true) then return false end

return info:find('not memorialized', 1, true) or info:find('ghost', 1, true)

end

function SlabOverlay:matches_filters(if_button)

local only_needs_slab = self.subviews.only_needs_slab:getOptionValue()

return not only_needs_slab or needs_slab(if_button)

end

return _ENV