CC - BigReactor Control for Multiturbine (startup)

os.loadAPI("button")

p = peripheral.find("tile_blockcapacitorbank_name")
m = peripheral.find("monitor")
r = peripheral.find("BigReactors-Reactor")
redstoneOutput = "left"

turbines = {}
turbines[0] = peripheral.wrap("BigReactors-Turbine_0")
turbines[1] = peripheral.wrap("BigReactors-Turbine_2")
turbines[2] = peripheral.wrap("BigReactors-Turbine_3")
turbines[3] = peripheral.wrap("BigReactors-Turbine_4")

turbinesColor = {}
turbinesColor[0] = colors.blue
turbinesColor[1] = colors.yellow
turbinesColor[2] = colors.black
turbinesColor[3] = colors.white

local numTurbines = 4
local numCapacitors = 16

turbinesOnline = {}
turbinesRotorSpeed = {}
turbinesRFGen = {}
turbinesFluidRate = {}
turbinesInductor = {}
optRodLevels = {}
turbinesActiveFlow = {}

local steamReactor = r.isActivelyCooled()
local menuType = "Reactor"
local turbineIndex = 0

local turnOnAt = 50
local turnOffAt = 90

local turbinesRequireSteam = 0

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 turbineRFGen = 0

local currDir = shell.dir()

function loadConfig()
    if not fs.exists(currDir.."config.txt") or fs.isDir(currDir.."config.txt") then
        return false
    else
        local configFile = fs.open(currDir.."config.txt","r")
        for i = 0,numTurbines-1,1 do
            optRodLevels[i] = tonumber(configFile.readLine())
        end
        configFile.close()
    end
    return true
end

function saveConfig()
    local configFile = fs.open(currDir.."config.txt","w")
    --configFile.writeLine("#ReactorRodLevels:")
    for i = 0,numTurbines-1,1 do
        configFile.writeLine(optRodLevels[i])
    end
    configFile.close()
    return true
end

function initalizeAll()
    for i = 0,numTurbines-1,1 do
        turbinesOnline[i] = false
        turbinesRotorSpeed[i] = 0
        turbinesRFGen[i] = 0
        turbinesFluidRate[i] = 0
        turbinesInductor[i] = 0
        optRodLevels[i] = 0
        turbinesActiveFlow[i] = true
    end
end

function setFluid()
    local output = 0
    for i = 0,numTurbines-1,1 do
        if not turbinesActiveFlow[i] then
            --print("FluidOff: "..i)
            if output == 0 then
              output = turbinesColor[i]
            else
                output = colors.combine(output, turbinesColor[i])
                --print("Combine: "..output.." + "..turbinesColor[i].." = "..output)
            end
        end
    end
    --print("output: "..output)
    redstone.setBundledOutput(redstoneOutput, output)
end

function setFluidOn(index)
    turbinesActiveFlow[index] = true
    setFluid()
end

function setFluidOff(index)
    turbinesActiveFlow[index] = false
    setFluid()
end

function fluidOnOff(index)
    button.setButton("Fluid On", turbinesActiveFlow[index])
    button.setButton("Fluid Off", not turbinesActiveFlow[index])
end

function addSteamReactorButtons()
   button.setTable("Reactor", reactorMenu, "", 2, 15, 19, 19)
   local line = 0
   local n = 0
    for i = 0,numTurbines-1,1 do
        if math.fmod(i,2) == 0 then
            line = line + 2
            n = 0
        end
    button.setTable("Turbine["..i.."]", turbineMenu, i, 47+(n*13), 57+(n*13), line, line)
    n = n + 1
    end
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
      addSteamReactorButtons()
   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)
   if menuType == "Reactor" then
       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)
   else
       turbineManualMenu(turbineIndex)
   end

   if steamReactor then
      addSteamReactorButtons()
   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)
   addSteamReactorButtons()
   button.screen()
   checkMode()
   menuMode()
end

function turbineManualMenu(index)
   m.setTextScale(1)
   button.clearTable()
   button.setTable("Automatic", autoMode, "", 3, 13, 6, 6)
   button.setTable("Manual", manualMode, "", 15, 25, 6, 6)
   addSteamReactorButtons()
   button.setTable("Online", setTurbineOnline, index, 3, 13, 8, 8)
   button.setTable("Offline", setTurbineOffline, index, 15, 25, 8, 8)
   button.setTable("Coils On", coilsOn, index, 3, 13, 10, 10)
   button.setTable("Coils Off", coilsOff, index, 15, 25, 10, 10)
   button.setTable("Fluid On", setFluidOn, index, 3, 13, 12, 12)
   button.setTable("Fluid Off", setFluidOff, index, 15, 25, 12, 12)
   button.screen()
   checkMode()
   turbineOnOff(index)
   coilsOnOff(index)
   fluidOnOff(index)
   menuMode()
end

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

function turbineMenu(index)
   menuType = "Turbine["..index.."]"
   turbineIndex = index
   displayScreen()
end

function online()
   r.setActive(true)
   --button.flash("Online")
end

function offline()
   r.setActive(false)
   --button.flash("Offline")
end

function setTurbineOnline( index)
   turbines[index].setActive(true)
   --button.flash("Online")
end

function setTurbineOffline( index)
   turbines[index].setActive(false)
   --button.flash("Offline")
end

function setAllTurbinesOnline()
   for i = 0, numTurbines-1,1 do
       turbines[i].setActive(true)
   end
end

function setAllTurbinesOffline()
   for i = 0, numTurbines-1,1 do
       turbines[i].setActive(false)
   end
end

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

function turbineOnOff( index)
   button.setButton("Online", turbines[index].getActive())
   button.setButton("Offline", not turbines[index].getActive())
end

function coilsOnOff( index)
   button.setButton("Coils On", turbines[index].getInductorEngaged())
   button.setButton("Coils Off", not turbines[index].getInductorEngaged())
end

function coilsOn( index)
   turbines[index].setInductorEngaged(true)
end

function coilsOff( index)
   turbines[index].setInductorEngaged(false)
end

function setAllCoilsOn()
    for i = 0, numTurbines-1,1 do
       turbines[i].setInductorEngaged(true)
    end
end

function setAllCoilsOff()
    for i = 0,numTurbines-1,1 do
       turbines[i].setInductorEngaged(false)
    end
end

function menuMode()
   if steamReactor then
   if menuType == "Reactor" then
     button.setButton("Reactor", true)
     for  i = 0,numTurbines-1,1 do
         button.setButton("Turbine["..i.."]",false)
     end
   else
     button.setButton("Reactor", false)
     for  i = 0, numTurbines-1, 1 do
         if menuType == "Turbine["..i.."]" then
             button.setButton("Turbine["..i.."]",true)
         else
             button.setButton("Turbine["..i.."]",false)
         end
     end
   end
   end
end

function setRods(setLevel)
   print("Setting Rod Level: "..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)
   --print("Energy Use: "..energy)
   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(28,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(28,2)
      m.write("Core Temp: "..math.floor(coreTemp).."c")
      m.setCursorPos(28,3)
      m.write("Fuel Use: "..fuelUse.."MB/t")
  else
      local currTurbine = menuType
      m.setCursorPos(1,3)
      m.write(currTurbine.." is: ")
      if turbinesOnline[turbineIndex] 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(28,1)
      m.write("RFGen: ")
      m.setTextColor(colors.green)
      m.write(comma_value(math.floor(turbinesRFGen[turbineIndex])).."RF/t")
      m.setTextColor(colors.white)
      m.setCursorPos(28,2)
      m.write("Rotor: "..comma_value(math.floor(turbinesRotorSpeed[turbineIndex])).." RPM")
      m.setCursorPos(28,3)
      m.write("Steam: "..comma_value(turbinesFluidRate[turbineIndex]).."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
     turbineRFGen = 0
     for  i = 0,numTurbines-1,1 do
         turbinesOnline[i] = turbines[i].getActive()
         turbinesRotorSpeed[i] = turbines[i].getRotorSpeed()
         turbinesRFGen[i] = turbines[i].getEnergyProducedLastTick()
         turbinesFluidRate[i] = turbines[i].getFluidFlowRate()
         turbineRFGen = turbineRFGen + turbinesRFGen[i]
     end
   end
end

function fuelRodLevel()
   rodLevel = r.getControlRodLevel(0)
   --print(rodLevel)
   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(index)
   turbineInductor = turbines[index].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 findOptFuelRods()
   m.clear()
   r.setActive(false)
   checkEn()
   displayEn()
   fuelRodLevel()
   while r.getFuelTemperature() > 99 do
      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
   while r.getHotFluidAmount() > 10000 do
     for i = 1,3 do
        checkEn()
        displayEn()
        fuelRodLevel()
        m.setCursorPos(3,6)
        m.write("Finding Optimal Rod Level, please wait....")
        m.setCursorPos(3,7)
        m.write("Fluid Amount: "..comma_value(r.getHotFluidAmount()).."mb")
        m.setCursorPos(3,8)
        m.write("Waiting for 10,000mb")
        sleep(1)
      end
   end
   r.setAllControlRodLevels(99)
   r.setActive(true)

   while r.getFuelTemperature() < 100 do
   for i = 1,5 do
     checkEn()
     displayEn()
     fuelRodLevel()
     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
     checkEn()
     displayEn()
     fuelRodLevel()
     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 tempMBlow = r.getEnergyProducedLastTick()
   print(tempMBlow.."MB/t of steam")
   for n = 1, numTurbines,1 do
       local tempRodLevels = math.floor((2000*n)/tempMBlow)
       print((2000*n).."/"..tempMBlow.." = "..tempRodLevels)
       tempRodLevels = 100-tempRodLevels+5
       print("Adding 5 to Rod Levels: "..tempRodLevels)
       r.setAllControlRodLevels(math.floor(tempRodLevels))
       print("Waiting 10 seconds to confirm...")
       for i = 1,10 do
         checkEn()
         displayEn()
         fuelRodLevel()
         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 > (2000*n) do
          tempRodLevels = tempRodLevels+1
          r.setAllControlRodLevels(math.floor(tempRodLevels))
          print("Setting Rod Levels to: "..tempRodLevels)
          for i = 1,5 do
           checkEn()
           displayEn()
           fuelRodLevel()
           m.setCursorPos(3,6)
           m.write("Getting below "..(2000*n).."mb/t")
           m.setCursorPos(3,7)
           m.write("Currently at: "..tempMB)
           sleep(1)
          end
          tempMB = r.getEnergyProducedLastTick()
       end
       while tempMB < (2000*n) do
          tempRodLevels = tempRodLevels -1
          r.setAllControlRodLevels(math.floor(tempRodLevels))
          print("Setting Rod Levels to: "..tempRodLevels)
          for i = 1,5 do
           checkEn()
           displayEn()
           fuelRodLevel()
           m.setCursorPos(3,6)
           m.write("Getting Above "..(2000*n).."mb/t")
           m.setCursorPos(3,7)
           m.write("Currently at: "..tempMB)
           sleep(1)
          end
          tempMB = r.getEnergyProducedLastTick()
       end
       optRodLevels[n-1] = tempRodLevels
   end
   r.setActive(false)
end

function adjustRods()
    print("NumTurbinesRequireSteam: "..turbinesRequireSteam)
    if turbinesRequireSteam <= 0 then
        r.setActive(false)
        turbinesRequireSteam = 0
    else
        r.setActive(true)
        r.setAllControlRodLevels(optRodLevels[turbinesRequireSteam-1])
    end
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
       if energyStoredPercent < turnOnAt then
         --online()
             setAllTurbinesOnline()
             setAllCoilsOn()
       end
       if energyStoredPercent > turnOffAt then
         --if turbineRotorSpeed > 1800 then
         --   offline()
         --else
            --   online()
         --end
         setAllTurbinesOnline()
         setAllCoilsOff()
       end
       turbinesRequireSteam = 0
    for  i = 0,numTurbines-1,1 do
        if turbines[i].getRotorSpeed() > 1800 then
            turbinesActiveFlow[i] = false
        else
            turbinesRequireSteam = turbinesRequireSteam + 1
            turbinesActiveFlow[i] = true
        end
    end
        setFluid()
        adjustRods()
    end
end

function displayScreen()
 --  repeat
      checkEn()
      displayEn()
      if menuType == "Reactor" then
        fuelRodLevel()
        if mode == "Automatic" then
           autoMenu()
           autoReactor()
        else
           manualMenu()
        end
      else
        turbineInductorDisplay(turbineIndex)
        if mode == "Automatic" then
           turbineAutoMenu()
           autoReactor()
        else
           turbineManualMenu(turbineIndex)
        end
      end

      timerCode = os.startTimer(1)
      local event, side, x, y
      repeat
        event, side, x, y = os.pullEvent()
        --print(event)
        if event == "timer" then
           --print(timerCode..":"..side)
           if timerCode ~= side then
              --print("Wrong Code")
            else
              --print("Right Code")
            end
        end
       until event~= "timer" or timerCode == side
       if event == "monitor_touch" then
            --print(x..":"..y)
            button.checkxy(x,y)
        end
 --  until event ~= "timer"
end

initalizeAll()

if steamReactor then
    if not loadConfig() then
        findOptFuelRods()
        saveConfig()
    end
end

while true do
   displayScreen()
end