UnitFrame.lua

Compare Download

UnitFrame.lua (8.3.7.35662; unchanged since 8.2.5.31960)
  2. PowerBarColor = {};
  3. PowerBarColor["MANA"] = { r = 0.00, g = 0.00, b = 1.00 };
  4. PowerBarColor["RAGE"] = { r = 1.00, g = 0.00, b = 0.00, fullPowerAnim=true };
  5. PowerBarColor["FOCUS"] = { r = 1.00, g = 0.50, b = 0.25, fullPowerAnim=true };
  6. PowerBarColor["ENERGY"] = { r = 1.00, g = 1.00, b = 0.00, fullPowerAnim=true };
  7. PowerBarColor["COMBO_POINTS"] = { r = 1.00, g = 0.96, b = 0.41 };
  8. PowerBarColor["RUNES"] = { r = 0.50, g = 0.50, b = 0.50 };
  9. PowerBarColor["RUNIC_POWER"] = { r = 0.00, g = 0.82, b = 1.00 };
  10. PowerBarColor["SOUL_SHARDS"] = { r = 0.50, g = 0.32, b = 0.55 };
  11. PowerBarColor["LUNAR_POWER"] = { r = 0.30, g = 0.52, b = 0.90, atlas="_Druid-LunarBar" };
  12. PowerBarColor["HOLY_POWER"] = { r = 0.95, g = 0.90, b = 0.60 };
  13. PowerBarColor["MAELSTROM"] = { r = 0.00, g = 0.50, b = 1.00, atlas = "_Shaman-MaelstromBar", fullPowerAnim=true };
  14. PowerBarColor["INSANITY"] = { r = 0.40, g = 0, b = 0.80, atlas = "_Priest-InsanityBar"};
  15. PowerBarColor["CHI"] = { r = 0.71, g = 1.0, b = 0.92 };
  16. PowerBarColor["ARCANE_CHARGES"] = { r = 0.1, g = 0.1, b = 0.98 };
  17. PowerBarColor["FURY"] = { r = 0.788, g = 0.259, b = 0.992, atlas = "_DemonHunter-DemonicFuryBar", fullPowerAnim=true };
  18. PowerBarColor["PAIN"] = { r = 255/255, g = 156/255, b = 0, atlas = "_DemonHunter-DemonicPainBar", fullPowerAnim=true };
  19. -- vehicle colors
  20. PowerBarColor["AMMOSLOT"] = { r = 0.80, g = 0.60, b = 0.00 };
  21. PowerBarColor["FUEL"] = { r = 0.0, g = 0.55, b = 0.5 };
  22. PowerBarColor["STAGGER"] = { {r = 0.52, g = 1.0, b = 0.52}, {r = 1.0, g = 0.98, b = 0.72}, {r = 1.0, g = 0.42, b = 0.42},};
  24. -- these are mostly needed for a fallback case (in case the code tries to index a power token that is missing from the table,
  25. -- it will try to index by power type instead)
  26. PowerBarColor[0] = PowerBarColor["MANA"];
  27. PowerBarColor[1] = PowerBarColor["RAGE"];
  28. PowerBarColor[2] = PowerBarColor["FOCUS"];
  29. PowerBarColor[3] = PowerBarColor["ENERGY"];
  30. PowerBarColor[4] = PowerBarColor["CHI"];
  31. PowerBarColor[5] = PowerBarColor["RUNES"];
  32. PowerBarColor[6] = PowerBarColor["RUNIC_POWER"];
  33. PowerBarColor[7] = PowerBarColor["SOUL_SHARDS"];
  34. PowerBarColor[8] = PowerBarColor["LUNAR_POWER"];
  35. PowerBarColor[9] = PowerBarColor["HOLY_POWER"];
  36. PowerBarColor[11] = PowerBarColor["MAELSTROM"];
  37. PowerBarColor[13] = PowerBarColor["INSANITY"];
  38. PowerBarColor[17] = PowerBarColor["FURY"];
  39. PowerBarColor[18] = PowerBarColor["PAIN"];
  41. --[[
  42. This system uses "update" functions as OnUpdate, and OnEvent handlers.
  43. This "Initialize" function registers the events to handle.
  44. The "update" function is set as the OnEvent handler (although they do not parse the event),
  45. as well as run from the parent's update handler.
  47. TT: I had to make the spellbar system differ from the norm.
  48. I needed a seperate OnUpdate and OnEvent handlers. And needed to parse the event.
  49. ]]--
  51. function UnitFrame_Initialize (self, unit, name, portrait, healthbar, healthtext, manabar, manatext, threatIndicator, threatFeedbackUnit, threatNumericIndicator,
  52. myHealPredictionBar, otherHealPredictionBar, totalAbsorbBar, totalAbsorbBarOverlay, overAbsorbGlow, overHealAbsorbGlow, healAbsorbBar, healAbsorbBarLeftShadow,
  53. healAbsorbBarRightShadow, myManaCostPredictionBar)
  54. self.unit = unit;
  55. self.name = name;
  56. self.portrait = portrait;
  57. self.healthbar = healthbar;
  58. self.manabar = manabar;
  59. self.threatIndicator = threatIndicator;
  60. self.threatNumericIndicator = threatNumericIndicator;
  61. self.myHealPredictionBar = myHealPredictionBar;
  62. self.otherHealPredictionBar = otherHealPredictionBar
  63. self.totalAbsorbBar = totalAbsorbBar;
  64. self.totalAbsorbBarOverlay = totalAbsorbBarOverlay;
  65. self.overAbsorbGlow = overAbsorbGlow;
  66. self.overHealAbsorbGlow = overHealAbsorbGlow;
  67. self.healAbsorbBar = healAbsorbBar;
  68. self.healAbsorbBarLeftShadow = healAbsorbBarLeftShadow;
  69. self.healAbsorbBarRightShadow = healAbsorbBarRightShadow;
  70. self.myManaCostPredictionBar = myManaCostPredictionBar;
  71. if ( self.myHealPredictionBar ) then
  72. self.myHealPredictionBar:ClearAllPoints();
  73. end
  74. if ( self.otherHealPredictionBar ) then
  75. self.otherHealPredictionBar:ClearAllPoints();
  76. end
  77. if ( self.totalAbsorbBar ) then
  78. self.totalAbsorbBar:ClearAllPoints();
  79. end
  80. if ( self.myManaCostPredictionBar ) then
  81. self.myManaCostPredictionBar:ClearAllPoints();
  82. end
  84. if ( self.totalAbsorbBarOverlay ) then
  85. self.totalAbsorbBar.overlay = self.totalAbsorbBarOverlay;
  86. self.totalAbsorbBarOverlay:SetAllPoints(self.totalAbsorbBar);
  87. self.totalAbsorbBarOverlay.tileSize = 32;
  88. end
  89. if ( self.overAbsorbGlow ) then
  90. self.overAbsorbGlow:ClearAllPoints();
  91. self.overAbsorbGlow:SetPoint("TOPLEFT", self.healthbar, "TOPRIGHT", -7, 0);
  92. self.overAbsorbGlow:SetPoint("BOTTOMLEFT", self.healthbar, "BOTTOMRIGHT", -7, 0);
  93. end
  94. if ( self.healAbsorbBar ) then
  95. self.healAbsorbBar:ClearAllPoints();
  96. self.healAbsorbBar:SetTexture("Interface\\RaidFrame\\Absorb-Fill", true, true);
  97. end
  98. if ( self.overHealAbsorbGlow ) then
  99. self.overHealAbsorbGlow:ClearAllPoints();
  100. self.overHealAbsorbGlow:SetPoint("BOTTOMRIGHT", self.healthbar, "BOTTOMLEFT", 7, 0);
  101. self.overHealAbsorbGlow:SetPoint("TOPRIGHT", self.healthbar, "TOPLEFT", 7, 0);
  102. end
  103. if ( healAbsorbBarLeftShadow ) then
  104. self.healAbsorbBarLeftShadow:ClearAllPoints();
  105. end
  106. if ( healAbsorbBarRightShadow ) then
  107. self.healAbsorbBarRightShadow:ClearAllPoints();
  108. end
  109. if (self.healthbar) then
  110. self.healthbar.capNumericDisplay = true;
  111. end
  112. if (self.manabar) then
  113. self.manabar.capNumericDisplay = true;
  114. end
  115. UnitFrameHealthBar_Initialize(unit, healthbar, healthtext, true);
  116. UnitFrameManaBar_Initialize(unit, manabar, manatext, (unit == "player" or unit == "pet" or unit == "vehicle" or unit == "target" or unit == "focus"));
  117. UnitFrameThreatIndicator_Initialize(unit, self, threatFeedbackUnit);
  118. UnitFrame_Update(self);
  119. self:RegisterForClicks("LeftButtonUp", "RightButtonUp");
  120. self:RegisterEvent("UNIT_NAME_UPDATE");
  121. self:RegisterEvent("UNIT_DISPLAYPOWER");
  122. self:RegisterEvent("UNIT_PORTRAIT_UPDATE")
  123. self:RegisterEvent("PORTRAITS_UPDATED");
  124. if ( self.healAbsorbBar ) then
  125. self:RegisterUnitEvent("UNIT_HEAL_ABSORB_AMOUNT_CHANGED", unit);
  126. end
  127. if ( self.myHealPredictionBar ) then
  128. self:RegisterUnitEvent("UNIT_MAXHEALTH", unit);
  129. self:RegisterUnitEvent("UNIT_HEAL_PREDICTION", unit);
  130. end
  131. if ( self.totalAbsorbBar ) then
  132. self:RegisterUnitEvent("UNIT_ABSORB_AMOUNT_CHANGED", unit);
  133. end
  134. if ( self.myManaCostPredictionBar ) then
  135. self:RegisterUnitEvent("UNIT_SPELLCAST_START", unit);
  136. self:RegisterUnitEvent("UNIT_SPELLCAST_STOP", unit);
  137. self:RegisterUnitEvent("UNIT_SPELLCAST_FAILED", unit);
  138. self:RegisterUnitEvent("UNIT_SPELLCAST_SUCCEEDED", unit);
  139. end
  140. end
  142. function UnitFrame_SetUnit (self, unit, healthbar, manabar)
  143. -- update unit events if unit changes
  144. if ( self.unit ~= unit ) then
  145. if ( self.myHealPredictionBar ) then
  146. self:RegisterUnitEvent("UNIT_MAXHEALTH", unit);
  147. self:RegisterUnitEvent("UNIT_HEAL_PREDICTION", unit);
  148. end
  149. if ( self.totalAbsorbBar ) then
  150. self:RegisterUnitEvent("UNIT_ABSORB_AMOUNT_CHANGED", unit);
  151. end
  152. if ( not healthbar.frequentUpdates ) then
  153. healthbar:RegisterUnitEvent("UNIT_HEALTH", unit);
  154. end
  155. if ( manabar and not manabar.frequentUpdates ) then
  156. UnitFrameManaBar_RegisterDefaultEvents(manabar);
  157. end
  158. healthbar:RegisterUnitEvent("UNIT_MAXHEALTH", unit);
  160. if ( self.PlayerFrameHealthBarAnimatedLoss ) then
  161. self.PlayerFrameHealthBarAnimatedLoss:SetUnitHealthBar(unit, healthbar);
  162. end
  163. end
  165. self.unit = unit;
  166. healthbar.unit = unit;
  167. if ( manabar ) then --Party Pet frames don't have a mana bar.
  168. manabar.unit = unit;
  169. end
  170. self:SetAttribute("unit", unit);
  171. securecall("UnitFrame_Update", self);
  172. end
  174. function UnitFrame_Update (self, isParty)
  175. if (self.name) then
  176. local name;
  177. if ( self.overrideName ) then
  178. name = self.overrideName;
  179. else
  180. name = self.unit;
  181. end
  182. if (isParty) then
  183. self.name:SetText(GetUnitName(name, true));
  184. else
  185. self.name:SetText(GetUnitName(name));
  186. end
  187. end
  189. UnitFramePortrait_Update(self);
  190. UnitFrameHealthBar_Update(self.healthbar, self.unit);
  191. UnitFrameManaBar_Update(self.manabar, self.unit);
  192. UnitFrame_UpdateThreatIndicator(self.threatIndicator, self.threatNumericIndicator);
  193. UnitFrameHealPredictionBars_UpdateMax(self);
  194. UnitFrameHealPredictionBars_Update(self);
  195. UnitFrameManaCostPredictionBars_Update(self);
  196. end
  198. function UnitFramePortrait_Update (self)
  199. if ( self.portrait ) then
  200. SetPortraitTexture(self.portrait, self.unit);
  201. end
  202. end
  204. function UnitFrame_OnEvent(self, event, ...)
  205. local eventUnit = ...
  207. local unit = self.unit;
  208. if ( eventUnit == unit ) then
  209. if ( event == "UNIT_NAME_UPDATE" ) then
  210. self.name:SetText(GetUnitName(unit));
  211. elseif ( event == "UNIT_PORTRAIT_UPDATE" ) then
  212. UnitFramePortrait_Update(self);
  213. elseif ( event == "UNIT_DISPLAYPOWER" ) then
  214. if ( self.manabar ) then
  215. UnitFrameManaBar_UpdateType(self.manabar);
  216. end
  217. elseif ( event == "UNIT_MAXHEALTH" ) then
  218. UnitFrameHealPredictionBars_UpdateMax(self);
  219. UnitFrameHealPredictionBars_Update(self);
  220. elseif ( event == "UNIT_HEAL_PREDICTION" ) then
  221. UnitFrameHealPredictionBars_Update(self);
  222. elseif ( event == "UNIT_ABSORB_AMOUNT_CHANGED" ) then
  223. UnitFrameHealPredictionBars_Update(self);
  224. elseif ( event == "UNIT_HEAL_ABSORB_AMOUNT_CHANGED" ) then
  225. UnitFrameHealPredictionBars_Update(self);
  226. elseif ( event == "UNIT_SPELLCAST_START" or event == "UNIT_SPELLCAST_STOP" or event == "UNIT_SPELLCAST_FAILED" or event == "UNIT_SPELLCAST_SUCCEEDED" ) then
  227. local name, text, texture, startTime, endTime, isTradeSkill, castID, notInterruptible, spellID = UnitCastingInfo(unit);
  228. UnitFrameManaCostPredictionBars_Update(self, event == "UNIT_SPELLCAST_START", startTime, endTime, spellID);
  229. end
  230. elseif ( event == "PORTRAITS_UPDATED" ) then
  231. UnitFramePortrait_Update(self);
  232. end
  233. end
  235. function UnitFrameHealPredictionBars_UpdateMax(self)
  236. if ( not self.myHealPredictionBar ) then
  237. return;
  238. end
  240. UnitFrameHealPredictionBars_Update(self);
  241. end
  243. function UnitFrameHealPredictionBars_UpdateSize(self)
  244. if ( not self.myHealPredictionBar or not self.otherHealPredictionBar ) then
  245. return;
  246. end
  248. UnitFrameHealPredictionBars_Update(self);
  249. end
  251. --WARNING: This function is very similar to the function CompactUnitFrame_UpdateHealPrediction in CompactUnitFrame.lua.
  252. --If you are making changes here, it is possible you may want to make changes there as well.
  253. local MAX_INCOMING_HEAL_OVERFLOW = 1.0;
  254. function UnitFrameHealPredictionBars_Update(frame)
  255. if ( not frame.myHealPredictionBar ) then
  256. return;
  257. end
  259. local _, maxHealth = frame.healthbar:GetMinMaxValues();
  260. local health = frame.healthbar:GetValue();
  261. if ( maxHealth <= 0 ) then
  262. return;
  263. end
  265. local myIncomingHeal = UnitGetIncomingHeals(frame.unit, "player") or 0;
  266. local allIncomingHeal = UnitGetIncomingHeals(frame.unit) or 0;
  267. local totalAbsorb = UnitGetTotalAbsorbs(frame.unit) or 0;
  269. local myCurrentHealAbsorb = 0;
  270. if ( frame.healAbsorbBar ) then
  271. myCurrentHealAbsorb = UnitGetTotalHealAbsorbs(frame.unit) or 0;
  273. --We don't fill outside the health bar with healAbsorbs. Instead, an overHealAbsorbGlow is shown.
  274. if ( health < myCurrentHealAbsorb ) then
  275. frame.overHealAbsorbGlow:Show();
  276. myCurrentHealAbsorb = health;
  277. else
  278. frame.overHealAbsorbGlow:Hide();
  279. end
  280. end
  282. --See how far we're going over the health bar and make sure we don't go too far out of the frame.
  283. if ( health - myCurrentHealAbsorb + allIncomingHeal > maxHealth * MAX_INCOMING_HEAL_OVERFLOW ) then
  284. allIncomingHeal = maxHealth * MAX_INCOMING_HEAL_OVERFLOW - health + myCurrentHealAbsorb;
  285. end
  287. local otherIncomingHeal = 0;
  289. --Split up incoming heals.
  290. if ( allIncomingHeal >= myIncomingHeal ) then
  291. otherIncomingHeal = allIncomingHeal - myIncomingHeal;
  292. else
  293. myIncomingHeal = allIncomingHeal;
  294. end
  296. --We don't fill outside the the health bar with absorbs. Instead, an overAbsorbGlow is shown.
  297. local overAbsorb = false;
  298. if ( health - myCurrentHealAbsorb + allIncomingHeal + totalAbsorb >= maxHealth or health + totalAbsorb >= maxHealth ) then
  299. if ( totalAbsorb > 0 ) then
  300. overAbsorb = true;
  301. end
  303. if ( allIncomingHeal > myCurrentHealAbsorb ) then
  304. totalAbsorb = max(0,maxHealth - (health - myCurrentHealAbsorb + allIncomingHeal));
  305. else
  306. totalAbsorb = max(0,maxHealth - health);
  307. end
  308. end
  310. if ( overAbsorb ) then
  311. frame.overAbsorbGlow:Show();
  312. else
  313. frame.overAbsorbGlow:Hide();
  314. end
  316. local healthTexture = frame.healthbar:GetStatusBarTexture();
  317. local myCurrentHealAbsorbPercent = 0;
  318. local healAbsorbTexture = nil;
  320. if ( frame.healAbsorbBar ) then
  321. myCurrentHealAbsorbPercent = myCurrentHealAbsorb / maxHealth;
  323. --If allIncomingHeal is greater than myCurrentHealAbsorb, then the current
  324. --heal absorb will be completely overlayed by the incoming heals so we don't show it.
  325. if ( myCurrentHealAbsorb > allIncomingHeal ) then
  326. local shownHealAbsorb = myCurrentHealAbsorb - allIncomingHeal;
  327. local shownHealAbsorbPercent = shownHealAbsorb / maxHealth;
  329. healAbsorbTexture = UnitFrameUtil_UpdateFillBar(frame, healthTexture, frame.healAbsorbBar, shownHealAbsorb, -shownHealAbsorbPercent);
  331. --If there are incoming heals the left shadow would be overlayed by the incoming heals
  332. --so it isn't shown.
  333. if ( allIncomingHeal > 0 ) then
  334. frame.healAbsorbBarLeftShadow:Hide();
  335. else
  336. frame.healAbsorbBarLeftShadow:SetPoint("TOPLEFT", healAbsorbTexture, "TOPLEFT", 0, 0);
  337. frame.healAbsorbBarLeftShadow:SetPoint("BOTTOMLEFT", healAbsorbTexture, "BOTTOMLEFT", 0, 0);
  338. frame.healAbsorbBarLeftShadow:Show();
  339. end
  341. -- The right shadow is only shown if there are absorbs on the health bar.
  342. if ( totalAbsorb > 0 ) then
  343. frame.healAbsorbBarRightShadow:SetPoint("TOPLEFT", healAbsorbTexture, "TOPRIGHT", -8, 0);
  344. frame.healAbsorbBarRightShadow:SetPoint("BOTTOMLEFT", healAbsorbTexture, "BOTTOMRIGHT", -8, 0);
  345. frame.healAbsorbBarRightShadow:Show();
  346. else
  347. frame.healAbsorbBarRightShadow:Hide();
  348. end
  349. else
  350. frame.healAbsorbBar:Hide();
  351. frame.healAbsorbBarLeftShadow:Hide();
  352. frame.healAbsorbBarRightShadow:Hide();
  353. end
  354. end
  356. --Show myIncomingHeal on the health bar.
  357. local incomingHealTexture = UnitFrameUtil_UpdateFillBar(frame, healthTexture, frame.myHealPredictionBar, myIncomingHeal, -myCurrentHealAbsorbPercent);
  359. --Append otherIncomingHeal on the health bar
  360. if (myIncomingHeal > 0) then
  361. incomingHealTexture = UnitFrameUtil_UpdateFillBar(frame, incomingHealTexture, frame.otherHealPredictionBar, otherIncomingHeal);
  362. else
  363. incomingHealTexture = UnitFrameUtil_UpdateFillBar(frame, healthTexture, frame.otherHealPredictionBar, otherIncomingHeal, -myCurrentHealAbsorbPercent);
  364. end
  366. --Append absorbs to the correct section of the health bar.
  367. local appendTexture = nil;
  368. if ( healAbsorbTexture ) then
  369. --If there is a healAbsorb part shown, append the absorb to the end of that.
  370. appendTexture = healAbsorbTexture;
  371. else
  372. --Otherwise, append the absorb to the end of the the incomingHeals part;
  373. appendTexture = incomingHealTexture;
  374. end
  375. UnitFrameUtil_UpdateFillBar(frame, appendTexture, frame.totalAbsorbBar, totalAbsorb)
  376. end
  378. function UnitFrameManaCostPredictionBars_Update(frame, isStarting, startTime, endTime, spellID)
  379. if (not frame.manabar or not frame.myManaCostPredictionBar) then
  380. return;
  381. end
  383. local cost = 0;
  384. if (not isStarting or startTime == endTime) then
  385. local currentSpellID = select(9, UnitCastingInfo(frame.unit));
  386. if(currentSpellID and frame.predictedPowerCost) then --if we're currently casting something with a power cost, then whatever cast
  387. cost = frame.predictedPowerCost; --just finished was allowed while casting, don't reset the original cast
  388. else
  389. frame.predictedPowerCost = nil;
  390. end
  391. else
  392. local costTable = GetSpellPowerCost(spellID);
  393. for _, costInfo in pairs(costTable) do
  394. if (costInfo.type == frame.manabar.powerType) then
  395. cost = costInfo.cost;
  396. break;
  397. end
  398. end
  399. frame.predictedPowerCost = cost;
  400. end
  401. local manaBarTexture = frame.manabar:GetStatusBarTexture();
  402. UnitFrameManaBar_Update(frame.manabar, frame.unit);
  403. UnitFrameUtil_UpdateManaFillBar(frame, manaBarTexture, frame.myManaCostPredictionBar, cost);
  404. end
  406. --WARNING: This function is very similar to the function CompactUnitFrameUtil_UpdateFillBar in CompactUnitFrame.lua.
  407. --If you are making changes here, it is possible you may want to make changes there as well.
  408. function UnitFrameUtil_UpdateFillBarBase(frame, realbar, previousTexture, bar, amount, barOffsetXPercent)
  409. if ( amount == 0 ) then
  410. bar:Hide();
  411. if ( bar.overlay ) then
  412. bar.overlay:Hide();
  413. end
  414. return previousTexture;
  415. end
  417. local barOffsetX = 0;
  418. if ( barOffsetXPercent ) then
  419. local realbarSizeX = realbar:GetWidth();
  420. barOffsetX = realbarSizeX * barOffsetXPercent;
  421. end
  423. bar:SetPoint("TOPLEFT", previousTexture, "TOPRIGHT", barOffsetX, 0);
  424. bar:SetPoint("BOTTOMLEFT", previousTexture, "BOTTOMRIGHT", barOffsetX, 0);
  426. local totalWidth, totalHeight = realbar:GetSize();
  427. local _, totalMax = realbar:GetMinMaxValues();
  429. local barSize = (amount / totalMax) * totalWidth;
  430. bar:SetWidth(barSize);
  431. bar:Show();
  432. if ( bar.overlay ) then
  433. bar.overlay:SetTexCoord(0, barSize / bar.overlay.tileSize, 0, totalHeight / bar.overlay.tileSize);
  434. bar.overlay:Show();
  435. end
  436. return bar;
  437. end
  439. function UnitFrameUtil_UpdateFillBar(frame, previousTexture, bar, amount, barOffsetXPercent)
  440. return UnitFrameUtil_UpdateFillBarBase(frame, frame.healthbar, previousTexture, bar, amount, barOffsetXPercent);
  441. end
  443. function UnitFrameUtil_UpdateManaFillBar(frame, previousTexture, bar, amount, barOffsetXPercent)
  444. return UnitFrameUtil_UpdateFillBarBase(frame, frame.manabar, previousTexture, bar, amount, barOffsetXPercent);
  445. end
  447. function UnitFrame_OnEnter (self)
  448. UnitFrame_UpdateTooltip(self);
  449. end
  451. function UnitFrame_OnLeave (self)
  452. self.UpdateTooltip = nil;
  453. GameTooltip:FadeOut();
  454. end
  456. function UnitFrame_UpdateTooltip (self)
  457. GameTooltip_SetDefaultAnchor(GameTooltip, self);
  458. if ( GameTooltip:SetUnit(self.unit, self.hideStatusOnTooltip) ) then
  459. self.UpdateTooltip = UnitFrame_UpdateTooltip;
  460. else
  461. self.UpdateTooltip = nil;
  462. end
  464. local r, g, b = GameTooltip_UnitColor(self.unit);
  465. GameTooltipTextLeft1:SetTextColor(r, g, b);
  466. end
  468. function UnitFrameManaBar_UpdateType (manaBar)
  469. if ( not manaBar ) then
  470. return;
  471. end
  472. local unitFrame = manaBar:GetParent();
  473. local powerType, powerToken, altR, altG, altB = UnitPowerType(manaBar.unit);
  474. local prefix = _G[powerToken];
  475. local info = PowerBarColor[powerToken];
  476. if ( info ) then
  477. if ( not manaBar.lockColor ) then
  478. local playerDeadOrGhost = (manaBar.unit == "player" and (UnitIsDead("player") or UnitIsGhost("player")));
  479. if ( info.atlas ) then
  480. manaBar:SetStatusBarAtlas(info.atlas);
  481. manaBar:SetStatusBarColor(1, 1, 1);
  482. manaBar:GetStatusBarTexture():SetDesaturated(playerDeadOrGhost);
  483. manaBar:GetStatusBarTexture():SetAlpha(playerDeadOrGhost and 0.5 or 1);
  484. else
  485. manaBar:SetStatusBarTexture("Interface\\TargetingFrame\\UI-StatusBar");
  486. if ( playerDeadOrGhost ) then
  487. manaBar:SetStatusBarColor(0.6, 0.6, 0.6, 0.5);
  488. else
  489. manaBar:SetStatusBarColor(info.r, info.g, info.b);
  490. end
  491. end
  493. if ( manaBar.FeedbackFrame ) then
  494. manaBar.FeedbackFrame:Initialize(info, manaBar.unit, powerType);
  495. end
  497. if ( manaBar.FullPowerFrame ) then
  498. manaBar.FullPowerFrame:Initialize(info.fullPowerAnim);
  499. end
  500. end
  501. else
  502. if ( not altR) then
  503. -- couldn't find a power token entry...default to indexing by power type or just mana if we don't have that either
  504. info = PowerBarColor[powerType] or PowerBarColor["MANA"];
  505. else
  506. if ( not manaBar.lockColor ) then
  507. manaBar:SetStatusBarColor(altR, altG, altB);
  508. end
  509. end
  510. end
  511. if ( manaBar.powerType ~= powerType or manaBar.powerType ~= powerType ) then
  512. manaBar.powerType = powerType;
  513. manaBar.powerToken = powerToken;
  514. if ( manaBar.FullPowerFrame ) then
  515. manaBar.FullPowerFrame:RemoveAnims();
  516. end
  517. if manaBar.FeedbackFrame then
  518. manaBar.FeedbackFrame:StopFeedbackAnim();
  519. end
  520. manaBar.currValue = UnitPower("player", powerType);
  521. if unitFrame.myManaCostPredictionBar then
  522. unitFrame.myManaCostPredictionBar:Hide();
  523. end
  524. unitFrame.predictedPowerCost = 0;
  525. end
  527. -- Update the manabar text
  528. if ( not unitFrame.noTextPrefix ) then
  529. SetTextStatusBarTextPrefix(manaBar, prefix);
  530. end
  531. TextStatusBar_UpdateTextString(manaBar);
  533. -- Setup newbie tooltip
  534. if ( manaBar.unit ~= "pet") then
  535. if ( unitFrame:GetName() == "PlayerFrame" ) then
  536. manaBar.tooltipTitle = prefix;
  537. manaBar.tooltipText = _G["NEWBIE_TOOLTIP_MANABAR_"..powerType];
  538. else
  539. manaBar.tooltipTitle = nil;
  540. manaBar.tooltipText = nil;
  541. end
  542. end
  543. end
  545. function UnitFrameHealthBar_Initialize (unit, statusbar, statustext, frequentUpdates)
  546. if ( not statusbar ) then
  547. return;
  548. end
  550. statusbar.unit = unit;
  551. SetTextStatusBarText(statusbar, statustext);
  553. statusbar.frequentUpdates = frequentUpdates;
  554. if ( frequentUpdates ) then
  555. statusbar:RegisterEvent("VARIABLES_LOADED");
  556. end
  557. if ( GetCVarBool("predictedHealth") and frequentUpdates ) then
  558. statusbar:SetScript("OnUpdate", UnitFrameHealthBar_OnUpdate);
  559. else
  560. statusbar:RegisterUnitEvent("UNIT_HEALTH", unit);
  561. end
  562. statusbar:RegisterUnitEvent("UNIT_MAXHEALTH", unit);
  563. statusbar:SetScript("OnEvent", UnitFrameHealthBar_OnEvent);
  565. -- Setup newbie tooltip
  566. if ( statusbar and (statusbar:GetParent() == PlayerFrame) ) then
  567. statusbar.tooltipTitle = HEALTH;
  568. statusbar.tooltipText = NEWBIE_TOOLTIP_HEALTHBAR;
  569. else
  570. statusbar.tooltipTitle = nil;
  571. statusbar.tooltipText = nil;
  572. end
  573. end
  575. function UnitFrameHealthBar_OnEvent(self, event, ...)
  576. if ( event == "CVAR_UPDATE" ) then
  577. TextStatusBar_OnEvent(self, event, ...);
  578. elseif ( event == "VARIABLES_LOADED" ) then
  579. self:UnregisterEvent("VARIABLES_LOADED");
  580. if ( GetCVarBool("predictedHealth") and self.frequentUpdates ) then
  581. self:SetScript("OnUpdate", UnitFrameHealthBar_OnUpdate);
  582. self:UnregisterEvent("UNIT_HEALTH");
  583. else
  584. self:RegisterUnitEvent("UNIT_HEALTH", self.unit);
  585. self:SetScript("OnUpdate", nil);
  586. end
  587. else
  588. if ( not self.ignoreNoUnit or UnitGUID(self.unit) ) then
  589. UnitFrameHealthBar_Update(self, ...);
  590. end
  591. end
  592. end
  594. AnimatedHealthLossMixin = {};
  596. function AnimatedHealthLossMixin:OnLoad()
  597. self:SetStatusBarColor(1, 0, 0, 1);
  598. self:SetDuration(.25);
  599. self:SetStartDelay(.1);
  600. self:SetPauseDelay(.05);
  601. self:SetPostponeDelay(.05);
  602. end
  604. function AnimatedHealthLossMixin:SetDuration(duration)
  605. self.animationDuration = duration or 0;
  606. end
  608. function AnimatedHealthLossMixin:SetStartDelay(delay)
  609. self.animationStartDelay = delay or 0;
  610. end
  612. function AnimatedHealthLossMixin:SetPauseDelay(delay)
  613. self.animationPauseDelay = delay or 0;
  614. end
  616. function AnimatedHealthLossMixin:SetPostponeDelay(delay)
  617. self.animationPostponeDelay = delay or 0;
  618. end
  620. function AnimatedHealthLossMixin:SetUnitHealthBar(unit, healthBar)
  621. if self.unit ~= unit then
  622. healthBar.AnimatedLossBar = self;
  624. self.unit = unit;
  625. self:SetAllPoints(healthBar);
  626. self:UpdateHealthMinMax();
  627. end
  628. end
  630. function AnimatedHealthLossMixin:UpdateHealthMinMax()
  631. local maxValue = UnitHealthMax(self.unit);
  632. self:SetMinMaxValues(0, maxValue);
  633. end
  635. function AnimatedHealthLossMixin:GetHealthLossAnimationData(currentHealth, previousHealth)
  636. if self.animationStartTime then
  637. local totalElapsedTime = GetTime() - self.animationStartTime;
  638. if totalElapsedTime > 0 then
  639. local animCompletePercent = totalElapsedTime / self.animationDuration;
  640. if animCompletePercent < 1 and previousHealth > currentHealth then
  641. local healthDelta = previousHealth - currentHealth;
  642. local animatedLossAmount = previousHealth - (animCompletePercent * healthDelta);
  643. return animatedLossAmount, animCompletePercent;
  644. end
  645. else
  646. return previousHealth, 0;
  647. end
  648. end
  649. return 0, 1; -- Animated loss amount is 0, and the animation is fully complete.
  650. end
  652. function AnimatedHealthLossMixin:CancelAnimation()
  653. self:Hide();
  654. self.animationStartTime = nil;
  655. self.animationCompletePercent = nil;
  656. end
  658. function AnimatedHealthLossMixin:BeginAnimation(value)
  659. self.animationStartValue = value;
  660. self.animationStartTime = GetTime() + self.animationStartDelay;
  661. self.animationCompletePercent = 0;
  662. self:Show();
  663. self:SetValue(self.animationStartValue);
  664. end
  666. function AnimatedHealthLossMixin:PostponeStartTime()
  667. self.animationStartTime = self.animationStartTime + self.animationPostponeDelay;
  668. end
  670. function AnimatedHealthLossMixin:UpdateHealth(currentHealth, previousHealth)
  671. local delta = currentHealth - previousHealth;
  672. local hasLoss = delta < 0;
  673. local hasBegun = self.animationStartTime ~= nil;
  674. local isAnimating = hasBegun and self.animationCompletePercent > 0;
  676. if hasLoss and not hasBegun then
  677. self:BeginAnimation(previousHealth);
  678. elseif hasLoss and hasBegun and not isAnimating then
  679. self:PostponeStartTime();
  680. elseif hasLoss and isAnimating then
  681. -- Reset the starting value of the health to what the animated loss bar was when the new incoming damage happened
  682. -- and pause briefly when new damage occurs.
  683. self.animationStartValue = self:GetHealthLossAnimationData(previousHealth, self.animationStartValue);
  684. self.animationStartTime = GetTime() + self.animationPauseDelay;
  685. elseif not hasLoss and hasBegun and currentHealth >= self.animationStartValue then
  686. self:CancelAnimation();
  687. end
  688. end
  690. function AnimatedHealthLossMixin:UpdateLossAnimation(currentHealth)
  691. local totalAbsorb = UnitGetTotalAbsorbs(self.unit) or 0;
  692. if totalAbsorb > 0 then
  693. self:CancelAnimation();
  694. end
  696. if self.animationStartTime then
  697. local animationValue, animationCompletePercent = self:GetHealthLossAnimationData(currentHealth, self.animationStartValue);
  698. self.animationCompletePercent = animationCompletePercent;
  699. if animationCompletePercent >= 1 then
  700. self:CancelAnimation();
  701. else
  702. self:SetValue(animationValue);
  703. end
  704. end
  705. end
  707. function UnitFrameHealthBar_OnUpdate(self)
  708. if ( not self.disconnected and not self.lockValues) then
  709. local currValue = UnitHealth(self.unit);
  710. local animatedLossBar = self.AnimatedLossBar;
  712. if ( currValue ~= self.currValue ) then
  713. if ( not self.ignoreNoUnit or UnitGUID(self.unit) ) then
  715. if animatedLossBar then
  716. animatedLossBar:UpdateHealth(currValue, self.currValue);
  717. end
  719. self:SetValue(currValue);
  720. self.currValue = currValue;
  721. TextStatusBar_UpdateTextString(self);
  722. UnitFrameHealPredictionBars_Update(self:GetParent());
  723. end
  724. end
  726. if animatedLossBar then
  727. animatedLossBar:UpdateLossAnimation(currValue);
  728. end
  729. end
  730. end
  732. function UnitFrameHealthBar_Update(statusbar, unit)
  733. if ( not statusbar or statusbar.lockValues ) then
  734. return;
  735. end
  737. if ( unit == statusbar.unit ) then
  738. local maxValue = UnitHealthMax(unit);
  740. -- Safety check to make sure we never get an empty bar.
  741. statusbar.forceHideText = false;
  742. if ( maxValue == 0 ) then
  743. maxValue = 1;
  744. statusbar.forceHideText = true;
  745. end
  747. statusbar:SetMinMaxValues(0, maxValue);
  749. if statusbar.AnimatedLossBar then
  750. statusbar.AnimatedLossBar:UpdateHealthMinMax();
  751. end
  753. statusbar.disconnected = not UnitIsConnected(unit);
  754. if ( statusbar.disconnected ) then
  755. if ( not statusbar.lockColor ) then
  756. statusbar:SetStatusBarColor(0.5, 0.5, 0.5);
  757. end
  758. statusbar:SetValue(maxValue);
  759. statusbar.currValue = maxValue;
  760. else
  761. local currValue = UnitHealth(unit);
  762. if ( not statusbar.lockColor ) then
  763. statusbar:SetStatusBarColor(0.0, 1.0, 0.0);
  764. end
  765. statusbar:SetValue(currValue);
  766. statusbar.currValue = currValue;
  767. end
  768. end
  769. TextStatusBar_UpdateTextString(statusbar);
  770. UnitFrameHealPredictionBars_Update(statusbar:GetParent());
  771. end
  773. function UnitFrameHealthBar_OnValueChanged(self, value)
  774. TextStatusBar_OnValueChanged(self, value);
  775. HealthBar_OnValueChanged(self, value);
  776. end
  778. function UnitFrameManaBar_UnregisterDefaultEvents(self)
  779. self:UnregisterEvent("UNIT_POWER_UPDATE");
  780. end
  782. function UnitFrameManaBar_RegisterDefaultEvents(self)
  783. self:RegisterUnitEvent("UNIT_POWER_UPDATE", self.unit);
  784. end
  786. function UnitFrameManaBar_Initialize (unit, statusbar, statustext, frequentUpdates)
  787. if ( not statusbar ) then
  788. return;
  789. end
  790. statusbar.unit = unit;
  791. statusbar.texture = statusbar:GetStatusBarTexture();
  792. SetTextStatusBarText(statusbar, statustext);
  794. statusbar.frequentUpdates = frequentUpdates;
  795. if ( frequentUpdates ) then
  796. statusbar:RegisterEvent("VARIABLES_LOADED");
  797. end
  798. if ( frequentUpdates ) then
  799. statusbar:SetScript("OnUpdate", UnitFrameManaBar_OnUpdate);
  800. else
  801. UnitFrameManaBar_RegisterDefaultEvents(statusbar);
  802. end
  803. statusbar:RegisterEvent("UNIT_DISPLAYPOWER");
  804. statusbar:RegisterUnitEvent("UNIT_MAXPOWER", unit);
  805. if ( statusbar.unit == "player" ) then
  806. statusbar:RegisterEvent("PLAYER_DEAD");
  807. statusbar:RegisterEvent("PLAYER_ALIVE");
  808. statusbar:RegisterEvent("PLAYER_UNGHOST");
  809. end
  810. statusbar:SetScript("OnEvent", UnitFrameManaBar_OnEvent);
  811. end
  813. function UnitFrameManaBar_OnEvent(self, event, ...)
  814. if ( event == "CVAR_UPDATE" ) then
  815. TextStatusBar_OnEvent(self, event, ...);
  816. elseif ( event == "VARIABLES_LOADED" ) then
  817. self:UnregisterEvent("VARIABLES_LOADED");
  818. if ( self.frequentUpdates ) then
  819. self:SetScript("OnUpdate", UnitFrameManaBar_OnUpdate);
  820. UnitFrameManaBar_UnregisterDefaultEvents(self);
  821. else
  822. UnitFrameManaBar_RegisterDefaultEvents(self);
  823. self:SetScript("OnUpdate", nil);
  824. end
  825. elseif ( event == "PLAYER_ALIVE" or event == "PLAYER_DEAD" or event == "PLAYER_UNGHOST" ) then
  826. UnitFrameManaBar_UpdateType(self);
  827. else
  828. if ( not self.ignoreNoUnit or UnitGUID(self.unit) ) then
  829. UnitFrameManaBar_Update(self, ...);
  830. end
  831. end
  832. end
  834. function UnitFrameManaBar_OnUpdate(self)
  835. if ( not self.disconnected and not self.lockValues ) then
  836. local predictedCost = self:GetParent().predictedPowerCost;
  837. local currValue = UnitPower(self.unit, self.powerType);
  838. if (predictedCost) then
  839. currValue = currValue - predictedCost;
  840. end
  841. if ( currValue ~= self.currValue or self.forceUpdate ) then
  842. self.forceUpdate = nil;
  843. if ( not self.ignoreNoUnit or UnitGUID(self.unit) ) then
  844. if ( self.FeedbackFrame ) then
  845. -- Only show anim if change is more than 10%
  846. local oldValue = self.currValue or 0;
  847. if ( self.FeedbackFrame.maxValue ~= 0 and math.abs(currValue - oldValue) / self.FeedbackFrame.maxValue > 0.1 ) then
  848. self.FeedbackFrame:StartFeedbackAnim(oldValue, currValue);
  849. end
  850. end
  851. if ( self.FullPowerFrame and self.FullPowerFrame.active ) then
  852. self.FullPowerFrame:StartAnimIfFull(self.currValue or 0, currValue);
  853. end
  854. self:SetValue(currValue);
  855. self.currValue = currValue;
  856. TextStatusBar_UpdateTextString(self);
  857. end
  858. end
  859. end
  860. end
  862. function UnitFrameManaBar_Update(statusbar, unit)
  863. if ( not statusbar or statusbar.lockValues ) then
  864. return;
  865. end
  867. if ( unit == statusbar.unit ) then
  868. -- be sure to update the power type before grabbing the max power!
  869. UnitFrameManaBar_UpdateType(statusbar);
  871. local maxValue = UnitPowerMax(unit, statusbar.powerType);
  873. statusbar:SetMinMaxValues(0, maxValue);
  875. statusbar.disconnected = not UnitIsConnected(unit);
  876. if ( statusbar.disconnected ) then
  877. statusbar:SetValue(maxValue);
  878. statusbar.currValue = maxValue;
  879. if ( not statusbar.lockColor ) then
  880. statusbar:SetStatusBarColor(0.5, 0.5, 0.5);
  881. end
  882. else
  883. local predictedCost = statusbar:GetParent().predictedPowerCost;
  884. local currValue = UnitPower(unit, statusbar.powerType);
  885. if (predictedCost) then
  886. currValue = currValue - predictedCost;
  887. end
  888. if ( statusbar.FullPowerFrame ) then
  889. statusbar.FullPowerFrame:SetMaxValue(maxValue);
  890. end
  892. statusbar:SetValue(currValue);
  893. statusbar.forceUpdate = true;
  894. end
  895. end
  896. TextStatusBar_UpdateTextString(statusbar);
  897. end
  899. function UnitFrameThreatIndicator_Initialize(unit, unitFrame, feedbackUnit)
  900. local indicator = unitFrame.threatIndicator;
  901. if ( not indicator ) then
  902. return;
  903. end
  905. indicator.unit = unit;
  906. indicator.feedbackUnit = feedbackUnit or unit;
  908. unitFrame:RegisterEvent("UNIT_THREAT_SITUATION_UPDATE");
  909. if ( unitFrame.OnEvent == nil ) then
  910. unitFrame.OnEvent = unitFrame:GetScript("OnEvent") or false;
  911. end
  912. unitFrame:SetScript("OnEvent", UnitFrameThreatIndicator_OnEvent);
  913. end
  915. function UnitFrameThreatIndicator_OnEvent(self, event, ...)
  916. if ( self.OnEvent ) then
  917. self.OnEvent(self, event, ...);
  918. end
  919. if ( event == "UNIT_THREAT_SITUATION_UPDATE" ) then
  920. UnitFrame_UpdateThreatIndicator(self.threatIndicator, self.threatNumericIndicator,...);
  921. end
  922. end
  924. function UnitFrame_UpdateThreatIndicator(indicator, numericIndicator, unit)
  925. if ( not indicator ) then
  926. return;
  927. end
  929. if ( not unit or unit == indicator.feedbackUnit ) then
  930. local status;
  931. if ( indicator.feedbackUnit ~= indicator.unit ) then
  932. status = UnitThreatSituation(indicator.feedbackUnit, indicator.unit);
  933. else
  934. status = UnitThreatSituation(indicator.feedbackUnit);
  935. end
  937. if ( IsThreatWarningEnabled() ) then
  938. if (status and status > 0) then
  939. indicator:SetVertexColor(GetThreatStatusColor(status));
  940. indicator:Show();
  941. else
  942. indicator:Hide();
  943. end
  945. if ( numericIndicator ) then
  946. if ( ShowNumericThreat() and not (UnitClassification(indicator.unit) == "minus") ) then
  947. local isTanking, status, percentage, rawPercentage = UnitDetailedThreatSituation(indicator.feedbackUnit, indicator.unit);
  948. local display = rawPercentage;
  949. if ( isTanking ) then
  950. display = UnitThreatPercentageOfLead(indicator.feedbackUnit, indicator.unit);
  951. end
  952. if ( display and display ~= 0 ) then
  953. numericIndicator.text:SetText(format("%1.0f", display).."%");
  954. numericIndicator.bg:SetVertexColor(GetThreatStatusColor(status));
  955. numericIndicator:Show();
  956. else
  957. numericIndicator:Hide();
  958. end
  959. else
  960. numericIndicator:Hide();
  961. end
  962. end
  963. else
  964. indicator:Hide();
  965. if ( numericIndicator ) then
  966. numericIndicator:Hide();
  967. end
  968. end
  969. end
  970. end
  972. function GetUnitName(unit, showServerName)
  973. local name, server = UnitName(unit);
  974. local relationship = UnitRealmRelationship(unit);
  975. if ( server and server ~= "" ) then
  976. if ( showServerName ) then
  977. return name.."-"..server;
  978. else
  979. if (relationship == LE_REALM_RELATION_VIRTUAL) then
  980. return name;
  981. else
  982. return name..FOREIGN_SERVER_LABEL;
  983. end
  984. end
  985. else
  986. return name;
  987. end
  988. end
  990. function ShowNumericThreat()
  991. if ( GetCVar("threatShowNumeric") == "1" ) then
  992. return true;
  993. else
  994. return false;
  995. end
  996. end