FilesExpand file tree

 develop

/

regrass.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

76 lines (65 loc) · 2.41 KB

 develop

/

regrass.lua

Copy path

Top

File metadata and controls

• Code
• Blame

76 lines (65 loc) · 2.41 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

local _ENV = mkmodule('plugins.regrass')

local argparse = require('argparse')

local utils = require('utils')

local function search_str(s)

return dfhack.upperCp437(dfhack.toSearchNormalized(s))

end

local function find_grass_idx(s) --find plant raw index by id string

local id_str = search_str(s)

for _,grass in ipairs(df.global.world.raws.plants.grasses) do

if search_str(grass.id) == id_str then

return grass.index

end

end

qerror('Plant raw not found: "'..s..'"')

end

local function find_grass(s) --accept index string or match id string

if tonumber(s) then

return argparse.nonnegativeInt(s, 'grass_id')

else

return find_grass_idx(s)

end

end

function parse_commandline(opts, pos_1, pos_2, args)

local plant_str, do_list

local positionals = argparse.processArgsGetopt(args,

{

{'l', 'list', handler=function() do_list = true end},

{'m', 'max', handler=function() opts.max_grass = true end},

{'n', 'new', handler=function() opts.new_grass = true end},

{'a', 'ashes', handler=function() opts.ashes = true end},

{'d', 'buildings', handler=function() opts.buildings = true end},

{'u', 'mud', handler=function() opts.mud = true end},

{'b', 'block', handler=function() opts.block = true end},

{'z', 'zlevel', handler=function() opts.zlevel = true end},

{'f', 'force', handler=function() opts.force = true end},

{'p', 'plant', hasArg=true, handler=function(optarg) plant_str = optarg end},

})

if do_list then

opts.forced_plant = -2 --will print all grass IDs

elseif plant_str then

if not opts.force then

qerror('Use of --plant without --force!')

else

opts.forced_plant = find_grass(plant_str)

end

elseif opts.force then

local grasses = df.global.world.raws.plants.grasses

if #grasses == 0 then

return -1 --no grass raws, world not loaded

else

local rng = dfhack.random.new()

opts.forced_plant = grasses[rng:random(#grasses)].index

end

end

if #positionals > 2 then

qerror('Too many positionals!')

end

if positionals[1] then

utils.assign(pos_1, argparse.coords(positionals[1], 'pos_1', true))

end

if positionals[2] then

utils.assign(pos_2, argparse.coords(positionals[2], 'pos_2', true))

end

end

return _ENV