--[[
Umaroth's Power Monitor v1.3

Monitors TE4 Energy Cells and EnderIO Capacitor Banks and outputs a redstone signal once energy storage drops below set limits.
Will automatically detect any compatible energy storage and monitors and connect to all of them. They can be either directly adjacent to the computer or connected through a wired modem network. Monitor size must be 2x3 or 1x2.
Redstone signal output for the engines is off by default but can be changed easily on line 18.
Also supports BigReactors Reactors and Turbines. Maximum 1 of each.

Requires either OpenPeripheralsIntegration or Computronics or both. Computronics is recommended because OpenP doesn't fully support Capacitor Banks. If you have OpenPeripherals but not Computronics you must use the workaround on line 19 for Capacitor Banks to work.

Script overhaul by Umaroth. Original script located here: http://tinyurl.com/kuaxv2p
Changelog/README: http://pastebin.com/erPf3t4t
--]]

--You can change these:
local upper = 0.98 --Upper limit for computer to stop transmitting redstone signal. 0.98=98% full.
local lower = 0.10 --Lower limit for computer to start transmitting redstone signal.
local redstoneSide = "none" -- Change this to the side you want to output the redstone signal to. ["left","right","top","bottom","front","back","none"]
local capacitorBankBlocks = 300 -- If you have OpenPeripherals without Computronics you need to specify how many blocks your Capacitor Bank contains. Only works properly for one Capacitor Bank. If you have Computronics, this variable won't do anything.

--Don't change these:
rednet.open("right")
cellCount = 0
connectedCells = {}
connectedOPCapBank = ""
monitorCount = 0
connectedMonitors = {}
TE4Cell = 0 EIOCell = 0 OPCapBank = 0 VEIOCell = 0
periList = peripheral.getNames()
validPeripherals = {
    -- "tile_thermalexpansion_cell",
    -- "powered_tile",
    "tile_blockcapacitorbank_name",
    "capacitor_bank",
	"io_config_tile",
    "monitor",
    "BigReactors%-Turbine",
    "BigReactors%-Reactor"
}

function checkValidity(periName)
    for n,b in pairs(validPeripherals) do
        if periName:find(b) then return b end
    end
    return false
end

for i,v in ipairs(periList) do
    local periFunctions = {
        -- ["tile_thermalexpansion_cell"] = function()
            -- cellCount = cellCount + 1
            -- TE4Cell= TE4Cell + 1
            -- connectedCells[cellCount] = periList[i]
        -- end,
        -- ["powered_tile"] = function()
            -- cellCount = cellCount + 1
            -- TE4Cell= TE4Cell + 1
            -- connectedCells[cellCount] = periList[i]
        -- end,
        ["tile_blockcapacitorbank_name"] = function()
            EIOCell = EIOCell + 1
            OPCapBank = OPCapBank + 1
            connectedOPCapBank = periList[i]
        end,
        ["capacitor_bank"] = function()
            cellCount = cellCount + 1
            EIOCell = EIOCell + 1
            connectedCells[cellCount] = periList[i]
        end,
		["io_config_tile"] = function()
            cellCount = cellCount + 1
            VEIOCell = VEIOCell + 1
            connectedCells[cellCount] = periList[i]
        end,
        ["monitor"] = function()
            monitorCount = monitorCount + 1
            connectedMonitors[monitorCount] = periList[i]
        end,
        ["BigReactors%-Turbine"] = function()
            turbine = peripheral.wrap(periList[i])
        end,
        ["BigReactors%-Reactor"] = function()
            reactor = peripheral.wrap(periList[i])
        end
    }

    local isValid = checkValidity(peripheral.getType(v))
    if isValid then periFunctions[isValid]() end
end

--Compatibility with OpenPeripherals
if OPCapBank > 1 then
    print("Error: Without Computronics this script can only support a maximum of one Capacitor Bank. Exiting Script!")
    return
elseif OPCapBank == 1 and capacitorBankBlocks == 0 then
    print("Warning: You have not entered how many blocks your Capacitor Bank contains, the script will not return the correct numbers. Please fix this by editing the script and changing the variable 'capacitorBankBlocks'.")
elseif OPCapBank == 1 then
    print("Warning: OpenPeripherals does not fully support Capacitor Banks, numbers may not be fully accurate.")
end

--Check for storage cells and monitors before continuing
if cellCount == 0 and OPCapBank == 0 then
    print("No RF storage found. Exiting script!")
    return
end
if monitorCount == 0 then
    print("No Monitor found. Exiting script!")
    return
end

--Function to set monitor sizes
function getMonitorSize(x, y)
    if x == 18 and y == 5 then
        return "small"
    elseif x == 29 and y == 12 then
        return "large"
    else
        print("Invalid monitor size detected. Exiting script!")
        return
    end
end

--Check monitor sizes before continuing
for i = 1, #connectedMonitors do
    local monitor = peripheral.wrap(connectedMonitors[i])
    if getMonitorSize(monitor.getSize()) == nil then
        return
    end
end

--Print connected peripherals
print("Peripherals connected:")
if monitorCount > 1 then print(monitorCount.." Monitors") else print(monitorCount.." Monitor") end
if TE4Cell ~= 1 then print(TE4Cell.." TE Energy Cells") else print(TE4Cell.." TE Energy Cell") end
if EIOCell ~= 1 then print(EIOCell.." Capacitor Banks") else print(EIOCell.." Capacitor Bank") end
if VEIOCell ~= 1 then print(VEIOCell.." Vibrant Capacitor Banks") else print(VEIOCell.." Vibrant Capacitor Bank") end
if turbine ~= nil then print ("1 Turbine") else print ("0 Turbines") end
if reactor ~= nil then print ("1 Reactor") else print ("0 Reactors") end

--Main code

--Set default output states to off
if redstoneSide ~= "none" then redstone.setOutput(redstoneSide, false) end
if turbine ~= nil then turbine.setActive(false) end
-- if reactor ~= nil then reactor.setActive(false) end
if reactor ~= nil then reactor.setAllControlRodLevels(100) end

--Write default engine status to all attached monitors
for i = 1, #connectedMonitors do
    local monitor = peripheral.wrap(connectedMonitors[i])
    if getMonitorSize(monitor.getSize()) == "large" then
        monitor.clear()
       monitor.setBackgroundColour((colours.grey))
        monitor.setCursorPos(11,4)
        monitor.write(" ON ")
       monitor.setBackgroundColour((colours.red))
        monitor.setCursorPos(15,4)
        monitor.write(" OFF ")
       monitor.setBackgroundColour((colours.black))
    else
        monitor.clear()
       monitor.setBackgroundColour((colours.grey))
        monitor.setCursorPos(1,4)
        monitor.write(" ON ")
       monitor.setBackgroundColour((colours.green))
        monitor.setCursorPos(5,4)
        monitor.write(" OFF ")
       monitor.setBackgroundColour((colours.black))
    end
end


--Main loop
while true do

  --Get all dynamic values
    --Get storage values
    local eNow = 0 eMax = 0 cellLoops = 0
    for i = 1, #connectedCells do
        cell = peripheral.wrap(connectedCells[i])
        eNow = eNow + cell.getEnergyStored()
        eMax = eMax + cell.getMaxEnergyStored()
        cellLoops = i
    end
    --Compatibility with OpenPeripherals
    -- if OPCapBank == 1 and cellLoops == #connectedCells then
        -- cell = peripheral.wrap(connectedOPCapBank)
        -- eNow = (eNow + cell.getEnergyStored()) * capacitorBankBlocks
        -- eMax = (eMax + cell.getMaxEnergyStored()) * capacitorBankBlocks
    -- end

	if VEIOCell >= 1 then
        eNow = eNow * capacitorBankBlocks
        eMax = eMax * capacitorBankBlocks
	end

    --Compute engine activation ratio
    local fill = eNow / eMax

    --Set storage scale
    if eNow >= 1000000000 then eNowScale = "billion"
    elseif eNow >= 1000000 then eNowScale = "million"
    else eNowScale = "none" end
    if eMax >= 1000000000 then eMaxScale = "billion"
    elseif eMax >= 1000000 then eMaxScale = "million"
    else eMaxScale = "none" end

    --Adjust number to scale
    if eNowScale == "billion" then eNowValue = math.ceil(eNow / 1000000)
    elseif eNowScale == "million" then eNowValue = math.ceil(eNow / 1000)
    else eNowValue = math.ceil(eNow) end
    if eMaxScale == "billion" then eMaxValue = math.ceil(eMax / 1000000)
    elseif eMaxScale == "million" then eMaxValue = math.ceil(eMax / 1000)
    else eMaxValue = math.ceil(eMax) end

    --Adjust suffix to scale
    if eNowScale == "billion" then eNowSuffixLarge = " Mrd RF" eNowSuffixSmall = "Mio RF"
    elseif eNowScale == "million" then eNowSuffixLarge = " Mio RF" eNowSuffixSmall = "Ki RF"
    else eNowSuffixLarge = " RF" eNowSuffixSmall = " RF" end
    if eMaxScale == "billion" then eMaxSuffixLarge = " Mrd RF" eMaxSuffixSmall = "Mio RF"
    elseif eMaxScale == "million" then eMaxSuffixLarge = " Mio RF" eMaxSuffixSmall = "Ki RF"
    else eMaxSuffixLarge = " RF" eMaxSuffixSmall = " RF" end

    --Get number of digits to write
    local eNowDigitCount = 0 eMaxDigitCount = 0
    for digit in string.gmatch(eNowValue, "%d") do eNowDigitCount = eNowDigitCount + 1 end
    for digit in string.gmatch(eMaxValue, "%d") do eMaxDigitCount = eMaxDigitCount + 1 end

	function comma_value(eNowValue)
	  local formatted = eNowValue
	  while true do
		formatted, k = string.gsub(formatted, "^(-?%d+)(%d%d%d)", '%1,%2')
		if (k==0) then
		  break
		end
	  end
	  return formatted
	end

	function comma_value(eMaxValue)
	  local formatted = eMaxValue
	  while true do
		formatted, k = string.gsub(formatted, "^(-?%d+)(%d%d%d)", '%1,%2')
		if (k==0) then
		  break
		end
	  end
	  return formatted
	end

    --Get location to write
    if eNowSuffixLarge ~= " RF" then eNowXLarge = 13 - eNowDigitCount
    else eNowXLarge = 17 - eNowDigitCount end
    eNowXSmall = 15 - eNowDigitCount
    if eMaxSuffixLarge ~= " RF" then eMaxXLarge = 13 - eMaxDigitCount
    else eMaxXLarge = 17 - eMaxDigitCount end
    eMaxXSmall = 15 - eMaxDigitCount

    --Loop to write to every monitor
    for i = 1, #connectedMonitors do
        local monitor=peripheral.wrap(connectedMonitors[i])

        if getMonitorSize(monitor.getSize()) == "large" then
            --Erase old data
            monitor.setCursorPos(10,9)
            monitor.write("       ")
            monitor.setCursorPos(10,11)
            monitor.write("       ")
            --Write constant/new data
            monitor.setCursorPos(12,2)
            monitor.write("Reaktor:")
            monitor.setCursorPos(9,7)
            monitor.write("Stromspeicher:")
            monitor.setCursorPos(eNowXLarge,9)
            monitor.write(comma_value(eNowValue)..eNowSuffixLarge)
            monitor.setCursorPos(eMaxXLarge,10)
			monitor.setCursorPos(12.5,10)
            monitor.write("von:")
            monitor.setCursorPos(eMaxXLarge,11)
            monitor.write(comma_value(eMaxValue)..eMaxSuffixLarge)
            if fill > upper then
                --Energy level is over upper level, turning redstone/reactors off
                if redstoneSide ~= "none" then redstone.setOutput(redstoneSide, false) end
                if turbine ~= nil then turbine.setActive(false) end
                -- if reactor ~= nil then reactor.setActive(false) end
                if reactor ~= nil then reactor.setAllControlRodLevels(100) end
				rednet.broadcast("an")
                monitor.setBackgroundColour((colours.grey))
                monitor.setCursorPos(11,4)
                monitor.write(" ON ")
                monitor.setBackgroundColour((colours.red))
                monitor.setCursorPos(15,4)
                monitor.write(" OFF ")
                monitor.setBackgroundColour((colours.black))
            elseif fill < lower then
                --Energy level is below lower limit, turning redstone/reactors on
                if redstoneSide ~= "none" then redstone.setOutput(redstoneSide, true) end
                if turbine ~= nil then turbine.setActive(true) end
                -- if reactor ~= nil then reactor.setActive(true) end
                if reactor ~= nil then reactor.setAllControlRodLevels(0) end
				rednet.broadcast("aus")
                monitor.setBackgroundColour((colours.green))
                monitor.setCursorPos(11,4)
                monitor.write(" ON ")
                monitor.setBackgroundColour((colours.grey))
                monitor.setCursorPos(15,4)
                monitor.write(" OFF ")
                 monitor.setBackgroundColour((colours.black))
            end
            for i = 1, math.ceil(fill * 10) do
                monitor.setBackgroundColour((colours.green))
                monitor.setCursorPos(24,12-i)
                monitor.write(" ")
                monitor.setBackgroundColour((colours.black))
            end
            for i = 1, 10 - math.ceil(fill * 10) do
                 monitor.setBackgroundColour((colours.red))
                monitor.setCursorPos(24,1+i)
                monitor.write(" ")
                monitor.setBackgroundColour((colours.black))
            end
        elseif getMonitorSize(monitor.getSize()) == "small" then
            --erase old data
            monitor.setCursorPos(10,3)
            monitor.write("       ")
            monitor.setCursorPos(10,5)
            monitor.write("       ")
            --write constant/new data
            monitor.setCursorPos(2,2)
            monitor.write("Engines:")
            monitor.setCursorPos(11,2)
            monitor.write("Storage:")
            monitor.setCursorPos(eNowXSmall,3)
            monitor.write(eNowValue..eNowSuffixSmall)
            monitor.setCursorPos(eMaxXSmall,4)
            monitor.write("of:")
            monitor.setCursorPos(eMaxXSmall,5)
            monitor.write(eMaxValue..eMaxSuffixSmall)
            if fill > upper then
                --Energy level is over upper level, turning redstone/reactors off
                if redstoneSide ~= "none" then redstone.setOutput(redstoneSide, false) end
                if turbine ~= nil then turbine.setActive(false) end
                -- if reactor ~= nil then reactor.setActive(false) end
                if reactor ~= nil then reactor.setAllControlRodLevels(100) end
                monitor.setBackgroundColour((colours.grey))
                monitor.setCursorPos(1,4)
                monitor.write(" ON ")
                monitor.setBackgroundColour((colours.green))
                monitor.setCursorPos(5,4)
                monitor.write(" OFF ")
                monitor.setBackgroundColour((colours.black))
            elseif fill < lower then
                --Energy level is below lower limit, turning redstone/reactors on
                if redstoneSide ~= "none" then redstone.setOutput(redstoneSide, true) end
                if turbine ~= nil then turbine.setActive(true) end
                -- if reactor ~= nil then reactor.setActive(true) end
                if reactor ~= nil then reactor.setAllControlRodLevels(0) end
                 monitor.setBackgroundColour((colours.green))
                monitor.setCursorPos(1,4)
                monitor.write(" ON ")
                monitor.setBackgroundColour((colours.grey))
                monitor.setCursorPos(5,4)
                monitor.write(" OFF ")
                 monitor.setBackgroundColour((colours.black))
            end
        end
    end
    sleep(1)
end --while