mirror of
https://github.com/parnic/ice-hud.git
synced 2025-06-15 22:30:13 -05:00
304 lines
7.9 KiB
Lua
304 lines
7.9 KiB
Lua
local L = LibStub("AceLocale-3.0"):GetLocale("IceHUD", false)
|
|
local StaggerBar = IceCore_CreateClass(IceUnitBar)
|
|
local IceHUD = _G.IceHUD
|
|
|
|
local tostring = tostring
|
|
local floor = math.floor
|
|
local min = math.min
|
|
local strform = string.format
|
|
|
|
local playerName = ""
|
|
local LightID = 124275
|
|
local ModerateID = 124274
|
|
local HeavyID = 124273
|
|
local StaggerID = 124255
|
|
local staggerNames = {"", "", ""}
|
|
|
|
local MinLevel = 10
|
|
|
|
StaggerBar.prototype.StaggerDuration = 0
|
|
StaggerBar.prototype.StaggerEndTime = 0
|
|
|
|
local function ReadableNumber(num, places)
|
|
local ret
|
|
local placeValue = ("%%.%df"):format(places or 0)
|
|
|
|
if not num then
|
|
ret = 0
|
|
elseif num >= 1000000 then
|
|
ret = placeValue:format(num / 1000000) .. "M" -- million
|
|
elseif num >= 1000 then
|
|
ret = placeValue:format(num / 1000) .. "k" -- thousand
|
|
else
|
|
ret = num -- hundreds
|
|
end
|
|
|
|
return ret
|
|
end
|
|
|
|
function StaggerBar.prototype:init()
|
|
StaggerBar.super.prototype.init(self, "Stagger", "player")
|
|
|
|
self:SetDefaultColor("Stagger1", 200, 180, 20)
|
|
self:SetDefaultColor("Stagger2", 200, 90, 10)
|
|
self:SetDefaultColor("Stagger3", 200, 0, 0)
|
|
self:SetDefaultColor("StaggerTime", 255, 255, 255)
|
|
|
|
self.bTreatEmptyAsFull = false
|
|
end
|
|
|
|
function StaggerBar.prototype:Redraw()
|
|
StaggerBar.super.prototype.Redraw(self)
|
|
|
|
self:MyOnUpdate()
|
|
end
|
|
|
|
function StaggerBar.prototype:GetDefaultSettings()
|
|
local settings = StaggerBar.super.prototype.GetDefaultSettings(self)
|
|
|
|
settings["enabled"] = true
|
|
settings["shouldAnimate"] = true
|
|
settings["lowThreshold"] = 0
|
|
settings["side"] = IceCore.Side.Left
|
|
settings["offset"] = 3
|
|
settings["upperText"]=""
|
|
settings["showAsPercentOfMax"] = true
|
|
settings["maxPercent"] = 20
|
|
settings["timerAlpha"] = 0.3
|
|
settings["usesDogTagStrings"] = false
|
|
settings["lockLowerFontAlpha"] = false
|
|
settings["lowerTextString"] = ""
|
|
settings["lowerTextVisible"] = false
|
|
settings["hideAnimationSettings"] = true
|
|
settings["bAllowExpand"] = true
|
|
settings["bShowWithNoTarget"] = true
|
|
|
|
return settings
|
|
end
|
|
|
|
function StaggerBar.prototype:GetOptions()
|
|
local opts = StaggerBar.super.prototype.GetOptions(self)
|
|
|
|
opts.reverse.hidden = true
|
|
|
|
opts["maxPercent"] =
|
|
{
|
|
type = "range",
|
|
name = "Max Percent",
|
|
desc = "Maximum percentage of your maximum health for the Stagger bar to represent. I.e, if set to 20%, the bar will be full when the Stagger damage over time effect is dealing 20% of your maximum health per second.",
|
|
min = 0,
|
|
max = 50,
|
|
step = 1,
|
|
get = function()
|
|
return self.moduleSettings.maxPercent
|
|
end,
|
|
set = function(info, v)
|
|
self.moduleSettings.maxPercent = v
|
|
self:Redraw()
|
|
end,
|
|
disabled = function()
|
|
return not self.moduleSettings.enabled
|
|
end
|
|
}
|
|
|
|
opts["timerAlpha"] =
|
|
{
|
|
type = "range",
|
|
name = "Timer bar alpha",
|
|
desc = "What alpha value to use for the bar that displays how long until Stagger wears off.",
|
|
min = 0,
|
|
max = 100,
|
|
step = 5,
|
|
get = function()
|
|
return self.moduleSettings.timerAlpha * 100
|
|
end,
|
|
set = function(info, v)
|
|
self.moduleSettings.timerAlpha = v / 100
|
|
self:Redraw()
|
|
end,
|
|
disabled = function()
|
|
return not self.moduleSettings.enabled
|
|
end
|
|
}
|
|
|
|
return opts
|
|
end
|
|
|
|
function StaggerBar.prototype:Enable(core)
|
|
StaggerBar.super.prototype.Enable(self, core)
|
|
|
|
playerName = UnitName(self.unit)
|
|
staggerNames[1] = GetSpellInfo(LightID)
|
|
staggerNames[2] = GetSpellInfo(ModerateID)
|
|
staggerNames[3] = GetSpellInfo(HeavyID)
|
|
|
|
self:RegisterEvent("COMBAT_LOG_EVENT_UNFILTERED")
|
|
self:RegisterEvent("PLAYER_ENTERING_WORLD")
|
|
self:RegisterEvent("ACTIVE_TALENT_GROUP_CHANGED")
|
|
|
|
self:UpdateShown()
|
|
end
|
|
|
|
function StaggerBar.prototype:Disable(core)
|
|
StaggerBar.super.prototype.Disable(self, core)
|
|
end
|
|
|
|
function StaggerBar.prototype:CreateFrame()
|
|
StaggerBar.super.prototype.CreateFrame(self)
|
|
self:CreateTimerBar()
|
|
|
|
self:UpdateShown()
|
|
self:UpdateAlpha()
|
|
end
|
|
|
|
function StaggerBar.prototype:CreateTimerBar()
|
|
self.timerFrame = self:BarFactory(self.timerFrame, "MEDIUM","ARTWORK")
|
|
|
|
self.CurrScale = 0
|
|
|
|
self.timerFrame.bar:SetVertexColor(self:GetColor("StaggerTime", self.moduleSettings.timerAlpha))
|
|
self.timerFrame.bar:SetHeight(0)
|
|
|
|
self:UpdateBar(1, "undef")
|
|
self:UpdateTimerFrame()
|
|
end
|
|
|
|
function StaggerBar.prototype:UpdateShown()
|
|
if GetSpecialization() == SPEC_MONK_BREWMASTER and not UnitInVehicle(self.unit) and UnitLevel(self.unit) >= MinLevel then
|
|
self:Show(true)
|
|
else
|
|
self:Show(false)
|
|
end
|
|
end
|
|
|
|
function StaggerBar.prototype:PLAYER_ENTERING_WORLD()
|
|
self:UpdateStaggerBar()
|
|
end
|
|
|
|
function StaggerBar.prototype:ACTIVE_TALENT_GROUP_CHANGED()
|
|
self:UpdateStaggerBar()
|
|
end
|
|
|
|
function StaggerBar.prototype:GetDebuffInfo()
|
|
local amount = 0
|
|
local duration = 0
|
|
local staggerLevel = 1
|
|
|
|
for i = 1, IceCore.BuffLimit do
|
|
local debuffID = select(11, UnitDebuff(self.unit, i))
|
|
|
|
if debuffID == LightID or debuffID == ModerateID or debuffID == HeavyID then
|
|
local spellName = select(1, UnitDebuff(self.unit, i))
|
|
|
|
duration = select(6, UnitAura(self.unit, spellName, "", "HARMFUL"))
|
|
amount = select(15, UnitAura(self.unit, spellName, "", "HARMFUL"))
|
|
staggerLevel = (debuffID == LightID) and 1 or (debuffID == ModerateID) and 2 or 3
|
|
|
|
break
|
|
end
|
|
end
|
|
|
|
self.amount = amount or 0
|
|
self.duration = duration or 0
|
|
self.staggerLevel = staggerLevel or 1
|
|
end
|
|
|
|
function StaggerBar.prototype:COMBAT_LOG_EVENT_UNFILTERED(_, timestamp, event, hideCaster, sourceGUID, sourceName, sourceFlags, sourceFlags2, destGUID, destName, destFlags, destFlags2, spellID)
|
|
if destName == playerName then
|
|
if spellID == StaggerID or event == "SWING_DAMAGE" or event == "SPELL_AURA_APPLIED" or event == "SPELL_AURA_REMOVED" then
|
|
self:UpdateStaggerBar()
|
|
end
|
|
end
|
|
end
|
|
|
|
function StaggerBar.prototype:UpdateStaggerBar()
|
|
self:GetDebuffInfo()
|
|
|
|
-- local health = UnitHealth(self.unit)
|
|
local maxHealth = UnitHealthMax(self.unit)
|
|
local percent = (self.amount / maxHealth) * 100
|
|
local percentText = percent >= 10 and floor(percent) or strform("%.1f", percent)
|
|
local scale = IceHUD:Clamp((self.amount / maxHealth) * (100 / self.moduleSettings.maxPercent), 0, 1)
|
|
|
|
if self.amount > 0 and self.duration <= 10 then
|
|
-- self.timerFrame.bar:SetVertexColor(self:GetColor("StaggerTime", self.moduleSettings.timerAlpha))
|
|
self:UpdateBar(scale or 0, "Stagger"..self.staggerLevel)
|
|
self:SetBottomText1(self.moduleSettings.upperText .. " " .. ReadableNumber(self.amount, 1) .. " (" .. percentText .. "%)")
|
|
self:UpdateShown()
|
|
self:UpdateTimerFrame()
|
|
else
|
|
self:UpdateBar(0, "Stagger1")
|
|
self:SetBottomText1("")
|
|
self:Show(false)
|
|
end
|
|
end
|
|
|
|
function StaggerBar.prototype:GetDebuffDuration(unitName, buffName)
|
|
local name, _, _, _, _, duration, endTime = UnitDebuff(unitName, buffName)
|
|
|
|
if name then
|
|
return duration, endTime - GetTime()
|
|
end
|
|
|
|
return nil, nil
|
|
end
|
|
|
|
function StaggerBar.prototype:MyOnUpdate()
|
|
StaggerBar.super.prototype.MyOnUpdate(self)
|
|
|
|
if self.bUpdateTimer then
|
|
self:UpdateTimerFrame(nil, self.unit, true)
|
|
end
|
|
end
|
|
|
|
function StaggerBar.prototype:UpdateTimerFrame(event, unit, fromUpdate)
|
|
if unit and unit ~= self.unit then
|
|
return
|
|
end
|
|
|
|
local now = GetTime()
|
|
local remaining = nil
|
|
|
|
if not fromUpdate then
|
|
for i = 1, 3 do
|
|
self.StaggerDuration, remaining = self:GetDebuffDuration(self.unit, staggerNames[i])
|
|
|
|
if remaining then
|
|
break
|
|
end
|
|
end
|
|
|
|
if not remaining then
|
|
self.StaggerEndTime = 0
|
|
else
|
|
self.StaggerEndTime = remaining + now
|
|
end
|
|
end
|
|
|
|
if self.StaggerEndTime and self.StaggerEndTime >= now then
|
|
if not fromUpdate then
|
|
self.bUpdateTimer = true
|
|
end
|
|
|
|
if not remaining and (self.StaggerEndTime and self.StaggerEndTime >= now) then
|
|
remaining = self.StaggerEndTime - now
|
|
end
|
|
|
|
if remaining then
|
|
self:SetBarCoord(self.timerFrame, IceHUD:Clamp(remaining / 10, 0, 1))
|
|
self.timerFrame:Show()
|
|
else
|
|
self:SetBarCoord(self.timerFrame, 0)
|
|
self.timerFrame:Hide()
|
|
end
|
|
else
|
|
self:SetBarCoord(self.timerFrame, 0)
|
|
self.timerFrame:Hide()
|
|
self.bUpdateTimer = false
|
|
end
|
|
end
|
|
|
|
local _, unitClass = UnitClass("player")
|
|
if unitClass == "MONK" then
|
|
IceHUD.StaggerBar = StaggerBar:new()
|
|
end |