Base_Power_v2

-- This program is designed to practice printing fancy graphics on the monitor/terminal
-- Program Name: Base_System_v2
local programName = "DrtOS v1.1"
print(programName.." now running...\n")
local mon = peripheral.wrap("right")

local c1 = peripheral.wrap("Ultimate Energy Cube_1")
local c2 = peripheral.wrap("Ultimate Energy Cube_2")
local c3 = peripheral.wrap("Ultimate Energy Cube_3")
local c4 = peripheral.wrap("Ultimate Energy Cube_4")
local c5 = peripheral.wrap("Ultimate Energy Cube_5")

local r1 = peripheral.wrap("BigReactors-Reactor_1")

mon.setTextScale(0.5)
mon.clear()

local percent = 0
local percentBarValue = 0
local energyLastScan = 0
local levelTotal = 0
local count = 0

local hydroGenStatus = 0
local reactorStatus = 0


--==========================================================================================--
--==========================================================================================--
--==========================================================================================--
local function clear()
    mon.setBackgroundColor(colors.black)
    mon.setTextColor(colors.white)
    mon.setCursorPos(1,1)
end

local function clearLine(x,y) --Clears the specific line of the cursor position, then returns cursor to original position
    local xPos = 0
    local yPos = 0
    mon.getCursorPos(xPos,yPos)
    mon.setCursorPos(x,y)
    mon.clearLine()
    mon.setCursorPos(xPos,yPos)
end

local function printMon(str, x, y, color_txt, color_bg) --Prints a message to the monitor, at locations x,y
    mon.setCursorPos(x, y)
    mon.setBackgroundColor(color_bg)
    mon.setTextColor(color_txt)
    mon.write(str)
    mon.setBackgroundColor(colors.black)
end

local function printBar(x, y, length, height, color_bar) --Draws a bar on the monitor, at location x,y, with specified values.
    for yPos = y, y+height -1 do
        mon.setBackgroundColor(color_bar)
        mon.setCursorPos(x,yPos)
        mon.write(string.rep(" ", length))
        mon.setBackgroundColor(colors.black)
    end
end

local function getBatStats() --Pulls battery levels and max energy levels, then calculates total percentage.
    local c1Level = c1.getStored()
    local c2Level = c2.getStored()
    local c3Level = c3.getStored()
    local c4Level = c4.getStored()
    local c5Level = c5.getStored()

    local c1Max = c1.getMaxEnergy()
    local c2Max = c2.getMaxEnergy()
    local c3Max = c3.getMaxEnergy()
    local c4Max = c4.getMaxEnergy()
    local c5Max = c5.getMaxEnergy()

    levelTotal = c1Level + c2Level + c3Level + c4Level + c5Level
    local maxTotal = c1Max + c2Max + c3Max + c4Max + c5Max

    percent = (levelTotal / maxTotal) * 100
    if percent < 1 then
        percent = 0
    end
    percent = math.ceil(percent)
end

local function getMath() --Calculates numbers for percent bar, and controls reactor/hydrogen status
    percentBarValue = percent * 0.26
    if percentBarValue > 25 then
        percentBarValue = 25
    end
    if percentBarValue < 1 then
        percenBar = 1
    end


    if percent <= 80 then
        hydroGenStatus = 1
    end
    if percent >= 95 then
        hydroGenStatus = 0
    end

    if hydroGenStatus == 1 then
        rs.setOutput("left",true)
    else
        rs.setOutput("left",false)
    end



    if percent <= 60 then
        reactorStatus = 1
    end
    if percent >= 99 then
        reactorStatus = 0
    end

    if reactorStatus == 1 then
        r1.setActive(true)
    else
        r1.setActive(false)
    end


end

local function energyChangeRate() --Takes average over 5 seconds of current base charge/delpetion rate
    local count2 = 0
    count = count + 1
    if count > 5 then count = 1 end

    local chgValue = {ch1, ch2, chg3, chg4, chg5}
    local chgValue2 = {total1, total2, total3, total4, total5}
    local chgAvgTotal = {avg1, avg2, avg3, avg4, avg5}

    chgValue[count] = energyLastScan
    chgValue2[count] = levelTotal
    chgAvgTotal[count] = (chgValue2[count]) - (chgValue[count])

    while count2 < 5 do --Checks each value for a null, if found sets it to 1. May slightly skew data when first starting up.
        if  count2 == 5 then
            count2 = 0
        end
        count2 = count2 +1
        if chgAvgTotal[count2] == null then
            chgAvgTotal[count2] = 1
        end
    end

    local energyAvg = ((chgAvgTotal[1]) + (chgAvgTotal[2]) + (chgAvgTotal[3])+ (chgAvgTotal[4]) + (chgAvgTotal[5])) / 5
    energyAvg = math.floor(((energyAvg * 0.4) * 10^-3) /20)
    clearLine(27,6)
    printMon(energyAvg.." kRF/t", 27,6,colors.white,colors.black)

end
--==========================================================================================--
--==========================================================================================--
--==========================================================================================--
clear()
sleep(0.5)

printBar(1,1,36,1,colors.gray)
printMon("Drt OS_v1.1",14,1, colors.orange, colors.gray)

while true do
    getMath()
    getBatStats()
    energyChangeRate()

    printMon("Energy Level",1, 4,colors.white, colors.black)
    clearLine(20,5)
    printMon(percent.."%",27,5, colors.white, colors.black)
    printBar(1,5,25,2,colors.gray)
    printBar(1,5,percentBarValue,2,colors.green)

    clearLine(1,9)
    clearLine(1,10)
    printMon("Backup Generator Status: ",1,9,colors.white,colors.black)
    printMon("Nuclear Reactor Status: ",1,10,colors.white,colors.black)

    if hydroGenStatus == 1 then
        printMon("ACTIVE",28,9,colors.green,colors.black)
    else
        printMon("STANDBY",27.5,9,colors.red,colors.black)
    end

    if reactorStatus == 1 then
        printMon("ACTIVE",28,10,colors.green,colors.black)
    else
        printMon("STANDBY",27.5,10,colors.red,colors.black)
    end
    sleep(1)
    energyLastScan = levelTotal
end