mirror of
https://github.com/parnic/ice-hud.git
synced 2025-06-16 06:40:13 -05:00
129 lines
3.2 KiB
Lua
129 lines
3.2 KiB
Lua
local FocusMana = IceCore_CreateClass(IceUnitBar)
|
|
|
|
|
|
-- Constructor --
|
|
function FocusMana.prototype:init()
|
|
FocusMana.super.prototype.init(self, "FocusMana", "focus")
|
|
|
|
self:SetDefaultColor("FocusMana", 52, 64, 221)
|
|
self:SetDefaultColor("FocusRage", 235, 44, 26)
|
|
self:SetDefaultColor("FocusEnergy", 228, 242, 31)
|
|
self:SetDefaultColor("FocusFocus", 242, 149, 98)
|
|
end
|
|
|
|
|
|
function FocusMana.prototype:GetDefaultSettings()
|
|
local settings = FocusMana.super.prototype.GetDefaultSettings(self)
|
|
|
|
settings["enabled"] = false
|
|
settings["side"] = IceCore.Side.Right
|
|
settings["offset"] = -2
|
|
settings["scale"] = 0.7
|
|
settings["upperText"] = "[PercentMP:Round]"
|
|
settings["lowerText"] = ""
|
|
settings["barVerticalOffset"] = 35
|
|
|
|
return settings
|
|
end
|
|
|
|
|
|
function FocusMana.prototype:Enable(core)
|
|
FocusMana.super.prototype.Enable(self, core)
|
|
|
|
if IceHUD.WowVer >= 40000 then
|
|
self:RegisterEvent("UNIT_POWER", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_MAXPOWER", "UpdateEvent")
|
|
else
|
|
self:RegisterEvent("UNIT_MANA", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_MAXMANA", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_RAGE", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_MAXRAGE", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_ENERGY", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_MAXENERGY", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_FOCUS", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_MAXFOCUS", "UpdateEvent")
|
|
end
|
|
self:RegisterEvent("UNIT_AURA", "UpdateEvent")
|
|
self:RegisterEvent("UNIT_FLAGS", "UpdateEvent")
|
|
self:RegisterEvent("PLAYER_FOCUS_CHANGED", "UpdateFocus")
|
|
|
|
self:Update(self.unit)
|
|
end
|
|
|
|
function FocusMana.prototype:UpdateFocus()
|
|
self:Update(self.unit)
|
|
end
|
|
|
|
function FocusMana.prototype:UpdateEvent(event, unit)
|
|
self:Update(unit)
|
|
end
|
|
|
|
function FocusMana.prototype:Update(unit)
|
|
FocusMana.super.prototype.Update(self)
|
|
if (unit and (unit ~= self.unit)) then
|
|
return
|
|
end
|
|
|
|
if ((not UnitExists(unit)) or (self.maxMana == 0)) then
|
|
self:Show(false)
|
|
return
|
|
else
|
|
self:Show(true)
|
|
end
|
|
|
|
|
|
local manaType = UnitPowerType(self.unit)
|
|
|
|
local color = "FocusMana"
|
|
if (self.moduleSettings.scaleManaColor) then
|
|
color = "ScaledManaColor"
|
|
end
|
|
if (manaType == SPELL_POWER_RAGE) then
|
|
color = "FocusRage"
|
|
elseif (manaType == SPELL_POWER_FOCUS) then
|
|
color = "FocusFocus"
|
|
elseif (manaType == SPELL_POWER_ENERGY) then
|
|
color = "FocusEnergy"
|
|
end
|
|
|
|
if (self.tapped) then
|
|
color = "Tapped"
|
|
end
|
|
|
|
self:UpdateBar(self.manaPercentage, color)
|
|
|
|
if not IceHUD.IceCore:ShouldUseDogTags() then
|
|
self:SetBottomText1(math.floor(self.manaPercentage * 100))
|
|
self:SetBottomText2(self:GetFormattedText(self.mana, self.maxMana), color)
|
|
end
|
|
end
|
|
|
|
|
|
-- OVERRIDE
|
|
function FocusMana.prototype:GetOptions()
|
|
local opts = FocusMana.super.prototype.GetOptions(self)
|
|
|
|
opts["scaleManaColor"] = {
|
|
type = "toggle",
|
|
name = "Color bar by mana %",
|
|
desc = "Colors the mana bar from MaxManaColor to MinManaColor based on current mana %",
|
|
get = function()
|
|
return self.moduleSettings.scaleManaColor
|
|
end,
|
|
set = function(info, value)
|
|
self.moduleSettings.scaleManaColor = value
|
|
self:Redraw()
|
|
end,
|
|
disabled = function()
|
|
return not self.moduleSettings.enabled
|
|
end,
|
|
order = 51
|
|
}
|
|
|
|
return opts
|
|
end
|
|
|
|
|
|
-- Load us up
|
|
IceHUD.FocusMana = FocusMana:new()
|