buildvraft power

-- Load the components
local component = require("component")
local gpu = component.gpu
local sides = require("sides")
local computer = require("computer")

-- Load the energy cube
local energyCube = component.elite_energy_cube
local tankController = component.tank_controller

local engine_1_tank_bank_1 = "08a57843-b7b5-4a5f-be68-6693ede070ef"
local engine_2_tank_bank_1 = "a72b707d-9902-4b4e-b33d-c59c91057cd4"
local engine_3_tank_bank_1 = "9ad89be2-98f6-488f-b07c-e6438f3a639e"
local engine_4_tank_bank_1 = "bebbcce8-435f-448c-a6bd-5b5e8566c39c"

local engine_1_tank_bank_2 = "fa76b96e-5c83-4944-be81-4d2ceb3dbb61"
local engine_2_tank_bank_2 = "f04fbbe0-fd11-4748-9eaf-a95d7a1409ec"
local engine_3_tank_bank_2 = "fcd32370-6b57-4c50-b224-97afb2598de6"
local engine_4_tank_bank_2 = "14470fcc-cb6d-4acd-90c1-4975ed9ce48f"



-- Variables
local energy = 0
local maxEnergy = 0

local fluidLevel = 0
local maxFluidLevel = 0

-- Thresholds for bundled wire outputs
local offThreshold = 11000000 -- Energy threshold to turn the engine off
local onThreshold = 1800000 -- Energy threshold to turn the engine on
local totalDischarge = 0
local totalCharge = 0
local dischargeCount = 0
local chargeCount = 0

local fuelConsumptionSamples = {} -- Table to store fuel consumption samples
local maxFuelSamples = 10 -- Maximum number of fuel samples to keep
local fuelConsumptionRate = 0 -- Fuel consumption rate


-- Progress bar object
local progressBar = {
    x = 25,
    y = 6,
    width = 120,
    height = 5,
    progress = 0,
    maxProgress = 100,
    backgroundColor = 0x808080,
    filledColor = 0x00BFFF,
    shadowColor = 0x404040,
    highlightColor = 0xFFFFFF,
    orientation = "horizontal" -- Default orientation is horizontal
}

-- Function to load the progress bar
local function loadProgressBar()
    progressBar.x = 25
    progressBar.y = 6
    progressBar.width = 120
    progressBar.height = 5
    progressBar.maxProgress = maxEnergy
end

-- Function to update the progress bar
local function updateProgressBar()
    progressBar.progress = energy
end

-- Function to draw the progress bar
local function drawProgressBar()
    local filledWidth = math.floor((progressBar.progress / progressBar.maxProgress) * progressBar.width)

    -- Set color for filled part
    gpu.setBackground(progressBar.filledColor)
    gpu.fill(progressBar.x, progressBar.y, filledWidth, progressBar.height, ' ')

    -- Set color for unfilled part
    gpu.setBackground(progressBar.backgroundColor)
    gpu.fill(progressBar.x + filledWidth, progressBar.y, progressBar.width - filledWidth, progressBar.height, ' ')

    -- Draw threshold line
    local offThresholdX = math.floor((offThreshold / progressBar.maxProgress) * progressBar.width) + progressBar.x
    local onThresholdX = math.floor((onThreshold / progressBar.maxProgress) * progressBar.width) + progressBar.x
    gpu.setBackground(0xFF0000) -- Red color for off threshold
    gpu.fill(offThresholdX, progressBar.y + progressBar.height / 2, 1, 1, ' ')
    gpu.setBackground(0x00FF00) -- Green color for on threshold
    gpu.fill(onThresholdX, progressBar.y + progressBar.height / 2, 1, 1, ' ')

    -- Reset color to white
    gpu.setBackground(progressBar.highlightColor)
end

-- Fluid tank object
local fluidTank = {
    x = 120,
    y = 5,
    width = 15,
    height = 8,
    tankCapacity = 0,
    fluidAmount = 0,
    fluidName = "",
}

-- Function to load the fluid tank
local function loadFluidTank()
    fluidTank.tankCapacity = tankController.getTankCapacity(3)
    local fluidInTank = tankController.getFluidInTank(3)
    if fluidInTank[1] then
        fluidTank.fluidAmount = fluidInTank[1].amount
        fluidTank.fluidName = fluidInTank[1].name
    else
        fluidTank.fluidAmount = 0
        fluidTank.fluidName = "Empty"
    end
end

-- Function to update the fluid tank
local function updateFluidTank()
    fluidTank.fluidAmount = tankController.getTankLevel(3)
    local fluidInTank = tankController.getFluidInTank(3)
    if fluidInTank[1] then
        fluidTank.fluidName = fluidInTank[1].name
    else
        fluidTank.fluidName = "Empty"
    end
end

-- Function to draw the fluid tank
local function drawFluidTank()
    -- Calculate fluid level height
    local fluidLevelHeight = math.floor((fluidTank.fluidAmount / fluidTank.tankCapacity) * (fluidTank.height - 2))
    if fluidLevelHeight > fluidTank.height - 2 then
        fluidLevelHeight = fluidTank.height - 2
    end
    -- Draw fluid tank background
    gpu.setBackground(fluidTank.backgroundColor)
    gpu.fill(fluidTank.x, fluidTank.y, fluidTank.width, fluidTank.height, ' ')

    -- Draw fluid level if tank is not empty
    if fluidTank.fluidAmount > 1 then
        gpu.setBackground(fluidTank.filledColor)
        gpu.fill(fluidTank.x + 1, fluidTank.y + fluidTank.height - fluidLevelHeight - 1, fluidTank.width - 2, fluidLevelHeight, ' ')
        gpu.setBackground(fluidTank.highlightColor)

    end

    -- Draw unfilled part of the tank
    gpu.setBackground(fluidTank.highlightColor)
    gpu.fill(fluidTank.x + 1, fluidTank.y + 1, fluidTank.width - 2, fluidTank.height - fluidLevelHeight - 2, ' ')

    -- Draw fluid name and amount
    gpu.setForeground(fluidTank.highlightColor)
    gpu.set(fluidTank.x - 2, fluidTank.y - 1, "Fluid: " .. fluidTank.fluidName)
    gpu.set(fluidTank.x, fluidTank.y + fluidTank.height, "Amount: " .. fluidTank.fluidAmount .. " mB")
end

-- Engine object
local engine = {
    body = {x = 100, y = 11, width = 1, height = 3, color = 0xFFFFFF},
    head = {x = 100+1, y = 11, width = 1, height = 3, color = 0xC0C0C0}, -- Silver (0xC0C0C0) or White (0xFFFFFF)
    core = {x = 100+1, y = 11+1, width = 4, height = 1, color = 0x8B4513},
}

-- Function to load the engine
local function loadEngine()
    engine.core.x = engine.core.x
    engine.core.y = engine.core.y
    engine.body.x = engine.body.x
    engine.body.y = engine.body.y
    engine.head.x = engine.head.x
    engine.head.y = engine.head.y
end

-- Function to update the engine
local function updateEngine()
    if engine.movement == nil then
        engine.movement = 0
        engine.direction = true
    end

    if engine.direction then
        engine.head.x = engine.head.x + 1
    else
        engine.head.x = engine.head.x - 1
    end

    engine.movement = engine.movement + 1

    if engine.movement >= 3 then
        engine.movement = 0
        engine.direction = not engine.direction
        resetCycle() -- Reset the cycle when the engine completes a cycle
    end
end

-- Function to draw the engine
local function drawEngine()
    -- Draw core
    gpu.setBackground(engine.core.color)
    gpu.fill(engine.core.x, engine.core.y, engine.core.width, engine.core.height, ' ')
    -- Draw body
    gpu.setBackground(engine.body.color)
    gpu.fill(engine.body.x, engine.body.y, engine.body.width, engine.body.height, ' ')
    -- Draw head
    gpu.setBackground(engine.head.color)
    gpu.fill(engine.head.x, engine.head.y, engine.head.width, engine.head.height, ' ')
    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
end

-- Graph object
local graph = {
    x = progressBar.x,
    y = progressBar.y + progressBar.height + 10,
    width = progressBar.width,
    height = 10,
    data = {},
    maxDataPoints = 100,
}

-- Function to add data to the graph
local function addGraphData(value)
    table.insert(graph.data, 1, value)
    if #graph.data > graph.maxDataPoints then
        table.remove(graph.data)
    end
end

-- Function to draw the graph
local function drawGraph()
    -- Draw graph background
    gpu.setBackground(0x808080)
    gpu.fill(graph.x, graph.y, graph.width, graph.height, ' ')

    -- Calculate scale and offset for mapping data to graph coordinates
    local minData = math.min(table.unpack(graph.data))
    local maxData = math.max(table.unpack(graph.data))
    local scale = (maxData - minData) / graph.height
    local xOffset = graph.width / (#graph.data - 1)

    -- Draw data points
    gpu.setBackground(0x00BFFF)
    for i = 2, #graph.data do
        local x1 = graph.x + (i - 2) * xOffset
        local x2 = graph.x + (i - 1) * xOffset
        local y1 = graph.y + graph.height - math.floor((graph.data[i - 1] - minData) / scale)
        local y2 = graph.y + graph.height - math.floor((graph.data[i] - minData) / scale)
        gpu.fill(x1, y1, 1, 1, ' ')
        gpu.fill(x2, y2, 1, 1, ' ')
        gpu.fill(x1, y1, math.abs(x2 - x1) + 1, math.abs(y2 - y1) + 1, ' ')
    end

    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
end

-- Function to load the program components
local function load()
    maxEnergy = energyCube.getMaxEnergyStored()
    gpu.setResolution(160, 50) -- Set screen resolution
    loadProgressBar()
    loadEngine()
    loadFluidTank()
end

-- Function to update the program components
local function update()
    -- Check bundled inputs
    local orangeInput = component.redstone.getBundledInput(3, 1) or 0
    local whiteInput = component.redstone.getBundledInput(3, 0) or 0

    -- Check if animation should run
    local shouldAnimate = orangeInput > 1 or whiteInput > 1

    -- Update energy and progress bar
    energy = energyCube.getEnergyStored()
    updateProgressBar()

    -- Calculate discharge and charge rates
    local currentTick = computer.uptime()
    local lastSample = energySamples[#energySamples]
    if lastSample then
        local deltaTime = currentTick - lastSample.tick
        local deltaEnergy = energy - lastSample.energy

        if deltaTime > 0 then
            if deltaEnergy > 0 then
                chargeRate = deltaEnergy / deltaTime
                dischargeRate = 0
            elseif deltaEnergy < 0 then
                dischargeRate = -deltaEnergy / deltaTime
                chargeRate = 0
            end
        end
        if deltaEnergy > 0 then
            chargeRate = deltaEnergy / deltaTime
            totalCharge = totalCharge + chargeRate
            chargeCount = chargeCount + 1
            dischargeRate = 0
        elseif deltaEnergy < 0 then
            dischargeRate = -deltaEnergy / deltaTime
            totalDischarge = totalDischarge + dischargeRate
            dischargeCount = dischargeCount + 1
            chargeRate = 0
        end
    end

    -- Store energy sample
    table.insert(energySamples, 1, { tick = currentTick, energy = energy })
    if #energySamples > maxSamples then
        table.remove(energySamples)
    end

    if shouldAnimate then
        updateEngine()
    end

    -- Check thresholds and set bundled wire outputs
    if energy <= onThreshold then
        -- Set bundled wire outputs to white (bit 0) and orange (bit 1)
        component.redstone.setBundledOutput(3, 0, 255)
        component.redstone.setBundledOutput(3, 1, 255) -- Disable orange wire output
    elseif energy >= offThreshold then
        -- Set bundled wire outputs to white (bit 0) and orange (bit 1)
        component.redstone.setBundledOutput(3, 0, 0) -- Disable orange wire output
        component.redstone.setBundledOutput(3, 1, 0) -- Disable white wire output
    end

    updateFluidTank()
end

-- Function to draw the program components
local function draw()
    -- Clear screen
    gpu.setBackground(0x000000)
    gpu.setForeground(0xFFFFFF)
    gpu.fill(1, 1, 160, 50, ' ')
    drawProgressBar()
    gpu.setBackground(0x000000)
    gpu.setForeground(0xFFFFFF)
    gpu.set(progressBar.x, progressBar.y + progressBar.height + 1, "Current Energy: " .. tostring(energy))
    gpu.set(progressBar.x, progressBar.y + progressBar.height + 2, "Max Energy: " .. tostring(maxEnergy))
    drawEngine()

    -- Render discharge and charge rates
    local boxX = 25
    local boxY = progressBar.y + progressBar.height + 5
    local boxWidth = 48
    local boxHeight = 10
    local averageDischarge = totalDischarge / (dischargeCount > 0 and dischargeCount or 1)
    local averageCharge = totalCharge / (chargeCount > 0 and chargeCount or 1)

    -- Draw gray box
    gpu.setBackground(0x808080)
    gpu.fill(boxX, boxY, boxWidth, boxHeight, ' ')

    -- Render discharge rate
    local dischargeText = string.format("Discharge Rate: %.2f RF/t", dischargeRate)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 1, dischargeText)

    -- Render charge rate
    local chargeText = string.format("Charge Rate: %.2f RF/t", chargeRate)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 2, chargeText)

    -- Render average discharge rate
    local averageDischargeText = string.format("Average Discharge Rate: %.2f RF/t", averageDischarge)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 3, averageDischargeText)

    -- Render average charge rate
    local averageChargeText = string.format("Average Charge Rate: %.2f RF/t", averageCharge)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 4, averageChargeText)

    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
    -- Render fuel consumption rate
    local fuelConsumptionText = string.format("Fuel Consumption Rate: %.2f units/tick", fuelConsumptionRate)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 5, fuelConsumptionText)

    drawFluidTank()
end

-- Main function to run the program
local function main()
    load()
    while true do
        update()
        draw()
        os.sleep(0.1) -- Wait for 0.1 seconds before updating again
    end
end

-- Run the program
main()
-- Load the components
local component = require("component")
local gpu = component.gpu
local sides = require("sides")
local computer = require("computer")

-- Load the energy cube
local energyCube = component.elite_energy_cube
local tankController = component.tank_controller

-- Variables
local energy = 0
local maxEnergy = 0

local fluidLevel = 0
local maxFluidLevel = 0

-- Thresholds for bundled wire outputs
local offThreshold = 11000000 -- Energy threshold to turn the engine off
local onThreshold = 1800000 -- Energy threshold to turn the engine on
local totalDischarge = 0
local totalCharge = 0
local dischargeCount = 0
local chargeCount = 0




-- Progress bar object
local progressBar = {
    x = 25,
    y = 6,
    width = 120,
    height = 5,
    progress = 0,
    maxProgress = 100,
    backgroundColor = 0x808080,
    filledColor = 0x00BFFF,
    shadowColor = 0x404040,
    highlightColor = 0xFFFFFF,
    orientation = "horizontal" -- Default orientation is horizontal
}

function progressBar:load(x, y, width, height, maxProgress)
    self.x = x or self.x
    self.y = y or self.y
    self.width = width or self.width
    self.height = height or self.height
    self.maxProgress = maxProgress or self.maxProgress
    self.orientation = orientation or self.orientation
end

-- Progress bar update
function progressBar:update(progress)
    self.progress = progress
end

-- Progress bar draw
function progressBar:draw()
    local filledWidth = math.floor((self.progress / self.maxProgress) * self.width)

    -- Set color for filled part
    gpu.setBackground(0x00BFFF)
    gpu.fill(self.x, self.y, filledWidth, self.height, ' ')

    -- Set color for unfilled part
    gpu.setBackground(0x808080)
    gpu.fill(self.x + filledWidth, self.y, self.width - filledWidth, self.height, ' ')

    -- Draw threshold line
    local offThresholdX = math.floor((offThreshold / self.maxProgress) * self.width) + self.x
    local onThresholdX = math.floor((onThreshold / self.maxProgress) * self.width) + self.x
    gpu.setBackground(0xFF0000) -- Red color for off threshold
    gpu.fill(offThresholdX, self.y + self.height / 2, 1, 1, ' ')
    gpu.setBackground(0x00FF00) -- Green color for on threshold
    gpu.fill(onThresholdX, self.y + self.height / 2, 1, 1, ' ')

    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
end

-- Fluid tank object
local fluidTank = {
    x = 120,
    y = 5,
    width = 15,
    height = 8,
    tankCapacity = 0,
    fluidAmount = 0,
    fluidName = "",
}

-- Fluid tank load
function fluidTank:load()
    self.tankCapacity = tankController.getTankCapacity(3)
    local fluidInTank = tankController.getFluidInTank(3)
    if fluidInTank[1] then
        self.fluidAmount = fluidInTank[1].amount
        self.fluidName = fluidInTank[1].name
    else
        self.fluidAmount = 0
        self.fluidName = "Empty"
    end
end


-- Fluid tank update
function fluidTank:update()
    self.fluidAmount = tankController.getTankLevel(3)
    local fluidInTank = tankController.getFluidInTank(3)
    if fluidInTank[1] then
        self.fluidName = fluidInTank[1].name
    else
        self.fluidName = "Empty"
    end
end

-- Fluid tank draw
function fluidTank:draw()
    -- Calculate fluid level height
    local fluidLevelHeight = math.floor((self.fluidAmount / self.tankCapacity) * (self.height - 2))
    if fluidLevelHeight > self.height - 2 then
        fluidLevelHeight = self.height - 2
    end
    -- Draw fluid tank background
    gpu.setBackground(0x808080)
    gpu.fill(self.x, self.y, self.width, self.height, ' ')

    -- Draw fluid level if tank is not empty
    if self.fluidAmount > 1 then
        gpu.setBackground(0xFFFF00)
        gpu.fill(self.x + 1, self.y + self.height - fluidLevelHeight - 1, self.width - 2, fluidLevelHeight, ' ')
        gpu.setBackground(0x000000)

    end

    -- Draw unfilled part of the tank
    gpu.setBackground(0x000000)
    gpu.fill(self.x + 1, self.y + 1, self.width - 2, self.height - fluidLevelHeight - 2, ' ')

    -- Draw fluid name and amount

    gpu.setForeground(0xFFFFFF)
    gpu.set(self.x-2, self.y - 1, "Fluid: " .. self.fluidName)
    gpu.set(self.x, self.y + self.height, "Amount: " .. self.fluidAmount .. " mB")
end







-- Engine object
local engine = {
    body = {x = 100, y = 11, width = 1, height = 3, color = 0xFFFFFF},
    head = {x = 100+1, y = 11, width = 1, height = 3, color = 0xC0C0C0}, -- Silver (0xC0C0C0) or White (0xFFFFFF)
    core = {x = 100+1, y = 11+1, width = 4, height = 1, color = 0x8B4513},
}

-- Engine load
function engine:load()
    self.core.x = engine.core.x
    self.core.y = engine.core.y
    self.body.x = engine.body.x
    self.body.y = engine.body.y
    self.head.x = engine.head.x
    self.head.y = engine.head.y
end

function resetCycle()
    totalDischarge = 0
    totalCharge = 0
    dischargeCount = 0
    chargeCount = 0
end


-- Engine update
function engine:update()
    if self.movement == nil then
        self.movement = 0
        self.direction = true
    end

    if self.direction then
        self.head.x = self.head.x + 1
    else
        self.head.x = self.head.x - 1
    end

    self.movement = self.movement + 1

    if self.movement >= 3 then
        self.movement = 0
        self.direction = not self.direction
        resetCycle() -- Reset the cycle when the engine completes a cycle
    end
end

-- Engine draw
function engine:draw()
    -- Draw core
    gpu.setBackground(self.core.color)
    gpu.fill(self.core.x, self.core.y, self.core.width, self.core.height, ' ')
    -- Draw body
    gpu.setBackground(self.body.color)
    gpu.fill(self.body.x, self.body.y, self.body.width, self.body.height, ' ')
    -- Draw head
    gpu.setBackground(self.head.color)
    gpu.fill(self.head.x, self.head.y, self.head.width, self.head.height, ' ')
    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
end

local graph = {
    x = progressBar.x,
    y = progressBar.y + progressBar.height + 10,
    width = progressBar.width,
    height = 10,
    data = {},
    maxDataPoints = 100,
}

-- Graph add data
function graph:addData(value)
    table.insert(self.data, 1, value)
    if #self.data > self.maxDataPoints then
        table.remove(self.data)
    end
end

-- Graph draw
function graph:draw()
    -- Draw graph background
    gpu.setBackground(0x808080)
    gpu.fill(self.x, self.y, self.width, self.height, ' ')

    -- Calculate scale and offset for mapping data to graph coordinates
    local minData = math.min(table.unpack(self.data))
    local maxData = math.max(table.unpack(self.data))
    local scale = (maxData - minData) / self.height
    local xOffset = self.width / (#self.data - 1)

    -- Draw data points
    gpu.setBackground(0x00BFFF)
    for i = 2, #self.data do
        local x1 = self.x + (i - 2) * xOffset
        local x2 = self.x + (i - 1) * xOffset
        local y1 = self.y + self.height - math.floor((self.data[i - 1] - minData) / scale)
        local y2 = self.y + self.height - math.floor((self.data[i] - minData) / scale)
        gpu.fill(x1, y1, 1, 1, ' ')
        gpu.fill(x2, y2, 1, 1, ' ')
        gpu.fill(x1, y1, math.abs(x2 - x1) + 1, math.abs(y2 - y1) + 1, ' ')
    end

    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
end


function getAverageChargeRate()
    local total = 0
    for i = 1, #chargeRateSamples do
        total = total + chargeRateSamples[i]
    end
    return total / (#chargeRateSamples > 0 and #chargeRateSamples or 1)
end

function getAverageDischargeRate()
    local total = 0
    for i = 1, #dischargeRateSamples do
        total = total + dischargeRateSamples[i]
    end
    return total / (#dischargeRateSamples > 0 and #dischargeRateSamples or 1)
end


local function enableWireBank(bank)
    component.redstone.setBundledOutput(3, bank, 255)
end

-- Function to disable a specific bank of bundled wire outputs
local function disableWireBank(bank)
    component.redstone.setBundledOutput(3, bank, 0)
end

--[[
    ╔════════════════════════════╗
    ║ MAIN PROGRAM INITIALIZATION║
    ╚════════════════════════════╝
--]]
function load()
    maxEnergy = energyCube.getMaxEnergyStored()
    gpu.setResolution(160, 50) -- Set screen resolution
    progressBar:load(25, 6, 48, 5, maxEnergy) -- Initialize progress bar
    engine:load() -- Initialize engine
    fluidTank:load()
end

--[[
    ╔═══════════════════════╗
    ║ MAIN PROGRAM LOOP     ║
    ╚═══════════════════════╝
--]]
local energySamples = {} -- Table to store energy samples
local sampleRate = 10 -- Number of ticks between energy samples
local maxSamples = 10 -- Maximum number of energy samples to keep
local dischargeRate = 0 -- Energy discharge rate
local chargeRate = 0 -- Energy charge rate

local sampleSize = 10 -- Number of samples to use when calculating average rates
local chargeRateSamples = {} -- Table to store charge rate samples
local dischargeRateSamples = {} -- Table to store discharge rate samples

function update()
    -- Check bundled inputs
    local orangeInput = component.redstone.getBundledInput(3, 1) or 0
    local whiteInput = component.redstone.getBundledInput(3, 0) or 0

    -- Check if animation should run
    local shouldAnimate = orangeInput > 1 or whiteInput > 1

    -- Update energy and progress bar
    energy = energyCube.getEnergyStored()
    progressBar:update(energy)

    -- Calculate discharge and charge rates
    local currentTick = computer.uptime()
    local lastSample = energySamples[#energySamples]
    if lastSample then
        local deltaTime = currentTick - lastSample.tick
        local deltaEnergy = energy - lastSample.energy

        if deltaTime > 0 then
            if deltaEnergy > 0 then
                chargeRate = deltaEnergy / deltaTime
                dischargeRate = 0
            elseif deltaEnergy < 0 then
                dischargeRate = -deltaEnergy / deltaTime
                chargeRate = 0
            end
        end
        if deltaEnergy > 0 then
            chargeRate = deltaEnergy / deltaTime
            totalCharge = totalCharge + chargeRate
            chargeCount = chargeCount + 1
            dischargeRate = 0
        elseif deltaEnergy < 0 then
            dischargeRate = -deltaEnergy / deltaTime
            totalDischarge = totalDischarge + dischargeRate
            dischargeCount = dischargeCount + 1
            chargeRate = 0
        end
    end

    -- Store energy sample
    table.insert(energySamples, 1, { tick = currentTick, energy = energy })
    if #energySamples > maxSamples then
        table.remove(energySamples)
    end

    if shouldAnimate then
        engine:update()
    end

    -- Check thresholds and set bundled wire outputs
    if energy <= onThreshold then
        -- Set bundled wire outputs to white (bit 0) and orange (bit 1)
        enableWireBank(0)
        enableWireBank(1) -- Disable orange wire output
    elseif energy >= offThreshold  then
        -- Set bundled wire outputs to white (bit 0) and orange (bit 1)
        disableWireBank(0) -- Disable orange wire output
        disableWireBank(1) -- Disable white wire output

    end
    fluidTank:update()
    -- Render discharge and charge rates
    local boxX = 25
    local boxY = progressBar.y + progressBar.height + 5
    local boxWidth = 48
    local boxHeight = 10
    local averageDischarge = totalDischarge / (dischargeCount > 0 and dischargeCount or 1)
    local averageCharge = totalCharge / (chargeCount > 0 and chargeCount or 1)

    -- Draw gray box
    gpu.setBackground(0x808080)
    gpu.fill(boxX, boxY, boxWidth, boxHeight, ' ')

    -- Render discharge rate
    local dischargeText = string.format("Discharge Rate: %.2f RF/t", dischargeRate)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 1, dischargeText)

    -- Render charge rate
    local chargeText = string.format("Charge Rate: %.2f RF/t", chargeRate)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 2, chargeText)

    -- Render average discharge rate
    local averageDischargeText = string.format("Average Discharge Rate: %.2f RF/t", averageDischarge)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 3, averageDischargeText)

    -- Render average charge rate
    local averageChargeText = string.format("Average Charge Rate: %.2f RF/t", averageCharge)
    gpu.setBackground(0x808080)
    gpu.setForeground(0xFFFFFF)
    gpu.set(boxX + 2, boxY + 4, averageChargeText)

    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
    -- Calculate fuel consumption rate
    local currentFuel = fluidTank.getFluidAmount()
    local lastFuelSample = fuelConsumptionSamples[#fuelConsumptionSamples]
    if lastFuelSample then
        local deltaFuel = lastFuelSample.fuel - currentFuel
        if deltaFuel > 0 then
            fuelConsumptionRate = deltaFuel / deltaTime
        else
            fuelConsumptionRate = 0
        end
    end

-- Store fuel consumption sample
table.insert(fuelConsumptionSamples, 1, { tick = currentTick, fuel = currentFuel })
if #fuelConsumptionSamples > maxFuelSamples then
    table.remove(fuelConsumptionSamples)
end

    drawFluidTank()
end

-- Main function to run the program
local function main()
    load()
    while true do
        update()
        draw()
        os.sleep(0.1) -- Wait for 0.1 seconds before updating again
    end
end

-- Run the program
main()
-- Load the components
local component = require("component")
local gpu = component.gpu
local sides = require("sides")
local computer = require("computer")

-- Load the energy cube
local energyCube = component.elite_energy_cube
local tankController = component.proxy("a72b707d-9902-4b4e-b33d-c59c91057cd4")

-- Variables
local energy = 0
local maxEnergy = 0

local fluidLevel = 0
local maxFluidLevel = 0

-- Thresholds for bundled wire outputs
local offThreshold = 11000000 -- Energy threshold to turn the engine off
local onThreshold = 1800000 -- Energy threshold to turn the engine on
local totalDischarge = 0
local totalCharge = 0
local dischargeCount = 0
local chargeCount = 0




-- Progress bar object
local progressBar = {
    x = 25,
    y = 6,
    width = 120,
    height = 5,
    progress = 0,
    maxProgress = 100,
    backgroundColor = 0x808080,
    filledColor = 0x00BFFF,
    shadowColor = 0x404040,
    highlightColor = 0xFFFFFF,
    orientation = "horizontal" -- Default orientation is horizontal
}

function progressBar:load(x, y, width, height, maxProgress)
    self.x = x or self.x
    self.y = y or self.y
    self.width = width or self.width
    self.height = height or self.height
    self.maxProgress = maxProgress or self.maxProgress
    self.orientation = orientation or self.orientation
end

-- Progress bar update
function progressBar:update(progress)
    self.progress = progress
end

-- Progress bar draw
function progressBar:draw()
    local filledWidth = math.floor((self.progress / self.maxProgress) * self.width)

    -- Set color for filled part
    gpu.setBackground(0x00BFFF)
    gpu.fill(self.x, self.y, filledWidth, self.height, ' ')

    -- Set color for unfilled part
    gpu.setBackground(0x808080)
    gpu.fill(self.x + filledWidth, self.y, self.width - filledWidth, self.height, ' ')

    -- Draw threshold line
    local offThresholdX = math.floor((offThreshold / self.maxProgress) * self.width) + self.x
    local onThresholdX = math.floor((onThreshold / self.maxProgress) * self.width) + self.x
    gpu.setBackground(0xFF0000) -- Red color for off threshold
    gpu.fill(offThresholdX, self.y + self.height / 2, 1, 1, ' ')
    gpu.setBackground(0x00FF00) -- Green color for on threshold
    gpu.fill(onThresholdX, self.y + self.height / 2, 1, 1, ' ')

    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
end

-- Fluid tank object
local fluidTank = {
    x = 120,
    y = 5,
    width = 15,
    height = 8,
    tankCapacity = 0,
    fluidAmount = 0,
    fluidName = "",
}

-- Fluid tank load
function fluidTank:load()
    self.tankCapacity = tankController.getTankCapacity(3)
    local fluidInTank = tankController.getFluidInTank(3)
    if fluidInTank[1] then
        self.fluidAmount = fluidInTank[1].amount
        self.fluidName = fluidInTank[1].name
    else
        self.fluidAmount = 0
        self.fluidName = "Empty"
    end
end


-- Fluid tank update
function fluidTank:update()
    self.fluidAmount = tankController.getTankLevel(3)
    local fluidInTank = tankController.getFluidInTank(3)
    if fluidInTank[1] then
        self.fluidName = fluidInTank[1].name
    else
        self.fluidName = "Empty"
    end
end

-- Fluid tank draw
function fluidTank:draw()
    -- Calculate fluid level height
    local fluidLevelHeight = math.floor((self.fluidAmount / self.tankCapacity) * (self.height - 2))
    if fluidLevelHeight > self.height - 2 then
        fluidLevelHeight = self.height - 2
    end
    -- Draw fluid tank background
    gpu.setBackground(0x808080)
    gpu.fill(self.x, self.y, self.width, self.height, ' ')

    -- Draw fluid level if tank is not empty
    if self.fluidAmount > 1 then
        gpu.setBackground(0xFFFF00)
        gpu.fill(self.x + 1, self.y + self.height - fluidLevelHeight - 1, self.width - 2, fluidLevelHeight, ' ')
        gpu.setBackground(0x000000)

    end

    -- Draw unfilled part of the tank
    gpu.setBackground(0x000000)
    gpu.fill(self.x + 1, self.y + 1, self.width - 2, self.height - fluidLevelHeight - 2, ' ')

    -- Draw fluid name and amount

    gpu.setForeground(0xFFFFFF)
    gpu.set(self.x-2, self.y - 1, "Fluid: " .. self.fluidName)
    gpu.set(self.x, self.y + self.height, "Amount: " .. self.fluidAmount .. " mB")
end







-- Engine object
local engine = {
    body = {x = 100, y = 11, width = 1, height = 3, color = 0xFFFFFF},
    head = {x = 100+1, y = 11, width = 1, height = 3, color = 0xC0C0C0}, -- Silver (0xC0C0C0) or White (0xFFFFFF)
    core = {x = 100+1, y = 11+1, width = 4, height = 1, color = 0x8B4513},
}

-- Engine load
function engine:load()
    self.core.x = engine.core.x
    self.core.y = engine.core.y
    self.body.x = engine.body.x
    self.body.y = engine.body.y
    self.head.x = engine.head.x
    self.head.y = engine.head.y
end

function resetCycle()
    totalDischarge = 0
    totalCharge = 0
    dischargeCount = 0
    chargeCount = 0
end


-- Engine update
function engine:update()
    if self.movement == nil then
        self.movement = 0
        self.direction = true
    end

    if self.direction then
        self.head.x = self.head.x + 1
    else
        self.head.x = self.head.x - 1
    end

    self.movement = self.movement + 1

    if self.movement >= 3 then
        self.movement = 0
        self.direction = not self.direction
        resetCycle() -- Reset the cycle when the engine completes a cycle
    end
end

-- Engine draw
function engine:draw()
    -- Draw core
    gpu.setBackground(self.core.color)
    gpu.fill(self.core.x, self.core.y, self.core.width, self.core.height, ' ')
    -- Draw body
    gpu.setBackground(self.body.color)
    gpu.fill(self.body.x, self.body.y, self.body.width, self.body.height, ' ')
    -- Draw head
    gpu.setBackground(self.head.color)
    gpu.fill(self.head.x, self.head.y, self.head.width, self.head.height, ' ')
    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
end

local graph = {
    x = progressBar.x,
    y = progressBar.y + progressBar.height + 10,
    width = progressBar.width,
    height = 10,
    data = {},
    maxDataPoints = 100,
}

-- Graph add data
function graph:addData(value)
    table.insert(self.data, 1, value)
    if #self.data > self.maxDataPoints then
        table.remove(self.data)
    end
end

-- Graph draw
function graph:draw()
    -- Draw graph background
    gpu.setBackground(0x808080)
    gpu.fill(self.x, self.y, self.width, self.height, ' ')

    -- Calculate scale and offset for mapping data to graph coordinates
    local minData = math.min(table.unpack(self.data))
    local maxData = math.max(table.unpack(self.data))
    local scale = (maxData - minData) / self.height
    local xOffset = self.width / (#self.data - 1)

    -- Draw data points
    gpu.setBackground(0x00BFFF)
    for i = 2, #self.data do
        local x1 = self.x + (i - 2) * xOffset
        local x2 = self.x + (i - 1) * xOffset
        local y1 = self.y + self.height - math.floor((self.data[i - 1] - minData) / scale)
        local y2 = self.y + self.height - math.floor((self.data[i] - minData) / scale)
        gpu.fill(x1, y1, 1, 1, ' ')
        gpu.fill(x2, y2, 1, 1, ' ')
        gpu.fill(x1, y1, math.abs(x2 - x1) + 1, math.abs(y2 - y1) + 1, ' ')
    end

    -- Reset color to white
    gpu.setBackground(0xFFFFFF)
end


function getAverageChargeRate()
    local total = 0
    for i = 1, #chargeRateSamples do
        total = total + chargeRateSamples[i]
    end
    return total / (#chargeRateSamples > 0 and #chargeRateSamples or 1)
end

function getAverageDischargeRate()
    local total = 0
    for i = 1, #dischargeRateSamples do
        total = total + dischargeRateSamples[i]
    end
    return total / (#dischargeRateSamples > 0 and #dischargeRateSamples or 1)
end


local function enableWireBank(bank)
    component.redstone.setBundledOutput(3, bank, 255)
end

-- Function to disable a specific bank of bundled wire outputs
local function disableWireBank(bank)
    component.redstone.setBundledOutput(3, bank, 0)
end

--[[
    ╔════════════════════════════╗
    ║ MAIN PROGRAM INITIALIZATION║
    ╚════════════════════════════╝
--]]
function load()
    maxEnergy = energyCube.getMaxEnergyStored()
    gpu.setResolution(160, 50) -- Set screen resolution
    progressBar:load(25, 6, 48, 5, maxEnergy) -- Initialize progress bar
    engine:load() -- Initialize engine
    fluidTank:load()
end

--[[
    ╔═══════════════════════╗
    ║ MAIN PROGRAM LOOP     ║
    ╚═══════════════════════╝
--]]
local energySamples = {} -- Table to store energy samples
local sampleRate = 10 -- Number of ticks between energy samples
local maxSamples = 10 -- Maximum number of energy samples to keep
local dischargeRate = 0 -- Energy discharge rate
local chargeRate = 0 -- Energy charge rate

local sampleSize = 10 -- Number of samples to use when calculating average rates
local chargeRateSamples = {} -- Table to store charge rate samples
local dischargeRateSamples = {} -- Table to store discharge rate samples

function update()
    -- Check bundled inputs
    local orangeInput = component.redstone.getBundledInput(3, 1) or 0
    local whiteInput = component.redstone.getBundledInput(3, 0) or 0

    -- Check if animation should run
    local shouldAnimate = orangeInput > 1 or whiteInput > 1

    -- Update energy and progress bar
    energy = energyCube.getEnergyStored()
    progressBar:update(energy)

    -- Calculate discharge and charge rates
    local currentTick = computer.uptime()
    local lastSample = energySamples[#energySamples]
    if lastSample then
        local deltaTime = currentTick - lastSample.tick
        local deltaEnergy = energy - lastSample.energy

        if deltaTime > 0 then
            if deltaEnergy > 0 then
                chargeRate = deltaEnergy / deltaTime
                dischargeRate = 0
            elseif deltaEnergy < 0 then
                dischargeRate = -deltaEnergy / deltaTime
                chargeRate = 0
            end
        end
        if deltaEnergy > 0 then
            chargeRate = deltaEnergy / deltaTime
            totalCharge = totalCharge + chargeRate
            chargeCount = chargeCount + 1
            dischargeRate = 0
        elseif deltaEnergy < 0 then
            dischargeRate = -deltaEnergy / deltaTime
            totalDischarge = totalDischarge + dischargeRate
            dischargeCount = dischargeCount + 1
            chargeRate = 0
        end
    end

    -- Store energy sample
    table.insert(energySamples, 1, { tick = currentTick, energy = energy })
    if #energySamples > maxSamples then
        table.remove(energySamples)
    end

    if shouldAnimate then
        engine:update()
    end

    -- Check thresholds and set bundled wire outputs
    if energy <= onThreshold then
        -- Set bundled wire outputs to white (bit 0) and orange (bit 1)
        enableWireBank(0)
        enableWireBank(1) -- Disable orange wire output
    elseif energy >= offThreshold  then
        -- Set bundled wire outputs to white (bit 0) and orange (bit 1)
        disableWireBank(0) -- Disable orange wire output
        disableWireBank(1) -- Disable white wire output

    end
    fluidTank:update()

end