local component = require("component")
local sides = require("sides")
local rs = component.redstone
local gpu = component.gpu
local event = require("event")
local colours = require("colors")

function trainDirection(direction)
    if direction == "north" then
        return 0
    elseif direction == "east" then
        return 1
    elseif direction == "south" then
        return 2
    else
        return 3
    end
end

function getStoredTrain()
    for i = 1, 16, 1 do
        if rs.getBundledInput(sides.north, i) == 1 then
            return i
        end
    end
end

function countTrains()
    local count = 0
    for i = 0, 16, 1 do
        if rs.getBundledInput(sides.north, i) == 1 then
            count = count + 1
        end
    end
    return count
end

function releaseTrain(id)
    rs.setBundledOutput(sides.east, id, 1)
    os.sleep(2)
    rs.setBundledOutput(sides.east, id, 0)
end

function setDestination(dire)
    rs.setBundledOutput(sides.bottom, dire, 1)
end

function resetDestination()
    for i = 0, 16, 1 do
        rs.setBundledOutput(sides.bottom, i, 0)
    end
end

function drawButtons()
    old = gpu.setBackground(colours.red)
    gpu.fill(10, 5, 20, 3, " ")
    gpu.setBackground(old)
end

drawButtons()
while true do
    local _, addr, x, y, but, player = event.pull("touch")
    os.sleep(1)
end