Blizzard_ArenaUI/Blizzard_ArenaUI.lua

Compare Download

Blizzard_ArenaUI/Blizzard_ArenaUI.lua (9.0.2.37176; unchanged since 9.0.1.36230)
  1. MAX_ARENA_ENEMIES = 5;
  3. function ArenaEnemyFrames_OnLoad(self)
  4. self:RegisterEvent("CVAR_UPDATE");
  5. self:RegisterEvent("VARIABLES_LOADED");
  6. self:RegisterEvent("PLAYER_ENTERING_WORLD");
  8. ArenaEnemyFrames_CheckEffectiveEnableState(self);
  9. local showCastbars = GetCVarBool("showArenaEnemyCastbar");
  10. local castFrame;
  11. for i = 1, MAX_ARENA_ENEMIES do
  12. castFrame = _G["ArenaEnemyFrame"..i.."CastingBar"];
  13. castFrame:SetPoint("RIGHT", _G["ArenaEnemyFrame"..i], "LEFT", -32, -3);
  14. castFrame.showCastbar = showCastbars;
  15. CastingBarFrame_UpdateIsShown(castFrame);
  16. end
  18. UpdateArenaEnemyBackground(GetCVarBool("showPartyBackground"));
  19. ArenaEnemyBackground_SetOpacity(tonumber(GetCVar("partyBackgroundOpacity")));
  20. ArenaEnemyFrames_UpdateVisible();
  21. ArenaEnemyFrames_ResetCrowdControlCooldownData();
  22. end
  24. function ArenaEnemyFrames_OnEvent(self, event, ...)
  25. local arg1, arg2 = ...;
  26. if ( (event == "CVAR_UPDATE") and (arg1 == "SHOW_ARENA_ENEMY_FRAMES_TEXT") ) then
  27. ArenaEnemyFrames_CheckEffectiveEnableState(self, arg2 == "1");
  28. elseif ( event == "VARIABLES_LOADED" ) then
  29. ArenaEnemyFrames_CheckEffectiveEnableState(self);
  30. local showCastbars = GetCVarBool("showArenaEnemyCastbar");
  31. local castFrame;
  32. for i = 1, MAX_ARENA_ENEMIES do
  33. castFrame = _G["ArenaEnemyFrame"..i.."CastingBar"];
  34. castFrame.showCastbar = showCastbars;
  35. CastingBarFrame_UpdateIsShown(castFrame);
  36. end
  37. for i=1, MAX_ARENA_ENEMIES do
  38. ArenaEnemyFrame_UpdatePet(_G["ArenaEnemyFrame"..i], i, true);
  39. end
  40. UpdateArenaEnemyBackground(GetCVarBool("showPartyBackground"));
  41. ArenaEnemyBackground_SetOpacity(tonumber(GetCVar("partyBackgroundOpacity")));
  42. elseif ( event == "PLAYER_ENTERING_WORLD" ) then
  43. ArenaEnemyFrames_CheckEffectiveEnableState(self);
  44. ArenaEnemyFrames_UpdateVisible();
  45. ArenaEnemyFrames_ResetCrowdControlCooldownData();
  46. end
  47. end
  49. function ArenaEnemyFrames_ResetCrowdControlCooldownData()
  50. for i=1, MAX_ARENA_ENEMIES do
  51. local frame = _G["ArenaEnemyFrame"..i];
  52. frame.CC.spellID = nil;
  53. frame.CC.Icon:SetTexture(nil);
  54. frame.CC.Cooldown:Clear();
  55. ArenaEnemyFrame_UpdateCrowdControl(frame);
  56. end
  57. end
  59. function ArenaEnemyFrames_OnShow(self)
  60. DurabilityFrame:SetAlerts();
  61. UIParent_ManageFramePositions();
  62. end
  64. function ArenaEnemyFrames_OnHide(self)
  65. DurabilityFrame:SetAlerts();
  66. UIParent_ManageFramePositions();
  67. end
  69. function ArenaEnemyFrames_CheckEffectiveEnableState(self, cvarUpdate)
  70. if (C_PvP.IsInBrawl()) then
  71. ArenaEnemyFrames_Disable(self);
  72. else
  73. if ( GetCVarBool("showArenaEnemyFrames") or cvarUpdate ) then
  74. ArenaEnemyFrames_Enable(self);
  75. else
  76. ArenaEnemyFrames_Disable(self);
  77. end
  78. end
  79. end
  81. function ArenaEnemyFrames_Enable(self)
  82. self.show = true;
  83. ArenaEnemyFrames_UpdateVisible();
  84. end
  86. function ArenaEnemyFrames_Disable(self)
  87. self.show = false;
  88. ArenaEnemyFrames_UpdateVisible();
  89. end
  91. function ArenaEnemyFrames_UpdateVisible()
  92. local _, instanceType = IsInInstance();
  93. if ( ArenaEnemyFrames.show and ((instanceType == "arena") or (GetNumArenaOpponents() > 0))) then
  94. ArenaEnemyFrames:Show();
  95. else
  96. ArenaEnemyFrames:Hide();
  97. end
  98. end
  100. function ArenaEnemyFrame_OnLoad(self)
  101. local id = self:GetID();
  102. self.debuffCountdown = 0;
  103. self.numDebuffs = 0;
  104. self.noTextPrefix = 1;
  105. local prefix = "ArenaEnemyFrame"..id;
  106. UnitFrame_Initialize(self, "arena"..id, _G[prefix.."Name"], nil,
  107. _G[prefix.."HealthBar"], _G[prefix.."HealthBarText"],
  108. _G[prefix.."ManaBar"], _G[prefix.."ManaBarText"], nil, nil, nil,
  109. _G[prefix.."MyHealPredictionBar"], _G[prefix.."OtherHealPredictionBar"],
  110. _G[prefix.."TotalAbsorbBar"], _G[prefix.."TotalAbsorbBarOverlay"], _G[prefix.."OverAbsorbGlow"],
  111. _G[prefix.."OverHealAbsorbGlow"], _G[prefix.."HealAbsorbBar"], _G[prefix.."HealAbsorbBarLeftShadow"],
  112. _G[prefix.."HealAbsorbBarRightShadow"]);
  113. SetTextStatusBarTextZeroText(_G[prefix.."HealthBar"], DEAD);
  115. self.statusCounter = 0;
  116. self.statusSign = -1;
  117. self.unitHPPercent = 1;
  119. self.classPortrait = _G[self:GetName().."ClassPortrait"];
  120. self.specPortrait = _G[self:GetName().."SpecPortrait"];
  121. self.specBorder = _G[self:GetName().."SpecBorder"];
  122. self.castBar = _G[self:GetName().."CastingBar"];
  124. ArenaEnemyFrame_UpdatePlayer(self, true);
  125. self:RegisterEvent("UNIT_PET");
  126. self:RegisterEvent("ARENA_OPPONENT_UPDATE");
  127. self:RegisterEvent("UNIT_NAME_UPDATE");
  128. self:RegisterEvent("ARENA_COOLDOWNS_UPDATE");
  129. self:RegisterEvent("ARENA_CROWD_CONTROL_SPELL_UPDATE");
  131. UIDropDownMenu_Initialize(self.DropDown, ArenaEnemyDropDown_Initialize, "MENU");
  133. local setfocus = function()
  134. FocusUnit("arena"..self:GetID());
  135. end
  136. SecureUnitButton_OnLoad(self, "arena"..self:GetID(), setfocus);
  138. local id = self:GetID();
  139. if ( UnitClass("arena"..id) and (not UnitExists("arena"..id))) then --It is possible for the unit itself to no longer exist on the client, but some of the information to remain (after reloading the UI)
  140. self:Show();
  141. ArenaEnemyFrame_Lock(self);
  142. elseif ( UnitExists("arenapet"..id) and ( not UnitClass("arena"..id) ) ) then --We use UnitClass because even if the unit doesn't exist on the client, we may still have enough info to populate the frame.
  143. ArenaEnemyFrame_SetMysteryPlayer(self);
  144. end
  145. end
  147. function ArenaEnemyFrame_UpdatePlayer(self, useCVars)--At some points, we need to use CVars instead of UVars even though UVars are faster.
  148. local id = self:GetID();
  149. if ( UnitGUID(self.unit) ) then --Use UnitGUID instead of UnitExists in case the unit is a remote update.
  150. self:Show();
  151. _G["ArenaPrepFrame"..id]:Hide();
  152. UnitFrame_Update(self);
  153. end
  155. local _, class = UnitClass(self.unit);
  156. if( class ) then
  157. self.classPortrait:SetTexture("Interface\\TargetingFrame\\UI-Classes-Circles");
  158. self.classPortrait:SetTexCoord(unpack(CLASS_ICON_TCOORDS[class]));
  159. end
  160. local specID = GetArenaOpponentSpec(id);
  161. if (specID and specID > 0) then
  162. local _, _, _, specIcon = GetSpecializationInfoByID(specID);
  163. self.specBorder:Show();
  164. SetPortraitToTexture(self.specPortrait, specIcon);
  165. else
  166. self.specPortrait:SetTexture(nil);
  167. self.specBorder:Hide();
  168. end
  170. -- When not in an arena, show their faction icon (these are really flag carriers, not arena opponents)
  171. local _, instanceType = IsInInstance();
  172. local factionGroup, factionName = UnitFactionGroup(self.unit);
  173. local pvpIcon = _G[self:GetName() .. "PVPIcon"];
  174. if ( factionGroup and factionGroup ~= "Neutral" and instanceType ~= "arena" ) then
  175. pvpIcon:SetTexture("Interface\\TargetingFrame\\UI-PVP-"..factionGroup);
  176. pvpIcon:Show();
  177. self:SetPoint("RIGHT", self:GetParent(), "RIGHT", -18, 0);
  178. else
  179. pvpIcon:Hide();
  180. self:SetPoint("RIGHT", self:GetParent(), "RIGHT", -2, 0);
  181. end
  183. CastingBarFrame_SetUnit(self.castBar, self.unit, false, true);
  185. ArenaEnemyFrames_UpdateVisible();
  186. end
  188. function ArenaEnemyFrame_Lock(self)
  189. self.healthbar:SetStatusBarColor(0.5, 0.5, 0.5);
  190. self.healthbar.lockColor = true;
  191. self.manabar:SetStatusBarColor(0.5, 0.5, 0.5);
  192. self.manabar.lockColor = true;
  193. self.hideStatusOnTooltip = true;
  194. end
  196. function ArenaEnemyFrame_Unlock(self)
  197. self.healthbar.lockColor = false;
  198. self.healthbar.forceHideText = false;
  199. self.manabar.lockColor = false;
  200. self.manabar.forceHideText = false;
  201. self.hideStatusOnTooltip = false;
  202. end
  204. function ArenaEnemyFrame_SetMysteryPlayer(self)
  205. self.healthbar:SetMinMaxValues(0,100);
  206. self.healthbar:SetValue(100);
  207. self.healthbar.forceHideText = true;
  208. self.manabar:SetMinMaxValues(0,100);
  209. self.manabar:SetValue(100);
  210. self.manabar.forceHideText = true;
  211. self.classPortrait:SetTexture("Interface\\CharacterFrame\\TempPortrait");
  212. self.classPortrait:SetTexCoord(0, 1, 0, 1);
  213. self.name:SetText("");
  214. ArenaEnemyFrame_Lock(self);
  215. self:Show();
  216. end
  218. function ArenaEnemyFrame_OnEvent(self, event, unit, ...)
  219. if ( unit == self.unit ) then
  220. if ( event == "ARENA_OPPONENT_UPDATE" ) then
  221. local unitEvent = ...;
  222. if ( unitEvent == "seen" or unitEvent == "destroyed") then
  223. ArenaEnemyFrame_Unlock(self);
  224. ArenaEnemyFrame_UpdatePlayer(self);
  226. if ( self.healthbar.frequentUpdates and GetCVarBool("predictedHealth") ) then
  227. self.healthbar:SetScript("OnUpdate", UnitFrameHealthBar_OnUpdate);
  228. self.healthbar:UnregisterEvent("UNIT_HEALTH");
  229. end
  230. if ( self.manabar.frequentUpdates ) then
  231. self.manabar:SetScript("OnUpdate", UnitFrameManaBar_OnUpdate);
  232. UnitFrameManaBar_UnregisterDefaultEvents(self.manabar);
  233. end
  234. ArenaEnemyFrame_UpdatePet(self);
  235. UpdateArenaEnemyBackground();
  236. UIParent_ManageFramePositions();
  237. elseif ( unitEvent == "unseen" ) then
  238. ArenaEnemyFrame_Lock(self);
  240. self.healthbar:RegisterEvent("UNIT_HEALTH");
  241. self.healthbar:SetScript("OnUpdate", nil);
  242. UnitFrameManaBar_RegisterDefaultEvents(self.manabar);
  243. self.manabar:SetScript("OnUpdate", nil);
  244. elseif ( unitEvent == "cleared" ) then
  245. ArenaEnemyFrame_Unlock(self);
  246. self:Hide();
  247. ArenaEnemyFrames_UpdateVisible();
  248. local _, instanceType = IsInInstance();
  249. if (instanceType ~= "arena") then
  250. ArenaPrepFrames:Hide()
  251. end
  252. end
  253. elseif ( event == "UNIT_PET" ) then
  254. ArenaEnemyFrame_UpdatePet(self);
  255. elseif ( event == "UNIT_NAME_UPDATE" ) then
  256. ArenaEnemyFrame_UpdatePlayer(self);
  257. elseif ( event == "UNIT_MAXHEALTH" or event == "UNIT_HEAL_PREDICTION" or event == "UNIT_ABSORB_AMOUNT_CHANGED" or event == "UNIT_HEAL_ABSORB_AMOUNT_CHANGED" ) then
  258. UnitFrameHealPredictionBars_Update(self);
  259. elseif ( event == "ARENA_COOLDOWNS_UPDATE" ) then
  260. ArenaEnemyFrame_UpdateCrowdControl(self);
  261. elseif ( event == "ARENA_CROWD_CONTROL_SPELL_UPDATE" ) then
  262. local spellID = ...;
  263. if (spellID ~= self.CC.spellID) then
  264. local spellTexture, spellTextureNoOverride = GetSpellTexture(spellID);
  265. self.CC.spellID = spellID;
  266. self.CC.Icon:SetTexture(spellTextureNoOverride);
  267. end
  268. end
  269. end
  270. end
  272. function ArenaEnemyFrame_UpdateCrowdControl(self)
  273. local spellID, startTime, duration = C_PvP.GetArenaCrowdControlInfo(self.unit);
  274. if (spellID) then
  275. if (spellID ~= self.CC.spellID) then
  276. local spellTexture, spellTextureNoOverride = GetSpellTexture(spellID);
  277. self.CC.spellID = spellID;
  278. self.CC.Icon:SetTexture(spellTextureNoOverride);
  279. end
  280. if (startTime ~= 0 and duration ~= 0) then
  281. self.CC.Cooldown:SetCooldown(startTime/1000.0, duration/1000.0);
  282. else
  283. self.CC.Cooldown:Clear();
  284. end
  285. end
  286. end
  288. function ArenaEnemyFrame_OnShow(self)
  289. self:SetFrameLevel(2);
  291. C_PvP.RequestCrowdControlSpell(self.unit);
  292. end
  294. function ArenaEnemyFrames_GetBestAnchorUnitFrameForOppponent(opponentNumber)
  295. return _G["ArenaEnemyFrame" .. math.min(opponentNumber, MAX_ARENA_ENEMIES)];
  296. end
  298. function ArenaEnemyFrame_UpdatePet(self, id, useCVars) --At some points, we need to use CVars instead of UVars even though UVars are faster.
  299. if ( not id ) then
  300. id = self:GetID();
  301. end
  303. local unitFrame = _G["ArenaEnemyFrame"..id];
  304. local petFrame = _G["ArenaEnemyFrame"..id.."PetFrame"];
  306. local showArenaEnemyPets = (SHOW_ARENA_ENEMY_PETS == "1");
  307. if ( useCVars ) then
  308. showArenaEnemyPets = GetCVarBool("showArenaEnemyPets");
  309. end
  311. if ( UnitGUID(petFrame.unit) and showArenaEnemyPets) then
  312. petFrame:Show();
  313. else
  314. petFrame:Hide();
  315. end
  317. UnitFrame_Update(petFrame);
  318. end
  320. function ArenaEnemyPetFrame_OnLoad(self)
  321. local id = self:GetParent():GetID();
  322. local prefix = "ArenaEnemyFrame"..id.."PetFrame";
  323. local unit = "arenapet"..id;
  324. UnitFrame_Initialize(self, unit, _G[prefix.."Name"], _G[prefix.."Portrait"],
  325. _G[prefix.."HealthBar"], _G[prefix.."HealthBarText"], _G[prefix.."ManaBar"], _G[prefix.."ManaBarText"]);
  326. SetTextStatusBarTextZeroText(_G[prefix.."HealthBar"], DEAD);
  327. _G[prefix.."Name"]:Hide();
  328. SecureUnitButton_OnLoad(self, unit);
  329. self:SetID(id);
  330. self:SetParent(ArenaEnemyFrames);
  331. ArenaEnemyFrame_UpdatePet(self, id, true);
  332. self:RegisterEvent("ARENA_OPPONENT_UPDATE");
  333. self:RegisterEvent("UNIT_CLASSIFICATION_CHANGED");
  335. UIDropDownMenu_Initialize(self.DropDown, ArenaEnemyPetDropDown_Initialize, "MENU");
  337. local setfocus = function()
  338. FocusUnit("arenapet"..self:GetID());
  339. end
  340. SecureUnitButton_OnLoad(self, "arenapet"..self:GetID(), setfocus);
  341. end
  343. function ArenaEnemyPetFrame_OnEvent(self, event, ...)
  344. local arg1, arg2 = ...;
  345. if ( event == "ARENA_OPPONENT_UPDATE" and arg1 == self.unit ) then
  346. if ( arg2 == "seen" or arg2 == "destroyed") then
  347. ArenaEnemyFrame_Unlock(self);
  348. ArenaEnemyFrame_UpdatePet(self);
  349. UpdateArenaEnemyBackground();
  350. local ownerFrame = _G["ArenaEnemyFrame"..self:GetID()];
  351. if ( not ownerFrame:IsShown() ) then
  352. ArenaEnemyFrame_SetMysteryPlayer(ownerFrame);
  353. ownerFrame:Show();
  354. end
  355. if ( self.healthbar.frequentUpdates and GetCVarBool("predictedHealth") ) then
  356. self.healthbar:SetScript("OnUpdate", UnitFrameHealthBar_OnUpdate);
  357. self.healthbar:UnregisterEvent("UNIT_HEALTH");
  358. end
  359. if ( self.manabar.frequentUpdates ) then
  360. self.manabar:SetScript("OnUpdate", UnitFrameManaBar_OnUpdate);
  361. UnitFrameManaBar_UnregisterDefaultEvents(self.manabar);
  362. end
  363. elseif ( arg2 == "unseen" ) then
  364. ArenaEnemyFrame_Lock(self);
  365. self.healthbar:RegisterEvent("UNIT_HEALTH");
  366. self.healthbar:SetScript("OnUpdate", nil);
  367. UnitFrameManaBar_RegisterDefaultEvents(self.manabar);
  368. self.manabar:SetScript("OnUpdate", nil);
  369. elseif ( arg2 == "cleared" ) then
  370. ArenaEnemyFrame_Unlock(self);
  371. self:Hide()
  372. end
  373. elseif ( event == "UNIT_CLASSIFICATION_CHANGED" and arg1 == self.unit ) then
  374. UnitFrame_Update(self);
  375. end
  376. UnitFrame_OnEvent(self, event, ...);
  377. end
  379. function ArenaEnemyDropDown_Initialize(self)
  380. UnitPopup_ShowMenu(self, "ARENAENEMY", "arena"..self:GetParent():GetID());
  381. end
  383. function ArenaEnemyPetDropDown_Initialize(self)
  384. UnitPopup_ShowMenu(self, "ARENAENEMY", "arenapet"..self:GetParent():GetID());
  385. end
  387. function UpdateArenaEnemyBackground(force)
  388. if ( (SHOW_PARTY_BACKGROUND == "1") or force ) then
  389. ArenaEnemyBackground:Show();
  390. local numOpps = min(GetNumArenaOpponents(), MAX_ARENA_ENEMIES);
  391. if ( numOpps > 0 ) then
  392. ArenaEnemyBackground:SetPoint("BOTTOMLEFT", "ArenaEnemyFrame"..numOpps.."PetFrame", "BOTTOMLEFT", -15, -10);
  393. else
  394. ArenaEnemyBackground:Hide();
  395. end
  396. else
  397. ArenaEnemyBackground:Hide();
  398. end
  400. end
  402. function ArenaEnemyBackground_SetOpacity(opacity)
  403. local alpha;
  404. if ( not opacity ) then
  405. alpha = 1.0 - OpacityFrameSlider:GetValue();
  406. else
  407. alpha = 1.0 - opacity;
  408. end
  409. ArenaEnemyBackground:SetAlpha(alpha);
  410. end
  412. -----------------------------------------------------------------------------
  413. --Arena preparation stuff, shows class and spec of opponents during countdown
  414. ------------------------------------------------------------------------------
  417. function ArenaPrepFrames_OnLoad(self)
  418. self:RegisterEvent("ARENA_PREP_OPPONENT_SPECIALIZATIONS");
  419. local numOpps = GetNumArenaOpponentSpecs();
  420. if (numOpps and numOpps > 0) then
  421. ArenaPrepFrames_OnEvent(self, "ARENA_PREP_OPPONENT_SPECIALIZATIONS");
  422. end
  423. end
  425. function ArenaPrepFrames_OnEvent(self, event, ...) --also called in OnLoad
  426. if (event == "ARENA_PREP_OPPONENT_SPECIALIZATIONS") then
  427. ArenaPrepFrames_UpdateFrames();
  428. self:Show()
  429. end
  430. end
  432. function ArenaPrepFrames_OnShow(self)
  433. DurabilityFrame:SetAlerts();
  434. UIParent_ManageFramePositions()
  435. end
  437. function ArenaPrepFrames_OnHide(self)
  438. DurabilityFrame:SetAlerts();
  439. UIParent_ManageFramePositions();
  440. end
  442. function ArenaPrepFrames_GetBestAnchorUnitFrameForOppponent(opponentNumber)
  443. return _G["ArenaPrepFrame" .. math.min(opponentNumber, MAX_ARENA_ENEMIES)];
  444. end
  446. function ArenaPrepFrames_UpdateFrames()
  447. local numOpps = GetNumArenaOpponentSpecs();
  448. for i=1, MAX_ARENA_ENEMIES do
  449. local prepFrame = _G["ArenaPrepFrame"..i];
  450. if (i <= numOpps) then
  451. prepFrame.specPortrait = _G["ArenaPrepFrame"..i.."SpecPortrait"];
  452. local specID, gender = GetArenaOpponentSpec(i);
  453. if (specID > 0) then
  454. local _, _, _, specIcon, _, class = GetSpecializationInfoByID(specID, gender);
  455. if( class ) then
  456. prepFrame.classPortrait:SetTexture("Interface\\TargetingFrame\\UI-Classes-Circles");
  457. prepFrame.classPortrait:SetTexCoord(unpack(CLASS_ICON_TCOORDS[strupper(class)]));
  458. end
  459. SetPortraitToTexture(prepFrame.specPortrait, specIcon);
  460. prepFrame:Show();
  461. else
  462. prepFrame:Hide();
  463. end
  464. else
  465. prepFrame:Hide();
  466. end
  467. end
  469. end
  471. function ArenaPrepFrames_UpdateBackground(force)
  472. if ( (SHOW_PARTY_BACKGROUND == "1") or force ) then
  473. ArenaPrepBackground:Show();
  474. local numOpps = GetNumArenaOpponents();
  475. if ( numOpps > 0 ) then
  476. ArenaPrepBackground:SetPoint("BOTTOMLEFT", "ArenaEnemyFrame"..numOpps.."PetFrame", "BOTTOMLEFT", -15, -10);
  477. else
  478. ArenaPrepBackground:Hide();
  479. end
  480. else
  481. ArenaPrepBackground:Hide();
  482. end
  484. end
  486. function ArenaPrepBackground_SetOpacity(opacity)
  487. local alpha;
  488. if ( not opacity ) then
  489. alpha = 1.0 - OpacityFrameSlider:GetValue();
  490. else
  491. alpha = 1.0 - opacity;
  492. end
  493. ArenaPrepBackground:SetAlpha(alpha);
  494. end