FilesExpand file tree

 develop

/

dig-now.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

70 lines (62 loc) · 2.64 KB

 develop

/

dig-now.lua

Copy path

Top

File metadata and controls

• Code
• Blame

70 lines (62 loc) · 2.64 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

local _ENV = mkmodule('plugins.dig-now')

local argparse = require('argparse')

local guidm = require('gui.dwarfmode')

local utils = require('utils')

local function parse_coords(opts, configname, arg)

local cursor = argparse.coords(arg, configname)

utils.assign(opts[configname], cursor)

end

local function parse_percentages(opts, arg)

local nums = argparse.numberList(arg, 'percentages', 4)

for _,percentage in ipairs(nums) do

if percentage < 0 or percentage > 100 then

qerror(('invalid percentages: "%s"; expected format is "<layer>,' ..

'<vein>,<small cluster>,<deep>", where each number is'..

' between 0 and 100, inclusive (e.g. "0,33,100,100")')

:format(arg))

end

end

local config = opts.boulder_percents

config.layer, config.vein, config.small_cluster, config.deep =

nums[1], nums[2], nums[3], nums[4]

end

local function min_to_max(...)

local args = {...}

table.sort(args, function(a, b) return a < b end)

return table.unpack(args)

end

function parse_commandline(opts, ...)

local use_zlevel = false

local positionals = argparse.processArgsGetopt({...}, {

{'c', 'clean',

handler=function() parse_percentages(opts, '0,0,0,0') end},

{'d', 'dump', hasArg=true,

handler=function(arg) parse_coords(opts, 'dump_pos', arg) end},

{'e', 'everywhere',

handler=function() parse_percentages(opts, '100,100,100,100') end},

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

{'p', 'percentages', hasArg=true,

handler=function(arg) parse_percentages(opts, arg) end},

{'z', 'cur-zlevel', handler=function() use_zlevel = true end},

})

if positionals[1] == 'help' then opts.help = true end

if opts.help then return end

if use_zlevel then

local x, y, z = df.global.world.map.x_count - 1,

df.global.world.map.y_count - 1,

df.global.window_z

parse_coords(opts, 'start', ('0,0,%d'):format(z))

parse_coords(opts, 'end', ('%d,%d,%d'):format(x, y, z))

elseif #positionals >= 1 then

parse_coords(opts, 'start', positionals[1])

if #positionals >= 2 then

parse_coords(opts, 'end', positionals[2])

opts.start.x, opts['end'].x = min_to_max(opts.start.x,opts['end'].x)

opts.start.y, opts['end'].y = min_to_max(opts.start.y,opts['end'].y)

opts.start.z, opts['end'].z = min_to_max(opts.start.z,opts['end'].z)

else

utils.assign(opts['end'], opts.start)

end

end

end

return _ENV