Mercurial > wow > kbf

annotate KBF.lua @ 12:5f07c89cb6af

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
tooltips for buffs
author Chris Mellon <arkanes@gmai.com>
date Wed, 13 Oct 2010 11:11:13 -0500
parents 76650e326a57
children be6c10697c59
rev   line source
arkanes@0 1 local _, kbf = ...
arkanes@0 2
arkanes@0 3 KBF = kbf -- make global for debugging
arkanes@0 4
arkanes@0 5 local kbf = LibStub("AceAddon-3.0"):NewAddon(kbf, "KBF", "AceEvent-3.0", "AceConsole-3.0")
arkanes@0 6
arkanes@0 7
arkanes@0 8 function kbf:OnInitialize()
arkanes@10 9 self.debuffFrames = {}
arkanes@0 10 self.anchor = self:CreateAnchorFrame()
arkanes@0 11 self.anchor:Show()
arkanes@0 12 self:RegisterEvent("UNIT_AURA")
arkanes@4 13 -- set up the countdown timer
arkanes@4 14 -- TODO: Fancy enable/disable based on whether you have any timed buffs.
arkanes@4 15 -- Not a big deal, how often do you care about that
arkanes@8 16 -- also TODO: Maybe should bucket OnUpdates somehow
arkanes@4 17 self.anchor:SetScript("OnUpdate", function() self:OnUpdate() end)
arkanes@7 18 self.dirty = true -- force an immediate scan on login
arkanes@8 19 --self:HideBlizzardBuffFrames()
arkanes@0 20 end
arkanes@1 21 -- naming convention
arkanes@1 22 -- "frame" is the secure aura button created by the group handler
arkanes@1 23 -- "bar" is the set of icon + status bars that we create to show the buff time
arkanes@0 24
arkanes@8 25 function kbf:HideBlizzardBuffFrames()
arkanes@8 26 local function HideBlizFrame(frame)
arkanes@8 27 if not frame then return end
arkanes@8 28 frame:UnregisterAllEvents()
arkanes@8 29 frame:SetScript("OnUpdate", nil)
arkanes@8 30 frame:Hide()
arkanes@8 31 frame.Show = function() end
arkanes@8 32 end
arkanes@8 33 HideBlizFrame(BuffFrame)
arkanes@8 34 HideBlizFrame(ConsolidatedBuffs)
arkanes@8 35 HideBlizFrame(TemporaryEnchantFrame)
arkanes@8 36
arkanes@8 37 end
arkanes@8 38
arkanes@4 39 function kbf:OnUpdate()
arkanes@10 40 local unit = self.secureFrame:GetAttribute("unit")
arkanes@10 41 local buffCount = 0
arkanes@4 42 for idx=1,99 do
arkanes@4 43 local frame = self.secureFrame:GetAttribute("child"..idx)
arkanes@10 44 if not (frame and frame:IsShown()) then break end
arkanes@10 45 buffCount = buffCount + 1
arkanes@6 46 if self.dirty then
arkanes@10 47 if self:BindBarToBuff(frame, unit) then break end
arkanes@9 48 end
arkanes@10 49 self:UpdateBarExpirationTime(frame)
arkanes@12 50 -- Don't forget to refresh shown tooltips
arkanes@12 51 if ( GameTooltip:IsOwned(frame) ) then
arkanes@12 52 GameTooltip:SetUnitAura(unit, idx, filter);
arkanes@12 53 end
arkanes@10 54 end
arkanes@10 55 -- temporary enchants
arkanes@10 56 -- TODO: The blizz secure aura header binds both temp enchants
arkanes@10 57 -- to the main hand. No support for cancelling weapon enchants
arkanes@10 58 -- until this gets fixed up
arkanes@10 59
arkanes@10 60 -- debuffs
arkanes@11 61 -- Since debuffs aren't cancellable, don't need to use the secure header
arkanes@11 62 -- for them. This could be rewritten to support useful features like
arkanes@11 63 -- sorting & scaling and stuff. Honestly, should at least be alphabetical.
arkanes@10 64 for idx=1,99 do
arkanes@10 65 local frame = self.debuffFrames[idx]
arkanes@10 66 if self.dirty then
arkanes@10 67 local name, rank, icon, stacks, debuffType, duration, expirationTime = UnitAura(unit, idx, "HARMFUL")
arkanes@10 68 if not name then
arkanes@10 69 -- out of debuffs, hide all the rest of them
arkanes@10 70 for jdx = idx, 99 do
arkanes@10 71 local bar = self.debuffFrames[jdx]
arkanes@10 72 if bar then bar:Hide() else break end
arkanes@10 73 end
arkanes@10 74 break
arkanes@10 75 end
arkanes@10 76 if not frame then
arkanes@10 77 frame = self:ConstructBar(nil, 1, 0, 0)
arkanes@10 78 self.debuffFrames[idx] = frame
arkanes@10 79 end
arkanes@10 80 self:SetBarAppearance(frame, name, icon, stacks, duration, expirationTime)
arkanes@10 81 frame:ClearAllPoints()
arkanes@10 82 -- position it under all the buffs, with a half-bar spacing
arkanes@10 83 frame:SetPoint("TOP", self.anchor, "BOTTOM", 0, (buffCount * -16) - 8)
arkanes@10 84 frame:Show()
arkanes@10 85 buffCount = buffCount + 1
arkanes@10 86 else
arkanes@10 87 -- not dirty, so no frame means we're done
arkanes@10 88 if not frame then break end
arkanes@6 89 end
arkanes@10 90 self:UpdateBarExpirationTime(frame)
arkanes@4 91 end
arkanes@6 92 self.dirty = nil
arkanes@4 93 end
arkanes@4 94
arkanes@0 95 function kbf:UNIT_AURA(event, unit)
arkanes@2 96 if unit ~= self.secureFrame:GetAttribute("unit") then return end
arkanes@6 97 self.dirty = true
arkanes@0 98 end
arkanes@0 99
arkanes@10 100 function kbf:UpdateBarExpirationTime(frame)
arkanes@10 101 if frame.expirationTime then
arkanes@10 102 local remaining = frame.expirationTime - GetTime()
arkanes@10 103 remaining = math.max(0, remaining)
arkanes@10 104 local perc = remaining / frame.duration
arkanes@10 105 frame.timertext:SetText(self:FormatTimeText(remaining))
arkanes@10 106 frame.statusbar:SetValue(remaining)
arkanes@10 107 end
arkanes@10 108 end
arkanes@10 109
arkanes@3 110 function kbf:BindBarToBuff(parentFrame, unit)
arkanes@3 111 local index = parentFrame:GetAttribute("index")
arkanes@3 112 local filter = parentFrame:GetAttribute("filter")
arkanes@10 113 local name, rank, icon, stacks, debuffType, duration, expirationTime,
arkanes@3 114 unitCaster, isStealable, shouldConsolidate, spellId = UnitAura(unit, index, filter)
arkanes@10 115 if not name then return end
arkanes@2 116 if not parentFrame.icon then
arkanes@2 117 self:ConstructBar(parentFrame)
arkanes@0 118 end
arkanes@10 119 self:SetBarAppearance(parentFrame, name, icon, stacks, duration, expirationTime)
arkanes@10 120 end
arkanes@10 121
arkanes@10 122 function kbf:SetBarAppearance(parentFrame, name, icon, stacks, duration, expirationTime)
arkanes@2 123 parentFrame.icon:SetNormalTexture(icon)
arkanes@0 124 if stacks and stacks > 0 then
arkanes@2 125 parentFrame.text:SetText(string.format("%s(%d)", name, stacks))
arkanes@0 126 else
arkanes@2 127 parentFrame.text:SetText(name)
arkanes@0 128 end
arkanes@4 129 parentFrame.timertext:SetText(self:FormatTimeText(duration))
arkanes@4 130 -- store duration information
arkanes@6 131 if duration and duration > 0 then
arkanes@4 132 parentFrame.expirationTime = expirationTime
arkanes@4 133 parentFrame.duration = duration
arkanes@4 134 parentFrame.statusbar:SetMinMaxValues(0, duration)
arkanes@4 135 else
arkanes@4 136 parentFrame.expirationTime = nil
arkanes@4 137 parentFrame.duration = 0
arkanes@4 138 parentFrame.statusbar:SetMinMaxValues(0,1)
arkanes@4 139 parentFrame.statusbar:SetValue(1)
arkanes@4 140 end
arkanes@4 141 end
arkanes@4 142
arkanes@4 143 function kbf:FormatTimeText(time)
arkanes@4 144 if not time or time == 0 then return "" end
arkanes@4 145 local timetext
arkanes@4 146 local h = floor(time/3600)
arkanes@4 147 local m = time - (h*3600)
arkanes@4 148 m = floor(m/60)
arkanes@4 149 local s = time - ((h*3600) + (m*60))
arkanes@4 150 if h > 0 then
arkanes@4 151 timetext = ("%d:%02d"):format(h, m)
arkanes@4 152 elseif m > 0 then
arkanes@4 153 timetext = string.format("%d:%02d", m, floor(s))
arkanes@4 154 elseif s < 10 then
arkanes@4 155 timetext = string.format("%1.1f", s)
arkanes@4 156 else
arkanes@4 157 timetext = string.format("%.0f", floor(s))
arkanes@4 158 end
arkanes@4 159 return timetext
arkanes@0 160 end
arkanes@0 161
arkanes@12 162 function KBF:OnEnter(button, motion)
arkanes@12 163 -- this is for the secure buttons, so use the attributes
arkanes@12 164 -- I'd like a better place to position this but it's funky for right now, handle it later
arkanes@12 165 GameTooltip:SetOwner(button, "ANCHOR_BOTTOMLEFT");
arkanes@12 166 GameTooltip:SetFrameLevel(button:GetFrameLevel() + 2);
arkanes@12 167 GameTooltip:SetUnitAura(button:GetAttribute("unit"), button:GetID(), button:GetAttribute("filter"));
arkanes@12 168
arkanes@12 169 end
arkanes@12 170
arkanes@1 171 -- creates a icon + statusbar bar
arkanes@10 172 function kbf:ConstructBar(frame, r, g, b)
arkanes@0 173 local texture = "Interface\\TargetingFrame\\UI-StatusBar"
arkanes@1 174 -- Because of secureframe suckiness, these height & width numbers
arkanes@1 175 -- have to be consistent with the stuff in KBF.xml
arkanes@0 176 local height = 16
arkanes@1 177 local width = 200 -- this is the width *without* the icon
arkanes@0 178 local font, _ style = GameFontHighlight:GetFont()
arkanes@10 179 local r = r or 0
arkanes@10 180 local g = g or 1
arkanes@10 181 local b = b or 0
arkanes@10 182 local bgcolor = {r, g, b, 0.5}
arkanes@10 183 local color = {r, g, b, 1}
arkanes@0 184 local fontsize = 11
arkanes@0 185 local timertextwidth = fontsize * 3.6
arkanes@0 186 local textcolor = {1, 1, 1, 1}
arkanes@0 187 local timertextcolor = {1, 1, 1, 1}
arkanes@10 188 if not frame then
arkanes@10 189 frame = CreateFrame("Button", nil, UIParent) -- the "top level" frame that represents the bar as a whole
arkanes@10 190 frame:SetHeight(16)
arkanes@10 191 frame:SetWidth(200 + 16)
arkanes@10 192 end
arkanes@1 193 local bar = frame
arkanes@0 194 bar.icon = CreateFrame("Button", nil, bar) -- the icon
arkanes@0 195 bar.statusbarbg = CreateFrame("StatusBar", nil, bar) -- the bars background
arkanes@0 196 bar.statusbar = CreateFrame("StatusBar", nil, bar) -- and the bars foreground
arkanes@0 197 bar.text = bar.statusbar:CreateFontString(nil, "OVERLAY") -- the label text
arkanes@0 198 bar.timertext = bar.statusbar:CreateFontString(nil, "OVERLAY") -- and the timer text
arkanes@0 199
arkanes@0 200 -- the icon
arkanes@0 201 bar.icon:ClearAllPoints()
arkanes@0 202 bar.icon:SetPoint("LEFT", bar, "LEFT", 0, 0)
arkanes@0 203 -- icons are square
arkanes@0 204 bar.icon:SetWidth(height)
arkanes@0 205 bar.icon:SetHeight(height)
arkanes@2 206 --bar.icon:EnableMouse(false)
arkanes@0 207 -- the status bar background & foreground
arkanes@0 208 local function setupStatusBar(sb, color)
arkanes@0 209 sb:ClearAllPoints()
arkanes@0 210 sb:SetHeight(height)
arkanes@0 211 sb:SetWidth(width)
arkanes@0 212 -- offset the height of the frame on the x-axis for the icon.
arkanes@0 213 sb:SetPoint("TOPLEFT", bar, "TOPLEFT", height, 0)
arkanes@0 214 sb:SetStatusBarTexture(texture)
arkanes@0 215 sb:GetStatusBarTexture():SetVertTile(false)
arkanes@0 216 sb:GetStatusBarTexture():SetHorizTile(false)
arkanes@0 217 sb:SetStatusBarColor(unpack(color))
arkanes@0 218 sb:SetMinMaxValues(0,1)
arkanes@0 219 sb:SetValue(1)
arkanes@0 220 end
arkanes@0 221 setupStatusBar(bar.statusbarbg, bgcolor)
arkanes@0 222 setupStatusBar(bar.statusbar, color)
arkanes@0 223 bar.statusbarbg:SetFrameLevel(bar.statusbarbg:GetFrameLevel()-1) -- make sure the bg frame stays in the back
arkanes@0 224 -- timer text
arkanes@0 225 bar.timertext:SetFontObject(GameFontHighlight)
arkanes@0 226 bar.timertext:SetFont(GameFontHighlight:GetFont())
arkanes@0 227 bar.timertext:SetHeight(height)
arkanes@0 228 bar.timertext:SetWidth(timertextwidth)
arkanes@0 229 bar.timertext:SetPoint("LEFT", bar.statusbar, "LEFT", 0, 0)
arkanes@0 230 bar.timertext:SetJustifyH("RIGHT")
arkanes@0 231 bar.timertext:SetText("time")
arkanes@0 232 bar.timertext:SetTextColor(timertextcolor[1], timertextcolor[2], timertextcolor[3], timertextcolor[4])
arkanes@0 233
arkanes@0 234 -- and the label text
arkanes@0 235 bar.text:SetFontObject(GameFontHighlight)
arkanes@0 236 bar.text:SetFont(GameFontHighlight:GetFont())
arkanes@0 237 bar.text:SetHeight(height)
arkanes@0 238 bar.text:SetWidth((width - timertextwidth) *.9)
arkanes@0 239 bar.text:SetPoint("RIGHT", bar.statusbar, "RIGHT", 0, 0)
arkanes@0 240 bar.text:SetJustifyH("LEFT")
arkanes@0 241 bar.text:SetText("text")
arkanes@0 242 bar.text:SetTextColor(textcolor[1], textcolor[2], textcolor[3], textcolor[4])
arkanes@0 243
arkanes@0 244 bar:EnableMouse(true)
arkanes@0 245 return bar
arkanes@0 246 end
arkanes@0 247
arkanes@0 248 function kbf:CreateAnchorFrame()
arkanes@0 249 -- give it a name so it'll remember its position
arkanes@0 250 local anchor = CreateFrame("FRAME", "KBFAnchorFrame", UIParent)
arkanes@0 251 anchor:SetClampedToScreen(true)
arkanes@0 252 anchor:SetBackdrop({bgFile = "Interface/Tooltips/UI-Tooltip-Background",
arkanes@0 253 edgeFile = "Interface/Tooltips/UI-Tooltip-Border",
arkanes@0 254 tile = true, tileSize = 16, edgeSize = 16,
arkanes@0 255 insets = { left = 4, right = 4, top = 4, bottom = 4 },
arkanes@0 256 })
arkanes@0 257 local text = anchor:CreateFontString(nil, "OVERLAY") -- the label text
arkanes@0 258 text:SetFontObject(GameFontHighlight)
arkanes@0 259 text:SetFont(GameFontHighlight:GetFont())
arkanes@0 260 text:SetPoint("TOPLEFT", anchor, "TOPLEFT", 0, 0)
arkanes@0 261 text:SetPoint("BOTTOMRIGHT", anchor, "BOTTOMRIGHT", 0, 0)
arkanes@0 262 text:SetText("KBF ANCHOR")
arkanes@10 263 anchor:SetWidth(200 + 16 + 8)
arkanes@10 264 anchor:SetHeight(16 + 8)
arkanes@0 265 -- movability
arkanes@0 266 anchor:EnableMouse(true)
arkanes@0 267 anchor:SetMovable(true)
arkanes@0 268 anchor:RegisterForDrag("LeftButton")
arkanes@0 269 anchor:SetScript("OnDragStart", anchor.StartMoving)
arkanes@0 270 anchor:SetScript("OnDragStop", anchor.StopMovingOrSizing)
arkanes@0 271 anchor:ClearAllPoints()
arkanes@0 272 anchor:SetPoint("CENTER", UIParent, "CENTER", 0, 0)
arkanes@0 273 anchor:Hide()
arkanes@0 274
arkanes@0 275 local frame = CreateFrame("FRAME", "KBFBuffFrame", anchor, "SecureAuraHeaderTemplate")
arkanes@0 276 --local frame = anchor
arkanes@0 277 frame:SetAttribute("filter", "HELPFUL")
arkanes@0 278 frame:SetAttribute("template", "KBFSecureUnitAuraTemplate")
arkanes@3 279 frame:SetAttribute("point", "TOP")
arkanes@0 280 frame:SetAttribute("wrapAfter", 100) -- required due to bugs in secure header
arkanes@0 281 frame:SetAttribute("consolidateTo", nil)
arkanes@2 282 frame:SetAttribute("xOffset", 0)
arkanes@3 283 frame:SetAttribute("yOffset", -16)
arkanes@3 284 frame:SetAttribute("minWidth", 216)
arkanes@3 285 frame:SetAttribute("minHeight", 16)
arkanes@2 286 frame:SetAttribute("unit", "player") -- TODO: figure out the vehicle swapping stuff
arkanes@3 287 frame:SetAttribute("sortMethod", "NAME")
arkanes@3 288 frame:SetAttribute("sortOrder", "-")
arkanes@10 289 -- TODO: SecureAuraHeader doesn't correcltly implement the temp enchants
arkanes@10 290 --frame:SetAttribute("weaponTemplate", "KBFSecureUnitAuraTemplate")
arkanes@10 291 --frame:SetAttribute("includeWeapons", 1)
arkanes@3 292 frame:SetPoint("TOP", anchor, "BOTTOM", 0, 0)
arkanes@5 293 frame:Show() -- has to be shown, otherwise the child frames don't show
arkanes@0 294 self.secureFrame = frame
arkanes@0 295 return anchor
arkanes@0 296 end