Mercurial > wow > reaction

annotate Bar.lua @ 283:f2ed8a8e2320

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add editor sliders for button grid and size
author Flick
date Thu, 12 May 2011 17:16:43 -0700
parents 5b9c0164a491
children d931fa418e17
rev   line source
Flick@276 1 local _, ns = ...
Flick@276 2 local ReAction = ns.ReAction
flickerstreak@122 3 local L = ReAction.L
flickerstreak@183 4 local LKB = ReAction.LKB
flickerstreak@122 5 local _G = _G
flickerstreak@122 6 local CreateFrame = CreateFrame
flickerstreak@122 7 local floor = math.floor
flickerstreak@122 8 local fmod = math.fmod
flickerstreak@122 9 local format = string.format
Flick@276 10 local tfetch = ns.tfetch
Flick@276 11 local tbuild = ns.tbuild
Flick@276 12 local fieldsort = ns.fieldsort
flickerstreak@122 13
flickerstreak@193 14 local LSG = LibStub("ReAction-LibShowActionGrid-1.0")
flickerstreak@147 15
flickerstreak@155 16 ---- Secure snippets ----
flickerstreak@155 17 local _reaction_init =
flickerstreak@155 18 [[
flickerstreak@155 19 anchorKeys = newtable("point","relPoint","x","y")
flickerstreak@155 20
flickerstreak@155 21 state = nil
flickerstreak@155 22 set_state = nil
flickerstreak@155 23 state_override = nil
flickerstreak@156 24 unit_exists = nil
flickerstreak@155 25
flickerstreak@155 26 showAll = false
flickerstreak@155 27 hidden = false
flickerstreak@155 28
flickerstreak@155 29 defaultAlpha = 1.0
flickerstreak@155 30 defaultScale = 1.0
flickerstreak@155 31 defaultAnchor = newtable()
flickerstreak@155 32
flickerstreak@155 33 activeStates = newtable()
flickerstreak@155 34 settings = newtable()
flickerstreak@155 35 extensions = newtable()
flickerstreak@155 36 ]]
flickerstreak@155 37
flickerstreak@155 38 local _reaction_refresh =
flickerstreak@155 39 [[
flickerstreak@155 40 local oldState = state
flickerstreak@155 41 state = state_override or set_state or state
flickerstreak@155 42
flickerstreak@156 43 local hide = nil
flickerstreak@155 44 if state then
flickerstreak@155 45 local settings = settings[state]
flickerstreak@155 46 if settings then
flickerstreak@155 47 -- show/hide
flickerstreak@156 48 hide = settings.hide
flickerstreak@155 49 -- re-anchor
flickerstreak@155 50 local old_anchor = activeStates.anchor
flickerstreak@155 51 activeStates.anchor = settings.anchorEnable and state
flickerstreak@155 52 if old_anchor ~= activeStates.anchor or not set_state then
flickerstreak@155 53 if activeStates.anchor then
flickerstreak@155 54 if settings.anchorPoint then
flickerstreak@155 55 self:ClearAllPoints()
flickerstreak@155 56 local f = self:GetAttribute("frameref-anchor-"..state)
flickerstreak@155 57 if f then
flickerstreak@155 58 self:SetPoint(settings.anchorPoint, f, settings.anchorRelPoint, settings.anchorX, settings.anchorY)
flickerstreak@155 59 end
flickerstreak@155 60 end
flickerstreak@155 61 elseif defaultAnchor.point then
flickerstreak@155 62 self:ClearAllPoints()
flickerstreak@155 63 self:SetPoint(defaultAnchor.point, defaultAnchor.frame,
flickerstreak@155 64 defaultAnchor.relPoint, defaultAnchor.x, defaultAnchor.y)
flickerstreak@155 65 end
flickerstreak@155 66 end
flickerstreak@155 67 -- re-scale
flickerstreak@155 68 local old_scale = activeStates.scale
flickerstreak@155 69 activeStates.scale = settings.enableScale and state
flickerstreak@155 70 if old_scale ~= activeStates.scale or not set_state then
flickerstreak@155 71 self:SetScale(activeStates.scale and settings.scale or defaultScale)
flickerstreak@155 72 end
flickerstreak@155 73 -- alpha
flickerstreak@155 74 local old_alpha = activeStates.alpha
flickerstreak@155 75 activeStates.alpha = settings.enableAlpha and state
flickerstreak@155 76 if old_alpha ~= activeStates.alpha or not set_state then
flickerstreak@155 77 self:SetAlpha(activeStates.alpha and settings.alpha or defaultAlpha)
flickerstreak@155 78 end
flickerstreak@155 79 end
flickerstreak@155 80 end
flickerstreak@155 81
flickerstreak@156 82 -- hide if state or unit_exists says to
flickerstreak@156 83 hide = not showAll and (hide or unithide)
flickerstreak@156 84 if hide ~= hidden then
flickerstreak@156 85 hidden = hide
flickerstreak@156 86 if hide then
flickerstreak@156 87 self:Hide()
flickerstreak@156 88 else
flickerstreak@156 89 self:Show()
flickerstreak@156 90 end
flickerstreak@156 91 end
flickerstreak@156 92
flickerstreak@155 93 for _, attr in pairs(extensions) do
flickerstreak@155 94 control:RunAttribute(attr)
flickerstreak@155 95 end
flickerstreak@155 96
flickerstreak@155 97 control:ChildUpdate()
flickerstreak@155 98
flickerstreak@155 99 if showAll then
flickerstreak@155 100 control:CallMethod("UpdateHiddenLabel", state and settings[state] and settings[state].hide)
flickerstreak@155 101 end
flickerstreak@155 102
flickerstreak@155 103 if oldState ~= state then
flickerstreak@155 104 control:CallMethod("StateRefresh", state)
flickerstreak@155 105 end
flickerstreak@155 106 ]]
flickerstreak@155 107
flickerstreak@155 108 local _onstate_reaction = -- function( self, stateid, newstate )
flickerstreak@155 109 [[
flickerstreak@155 110 set_state = newstate
flickerstreak@155 111 ]] .. _reaction_refresh
flickerstreak@155 112
flickerstreak@155 113 local _onstate_showgrid = -- function( self, stateid, newstate )
flickerstreak@155 114 [[
flickerstreak@155 115 control:ChildUpdate(stateid,newstate)
flickerstreak@155 116 control:CallMethod("UpdateShowGrid")
flickerstreak@155 117 ]]
flickerstreak@155 118
flickerstreak@155 119 local _onstate_unitexists = -- function( self, stateid, newstate )
flickerstreak@155 120 [[
flickerstreak@159 121 unithide = not newstate or newstate == "hide"
flickerstreak@155 122 ]] .. _reaction_refresh
flickerstreak@155 123
flickerstreak@155 124 local _onclick = -- function( self, button, down )
flickerstreak@155 125 [[
flickerstreak@155 126 if state_override == button then
flickerstreak@155 127 state_override = nil -- toggle
flickerstreak@155 128 else
flickerstreak@155 129 state_override = button
flickerstreak@155 130 end
flickerstreak@155 131 ]] .. _reaction_refresh
flickerstreak@155 132
flickerstreak@157 133 -- For reference
flickerstreak@157 134 -- the option field names must match the field names of the options table, below
flickerstreak@157 135 local stateProperties = {
flickerstreak@157 136 hide = true,
flickerstreak@157 137 --keybindState = true, TODO: broken
flickerstreak@157 138 anchorEnable = true,
flickerstreak@157 139 anchorFrame = true,
flickerstreak@157 140 anchorPoint = true,
flickerstreak@157 141 anchorRelPoint = true,
flickerstreak@157 142 anchorX = true,
flickerstreak@157 143 anchorY = true,
flickerstreak@157 144 enableScale = true,
flickerstreak@157 145 scale = true,
flickerstreak@157 146 enableAlpha = true,
flickerstreak@157 147 alpha = true,
flickerstreak@157 148 }
flickerstreak@157 149
flickerstreak@157 150
flickerstreak@155 151
flickerstreak@147 152 ---- Bar class ----
flickerstreak@122 153 local Bar = { }
flickerstreak@147 154 local frameList = { }
flickerstreak@122 155
flickerstreak@122 156 ReAction.Bar = Bar -- export to ReAction
flickerstreak@122 157
Flick@234 158 function Bar:New( name, config, buttonClass )
flickerstreak@122 159 if type(config) ~= "table" then
flickerstreak@122 160 error("ReAction.Bar: config table required")
flickerstreak@122 161 end
flickerstreak@122 162
flickerstreak@122 163 -- create new self
flickerstreak@122 164 self = setmetatable(
flickerstreak@122 165 {
Flick@234 166 config = config,
Flick@234 167 name = name,
Flick@234 168 buttons = { },
Flick@234 169 buttonClass = buttonClass,
Flick@234 170 width = config.width or 480,
Flick@234 171 height = config.height or 40,
flickerstreak@122 172 },
flickerstreak@147 173 {__index = self} )
flickerstreak@122 174
flickerstreak@122 175 -- The frame type is 'Button' in order to have an OnClick handler. However, the frame itself is
flickerstreak@122 176 -- not mouse-clickable by the user.
flickerstreak@122 177 local parent = config.parent and (ReAction:GetBar(config.parent) or _G[config.parent]) or UIParent
flickerstreak@147 178 name = name and "ReAction-"..name
flickerstreak@147 179 local f = name and frameList[name]
flickerstreak@147 180 if not f then
flickerstreak@147 181 f = CreateFrame("Button", name, parent, "SecureHandlerStateTemplate, SecureHandlerClickTemplate")
flickerstreak@147 182 if name then
flickerstreak@147 183 frameList[name] = f
flickerstreak@147 184 end
flickerstreak@147 185 end
flickerstreak@122 186 f:SetFrameStrata("MEDIUM")
flickerstreak@122 187 f:SetWidth(self.width)
flickerstreak@146 188 f:SetHeight(self.height)
flickerstreak@122 189 f:SetAlpha(config.alpha or 1.0)
flickerstreak@122 190 f:Show()
flickerstreak@122 191 f:EnableMouse(false)
flickerstreak@122 192 f:SetClampedToScreen(true)
flickerstreak@171 193 LSG:AddFrame(f)
flickerstreak@122 194
flickerstreak@155 195 -- secure handlers
flickerstreak@155 196 f:Execute(_reaction_init)
flickerstreak@155 197 f:SetAttribute("_onstate-reaction", _onstate_reaction)
flickerstreak@155 198 f:SetAttribute("_onstate-showgrid", _onstate_showgrid)
flickerstreak@155 199 f:SetAttribute("_onstate-unitexists", _onstate_unitexists)
flickerstreak@155 200 f:SetAttribute("_onclick", _onclick)
flickerstreak@155 201
flickerstreak@155 202 -- secure handler CallMethod()s
flickerstreak@155 203 f.UpdateShowGrid = function() self:UpdateShowGrid() end
flickerstreak@155 204 f.StateRefresh = function() self:RefreshControls() end
flickerstreak@155 205 f.UpdateHiddenLabel = function(f,hidden) self:SetLabelSubtext(hidden and L["Hidden"]) end
flickerstreak@122 206
flickerstreak@122 207 -- Override the default frame accessor to provide strict read-only access
flickerstreak@122 208 function self:GetFrame()
flickerstreak@122 209 return f
flickerstreak@122 210 end
flickerstreak@122 211
flickerstreak@122 212 self:ApplyAnchor()
flickerstreak@147 213 self:SetConfigMode(ReAction:GetConfigMode())
flickerstreak@147 214 self:SetKeybindMode(ReAction:GetKeybindMode())
flickerstreak@147 215
flickerstreak@213 216 if ReAction.LBF then
flickerstreak@213 217 local g = ReAction.LBF:Group(L["ReAction"], self.name)
flickerstreak@213 218 self.config.ButtonFacade = self.config.ButtonFacade or {
flickerstreak@213 219 skinID = "Blizzard",
flickerstreak@213 220 backdrop = true,
flickerstreak@213 221 gloss = 0,
flickerstreak@213 222 colors = {},
flickerstreak@213 223 }
flickerstreak@213 224 local c = self.config.ButtonFacade
flickerstreak@213 225 g:Skin(c.skinID, c.gloss, c.backdrop, c.colors)
flickerstreak@213 226 self.LBFGroup = g
flickerstreak@213 227 end
flickerstreak@213 228
flickerstreak@122 229 ReAction.RegisterCallback(self, "OnConfigModeChanged")
flickerstreak@122 230
Flick@234 231 buttonClass:SetupBar(self)
Flick@242 232 self:ApplyStates()
Flick@234 233
flickerstreak@122 234 return self
flickerstreak@122 235 end
flickerstreak@122 236
flickerstreak@122 237 function Bar:Destroy()
flickerstreak@122 238 local f = self:GetFrame()
Flick@242 239 self:CleanupStates()
Flick@234 240 for idx, b in self:IterateButtons() do
Flick@234 241 b:Destroy()
Flick@234 242 end
flickerstreak@122 243 f:UnregisterAllEvents()
flickerstreak@125 244 self:ShowControls(false)
flickerstreak@122 245 ReAction.UnregisterAllCallbacks(self)
flickerstreak@171 246 LKB.UnregisterAllCallbacks(self)
flickerstreak@213 247 if self.LBFGroup then
flickerstreak@213 248 self.LBFGroup:Delete(true)
flickerstreak@213 249 end
flickerstreak@171 250 LSG:RemoveFrame(f)
flickerstreak@122 251 f:SetParent(UIParent)
flickerstreak@122 252 f:ClearAllPoints()
flickerstreak@147 253 f:Hide()
flickerstreak@122 254 end
flickerstreak@122 255
flickerstreak@147 256 --
flickerstreak@147 257 -- Events
flickerstreak@147 258 --
flickerstreak@147 259
flickerstreak@147 260 function Bar:OnConfigModeChanged(event, mode)
flickerstreak@147 261 self:SetConfigMode(mode)
flickerstreak@123 262 end
flickerstreak@123 263
flickerstreak@153 264 --
flickerstreak@153 265 -- Accessors
flickerstreak@153 266 --
flickerstreak@153 267
flickerstreak@153 268 function Bar:GetName()
flickerstreak@153 269 return self.name
flickerstreak@153 270 end
flickerstreak@153 271
flickerstreak@153 272 -- only ReAction:RenameBar() should call this function. Calling from any other
flickerstreak@153 273 -- context will desync the bar list in the ReAction class.
flickerstreak@153 274 function Bar:SetName(name)
flickerstreak@213 275 if self.LBFGroup then
flickerstreak@213 276 -- LBF doesn't offer a method of renaming a group, so delete and remake the group.
flickerstreak@213 277 local c = self.config.ButtonFacade
flickerstreak@213 278 local g = ReAction.LBF:Group(L["ReAction"], name)
Flick@234 279 for idx, b in self:IterateButtons() do
flickerstreak@213 280 self.LBFGroup:RemoveButton(b:GetFrame(), true)
flickerstreak@213 281 g:AddButton(b:GetFrame())
flickerstreak@213 282 end
flickerstreak@213 283 self.LBFGroup:Delete(true)
flickerstreak@213 284 self.LBFGroup = g
flickerstreak@213 285 self.LBFGroup:Skin(c.skinID, c.gloss, c.backdrop, c.colors)
flickerstreak@213 286 end
flickerstreak@153 287 self.name = name
flickerstreak@154 288 if self.overlay then
flickerstreak@154 289 self.overlay:SetLabel(self.name)
flickerstreak@154 290 end
flickerstreak@153 291 end
flickerstreak@153 292
flickerstreak@153 293 function Bar:GetFrame()
flickerstreak@153 294 -- this method is included for documentation purposes. It is overridden
flickerstreak@153 295 -- for each object in the :New() method.
flickerstreak@153 296 error("Invalid Bar object: used without initialization")
flickerstreak@153 297 end
flickerstreak@153 298
Flick@234 299 function Bar:GetButton(idx)
Flick@234 300 return self.buttons[idx]
Flick@234 301 end
Flick@234 302
Flick@237 303 function Bar:GetButtonClass()
Flick@237 304 return self.buttonClass
Flick@237 305 end
Flick@237 306
flickerstreak@153 307 function Bar:GetConfig()
flickerstreak@153 308 return self.config
flickerstreak@153 309 end
flickerstreak@153 310
flickerstreak@153 311 function Bar:GetAnchor()
flickerstreak@122 312 local c = self.config
flickerstreak@153 313 return (c.point or "CENTER"),
flickerstreak@153 314 (c.anchor or self:GetFrame():GetParent():GetName()),
flickerstreak@153 315 (c.relpoint or c.point or "CENTER"),
flickerstreak@153 316 (c.x or 0),
flickerstreak@153 317 (c.y or 0)
flickerstreak@122 318 end
flickerstreak@122 319
flickerstreak@122 320 function Bar:SetAnchor(point, frame, relativePoint, x, y)
flickerstreak@122 321 local c = self.config
flickerstreak@122 322 c.point = point or c.point
flickerstreak@122 323 c.anchor = frame or c.anchor
flickerstreak@122 324 c.relpoint = relativePoint or c.relpoint
flickerstreak@122 325 c.x = x or c.x
flickerstreak@122 326 c.y = y or c.y
flickerstreak@122 327 self:ApplyAnchor()
flickerstreak@122 328 end
flickerstreak@122 329
flickerstreak@122 330 function Bar:GetSize()
flickerstreak@122 331 local f = self:GetFrame()
flickerstreak@122 332 return f:GetWidth(), f:GetHeight()
flickerstreak@122 333 end
flickerstreak@122 334
flickerstreak@122 335 function Bar:SetSize(w,h)
flickerstreak@122 336 local f = self:GetFrame()
flickerstreak@122 337 self.config.width = w
flickerstreak@122 338 self.config.height = h
flickerstreak@122 339 f:SetWidth(w)
flickerstreak@122 340 f:SetHeight(h)
flickerstreak@122 341 end
flickerstreak@122 342
flickerstreak@122 343 function Bar:GetButtonSize()
flickerstreak@122 344 local w = self.config.btnWidth or 32
flickerstreak@122 345 local h = self.config.btnHeight or 32
flickerstreak@122 346 -- TODO: get from modules?
flickerstreak@122 347 return w,h
flickerstreak@122 348 end
flickerstreak@122 349
flickerstreak@122 350 function Bar:SetButtonSize(w,h)
flickerstreak@122 351 if w > 0 and h > 0 then
flickerstreak@122 352 self.config.btnWidth = w
flickerstreak@122 353 self.config.btnHeight = h
Flick@279 354 for _, b in pairs(self.buttons) do
Flick@279 355 b:Refresh()
Flick@279 356 end
flickerstreak@122 357 end
flickerstreak@122 358 end
flickerstreak@122 359
flickerstreak@153 360 function Bar:GetNumButtons()
flickerstreak@153 361 local r,c = self:GetButtonGrid()
flickerstreak@153 362 return r*c
flickerstreak@153 363 end
flickerstreak@153 364
flickerstreak@122 365 function Bar:GetButtonGrid()
flickerstreak@122 366 local cfg = self.config
flickerstreak@122 367 local r = cfg.btnRows or 1
flickerstreak@122 368 local c = cfg.btnColumns or 1
flickerstreak@122 369 local s = cfg.spacing or 4
flickerstreak@122 370 return r,c,s
flickerstreak@122 371 end
flickerstreak@122 372
flickerstreak@122 373 function Bar:SetButtonGrid(r,c,s)
Flick@283 374 local cfg = self.config
Flick@283 375 r = r or cfg.btnRows
Flick@283 376 c = c or cfg.btnColumns
Flick@283 377 s = s or cfg.spacing
flickerstreak@122 378 if r > 0 and c > 0 and s > 0 then
flickerstreak@122 379 cfg.btnRows = r
flickerstreak@122 380 cfg.btnColumns = c
flickerstreak@122 381 cfg.spacing = s
flickerstreak@122 382 end
Flick@234 383 self.buttonClass:SetupBar(self)
flickerstreak@122 384 end
flickerstreak@122 385
flickerstreak@122 386 function Bar:GetAlpha()
flickerstreak@122 387 return self.config.alpha or 1.0
flickerstreak@122 388 end
flickerstreak@122 389
flickerstreak@122 390 function Bar:SetAlpha(value)
flickerstreak@122 391 self.config.alpha = value
flickerstreak@122 392 self:GetFrame():SetAlpha(value or 1.0)
flickerstreak@155 393 self:UpdateDefaultStateAlpha()
flickerstreak@122 394 end
flickerstreak@122 395
flickerstreak@122 396 function Bar:IterateButtons()
Flick@234 397 -- iterator returns idx, button, but does NOT iterate in index order
flickerstreak@122 398 return pairs(self.buttons)
flickerstreak@122 399 end
flickerstreak@122 400
flickerstreak@153 401 --
flickerstreak@153 402 -- Methods
flickerstreak@153 403 --
flickerstreak@153 404
flickerstreak@147 405 function Bar:SetConfigMode(mode)
flickerstreak@155 406 self:SetSecureData("showAll",mode)
flickerstreak@147 407 self:ShowControls(mode)
Flick@234 408 for idx, b in self:IterateButtons() do
flickerstreak@147 409 b:ShowGridTemp(mode)
flickerstreak@147 410 b:UpdateActionIDLabel(mode)
flickerstreak@147 411 end
Flick@279 412 self.buttonClass:SetupBar(self) -- force a full refresh
flickerstreak@147 413 end
flickerstreak@147 414
flickerstreak@147 415 function Bar:SetKeybindMode(mode)
flickerstreak@155 416 self:SetSecureData("showAll",mode)
Flick@234 417 for idx, b in self:IterateButtons() do
flickerstreak@147 418 b:SetKeybindMode(mode)
flickerstreak@147 419 end
flickerstreak@147 420 end
flickerstreak@147 421
flickerstreak@153 422 function Bar:ApplyAnchor()
flickerstreak@153 423 local f = self:GetFrame()
flickerstreak@153 424 local c = self.config
flickerstreak@153 425 local p = c.point
flickerstreak@153 426
flickerstreak@153 427 f:SetWidth(c.width)
flickerstreak@153 428 f:SetHeight(c.height)
flickerstreak@153 429 f:ClearAllPoints()
flickerstreak@153 430
flickerstreak@153 431 if p then
flickerstreak@153 432 local a = f:GetParent()
flickerstreak@153 433 if c.anchor then
flickerstreak@153 434 local bar = ReAction:GetBar(c.anchor)
flickerstreak@153 435 if bar then
flickerstreak@153 436 a = bar:GetFrame()
flickerstreak@153 437 else
flickerstreak@153 438 a = _G[c.anchor]
flickerstreak@153 439 end
flickerstreak@153 440 end
flickerstreak@153 441 local fr = a or f:GetParent()
flickerstreak@153 442 f:SetPoint(p, a or f:GetParent(), c.relpoint, c.x or 0, c.y or 0)
flickerstreak@153 443 else
flickerstreak@153 444 f:SetPoint("CENTER")
flickerstreak@153 445 end
flickerstreak@155 446
flickerstreak@155 447 self:UpdateDefaultStateAnchor()
flickerstreak@153 448 end
flickerstreak@153 449
flickerstreak@153 450 function Bar:ClipNButtons( n )
flickerstreak@153 451 local cfg = self.config
flickerstreak@153 452 local r = cfg.btnRows or 1
flickerstreak@153 453 local c = cfg.btnColumns or 1
flickerstreak@153 454
flickerstreak@153 455 cfg.btnRows = ceil(n/c)
flickerstreak@153 456 cfg.btnColumns = min(n,c)
flickerstreak@153 457 end
flickerstreak@153 458
flickerstreak@153 459 function Bar:AddButton(idx, button)
flickerstreak@153 460 local f = self:GetFrame()
flickerstreak@153 461
Flick@234 462 self.buttons[idx] = button
flickerstreak@153 463
flickerstreak@153 464 -- Store a properly wrapped reference to the child frame as an attribute
flickerstreak@153 465 -- (accessible via "frameref-btn#")
flickerstreak@153 466 f:SetFrameRef(format("btn%d",idx), button:GetFrame())
flickerstreak@213 467
flickerstreak@213 468 -- button constructors are responsible for calling SkinButton
flickerstreak@153 469 end
flickerstreak@153 470
flickerstreak@153 471 function Bar:RemoveButton(button)
Flick@234 472 local idx = button:GetIndex()
flickerstreak@153 473 if idx then
flickerstreak@153 474 self:GetFrame():SetAttribute(format("frameref-btn%d",idx),nil)
Flick@234 475 self.buttons[idx] = nil
flickerstreak@153 476 end
flickerstreak@213 477 if self.LBFGroup then
flickerstreak@213 478 self.LBFGroup:RemoveButton(button:GetFrame(),true)
flickerstreak@213 479 end
flickerstreak@153 480 end
flickerstreak@153 481
flickerstreak@122 482 function Bar:PlaceButton(button, baseW, baseH)
Flick@234 483 local idx = button:GetIndex()
flickerstreak@122 484 if idx then
flickerstreak@122 485 local r, c, s = self:GetButtonGrid()
flickerstreak@122 486 local bh, bw = self:GetButtonSize()
flickerstreak@122 487 local row, col = floor((idx-1)/c), fmod((idx-1),c) -- zero-based
flickerstreak@122 488 local x, y = col*bw + (col+0.5)*s, -(row*bh + (row+0.5)*s)
flickerstreak@122 489 local scale = bw/baseW
flickerstreak@122 490 local b = button:GetFrame()
flickerstreak@122 491
flickerstreak@122 492 b:ClearAllPoints()
flickerstreak@122 493 b:SetPoint("TOPLEFT",x/scale,y/scale)
flickerstreak@122 494 b:SetScale(scale)
flickerstreak@122 495 end
flickerstreak@122 496 end
flickerstreak@122 497
flickerstreak@213 498 function Bar:SkinButton( button, data )
flickerstreak@213 499 if self.LBFGroup then
flickerstreak@213 500 self.LBFGroup:AddButton(button:GetFrame(), data)
flickerstreak@213 501 end
flickerstreak@147 502 end
flickerstreak@148 503
flickerstreak@155 504 function Bar:UpdateShowGrid()
Flick@234 505 for idx, button in self:IterateButtons() do
flickerstreak@155 506 button:UpdateShowGrid()
flickerstreak@155 507 end
flickerstreak@155 508 end
flickerstreak@155 509
flickerstreak@154 510 function Bar:ShowControls(show)
flickerstreak@154 511 if show then
flickerstreak@155 512 if not self.overlay then
flickerstreak@155 513 self.overlay = Bar.Overlay:New(self) -- see Overlay.lua
flickerstreak@154 514 end
flickerstreak@155 515 self.overlay:Show()
flickerstreak@155 516 self:RefreshSecureState()
flickerstreak@155 517 elseif self.overlay then
flickerstreak@155 518 self.overlay:Hide()
flickerstreak@154 519 end
flickerstreak@154 520 end
flickerstreak@154 521
flickerstreak@154 522 function Bar:RefreshControls()
flickerstreak@154 523 if self.overlay and self.overlay:IsShown() then
flickerstreak@154 524 self.overlay:RefreshControls()
flickerstreak@154 525 end
flickerstreak@154 526 end
flickerstreak@154 527
flickerstreak@154 528 function Bar:SetLabelSubtext(text)
flickerstreak@154 529 if self.overlay then
flickerstreak@154 530 self.overlay:SetLabelSubtext(text)
flickerstreak@154 531 end
flickerstreak@154 532 end
flickerstreak@154 533
flickerstreak@154 534 --
flickerstreak@154 535 -- Secure state functions
flickerstreak@154 536 --
flickerstreak@154 537
flickerstreak@155 538 function Bar:GetSecureState()
flickerstreak@155 539 local env = GetManagedEnvironment(self:GetFrame())
flickerstreak@155 540 return env and env.state
flickerstreak@155 541 end
flickerstreak@155 542
flickerstreak@155 543 function Bar:GetStateProperty(state, propname)
Flick@242 544 return tfetch(self:GetConfig(), "states", state, propname)
flickerstreak@155 545 end
flickerstreak@155 546
flickerstreak@155 547 function Bar:SetStateProperty(state, propname, value)
Flick@242 548 local s = tbuild(self:GetConfig(), "states", state)
Flick@242 549 s[propname] = value
Flick@242 550 self:SetSecureStateData(state, propname, value)
Flick@242 551 end
Flick@242 552
Flick@242 553 function Bar:ApplyStates()
Flick@242 554 local states = tfetch(self:GetConfig(), "states")
Flick@242 555 if states then
Flick@242 556 self:SetStateDriver(states)
Flick@242 557 end
Flick@242 558 end
Flick@242 559
Flick@242 560 function Bar:CleanupStates()
Flick@242 561 self:SetStateDriver(nil)
flickerstreak@155 562 end
flickerstreak@155 563
flickerstreak@155 564 function Bar:RefreshSecureState()
flickerstreak@155 565 self:GetFrame():Execute(_reaction_refresh)
flickerstreak@155 566 end
flickerstreak@155 567
flickerstreak@155 568 -- usage: SetSecureData(globalname, [tblkey1, tblkey2, ...], value)
flickerstreak@155 569 function Bar:SetSecureData( ... )
flickerstreak@155 570 local n = select('#',...)
flickerstreak@155 571 if n < 2 then
flickerstreak@155 572 error("ReAction.Bar:SetSecureData() requires at least 2 arguments")
flickerstreak@155 573 end
flickerstreak@155 574 local f = self:GetFrame()
flickerstreak@155 575 f:SetAttribute("data-depth",n-1)
flickerstreak@155 576 f:SetAttribute("data-value",select(n,...))
flickerstreak@155 577 for i = 1, n-1 do
flickerstreak@155 578 local key = select(i,...)
flickerstreak@155 579 if key == nil then
flickerstreak@155 580 error("ReAction.Bar:SetSecureData() - nil table key in argument list (#"..i..")")
flickerstreak@155 581 end
flickerstreak@155 582 f:SetAttribute("data-key-"..i, key)
flickerstreak@155 583 end
flickerstreak@155 584 f:Execute(
flickerstreak@155 585 [[
flickerstreak@155 586 local n = self:GetAttribute("data-depth")
flickerstreak@155 587 if n > 0 then
flickerstreak@155 588 local value = self:GetAttribute("data-value")
flickerstreak@155 589 local t = _G
flickerstreak@155 590 for i = 1, n do
flickerstreak@155 591 local key = self:GetAttribute("data-key-"..i)
flickerstreak@155 592 if not key then return end
flickerstreak@155 593 if not t[key] then
flickerstreak@155 594 t[key] = newtable()
flickerstreak@155 595 end
flickerstreak@155 596 if i == n then
flickerstreak@155 597 t[key] = value
flickerstreak@155 598 else
flickerstreak@155 599 t = t[key]
flickerstreak@155 600 end
flickerstreak@155 601 end
flickerstreak@155 602 end
flickerstreak@155 603 ]])
flickerstreak@155 604 self:RefreshSecureState()
flickerstreak@155 605 end
flickerstreak@155 606
flickerstreak@155 607 function Bar:SetSecureStateData( state, key, value )
flickerstreak@155 608 self:SetSecureData("settings",state,key,value)
flickerstreak@155 609 end
flickerstreak@155 610
flickerstreak@155 611 -- sets a snippet to be run as an extension to _onstate-reaction
flickerstreak@155 612 function Bar:SetSecureStateExtension( id, snippet )
flickerstreak@155 613 if id == nil then
flickerstreak@155 614 error("ReAction.Bar:SetSecureStateExtension() requires an id")
flickerstreak@155 615 end
flickerstreak@155 616 local f = self:GetFrame()
flickerstreak@155 617 f:SetAttribute("input-secure-ext-id",id)
flickerstreak@155 618 f:SetAttribute("secure-ext-"..id,snippet)
flickerstreak@155 619 f:Execute(
flickerstreak@155 620 [[
flickerstreak@155 621 local id = self:GetAttribute("input-secure-ext-id")
flickerstreak@155 622 if id then
flickerstreak@155 623 extensions[id] = self:GetAttribute("secure-ext-"..id) or nil
flickerstreak@155 624 end
flickerstreak@155 625 ]])
flickerstreak@155 626 self:RefreshSecureState()
flickerstreak@155 627 end
flickerstreak@155 628
flickerstreak@155 629 function Bar:SetFrameRef( name, refFrame )
flickerstreak@155 630 if refFrame then
flickerstreak@155 631 local _, explicit = refFrame:IsProtected()
flickerstreak@155 632 if not explicit then
flickerstreak@155 633 refFrame = nil
flickerstreak@155 634 end
flickerstreak@155 635 end
flickerstreak@155 636 if refFrame then
flickerstreak@155 637 self:GetFrame():SetFrameRef(name,refFrame)
flickerstreak@155 638 else
flickerstreak@155 639 self:GetFrame():SetAttribute("frameref-"..name,nil)
flickerstreak@155 640 end
flickerstreak@155 641 end
flickerstreak@155 642
flickerstreak@157 643 function Bar:SetStateDriver( states )
flickerstreak@157 644 if states then
flickerstreak@157 645 for state, props in pairs(states) do
flickerstreak@157 646 self:SetSecureStateData(state, "active_", true) -- make sure there's a 'settings' field for this state
flickerstreak@157 647 for propname, value in pairs(props) do
flickerstreak@157 648 if propname == "anchorFrame" then
flickerstreak@157 649 self:SetFrameRef("anchor-"..state, _G[value])
flickerstreak@157 650 elseif propname == "rule" then
flickerstreak@157 651 -- do nothing
flickerstreak@157 652 else
flickerstreak@157 653 self:SetSecureStateData(state, propname, value)
flickerstreak@157 654 end
flickerstreak@157 655 end
flickerstreak@157 656 end
flickerstreak@157 657 end
flickerstreak@157 658 local rule = states and self:BuildStateRule(states)
flickerstreak@155 659 if rule then
flickerstreak@155 660 RegisterStateDriver(self:GetFrame(),"reaction",rule)
flickerstreak@155 661 elseif self.statedriver then
flickerstreak@155 662 UnregisterStateDriver(self:GetFrame(),"reaction")
flickerstreak@155 663 end
flickerstreak@155 664 self.statedriver = rule
flickerstreak@157 665 self:BuildStateKeybinds(states)
flickerstreak@155 666 self:RefreshSecureState()
flickerstreak@155 667 end
flickerstreak@155 668
flickerstreak@148 669 -- pass unit=nil to set up the unit elsewhere, if you want something more complex
flickerstreak@148 670 function Bar:RegisterUnitWatch( unit, enable )
flickerstreak@148 671 local f = self:GetFrame()
flickerstreak@148 672 if unit then
flickerstreak@148 673 f:SetAttribute("unit",unit)
flickerstreak@148 674 end
flickerstreak@156 675 if enable then
Flick@234 676 if not self.unitwatch then
Flick@279 677 RegisterUnitWatch(f,true)
Flick@234 678 end
flickerstreak@156 679 elseif self.unitwatch then
Flick@279 680 UnregisterUnitWatch(f)
flickerstreak@156 681 end
flickerstreak@148 682 self.unitwatch = enable
flickerstreak@156 683 self:RefreshSecureState()
flickerstreak@148 684 end
flickerstreak@148 685
flickerstreak@155 686 function Bar:SetStateKeybind( key, state )
flickerstreak@155 687 local f = self:GetFrame()
flickerstreak@155 688 local binds = self.statebinds
flickerstreak@155 689 if not binds then
flickerstreak@155 690 binds = { }
flickerstreak@155 691 self.statebinds = binds
flickerstreak@155 692 end
flickerstreak@155 693
flickerstreak@155 694 -- clear the old binding, if any
flickerstreak@155 695 if binds[state] then
flickerstreak@155 696 SetOverrideBinding(f, false, binds[state], nil)
flickerstreak@155 697 end
flickerstreak@155 698
flickerstreak@155 699 if key then
flickerstreak@157 700 SetOverrideBindingClick(f, false, key, f:GetName(), state) -- state name is virtual mouse button
flickerstreak@155 701 end
flickerstreak@155 702 binds[state] = key
flickerstreak@155 703 end
flickerstreak@155 704
flickerstreak@155 705 function Bar:GetStateKeybind( state )
flickerstreak@155 706 if self.statebinds and state then
flickerstreak@155 707 return self.statebinds[state]
flickerstreak@155 708 end
flickerstreak@155 709 end
flickerstreak@155 710
flickerstreak@155 711 function Bar:UpdateDefaultStateAnchor()
flickerstreak@155 712 local point, frame, relPoint, x, y = self:GetAnchor()
flickerstreak@155 713 local f = self:GetFrame()
flickerstreak@155 714 f:SetAttribute("defaultAnchor-point",point)
flickerstreak@155 715 f:SetAttribute("defaultAnchor-relPoint",relPoint)
flickerstreak@155 716 f:SetAttribute("defaultAnchor-x",x)
flickerstreak@155 717 f:SetAttribute("defaultAnchor-y",y)
flickerstreak@155 718 self:SetFrameRef("defaultAnchor",_G[frame or "UIParent"])
flickerstreak@155 719 f:Execute([[
flickerstreak@155 720 for _, k in pairs(anchorKeys) do
flickerstreak@155 721 defaultAnchor[k] = self:GetAttribute("defaultAnchor-"..k)
flickerstreak@155 722 end
flickerstreak@155 723 defaultAnchor.frame = self:GetAttribute("frameref-defaultAnchor")
flickerstreak@155 724 ]])
flickerstreak@155 725 end
flickerstreak@155 726
flickerstreak@155 727 function Bar:UpdateDefaultStateAlpha()
flickerstreak@155 728 local f = self:GetFrame()
flickerstreak@155 729 f:SetAttribute("defaultAlpha",self:GetAlpha())
flickerstreak@155 730 f:Execute([[
flickerstreak@155 731 defaultAlpha = self:GetAttribute("defaultAlpha")
flickerstreak@155 732 ]])
flickerstreak@155 733 end
flickerstreak@157 734
flickerstreak@157 735 ---- secure state driver rules ----
flickerstreak@157 736
flickerstreak@157 737 local playerClass = select(2, UnitClass("player"))
flickerstreak@157 738 local function ClassFilter(...)
flickerstreak@157 739 for i = 1, select('#',...) do
flickerstreak@157 740 if playerClass == select(i,...) then
flickerstreak@157 741 return false
flickerstreak@157 742 end
flickerstreak@157 743 end
flickerstreak@157 744 return true
flickerstreak@157 745 end
flickerstreak@157 746
flickerstreak@157 747 local ruleformats = {
flickerstreak@157 748 stealth = { format = "stealth", filter = ClassFilter("ROGUE","DRUID") },
flickerstreak@157 749 nostealth = { format = "nostealth", filter = ClassFilter("ROGUE","DRUID") },
flickerstreak@157 750 shadowdance = { format = "bonusbar:2", filter = ClassFilter("ROGUE") },
flickerstreak@157 751 shadowform = { format = "form:1", filter = ClassFilter("PRIEST") },
flickerstreak@157 752 noshadowform = { format = "noform", filter = ClassFilter("PRIEST") },
flickerstreak@157 753 battle = { format = "stance:1", filter = ClassFilter("WARRIOR") },
flickerstreak@157 754 defensive = { format = "stance:2", filter = ClassFilter("WARRIOR") },
flickerstreak@157 755 berserker = { format = "stance:3", filter = ClassFilter("WARRIOR") },
flickerstreak@157 756 caster = { format = "form:0/2/4/5/6", filter = ClassFilter("DRUID") },
flickerstreak@157 757 bear = { format = "form:1", filter = ClassFilter("DRUID") },
flickerstreak@157 758 cat = { format = "form:3", filter = ClassFilter("DRUID") },
flickerstreak@157 759 tree = { format = "form:5", filter = ClassFilter("DRUID") },
flickerstreak@157 760 moonkin = { format = "form:5", filter = ClassFilter("DRUID") },
flickerstreak@178 761 demon = { format = "form:2", filter = ClassFilter("WARLOCK") },
flickerstreak@178 762 nodemon = { format = "noform", filter = ClassFilter("WARLOCK") },
flickerstreak@157 763 pet = { format = "pet" },
flickerstreak@157 764 nopet = { format = "nopet" },
flickerstreak@176 765 harm = { format = "@target,harm" },
flickerstreak@176 766 help = { format = "@target,help" },
flickerstreak@176 767 notarget = { format = "@target,noexists" },
flickerstreak@176 768 focusharm = { format = "@focus,harm" },
flickerstreak@176 769 focushelp = { format = "@focus,help" },
flickerstreak@176 770 nofocus = { format = "@focus,noexists" },
flickerstreak@157 771 raid = { format = "group:raid" },
flickerstreak@157 772 party = { format = "group:party" },
flickerstreak@157 773 solo = { format = "nogroup" },
flickerstreak@157 774 combat = { format = "combat" },
flickerstreak@157 775 nocombat = { format = "nocombat" },
flickerstreak@176 776 possess = { format = "@vehicle,noexists,bonusbar:5" },
flickerstreak@176 777 vehicle = { format = "@vehicle,exists,bonusbar:5" },
flickerstreak@157 778 }
flickerstreak@157 779
flickerstreak@157 780 function Bar.InitRuleFormats()
flickerstreak@157 781 local forms = { }
flickerstreak@157 782 for i = 1, GetNumShapeshiftForms() do
flickerstreak@157 783 local _, name = GetShapeshiftFormInfo(i)
flickerstreak@157 784 forms[name] = i;
flickerstreak@157 785 end
flickerstreak@157 786 -- use 9 if not found since 9 is never a valid stance/form
flickerstreak@157 787 local defensive = forms[GetSpellInfo(71)] or 9
flickerstreak@157 788 local berserker = forms[GetSpellInfo(2458)] or 9
flickerstreak@176 789 local bear = forms[GetSpellInfo(5487)] or 9
flickerstreak@157 790 local aquatic = forms[GetSpellInfo(1066)] or 9
flickerstreak@157 791 local cat = forms[GetSpellInfo(768)] or 9
flickerstreak@157 792 local travel = forms[GetSpellInfo(783)] or 9
flickerstreak@157 793 local tree = forms[GetSpellInfo(33891)] or 9
flickerstreak@157 794 local moonkin = forms[GetSpellInfo(24858)] or 9
flickerstreak@157 795 local flight = forms[GetSpellInfo(40120)] or forms[GetSpellInfo(33943)] or 9
flickerstreak@157 796
flickerstreak@157 797 ruleformats.defensive.format = "stance:"..defensive
flickerstreak@157 798 ruleformats.berserker.format = "stance:"..berserker
flickerstreak@157 799 ruleformats.caster.format = format("form:0/%d/%d/%d", aquatic, travel, flight)
flickerstreak@157 800 ruleformats.bear.format = "form:"..bear
flickerstreak@157 801 ruleformats.cat.format = "form:"..cat
flickerstreak@157 802 ruleformats.tree.format = "form:"..tree
flickerstreak@157 803 ruleformats.moonkin.format = "form:"..moonkin
flickerstreak@157 804 end
flickerstreak@157 805
flickerstreak@157 806 function Bar:BuildStateRule(states)
flickerstreak@157 807 -- states is a table :
flickerstreak@157 808 -- states[statename].rule = {
flickerstreak@157 809 -- order = #,
flickerstreak@157 810 -- type = "default"/"custom"/"any"/"all",
flickerstreak@157 811 -- values = { ... }, -- keys of ruleformats[]
flickerstreak@157 812 -- custom = "...",
flickerstreak@157 813 -- }
flickerstreak@157 814 local rules = { }
flickerstreak@157 815 local default
flickerstreak@157 816
flickerstreak@157 817 for idx, state in ipairs(fieldsort(states, "rule", "order")) do
flickerstreak@157 818 local c = states[state].rule
flickerstreak@157 819 local type = c.type
flickerstreak@157 820 if type == "default" then
flickerstreak@157 821 default = default or state
flickerstreak@157 822 elseif type == "custom" then
flickerstreak@157 823 if c.custom then
flickerstreak@157 824 -- strip out all spaces from the custom rule
flickerstreak@157 825 table.insert(rules, format("%s %s", c.custom:gsub("%s",""), state))
flickerstreak@157 826 end
flickerstreak@157 827 elseif type == "any" or type == "all" then
flickerstreak@157 828 if c.values then
flickerstreak@157 829 local clauses = { }
flickerstreak@157 830 for key, value in pairs(c.values) do
flickerstreak@157 831 if ruleformats[key] and not ruleformats[key].filter then
flickerstreak@157 832 table.insert(clauses, ruleformats[key].format)
flickerstreak@157 833 end
flickerstreak@157 834 end
flickerstreak@157 835 if #clauses > 0 then
flickerstreak@157 836 local sep = (type == "any") and "][" or ","
flickerstreak@157 837 table.insert(rules, format("[%s] %s", table.concat(clauses,sep), state))
flickerstreak@157 838 end
flickerstreak@157 839 end
flickerstreak@157 840 end
flickerstreak@157 841 end
flickerstreak@157 842 -- make sure that the default, if any, is last
flickerstreak@157 843 if default then
flickerstreak@157 844 table.insert(rules, default)
flickerstreak@157 845 end
flickerstreak@157 846 return table.concat(rules,";")
flickerstreak@157 847 end
flickerstreak@157 848
flickerstreak@157 849 function Bar:BuildStateKeybinds( states )
flickerstreak@157 850 if states then
flickerstreak@157 851 for name, state in pairs(states) do
flickerstreak@157 852 local rule = tfetch(state, "rule")
flickerstreak@157 853 if rule and rule.type == "keybind" then
flickerstreak@157 854 self:SetStateKeybind(rule.keybind, name)
flickerstreak@157 855 else
flickerstreak@157 856 self:SetStateKeybind(nil, name) -- this clears an existing keybind
flickerstreak@157 857 end
flickerstreak@157 858 end
flickerstreak@157 859 end
flickerstreak@157 860 end
flickerstreak@157 861