Mercurial > wow > reaction

annotate StanceButton.lua @ 279:5b9c0164a491

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fixed a number of problems with drag corners in config mode - Fixed button resizing via dragging not updating until a reload - Fixed pet/stance/vehicle buttons not always displaying in config mode - Fixed error spam when making too many pet/bag buttons
author Flick
date Wed, 11 May 2011 16:19:17 -0700
parents 4e325f1ea6e1
children 276165a0e860
rev   line source
Flick@276 1 local _, ns = ...
Flick@276 2 local ReAction = ns.ReAction
flickerstreak@134 3 local L = ReAction.L
flickerstreak@134 4 local _G = _G
flickerstreak@134 5 local CreateFrame = CreateFrame
flickerstreak@134 6 local format = string.format
flickerstreak@134 7 local GetCVar = GetCVar
flickerstreak@134 8 local GameTooltip_SetDefaultAnchor = GameTooltip_SetDefaultAnchor
flickerstreak@134 9 local CooldownFrame_SetTimer = CooldownFrame_SetTimer
flickerstreak@134 10 local InCombatLockdown = InCombatLockdown
flickerstreak@134 11 local GetNumShapeshiftForms = GetNumShapeshiftForms
flickerstreak@134 12 local GetShapeshiftFormInfo = GetShapeshiftFormInfo
flickerstreak@134 13 local IsUsableSpell = IsUsableSpell
flickerstreak@134 14 local GetSpellInfo = GetSpellInfo
flickerstreak@134 15
flickerstreak@134 16 --
flickerstreak@134 17 -- private
flickerstreak@134 18 --
flickerstreak@134 19 local playerClass = select(2,UnitClass("player"))
flickerstreak@134 20
flickerstreak@134 21 local eventList = {
flickerstreak@134 22 "PLAYER_REGEN_ENABLED",
flickerstreak@134 23 "PLAYER_ENTERING_WORLD",
flickerstreak@134 24 "UPDATE_SHAPESHIFT_FORM",
flickerstreak@134 25 "UPDATE_SHAPESHIFT_FORMS",
flickerstreak@134 26 "UPDATE_SHAPESHIFT_USABLE",
flickerstreak@134 27 "UPDATE_SHAPESHIFT_COOLDOWN",
flickerstreak@137 28 "UPDATE_BINDINGS",
flickerstreak@134 29 }
flickerstreak@134 30
flickerstreak@134 31 --
flickerstreak@134 32 -- Stance Button class
flickerstreak@134 33 --
flickerstreak@218 34 local buttonTypeID = "Stance"
flickerstreak@134 35 local Super = ReAction.Button
flickerstreak@218 36 local Stance = setmetatable(
flickerstreak@218 37 {
Flick@248 38 defaultBarConfig = {
flickerstreak@218 39 type = buttonTypeID,
flickerstreak@218 40 btnHeight = 36,
flickerstreak@218 41 btnWidth = 36,
flickerstreak@218 42 btnRows = 1,
flickerstreak@218 43 btnColumns = 6,
flickerstreak@218 44 spacing = 3
flickerstreak@218 45 },
flickerstreak@218 46
flickerstreak@218 47 barType = L["Stance Bar"],
Flick@231 48 buttonTypeID = buttonTypeID
flickerstreak@218 49 },
flickerstreak@218 50 { __index = Super } )
flickerstreak@223 51
flickerstreak@134 52 ReAction.Button.Stance = Stance
flickerstreak@223 53 ReAction:RegisterBarType(Stance)
flickerstreak@134 54
Flick@234 55 function Stance:New( config, bar, idx, idHint )
Flick@277 56 self = Super.New(self, config, bar, idx, "SecureActionButtonTemplate, ActionButtonTemplate" )
flickerstreak@134 57
flickerstreak@134 58 local f = self:GetFrame()
flickerstreak@134 59 local barFrame = bar:GetFrame()
flickerstreak@134 60 local config = self:GetConfig()
flickerstreak@134 61
Flick@249 62 -- set up the stance ID
Flick@249 63 config.stanceID = config.stanceID or idx
flickerstreak@134 64
flickerstreak@134 65 -- attribute setup
flickerstreak@134 66 f:SetAttribute("type","spell")
flickerstreak@134 67
flickerstreak@134 68 -- non secure scripts
flickerstreak@134 69 f:SetScript("OnEvent", function(frame, ...) self:OnEvent(...) end)
flickerstreak@134 70 f:SetScript("OnEnter", function(frame) self:OnEnter() end)
flickerstreak@134 71 f:SetScript("OnLeave", function(frame) self:OnLeave() end)
flickerstreak@134 72 f:SetScript("PreClick", function(frame, ...) self:PreClick(...) end)
flickerstreak@134 73
flickerstreak@134 74 -- secure handlers
flickerstreak@134 75 -- (none)
flickerstreak@134 76
flickerstreak@134 77 -- event registration
flickerstreak@134 78 for _, evt in pairs(eventList) do
flickerstreak@134 79 f:RegisterEvent(evt)
flickerstreak@134 80 end
flickerstreak@134 81
flickerstreak@134 82 -- attach to skinner
flickerstreak@134 83 bar:SkinButton(self)
flickerstreak@134 84
flickerstreak@134 85 -- initial display
flickerstreak@134 86 if ReAction:GetConfigMode() then
flickerstreak@134 87 self:GetFrame():Show()
flickerstreak@134 88 end
flickerstreak@134 89
flickerstreak@134 90 self:Refresh()
flickerstreak@134 91
flickerstreak@134 92 return self
flickerstreak@134 93 end
flickerstreak@134 94
flickerstreak@134 95 function Stance:GetActionID()
flickerstreak@134 96 return self.config.stanceID
flickerstreak@134 97 end
flickerstreak@134 98
Flick@249 99 function Stance:AcquireActionID()
Flick@249 100 -- don't use pool
Flick@249 101 end
Flick@249 102
Flick@249 103 function Stance:ReleaseActionID()
Flick@249 104 -- don't use pool
Flick@249 105 end
Flick@249 106
flickerstreak@134 107 function Stance:UpdateAction()
flickerstreak@134 108 if InCombatLockdown() then
flickerstreak@134 109 self.updatePending = true
flickerstreak@134 110 else
flickerstreak@134 111 self.updatePending = false
flickerstreak@134 112 local idx = self:GetActionID()
flickerstreak@134 113 local f = self:GetFrame()
Flick@279 114 if idx > GetNumShapeshiftForms() and not ReAction:GetConfigMode() then
flickerstreak@134 115 f:Hide()
flickerstreak@134 116 else
flickerstreak@134 117 f:SetAttribute("spell", select(2,GetShapeshiftFormInfo(idx)))
flickerstreak@134 118 f:Show()
flickerstreak@134 119 self:Update()
flickerstreak@134 120 end
flickerstreak@134 121 end
flickerstreak@134 122 end
flickerstreak@134 123
flickerstreak@134 124 function Stance:Refresh()
flickerstreak@134 125 Super.Refresh(self)
flickerstreak@137 126 self:UpdateHotkey()
flickerstreak@136 127 self:UpdateAction()
flickerstreak@134 128 end
flickerstreak@134 129
flickerstreak@134 130 function Stance:Update()
flickerstreak@179 131 local texture, _, isActive, isCastable = GetShapeshiftFormInfo(self:GetActionID())
flickerstreak@134 132
flickerstreak@134 133 local icon = self.frames.icon
flickerstreak@134 134 icon:SetTexture(texture)
flickerstreak@134 135 self:GetFrame():SetChecked( isActive and 1 or 0 )
flickerstreak@134 136 if isCastable then
flickerstreak@137 137 self.frames.hotkey:Show()
flickerstreak@134 138 icon:SetVertexColor(1.0, 1.0, 1.0)
flickerstreak@134 139 else
flickerstreak@134 140 icon:SetVertexColor(0.4, 0.4, 0.4)
flickerstreak@134 141 end
flickerstreak@134 142
flickerstreak@134 143 self:UpdateCooldown()
flickerstreak@134 144 end
flickerstreak@134 145
flickerstreak@134 146 function Stance:UpdateCooldown()
flickerstreak@179 147 local start, duration, enabled = GetShapeshiftFormCooldown(self:GetActionID())
flickerstreak@134 148 if start then
flickerstreak@134 149 CooldownFrame_SetTimer(self.frames.cooldown, start, duration, enabled)
flickerstreak@134 150 end
flickerstreak@134 151 end
flickerstreak@134 152
flickerstreak@134 153 function Stance:SetTooltip()
flickerstreak@134 154 if GetCVar("UberTooltips") == "1" then
flickerstreak@134 155 GameTooltip_SetDefaultAnchor(GameTooltip, self:GetFrame())
flickerstreak@134 156 else
flickerstreak@134 157 GameTooltip:SetOwner(self:GetFrame(), "ANCHOR_RIGHT")
flickerstreak@134 158 end
flickerstreak@179 159 GameTooltip:SetShapeshift(self:GetActionID())
flickerstreak@134 160 end
flickerstreak@134 161
flickerstreak@134 162 function Stance:OnEnter()
flickerstreak@134 163 self:SetTooltip()
flickerstreak@134 164 end
flickerstreak@134 165
flickerstreak@134 166 function Stance:OnLeave()
flickerstreak@134 167 GameTooltip:Hide()
flickerstreak@134 168 end
flickerstreak@134 169
flickerstreak@134 170 function Stance:PreClick()
flickerstreak@134 171 local f = self:GetFrame()
flickerstreak@134 172 f:SetChecked( not f:GetChecked() )
flickerstreak@134 173 end
flickerstreak@134 174
flickerstreak@134 175 function Stance:OnEvent(event, arg)
flickerstreak@134 176 if event == "PLAYER_REGEN_ENABLED" then
flickerstreak@134 177 if self.updatePending then
flickerstreak@134 178 self:UpdateAction()
flickerstreak@134 179 end
flickerstreak@134 180 elseif event == "UPDATE_SHAPESHIFT_COOLDOWN" then
flickerstreak@134 181 self:UpdateCooldown()
flickerstreak@179 182 elseif event == "UPDATE_SHAPESHIFT_FORMS" then
flickerstreak@134 183 self:UpdateAction()
flickerstreak@134 184 elseif event == "UNIT_AURA" then
flickerstreak@134 185 if arg == "player" then
flickerstreak@134 186 self:Update()
flickerstreak@134 187 end
flickerstreak@137 188 elseif event == "UPDATE_BINDINGS" then
flickerstreak@137 189 self:UpdateHotkey()
flickerstreak@134 190 else
flickerstreak@134 191 self:Update()
flickerstreak@134 192 end
flickerstreak@134 193 end
flickerstreak@134 194
flickerstreak@134 195 function Stance:ShowGridTemp(show)
flickerstreak@134 196 if show then
flickerstreak@134 197 self:GetFrame():Show()
flickerstreak@134 198 else
flickerstreak@134 199 self:UpdateAction()
flickerstreak@134 200 end
flickerstreak@134 201 end