From 8ab28351419ab431fcfaa18bdda70956deb23e24 Mon Sep 17 00:00:00 2001
From: Joachim Stolberg <joe.stolberg@gmx.de>
Date: Sun, 21 Jan 2024 12:36:44 +0100
Subject: [PATCH] Add reverse mode for ta5 pump

---
 fusion_reactor/ta5_pump.lua | 49 ++++++++++++++++++++++++++++---------
 1 file changed, 38 insertions(+), 11 deletions(-)

diff --git a/fusion_reactor/ta5_pump.lua b/fusion_reactor/ta5_pump.lua
index f054f5c..9a8a299 100644
--- a/fusion_reactor/ta5_pump.lua
+++ b/fusion_reactor/ta5_pump.lua
@@ -31,6 +31,14 @@ local WRENCH_MENU =	{{
 	name = "flowrate",
 	label = S("Total flow rate"),
 	tooltip = S("Total flow rate in liquid units"),
+},{
+	type = "dropdown",
+	choices = "normal,reverse",
+	name = "operation",
+	label = S("Operation"),
+	tooltip = S("Pump direction"),
+	values = {0, 1},
+	default = "1",
 }}
 
 local State = techage.NodeStates:new({
@@ -43,20 +51,39 @@ local State = techage.NodeStates:new({
 
 local function pumping(pos, nvm)
 	local outdir = M(pos):get_int("outdir")
-	local taken, name = liquid.take(pos, Pipe2, Flip[outdir], nil, CAPA)
-	if taken > 0 then
-		local leftover = liquid.put(pos, Pipe3, outdir, name, taken)
-		if leftover and leftover > 0 then
-			liquid.untake(pos, Pipe2, Flip[outdir], name, leftover)
-			if leftover == taken then
-				State:blocked(pos, nvm)
-				return 0
+	local reverse = M(pos):get_int("operation")
+	if reverse == 1 then
+		local taken, name = liquid.take(pos, Pipe3, outdir, nil, CAPA)
+		if taken > 0 then
+			local leftover = liquid.put(pos, Pipe2, Flip[outdir], name, taken)
+			if leftover and leftover > 0 then
+				liquid.untake(pos, Pipe3, outdir, name, leftover)
+				if leftover == taken then
+					State:blocked(pos, nvm)
+					return 0
+				end
+				State:keep_running(pos, nvm, COUNTDOWN_TICKS)
+				return taken - leftover
 			end
 			State:keep_running(pos, nvm, COUNTDOWN_TICKS)
-			return taken - leftover
+			return taken
+		end
+	else
+		local taken, name = liquid.take(pos, Pipe2, Flip[outdir], nil, CAPA)
+		if taken > 0 then
+			local leftover = liquid.put(pos, Pipe3, outdir, name, taken)
+			if leftover and leftover > 0 then
+				liquid.untake(pos, Pipe2, Flip[outdir], name, leftover)
+				if leftover == taken then
+					State:blocked(pos, nvm)
+					return 0
+				end
+				State:keep_running(pos, nvm, COUNTDOWN_TICKS)
+				return taken - leftover
+			end
+			State:keep_running(pos, nvm, COUNTDOWN_TICKS)
+			return taken
 		end
-		State:keep_running(pos, nvm, COUNTDOWN_TICKS)
-		return taken
 	end
 	State:idle(pos, nvm)
 	return 0