Reactor DW20 Big Reactors control program

--[[
Installer pastebin code http://pastebin.com/MuS68wCr
Reactor pastebin code http://pastebin.com/YbHGQ2wY
Button pastebin code http://pastebin.com/wGAt78pv
Startup pastebin code http://pastebin.com/irim4gRD

Changes:
   -2/16/2015
   - pulling hot water capacity from reactor in active cooled mode
   - verfied 3 peak speed for turbines default speed is 1820. Low is 900, High 2728
   - user set-able reactor offset for cold start-up default % increase is 5
   - build Histarics to help turbines stay in high output zones
   - corrected some venting issues causing long start-up time is odd cases
   - functions turbineWarmup and turbineRunning to help reduce code and help histarics implementation.


   -2/15/2015
   - adjusted hot water start to 20,000 mb
   - set turnOnAt to 60 from 50
   - set numCapacitors to 10 for house from 3
   - function setVenting() for dumping water in reactor
   - function setVentRuning() closing turbine before start-up
]]--

os.loadAPI("button")

p = peripheral.find("tile_blockcapacitorbank_name")
m = peripheral.find("monitor")
r = peripheral.find("BigReactors-Reactor")
t = peripheral.find("BigReactors-Turbine")

local steamReactor = r.isActivelyCooled()

local numCapacitors = 27 -- number of capacitors in your capacitor bank
local turnOnAt = 60 -- capacitor percentage to turn coils on in turbine or turn on the reactor
local turnOffAt = 90 -- capacitor percentage to shut coils off on turbine or shut reactor off
local reactorOffset = 5 -- used to help cold start-ups this is % of rods in during start-up
local steamMin = 2000 -- Min um steam to start the program lower only to accommodate 1 rod reactors
local targetSpeed = 1840 -- valid speeds peaks 900, (1840), 2728

local energy = 0
local energyStored = 0
local energyMax = 0
local energyStoredPercent = 0
local timerCode
local mode = "Automatic"
local RFProduction = 0
local fuelUse = 0
local coreTemp = 0
local reactorOnline = false
local rodLevel = 0

local turbineOnline = false
local turbineRotorSpeed = 0
local turbineRFGen = 0
local turbineFluidRate = 0
local turbineInductor = false

local OptFuelRodLevel = 0

if (steamReactor ~= true) then
    local menuType = "Reactor"
else
    local menuType = "Turbine"
end

function autoMenu()
    m.setTextScale(1)
    button.clearTable()
    button.setTable("Automatic", autoMode, "", 3, 13, 6, 6)
    button.setTable("Manual", manualMode, "", 15, 25, 6, 6)

    if steamReactor then
       button.setTable("Reactor", reactorMenu, "", 5, 18, 19, 19)
       button.setTable("Turbine", turbineMenu, "", 22, 35, 19, 19)
    end
    button.screen()
    checkMode()
    menuMode()
end

function manualMenu()
    m.setTextScale(1)
    button.clearTable()
    button.setTable("Automatic", autoMode, "", 3, 13, 6, 6)
    button.setTable("Manual", manualMode, "", 15, 25, 6, 6)
    button.setTable("Online", online, "", 3, 13, 8, 8)
    button.setTable("Offline", offline, "", 15, 25, 8, 8)
    button.setTable("0", setRods, 0, 11,14, 10, 10)
    button.setTable("10", setRods, 10, 5,8, 12, 12)
    button.setTable("20", setRods, 20, 11,14, 12, 12)
    button.setTable("30", setRods, 30, 17,20, 12, 12)
    button.setTable("40", setRods, 40, 5,8, 14, 14)
    button.setTable("50", setRods, 50, 11,14, 14, 14)
    button.setTable("60", setRods, 60, 17,20, 14, 14)
    button.setTable("70", setRods, 70, 5,8, 16, 16)
    button.setTable("80", setRods, 80, 11,14, 16, 16)
    button.setTable("90", setRods, 90, 17,20, 16, 16)
    button.setTable("+", rodPlus, "", 23, 25, 12, 12)
    button.setTable("-", rodMinus, "", 23, 25, 16, 16)
    if steamReactor then
        button.setTable("Reactor", reactorMenu, "", 5, 18, 19, 19)
        button.setTable("Turbine", turbineMenu, "", 22, 35, 19, 19)
    end
    button.screen()
    checkMode()
    reactorOnOff()
    menuMode()
end

function turbineAutoMenu()
    m.setTextScale(1)
    button.clearTable()
    button.setTable("Automatic", autoMode, "", 3, 13, 6, 6)
    button.setTable("Manual", manualMode, "", 15, 25, 6, 6)
    button.setTable("Reactor", reactorMenu, "", 5, 18, 19, 19)
    button.setTable("Turbine", turbineMenu, "", 22, 35, 19, 19)
    button.screen()
    checkMode()
    menuMode()
end

function turbineManualMenu()
    m.setTextScale(1)
    button.clearTable()
    button.setTable("Automatic", autoMode, "", 3, 13, 6, 6)
    button.setTable("Manual", manualMode, "", 15, 25, 6, 6)
    button.setTable("Reactor", reactorMenu, "", 5, 18, 19, 19)
    button.setTable("Turbine", turbineMenu, "", 22, 35, 19, 19)
    button.setTable("Online", setTurbineOnline, "", 3, 13, 8, 8)
    button.setTable("Offline", setTurbineOffline, "", 15, 25, 8, 8)
    button.setTable("Coils On", coilsOn, "", 3, 13, 10, 10)
    button.setTable("Coils Off", coilsOff, "", 15, 25, 10, 10)
    button.screen()
    checkMode()
    turbineOnOff()
    coilsOnOff()
    menuMode()
end

function reactorMenu()
    menuType = "Reactor"
    displayScreen()
end

function turbineMenu()
    menuType = "Turbine"
    displayScreen()
end

function online()
    r.setActive(true)
end

function offline()
    r.setActive(false)
end

function setTurbineOnline()
    t.setActive(true)
end

function setTurbineOffline()
    t.setActive(false)
end

function reactorOnOff()
    button.setButton("Online", r.getActive())
    button.setButton("Offline", not r.getActive())
end

function turbineOnOff()
    button.setButton("Online", t.getActive())
    button.setButton("Offline", not t.getActive())
end

function coilsOnOff()
    button.setButton("Coils On", t.getInductorEngaged())
    button.setButton("Coils Off", not t.getInductorEngaged())
end

function coilsOn()
    t.setInductorEngaged(true)
end

function coilsOff()
    t.setInductorEngaged(false)
end

function menuMode()
    if steamReactor then
        if menuType == "Reactor" then
            button.setButton("Reactor", true)
            button.setButton("Turbine", false)
        else
            button.setButton("Reactor", false)
            button.setButton("Turbine", true)
        end
    end
end

function setRods(setLevel)
    button.flash(tostring(setLevel))
    r.setAllControlRodLevels(setLevel)
    fuelRodLevel()
end

function rodPlus()
    button.flash("+")
    r.setAllControlRodLevels(rodLevel+1)
    fuelRodLevel()
end

function rodMinus()
    button.flash("-")
    r.setAllControlRodLevels(rodLevel-1)
    fuelRodLevel()
end

function checkMode()
    button.toggleButton(mode)
end

function manualMode()
    mode = "Manual"
    manualMenu()
end

function autoMode()
    mode = "Automatic"
    displayScreen()
end

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

function displayEn()
    m.clear()
    m.setCursorPos(1,1)
    m.write("Energy Use: ")
    if energy < 0 then
        m.setTextColor(colors.red)
    else
        m.setTextColor(colors.green)
    end
    m.write(comma_value(math.floor(energy)).. "RF/t")
    m.setTextColor(colors.white)
    m.setCursorPos(1,2)
    m.write("Energy Stored: "..energyStoredPercent.."%")
    if menuType == "Reactor" then
        m.setCursorPos(1,3)
        m.write("Reactor is: ")
        if reactorOnline then
            m.setTextColor(colors.green)
            m.write("Online")
        else
            m.setTextColor(colors.red)
            m.write("Offline")
        end
        m.setTextColor(colors.white)
        m.setCursorPos(22,1)
        if steamReactor then
            m.write("Steam: ")
            m.setTextColor(colors.green)
            m.write(comma_value(math.floor(RFProduction)).."MB/t")
        else
            m.write("RF Gen: ")
            m.setTextColor(colors.green)
            m.write(comma_value(math.floor(RFProduction)).."RF/t")
        end
        m.setTextColor(colors.white)
        m.setCursorPos(22,2)
        m.write("Core Temp: "..math.floor(coreTemp).."c")
        m.setCursorPos(22,3)
        m.write("Fuel Use: "..fuelUse.."MB/t")
    else
        m.setCursorPos(1,3)
        m.write("Turbine is: ")
        if turbineOnline then
            m.setTextColor(colors.green)
            m.write("Online")
        else
            m.setTextColor(colors.red)
            m.write("Offline")
        end
        m.setCursorPos(1,4)
        m.setTextColor(colors.white)
        m.write("Reactor is: ")
        if reactorOnline then
            m.setTextColor(colors.green)
            m.write("Online")
        else
            m.setTextColor(colors.red)
            m.write("Offline")
        end
        m.setTextColor(colors.white)
        m.setCursorPos(22,1)
        m.write("RFGen: ")
        m.setTextColor(colors.green)
        m.write(comma_value(math.floor(turbineRFGen)).."RF/t")
        m.setTextColor(colors.white)
        m.setCursorPos(22,2)
        m.write("Rotor: "..comma_value(math.floor(turbineRotorSpeed)).." RPM")
        m.setCursorPos(22,3)
        m.write("Steam: "..comma_value(turbineFluidRate).."MB/t")
    end
end

function checkEn()
    local tempEnergy = 0
    energyStored = p.getEnergyStored()
    energyMax = p.getMaxEnergyStored()
    energyStoredPercent = math.floor((energyStored/energyMax)*100)
    RFProduction = r.getEnergyProducedLastTick()
    fuelUse = r.getFuelConsumedLastTick()
    fuelUse = math.floor(fuelUse*100)
    fuelUse = fuelUse/100
    coreTemp = r.getFuelTemperature()
    reactorOnline = r.getActive()
    tempEnergy = p.getEnergyStored()
    sleep(0.1)
    energy = (p.getEnergyStored()-tempEnergy)/2
    energy = energy*numCapacitors
    if steamReactor then
        turbineOnline = t.getActive()
        turbineRotorSpeed = t.getRotorSpeed()
        turbineRFGen = t.getEnergyProducedLastTick()
        turbineFluidRate = t.getFluidFlowRate()
    end
end

function fuelRodLevel()
    rodLevel = r.getControlRodLevel(0)
    m.setCursorPos(30,5)
    m.write(tostring(rodLevel).."%")
    m.setBackgroundColor(colors.white)
    m.setCursorPos(28,6)
    m.write("       ")
    for i = 1,10 do
        m.setCursorPos(28,i+6)
        m.setBackgroundColor(colors.white)
        m.write(" ")
        m.setBackgroundColor(colors.yellow)
        m.write(" ")
        if rodLevel/10 >= i then
            m.setBackgroundColor(colors.red)
        else
            m.setBackgroundColor(colors.yellow)
        end
        m.write("   ")
        m.setBackgroundColor(colors.yellow)
        m.write(" ")
        m.setBackgroundColor(colors.white)
        m.write(" ")
    end
    m.setCursorPos(28,17)
    m.write("       ")
    m.setBackgroundColor(colors.black)
end

function turbineInductorDisplay()
    turbineInductor = t.getInductorEngaged()
    m.setCursorPos(30,5)
    if turbineInductor then
        m.write("On")
    else
        m.write("Off")
    end
    m.setBackgroundColor(colors.gray)
    m.setCursorPos(28,6)
    m.write("       ")
    for i = 1,7 do
        m.setCursorPos(28,i+6)
        m.setBackgroundColor(colors.gray)
        m.write(" ")
        m.setBackgroundColor(colors.lightGray)
        m.write(" ")
        if i % 2 == 0 then
            m.setBackgroundColor(colors.gray)
        end
        m.write(" ")
        m.setBackgroundColor(colors.gray)
        m.write(" ")
        if i % 2 ~= 0 then
            m.setBackgroundColor(colors.lightGray)
        end
        m.write(" ")
        m.setBackgroundColor(colors.lightGray)
        m.write(" ")
        m.setBackgroundColor(colors.gray)
        m.write(" ")
    end
    for i = 8,10 do
        m.setCursorPos(28,i+6)
        m.setBackgroundColor(colors.gray)
        m.write(" ")
        m.setBackgroundColor(colors.lightGray)
        m.write(" ")
        if turbineInductor then
            m.setBackgroundColor(colors.red)
        else
            m.setBackgroundColor(colors.blue)
        end
        m.write(" ")
        m.setBackgroundColor(colors.gray)
        m.write(" ")
        if turbineInductor then
            m.setBackgroundColor(colors.red)
        else
            m.setBackgroundColor(colors.blue)
        end
        m.write(" ")
        m.setBackgroundColor(colors.lightGray)
        m.write(" ")
        m.setBackgroundColor(colors.gray)
        m.write(" ")
    end
    m.setCursorPos(28,17)
    m.write("       ")
    m.setBackgroundColor(colors.black)
end

function getClick()
    local event, side, x, y = os.pullEvent("monitor_touch")
    button.checkxy(x,y)
end

function setVenting()
    t.setVentAll()
    t.setActive(true)
end

function setVentRuning()
    t.setVentOverflow()
    t.setActive(true)
end

function turbineWarmup()
    setTurbineOnline()
    coilsOff()
end

function turbineRunning()
    setTurbineOnline()
    coilsOn()
end


function findOptFuelRods()
    m.clear()
    r.setActive(false)
    checkEn()
    displayEn()
    fuelRodLevel()
    while r.getFuelTemperature() > 99 do
        turbineWarmup()
        for i= 1,3 do
            checkEn()
            displayEn()
            fuelRodLevel()
            m.setCursorPos(3,6)
            m.write("Finding Optimal Rod Level")
            m.setCursorPos(3,7)
            m.write("Core Temp: "..r.getFuelTemperature())
            m.setCursorPos(3,8)
            m.write("Waiting for 99c")
            sleep(1)
        end
    end
    local reactorFluidCap = r.getCoolantAmountMax()
    while r.getHotFluidAmount() > reactorFluidCap do
        turbineWarmup()
        for i = 1,3 do
            checkEn()
            displayEn()
            fuelRodLevel()
            setVentRuning()
            m.setCursorPos(2,6)
            m.write("Finding Optimal Rod Level, please wait...")
            m.setCursorPos(2,7)
            m.write("Fluid Amount: "..comma_value(r.getHotFluidAmount()).."mb")
            m.setCursorPos(2,8)
            m.write("Waiting for fluid levels.")
            sleep(1)
        end
    end
    setVentRuning()
    r.setAllControlRodLevels(99)
    r.setActive(true)
    while r.getFuelTemperature() < 100 do
        for i = 1,5 do
            turbineWarmup()
            checkEn()
            displayEn()
            fuelRodLevel()
            setVentRuning()
            m.setCursorPos(3,6)
            m.write("Set all rod levels to 99")
            m.setCursorPos(3,7)
            m.write("Waiting 5 seconds...")
            sleep(1)
        end
    end
    for i = 1,5 do
        turbineWarmup()
        checkEn()
        displayEn()
        fuelRodLevel()
        setVentRuning()
        m.setCursorPos(3,6)
        m.write("Set all rod levels to 99")
        m.setCursorPos(3,7)
        m.write("Waiting 5 seconds...")
        sleep(1)
    end
    local tempMB = r.getEnergyProducedLastTick()
    local tempRodLevels = math.floor(steamMin/tempMB)
    tempRodLevels = 100-tempRodLevels+reactorOffset
    if tempRodLevels < 0 then
        if reactorOffset <0 then
            reactorOffset = 5
        end
        tempRodLevels = (0 + reactorOffset)
    end
    r.setAllControlRodLevels(math.floor(tempRodLevels))
    for i = 1,10 do
        turbineWarmup()
        checkEn()
        displayEn()
        fuelRodLevel()
        setVentRuning()
        m.setCursorPos(3,6)
        m.write("Estimated Level: "..tempRodLevels)
        m.setCursorPos(3,7)
        m.write("Waiting 10 seconds...")
        sleep(1)
    end
    tempMB = r.getEnergyProducedLastTick()
    while tempMB > steamMin do
        turbineWarmup()
        tempRodLevels = tempRodLevels+1
        r.setAllControlRodLevels(math.floor(tempRodLevels))
        for i = 1,5 do
            turbineWarmup()
            checkEn()
            displayEn()
            fuelRodLevel()
            setVentRuning()
            m.setCursorPos(3,6)
            m.write("Getting below "..steamMin.."mb/t")
            m.setCursorPos(3,7)
            m.write("Currently at: "..tempMB)
            sleep(1)
        end
        tempMB = r.getEnergyProducedLastTick()
    end
    while tempMB < steamMin do
        tempRodLevels = tempRodLevels -1
        r.setAllControlRodLevels(math.floor(tempRodLevels))
        for i = 1,5 do
            turbineWarmup()
            checkEn()
            displayEn()
            fuelRodLevel()
            setVentRuning()
            m.setCursorPos(3,6)
            m.write("Getting Above "..steamMin.."mb/t")
            m.setCursorPos(3,7)
            m.write("Currently at: "..tempMB)
            sleep(1)
        end
        tempMB = r.getEnergyProducedLastTick()
    end
    OptFuelRodLevel = tempRodLevels
end

function autoReactor()
    if not steamReactor then
        r.setAllControlRodLevels(0)
        if energyStoredPercent < turnOnAt then
            if not reactorOnline then
                online()
            end
        end
        if energyStoredPercent > turnOffAt then
            if reactorOnline then
                offline()
            end
        end
    else
        r.setAllControlRodLevels(OptFuelRodLevel)
        if energyStoredPercent < turnOnAt then
            turbineRunning()
        end
        if energyStoredPercent > turnOffAt then
            turbineWarmup()
        end
        if turbineRotorSpeed > targetSpeed then
            offline()
        else
            online()
        end
    end
end

function displayScreen()
    checkEn()
    displayEn()
    if menuType == "Reactor" then
        fuelRodLevel()
        if mode == "Automatic" then
            autoMenu()
            autoReactor()
        else
            manualMenu()
        end
    else
        turbineInductorDisplay()
        if mode == "Automatic" then
            turbineAutoMenu()
            autoReactor()
        else
            turbineManualMenu()
        end
    end
    timerCode = os.startTimer(1)
    local event, side, x, y
    repeat
    event, side, x, y = os.pullEvent()
    until event~= "timer" or timerCode == side
    if event == "monitor_touch" then
    button.checkxy(x,y)
    end
end

if steamReactor then
   findOptFuelRods()
end

while true do
   displayScreen()
end