FilesExpand file tree

 develop

/

dwarfmonitor.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

175 lines (152 loc) · 4.25 KB

 develop

/

dwarfmonitor.lua

Copy path

Top

File metadata and controls

• Code
• Blame

175 lines (152 loc) · 4.25 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

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

local _ENV = mkmodule('plugins.dwarfmonitor')

local gps = df.global.gps

local gui = require 'gui'

config = {}

widgets = {}

function dmerror(desc)

qerror('dwarfmonitor: ' .. tostring(desc))

end

Widget = defclass(Widget)

function Widget:init(opts)

self.opts = opts

end

function Widget:get_pos()

local x = self.opts.x >= 0 and self.opts.x or gps.dimx + self.opts.x

local y = self.opts.y >= 0 and self.opts.y or gps.dimy + self.opts.y

if self.opts.anchor == 'right' then

x = x - (self:get_width() or 0) + 1

end

return x, y

end

function Widget:render()

if monitor_state(self.opts.type) == false then

return

end

self:update()

local x, y = self:get_pos()

local p = gui.Painter.new_xy(x, y, gps.dimx - 1, y)

self:render_body(p)

end

function Widget:update() end

function Widget:get_width() end

function Widget:render_body() end

Widget_weather = defclass(Widget_weather, Widget)

function Widget_weather:update()

self.counts = get_weather_counts()

end

function Widget_weather:get_width()

if self.counts.rain > 0 then

if self.counts.snow > 0 then

return 9

end

return 4

elseif self.counts.snow > 0 then

return 4

end

return 0

end

function Widget_weather:render_body(p)

if self.counts.rain > 0 then

p:string('Rain', COLOR_LIGHTBLUE):advance(1)

end

if self.counts.snow > 0 then

p:string('Snow', COLOR_WHITE)

end

end

Widget_date = defclass(Widget_date, Widget)

Widget_date.ATTRS = {

output = ''

}

function Widget_date:update()

if not self.opts.format then

self.opts.format = 'Y-M-D'

end

local year = dfhack.world.ReadCurrentYear()

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

local day = dfhack.world.ReadCurrentDay()

self.output = 'Date:'

for i = 1, #self.opts.format do

local c = self.opts.format:sub(i, i)

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

self.output = self.output .. year

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

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

self.output = self.output .. '0'

end

self.output = self.output .. month

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

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

self.output = self.output .. '0'

end

self.output = self.output .. day

else

self.output = self.output .. c

end

end

end

function Widget_date:get_width()

return #self.output

end

function Widget_date:render_body(p)

p:string(self.output, COLOR_GREY)

end

Widget_misery = defclass(Widget_misery, Widget)

function Widget_misery:update()

self.data = get_misery_data()

end

function Widget_misery:get_width()

local w = 2 + 6

for k, v in pairs(self.data) do

w = w + #tostring(v.value)

end

return w

end

function Widget_misery:render_body(p)

p:string("H:", COLOR_WHITE)

for i = 0, 6 do

local v = self.data[i]

p:string(tostring(v.value), v.color)

if not v.last then

p:string("/", COLOR_WHITE)

end

end

end

Widget_cursor = defclass(Widget_cursor, Widget)

function Widget_cursor:update()

if gps.mouse_x == -1 and not self.opts.show_invalid then

self.output = ''

return

end

self.output = (self.opts.format or '(x,y)'):gsub('[xX]', gps.mouse_x):gsub('[yY]', gps.mouse_y)

end

function Widget_cursor:get_width()

return #self.output

end

function Widget_cursor:render_body(p)

p:string(self.output)

end

function render_all()

for _, w in pairs(widgets) do

w:render()

end

end

function load_config()

config = require('json').decode_file('dfhack-config/dwarfmonitor.json')

if not config.widgets then

dmerror('No widgets enabled')

end

if type(config.widgets) ~= 'table' then

dmerror('"widgets" is not a table')

end

widgets = {}

for _, opts in pairs(config.widgets) do

if type(opts) ~= 'table' then dmerror('"widgets" is not an array') end

if not opts.type then dmerror('Widget missing type field') end

local cls = _ENV['Widget_' .. opts.type]

if not cls then

dmerror('Invalid widget type: ' .. opts.type)

end

table.insert(widgets, cls(opts))

end

end

return _ENV