1 files changed, 40 insertions, 0 deletions
diff --git a/register_flow_logic.lua b/register_flow_logic.lua
new file mode 100644
index 0000000..e027dae
--- /dev/null
+++ b/register_flow_logic.lua
@@ -0,0 +1,40 @@
+-- register new flow logic ABMs
+
+local pipes_full_nodenames = pipeworks.pipes_full_nodenames
+local pipes_empty_nodenames = pipeworks.pipes_empty_nodenames
+
+-- run pressure balancing ABM over all water-moving nodes
+-- FIXME: DRY principle, get this from elsewhere in the code
+local pump_on = "pipeworks:pump_on"
+local pump_off = "pipeworks:pump_off"
+
+local pipes_all_nodenames = pipes_full_nodenames
+for _, pipe in ipairs(pipes_empty_nodenames) do
+	table.insert(pipes_all_nodenames, pipe)
+end
+table.insert(pipes_all_nodenames, pump_off)
+table.insert(pipes_all_nodenames, pump_on)
+
+
+if pipeworks.enable_pipes then
+	minetest.register_abm({
+		nodenames = pipes_all_nodenames,
+		interval = 1,
+		chance = 1,
+		action = function(pos, node, active_object_count, active_object_count_wider)
+			pipeworks.balance_pressure(pos, node)
+		end
+	})
+end
+
+if pipeworks.enable_pipe_devices then
+	-- absorb water into pumps if it'll fit
+	minetest.register_abm({
+		nodenames = { pump_on },
+		interval = 1,
+		chance = 1,
+		action = function(pos, node, active_object_count, active_object_count_wider)
+			pipeworks.run_pump_intake(pos, node)
+		end
+	})
+end