]> git.lizzy.rs Git - xdecor.git/blob - src/workbench.lua
Workbench: Make custom cuttable nodes table global
[xdecor.git] / src / workbench.lua
1 local workbench = {}
2 WB = {}
3 screwdriver = screwdriver or {}
4 local min, ceil = math.min, math.ceil
5
6 -- Nodes allowed to be cut.
7 -- Only the regular, solid blocks without metas or explosivity can be cut.
8 local nodes = {}
9 for node, def in pairs(minetest.registered_nodes) do
10         if (def.drawtype == "normal" or def.drawtype:sub(1,5) == "glass") and
11            (def.groups.cracky or def.groups.choppy) and
12            not def.on_construct and
13            not def.after_place_node and
14            not def.on_rightclick and
15            not def.on_blast and
16            not def.allow_metadata_inventory_take and
17            not (def.groups.not_in_creative_inventory == 1) and
18            not (def.groups.not_cuttable == 1) and
19            not def.groups.wool and
20            (def.tiles and type(def.tiles[1]) == "string" and not
21                 def.tiles[1]:find("default_mineral")) and
22            not def.mesecons and
23            def.description and
24            def.description ~= "" and
25            def.light_source == 0
26         then
27                 nodes[#nodes+1] = node
28         end
29 end
30
31 -- Optionally, you can register custom cuttable nodes in the workbench.
32 WB.custom_nodes_register = {
33         -- "default:leaves",
34 }
35
36 setmetatable(nodes, {
37         __concat = function(t1, t2)
38                 for i=1, #t2 do
39                         t1[#t1+1] = t2[i]
40                 end
41                 return t1
42         end
43 })
44
45 nodes = nodes..WB.custom_nodes_register
46
47 -- Nodeboxes definitions.
48 workbench.defs = {
49         -- Name       Yield   X  Y   Z  W   H  L
50         {"nanoslab",    16, { 0, 0,  0, 8,  1, 8  }},
51         {"micropanel",  16, { 0, 0,  0, 16, 1, 8  }},
52         {"microslab",   8,  { 0, 0,  0, 16, 1, 16 }},
53         {"thinstair",   8,  { 0, 7,  0, 16, 1, 8  },
54                             { 0, 15, 8, 16, 1, 8  }},
55         {"cube",        4,  { 0, 0,  0, 8,  8, 8  }},
56         {"panel",       4,  { 0, 0,  0, 16, 8, 8  }},
57         {"slab",        2,  nil                   },
58         {"doublepanel", 2,  { 0, 0,  0, 16, 8, 8  },
59                             { 0, 8,  8, 16, 8, 8  }},
60         {"halfstair",   2,  { 0, 0,  0, 8,  8, 16 },
61                             { 0, 8,  8, 8,  8, 8  }},
62         {"outerstair",  1,  { 0, 0,  0, 16, 8, 16 },
63                             { 0, 8,  8, 8,  8, 8  }},
64         {"stair",       1,  nil                   },
65         {"innerstair",  1,  { 0, 0,  0, 16, 8, 16 },
66                             { 0, 8,  8, 16, 8, 8  },
67                             { 0, 8,  0, 8,  8, 8  }}
68 }
69
70 -- Tools allowed to be repaired.
71 function workbench:repairable(stack)
72         local tools = {"pick", "axe", "shovel", "sword", "hoe", "armor", "shield"}
73         for _, t in pairs(tools) do
74                 if stack:find(t) then return true end
75         end
76         return false
77 end
78
79 function workbench:get_output(inv, input, name)
80         if inv:is_empty("input") then
81                 inv:set_list("forms", {}) return
82         end
83
84         local output = {}
85         for _, n in pairs(self.defs) do
86                 local count = min(n[2] * input:get_count(), input:get_stack_max())
87                 local item = name.."_"..n[1]
88                 if not n[3] then item = "stairs:"..n[1].."_"..name:match(":(.*)") end
89                 output[#output+1] = item.." "..count
90         end
91         inv:set_list("forms", output)
92 end
93
94 local formspecs = {
95         -- Main formspec.
96         [[ label[0.9,1.23;Cut]
97            label[0.9,2.23;Repair]
98            box[-0.05,1;2.05,0.9;#555555]
99            box[-0.05,2;2.05,0.9;#555555]
100            button[0,0;2,1;craft;Crafting]
101            button[2,0;2,1;storage;Storage]
102            image[3,1;1,1;gui_furnace_arrow_bg.png^[transformR270]
103            image[0,1;1,1;worktable_saw.png]
104            image[0,2;1,1;worktable_anvil.png]
105            image[3,2;1,1;hammer_layout.png]
106            list[context;input;2,1;1,1;]
107            list[context;tool;2,2;1,1;]
108            list[context;hammer;3,2;1,1;]
109            list[context;forms;4,0;4,3;] ]],
110         -- Crafting formspec.
111         [[ image[5,1;1,1;gui_furnace_arrow_bg.png^[transformR270]
112            button[0,0;1.5,1;back;< Back]
113            list[current_player;craft;2,0;3,3;]
114            list[current_player;craftpreview;6,1;1,1;]
115            listring[current_player;main]
116            listring[current_player;craft] ]],
117         -- Storage formspec.
118         [[ list[context;storage;0,1;8,2;]
119            button[0,0;1.5,1;back;< Back]
120            listring[context;storage]
121            listring[current_player;main] ]]
122 }
123
124 function workbench:set_formspec(meta, id)
125         meta:set_string("formspec", "size[8,7;]list[current_player;main;0,3.25;8,4;]"..
126                         formspecs[id]..xbg..default.get_hotbar_bg(0,3.25))
127 end
128
129 function workbench.construct(pos)
130         local meta = minetest.get_meta(pos)
131         local inv = meta:get_inventory()
132
133         inv:set_size("tool", 1)
134         inv:set_size("input", 1)
135         inv:set_size("hammer", 1)
136         inv:set_size("forms", 4*3)
137         inv:set_size("storage", 8*2)
138
139         meta:set_string("infotext", "Work Bench")
140         workbench:set_formspec(meta, 1)
141 end
142
143 function workbench.fields(pos, _, fields)
144         local meta = minetest.get_meta(pos)
145         if     fields.back    then workbench:set_formspec(meta, 1)
146         elseif fields.craft   then workbench:set_formspec(meta, 2)
147         elseif fields.storage then workbench:set_formspec(meta, 3) end
148 end
149
150 function workbench.dig(pos)
151         local inv = minetest.get_meta(pos):get_inventory()
152         return inv:is_empty("input") and inv:is_empty("hammer") and
153                 inv:is_empty("tool") and inv:is_empty("storage")
154 end
155
156 function workbench.timer(pos)
157         local timer = minetest.get_node_timer(pos)
158         local inv = minetest.get_meta(pos):get_inventory()
159         local tool = inv:get_stack("tool", 1)
160         local hammer = inv:get_stack("hammer", 1)
161
162         if tool:is_empty() or hammer:is_empty() or tool:get_wear() == 0 then
163                 timer:stop() return
164         end
165
166         -- Tool's wearing range: 0-65535 | 0 = new condition.
167         tool:add_wear(-500)
168         hammer:add_wear(700)
169
170         inv:set_stack("tool", 1, tool)
171         inv:set_stack("hammer", 1, hammer)
172         return true
173 end
174
175 function workbench.put(_, listname, _, stack)
176         local stackname = stack:get_name()
177         if (listname == "tool" and stack:get_wear() > 0 and
178             workbench:repairable(stackname)) or
179            (listname == "input" and minetest.registered_nodes[stackname.."_cube"]) or
180            (listname == "hammer" and stackname == "xdecor:hammer") or
181             listname == "storage" then
182                 return stack:get_count()
183         end
184         return 0
185 end
186
187 function workbench.take(_, listname, _, stack, player)
188         if listname == "forms" then
189                 local inv = player:get_inventory()
190                 if inv:room_for_item("main", stack:get_name()) then return -1 end
191                 return 0
192         end
193         return stack:get_count()
194 end
195
196 function workbench.move(_, from_list, _, to_list, _, count)
197         if to_list == "storage" and from_list ~= "forms" then return count end
198         return 0
199 end
200
201 function workbench.on_put(pos, listname, _, stack)
202         local inv = minetest.get_meta(pos):get_inventory()
203         if listname == "input" then
204                 local input = inv:get_stack("input", 1)
205                 workbench:get_output(inv, input, stack:get_name())
206         elseif listname == "tool" or listname == "hammer" then
207                 local timer = minetest.get_node_timer(pos)
208                 timer:start(3.0)
209         end
210 end
211
212 function workbench.on_take(pos, listname, index, stack)
213         local inv = minetest.get_meta(pos):get_inventory()
214         local input = inv:get_stack("input", 1)
215
216         if listname == "input" then
217                 if stack:get_name() == input:get_name() then
218                         workbench:get_output(inv, input, stack:get_name())
219                 else
220                         inv:set_list("forms", {})
221                 end
222         elseif listname == "forms" then
223                 input:take_item(ceil(stack:get_count() / workbench.defs[index][2]))
224                 inv:set_stack("input", 1, input)
225                 workbench:get_output(inv, input, input:get_name())
226         end
227 end
228
229 xdecor.register("workbench", {
230         description = "Work Bench",
231         groups = {cracky=2, choppy=2, oddly_breakable_by_hand=1},
232         sounds = default.node_sound_wood_defaults(),
233         tiles = {"xdecor_workbench_top.png",   "xdecor_workbench_top.png",
234                  "xdecor_workbench_sides.png", "xdecor_workbench_sides.png",
235                  "xdecor_workbench_front.png", "xdecor_workbench_front.png"},
236         on_rotate = screwdriver.rotate_simple,
237         can_dig = workbench.dig,
238         on_timer = workbench.timer,
239         on_construct = workbench.construct,
240         on_receive_fields = workbench.fields,
241         on_metadata_inventory_put = workbench.on_put,
242         on_metadata_inventory_take = workbench.on_take,
243         allow_metadata_inventory_put = workbench.put,
244         allow_metadata_inventory_take = workbench.take,
245         allow_metadata_inventory_move = workbench.move
246 })
247
248 for _, d in pairs(workbench.defs) do
249 for i=1, #nodes do
250         local node = nodes[i]
251         local def = minetest.registered_nodes[node]
252
253         if d[3] then
254                 local groups = {}
255                 local tiles
256                 groups.not_in_creative_inventory = 1
257
258                 for k, v in pairs(def.groups) do
259                         if k ~= "wood" and k ~= "stone" and k ~= "level" then
260                                 groups[k] = v
261                         end
262                 end
263
264                 if def.tiles then
265                         if #def.tiles > 1 and not (def.drawtype:sub(1,5) == "glass") then
266                                 tiles = def.tiles
267                         else
268                                 tiles = {def.tiles[1]}
269                         end
270                 else
271                         tiles = {def.tile_images[1]}
272                 end
273
274                 if not minetest.registered_nodes["stairs:slab_"..node:match(":(.*)")] then
275                         stairs.register_stair_and_slab(node:match(":(.*)"), node,
276                                 groups, tiles, def.description.." Stair",
277                                 def.description.." Slab", def.sounds)
278                 end
279
280                 minetest.register_node(":"..node.."_"..d[1], {
281                         description = def.description.." "..d[1]:gsub("^%l", string.upper),
282                         paramtype = "light",
283                         paramtype2 = "facedir",
284                         drawtype = "nodebox",
285                         sounds = def.sounds,
286                         tiles = tiles,
287                         groups = groups,
288                         -- `unpack` has been changed to `table.unpack` in newest Lua versions.
289                         node_box = xdecor.pixelbox(16, {unpack(d, 3)}),
290                         sunlight_propagates = true,
291                         on_place = minetest.rotate_node
292                 })
293         end
294 end
295 end
296