From fdf679f405b1190275ef7ac2239fd1052607ea3d Mon Sep 17 00:00:00 2001
From: numzero <numzer0@yandex.ru>
Date: Mon, 10 Aug 2020 20:36:28 +0300
Subject: [PATCH] Limit piston speed

---
 mesecons_pistons/init.lua | 39 ++++++++++++++++++++++++++++++++++-----
 settingtypes.txt          |  5 +++++
 2 files changed, 39 insertions(+), 5 deletions(-)

diff --git a/mesecons_pistons/init.lua b/mesecons_pistons/init.lua
index 66c74b2..b3c63dc 100644
--- a/mesecons_pistons/init.lua
+++ b/mesecons_pistons/init.lua
@@ -78,8 +78,11 @@ local function piston_after_dig(pos, node)
 	piston_remove_pusher(pos, node, true)
 end
 
-local piston_on = function(pos, node)
+local function piston_on(pos, node)
 	local pistonspec = get_pistonspec(node.name, "offname")
+	if not pistonspec then -- it may be called asynchronously now, don’t crash if something goes wrong
+		return
+	end
 	local dir = vector.multiply(minetest.facedir_to_dir(node.param2), -1)
 	local pusher_pos = vector.add(pos, dir)
 	local meta = minetest.get_meta(pos)
@@ -99,6 +102,9 @@ end
 
 local function piston_off(pos, node)
 	local pistonspec = get_pistonspec(node.name, "onname")
+	if not pistonspec then
+		return
+	end
 	minetest.swap_node(pos, {param2 = node.param2, name = pistonspec.offname})
 	piston_remove_pusher(pos, node, not pistonspec.sticky) -- allow that even in protected area
 
@@ -114,6 +120,29 @@ local function piston_off(pos, node)
 	end
 end
 
+-- not on/off as power state may change faster than the piston state
+mesecon.queue:add_function("piston_switch", function(pos)
+	local node = mesecon.get_node_force(pos)
+	if mesecon.is_powered(pos) then
+		piston_on(pos, node)
+	else
+		piston_off(pos, node)
+	end
+end)
+
+local piston_on_delayed, piston_off_delayed
+local delay = mesecon.setting("piston_delay", 0.15)
+if delay >= 0 then
+	local function piston_switch_delayed(pos, node)
+		mesecon.queue:add_action(pos, "piston_switch", {}, delay, "piston_switch")
+	end
+	piston_on_delayed = piston_switch_delayed
+	piston_off_delayed = piston_switch_delayed
+else
+	piston_on_delayed = piston_on
+	piston_off_delayed = piston_off
+end
+
 local orientations = {
 	[0] = { 4,  8},
 	      {13, 17},
@@ -281,7 +310,7 @@ minetest.register_node("mesecons_pistons:piston_normal_off", {
 	after_place_node = piston_orientate,
 	sounds = default.node_sound_wood_defaults(),
 	mesecons = {effector={
-		action_on = piston_on,
+		action_on = piston_on_delayed,
 		rules = piston_get_rules,
 	}},
 	on_punch = piston_punch,
@@ -311,7 +340,7 @@ minetest.register_node("mesecons_pistons:piston_normal_on", {
 	selection_box = piston_on_box,
 	sounds = default.node_sound_wood_defaults(),
 	mesecons = {effector={
-		action_off = piston_off,
+		action_off = piston_off_delayed,
 		rules = piston_get_rules,
 	}},
 	on_rotate = piston_rotate_on,
@@ -360,7 +389,7 @@ minetest.register_node("mesecons_pistons:piston_sticky_off", {
 	after_place_node = piston_orientate,
 	sounds = default.node_sound_wood_defaults(),
 	mesecons = {effector={
-		action_on = piston_on,
+		action_on = piston_on_delayed,
 		rules = piston_get_rules,
 	}},
 	on_punch = piston_punch,
@@ -390,7 +419,7 @@ minetest.register_node("mesecons_pistons:piston_sticky_on", {
 	selection_box = piston_on_box,
 	sounds = default.node_sound_wood_defaults(),
 	mesecons = {effector={
-		action_off = piston_off,
+		action_off = piston_off_delayed,
 		rules = piston_get_rules,
 	}},
 	on_rotate = piston_rotate_on,
diff --git a/settingtypes.txt b/settingtypes.txt
index 5627440..640bebc 100644
--- a/settingtypes.txt
+++ b/settingtypes.txt
@@ -52,6 +52,11 @@ mesecon.movestone_max_pull (Max pull) int 50 1 100
 mesecon.piston_max_push (Max push) int 15 1 100
 mesecon.piston_max_pull (Max pull) int 15 1 100
 
+# Delay between piston power on/off and actual push/pull.
+# Zero means minimal possible delay. Negative value means
+# synchronous action (old behavior).
+mesecon.piston_delay (Delay) float 0.15 -1 3.0
+
 
 [mesecons_pressureplates]