ReactorControl2

-- Get args
local arg1, arg2, arg3 = ...
-- Startup
print('ReactorControl Engaged. View Monitor.')
local w, h = term.getSize()
-- Vars
emptyflag = 0
offlineflag = 0
flashflag = 0

reactorNameInput = "BigReactors-Reactor_" .. arg1
reactor = peripheral.wrap(reactorNameInput)
if reactor == nil then
   print("Reactor null")
end

monitorSideInput = arg2
monitor = peripheral.wrap(monitorSideInput)
if monitor == nil then
   print("Monitor null")
end

controlRodNumber = arg3
if controlRodNumber == nil then
   print("controlRodNumber null")
end

-- Monitor Settings
monitor.setTextScale(1)
monitor.setBackgroundColor(colors.black)

if fs.exists("ProgressBar") then
    print("You have the ProgressBar API")
 else
    print("Downloading API")
    shell.run("pastebin get Rzxj0KKM ProgressBar")
    print("Restarting Program")
    restart()
end

--Load ProgressBar API
os.loadAPI('ProgressBar')
ProgressBar.SetPeripheral(monitorSideInput)

function MakeFuelInfo ()
    fuelMax = reactor.getFuelAmountMax()
    fuelAmount = reactor.getFuelAmount()
    fuelAmountText = math.floor(((reactor.getFuelAmount()/reactor.getFuelAmountMax())*100)+0.5)
    ProgressBar.SetTable("Fuel", fuelMax, fuelAmount, 7, w - 7,2)
    monitor.setCursorPos(w - 5, 2)
    monitor.setTextColor(colors.yellow)
    monitor.write(fuelAmountText..'%')
end

function MakeWasteInfo()
    wasteMax = reactor.getFuelAmountMax()
    wasteAmount = reactor.getWasteAmount()
    wasteAmountText = math.floor(((reactor.getWasteAmount()/reactor.getFuelAmountMax())*100)+0.5)
    ProgressBar.SetTable("Waste", wasteMax, wasteAmount, 7, w - 7, 3)
    monitor.setCursorPos(w - 5, 3)
    monitor.setTextColor(colors.cyan)
    monitor.write(wasteAmountText..'%')
end

function ControlRods(numberOfRods)
local cursorPos = 6
local rodLevelAvg = 0
    --Run for each rod
        for i=1, numberOfRods do
            rodLevelAvg = rodLevelAvg + (reactor.getControlRodLevel(0))
        end
        rodLevelAvg = rodLevelAvg / numberOfRods

        monitor.setTextColor(colors.green)
        monitor.setCursorPos(1,cursorPos)
        monitor.write('Rod Depths: ')
        ProgressBar.SetTable("RODS", 100, rodLevelAvg, 12, w - 7, 6)
        monitor.setCursorPos(w - 5,6)
        monitor.write((100-(reactor.getControlRodLevel(0)))..'%')
end

function GetCasingHeatinfo()

        if reactor.getCasingTemperature()>=650 then
            monitor.setTextColor(colors.purple)
            else if reactor.getCasingTemperature()>=950 then
                monitor.setTextColor(colors.red)
            else
            monitor.setTextColor(colors.green)
            end
        end
        casingTemp = reactor.getCasingTemperature()
        ProgressBar.SetTable("CasingTemp", 10000, casingTemp, 9, w - 7, 9)
        monitor.setCursorPos(w - 5, 9)
        monitor.write(casingTemp..' C')
end

 function GetFuelHeatInfo()

        if reactor.getFuelTemperature()>=650 then
          monitor.setTextColor(colors.purple)
          else if reactor.getFuelTemperature()>=950 then
            monitor.setTextColor(colors.red)
          else
         monitor.setTextColor(colors.green)
          end
        end
        fuelTemp = reactor.getFuelTemperature()
        ProgressBar.SetTable("FuelTemp", 10000, fuelTemp, 9, w - 7, 10)
        monitor.setCursorPos(w - 5,10)
        monitor.write(fuelTemp..' C')
 end

 function GetFluxInfo()
        -- RF Stored
        rfStored = reactor.getEnergyStored()
        rfStoredFormat = rfStored / 1000
        ProgressBar.SetTable("RFStored", 10000000, rfStored, 8, w-10,13)
        monitor.setCursorPos(w-7,13)
        monitor.write(rfStoredFormat..'K Stored')
        --------------------------------------------------
        -- RF Per Tick
        monitor.setCursorPos(1,14)
        monitor.setTextColor(colors.orange)
        rfPerTick = reactor.getEnergyProducedLastTick()
        --ProgressBar.SetTable("RFPerTick", 10000000, rfPerTick, 2,w-11,15)

        monitor.setCursorPos(1,14)
        monitor.write('RF/t: '..(math.floor(reactor.getEnergyProducedLastTick()+0.5)))
        --------------------------------------------------

 end

-- Main Function
function main()
    while true do
    -- Initial Setup
        monitor.clear()
        monitor.setCursorPos(1,1)
        monitor.setTextColor(colors.white)
    -----------------------------------------------

    -- Make Fuel info
        --Set POS, Color, Title
        monitor.setCursorPos(1,2)
        monitor.setTextColor(colors.yellow)
        monitor.write('Fuel ')
        MakeFuelInfo()
    -----------------------------------------------

    -- Waste Bar
        --Set POS, Color, Title
        monitor.setCursorPos(1,3)
        monitor.setTextColor(colors.lightBlue)
        monitor.write('Waste ')
        MakeWasteInfo()
    -----------------------------------------------

    -- Control Rod
        --Set POS, Color, Title
        monitor.setCursorPos(1,5)
        monitor.setTextColor(colors.white)
        monitor.write('Control Rod Levels:')
        ControlRods(controlRodNumber)
    -----------------------------------------------

    -- Temps and Heat
        monitor.setCursorPos(1,8)
        monitor.setTextColor(colors.white)
        monitor.write('Temperature:')

        monitor.setCursorPos(1,9)
        monitor.setTextColor(colors.lightGray)
        monitor.write('Casing: ')
        GetCasingHeatinfo()
        --------------------------------------------------
        monitor.setCursorPos(1,10)
        monitor.setTextColor(colors.yellow)
        monitor.write('Fuel: ')
        GetFuelHeatInfo()
    --------------------------------------------------

    -- Flux Information
        monitor.setCursorPos(1,12)
        monitor.setTextColor(colors.white)
        monitor.write('Flux:')

        monitor.setCursorPos(1,13)
        monitor.setTextColor(colors.yellow)
        monitor.write('Stored: ')


        monitor.setCursorPos(1,13)
        monitor.setTextColor(colors.green)
        GetFluxInfo()
    --------------------------------------------------

    ProgressBar.DrawToPeripheral()
        monitor.setCursorPos(1,18)

        if flashflag==0 then
          flashflag=1
          if offlineflag==1 then
            monitor.setCursorPos(1,17)
            monitor.setTextColor(colors.lightGray)
            monitor.write('OFFLINE - Manual Override')
          end
          if emptyflag==1 then
            monitor.setCursorPos(1,17)
            monitor.setTextColor(colors.pink)
            monitor.write('OFFLINE - Fuel Exhausted')
          end
          if emptyflag==0 and offlineflag==0 and reactor.getControlRodLevel(0)>75 then
            monitor.setCursorPos(1,17)
            monitor.setTextColor(colors.yellow)
            monitor.write('ONLINE - Low Power Mode')
          end
          if emptyflag==0 and offlineflag==0 and reactor.getControlRodLevel(0)<=75 then
            monitor.setCursorPos(1,17)
            monitor.setTextColor(colors.orange)
            monitor.write('ONLINE - High Power Mode')
          end
        else
          flashflag=0
          monitor.setCursorPos(1,17)
          monitor.clearLine()
        end

        if reactor.getEnergyStored()<=10000000 and reactor.getEnergyStored()>100 then
            reactor.setAllControlRodLevels(0+(math.floor(reactor.getEnergyStored()/100000)))
        else
            reactor.setAllControlRodLevels(0)
        end

        if reactor.getFuelAmount()<=100 and offlineflag==0 then
            reactor.setAllControlRodLevels(100)
            reactor.setActive(false)
            emptyflag=1
        else
            emptyflag=0
        end

        if rs.getInput('bottom')==false and emptyflag==0 then
            reactor.setActive(true)
            offlineflag=0
        end

        if rs.getInput('bottom')==true and emptyflag==0 then
            reactor.setActive(false)
            reactor.setAllControlRodLevels(100)
            offlineflag=1
        end
    sleep(1)
    end
end

function restart()
    main()
end

-- Reactor Connection Checl
if (reactor.getConnected) then
    main()
end