ActionButton.lua

Compare Download

ActionButton.lua (9.0.2.37176; unchanged since 9.0.1.36230)
  1. CURRENT_ACTIONBAR_PAGE = 1;
  2. NUM_ACTIONBAR_PAGES = 6;
  3. NUM_ACTIONBAR_BUTTONS = 12;
  4. NUM_OVERRIDE_BUTTONS = 6;
  5. ATTACK_BUTTON_FLASH_TIME = 0.4;
  7. BOTTOMLEFT_ACTIONBAR_PAGE = 6;
  8. BOTTOMRIGHT_ACTIONBAR_PAGE = 5;
  9. LEFT_ACTIONBAR_PAGE = 4;
  10. RIGHT_ACTIONBAR_PAGE = 3;
  11. RANGE_INDICATOR = "●";
  13. COOLDOWN_TYPE_LOSS_OF_CONTROL = 1;
  14. COOLDOWN_TYPE_NORMAL = 2;
  16. -- Table of actionbar pages and whether they're viewable or not
  17. VIEWABLE_ACTION_BAR_PAGES = {1, 1, 1, 1, 1, 1};
  19. ACTION_HIGHLIGHT_MARKS = { };
  20. ON_BAR_HIGHLIGHT_MARKS = { };
  22. ACTION_BUTTON_SHOW_GRID_REASON_CVAR = 1;
  23. ACTION_BUTTON_SHOW_GRID_REASON_EVENT = 2;
  24. ACTION_BUTTON_SHOW_GRID_REASON_SPELLBOOK = 4;
  26. function IsOnPrimaryActionBar(action)
  27. return action >= 1 and action <= NUM_ACTIONBAR_BUTTONS;
  28. end
  30. function MarkNewActionHighlight(action)
  31. ACTION_HIGHLIGHT_MARKS[action] = true;
  32. end
  34. function ClearNewActionHighlight(action, preventIdenticalActionsFromClearing)
  35. ACTION_HIGHLIGHT_MARKS[action] = nil;
  37. if preventIdenticalActionsFromClearing then
  38. return;
  39. end
  41. -- If we're unhighlighting this one because it was used/moused over/etc...
  42. -- then go find all other current actions that match this one that are also
  43. -- marked for highlight and unmark them. The next time they update the highlight
  44. -- will update; may need to actually force update the action button in some cases
  45. -- and that means that ACTION_HIGHLIGHT_MARKS needs to store more information
  46. local unmarkedType, unmarkedID = GetActionInfo(action);
  48. for actionKey, markValue in pairs(ACTION_HIGHLIGHT_MARKS) do
  49. if markValue then
  50. local actionType, actionID = GetActionInfo(actionKey);
  51. if actionType == unmarkedType and actionID == unmarkedID then
  52. ACTION_HIGHLIGHT_MARKS[actionKey] = nil;
  53. end
  54. end
  55. end
  56. end
  58. function GetNewActionHighlightMark(action)
  59. return ACTION_HIGHLIGHT_MARKS[action];
  60. end
  62. function ClearOnBarHighlightMarks()
  63. ON_BAR_HIGHLIGHT_MARKS = {};
  64. end
  66. function GetOnBarHighlightMark(action)
  67. return ON_BAR_HIGHLIGHT_MARKS[action];
  68. end
  70. local function UpdateOnBarHighlightMarks(actionButtonSlots)
  71. if actionButtonSlots then
  72. ON_BAR_HIGHLIGHT_MARKS = tInvert(actionButtonSlots);
  73. else
  74. ClearOnBarHighlightMarks();
  75. end
  76. end
  78. function UpdateOnBarHighlightMarksBySpell(spellID)
  79. UpdateOnBarHighlightMarks(C_ActionBar.FindSpellActionButtons(spellID));
  80. end
  82. function UpdateOnBarHighlightMarksByFlyout(flyoutID)
  83. UpdateOnBarHighlightMarks(C_ActionBar.FindFlyoutActionButtons(flyoutID));
  84. end
  86. function UpdateOnBarHighlightMarksByPetAction(petAction)
  87. UpdateOnBarHighlightMarks(C_ActionBar.FindPetActionButtons(petAction));
  88. end
  90. function GetActionButtonForID(id)
  91. if OverrideActionBar and OverrideActionBar:IsShown() then
  92. if id > NUM_OVERRIDE_BUTTONS then
  93. return;
  94. end
  96. return _G["OverrideActionBarButton"..id];
  97. end
  99. return _G["ActionButton"..id];
  100. end
  102. local function CheckUseActionButton(button, checkingFromDown)
  103. local actionButtonUseKeyDown = GetCVarBool("ActionButtonUseKeyDown");
  104. local doAction = (checkingFromDown and actionButtonUseKeyDown) or not (checkingFromDown or actionButtonUseKeyDown);
  106. if doAction then
  107. if not button.ZoneAbilityDisabled then
  108. SecureActionButton_OnClick(button, "LeftButton");
  110. if GetNewActionHighlightMark(button.action) then
  111. ClearNewActionHighlight(button.action);
  112. button:UpdateHighlightMark();
  113. end
  114. end
  115. button:UpdateState();
  116. end
  117. end
  119. local isInPetBattle = C_PetBattles.IsInBattle;
  120. local function CheckPetActionButtonEvent(id, isDown)
  121. if isInPetBattle() and PetBattleFrame then
  122. if isDown then
  123. PetBattleFrame_ButtonDown(id);
  124. else
  125. PetBattleFrame_ButtonUp(id);
  126. end
  127. return true;
  128. end
  130. return false;
  131. end
  133. function ActionButtonDown(id)
  134. if CheckPetActionButtonEvent(id, true) then
  135. return;
  136. end
  138. local button = GetActionButtonForID(id);
  139. if button then
  140. if button:GetButtonState() == "NORMAL" then
  141. button:SetButtonState("PUSHED");
  142. end
  144. CheckUseActionButton(button, true);
  145. end
  146. end
  148. function ActionButtonUp(id)
  149. if CheckPetActionButtonEvent(id, false) then
  150. return;
  151. end
  153. local button = GetActionButtonForID(id);
  154. if button then
  155. if ( button:GetButtonState() == "PUSHED" ) then
  156. button:SetButtonState("NORMAL");
  157. CheckUseActionButton(button, false);
  158. end
  159. end
  160. end
  162. function ActionBar_PageUp()
  163. local nextPage;
  164. for i=GetActionBarPage() + 1, NUM_ACTIONBAR_PAGES do
  165. if ( VIEWABLE_ACTION_BAR_PAGES[i] ) then
  166. nextPage = i;
  167. break;
  168. end
  169. end
  171. if ( not nextPage ) then
  172. nextPage = 1;
  173. end
  174. ChangeActionBarPage(nextPage);
  175. end
  177. function ActionBar_PageDown()
  178. local prevPage;
  179. for i=GetActionBarPage() - 1, 1, -1 do
  180. if ( VIEWABLE_ACTION_BAR_PAGES[i] ) then
  181. prevPage = i;
  182. break;
  183. end
  184. end
  186. if ( not prevPage ) then
  187. for i=NUM_ACTIONBAR_PAGES, 1, -1 do
  188. if ( VIEWABLE_ACTION_BAR_PAGES[i] ) then
  189. prevPage = i;
  190. break;
  191. end
  192. end
  193. end
  194. ChangeActionBarPage(prevPage);
  195. end
  197. ActionBarButtonEventsFrameMixin = {};
  199. function ActionBarButtonEventsFrameMixin:OnLoad()
  200. self.frames = {};
  201. self:RegisterEvent("PLAYER_ENTERING_WORLD");
  202. self:RegisterEvent("ACTIONBAR_SHOWGRID");
  203. self:RegisterEvent("ACTIONBAR_HIDEGRID");
  204. self:RegisterEvent("ACTIONBAR_SLOT_CHANGED");
  205. self:RegisterEvent("UPDATE_BINDINGS");
  206. self:RegisterEvent("UPDATE_SHAPESHIFT_FORM");
  207. self:RegisterEvent("ACTIONBAR_UPDATE_COOLDOWN");
  208. self:RegisterEvent("PET_BAR_UPDATE");
  209. self:RegisterUnitEvent("UNIT_FLAGS", "pet");
  210. self:RegisterUnitEvent("UNIT_AURA", "pet");
  211. self:RegisterEvent("PLAYER_MOUNT_DISPLAY_CHANGED");
  212. end
  214. function ActionBarButtonEventsFrameMixin:OnEvent(event, ...)
  215. -- pass event down to the buttons
  216. for k, frame in pairs(self.frames) do
  217. frame:OnEvent(event, ...);
  218. end
  219. end
  221. function ActionBarButtonEventsFrameMixin:RegisterFrame(frame)
  222. tinsert(self.frames, frame);
  223. end
  225. ActionBarActionEventsFrameMixin = {};
  227. function ActionBarActionEventsFrameMixin:OnLoad()
  228. self.frames = {};
  229. --self:RegisterEvent("ACTIONBAR_UPDATE_STATE"); not updating state from lua anymore, see SetActionUIButton
  230. self:RegisterEvent("ACTIONBAR_UPDATE_USABLE");
  231. --self:RegisterEvent("ACTIONBAR_UPDATE_COOLDOWN"); not updating cooldown from lua anymore, see SetActionUIButton
  232. self:RegisterEvent("SPELL_UPDATE_CHARGES");
  233. self:RegisterEvent("UPDATE_INVENTORY_ALERTS");
  234. self:RegisterEvent("PLAYER_TARGET_CHANGED");
  235. self:RegisterEvent("TRADE_SKILL_SHOW");
  236. self:RegisterEvent("TRADE_SKILL_CLOSE");
  237. self:RegisterEvent("ARCHAEOLOGY_CLOSED");
  238. self:RegisterEvent("PLAYER_ENTER_COMBAT");
  239. self:RegisterEvent("PLAYER_LEAVE_COMBAT");
  240. self:RegisterEvent("START_AUTOREPEAT_SPELL");
  241. self:RegisterEvent("STOP_AUTOREPEAT_SPELL");
  242. self:RegisterEvent("UNIT_ENTERED_VEHICLE");
  243. self:RegisterEvent("UNIT_EXITED_VEHICLE");
  244. self:RegisterEvent("COMPANION_UPDATE");
  245. self:RegisterEvent("UNIT_INVENTORY_CHANGED");
  246. self:RegisterEvent("LEARNED_SPELL_IN_TAB");
  247. self:RegisterEvent("PET_STABLE_UPDATE");
  248. self:RegisterEvent("PET_STABLE_SHOW");
  249. self:RegisterEvent("SPELL_ACTIVATION_OVERLAY_GLOW_SHOW");
  250. self:RegisterEvent("SPELL_ACTIVATION_OVERLAY_GLOW_HIDE");
  251. self:RegisterEvent("UPDATE_SUMMONPETS_ACTION");
  252. self:RegisterEvent("LOSS_OF_CONTROL_ADDED");
  253. self:RegisterEvent("LOSS_OF_CONTROL_UPDATE");
  254. self:RegisterEvent("SPELL_UPDATE_ICON");
  255. end
  257. function ActionBarActionEventsFrameMixin:OnEvent(event, ...)
  258. if ( event == "UNIT_INVENTORY_CHANGED" ) then
  259. local unit = ...;
  260. if ( unit == "player" and self.tooltipOwner and GameTooltip:GetOwner() == self.tooltipOwner ) then
  261. self.tooltipOwner:SetTooltip();
  262. end
  263. else
  264. for k, frame in pairs(self.frames) do
  265. frame:OnEvent(event, ...);
  266. end
  267. end
  268. end
  270. function ActionBarActionEventsFrameMixin:RegisterFrame(frame)
  271. self.frames[frame] = frame;
  272. end
  274. function ActionBarActionEventsFrameMixin:UnregisterFrame(frame)
  275. self.frames[frame] = nil;
  276. end
  278. ActionBarActionButtonMixin = {};
  280. function ActionBarActionButtonMixin:OnLoad()
  281. self.flashing = 0;
  282. self.flashtime = 0;
  283. self:SetAttribute("showgrid", 0);
  284. self:SetAttribute("type", "action");
  285. self:SetAttribute("checkselfcast", true);
  286. self:SetAttribute("checkfocuscast", true);
  287. self:SetAttribute("useparent-unit", true);
  288. self:SetAttribute("useparent-actionpage", true);
  289. self:RegisterForDrag("LeftButton", "RightButton");
  290. self:RegisterForClicks("AnyUp");
  291. ActionBarButtonEventsFrame:RegisterFrame(self);
  292. self:UpdateAction();
  293. self:UpdateHotkeys(self.buttonType);
  294. end
  296. function ActionBarActionButtonMixin:UpdateHotkeys(actionButtonType)
  297. local id;
  298. if ( not actionButtonType ) then
  299. actionButtonType = "ACTIONBUTTON";
  300. id = self:GetID();
  301. else
  302. if ( actionButtonType == "MULTICASTACTIONBUTTON" ) then
  303. id = self.buttonIndex;
  304. else
  305. id = self:GetID();
  306. end
  307. end
  309. local hotkey = self.HotKey;
  310. local key = GetBindingKey(actionButtonType..id) or
  311. GetBindingKey("CLICK "..self:GetName()..":LeftButton");
  313. local text = GetBindingText(key, 1);
  314. if ( text == "" ) then
  315. hotkey:SetText(RANGE_INDICATOR);
  316. hotkey:Hide();
  317. else
  318. hotkey:SetText(text);
  319. hotkey:Show();
  320. end
  321. end
  323. function ActionBarActionButtonMixin:UpdateAction(force)
  324. local action = self:CalculateAction();
  325. if ( action ~= self.action or force ) then
  326. self.action = action;
  327. SetActionUIButton(self, action, self.cooldown);
  328. self:Update();
  329. end
  330. end
  332. function ActionBarActionButtonMixin:Update()
  333. local action = self.action;
  334. local icon = self.icon;
  335. local buttonCooldown = self.cooldown;
  336. local texture = GetActionTexture(action);
  338. self.zoneAbilityDisabled = false;
  339. icon:SetDesaturated(false);
  340. local type, id = GetActionInfo(action);
  341. if ( HasAction(action) ) then
  342. if ( not self.eventsRegistered ) then
  343. ActionBarActionEventsFrame:RegisterFrame(self);
  344. self.eventsRegistered = true;
  345. end
  347. if ( not self:GetAttribute("statehidden") ) then
  348. self:Show();
  349. end
  350. self:UpdateState();
  351. self:UpdateUsable();
  352. ActionButton_UpdateCooldown(self);
  353. self:UpdateFlash();
  354. self:UpdateHighlightMark();
  355. self:UpdateSpellHighlightMark();
  356. else
  357. if ( self.eventsRegistered ) then
  358. ActionBarActionEventsFrame:UnregisterFrame(self);
  359. self.eventsRegistered = nil;
  360. end
  362. if ( self:GetAttribute("showgrid") == 0 ) then
  363. self:Hide();
  364. else
  365. buttonCooldown:Hide();
  366. end
  368. ClearChargeCooldown(self);
  370. self:ClearFlash();
  371. self:SetChecked(false);
  373. if self.LevelLinkLockIcon then
  374. self.LevelLinkLockIcon:SetShown(false);
  375. end
  376. end
  378. -- Add a green border if button is an equipped item
  379. local border = self.Border;
  380. if border then
  381. if ( IsEquippedAction(action) ) then
  382. border:SetVertexColor(0, 1.0, 0, 0.35);
  383. border:Show();
  384. else
  385. border:Hide();
  386. end
  387. end
  389. -- Update Action Text
  390. local actionName = self.Name;
  391. if actionName then
  392. if ( not IsConsumableAction(action) and not IsStackableAction(action) and (IsItemAction(action) or GetActionCount(action) == 0) ) then
  393. actionName:SetText(GetActionText(action));
  394. else
  395. actionName:SetText("");
  396. end
  397. end
  399. -- Update icon and hotkey text
  400. if ( texture ) then
  401. icon:SetTexture(texture);
  402. icon:Show();
  403. self.rangeTimer = -1;
  404. self:UpdateCount();
  405. else
  406. self.Count:SetText("");
  407. icon:Hide();
  408. buttonCooldown:Hide();
  409. self.rangeTimer = nil;
  410. local hotkey = self.HotKey;
  411. if ( hotkey:GetText() == RANGE_INDICATOR ) then
  412. hotkey:Hide();
  413. else
  414. hotkey:SetVertexColor(0.6, 0.6, 0.6);
  415. end
  416. end
  418. -- Update flyout appearance
  419. ActionButton_UpdateFlyout(self);
  421. self:UpdateOverlayGlow();
  423. -- Update tooltip
  424. if ( GameTooltip:GetOwner() == self ) then
  425. self:SetTooltip();
  426. end
  428. self.feedback_action = action;
  429. end
  431. function ActionBarActionButtonMixin:UpdateHighlightMark()
  432. if ( self.NewActionTexture ) then
  433. self.NewActionTexture:SetShown(GetNewActionHighlightMark(self.action));
  434. end
  435. end
  437. -- Shared between the action bar and the pet bar.
  438. function SharedActionButton_RefreshSpellHighlight(button, shown)
  439. if ( shown ) then
  440. button.SpellHighlightTexture:Show();
  441. button.SpellHighlightAnim:Play();
  442. else
  443. button.SpellHighlightTexture:Hide();
  444. button.SpellHighlightAnim:Stop();
  445. end
  446. end
  448. function ActionBarActionButtonMixin:UpdateSpellHighlightMark()
  449. if ( self.SpellHighlightTexture and self.SpellHighlightAnim ) then
  450. SharedActionButton_RefreshSpellHighlight(self, GetOnBarHighlightMark(self.action));
  451. end
  452. end
  454. function ActionBarActionButtonMixin:ShowGrid(reason)
  455. assert(reason);
  456. if ( issecure() ) then
  457. self:SetAttribute("showgrid", bit.bor(self:GetAttribute("showgrid"), reason));
  458. end
  460. if ( self.NormalTexture ) then
  461. self.NormalTexture:SetVertexColor(1.0, 1.0, 1.0, 0.5);
  462. end
  464. if ( self:GetAttribute("showgrid") > 0 and not self:GetAttribute("statehidden") ) then
  465. self:Show();
  466. end
  467. end
  469. function ActionBarActionButtonMixin:HideGrid(reason)
  470. assert(reason);
  472. local showgrid = self:GetAttribute("showgrid");
  474. if ( issecure() ) then
  475. if ( showgrid > 0 ) then
  476. self:SetAttribute("showgrid", bit.band(showgrid, bit.bnot(reason)));
  477. end
  478. end
  480. if ( self:GetAttribute("showgrid") == 0 and not HasAction(self.action) ) then
  481. self:Hide();
  482. end
  483. end
  485. function ActionBarActionButtonMixin:UpdateState()
  486. local action = self.action;
  487. local isChecked = IsCurrentAction(action) or IsAutoRepeatAction(action);
  488. self:SetChecked(isChecked);
  489. end
  491. function ActionBarActionButtonMixin:UpdateUsable()
  492. local icon = self.icon;
  493. local normalTexture = self.NormalTexture;
  494. if ( not normalTexture ) then
  495. return;
  496. end
  498. local isUsable, notEnoughMana = IsUsableAction(self.action);
  499. if ( isUsable ) then
  500. icon:SetVertexColor(1.0, 1.0, 1.0);
  501. normalTexture:SetVertexColor(1.0, 1.0, 1.0);
  502. elseif ( notEnoughMana ) then
  503. icon:SetVertexColor(0.5, 0.5, 1.0);
  504. normalTexture:SetVertexColor(0.5, 0.5, 1.0);
  505. else
  506. icon:SetVertexColor(0.4, 0.4, 0.4);
  507. normalTexture:SetVertexColor(1.0, 1.0, 1.0);
  508. end
  510. local isLevelLinkLocked = C_LevelLink.IsActionLocked(self.action);
  511. if not icon:IsDesaturated() then
  512. icon:SetDesaturated(isLevelLinkLocked);
  513. end
  515. if self.LevelLinkLockIcon then
  516. self.LevelLinkLockIcon:SetShown(isLevelLinkLocked);
  517. end
  518. end
  520. function ActionBarActionButtonMixin:UpdateCount()
  521. local text = self.Count;
  522. local action = self.action;
  523. if ( IsConsumableAction(action) or IsStackableAction(action) or (not IsItemAction(action) and GetActionCount(action) > 0) ) then
  524. local count = GetActionCount(action);
  525. if ( count > (self.maxDisplayCount or 9999 ) ) then
  526. text:SetText("*");
  527. else
  528. text:SetText(count);
  529. end
  530. else
  531. local charges, maxCharges, chargeStart, chargeDuration = GetActionCharges(action);
  532. if (maxCharges > 1) then
  533. text:SetText(charges);
  534. else
  535. text:SetText("");
  536. end
  537. end
  538. end
  540. -- Shared between action bar buttons and spell flyout buttons.
  541. function ActionButton_UpdateCooldown(self)
  542. local locStart, locDuration;
  543. local start, duration, enable, charges, maxCharges, chargeStart, chargeDuration;
  544. local modRate = 1.0;
  545. local chargeModRate = 1.0;
  546. if ( self.spellID ) then
  547. locStart, locDuration = GetSpellLossOfControlCooldown(self.spellID);
  548. start, duration, enable, modRate = GetSpellCooldown(self.spellID);
  549. charges, maxCharges, chargeStart, chargeDuration, chargeModRate = GetSpellCharges(self.spellID);
  550. else
  551. locStart, locDuration = GetActionLossOfControlCooldown(self.action);
  552. start, duration, enable, modRate = GetActionCooldown(self.action);
  553. charges, maxCharges, chargeStart, chargeDuration, chargeModRate = GetActionCharges(self.action);
  554. end
  556. if ( (locStart + locDuration) > (start + duration) ) then
  557. if ( self.cooldown.currentCooldownType ~= COOLDOWN_TYPE_LOSS_OF_CONTROL ) then
  558. self.cooldown:SetEdgeTexture("Interface\\Cooldown\\edge-LoC");
  559. self.cooldown:SetSwipeColor(0.17, 0, 0);
  560. self.cooldown:SetHideCountdownNumbers(true);
  561. self.cooldown.currentCooldownType = COOLDOWN_TYPE_LOSS_OF_CONTROL;
  562. end
  564. CooldownFrame_Set(self.cooldown, locStart, locDuration, true, true, modRate);
  565. ClearChargeCooldown(self);
  566. else
  567. if ( self.cooldown.currentCooldownType ~= COOLDOWN_TYPE_NORMAL ) then
  568. self.cooldown:SetEdgeTexture("Interface\\Cooldown\\edge");
  569. self.cooldown:SetSwipeColor(0, 0, 0);
  570. self.cooldown:SetHideCountdownNumbers(false);
  571. self.cooldown.currentCooldownType = COOLDOWN_TYPE_NORMAL;
  572. end
  574. if( locStart > 0 ) then
  575. self.cooldown:SetScript("OnCooldownDone", ActionButtonCooldown_OnCooldownDone);
  576. end
  578. if ( charges and maxCharges and maxCharges > 1 and charges < maxCharges ) then
  579. StartChargeCooldown(self, chargeStart, chargeDuration, chargeModRate);
  580. else
  581. ClearChargeCooldown(self);
  582. end
  584. CooldownFrame_Set(self.cooldown, start, duration, enable, false, modRate);
  585. end
  586. end
  588. function ActionButtonCooldown_OnCooldownDone(self)
  589. self:SetScript("OnCooldownDone", nil);
  590. ActionButton_UpdateCooldown(self:GetParent());
  591. end
  593. -- Charge Cooldown stuff
  595. local numChargeCooldowns = 0;
  596. local function CreateChargeCooldownFrame(parent)
  597. numChargeCooldowns = numChargeCooldowns + 1;
  598. cooldown = CreateFrame("Cooldown", "ChargeCooldown"..numChargeCooldowns, parent, "CooldownFrameTemplate");
  599. cooldown:SetHideCountdownNumbers(true);
  600. cooldown:SetDrawSwipe(false);
  602. cooldown:SetAllPoints(parent);
  603. cooldown:SetFrameStrata("TOOLTIP");
  605. return cooldown;
  606. end
  608. function StartChargeCooldown(parent, chargeStart, chargeDuration, chargeModRate)
  609. if chargeStart == 0 then
  610. ClearChargeCooldown(parent);
  611. return;
  612. end
  614. parent.chargeCooldown = parent.chargeCooldown or CreateChargeCooldownFrame(parent);
  616. CooldownFrame_Set(parent.chargeCooldown, chargeStart, chargeDuration, true, true, chargeModRate);
  617. end
  619. function ClearChargeCooldown(parent)
  620. if parent.chargeCooldown then
  621. CooldownFrame_Clear(parent.chargeCooldown);
  622. end
  623. end
  626. --Overlay stuff
  627. local unusedOverlayGlows = {};
  628. local numOverlays = 0;
  629. function ActionButton_GetOverlayGlow()
  630. local overlay = tremove(unusedOverlayGlows);
  631. if ( not overlay ) then
  632. numOverlays = numOverlays + 1;
  633. overlay = CreateFrame("Frame", "ActionButtonOverlay"..numOverlays, UIParent, "ActionBarButtonSpellActivationAlert");
  634. end
  635. return overlay;
  636. end
  638. function ActionBarActionButtonMixin:UpdateOverlayGlow()
  639. local spellType, id, subType = GetActionInfo(self.action);
  640. if ( spellType == "spell" and IsSpellOverlayed(id) ) then
  641. ActionButton_ShowOverlayGlow(self);
  642. elseif ( spellType == "macro" ) then
  643. local spellId = GetMacroSpell(id);
  644. if ( spellId and IsSpellOverlayed(spellId) ) then
  645. ActionButton_ShowOverlayGlow(self);
  646. else
  647. ActionButton_HideOverlayGlow(self);
  648. end
  649. else
  650. ActionButton_HideOverlayGlow(self);
  651. end
  652. end
  654. -- Shared between action button and MainMenuBarMicroButton
  655. function ActionButton_ShowOverlayGlow(button)
  656. if ( button.overlay ) then
  657. if ( button.overlay.animOut:IsPlaying() ) then
  658. button.overlay.animOut:Stop();
  659. button.overlay.animIn:Play();
  660. end
  661. else
  662. button.overlay = ActionButton_GetOverlayGlow();
  663. local frameWidth, frameHeight = button:GetSize();
  664. button.overlay:SetParent(button);
  665. button.overlay:ClearAllPoints();
  666. --Make the height/width available before the next frame:
  667. button.overlay:SetSize(frameWidth * 1.4, frameHeight * 1.4);
  668. button.overlay:SetPoint("TOPLEFT", button, "TOPLEFT", -frameWidth * 0.2, frameHeight * 0.2);
  669. button.overlay:SetPoint("BOTTOMRIGHT", button, "BOTTOMRIGHT", frameWidth * 0.2, -frameHeight * 0.2);
  670. button.overlay.animIn:Play();
  671. end
  672. end
  674. -- Shared between action button and MainMenuBarMicroButton
  675. function ActionButton_HideOverlayGlow(button)
  676. if ( button.overlay ) then
  677. if ( button.overlay.animIn:IsPlaying() ) then
  678. button.overlay.animIn:Stop();
  679. end
  680. if ( button:IsVisible() ) then
  681. button.overlay.animOut:Play();
  682. else
  683. button.overlay.animOut:OnFinished(); --We aren't shown anyway, so we'll instantly hide it.
  684. end
  685. end
  686. end
  688. ActionBarOverlayGlowAnimOutMixin = {};
  690. function ActionBarOverlayGlowAnimOutMixin:OnFinished()
  691. local overlay = self:GetParent();
  692. local actionButton = overlay:GetParent();
  693. overlay:Hide();
  694. tinsert(unusedOverlayGlows, overlay);
  695. actionButton.overlay = nil;
  696. end
  698. ActionBarOverlayGlowAnimInMixin = {};
  700. function ActionBarOverlayGlowAnimInMixin:OnPlay()
  701. local frame = self:GetParent();
  702. local frameWidth, frameHeight = frame:GetSize();
  703. frame.spark:SetSize(frameWidth, frameHeight);
  704. frame.spark:SetAlpha(0.3);
  705. frame.innerGlow:SetSize(frameWidth / 2, frameHeight / 2);
  706. frame.innerGlow:SetAlpha(1.0);
  707. frame.innerGlowOver:SetAlpha(1.0);
  708. frame.outerGlow:SetSize(frameWidth * 2, frameHeight * 2);
  709. frame.outerGlow:SetAlpha(1.0);
  710. frame.outerGlowOver:SetAlpha(1.0);
  711. frame.ants:SetSize(frameWidth * 0.85, frameHeight * 0.85)
  712. frame.ants:SetAlpha(0);
  713. frame:Show();
  714. end
  716. function ActionBarOverlayGlowAnimInMixin:OnFinished()
  717. local frame = self:GetParent();
  718. local frameWidth, frameHeight = frame:GetSize();
  719. frame.spark:SetAlpha(0);
  720. frame.innerGlow:SetAlpha(0);
  721. frame.innerGlow:SetSize(frameWidth, frameHeight);
  722. frame.innerGlowOver:SetAlpha(0.0);
  723. frame.outerGlow:SetSize(frameWidth, frameHeight);
  724. frame.outerGlowOver:SetAlpha(0.0);
  725. frame.outerGlowOver:SetSize(frameWidth, frameHeight);
  726. frame.ants:SetAlpha(1.0);
  727. end
  729. ActionBarButtonSpellActivationAlertMixin = {};
  731. function ActionBarButtonSpellActivationAlertMixin:OnUpdate(elapsed)
  732. AnimateTexCoords(self.ants, 256, 256, 48, 48, 22, elapsed, 0.01);
  733. local cooldown = self:GetParent().cooldown;
  734. -- we need some threshold to avoid dimming the glow during the gdc
  735. -- (using 1500 exactly seems risky, what if casting speed is slowed or something?)
  736. if(cooldown and cooldown:IsShown() and cooldown:GetCooldownDuration() > 3000) then
  737. self:SetAlpha(0.5);
  738. else
  739. self:SetAlpha(1.0);
  740. end
  741. end
  743. function ActionBarButtonSpellActivationAlertMixin:OnHide()
  744. if ( self.animOut:IsPlaying() ) then
  745. self.animOut:Stop();
  746. self.animOut:OnFinished();
  747. end
  748. end
  750. function ActionBarActionButtonMixin:OnEvent(event, ...)
  751. local arg1 = ...;
  752. if ((event == "UNIT_INVENTORY_CHANGED" and arg1 == "player") or event == "LEARNED_SPELL_IN_TAB") then
  753. if ( GameTooltip:GetOwner() == self ) then
  754. self:SetTooltip();
  755. end
  756. elseif ( event == "ACTIONBAR_SLOT_CHANGED" ) then
  757. if ( arg1 == 0 or arg1 == tonumber(self.action) ) then
  758. ClearNewActionHighlight(self.action, true);
  759. self:Update();
  760. end
  761. elseif ( event == "PLAYER_ENTERING_WORLD" ) then
  762. self:Update();
  763. elseif ( event == "UPDATE_SHAPESHIFT_FORM" ) then
  764. -- need to listen for UPDATE_SHAPESHIFT_FORM because attack icons change when the shapeshift form changes
  765. -- This is NOT intended to update everything about shapeshifting; most stuff should be handled by ActionBar-specific events such as UPDATE_BONUS_ACTIONBAR, UPDATE_USABLE, etc.
  766. local texture = GetActionTexture(self.action);
  767. if (texture) then
  768. self.icon:SetTexture(texture);
  769. end
  770. elseif ( event == "ACTIONBAR_SHOWGRID" ) then
  771. self:ShowGrid(ACTION_BUTTON_SHOW_GRID_REASON_EVENT);
  772. elseif ( event == "ACTIONBAR_HIDEGRID" ) then
  773. if ( not KeybindFrames_InQuickKeybindMode() ) then
  774. self:HideGrid(ACTION_BUTTON_SHOW_GRID_REASON_EVENT);
  775. end
  776. elseif ( event == "UPDATE_BINDINGS" ) then
  777. self:UpdateHotkeys(self.buttonType);
  778. elseif ( event == "PLAYER_TARGET_CHANGED" ) then -- All event handlers below this line are only set when the button has an action
  779. self.rangeTimer = -1;
  780. elseif ( event == "UNIT_FLAGS" or event == "UNIT_AURA" or event == "PET_BAR_UPDATE" ) then
  781. -- Pet actions can also change the state of action buttons.
  782. self:UpdateState();
  783. self:UpdateFlash();
  784. elseif ( (event == "ACTIONBAR_UPDATE_STATE") or
  785. ((event == "UNIT_ENTERED_VEHICLE" or event == "UNIT_EXITED_VEHICLE") and (arg1 == "player")) or
  786. ((event == "COMPANION_UPDATE") and (arg1 == "MOUNT")) ) then
  787. self:UpdateState();
  788. elseif ( event == "ACTIONBAR_UPDATE_USABLE" or event == "PLAYER_MOUNT_DISPLAY_CHANGED" ) then
  789. self:UpdateUsable();
  790. elseif ( event == "LOSS_OF_CONTROL_UPDATE" ) then
  791. ActionButton_UpdateCooldown(self);
  792. elseif ( event == "ACTIONBAR_UPDATE_COOLDOWN" or event == "LOSS_OF_CONTROL_ADDED" ) then
  793. ActionButton_UpdateCooldown(self);
  794. -- Update tooltip
  795. if ( GameTooltip:GetOwner() == self ) then
  796. self:SetTooltip();
  797. end
  798. elseif ( event == "TRADE_SKILL_SHOW" or event == "TRADE_SKILL_CLOSE" or event == "ARCHAEOLOGY_CLOSED" ) then
  799. self:UpdateState();
  800. elseif ( event == "PLAYER_ENTER_COMBAT" ) then
  801. if ( IsAttackAction(self.action) ) then
  802. self:StartFlash();
  803. end
  804. elseif ( event == "PLAYER_LEAVE_COMBAT" ) then
  805. if ( IsAttackAction(self.action) ) then
  806. self:StopFlash();
  807. end
  808. elseif ( event == "START_AUTOREPEAT_SPELL" ) then
  809. if ( IsAutoRepeatAction(self.action) ) then
  810. self:StartFlash();
  811. end
  812. elseif ( event == "STOP_AUTOREPEAT_SPELL" ) then
  813. if ( self:IsFlashing() and not IsAttackAction(self.action) ) then
  814. self:StopFlash();
  815. end
  816. elseif ( event == "PET_STABLE_UPDATE" or event == "PET_STABLE_SHOW") then
  817. -- Has to update everything for now, but this event should happen infrequently
  818. self:Update();
  819. elseif ( event == "SPELL_ACTIVATION_OVERLAY_GLOW_SHOW" ) then
  820. local actionType, id, subType = GetActionInfo(self.action);
  821. if ( actionType == "spell" and id == arg1 ) then
  822. ActionButton_ShowOverlayGlow(self);
  823. elseif ( actionType == "macro" ) then
  824. local spellId = GetMacroSpell(id);
  825. if ( spellId and spellId == arg1 ) then
  826. ActionButton_ShowOverlayGlow(self);
  827. end
  828. elseif (actionType == "flyout" and FlyoutHasSpell(id, arg1)) then
  829. ActionButton_ShowOverlayGlow(self);
  830. end
  831. elseif ( event == "SPELL_ACTIVATION_OVERLAY_GLOW_HIDE" ) then
  832. local actionType, id, subType = GetActionInfo(self.action);
  833. if ( actionType == "spell" and id == arg1 ) then
  834. ActionButton_HideOverlayGlow(self);
  835. elseif ( actionType == "macro" ) then
  836. local spellId = GetMacroSpell(id);
  837. if (spellId and spellId == arg1 ) then
  838. ActionButton_HideOverlayGlow(self);
  839. end
  840. elseif (actionType == "flyout" and FlyoutHasSpell(id, arg1)) then
  841. ActionButton_HideOverlayGlow(self);
  842. end
  843. elseif ( event == "SPELL_UPDATE_CHARGES" ) then
  844. self:UpdateCount();
  845. elseif ( event == "UPDATE_SUMMONPETS_ACTION" ) then
  846. local actionType, id = GetActionInfo(self.action);
  847. if (actionType == "summonpet") then
  848. local texture = GetActionTexture(self.action);
  849. if (texture) then
  850. self.icon:SetTexture(texture);
  851. end
  852. end
  853. elseif ( event == "SPELL_UPDATE_ICON" ) then
  854. self:Update();
  855. end
  856. end
  858. function ActionBarActionButtonMixin:SetTooltip()
  859. local inQuickKeybind = KeybindFrames_InQuickKeybindMode();
  860. if ( GetCVar("UberTooltips") == "1" or inQuickKeybind ) then
  861. GameTooltip_SetDefaultAnchor(GameTooltip, self);
  862. else
  863. local parent = self:GetParent();
  864. if ( parent == MultiBarBottomRight or parent == MultiBarRight or parent == MultiBarLeft ) then
  865. GameTooltip:SetOwner(self, "ANCHOR_LEFT");
  866. else
  867. GameTooltip:SetOwner(self, "ANCHOR_RIGHT");
  868. end
  869. end
  870. if ( GameTooltip:SetAction(self.action) ) then
  871. self.UpdateTooltip = self.SetTooltip;
  872. else
  873. self.UpdateTooltip = nil;
  874. end
  875. end
  877. function ActionBarActionButtonMixin:OnUpdate(elapsed)
  878. if ( self:IsFlashing() ) then
  879. local flashtime = self.flashtime;
  880. flashtime = flashtime - elapsed;
  882. if ( flashtime <= 0 ) then
  883. local overtime = -flashtime;
  884. if ( overtime >= ATTACK_BUTTON_FLASH_TIME ) then
  885. overtime = 0;
  886. end
  887. flashtime = ATTACK_BUTTON_FLASH_TIME - overtime;
  889. local flashTexture = self.Flash;
  890. if ( flashTexture:IsShown() ) then
  891. flashTexture:Hide();
  892. else
  893. flashTexture:Show();
  894. end
  895. end
  897. self.flashtime = flashtime;
  898. end
  900. -- Handle range indicator
  901. local rangeTimer = self.rangeTimer;
  902. if ( rangeTimer ) then
  903. rangeTimer = rangeTimer - elapsed;
  905. if ( rangeTimer <= 0 ) then
  906. local valid = IsActionInRange(self.action);
  907. local checksRange = (valid ~= nil);
  908. local inRange = checksRange and valid;
  909. ActionButton_UpdateRangeIndicator(self, checksRange, inRange);
  910. rangeTimer = TOOLTIP_UPDATE_TIME;
  911. end
  913. self.rangeTimer = rangeTimer;
  914. end
  915. end
  917. -- Shared between the action bar and the pet bar.
  918. function ActionButton_UpdateRangeIndicator(self, checksRange, inRange)
  919. if ( self.HotKey:GetText() == RANGE_INDICATOR ) then
  920. if ( checksRange ) then
  921. self.HotKey:Show();
  922. if ( inRange ) then
  923. self.HotKey:SetVertexColor(LIGHTGRAY_FONT_COLOR:GetRGB());
  924. else
  925. self.HotKey:SetVertexColor(RED_FONT_COLOR:GetRGB());
  926. end
  927. else
  928. self.HotKey:Hide();
  929. end
  930. else
  931. if ( checksRange and not inRange ) then
  932. self.HotKey:SetVertexColor(RED_FONT_COLOR:GetRGB());
  933. else
  934. self.HotKey:SetVertexColor(LIGHTGRAY_FONT_COLOR:GetRGB());
  935. end
  936. end
  937. end
  939. function ActionBarActionButtonMixin:GetPagedID()
  940. return self.action;
  941. end
  943. function ActionBarActionButtonMixin:UpdateFlash()
  944. local action = self.action;
  945. if ( (IsAttackAction(action) and IsCurrentAction(action)) or IsAutoRepeatAction(action) ) then
  946. self:StartFlash();
  948. local actionType, actionID, actionSubtype = GetActionInfo(action);
  949. if ( actionSubtype == "pet" ) then
  950. self:GetCheckedTexture():SetAlpha(0.5);
  951. else
  952. self:GetCheckedTexture():SetAlpha(1.0);
  953. end
  954. else
  955. self:StopFlash();
  956. end
  958. if ( self.AutoCastable ) then
  959. self.AutoCastable:SetShown(C_ActionBar.IsAutoCastPetAction(action));
  960. if ( C_ActionBar.IsEnabledAutoCastPetAction(action) ) then
  961. self.AutoCastShine:Show();
  962. AutoCastShine_AutoCastStart(self.AutoCastShine);
  963. else
  964. self.AutoCastShine:Hide();
  965. AutoCastShine_AutoCastStop(self.AutoCastShine);
  966. end
  967. end
  968. end
  970. function ActionBarActionButtonMixin:ClearFlash()
  971. if ( self.AutoCastable ) then
  972. self.AutoCastable:Hide();
  973. self.AutoCastShine:Hide();
  974. AutoCastShine_AutoCastStop(self.AutoCastShine);
  975. end
  976. end
  978. function ActionBarActionButtonMixin:StartFlash()
  979. self.flashing = 1;
  980. self.flashtime = 0;
  981. self:UpdateState();
  982. end
  984. function ActionBarActionButtonMixin:StopFlash()
  985. self.flashing = 0;
  986. self.Flash:Hide();
  987. self:UpdateState();
  988. end
  990. function ActionBarActionButtonMixin:IsFlashing()
  991. if ( self.flashing == 1 ) then
  992. return 1;
  993. end
  995. return nil;
  996. end
  998. -- Shared between action bar buttons and spell flyout buttons.
  999. function ActionButton_UpdateFlyout(self)
  1000. if not self.FlyoutArrow then
  1001. return;
  1002. end
  1004. local actionType = GetActionInfo(self.action);
  1005. if (actionType == "flyout") then
  1006. -- Update border and determine arrow position
  1007. local arrowDistance;
  1008. if ((SpellFlyout and SpellFlyout:IsShown() and SpellFlyout:GetParent() == self) or GetMouseFocus() == self) then
  1009. self.FlyoutBorder:Show();
  1010. self.FlyoutBorderShadow:Show();
  1011. arrowDistance = 5;
  1012. else
  1013. self.FlyoutBorder:Hide();
  1014. self.FlyoutBorderShadow:Hide();
  1015. arrowDistance = 2;
  1016. end
  1018. -- Update arrow
  1019. self.FlyoutArrow:Show();
  1020. self.FlyoutArrow:ClearAllPoints();
  1021. local direction = self:GetAttribute("flyoutDirection");
  1022. if (direction == "LEFT") then
  1023. self.FlyoutArrow:SetPoint("LEFT", self, "LEFT", -arrowDistance, 0);
  1024. SetClampedTextureRotation(self.FlyoutArrow, 270);
  1025. elseif (direction == "RIGHT") then
  1026. self.FlyoutArrow:SetPoint("RIGHT", self, "RIGHT", arrowDistance, 0);
  1027. SetClampedTextureRotation(self.FlyoutArrow, 90);
  1028. elseif (direction == "DOWN") then
  1029. self.FlyoutArrow:SetPoint("BOTTOM", self, "BOTTOM", 0, -arrowDistance);
  1030. SetClampedTextureRotation(self.FlyoutArrow, 180);
  1031. else
  1032. self.FlyoutArrow:SetPoint("TOP", self, "TOP", 0, arrowDistance);
  1033. SetClampedTextureRotation(self.FlyoutArrow, 0);
  1034. end
  1035. else
  1036. self.FlyoutBorder:Hide();
  1037. self.FlyoutBorderShadow:Hide();
  1038. self.FlyoutArrow:Hide();
  1039. end
  1040. end
  1042. function ActionBarActionButtonMixin:OnClick(button, down)
  1043. if ( KeybindFrames_InQuickKeybindMode() ) then
  1044. local cursorType = GetCursorInfo();
  1045. if ( cursorType ) then
  1046. local slotID = self:CalculateAction(button);
  1047. C_ActionBar.PutActionInSlot(slotID);
  1048. end
  1049. else
  1050. if button == "RightButton" and C_ActionBar.IsAutoCastPetAction(self.action) then
  1051. C_ActionBar.ToggleAutoCastPetAction(self.action);
  1052. elseif (not self.zoneAbilityDisabled) then
  1053. SecureActionButton_OnClick(self, button);
  1054. end
  1055. end
  1056. end
  1058. function ActionBarActionButtonMixin:OnDragStart()
  1059. if ( LOCK_ACTIONBAR ~= "1" or IsModifiedClick("PICKUPACTION") ) then
  1060. SpellFlyout:Hide();
  1061. PickupAction(self.action);
  1062. self:UpdateState();
  1063. self:UpdateFlash();
  1064. end
  1065. end
  1067. function ActionBarActionButtonMixin:OnReceiveDrag()
  1068. PlaceAction(self.action);
  1069. self:UpdateState();
  1070. self:UpdateFlash();
  1071. end
  1073. function ActionBarActionButtonMixin:OnEnter()
  1074. if (self.NewActionTexture) then
  1075. ClearNewActionHighlight(self.action);
  1076. self:UpdateAction(true);
  1077. end
  1078. self:SetTooltip();
  1079. ActionBarButtonEventsFrame.tooltipOwner = self;
  1080. ActionBarActionEventsFrame.tooltipOwner = self;
  1081. ActionButton_UpdateFlyout(self);
  1082. end
  1084. function ActionBarActionButtonMixin:OnLeave()
  1085. GameTooltip:Hide();
  1086. ActionBarButtonEventsFrame.tooltipOwner = nil;
  1087. ActionBarActionEventsFrame.tooltipOwner = nil;
  1088. ActionButton_UpdateFlyout(self);
  1089. end