finite_fuel_client.lua

-- math.round function
function math.round(number, decimals)
    local multiply = 10 ^ (decimals or 0)
    return math.floor(number * multiply + 0.5) / multiply
end

fill = Image.Create(AssetLocation.Game, "hud_heat_meter_dif.dds")
background = Image.Create(AssetLocation.Game, "hud_health_max_dif.dds")
fill:SetSize(Vector2(16, 512))



transform = Transform2()

class "FiniteFuel"

function FiniteFuel:__init()
    -- Configurable
    self.gasStationMinimapIcon = true -- Square on the minimap showing closest fuel station
    self.gasStationMinimapColor = Color(0, 255, 0) -- The color of the minimap square

    self.gasStationMarker = true -- 3D markers that show up at the gas station itself
    self.gasStationMarkerVisibleRadius = 100 -- The radius in which the 3D marker will be visible
    self.gasStationMarkerColor = Color(0, 255, 0, 100) -- The color of the 3D marker

    self.gasStationRefuelRadius = 10 -- The radius from the center of the gas station that you can refuel within
    self.gasStationRefuelMaxVelocity = 0.4 -- The maximum velocity you can have while refueling

    self.externalRefuel = false -- Set this to true if you want to control refuel with an external script
    self.refuelRate = 35 -- The amount of fuel to refuel every 500 milliseconds, if at a gas station, and externalRefuel = false

    self.enterVehicleFuelMessage = true -- Show message with current vehicle's fuel when entering one
    self.enterVehicleMessageColor = Color(255, 0, 0)

    -- Variables
    self.currentVehicle = nil

    self.tickTimer = Timer()
    self.tickTimeout = 500

    -- Fuel meter
    self.fuelMeterPosition = "BottomCenter" -- Options: BottomLeft, BottomRight, BottomCenter, TopRight, TopCenter
    self.fuelMeterRelativeWidth = 0.2 -- value * screen width
    self.fuelMeterRelativeHeight = 0.03 -- value * screen height
    self.fuelMeterBackground = Color(0, 0, 0, 100)
    self.fuelMeterForeground = Color(0, 255, 0)
    self.fuelMeterRelativeTextSize = 0.02 -- value * screen height
    self.fuelMeterTextColor = Color(255, 255, 255)

    self.fuelMeterWidth = 0
    self.fuelMeterHeight = 0
    self.fuelMeterLeft = 0
    self.fuelMeterTop = 0
    self.fuelMeterTextLeft = 0
    self.fuelMeterTextTop = 0
    self.fuelMeterIndicatorWidth = 0
    self.fuelMeterTextSize = 0
    self.fuelMeterText = "Fuel"

    self.gasStationClosest = {gasStation = nil, distance = nil}
    self.sentAtGasStation = nil

    self:CalculateMeterPosition({size = Vector2(Render.Width, Render.Height)})

    -- Events
    Events:Subscribe("LocalPlayerEnterVehicle", self, self.LocalPlayerEnterVehicle)
    Events:Subscribe("LocalPlayerExitVehicle", self, self.LocalPlayerExitVehicle)
    Events:Subscribe("LocalPlayerInput", self, self.LocalPlayerInput)
    Events:Subscribe("InputPoll", self, self.InputPoll)
    Events:Subscribe("PostTick", self, self.PostTick)
    Events:Subscribe("ResolutionChange", self, self.CalculateMeterPosition)
    Events:Subscribe("Render", self, self.Render)

    -- Custom events
    Events:Subscribe("FiniteFuelGetFuel", self, self.LocalGetFuel)
    Events:Subscribe("FiniteFuelSetFuel", self, self.LocalSetFuel)

    -- Networked events
    Network:Subscribe("FiniteFuelGetFuel", self, self.GetFuel)

    -- Get current vehicle's fuel if player in one
    if LocalPlayer:InVehicle() and IsValid(LocalPlayer:GetVehicle()) then
        Network:Send("FiniteFuelGetFuel", LocalPlayer:GetVehicle())
    end
end

-- ======================== Position calculations ========================
function FiniteFuel:CalculateMeterPosition(args)
    local size = args.size



    -- Calculate width and height
    self.fuelMeterWidth = size.x * self.fuelMeterRelativeWidth
    self.fuelMeterHeight = size.y * self.fuelMeterRelativeHeight

    -- Calculate left and top positions
    self.fuelMeterLeft = 0
    self.fuelMeterTop = 0

    if self.fuelMeterPosition == "BottomLeft" then
        self.fuelMeterLeft = 0
        self.fuelMeterTop = size.y - self.fuelMeterHeight
    elseif self.fuelMeterPosition == "BottomRight" then
        self.fuelMeterLeft = size.x - self.fuelMeterWidth
        self.fuelMeterTop = size.y - self.fuelMeterHeight
    elseif self.fuelMeterPosition == "BottomCenter" then
        self.fuelMeterLeft = (size.x / 2) - (self.fuelMeterWidth / 2)
        self.fuelMeterTop = size.y - self.fuelMeterHeight
    elseif self.fuelMeterPosition == "TopRight" then
        self.fuelMeterLeft = size.x - self.fuelMeterWidth
        self.fuelMeterTop = 0
    elseif self.fuelMeterPosition == "TopCenter" then
        self.fuelMeterLeft = (size.x / 2) - (self.fuelMeterWidth / 2)
        self.fuelMeterTop = 0
    end

    self.fuelMeterTextSize = size.y * self.fuelMeterRelativeTextSize

    self:CalculateTextPosition()
end

function FiniteFuel:CalculateTextPosition()
    local textSize = Render:GetTextSize(self.fuelMeterText, self.fuelMeterTextSize)

    self.fuelMeterTextLeft = self.fuelMeterLeft + (self.fuelMeterWidth / 2) - (textSize.x / 2)
    self.fuelMeterTextTop = self.fuelMeterTop + (self.fuelMeterHeight / 2) - (textSize.y / 2)
end

-- ======================== Rendering ========================
function FiniteFuel:Render(args)
    fill:SetPosition(Vector2(0, 120))
    background:SetPosition(Vector2(0, 120))
    if self.currentVehicle == nil or not IsValid(self.currentVehicle.vehicle) or not LocalPlayer:InVehicle() or Game:GetState() ~= GUIState.Game then return end
    v = LocalPlayer:GetVehicle()
    background:SetSize(Vector2(16, 512))
    fill:SetSize(Vector2(16, math.floor(self.currentVehicle.fuel / (v:GetMass() / 1024))))
    print(self.currentVehicle.fuel)
    transform = Transform2()
    transform:Translate(Vector2(Render.Size.x + 120, Render.Size.y - 128))
    transform:Rotate(math.rad(90))
    Render:SetTransform(transform)
    -- Draw background
    background:Draw()
    -- Draw indicator
    fill:Draw()

    Render:ResetTransform()


    -- Draw closest gas station on minimap
    if self.gasStationMinimapIcon and self.gasStationClosest.gasStation ~= nil then Render:FillArea(Render:WorldToMinimap(self.gasStationClosest.gasStation.position), Vector2(10, 10), self.gasStationMinimapColor) end

    -- Draw gas station marker
    if self.gasStationMarker and self.gasStationClosest.gasStation ~= nil and self.gasStationClosest.distance <= self.gasStationMarkerVisibleRadius then
        local position = self.gasStationClosest.gasStation.position
        local distance = self.gasStationClosest.distance
        local pos1 = position
        local pos2 = position + (Vector3(-1, 2, 0) * (distance / 20))
        local pos3 = position + (Vector3(1, 2, 0) * (distance / 20))
        Render:FillTriangle(pos1, pos2, pos3, self.gasStationMarkerColor)
        pos1 = position
        pos2 = position + (Vector3(0, 2, -1) * (distance / 20))
        pos3 = position + (Vector3(0, 2, 1) * (distance / 20))
        Render:FillTriangle(pos1, pos2, pos3, self.gasStationMarkerColor)
    end
end

-- ======================== Enter/exit vehicle ========================
function FiniteFuel:LocalPlayerEnterVehicle(args)
    if not IsValid(args.vehicle) then return end

    self.gasStationClosest = {gasStation = nil, distance = nil}

    Network:Send("FiniteFuelGetFuel", args.vehicle)
end

function FiniteFuel:LocalPlayerExitVehicle(args)
    if self.currentVehicle == nil or self.currentVehicle.vehicle ~= args.vehicle then return end

    Network:Send("FiniteFuelSetFuel", {vehicle = args.vehicle, fuel = self.currentVehicle.fuel})

    -- Notify external scripts of fuel station exit
    if self.externalRefuel and self.sentAtGasStation ~= nil then
        Events:Fire("FiniteFuelExitedGasStation", {vehicle = self.currentVehicle.vehicle, gasStation = self.sentAtGasStation})
        self.sentAtGasStation = nil
    end

    self.currentVehicle = nil
end

-- ======================== Tick update events ========================
function FiniteFuel:InputPoll()
    if self.currentVehicle == nil or
    not IsValid(self.currentVehicle.vehicle) or
    self.currentVehicle.fuel > 0 or
    (self.currentVehicle.vehicleGasType ~= FiniteFuelGasTypes.Aircraft and self.currentVehicle.vehicleGasType ~= FiniteFuelGasTypes.Aircraft) then return end

    Input:SetValue(Action.HeliDecAltitude, 1)
    Input:SetValue(Action.PlaneDecTrust, 1)
end

function FiniteFuel:PostTick()
    if self.tickTimer:GetMilliseconds() < self.tickTimeout or self.currentVehicle == nil or not IsValid(self.currentVehicle.vehicle) then return end
    self.tickTimer:Restart()

    local playerPosition = LocalPlayer:GetPosition()
    if self.gasStationClosest.gasStation ~= nil then self.gasStationClosest.distance = Vector3.Distance(self.gasStationClosest.gasStation.position, playerPosition) end
    for index, gasStation in ipairs(FiniteFuelGasStations) do
        -- Only evaluate if of same gas type
        if gasStation.gasType == self.currentVehicle.vehicleGasType then
            local gasStationPosition = gasStation.position
            local distance = Vector3.Distance(gasStationPosition, playerPosition)

            -- Closer than last gas station checked
            if self.gasStationClosest.distance == nil or distance < self.gasStationClosest.distance then
                self.gasStationClosest = {gasStation = gasStation, distance = distance}
            end
        end
    end

    -- Close and moving slow enough to refuel, and tank is not full
    local velocity = self.currentVehicle.vehicle:GetLinearVelocity():Length()
    local idling = velocity <= self.gasStationRefuelMaxVelocity

    if self.gasStationClosest.distance ~= nil and self.gasStationClosest.distance <= self.gasStationRefuelRadius and idling and self.currentVehicle.fuel < self.currentVehicle.tankSize then
        if not self.externalRefuel then
            -- Fuel up. Set to tank size if full
            self.currentVehicle.fuel = self.currentVehicle.fuel + self.refuelRate
            if self.currentVehicle.fuel > self.currentVehicle.tankSize then self.currentVehicle.fuel = self.currentVehicle.tankSize end

            -- Change text
            if self.fuelMeterText ~= "Refuelling..." then
                self.fuelMeterText = "Refuelling..."
                self:CalculateTextPosition()
            end
        else
            -- Send network event for external refuelling scripts
            if self.sentAtGasStation ~= self.gasStationClosest.gasStation then
                self.sentAtGasStation = self.gasStationClosest.gasStation
                Events:Fire("FiniteFuelEnteredGasStation", {vehicle = self.currentVehicle.vehicle, vehicleGasType = self.currentVehicle.vehicleGasType, gasStation = self.gasStationClosest.gasStation})
            end
        end
    elseif idling and self.currentVehicle.fuel > 0 then -- Idling
        -- Drain idle. Set tank to 0 if less than empty
        self.currentVehicle.fuel = self.currentVehicle.fuel - self.currentVehicle.idleDrainRate

        -- Change text
        if self.fuelMeterText ~= "Idle" then
            self.fuelMeterText = "Idle"
            self:CalculateTextPosition()
        end
        if self.currentVehicle.fuel < 0 then self.currentVehicle.fuel = 0 end
    elseif not idling and self.currentVehicle.fuel > 0 then -- Moving
        -- Drain moving. Set tank to 0 if less than empty
        local drain = (velocity * self.currentVehicle.drainRate)-- * 100 -- DEBUGGING
        self.currentVehicle.fuel = self.currentVehicle.fuel - drain
        if self.currentVehicle.fuel < 0 then self.currentVehicle.fuel = 0 end

        -- Change text
        if self.fuelMeterText ~= "Fuel" then
            self.fuelMeterText = "Fuel"
            self:CalculateTextPosition()
        end
    end

    -- Send left gas station to external scripts
    if self.externalRefuel and self.sentAtGasStation ~= nil and self.gasStationClosest.distance > self.gasStationRefuelRadius then
        Events:Fire("FiniteFuelExitedGasStation", {vehicle = self.currentVehicle.vehicle, gasStation = self.sentAtGasStation})
        self.sentAtGasStation = nil
    end

    -- Calculate indicator width
    self.fuelMeterIndicatorWidth = self.fuelMeterWidth / self.currentVehicle.tankSize * self.currentVehicle.fuel
    fill:SetSize(Vector2(16, self.fuelMeterWidth))
end

-- ======================== Movement blocker ========================
function FiniteFuel:LocalPlayerInput(args)
    if self.currentVehicle == nil or not IsValid(self.currentVehicle.vehicle) or self.currentVehicle.fuel > 0 then return end

    -- Fuel empty, block movement keys
    return not FiniteFuelVehicleKeys[args.input]
end

-- ======================== Vehicle info ========================
function FiniteFuel:GetFuel(args)
    local vehicle = args.vehicle
    local fuel = args.fuel

    self.currentVehicle = FiniteFuelVehicle(vehicle, fuel)
    if self.enterVehicleFuelMessage then
        Chat:Print("Vehicle currently has " .. math.round(self.currentVehicle.fuel) .. "/" .. math.round(self.currentVehicle.tankSize) .. " fuel.", self.enterVehicleMessageColor)
    end
end

-- ======================== Local events ========================
function FiniteFuel:LocalGetFuel()
    if self.currentVehicle == nil or not IsValid(self.currentVehicle.vehicle) then return end

    Events:Fire("FiniteFuelReturnGetFuel", {vehicle = self.currentVehicle.vehicle, vehicleGasType = self.currentVehicle.vehicleGasType, fuel = self.currentVehicle.fuel, tankSize = self.currentVehicle.tankSize})
end

function FiniteFuel:LocalSetFuel(fuel)
    if fuel < 0 or self.currentVehicle == nil or not IsValid(self.currentVehicle.vehicle) then return end

    if fuel > self.currentVehicle.tankSize then fuel = self.currentVehicle.tankSize end

    self.currentVehicle.fuel = fuel
end

-- ======================== Initialize ========================
Events:Subscribe("ModuleLoad", function()
    FiniteFuel()
end)