Mercurial > wow > reaction

annotate Button.lua @ 245:65f2805957a0

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
No real reason to store some of the code in a subdirectory.
author Flick
date Sat, 26 Mar 2011 12:35:08 -0700
parents classes/Button.lua@dcdc0235d489
children 46b59a9ded76
rev   line source
flickerstreak@122 1 --[[
flickerstreak@122 2 ReAction Button base class
flickerstreak@122 3 --]]
flickerstreak@122 4
flickerstreak@122 5 -- local imports
flickerstreak@175 6 local addonName, addonTable = ...
flickerstreak@175 7 local ReAction = addonTable.ReAction
flickerstreak@122 8 local L = ReAction.L
flickerstreak@183 9 local LKB = ReAction.LKB
flickerstreak@122 10 local _G = _G
flickerstreak@122 11 local CreateFrame = CreateFrame
flickerstreak@122 12 local GetBindingKey = GetBindingKey
flickerstreak@122 13 local format = string.format
flickerstreak@122 14
flickerstreak@122 15 -- private
flickerstreak@122 16 local trash = CreateFrame("Frame")
flickerstreak@122 17 local frameList = { }
flickerstreak@122 18 local idPools = { }
flickerstreak@122 19
flickerstreak@122 20 local function kb_onEnter( frame )
flickerstreak@183 21 LKB:Set(frame)
flickerstreak@122 22 end
flickerstreak@122 23
flickerstreak@122 24 -- Button class
flickerstreak@218 25 local buttonTypeID = "Button"
flickerstreak@218 26 local Button = {
flickerstreak@218 27 defaultBarConfig = {
flickerstreak@218 28 type = buttonTypeID,
flickerstreak@218 29 btnWidth = 36,
flickerstreak@218 30 btnHeight = 36,
flickerstreak@218 31 btnRows = 1,
flickerstreak@218 32 btnColumns = 12,
flickerstreak@218 33 spacing = 3
flickerstreak@218 34 },
flickerstreak@218 35 barType = L["Button Bar"]
flickerstreak@218 36 }
flickerstreak@122 37
flickerstreak@122 38 ReAction.Button = Button -- export to ReAction
flickerstreak@122 39
flickerstreak@129 40 function Button:New( name, config, bar, idx, inherits, buttonType )
flickerstreak@122 41 buttonType = buttonType or "CheckButton"
flickerstreak@122 42
flickerstreak@122 43 -- create new self
flickerstreak@122 44 self = setmetatable(
flickerstreak@122 45 {
flickerstreak@122 46 bar = bar,
flickerstreak@122 47 idx = idx,
flickerstreak@129 48 config = config,
flickerstreak@122 49 name = name,
flickerstreak@122 50 },
flickerstreak@122 51 { __index = self } )
flickerstreak@122 52
flickerstreak@122 53 -- have to recycle frames with the same name: CreateFrame() doesn't overwrite
flickerstreak@122 54 -- existing globals. Can't set to nil in the global because it's then tainted.
flickerstreak@122 55 -- Caller is responsible for ensuring global uniqueness of names.
flickerstreak@122 56 local f = name and frameList[name]
flickerstreak@122 57 if f then
flickerstreak@122 58 f:SetParent(bar:GetFrame())
flickerstreak@122 59 else
flickerstreak@122 60 f = CreateFrame(buttonType, name, bar:GetFrame(), inherits)
flickerstreak@122 61 if name then
flickerstreak@122 62 frameList[name] = f
flickerstreak@122 63 end
flickerstreak@122 64 end
flickerstreak@122 65
flickerstreak@122 66 self.frame = f
flickerstreak@122 67
flickerstreak@131 68 local frames = { }
flickerstreak@131 69 self.frames = frames
flickerstreak@131 70 frames.icon = _G[name.."Icon"]
flickerstreak@131 71 frames.flash = _G[name.."Flash"]
flickerstreak@131 72 frames.hotkey = _G[name.."HotKey"]
flickerstreak@131 73 frames.count = _G[name.."Count"]
flickerstreak@131 74 frames.name = _G[name.."Name"]
flickerstreak@131 75 frames.border = _G[name.."Border"]
flickerstreak@131 76 frames.cooldown = _G[name.."Cooldown"]
flickerstreak@131 77 frames.normalTexture = _G[name.."NormalTexture"]
flickerstreak@131 78
flickerstreak@122 79 if config then
flickerstreak@122 80 config.name = name
flickerstreak@122 81 end
flickerstreak@122 82
flickerstreak@122 83 -- install LibKeyBound handlers onto frame
flickerstreak@122 84 function f:GetActionName()
flickerstreak@122 85 return format("%s:%s", bar:GetName(), idx)
flickerstreak@122 86 end
flickerstreak@122 87
flickerstreak@122 88 local clickBinding = format("CLICK %s:LeftButton", name)
flickerstreak@122 89 function f:GetHotkey()
flickerstreak@183 90 return LKB:ToShortKey(GetBindingKey(clickBinding))
flickerstreak@122 91 end
flickerstreak@122 92
flickerstreak@122 93 return self
flickerstreak@122 94 end
flickerstreak@122 95
flickerstreak@122 96 function Button:Destroy()
flickerstreak@126 97 local f = self:GetFrame()
flickerstreak@132 98 f:UnregisterAllEvents()
flickerstreak@132 99 self:ReleaseActionID(self:GetActionID())
flickerstreak@126 100 if f then
flickerstreak@126 101 f:Hide()
flickerstreak@126 102 f:SetParent(trash)
flickerstreak@126 103 f:ClearAllPoints()
flickerstreak@126 104 end
flickerstreak@122 105 end
flickerstreak@122 106
flickerstreak@128 107 function Button:GetBar()
flickerstreak@128 108 return self.bar
flickerstreak@128 109 end
flickerstreak@128 110
flickerstreak@122 111 function Button:GetFrame()
flickerstreak@122 112 return self.frame
flickerstreak@122 113 end
flickerstreak@122 114
Flick@234 115 function Button:GetIndex()
Flick@234 116 return self.idx
Flick@234 117 end
Flick@234 118
flickerstreak@122 119 function Button:GetName()
flickerstreak@122 120 return self.name
flickerstreak@122 121 end
flickerstreak@122 122
flickerstreak@218 123 function Button:GetDefaultBarConfig()
flickerstreak@218 124 return self.defaultBarConfig
flickerstreak@218 125 end
flickerstreak@218 126
flickerstreak@218 127 function Button:GetBarType()
flickerstreak@218 128 return self.barType
flickerstreak@218 129 end
flickerstreak@218 130
Flick@231 131 function Button:GetButtonTypeID()
Flick@231 132 return self.buttonTypeID
Flick@231 133 end
Flick@231 134
flickerstreak@122 135 function Button:GetConfig()
flickerstreak@122 136 return self.config
flickerstreak@122 137 end
flickerstreak@122 138
flickerstreak@122 139 function Button:GetActionID()
flickerstreak@122 140 -- derived classes should override this
flickerstreak@122 141 return nil
flickerstreak@122 142 end
flickerstreak@122 143
flickerstreak@122 144 function Button:SetActionIDPool( poolID, maxID )
flickerstreak@122 145 self.actionPoolID = poolID
flickerstreak@122 146 self.actionMaxID = maxID
flickerstreak@122 147 end
flickerstreak@122 148
Flick@234 149 function Button:SetupBar( bar )
Flick@234 150 local config = bar:GetConfig()
Flick@234 151 if not config.buttons then
Flick@234 152 config.buttons = { }
Flick@234 153 end
Flick@234 154 local btnCfg = config.buttons
Flick@234 155
Flick@234 156 local r, c = bar:GetButtonGrid()
Flick@234 157 local n = r*c
Flick@235 158 local cfgN = n
Flick@234 159
Flick@234 160 local hint = nil
Flick@234 161 local i = 1
Flick@234 162 repeat
Flick@234 163 local b = bar:GetButton(i)
Flick@234 164 if b then
Flick@234 165 if i > n then
Flick@234 166 bar:RemoveButton(b)
Flick@234 167 b:Destroy()
Flick@235 168 if i > cfgN then
Flick@235 169 btnCfg[i] = nil
Flick@235 170 end
Flick@234 171 else
Flick@234 172 b:Refresh()
Flick@234 173 hint = b:GetActionID()
Flick@234 174 end
Flick@234 175 elseif i <= n then
Flick@234 176 local cfg = btnCfg[i] or { }
Flick@234 177 local success, r = pcall(self.New, self, cfg, bar, i, hint) -- note call semantics for derived class constructors
Flick@234 178 if success and r then
Flick@234 179 b = r
Flick@234 180 bar:AddButton(i,b)
Flick@234 181 btnCfg[i] = cfg
Flick@234 182 b:Refresh()
Flick@234 183 hint = b:GetActionID()
Flick@234 184 else
Flick@234 185 n = i - 1
Flick@234 186 if not success then
Flick@236 187 bar:ClipNButtons(n)
Flick@236 188 cfgN = n
Flick@234 189 geterrorhandler()(r)
Flick@234 190 end
Flick@234 191 end
Flick@234 192 end
Flick@234 193 i = i + 1
Flick@234 194 until b == nil
Flick@234 195 end
Flick@234 196
flickerstreak@122 197 function Button:AcquireActionID( id, hint, unique )
flickerstreak@122 198 local poolID = self.actionPoolID
flickerstreak@122 199 local maxID = self.actionMaxID
flickerstreak@122 200 if not poolID or not maxID then
flickerstreak@122 201 error("AcquireActionID: must setup pool first with SetActionIDPool")
flickerstreak@122 202 end
flickerstreak@123 203 local pool = idPools[poolID]
flickerstreak@122 204 if not pool then
flickerstreak@122 205 pool = { nWraps = 0, useCount = { } }
flickerstreak@122 206 for i = 1, maxID do
flickerstreak@122 207 pool.useCount[i] = 0
flickerstreak@122 208 end
flickerstreak@123 209 idPools[poolID] = pool
flickerstreak@122 210 end
flickerstreak@122 211 local useCount = pool.useCount
flickerstreak@122 212 if id == nil then
flickerstreak@122 213 repeat
flickerstreak@126 214 local nWraps = pool.nWraps or 0
flickerstreak@126 215 if hint and (useCount[hint] == nil or useCount[hint] == nWraps) then
flickerstreak@122 216 id = hint
flickerstreak@122 217 else
flickerstreak@122 218 local start = hint or 1
flickerstreak@122 219 for i = start, maxID do
flickerstreak@122 220 if useCount[i] == nil or useCount[i] == nWraps then
flickerstreak@122 221 id = i
flickerstreak@122 222 break
flickerstreak@122 223 end
flickerstreak@122 224 end
flickerstreak@122 225 if not id then
flickerstreak@122 226 for i = 1, start do
flickerstreak@122 227 if useCount[i] == nil or useCount[i] == nWraps then
flickerstreak@122 228 id = i
flickerstreak@122 229 break
flickerstreak@122 230 end
flickerstreak@122 231 end
flickerstreak@122 232 end
flickerstreak@122 233 end
flickerstreak@122 234 if id == nil then
flickerstreak@122 235 if unique then
flickerstreak@122 236 return nil
flickerstreak@122 237 end
flickerstreak@122 238 pool.nWraps = nWraps + 1
flickerstreak@122 239 end
flickerstreak@126 240 until id ~= nil
flickerstreak@122 241 end
flickerstreak@122 242 useCount[id] = (useCount[id] or 0) + 1
flickerstreak@122 243 return id
flickerstreak@122 244 end
flickerstreak@122 245
flickerstreak@122 246 function Button:ReleaseActionID( id )
flickerstreak@122 247 local poolID = self.actionPoolID
flickerstreak@122 248 if not poolID then
flickerstreak@122 249 error("ReleaseActionID: must setup pool first with SetActionIDPool")
flickerstreak@122 250 end
flickerstreak@123 251 local pool = idPools[poolID]
flickerstreak@122 252 if pool and id and pool.useCount[id] then
flickerstreak@122 253 pool.useCount[id] = pool.useCount[id] - 1
flickerstreak@122 254 pool.nWraps = min(pool.useCount[id], pool.nWraps)
flickerstreak@122 255 end
flickerstreak@122 256 end
flickerstreak@122 257
flickerstreak@122 258 function Button:Refresh()
flickerstreak@129 259 local f = self:GetFrame()
flickerstreak@129 260 self.bar:PlaceButton( self, f:GetWidth(), f:GetHeight() )
flickerstreak@122 261 end
flickerstreak@122 262
flickerstreak@122 263 function Button:SetKeybindMode( mode )
flickerstreak@122 264 local f = self.frame
flickerstreak@122 265 if mode then
flickerstreak@122 266 self.oldOnEnter = f:GetScript("OnEnter")
flickerstreak@122 267 f:SetScript("OnEnter", kb_onEnter)
flickerstreak@124 268 elseif self.oldOnEnter then
flickerstreak@122 269 f:SetScript("OnEnter", self.oldOnEnter)
flickerstreak@122 270 self.oldOnEnter = nil
flickerstreak@122 271 end
flickerstreak@128 272 self:ShowGridTemp(mode)
flickerstreak@122 273 self:UpdateKeybindModeDisplay( mode )
flickerstreak@122 274 end
flickerstreak@122 275
flickerstreak@122 276 function Button:UpdateKeybindModeDisplay( mode )
flickerstreak@131 277 local border = self.frames.border or _G[format("%sBorder",tostring(self:GetName()))]
flickerstreak@131 278 if border then
flickerstreak@122 279 if mode then
flickerstreak@183 280 border:SetVertexColor(LKB:GetColorKeyBoundMode())
flickerstreak@131 281 border:Show()
flickerstreak@122 282 else
flickerstreak@131 283 border:Hide()
flickerstreak@122 284 end
flickerstreak@122 285 end
flickerstreak@122 286 end
flickerstreak@122 287
flickerstreak@122 288 function Button:UpdateHotkey( hotkey )
flickerstreak@131 289 hotkey = hotkey or self.frames.hotkey
flickerstreak@122 290 if not hotkey then
flickerstreak@129 291 hotkey = _G[self:GetName().."HotKey"]
flickerstreak@131 292 self.frames.hotkey = hotkey
flickerstreak@122 293 end
flickerstreak@122 294 if hotkey then
flickerstreak@122 295 local txt = self.frame:GetHotkey()
flickerstreak@122 296 hotkey:SetText( txt )
flickerstreak@122 297 if txt == nil or txt == "" then
flickerstreak@122 298 hotkey:Hide()
flickerstreak@122 299 else
flickerstreak@122 300 hotkey:Show()
flickerstreak@122 301 end
flickerstreak@122 302 end
flickerstreak@122 303 end
flickerstreak@122 304
flickerstreak@122 305 function Button:GetActionIDLabel( create )
flickerstreak@129 306 local f = self:GetFrame()
flickerstreak@122 307 if not f.actionIDLabel and create then
flickerstreak@122 308 local label = f:CreateFontString(nil,"OVERLAY","GameFontNormalLarge")
flickerstreak@122 309 label:SetAllPoints()
flickerstreak@122 310 label:SetJustifyH("CENTER")
flickerstreak@122 311 label:SetShadowColor(0,0,0,1)
flickerstreak@122 312 label:SetShadowOffset(2,-2)
flickerstreak@122 313 f.actionIDLabel = label -- store the label with the frame for recycling
flickerstreak@122 314 end
flickerstreak@122 315 return f.actionIDLabel
flickerstreak@122 316 end
flickerstreak@122 317
flickerstreak@122 318 function Button:UpdateActionIDLabel( show )
flickerstreak@122 319 local label = self:GetActionIDLabel( show )
flickerstreak@122 320 if label then
flickerstreak@122 321 if show then
flickerstreak@122 322 local id = self:GetActionID()
flickerstreak@122 323 if id then
flickerstreak@122 324 label:SetText(tostring(id))
flickerstreak@122 325 label:Show()
flickerstreak@122 326 return
flickerstreak@122 327 end
flickerstreak@122 328 end
flickerstreak@122 329 label:Hide()
flickerstreak@122 330 end
flickerstreak@122 331 end
flickerstreak@122 332
flickerstreak@122 333 function Button:SetNormalVertexColor( r, g, b, a )
flickerstreak@213 334 if ReAction.LBF then
flickerstreak@213 335 ReAction.LBF:SetNormalVertexColor(self:GetFrame(), r, g, b, a)
flickerstreak@122 336 else
flickerstreak@129 337 self:GetFrame():GetNormalTexture():SetVertexColor(r,g,b,a)
flickerstreak@122 338 end
flickerstreak@122 339 end
flickerstreak@122 340
flickerstreak@122 341 function Button:GetNormalVertexColor()
flickerstreak@213 342 if ReAction.LBF then
flickerstreak@213 343 return ReAction.LBF:GetNormalVertexColor(self:GetFrame())
flickerstreak@122 344 else
flickerstreak@129 345 return self:GetFrame():GetNormalTexture():GetVertexColor()
flickerstreak@122 346 end
flickerstreak@122 347 end
flickerstreak@122 348
flickerstreak@122 349 function Button:UpdateShowGrid()
flickerstreak@122 350 -- does nothing by default
flickerstreak@122 351 end
flickerstreak@128 352
flickerstreak@128 353 function Button:ShowGridTemp(show)
flickerstreak@128 354 -- does nothing by default
flickerstreak@128 355 end
flickerstreak@128 356
flickerstreak@128 357 function Button:ShowGrid(show)
flickerstreak@128 358 -- does nothing by default
flickerstreak@128 359 end