techage/basis/networks.lua

--[[

	TechAge
	=======

	Copyright (C) 2019 Joachim Stolberg

	GPL v3
	See LICENSE.txt for more information
	
	Networks - the connection of tubelib2 tube/pipe/cable lines to networks
]]--

local S2P = minetest.string_to_pos
local P2S = minetest.pos_to_string
local M = minetest.get_meta
local N = techage.get_node_lvm
local S = techage.S
local hex = function(val) return string.format("%x", val) end

local Networks = {} -- cache for networks

techage.networks = {}  -- name space

local MAX_NUM_NODES = 500
local BEST_BEFORE = 5 * 60  -- 5 minutes
local Route = {} -- Used to determine the already passed nodes while walking
local NumNodes = 0
local DirToSide = {"B", "R", "F", "L", "D", "U"}
local Sides = {B = true, R = true, F = true, L = true, D = true, U = true}
local SideToDir = {B=1, R=2, F=3, L=4, D=5, U=6}
local Flip = {[0]=0,3,4,1,2,6,5} -- 180 degree turn

local function error(pos, msg)
	minetest.log("error", "[techage] "..msg.." at "..P2S(pos).." "..N(pos).name)
end

local function count_nodes(ntype, nodes)
	local num = 0
	for _,pos in ipairs(nodes or {}) do
		num = num + 1
	end
	return ntype.."="..num
end

local function output(network, valid)
	local tbl = {}
	for ntype,table in pairs(network) do
		if type(table) == "table" then
			tbl[#tbl+1] = count_nodes(ntype, table)
		end
	end
	print("Network ("..valid.."): "..table.concat(tbl, ", "))
end

-- return the node definition local networks table
local function net_def(pos, net_name) 
	local ndef = minetest.registered_nodes[techage.get_node_lvm(pos).name]
	return ndef and ndef.networks and ndef.networks[net_name] or {} 
end

local function net_def2(node_name, net_name) 
	local ndef = minetest.registered_nodes[node_name]
	return ndef and ndef.networks and ndef.networks[net_name] or {} 
end

-- Calculate the node outdir based on node.param2 and nominal dir (according to side)
local function dir_to_outdir(dir, param2)
	if dir < 5 then
		return ((dir + param2 - 1) % 4) + 1
	end
	return dir
end

local function indir_to_dir(indir, param2)
	if indir < 5 then
		return ((indir - param2 + 5) % 4) + 1
	end
	return Flip[indir]
end

local function outdir_to_dir(outdir, param2)
	if outdir < 5 then
		return ((outdir - param2 + 3) % 4) + 1
	end
	return outdir
end

local function side_to_outdir(pos, side)
	return dir_to_outdir(SideToDir[side], techage.get_node_lvm(pos).param2)
end

-- Get tlib2 connection dirs as table
-- used e.g. for the connection walk
local function get_node_connections(pos, net_name)
	local val = M(pos):get_int(net_name.."_conn")
    local tbl = {}
    if val % 0x40 >= 0x20 then tbl[#tbl+1] = 1 end
    if val % 0x20 >= 0x10 then tbl[#tbl+1] = 2 end
    if val % 0x10 >= 0x08 then tbl[#tbl+1] = 3 end
    if val % 0x08 >= 0x04 then tbl[#tbl+1] = 4 end
    if val % 0x04 >= 0x02 then tbl[#tbl+1] = 5 end
    if val % 0x02 >= 0x01 then tbl[#tbl+1] = 6 end
    return tbl
end

-- determine all node sides with tube connections
local function node_connections(pos, tlib2)
	local node = techage.get_node_lvm(pos)
	local val = 0
	local sides = net_def(pos, tlib2.tube_type).sides
	
	if sides then
		for dir = 1,6 do
			val = val * 2
			local side = DirToSide[outdir_to_dir(dir, node.param2)]
			if sides[side] then
				if tlib2:connected(pos, dir) then
					val = val + 1
				end
			end
		end
		M(pos):set_int(tlib2.tube_type.."_conn", val)
	else
		error(pos, "sides missing")
	end
end

local function pos_already_reached(pos)
	local key = minetest.hash_node_position(pos)
	if not Route[key] and NumNodes < MAX_NUM_NODES then
		Route[key] = true
		NumNodes = NumNodes + 1
		return false
	end
	return true
end

-- check if the given pipe dir out of the node is valid
local function valid_indir(pos, indir, param2, net_name)
	local sides = net_def(pos, net_name).sides
	if not sides then return false end
	local side = DirToSide[indir_to_dir(indir, param2)]
	if not sides[side] then return false end
	return true
end

-- do the walk through the tubelib2 network
-- indir is the direction which should not be covered by the walk
-- (coming from there or is a different network)
local function connection_walk(pos, indir, node, tlib2, clbk)
	if clbk then clbk(pos, indir, node) end
	for _,outdir in pairs(get_node_connections(pos, tlib2.tube_type)) do
		if outdir ~= Flip[indir] then
			local pos2, indir2 = tlib2:get_connected_node_pos(pos, outdir)
			local node = techage.get_node_lvm(pos2)
			if pos2 and not pos_already_reached(pos2) and 
						valid_indir(pos2, indir2, node.param2, tlib2.tube_type) then
				connection_walk(pos2, indir2, node, tlib2, clbk)
			end
		end
	end
end


local function collect_network_nodes(pos, outdir, tlib2)
	Route = {}
	NumNodes = 0
	pos_already_reached(pos) 
	local netw = {}
	local node = techage.get_node_lvm(pos)
	local net_name = tlib2.tube_type
	-- outdir corresponds to the indir coming from
	connection_walk(pos, outdir, node, tlib2, function(pos, indir, node)
		local ntype = net_def(pos, net_name).ntype
		print("collect_network_nodes", P2S(pos), ntype)
		if ntype then
			if not netw[ntype] then netw[ntype] = {} end
			netw[ntype][#netw[ntype] + 1] = {pos = pos, indir = indir}
		end
	end)
	netw.best_before = minetest.get_gametime() + BEST_BEFORE
	return netw
end

-- keep data base small and valid
local function remove_outdated_networks()
	local to_be_deleted = {}
	local t = minetest.get_gametime()
	for net_name,tbl in pairs(Networks) do
		for netID,network in pairs(tbl) do
			local valid = (network.best_before or 0) - t
			output(network, valid)
			if valid < 0 then
				to_be_deleted[#to_be_deleted+1] = {net_name, netID}
			end
		end
	end
	for _,item in ipairs(to_be_deleted) do
		local net_name, netID = unpack(item)
		print("delete", net_name, netID)
		Networks[net_name][netID] = nil
	end
	minetest.after(60, remove_outdated_networks)
end
minetest.after(60, remove_outdated_networks)

--
-- API Functions
--

-- Table fo a 180 degree turn
techage.networks.Flip = Flip

-- techage.networks.net_def(pos, net_name)
techage.networks.net_def = net_def

techage.networks.AllSides = Sides -- table for all 6 node sides

-- techage.networks.side_to_outdir(pos, side)
techage.networks.side_to_outdir = side_to_outdir

-- check if the given pipe dir into the node is valid
-- valid_indir(pos, indir, param2, net_name)
techage.networks.valid_indir = valid_indir

-- techage.networks.node_connections(pos, tlib2)
techage.networks.node_connections = node_connections

-- techage.networks.collect_network_nodes(pos, outdir, tlib2)
techage.networks.collect_network_nodes = collect_network_nodes

function techage.networks.connection_walk(pos, outdir, tlib2, clbk)
	Route = {}
	NumNodes = 0
	pos_already_reached(pos) -- don't consider the start pos
	local node = techage.get_node_lvm(pos)
	connection_walk(pos, outdir, node, tlib2, clbk)
	return NumNodes
end

function techage.networks.get_network(netID, tlib2)
	if Networks[tlib2.tube_type] and Networks[tlib2.tube_type][netID] then
		Networks[tlib2.tube_type][netID].best_before = minetest.get_gametime() + BEST_BEFORE
		return Networks[tlib2.tube_type][netID]
	end
end

function techage.networks.set_network(netID, tlib2, network)
	if netID then
		if not Networks[tlib2.tube_type] then
			Networks[tlib2.tube_type] = {}
		end
		Networks[tlib2.tube_type][netID] = network
		Networks[tlib2.tube_type][netID].best_before = minetest.get_gametime() + BEST_BEFORE
	end
end

function techage.networks.trigger_network(netID, tlib2)
	if not Networks[tlib2.tube_type] then
		Networks[tlib2.tube_type] = {}
	end
	Networks[tlib2.tube_type][netID].best_before = minetest.get_gametime() + BEST_BEFORE
end

function techage.networks.delete_network(netID, tlib2)
	if Networks[tlib2.tube_type] and Networks[tlib2.tube_type][netID] then
		Networks[tlib2.tube_type][netID] = nil
	end
end

function techage.networks.connections(pos, tlib2)
	for _,dir in ipairs(get_node_connections(pos, tlib2.tube_type)) do
		print(({"North", "East", "South", "West", "Down", "Up"})[dir])
	end
end