TotemFrame.lua

Compare Download

TotemFrame.lua (9.0.2.36294; unchanged since 7.1.0.22908)
  2. function TotemFrame_OnLoad(self)
  3. self:RegisterEvent("PLAYER_TOTEM_UPDATE");
  4. self:RegisterEvent("PLAYER_ENTERING_WORLD");
  5. self:RegisterEvent("UPDATE_SHAPESHIFT_FORM");
  6. self:RegisterEvent("PLAYER_TALENT_UPDATE");
  8. local _, class = UnitClass("player");
  9. if ( class == "DEATHKNIGHT" ) then
  10. self:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 65, -55);
  11. elseif ( class == "WARLOCK" ) then
  12. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 28, -85);
  13. end
  14. end
  16. function TotemFrame_Update()
  17. local _, class = UnitClass("player");
  18. local priorities = STANDARD_TOTEM_PRIORITIES;
  19. if (class == "SHAMAN") then
  20. priorities = SHAMAN_TOTEM_PRIORITIES;
  21. end
  23. local hasPet = PetFrame and PetFrame:IsShown();
  24. if ( class == "PALADIN" or class == "DEATHKNIGHT" ) then
  25. if ( hasPet ) then
  26. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 28, -75);
  27. else
  28. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 67, -63);
  29. end
  30. elseif ( class == "DRUID" ) then
  31. local form = GetShapeshiftFormID();
  32. if ( form == MOONKIN_FORM or not form ) then
  33. if ( GetSpecialization() == 1 ) then
  34. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 115, -88);
  35. else
  36. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "BOTTOMLEFT", 99, 38);
  37. end
  38. elseif ( form == BEAR_FORM or form == CAT_FORM ) then
  39. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 99, -78);
  40. else
  41. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "BOTTOMLEFT", 99, 38);
  42. end
  43. elseif ( class == "MONK" ) then
  44. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 28, -75);
  45. elseif ( class == "MAGE" ) then
  46. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 50, -75);
  47. elseif ( class == "SHAMAN" ) then
  48. if ( GetSpecialization() == 3) then
  49. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "BOTTOMLEFT", 99, 38);
  50. else
  51. TotemFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 99, -78);
  52. end
  53. elseif ( hasPet and class ~= "SHAMAN" and class ~= "WARLOCK" ) then
  54. TotemFrame:Hide();
  55. return;
  56. end
  58. local haveTotem, name, startTime, duration, icon;
  59. local slot;
  60. local button;
  61. local buttonIndex = 1;
  62. TotemFrame.activeTotems = 0;
  63. for i=1, MAX_TOTEMS do
  64. slot = priorities[i];
  65. haveTotem, name, startTime, duration, icon = GetTotemInfo(slot);
  66. if ( haveTotem ) then
  67. button = _G["TotemFrameTotem"..buttonIndex];
  68. button.slot = slot;
  69. TotemButton_Update(button, startTime, duration, icon);
  71. if ( button:IsShown() ) then
  72. TotemFrame.activeTotems = TotemFrame.activeTotems + 1;
  73. end
  75. buttonIndex = buttonIndex + 1;
  76. else
  77. button = _G["TotemFrameTotem"..MAX_TOTEMS - i + buttonIndex];
  78. button.slot = 0;
  79. button:Hide();
  80. end
  81. end
  82. if ( TotemFrame.activeTotems > 0 ) then
  83. TotemFrame:Show();
  84. else
  85. TotemFrame:Hide();
  86. end
  87. TotemFrame_AdjustPetFrame();
  88. PlayerFrame_AdjustAttachments();
  89. end
  91. function TotemFrame_OnEvent(self, event, ...)
  92. if ( event == "PLAYER_TOTEM_UPDATE" ) then
  93. local slot = ...;
  94. if ( slot <= MAX_TOTEMS ) then
  95. local haveTotem, name, startTime, duration, icon = GetTotemInfo(slot);
  96. local button;
  97. for i=1, MAX_TOTEMS do
  98. button = _G["TotemFrameTotem"..i];
  99. if ( button.slot == slot ) then
  100. local previouslyShown = button:IsShown();
  101. TotemButton_Update(button, startTime, duration, icon);
  102. -- if we have no active totems then we need to hide the whole frame, otherwise show it
  103. if ( previouslyShown ) then
  104. if ( not button:IsShown() ) then
  105. self.activeTotems = self.activeTotems - 1;
  106. end
  107. else
  108. if ( button:IsShown() ) then
  109. self.activeTotems = self.activeTotems + 1;
  110. end
  111. end
  112. if ( self.activeTotems > 0 ) then
  113. self:Show();
  114. else
  115. self:Hide();
  116. end
  117. TotemFrame_AdjustPetFrame();
  118. return;
  119. end
  120. end
  121. end
  122. end
  123. TotemFrame_Update();
  124. end
  126. function TotemButton_OnClick(self, mouseButton)
  127. local cannotDismiss = GetTotemCannotDismiss(self.slot)
  128. if ( not cannotDismiss ) then
  129. if ( mouseButton == "RightButton" and self.slot > 0 ) then
  130. DestroyTotem(self.slot);
  131. end
  132. end
  133. end
  135. function TotemButton_OnLoad(self)
  136. self:RegisterForClicks("RightButtonUp");
  137. end
  139. function TotemButton_OnUpdate(button, elapsed)
  140. AuraButton_UpdateDuration(button, GetTotemTimeLeft(button.slot));
  141. if ( GameTooltip:IsOwned(button) ) then
  142. GameTooltip:SetTotem(button.slot);
  143. end
  144. end
  146. function TotemButton_Update(button, startTime, duration, icon)
  147. local buttonName = button:GetName();
  148. local buttonIcon = _G[buttonName.."IconTexture"];
  149. local buttonDuration = _G[buttonName.."Duration"];
  150. local buttonCooldown = _G[buttonName.."IconCooldown"];
  152. if ( duration > 0 ) then
  153. buttonIcon:SetTexture(icon);
  154. buttonIcon:Show();
  155. CooldownFrame_Set(buttonCooldown, startTime, duration, true);
  156. buttonCooldown:Show();
  157. button:SetScript("OnUpdate", TotemButton_OnUpdate);
  158. button:Show();
  159. else
  160. buttonIcon:Hide();
  161. buttonDuration:Hide();
  162. buttonCooldown:Hide();
  163. button:SetScript("OnUpdate", nil);
  164. button:Hide();
  165. end
  166. end
  168. function TotemFrame_AdjustPetFrame()
  169. local _, class = UnitClass("player");
  170. if ( class == "WARLOCK" ) then
  171. if ( PetFrame:IsShown() and TotemFrameTotem4:IsShown() ) then
  172. PetFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 156, -90);
  173. elseif ( PetFrame:IsShown() and TotemFrameTotem3:IsShown() ) then
  174. PetFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 126, -90);
  175. elseif ( PetFrame:IsShown() and TotemFrameTotem2:IsShown() ) then
  176. PetFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 93, -90);
  177. else
  178. PetFrame:SetPoint("TOPLEFT", PlayerFrame, "TOPLEFT", 60, -90);
  179. end
  180. end
  181. end