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

- changed the pet health/mana bars to monitor the player whenever the player enters a vehicle since the player bars already change to display vehicle info in that situation

Browse Source
This commit is contained in:
Parnic
2010-10-21 02:49:59 +00:00
parent 78b825c79d
commit a51fe8e06f
2 changed files with 44 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
modules/PetHealth.lua
View File

@ -60,6 +60,9 @@ function PetHealth.prototype:Enable(core)
self:RegisterEvent("UNIT_HAPPINESS", "PetHappiness") self:RegisterEvent("UNIT_HAPPINESS", "PetHappiness")
end end
self:RegisterEvent("UNIT_ENTERED_VEHICLE", "EnteringVehicle")
self:RegisterEvent("UNIT_EXITED_VEHICLE", "ExitingVehicle")
self.frame:SetAttribute("unit", self.unit) self.frame:SetAttribute("unit", self.unit)
RegisterUnitWatch(self.frame) RegisterUnitWatch(self.frame)
@ -223,5 +226,22 @@ function PetHealth.prototype:EnableClickTargeting(bEnable)
end end
end end
function PetHealth.prototype:EnteringVehicle(event, unit, arg2)
if (self.unit == "pet" and IceHUD:ShouldSwapToVehicle(unit, arg2)) then
self.unit = "player"
self:RegisterFontStrings()
self:Update(self.unit)
end
end
function PetHealth.prototype:ExitingVehicle(event, unit)
if (unit == "player" and self.unit == "player") then
self.unit = "pet"
self:RegisterFontStrings()
self:Update(self.unit)
end
end
-- Load us up -- Load us up
IceHUD.PetHealth = PetHealth:new() IceHUD.PetHealth = PetHealth:new()

48
modules/PetMana.lua
View File

@ -50,11 +50,10 @@ end
function PetMana.prototype:Enable(core) function PetMana.prototype:Enable(core)
PetMana.super.prototype.Enable(self, core) PetMana.super.prototype.Enable(self, core)
self:RegisterEvent("PET_UI_UPDATE", "CheckPet"); self:RegisterEvent("PET_UI_UPDATE", "CheckPet")
self:RegisterEvent("PLAYER_PET_CHANGED", "CheckPet"); self:RegisterEvent("PLAYER_PET_CHANGED", "CheckPet")
self:RegisterEvent("PET_BAR_CHANGED", "CheckPet"); self:RegisterEvent("PET_BAR_CHANGED", "CheckPet")
self:RegisterEvent("UNIT_PET", "CheckPet"); self:RegisterEvent("UNIT_PET", "CheckPet")
self:RegisterEvent("UNIT_ENTERED_VEHICLE", "CheckPet")
if IceHUD.WowVer >= 40000 then if IceHUD.WowVer >= 40000 then
self:RegisterEvent("UNIT_POWER", "UpdateEvent") self:RegisterEvent("UNIT_POWER", "UpdateEvent")
@ -72,14 +71,15 @@ function PetMana.prototype:Enable(core)
self:RegisterEvent("UNIT_DISPLAYPOWER", "ManaType") self:RegisterEvent("UNIT_DISPLAYPOWER", "ManaType")
self:RegisterEvent("UNIT_ENTERED_VEHICLE", "EnteringVehicle")
self:RegisterEvent("UNIT_EXITED_VEHICLE", "ExitingVehicle")
self:CheckPet() self:CheckPet()
self:ManaType(nil, self.unit) self:ManaType(nil, self.unit)
end end
function PetMana.prototype:CheckPet() function PetMana.prototype:CheckPet()
self:CheckForVehicle()
if (UnitExists(self.unit)) then if (UnitExists(self.unit)) then
self:Show(true) self:Show(true)
self:ManaType(nil, self.unit) self:ManaType(nil, self.unit)
@ -123,28 +123,11 @@ function PetMana.prototype:MyOnUpdate()
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(nil, self.unit)
end
end
function PetMana.prototype:UpdateEvent(event, unit) function PetMana.prototype:UpdateEvent(event, unit)
self:Update(unit) self:Update(unit)
end end
function PetMana.prototype:Update(unit) function PetMana.prototype:Update(unit)
self:CheckForVehicle()
PetMana.super.prototype.Update(self) PetMana.super.prototype.Update(self)
if (unit and (unit ~= self.unit)) then if (unit and (unit ~= self.unit)) then
@ -215,7 +198,24 @@ function PetMana.prototype:GetOptions()
return opts return opts
end end
function PetMana.prototype:EnteringVehicle(event, unit, arg2)
if (self.unit == "pet" and IceHUD:ShouldSwapToVehicle(unit, arg2)) then
self.unit = "player"
self:RegisterFontStrings()
self:ManaType(nil, self.unit)
self:Update(self.unit)
end
end
function PetMana.prototype:ExitingVehicle(event, unit)
if (unit == "player" and self.unit == "player") then
self.unit = "pet"
self:RegisterFontStrings()
self:ManaType(nil, self.unit)
self:Update(self.unit)
end
end
-- Load us up -- Load us up
IceHUD.PetMana = PetMana:new() IceHUD.PetMana = PetMana:new()