Mercurial > wow > kbf

annotate KBF.lua @ 11:76650e326a57

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