FilesExpand file tree

 remove-modules

/

buildingplan.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

153 lines (130 loc) · 4.43 KB

 remove-modules

/

buildingplan.lua

Copy path

Top

File metadata and controls

• Code
• Blame

153 lines (130 loc) · 4.43 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

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

local _ENV = mkmodule('plugins.buildingplan')

--[[

Public native functions:

* bool isPlannableBuilding(df::building_type type, int16_t subtype, int32_t custom)

* bool isPlannedBuilding(df::building *bld)

* void addPlannedBuilding(df::building *bld)

* void doCycle()

* void scheduleCycle()

--]]

local argparse = require('argparse')

local inspector = require('plugins.buildingplan.inspectoroverlay')

local mechanisms = require('plugins.buildingplan.mechanisms')

local pens = require('plugins.buildingplan.pens')

local planner = require('plugins.buildingplan.planneroverlay')

local unlink_mechanisms = require('plugins.buildingplan.unlink_mechanisms')

require('dfhack.buildings')

local function process_args(opts, args)

if args[1] == 'help' then

opts.help = true

return

end

return argparse.processArgsGetopt(args, {

{'h', 'help', handler=function() opts.help = true end},

})

end

function parse_commandline(...)

local args, opts = {...}, {}

local positionals = process_args(opts, args)

if opts.help then

return false

end

local command = table.remove(positionals, 1)

if not command or command == 'status' then

printStatus()

elseif command == 'set' and positionals then

setSetting(positionals[1], positionals[2] == 'true')

elseif command == 'reset' then

resetFilters()

elseif command == 'ignore' and positionals then

setIgnoreBurrow(positionals[1] or "")

else

return false

end

return true

end

function is_suspendmanager_enabled()

return require('plugins.suspendmanager').isEnabled()

end

function get_num_filters(btype, subtype, custom)

local filters = dfhack.buildings.getFiltersByType({}, btype, subtype, custom)

return filters and #filters or 0

end

function get_job_item(btype, subtype, custom, index)

local filters = dfhack.buildings.getFiltersByType({}, btype, subtype, custom)

if not filters or not filters[index] then return nil end

local obj = df.job_item:new()

obj:assign(filters[index])

return obj

end

local function to_title_case(str)

str = str:gsub('(%a)([%w_]*)',

function (first, rest) return first:upper()..rest:lower() end)

str = str:gsub('_', ' ')

return str

end

function get_desc(filter)

local desc = 'Unknown'

if filter.has_tool_use and filter.has_tool_use > -1 then

desc = to_title_case(df.tool_uses[filter.has_tool_use])

elseif filter.flags2 and filter.flags2.screw then

desc = 'Screw'

elseif filter.item_type and filter.item_type > -1 then

desc = to_title_case(df.item_type[filter.item_type])

elseif filter.vector_id and filter.vector_id > -1 then

desc = to_title_case(df.job_item_vector_id[filter.vector_id])

elseif filter.flags2 and filter.flags2.building_material then

if filter.flags2.magma_safe then

desc = 'Magma-safe material'

elseif filter.flags2.fire_safe then

desc = 'Fire-safe material'

else

desc = 'Building material'

end

end

if desc:endswith('s') then

desc = desc:sub(1,-2)

end

if desc == 'Trappart' then

desc = 'Mechanism'

elseif desc == 'Wood' then

desc = 'Log'

elseif desc == 'Any weapon' then

desc = 'Weapon'

elseif desc == 'Any spike' then

desc = 'Spike'

elseif desc == 'Ballistapart' then

desc = 'Ballista part'

elseif desc == 'Catapultpart' then

desc = 'Catapult part'

elseif desc == 'Smallgem' then

desc = 'Small, cut gem'

end

return desc

end

function reload_pens()

pens.reload_pens()

end

function signal_reset()

planner.reset_counts_flag = true

inspector.reset_inspector_flag = true

end

-- for use during development to reload all buildingplan modules

function reload_modules()

reload('plugins.buildingplan.pens')

reload('plugins.buildingplan.filterselection')

reload('plugins.buildingplan.itemselection')

reload('plugins.buildingplan.planneroverlay')

reload('plugins.buildingplan.inspectoroverlay')

reload('plugins.buildingplan.mechanisms')

reload('plugins.buildingplan.unlink_mechanisms')

reload('plugins.buildingplan')

end

OVERLAY_WIDGETS = {

planner=planner.PlannerOverlay,

inspector=inspector.InspectorOverlay,

mechanisms=mechanisms.MechanismOverlay,

mechanism_free=unlink_mechanisms.MechItemOverlay,

mechanism_unlink=unlink_mechanisms.MechLinkOverlay,

}

return _ENV