From b81d1d3f2debcab4d18cf4dee824075ef0bbb116 Mon Sep 17 00:00:00 2001
From: Lejo <Lejo_1@web.de>
Date: Sat, 13 Apr 2019 20:45:49 +0200
Subject: [PATCH] Check protection of the forcefield and Quarry (#468)

---
 technic/machines/switching_station.lua |  805 ++++++++++++++++++++++++++++++++++-----------------------
 1 files changed, 482 insertions(+), 323 deletions(-)

diff --git a/technic/machines/switching_station.lua b/technic/machines/switching_station.lua
index 36133fa..d645847 100644
--- a/technic/machines/switching_station.lua
+++ b/technic/machines/switching_station.lua
@@ -1,352 +1,511 @@
--- SWITCHING STATION
--- The switching station is the center of all power distribution on an electric network.
--- The station will collect all produced power from producers (PR) and batteries (BA)
--- and distribute it to receivers (RE) and depleted batteries (BA).
---
--- It works like this:
---  All PR,BA,RE nodes are indexed and tagged with the switching station.
--- The tagging is to allow more stations to be built without allowing a cheat
--- with duplicating power.
---  All the RE nodes are queried for their current EU demand. Those which are off
--- would require no or a small standby EU demand, while those which are on would
--- require more.
--- If the total demand is less than the available power they are all updated with the
--- demand number.
--- If any surplus exists from the PR nodes the batteries will be charged evenly with this.
--- If the total demand requires draw on the batteries they will be discharged evenly.
---
--- If the total demand is more than the available power all RE nodes will be shut down.
--- We have a brown-out situation.
---
--- Hence all the power distribution logic resides in this single node.
---
---  Nodes connected to the network will have one or more of these parameters as meta data:
---   <LV|MV|HV>_EU_supply : Exists for PR and BA node types. This is the EU value supplied by the node. Output
---   <LV|MV|HV>_EU_demand : Exists for RE and BA node types. This is the EU value the node requires to run. Output
---   <LV|MV|HV>_EU_input  : Exists for RE and BA node types. This is the actual EU value the network can give the node. Input
---
---  The reason the LV|MV|HV type is prepended toe meta data is because some machine could require several supplies to work.
---  This way the supplies are separated per network.
-technic.DBG = 1
-local dprint = technic.dprint
+-- See also technic/doc/api.md
 
-minetest.register_craft(
-   {
-      output = 'technic:switching_station 1',
-      recipe = {
-	 {'technic:lv_transformer', 'technic:mv_transformer', 'technic:hv_transformer'},
-	 {'technic:lv_transformer', 'technic:mv_transformer', 'technic:hv_transformer'},
-	 {'technic:lv_cable',       'technic:mv_cable',       'technic:hv_cable'},
-      }
-   })
+technic.networks = {}
+technic.cables = {}
+technic.redundant_warn = {}
 
-minetest.register_node(
-   "technic:switching_station",
-   {description = "Switching Station",
-    tiles  = {"technic_water_mill_top_active.png", "technic_water_mill_top_active.png", "technic_water_mill_top_active.png",
-	      "technic_water_mill_top_active.png", "technic_water_mill_top_active.png", "technic_water_mill_top_active.png"},
-    groups = {snappy=2,choppy=2,oddly_breakable_by_hand=2},
-    sounds = default.node_sound_wood_defaults(),
-    drawtype = "nodebox",
-    paramtype = "light",
-    is_ground_content = true,
-    node_box = {
-	 type = "fixed",
-       fixed = {-0.5, -0.5, -0.5, 0.5, 0.5, 0.5},
-    },
-    selection_box = {
-       type = "fixed",
-       fixed = {-0.5, -0.5, -0.5, 0.5, 0.5, 0.5},
-    },
-    on_construct = function(pos)
-		      local meta = minetest.env:get_meta(pos)
-		      meta:set_string("infotext", "Switching Station")
---			minetest.chat_send_player(puncher:get_player_name(), "Switching station constructed. Punch the station to shut down the network.");
---			meta:set_int("active", 1)
-		   end,
---      on_punch = function(pos, node, puncher)
---		    local meta   = minetest.env:get_meta(pos)
---		    local active = meta:get_int("active")
---		    if active == 1 then
---		       meta:set_int("active", 0)
---		       minetest.chat_send_player(puncher:get_player_name(), "Electrical network shut down. Punch again to turn it on.");
---		    else
---		       meta:set_int("active", 1)
---		       minetest.chat_send_player(puncher:get_player_name(), "Electrical network turned on. Punch again to shut it down.");
---		    end
---		 end
- })
+local mesecons_path = minetest.get_modpath("mesecons")
+local digilines_path = minetest.get_modpath("digilines")
 
---------------------------------------------------
--- Functions to help the machines on the electrical network
---------------------------------------------------
--- This one provides a timeout for a node in case it was disconnected from the network
--- A node must be touched by the station continuously in order to function
-technic.switching_station_timeout_count = function(pos, machine_tier)
-					     local meta = minetest.env:get_meta(pos)
-					     timeout =  meta:get_int(machine_tier.."_EU_timeout")
-					     --print("Counting timeout "..timeout)
-					     if timeout == 0 then
-						--print("OFF")
-						meta:set_int(machine_tier.."_EU_input", 0)
-					     else
-						--print("ON")
-						meta:set_int(machine_tier.."_EU_timeout", timeout-1)
-					     end
-					  end
+local S = technic.getter
+
+local cable_entry = "^technic_cable_connection_overlay.png"
+
+minetest.register_craft({
+	output = "technic:switching_station",
+	recipe = {
+		{"",                     "technic:lv_transformer", ""},
+		{"default:copper_ingot", "technic:machine_casing", "default:copper_ingot"},
+		{"technic:lv_cable",     "technic:lv_cable",       "technic:lv_cable"}
+	}
+})
+
+local mesecon_def
+if mesecons_path then
+	mesecon_def = {effector = {
+		rules = mesecon.rules.default,
+	}}
+end
+
+minetest.register_node("technic:switching_station",{
+	description = S("Switching Station"),
+	tiles  = {
+		"technic_water_mill_top_active.png",
+		"technic_water_mill_top_active.png"..cable_entry,
+		"technic_water_mill_top_active.png",
+		"technic_water_mill_top_active.png",
+		"technic_water_mill_top_active.png",
+		"technic_water_mill_top_active.png"},
+	groups = {snappy=2, choppy=2, oddly_breakable_by_hand=2, technic_all_tiers=1},
+	connect_sides = {"bottom"},
+	sounds = default.node_sound_wood_defaults(),
+	on_construct = function(pos)
+		local meta = minetest.get_meta(pos)
+		meta:set_string("infotext", S("Switching Station"))
+		meta:set_string("active", 1)
+		meta:set_string("channel", "switching_station"..minetest.pos_to_string(pos))
+		meta:set_string("formspec", "field[channel;Channel;${channel}]")
+		local poshash = minetest.hash_node_position(pos)
+		technic.redundant_warn.poshash = nil
+	end,
+	after_dig_node = function(pos)
+		minetest.forceload_free_block(pos)
+		pos.y = pos.y - 1
+		minetest.forceload_free_block(pos)
+		local poshash = minetest.hash_node_position(pos)
+		technic.redundant_warn.poshash = nil
+	end,
+	on_receive_fields = function(pos, formname, fields, sender)
+		if not fields.channel then
+			return
+		end
+		local plname = sender:get_player_name()
+		if minetest.is_protected(pos, plname) then
+			minetest.record_protection_violation(pos, plname)
+			return
+		end
+		local meta = minetest.get_meta(pos)
+		meta:set_string("channel", fields.channel)
+	end,
+	mesecons = mesecon_def,
+	digiline = {
+		receptor = {action = function() end},
+		effector = {
+			action = function(pos, node, channel, msg)
+				if msg ~= "GET" and msg ~= "get" then
+					return
+				end
+				local meta = minetest.get_meta(pos)
+				if channel ~= meta:get_string("channel") then
+					return
+				end
+				digilines.receptor_send(pos, digilines.rules.default, channel, {
+					supply = meta:get_int("supply"),
+					demand = meta:get_int("demand")
+				})
+			end
+		},
+	},
+})
 
 --------------------------------------------------
 -- Functions to traverse the electrical network
 --------------------------------------------------
+local function flatten(map)
+	local list = {}
+	for key, value in pairs(map) do
+		list[#list + 1] = value
+	end
+	return list
+end
 
 -- Add a wire node to the LV/MV/HV network
-local add_new_cable_node = function(nodes,pos)
-			      local i = 1
-			      repeat
-				 if nodes[i]==nil then break end
-				 if pos.x==nodes[i].x and pos.y==nodes[i].y and pos.z==nodes[i].z then return false end
-				 i=i+1
-			      until false
-			      nodes[i] = {x=pos.x, y=pos.y, z=pos.z, visited=1} -- copy position
-			      return true
-			   end
+local function add_network_node(nodes, pos, network_id)
+	local node_id = minetest.hash_node_position(pos)
+	technic.cables[node_id] = network_id
+	if nodes[node_id] then
+		return false
+	end
+	nodes[node_id] = pos
+	return true
+end
+
+local function add_cable_node(nodes, pos, network_id, queue)
+	if add_network_node(nodes, pos, network_id) then
+		queue[#queue + 1] = pos
+	end
+end
 
 -- Generic function to add found connected nodes to the right classification array
-local check_node_subp = function(PR_nodes,RE_nodes,BA_nodes,all_nodes,pos,machines,cablename)
-			   local meta = minetest.env:get_meta(pos)
-			   local name = minetest.env:get_node(pos).name
-			   if meta:get_float(cablename)==1 then
-			      add_new_cable_node(all_nodes,pos)
-			   elseif machines[name] then
-			      --dprint(name.." is a "..machines[name])
-			      if     machines[name] == "PR" then
-				 add_new_cable_node(PR_nodes,pos)
-			      elseif machines[name] == "RE" then
-				 add_new_cable_node(RE_nodes,pos)
-			      elseif machines[name] == "BA" then
-				 add_new_cable_node(BA_nodes,pos)
-			      end
-			      if cablename == "cablelike" then
-				 meta:set_int("LV_EU_timeout", 2) -- Touch node
-			      elseif cablename == "mv_cablelike" then
-				 meta:set_int("MV_EU_timeout", 2) -- Touch node
-			      elseif cablename == "hv_cablelike" then
-				 meta:set_int("HV_EU_timeout", 2) -- Touch node
-			      end
-			   end
-			end
+local check_node_subp = function(PR_nodes, RE_nodes, BA_nodes, SP_nodes, all_nodes, pos, machines, tier, sw_pos, from_below, network_id, queue)
+	technic.get_or_load_node(pos)
+	local meta = minetest.get_meta(pos)
+	local name = minetest.get_node(pos).name
+
+	if technic.is_tier_cable(name, tier) then
+		add_cable_node(all_nodes, pos,network_id, queue)
+	elseif machines[name] then
+		--dprint(name.." is a "..machines[name])
+		meta:set_string(tier.."_network",minetest.pos_to_string(sw_pos))
+		if     machines[name] == technic.producer then
+			add_network_node(PR_nodes, pos, network_id)
+		elseif machines[name] == technic.receiver then
+			add_network_node(RE_nodes, pos, network_id)
+		elseif machines[name] == technic.producer_receiver then
+			add_network_node(PR_nodes, pos, network_id)
+			add_network_node(RE_nodes, pos, network_id)
+		elseif machines[name] == "SPECIAL" and
+				(pos.x ~= sw_pos.x or pos.y ~= sw_pos.y or pos.z ~= sw_pos.z) and
+				from_below then
+			-- Another switching station -> disable it
+			add_network_node(SP_nodes, pos, network_id)
+			meta:set_int("active", 0)
+		elseif machines[name] == technic.battery then
+			add_network_node(BA_nodes, pos, network_id)
+		end
+
+		meta:set_int(tier.."_EU_timeout", 2) -- Touch node
+	end
+end
 
 -- Traverse a network given a list of machines and a cable type name
-local traverse_network = function(PR_nodes,RE_nodes,BA_nodes,all_nodes, i, machines, cablename)
-			    local pos = {x=all_nodes[i].x, y=all_nodes[i].y, z=all_nodes[i].z} -- copy position
-			    pos.x=pos.x+1
-			    check_node_subp(PR_nodes,RE_nodes,BA_nodes,all_nodes,pos, machines, cablename)
-			    pos.x=pos.x-2
-			    check_node_subp(PR_nodes,RE_nodes,BA_nodes,all_nodes,pos, machines, cablename)
-			    pos.x=pos.x+1
-			    
-			    pos.y=pos.y+1
-			    check_node_subp(PR_nodes,RE_nodes,BA_nodes,all_nodes,pos, machines, cablename)
-			    pos.y=pos.y-2
-			    check_node_subp(PR_nodes,RE_nodes,BA_nodes,all_nodes,pos, machines, cablename)
-			    pos.y=pos.y+1
-			    
-			    pos.z=pos.z+1
-			    check_node_subp(PR_nodes,RE_nodes,BA_nodes,all_nodes,pos, machines, cablename)
-			    pos.z=pos.z-2
-			    check_node_subp(PR_nodes,RE_nodes,BA_nodes,all_nodes,pos, machines, cablename)
-			    pos.z=pos.z+1
-		      end
+local traverse_network = function(PR_nodes, RE_nodes, BA_nodes, SP_nodes, all_nodes, pos, machines, tier, sw_pos, network_id, queue)
+	local positions = {
+		{x=pos.x+1, y=pos.y,   z=pos.z},
+		{x=pos.x-1, y=pos.y,   z=pos.z},
+		{x=pos.x,   y=pos.y+1, z=pos.z},
+		{x=pos.x,   y=pos.y-1, z=pos.z},
+		{x=pos.x,   y=pos.y,   z=pos.z+1},
+		{x=pos.x,   y=pos.y,   z=pos.z-1}}
+	for i, cur_pos in pairs(positions) do
+		check_node_subp(PR_nodes, RE_nodes, BA_nodes, SP_nodes, all_nodes, cur_pos, machines, tier, sw_pos, i == 3, network_id, queue)
+	end
+end
 
-----------------------------------------------
--- The action code for the switching station
-----------------------------------------------
-minetest.register_abm(
-	{nodenames = {"technic:switching_station"},
+local touch_nodes = function(list, tier)
+	for _, pos in ipairs(list) do
+		local meta = minetest.get_meta(pos)
+		meta:set_int(tier.."_EU_timeout", 2) -- Touch node
+	end
+end
+
+local get_network = function(sw_pos, pos1, tier)
+	local network_id = minetest.hash_node_position(pos1)
+	local cached = technic.networks[network_id]
+	if cached and cached.tier == tier then
+		touch_nodes(cached.PR_nodes, tier)
+		touch_nodes(cached.BA_nodes, tier)
+		touch_nodes(cached.RE_nodes, tier)
+		for _, pos in ipairs(cached.SP_nodes) do
+			local meta = minetest.get_meta(pos)
+			meta:set_int("active", 0)
+			meta:set_string("active_pos", minetest.serialize(sw_pos))
+		end
+		return cached.PR_nodes, cached.BA_nodes, cached.RE_nodes
+	end
+	local PR_nodes = {}
+	local BA_nodes = {}
+	local RE_nodes = {}
+	local SP_nodes = {}
+	local all_nodes = {}
+	local queue = {}
+	add_cable_node(all_nodes, pos1, network_id, queue)
+	while next(queue) do
+		local to_visit = {}
+		for _, pos in ipairs(queue) do
+			traverse_network(PR_nodes, RE_nodes, BA_nodes, SP_nodes, all_nodes,
+					pos, technic.machines[tier], tier, sw_pos, network_id, to_visit)
+		end
+		queue = to_visit
+	end
+	PR_nodes = flatten(PR_nodes)
+	BA_nodes = flatten(BA_nodes)
+	RE_nodes = flatten(RE_nodes)
+	SP_nodes = flatten(SP_nodes)
+	all_nodes = flatten(all_nodes)
+	technic.networks[network_id] = {tier = tier, all_nodes = all_nodes, SP_nodes = SP_nodes,
+			PR_nodes = PR_nodes, RE_nodes = RE_nodes, BA_nodes = BA_nodes}
+	return PR_nodes, BA_nodes, RE_nodes
+end
+
+-----------------------------------------------
+-- The action code for the switching station --
+-----------------------------------------------
+
+technic.powerctrl_state = true
+
+minetest.register_chatcommand("powerctrl", {
+	params = "state",
+	description = "Enables or disables technic's switching station ABM",
+	privs = { basic_privs = true },
+	func = function(name, state)
+		if state == "on" then
+			technic.powerctrl_state = true
+		else
+			technic.powerctrl_state = false
+		end
+	end
+})
+
+minetest.register_abm({
+	nodenames = {"technic:switching_station"},
+	label = "Switching Station", -- allows the mtt profiler to profile this abm individually
 	interval   = 1,
 	chance     = 1,
 	action = function(pos, node, active_object_count, active_object_count_wider)
-		    local meta             = minetest.env:get_meta(pos)
-		    local meta1            = nil
-		    local pos1             = {}
-		    local PR_EU            = 0 -- EUs from PR nodes
-		    local BA_PR_EU         = 0 -- EUs from BA nodes (discharching)
-		    local BA_RE_EU         = 0 -- EUs to BA nodes (charging)
-		    local RE_EU            = 0 -- EUs to RE nodes
+		if not technic.powerctrl_state then return end
+		local meta             = minetest.get_meta(pos)
+		local meta1            = nil
+		local pos1             = {}
+		local PR_EU            = 0 -- EUs from PR nodes
+		local BA_PR_EU         = 0 -- EUs from BA nodes (discharching)
+		local BA_RE_EU         = 0 -- EUs to BA nodes (charging)
+		local RE_EU            = 0 -- EUs to RE nodes
 
-		    local network   = ""
-		    local all_nodes = {}
-		    local PR_nodes  = {}
-		    local BA_nodes  = {} 
-		    local RE_nodes  = {}
+		local tier      = ""
+		local PR_nodes
+		local BA_nodes
+		local RE_nodes
+		local machine_name = S("Switching Station")
 
---		    -- Possible to turn off the entire network
---		    if meta:get_int("active") == 0 then
---		       for _,pos1 in pairs(RE_nodes) do
---			  meta1  = minetest.env:get_meta(pos1)
---			  meta1:set_int("EU_input", 0)
---		       end
---		       for _,pos1 in pairs(BA_nodes) do
---			  meta1  = minetest.env:get_meta(pos1)
---			  meta1:set_int("EU_input", 0)
---		       end
---		       return
---		    end
+		-- Which kind of network are we on:
+		pos1 = {x=pos.x, y=pos.y-1, z=pos.z}
 
-		    -- Which kind of network are we on:
-		    pos1 = {x=pos.x, y=pos.y-1, z=pos.z}
-		    all_nodes[1] = pos1
+		--Disable if necessary
+		if meta:get_int("active") ~= 1 then
+			minetest.forceload_free_block(pos)
+			minetest.forceload_free_block(pos1)
+			meta:set_string("infotext",S("%s Already Present"):format(machine_name))
 
-		    meta1  = minetest.env:get_meta(pos1)
-		    if meta1:get_float("cablelike") ==1 then
-		       -- LV type
-		       --dprint("LV type")
-		       network = "LV"
-		       local table_index = 1
-		       repeat
-			  traverse_network(PR_nodes,RE_nodes,BA_nodes,all_nodes,table_index, technic.LV_machines, "cablelike")
-			  table_index = table_index + 1
-			  if all_nodes[table_index] == nil then break end
-		       until false
-		    elseif meta1:get_float("mv_cablelike") ==1 then
-		       -- MV type
-		       --dprint("MV type")
-		       network = "MV"
-		       local table_index = 1
-		       repeat
-			  traverse_network(PR_nodes,RE_nodes,BA_nodes,all_nodes,table_index, technic.MV_machines, "mv_cablelike")
-			  table_index = table_index + 1
-			  if all_nodes[table_index] == nil then break end
-		       until false
-		    elseif meta1:get_float("hv_cablelike") ==1 then
-		       -- HV type
-		       --dprint("HV type")
-		       network = "HV"
-		       local table_index = 1
-		       repeat
-			  traverse_network(PR_nodes,RE_nodes,BA_nodes,all_nodes,table_index, technic.HV_machines, "hv_cablelike")
-			  table_index = table_index + 1
-			  if all_nodes[table_index] == nil then break end
-		       until false
-		    else
-		       -- No type :-)
-		       --dprint("Not connected to a network")
-		       meta:set_string("infotext", "Switching Station - no network")
-		       return
-		    end
-		    --dprint("nodes="..table.getn(all_nodes).." PR="..table.getn(PR_nodes).." BA="..table.getn(BA_nodes).." RE="..table.getn(RE_nodes))
+			local poshash = minetest.hash_node_position(pos)
 
-		    -- Strings for the meta data
-		    local eu_demand_str    = network.."_EU_demand"
-		    local eu_input_str     = network.."_EU_input"
-		    local eu_supply_str    = network.."_EU_supply"
-		    local eu_from_fuel_str = network.."_EU_from_fuel"
+			if not technic.redundant_warn[poshash] then
+				technic.redundant_warn[poshash] = true
+				print("[TECHNIC] Warning: redundant switching station found near "..minetest.pos_to_string(pos))
+			end
+			return
+		end
 
-		    -- Get all the power from the PR nodes
-		    local PR_eu_supply = 0 -- Total power
-		    for _,pos1 in pairs(PR_nodes) do
-		       meta1  = minetest.env:get_meta(pos1)
-		       PR_eu_supply = PR_eu_supply + meta1:get_int(eu_supply_str)
-		    end
-		    --dprint("Total PR supply:"..PR_eu_supply)
+		local name = minetest.get_node(pos1).name
+		local tier = technic.get_cable_tier(name)
+		if tier then
+			-- Forceload switching station
+			minetest.forceload_block(pos)
+			minetest.forceload_block(pos1)
+			PR_nodes, BA_nodes, RE_nodes = get_network(pos, pos1, tier)
+		else
+			--dprint("Not connected to a network")
+			meta:set_string("infotext", S("%s Has No Network"):format(machine_name))
+			minetest.forceload_free_block(pos)
+			minetest.forceload_free_block(pos1)
+			return
+		end
 
-		    -- Get all the demand from the RE nodes
-		    local RE_eu_demand = 0
-		    for _,pos1 in pairs(RE_nodes) do
-		       meta1  = minetest.env:get_meta(pos1)
-		       RE_eu_demand = RE_eu_demand + meta1:get_int(eu_demand_str)
-		    end
-		    --dprint("Total RE demand:"..RE_eu_demand)
-
-		    -- Get all the power from the BA nodes
-		    local BA_eu_supply = 0
-		    for _,pos1 in pairs(BA_nodes) do
-		       meta1  = minetest.env:get_meta(pos1)
-		       BA_eu_supply = BA_eu_supply + meta1:get_int(eu_supply_str)
-		    end
-		    --dprint("Total BA supply:"..BA_eu_supply)
-
-		    -- Get all the demand from the BA nodes
-		    local BA_eu_demand = 0
-		    for _,pos1 in pairs(BA_nodes) do
-		       meta1  = minetest.env:get_meta(pos1)
-		       BA_eu_demand = BA_eu_demand + meta1:get_int(eu_demand_str)
-		    end
-		    --dprint("Total BA demand:"..BA_eu_demand)
-
-		    meta:set_string("infotext", "Switching Station. PR("..(PR_eu_supply+BA_eu_supply)..") RE("..(RE_eu_demand+BA_eu_demand)..")")
-
-		    -- If the PR supply is enough for the RE demand supply them all
-		    if PR_eu_supply >= RE_eu_demand then
-		       --dprint("PR_eu_supply"..PR_eu_supply.." >= RE_eu_demand"..RE_eu_demand)
-		       for _,pos1 in pairs(RE_nodes) do
-			  meta1  = minetest.env:get_meta(pos1)
-			  local eu_demand = meta1:get_int(eu_demand_str)
-			  meta1:set_int(eu_input_str, eu_demand)
-		       end
-		       -- We have a surplus, so distribute the rest equally to the BA nodes
-		       -- Let's calculate the factor of the demand
-		       PR_eu_supply = PR_eu_supply - RE_eu_demand
-		       local charge_factor = 0 -- Assume all batteries fully charged
-		       if BA_eu_demand > 0 then
-			  charge_factor = PR_eu_supply / BA_eu_demand
-		       end
-		       for n,pos1 in pairs(BA_nodes) do
-			  meta1  = minetest.env:get_meta(pos1)
-			  local eu_demand = meta1:get_int(eu_demand_str)
-			  meta1:set_int(eu_input_str, math.floor(eu_demand*charge_factor))
-			  --dprint("Charging battery:"..math.floor(eu_demand*charge_factor))
-		       end
-		       -- If still a surplus we can start giving back to the fuel burning generators
-		       -- Only full EU packages are given back. The rest is wasted.
-		       if BA_eu_demand == 0 then
-			  for _,pos1 in pairs(PR_nodes) do
-			     meta1  = minetest.env:get_meta(pos1)
-			     if meta1:get_int(eu_from_fuel_str) == 1 then
-				local eu_supply = meta1:get_int(eu_supply_str)
-				if PR_eu_supply < eu_supply then
-				   break
-				else
-				   -- Set the supply to 0 if we did not require it.
-				   meta1:set_int(eu_supply_str, 0)
-				   PR_eu_supply = PR_eu_supply - eu_supply
+		-- Run all the nodes
+		local function run_nodes(list, run_stage)
+			for _, pos2 in ipairs(list) do
+				technic.get_or_load_node(pos2)
+				local node2 = minetest.get_node(pos2)
+				local nodedef
+				if node2 and node2.name then
+					nodedef = minetest.registered_nodes[node2.name]
 				end
-			     end
-			  end
-		       end
-		       return
-		    end
+				if nodedef and nodedef.technic_run then
+					nodedef.technic_run(pos2, node2, run_stage)
+				end
+			end
+		end
 
-		    -- If the PR supply is not enough for the RE demand we will discharge the batteries too
-		    if PR_eu_supply+BA_eu_supply >= RE_eu_demand then
-		       --dprint("PR_eu_supply "..PR_eu_supply.."+BA_eu_supply "..BA_eu_supply.." >= RE_eu_demand"..RE_eu_demand)
-		       for _,pos1 in pairs(RE_nodes) do
-			  meta1  = minetest.env:get_meta(pos1)
-			  local eu_demand = meta1:get_int(eu_demand_str)
-			  meta1:set_int(eu_input_str, eu_demand)
-		       end
-		       -- We have a deficit, so distribute to the BA nodes
-		       -- Let's calculate the factor of the supply
-		       local charge_factor = 0 -- Assume all batteries depleted
-		       if BA_eu_supply > 0 then
-			  charge_factor = (PR_eu_supply - RE_eu_demand) / BA_eu_supply
-		       end
-		       for n,pos1 in pairs(BA_nodes) do
-			  meta1  = minetest.env:get_meta(pos1)
-			  local eu_supply = meta1:get_int(eu_supply_str)
-			  meta1:set_int(eu_input_str, math.floor(eu_supply*charge_factor))
-			  --dprint("Discharging battery:"..math.floor(eu_supply*charge_factor))
-		       end
-		       return
-		    end
+		run_nodes(PR_nodes, technic.producer)
+		run_nodes(RE_nodes, technic.receiver)
+		run_nodes(BA_nodes, technic.battery)
 
-		    -- If the PR+BA supply is not enough for the RE demand: Shut everything down!
-		    -- Note: another behaviour could also be imagined: provide the average power for all and let the node decide what happens.
-		    -- This is much simpler though: Not enough power for all==no power for all
-		    --print("NO POWER")
-		    for _,pos1 in pairs(RE_nodes) do
-		       meta1  = minetest.env:get_meta(pos1)
-		       meta1:set_int(eu_input_str, 0)
-		    end
+		-- Strings for the meta data
+		local eu_demand_str    = tier.."_EU_demand"
+		local eu_input_str     = tier.."_EU_input"
+		local eu_supply_str    = tier.."_EU_supply"
+
+		-- Distribute charge equally across multiple batteries.
+		local charge_total = 0
+		local battery_count = 0
+
+		for n, pos1 in pairs(BA_nodes) do
+			meta1 = minetest.get_meta(pos1)
+			local charge = meta1:get_int("internal_EU_charge")
+
+			if (meta1:get_int(eu_demand_str) ~= 0) then
+				charge_total = charge_total + charge
+				battery_count = battery_count + 1
+			end
+		end
+
+		local charge_distributed = math.floor(charge_total / battery_count)
+
+		for n, pos1 in pairs(BA_nodes) do
+			meta1 = minetest.get_meta(pos1)
+
+			if (meta1:get_int(eu_demand_str) ~= 0) then
+				meta1:set_int("internal_EU_charge", charge_distributed)
+			end
+		end
+
+		-- Get all the power from the PR nodes
+		local PR_eu_supply = 0 -- Total power
+		for _, pos1 in pairs(PR_nodes) do
+			meta1 = minetest.get_meta(pos1)
+			PR_eu_supply = PR_eu_supply + meta1:get_int(eu_supply_str)
+		end
+		--dprint("Total PR supply:"..PR_eu_supply)
+
+		-- Get all the demand from the RE nodes
+		local RE_eu_demand = 0
+		for _, pos1 in pairs(RE_nodes) do
+			meta1 = minetest.get_meta(pos1)
+			RE_eu_demand = RE_eu_demand + meta1:get_int(eu_demand_str)
+		end
+		--dprint("Total RE demand:"..RE_eu_demand)
+
+		-- Get all the power from the BA nodes
+		local BA_eu_supply = 0
+		for _, pos1 in pairs(BA_nodes) do
+			meta1 = minetest.get_meta(pos1)
+			BA_eu_supply = BA_eu_supply + meta1:get_int(eu_supply_str)
+		end
+		--dprint("Total BA supply:"..BA_eu_supply)
+
+		-- Get all the demand from the BA nodes
+		local BA_eu_demand = 0
+		for _, pos1 in pairs(BA_nodes) do
+			meta1 = minetest.get_meta(pos1)
+			BA_eu_demand = BA_eu_demand + meta1:get_int(eu_demand_str)
+		end
+		--dprint("Total BA demand:"..BA_eu_demand)
+
+		meta:set_string("infotext", S("@1. Supply: @2 Demand: @3",
+				machine_name, technic.EU_string(PR_eu_supply),
+				technic.EU_string(RE_eu_demand)))
+
+		-- If mesecon signal and power supply or demand changed then
+		-- send them via digilines.
+		if mesecons_path and digilines_path and mesecon.is_powered(pos) then
+			if PR_eu_supply ~= meta:get_int("supply") or
+					RE_eu_demand ~= meta:get_int("demand") then
+				local channel = meta:get_string("channel")
+				digilines.receptor_send(pos, digilines.rules.default, channel, {
+					supply = PR_eu_supply,
+					demand = RE_eu_demand
+				})
+			end
+		end
+
+		-- Data that will be used by the power monitor
+		meta:set_int("supply",PR_eu_supply)
+		meta:set_int("demand",RE_eu_demand)
+
+		-- If the PR supply is enough for the RE demand supply them all
+		if PR_eu_supply >= RE_eu_demand then
+		--dprint("PR_eu_supply"..PR_eu_supply.." >= RE_eu_demand"..RE_eu_demand)
+			for _, pos1 in pairs(RE_nodes) do
+				meta1 = minetest.get_meta(pos1)
+				local eu_demand = meta1:get_int(eu_demand_str)
+				meta1:set_int(eu_input_str, eu_demand)
+			end
+			-- We have a surplus, so distribute the rest equally to the BA nodes
+			-- Let's calculate the factor of the demand
+			PR_eu_supply = PR_eu_supply - RE_eu_demand
+			local charge_factor = 0 -- Assume all batteries fully charged
+			if BA_eu_demand > 0 then
+				charge_factor = PR_eu_supply / BA_eu_demand
+			end
+			for n, pos1 in pairs(BA_nodes) do
+				meta1 = minetest.get_meta(pos1)
+				local eu_demand = meta1:get_int(eu_demand_str)
+				meta1:set_int(eu_input_str, math.floor(eu_demand * charge_factor))
+				--dprint("Charging battery:"..math.floor(eu_demand*charge_factor))
+			end
+			return
+		end
+
+		-- If the PR supply is not enough for the RE demand we will discharge the batteries too
+		if PR_eu_supply + BA_eu_supply >= RE_eu_demand then
+			--dprint("PR_eu_supply "..PR_eu_supply.."+BA_eu_supply "..BA_eu_supply.." >= RE_eu_demand"..RE_eu_demand)
+			for _, pos1 in pairs(RE_nodes) do
+				meta1  = minetest.get_meta(pos1)
+				local eu_demand = meta1:get_int(eu_demand_str)
+				meta1:set_int(eu_input_str, eu_demand)
+			end
+			-- We have a deficit, so distribute to the BA nodes
+			-- Let's calculate the factor of the supply
+			local charge_factor = 0 -- Assume all batteries depleted
+			if BA_eu_supply > 0 then
+				charge_factor = (PR_eu_supply - RE_eu_demand) / BA_eu_supply
+			end
+			for n,pos1 in pairs(BA_nodes) do
+				meta1 = minetest.get_meta(pos1)
+				local eu_supply = meta1:get_int(eu_supply_str)
+				meta1:set_int(eu_input_str, math.floor(eu_supply * charge_factor))
+				--dprint("Discharging battery:"..math.floor(eu_supply*charge_factor))
+			end
+			return
+		end
+
+		-- If the PR+BA supply is not enough for the RE demand: Power only the batteries
+		local charge_factor = 0 -- Assume all batteries fully charged
+		if BA_eu_demand > 0 then
+			charge_factor = PR_eu_supply / BA_eu_demand
+		end
+		for n, pos1 in pairs(BA_nodes) do
+			meta1 = minetest.get_meta(pos1)
+			local eu_demand = meta1:get_int(eu_demand_str)
+			meta1:set_int(eu_input_str, math.floor(eu_demand * charge_factor))
+		end
+		for n, pos1 in pairs(RE_nodes) do
+			meta1 = minetest.get_meta(pos1)
+			meta1:set_int(eu_input_str, 0)
+		end
+
 	end,
 })
+
+-- Timeout ABM
+-- Timeout for a node in case it was disconnected from the network
+-- A node must be touched by the station continuously in order to function
+local function switching_station_timeout_count(pos, tier)
+	local meta = minetest.get_meta(pos)
+	local timeout = meta:get_int(tier.."_EU_timeout")
+	if timeout <= 0 then
+		meta:set_int(tier.."_EU_input", 0) -- Not needed anymore <-- actually, it is for supply converter
+		return true
+	else
+		meta:set_int(tier.."_EU_timeout", timeout - 1)
+		return false
+	end
+end
+minetest.register_abm({
+	label = "Machines: timeout check",
+	nodenames = {"group:technic_machine"},
+	interval   = 1,
+	chance     = 1,
+	action = function(pos, node, active_object_count, active_object_count_wider)
+		local meta = minetest.get_meta(pos)
+		for tier, machines in pairs(technic.machines) do
+			if machines[node.name] and switching_station_timeout_count(pos, tier) then
+				local nodedef = minetest.registered_nodes[node.name]
+				if nodedef and nodedef.technic_disabled_machine_name then
+					node.name = nodedef.technic_disabled_machine_name
+					minetest.swap_node(pos, node)
+				elseif nodedef and nodedef.technic_on_disable then
+					nodedef.technic_on_disable(pos, node)
+				end
+				if nodedef then
+					local meta = minetest.get_meta(pos)
+					meta:set_string("infotext", S("%s Has No Network"):format(nodedef.description))
+				end
+			end
+		end
+	end,
+})
+
+--Re-enable disabled switching station if necessary, similar to the timeout above
+minetest.register_abm({
+	label = "Machines: re-enable check",
+	nodenames = {"technic:switching_station"},
+	interval   = 1,
+	chance     = 1,
+	action = function(pos, node, active_object_count, active_object_count_wider)
+		local meta = minetest.get_meta(pos)
+		local pos1 = {x=pos.x,y=pos.y-1,z=pos.z}
+		local tier = technic.get_cable_tier(minetest.get_node(pos1).name)
+		if not tier then return end
+		if switching_station_timeout_count(pos, tier) then
+			local meta = minetest.get_meta(pos)
+			meta:set_int("active",1)
+		end
+	end,
+})
+
+for tier, machines in pairs(technic.machines) do
+	-- SPECIAL will not be traversed
+	technic.register_machine(tier, "technic:switching_station", "SPECIAL")
+end
+

--
Gitblit v1.8.0