wow-mods/ice-hud
1
0
Fork 0
mirror of https://github.com/parnic/ice-hud.git synced 2025-06-16 14:50:13 -05:00
Code Issues Projects Releases Wiki Activity

- vehicle fixes! vehicles now regenerate mana/energy properly instead of relying on events to fire (which seem to be too slow)

Browse Source 
- pet bar is now properly colored for all types of vehicles (was sometimes failing previously for vehicles with energy)
This commit is contained in:
Parnic
2008-12-04 04:03:37 +00:00
parent f61e554040
commit 062e994e7b
1 changed files with 57 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
modules/PetMana.lua
View File

@ -55,6 +55,7 @@ function PetMana.prototype:Enable(core)
self:RegisterEvent("PLAYER_PET_CHANGED", "CheckPet");
self:RegisterEvent("PET_BAR_CHANGED", "CheckPet");
self:RegisterEvent("UNIT_PET", "CheckPet");
self:RegisterEvent("UNIT_ENTERED_VEHICLE", "CheckPet")
self:RegisterEvent("UNIT_MANA", "Update")
self:RegisterEvent("UNIT_MAXMANA", "Update")
@ -73,8 +74,11 @@ end
function PetMana.prototype:CheckPet()
self:CheckForVehicle()
if (UnitExists(self.unit)) then
self:Show(true)
self:ManaType(self.unit)
self:Update(self.unit)
else
self:Show(false)
@ -92,8 +96,49 @@ function PetMana.prototype:ManaType(unit)
end
function PetMana.prototype:SetupOnUpdate(enable)
if enable then
self.frame:SetScript("OnUpdate", function() self:Update(self.unit) end)
else
-- make sure the animation has a chance to finish filling up the bar before we cut it off completely
if self.CurrScale ~= self.DesiredScale then
self.frame:SetScript("OnUpdate", function() self:MyOnUpdate() end)
else
self.frame:SetScript("OnUpdate", nil)
end
end
end
function PetMana.prototype:MyOnUpdate()
PetMana.super.prototype.MyOnUpdate(self)
if self.CurrScale == self.DesiredScale then
self:SetupOnUpdate(false)
end
end
function PetMana.prototype:CheckForVehicle()
local lastUnit = self.unit
if UnitIsUnit("pet", "vehicle") or UnitHasVehicleUI("player") then
self.unit = "vehicle"
else
self.unit = "pet"
end
if self.unit ~= lastUnit then
self:ManaType(self.unit)
end
end
function PetMana.prototype:Update(unit)
self:CheckForVehicle()
PetMana.super.prototype.Update(self)
if (unit and (unit ~= self.unit)) then
return
end
@ -105,6 +150,13 @@ function PetMana.prototype:Update(unit)
self:Show(true)
end
if (self.manaPercentage == 1 and self.manaType ~= 1 and self.manaType ~= 6)
or (self.manaPercentage == 0 and (self.manaType == 1 or self.manaType == 6)) then
self:SetupOnUpdate(false)
elseif GetCVarBool("predictedPower") then
self:SetupOnUpdate(true)
end
local color = "PetMana"
if (self.moduleSettings.scaleManaColor) then
color = "ScaledManaColor"