mirror of
https://github.com/parnic/ice-hud.git
synced 2025-06-16 06:40:13 -05:00
145 lines
3.8 KiB
Lua
145 lines
3.8 KiB
Lua
local L = LibStub("AceLocale-3.0"):GetLocale("IceHUD", false)
|
|
local GlobalCoolDown = IceCore_CreateClass(IceBarElement)
|
|
|
|
-- Constructor --
|
|
function GlobalCoolDown.prototype:init()
|
|
GlobalCoolDown.super.prototype.init(self, "GlobalCoolDown")
|
|
|
|
self.unit = "player"
|
|
self.startTime = nil
|
|
self.duration = nil
|
|
|
|
self:SetDefaultColor("GlobalCoolDown", 0.1, 0.1, 0.1)
|
|
end
|
|
|
|
-- OVERRIDE
|
|
function GlobalCoolDown.prototype:Enable(core)
|
|
GlobalCoolDown.super.prototype.Enable(self, core)
|
|
|
|
--self:RegisterEvent("ACTIONBAR_UPDATE_COOLDOWN", "CooldownStateChanged")
|
|
self:RegisterEvent("UNIT_SPELLCAST_START","CooldownStateChanged")
|
|
self:RegisterEvent("UNIT_SPELLCAST_SUCCEEDED","CooldownStateChanged")
|
|
|
|
self:RegisterEvent("UNIT_SPELLCAST_INTERRUPTED","CooldownAborted")
|
|
self:RegisterEvent("UNIT_SPELLCAST_FAILED","CooldownAborted")
|
|
|
|
self:Show(false)
|
|
|
|
self.frame:SetFrameStrata("TOOLTIP")
|
|
|
|
self.CDSpellId = self:GetSpellId()
|
|
end
|
|
|
|
function GlobalCoolDown.prototype:CooldownAborted(event, unit, spell)
|
|
if unit ~= "player" or not spell or not self.CurrSpell or self.CurrSpell ~= spell then
|
|
return
|
|
end
|
|
|
|
self.CurrLerpTime = self.moduleSettings.desiredLerpTime
|
|
self.CurrSpell = nil
|
|
end
|
|
|
|
-- OVERRIDE
|
|
function GlobalCoolDown.prototype:GetDefaultSettings()
|
|
local settings = GlobalCoolDown.super.prototype.GetDefaultSettings(self)
|
|
|
|
settings["enabled"] = false
|
|
settings["side"] = IceCore.Side.Left
|
|
settings["offset"] = 6
|
|
settings["shouldAnimate"] = true
|
|
settings["hideAnimationSettings"] = true
|
|
settings["desiredLerpTime"] = 1
|
|
settings["lowThreshold"] = 0
|
|
settings["barVisible"]["bg"] = false
|
|
settings["usesDogTagStrings"] = false
|
|
settings["bHideMarkerSettings"] = true
|
|
|
|
return settings
|
|
end
|
|
|
|
-- OVERRIDE
|
|
function GlobalCoolDown.prototype:GetOptions()
|
|
local opts = GlobalCoolDown.super.prototype.GetOptions(self)
|
|
|
|
opts["lowThreshold"] = nil
|
|
opts["textSettings"] = nil
|
|
opts.alwaysFullAlpha = nil
|
|
|
|
return opts
|
|
end
|
|
|
|
function GlobalCoolDown.prototype:IsFull(scale)
|
|
return false
|
|
end
|
|
|
|
function GlobalCoolDown.prototype:CooldownStateChanged(event, unit, spell)
|
|
if unit ~= "player" or not spell then
|
|
return
|
|
end
|
|
|
|
local start, dur = GetSpellCooldown(self.CDSpellId)
|
|
|
|
if start and dur ~= nil and dur > 0 and dur <= 1.5 then
|
|
local bRestart = not self.startTime or start > self.startTime + 0.5
|
|
if bRestart then
|
|
self.startTime = start
|
|
self.duration = dur
|
|
|
|
self:SetScale(1, true)
|
|
self.LastScale = 1
|
|
self.DesiredScale = 0
|
|
self.CurrLerpTime = 0
|
|
self.moduleSettings.desiredLerpTime = dur or 1
|
|
self.CurrSpell = spell
|
|
|
|
self.barFrame.bar:SetVertexColor(self:GetColor("GlobalCoolDown", 0.8))
|
|
self:Show(true)
|
|
end
|
|
end
|
|
|
|
if event == "UNIT_SPELLCAST_SUCCEEDED" then
|
|
self.CurrSpell = nil
|
|
end
|
|
end
|
|
|
|
function GlobalCoolDown.prototype:MyOnUpdate()
|
|
GlobalCoolDown.super.prototype.MyOnUpdate(self)
|
|
|
|
if self:IsVisible() and self.startTime ~= nil and self.duration ~= nil
|
|
and self.CurrScale <= 0.01 then
|
|
self:Show(false)
|
|
end
|
|
end
|
|
|
|
function GlobalCoolDown.prototype:CreateFrame()
|
|
GlobalCoolDown.super.prototype.CreateFrame(self)
|
|
|
|
self.barFrame.bar:SetVertexColor(self:GetColor("GlobalCoolDown", 0.8))
|
|
local r, g, b = self.settings.backgroundColor.r, self.settings.backgroundColor.g, self.settings.backgroundColor.b
|
|
self.frame.bg:SetVertexColor(r, g, b, 0.6)
|
|
end
|
|
|
|
function GlobalCoolDown.prototype:GetSpellId()
|
|
local defaultSpells
|
|
|
|
defaultSpells = {
|
|
ROGUE=1752, -- sinister strike
|
|
PRIEST=585, -- smite
|
|
DRUID=5176, -- wrath
|
|
WARRIOR=34428, -- victory rush (not available until 5, sadly)
|
|
MAGE=133, -- fireball
|
|
WARLOCK=686, -- shadow bolt
|
|
PALADIN=20154, -- seal of righteousness
|
|
SHAMAN=403, -- lightning bolt
|
|
HUNTER=3044, -- arcane shot
|
|
DEATHKNIGHT=47541, -- death coil
|
|
MONK=100780, -- jab
|
|
}
|
|
|
|
local _, unitClass = UnitClass("player")
|
|
return defaultSpells[unitClass]
|
|
end
|
|
|
|
-- Load us up
|
|
IceHUD.GlobalCoolDown = GlobalCoolDown:new()
|