ObjectRef: Add add_velocity() (#3208)
[oweals/minetest.git] / builtin / game / falling.lua
1 -- Minetest: builtin/item.lua
2
3 local builtin_shared = ...
4
5 --
6 -- Falling stuff
7 --
8
9 core.register_entity(":__builtin:falling_node", {
10         initial_properties = {
11                 visual = "wielditem",
12                 visual_size = {x = 0.667, y = 0.667},
13                 textures = {},
14                 physical = true,
15                 is_visible = false,
16                 collide_with_objects = false,
17                 collisionbox = {-0.5, -0.5, -0.5, 0.5, 0.5, 0.5},
18         },
19
20         node = {},
21         meta = {},
22
23         set_node = function(self, node, meta)
24                 self.node = node
25                 self.meta = meta or {}
26                 self.object:set_properties({
27                         is_visible = true,
28                         textures = {node.name},
29                 })
30         end,
31
32         get_staticdata = function(self)
33                 local ds = {
34                         node = self.node,
35                         meta = self.meta,
36                 }
37                 return core.serialize(ds)
38         end,
39
40         on_activate = function(self, staticdata)
41                 self.object:set_armor_groups({immortal = 1})
42
43                 local ds = core.deserialize(staticdata)
44                 if ds and ds.node then
45                         self:set_node(ds.node, ds.meta)
46                 elseif ds then
47                         self:set_node(ds)
48                 elseif staticdata ~= "" then
49                         self:set_node({name = staticdata})
50                 end
51         end,
52
53         on_step = function(self, dtime)
54                 -- Set gravity
55                 local acceleration = self.object:getacceleration()
56                 if not vector.equals(acceleration, {x = 0, y = -10, z = 0}) then
57                         self.object:setacceleration({x = 0, y = -10, z = 0})
58                 end
59                 -- Turn to actual node when colliding with ground, or continue to move
60                 local pos = self.object:getpos()
61                 -- Position of bottom center point
62                 local bcp = {x = pos.x, y = pos.y - 0.7, z = pos.z}
63                 -- 'bcn' is nil for unloaded nodes
64                 local bcn = core.get_node_or_nil(bcp)
65                 -- Delete on contact with ignore at world edges
66                 if bcn and bcn.name == "ignore" then
67                         self.object:remove()
68                         return
69                 end
70                 local bcd = bcn and core.registered_nodes[bcn.name]
71                 if bcn and
72                                 (not bcd or bcd.walkable or
73                                 (core.get_item_group(self.node.name, "float") ~= 0 and
74                                 bcd.liquidtype ~= "none")) then
75                         if bcd and bcd.leveled and
76                                         bcn.name == self.node.name then
77                                 local addlevel = self.node.level
78                                 if not addlevel or addlevel <= 0 then
79                                         addlevel = bcd.leveled
80                                 end
81                                 if core.add_node_level(bcp, addlevel) == 0 then
82                                         self.object:remove()
83                                         return
84                                 end
85                         elseif bcd and bcd.buildable_to and
86                                         (core.get_item_group(self.node.name, "float") == 0 or
87                                         bcd.liquidtype == "none") then
88                                 core.remove_node(bcp)
89                                 return
90                         end
91                         local np = {x = bcp.x, y = bcp.y + 1, z = bcp.z}
92                         -- Check what's here
93                         local n2 = core.get_node(np)
94                         local nd = core.registered_nodes[n2.name]
95                         -- If it's not air or liquid, remove node and replace it with
96                         -- it's drops
97                         if n2.name ~= "air" and (not nd or nd.liquidtype == "none") then
98                                 core.remove_node(np)
99                                 if nd and nd.buildable_to == false then
100                                         -- Add dropped items
101                                         local drops = core.get_node_drops(n2, "")
102                                         for _, dropped_item in pairs(drops) do
103                                                 core.add_item(np, dropped_item)
104                                         end
105                                 end
106                                 -- Run script hook
107                                 for _, callback in pairs(core.registered_on_dignodes) do
108                                         callback(np, n2)
109                                 end
110                         end
111                         -- Create node and remove entity
112                         local def = core.registered_nodes[self.node.name]
113                         if def then
114                                 core.add_node(np, self.node)
115                                 if self.meta then
116                                         local meta = core.get_meta(np)
117                                         meta:from_table(self.meta)
118                                 end
119                                 if def.sounds and def.sounds.place and def.sounds.place.name then
120                                         core.sound_play(def.sounds.place, {pos = np})
121                                 end
122                         end
123                         self.object:remove()
124                         core.check_for_falling(np)
125                         return
126                 end
127                 local vel = self.object:getvelocity()
128                 if vector.equals(vel, {x = 0, y = 0, z = 0}) then
129                         local npos = self.object:getpos()
130                         self.object:setpos(vector.round(npos))
131                 end
132         end
133 })
134
135 local function spawn_falling_node(p, node, meta)
136         local obj = core.add_entity(p, "__builtin:falling_node")
137         if obj then
138                 obj:get_luaentity():set_node(node, meta)
139         end
140 end
141
142 function core.spawn_falling_node(pos)
143         local node = core.get_node(pos)
144         if node.name == "air" or node.name == "ignore" then
145                 return false
146         end
147         local obj = core.add_entity(pos, "__builtin:falling_node")
148         if obj then
149                 obj:get_luaentity():set_node(node)
150                 core.remove_node(pos)
151                 return true
152         end
153         return false
154 end
155
156 local function drop_attached_node(p)
157         local n = core.get_node(p)
158         local drops = core.get_node_drops(n, "")
159         local def = core.registered_items[n.name]
160         if def and def.preserve_metadata then
161                 local oldmeta = core.get_meta(p):to_table().fields
162                 -- Copy pos and node because the callback can modify them.
163                 local pos_copy = {x=p.x, y=p.y, z=p.z}
164                 local node_copy = {name=n.name, param1=n.param1, param2=n.param2}
165                 local drop_stacks = {}
166                 for k, v in pairs(drops) do
167                         drop_stacks[k] = ItemStack(v)
168                 end
169                 drops = drop_stacks
170                 def.preserve_metadata(pos_copy, node_copy, oldmeta, drops)
171         end
172         core.remove_node(p)
173         for _, item in pairs(drops) do
174                 local pos = {
175                         x = p.x + math.random()/2 - 0.25,
176                         y = p.y + math.random()/2 - 0.25,
177                         z = p.z + math.random()/2 - 0.25,
178                 }
179                 core.add_item(pos, item)
180         end
181 end
182
183 function builtin_shared.check_attached_node(p, n)
184         local def = core.registered_nodes[n.name]
185         local d = {x = 0, y = 0, z = 0}
186         if def.paramtype2 == "wallmounted" or
187                         def.paramtype2 == "colorwallmounted" then
188                 -- The fallback vector here is in case 'wallmounted to dir' is nil due
189                 -- to voxelmanip placing a wallmounted node without resetting a
190                 -- pre-existing param2 value that is out-of-range for wallmounted.
191                 -- The fallback vector corresponds to param2 = 0.
192                 d = core.wallmounted_to_dir(n.param2) or {x = 0, y = 1, z = 0}
193         else
194                 d.y = -1
195         end
196         local p2 = vector.add(p, d)
197         local nn = core.get_node(p2).name
198         local def2 = core.registered_nodes[nn]
199         if def2 and not def2.walkable then
200                 return false
201         end
202         return true
203 end
204
205 --
206 -- Some common functions
207 --
208
209 function core.check_single_for_falling(p)
210         local n = core.get_node(p)
211         if core.get_item_group(n.name, "falling_node") ~= 0 then
212                 local p_bottom = {x = p.x, y = p.y - 1, z = p.z}
213                 -- Only spawn falling node if node below is loaded
214                 local n_bottom = core.get_node_or_nil(p_bottom)
215                 local d_bottom = n_bottom and core.registered_nodes[n_bottom.name]
216                 if d_bottom and
217
218                                 (core.get_item_group(n.name, "float") == 0 or
219                                 d_bottom.liquidtype == "none") and
220
221                                 (n.name ~= n_bottom.name or (d_bottom.leveled and
222                                 core.get_node_level(p_bottom) <
223                                 core.get_node_max_level(p_bottom))) and
224
225                                 (not d_bottom.walkable or d_bottom.buildable_to) then
226                         n.level = core.get_node_level(p)
227                         local meta = core.get_meta(p)
228                         local metatable = {}
229                         if meta ~= nil then
230                                 metatable = meta:to_table()
231                         end
232                         core.remove_node(p)
233                         spawn_falling_node(p, n, metatable)
234                         return true
235                 end
236         end
237
238         if core.get_item_group(n.name, "attached_node") ~= 0 then
239                 if not builtin_shared.check_attached_node(p, n) then
240                         drop_attached_node(p)
241                         return true
242                 end
243         end
244
245         return false
246 end
247
248 -- This table is specifically ordered.
249 -- We don't walk diagonals, only our direct neighbors, and self.
250 -- Down first as likely case, but always before self. The same with sides.
251 -- Up must come last, so that things above self will also fall all at once.
252 local check_for_falling_neighbors = {
253         {x = -1, y = -1, z = 0},
254         {x = 1, y = -1, z = 0},
255         {x = 0, y = -1, z = -1},
256         {x = 0, y = -1, z = 1},
257         {x = 0, y = -1, z = 0},
258         {x = -1, y = 0, z = 0},
259         {x = 1, y = 0, z = 0},
260         {x = 0, y = 0, z = 1},
261         {x = 0, y = 0, z = -1},
262         {x = 0, y = 0, z = 0},
263         {x = 0, y = 1, z = 0},
264 }
265
266 function core.check_for_falling(p)
267         -- Round p to prevent falling entities to get stuck.
268         p = vector.round(p)
269
270         -- We make a stack, and manually maintain size for performance.
271         -- Stored in the stack, we will maintain tables with pos, and
272         -- last neighbor visited. This way, when we get back to each
273         -- node, we know which directions we have already walked, and
274         -- which direction is the next to walk.
275         local s = {}
276         local n = 0
277         -- The neighbor order we will visit from our table.
278         local v = 1
279
280         while true do
281                 -- Push current pos onto the stack.
282                 n = n + 1
283                 s[n] = {p = p, v = v}
284                 -- Select next node from neighbor list.
285                 p = vector.add(p, check_for_falling_neighbors[v])
286                 -- Now we check out the node. If it is in need of an update,
287                 -- it will let us know in the return value (true = updated).
288                 if not core.check_single_for_falling(p) then
289                         -- If we don't need to "recurse" (walk) to it then pop
290                         -- our previous pos off the stack and continue from there,
291                         -- with the v value we were at when we last were at that
292                         -- node
293                         repeat
294                                 local pop = s[n]
295                                 p = pop.p
296                                 v = pop.v
297                                 s[n] = nil
298                                 n = n - 1
299                                 -- If there's nothing left on the stack, and no
300                                 -- more sides to walk to, we're done and can exit
301                                 if n == 0 and v == 11 then
302                                         return
303                                 end
304                         until v < 11
305                         -- The next round walk the next neighbor in list.
306                         v = v + 1
307                 else
308                         -- If we did need to walk the neighbor, then
309                         -- start walking it from the walk order start (1),
310                         -- and not the order we just pushed up the stack.
311                         v = 1
312                 end
313         end
314 end
315
316 --
317 -- Global callbacks
318 --
319
320 local function on_placenode(p, node)
321         core.check_for_falling(p)
322 end
323 core.register_on_placenode(on_placenode)
324
325 local function on_dignode(p, node)
326         core.check_for_falling(p)
327 end
328 core.register_on_dignode(on_dignode)
329
330 local function on_punchnode(p, node)
331         core.check_for_falling(p)
332 end
333 core.register_on_punchnode(on_punchnode)