UnitFrame.lua

Compare Download

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