mirror of
https://github.com/parnic/ice-hud.git
synced 2025-06-15 22:30:13 -05:00
77 lines
1.9 KiB
Lua
77 lines
1.9 KiB
Lua
local L = LibStub("AceLocale-3.0"):GetLocale("IceHUD", false)
|
|
local DruidMana = IceCore_CreateClass(IceUnitBar)
|
|
|
|
DruidMana.prototype.druidMana = nil
|
|
DruidMana.prototype.druidManaMax = nil
|
|
|
|
local MANA_POWER_INDEX = SPELL_POWER_MANA
|
|
|
|
-- Constructor --
|
|
function DruidMana.prototype:init()
|
|
DruidMana.super.prototype.init(self, "DruidMana", "player")
|
|
|
|
self.side = IceCore.Side.Right
|
|
self.offset = 0
|
|
|
|
self:SetDefaultColor("DruidMana", 87, 82, 141)
|
|
end
|
|
|
|
|
|
function DruidMana.prototype:GetDefaultSettings()
|
|
local settings = DruidMana.super.prototype.GetDefaultSettings(self)
|
|
|
|
settings["side"] = IceCore.Side.Right
|
|
settings["offset"] = 0
|
|
settings["textVisible"] = {upper = true, lower = false}
|
|
settings["upperText"] = "[PercentDruidMP:Round]"
|
|
settings["lowerText"] = "[FractionalDruidMP:Color('3071bf'):Bracket]"
|
|
|
|
return settings
|
|
end
|
|
|
|
|
|
function DruidMana.prototype:Enable(core)
|
|
DruidMana.super.prototype.Enable(self, core)
|
|
|
|
self:RegisterEvent("UPDATE_SHAPESHIFT_FORM", "Update")
|
|
if IceHUD.WowVer >= 40000 then
|
|
self:RegisterEvent("UNIT_POWER", "Update")
|
|
self:RegisterEvent("UNIT_MAXPOWER", "Update")
|
|
else
|
|
self:RegisterEvent("UNIT_MAXMANA", "Update")
|
|
self:RegisterEvent("UNIT_MANA", "Update")
|
|
end
|
|
end
|
|
|
|
|
|
function DruidMana.prototype:Disable(core)
|
|
DruidMana.super.prototype.Disable(self, core)
|
|
end
|
|
|
|
|
|
function DruidMana.prototype:Update()
|
|
DruidMana.super.prototype.Update(self)
|
|
|
|
local forms = (UnitPowerType(self.unit) ~= 0)
|
|
|
|
self.druidMana = UnitPower(self.unit, MANA_POWER_INDEX)
|
|
self.druidManaMax = UnitPowerMax(self.unit, MANA_POWER_INDEX)
|
|
|
|
if (not self.alive or not forms or not self.druidMana or not self.druidManaMax or self.druidManaMax == 0) then
|
|
self:Show(false)
|
|
return
|
|
else
|
|
self:Show(true)
|
|
end
|
|
|
|
self:UpdateBar(self.druidManaMax ~= 0 and self.druidMana / self.druidManaMax or 0, "DruidMana")
|
|
end
|
|
|
|
|
|
|
|
-- Load us up (if we are a druid)
|
|
local _, unitClass = UnitClass("player")
|
|
if (unitClass == "DRUID") then
|
|
IceHUD.DruidMana = DruidMana:new()
|
|
end
|