mirror of
https://github.com/parnic/ice-hud.git
synced 2025-06-16 14:50:13 -05:00
747bc0bc42
- bumped up the default position of the pet mana/health values to get them out of the way of nearby bars - changed pet health/mana dogtags to display on the uppertext block instead of the lowertext block so it behaves and alphas like all other bars
148 lines
3.2 KiB
Lua
148 lines
3.2 KiB
Lua
local AceOO = AceLibrary("AceOO-2.0")
|
|
|
|
local PetHealth = AceOO.Class(IceUnitBar)
|
|
|
|
PetHealth.prototype.happiness = nil
|
|
|
|
|
|
-- Constructor --
|
|
function PetHealth.prototype:init()
|
|
PetHealth.super.prototype.init(self, "PetHealth", "pet")
|
|
|
|
self:SetDefaultColor("PetHealthHappy", 37, 164, 30)
|
|
self:SetDefaultColor("PetHealthContent", 164, 164, 30)
|
|
self:SetDefaultColor("PetHealthUnhappy", 164, 30, 30)
|
|
|
|
self.scalingEnabled = true
|
|
end
|
|
|
|
|
|
-- OVERRIDE
|
|
function PetHealth.prototype:GetDefaultSettings()
|
|
local settings = PetHealth.super.prototype.GetDefaultSettings(self)
|
|
|
|
settings["side"] = IceCore.Side.Left
|
|
settings["offset"] = -1
|
|
settings.scale = 0.7
|
|
settings["textVerticalOffset"] = 4
|
|
settings["upperText"] = "[PercentHP:Round]"
|
|
settings["lowerText"] = ""
|
|
|
|
return settings
|
|
end
|
|
|
|
|
|
-- OVERRIDE
|
|
--[[
|
|
function PetHealth.prototype:CreateFrame()
|
|
PetHealth.super.prototype.CreateFrame(self)
|
|
|
|
local point, relativeTo, relativePoint, xoff, yoff = self.frame.bottomUpperText:GetPoint()
|
|
if (point == "TOPLEFT") then
|
|
point = "BOTTOMLEFT"
|
|
else
|
|
point = "BOTTOMRIGHT"
|
|
end
|
|
|
|
self.frame.bottomUpperText:ClearAllPoints()
|
|
self.frame.bottomUpperText:SetPoint(point, relativeTo, relativePoint, 0, 0)
|
|
end
|
|
]]
|
|
|
|
function PetHealth.prototype:Enable(core)
|
|
PetHealth.super.prototype.Enable(self, core)
|
|
|
|
self:RegisterEvent("PET_UI_UPDATE", "CheckPet");
|
|
self:RegisterEvent("PLAYER_PET_CHANGED", "CheckPet");
|
|
self:RegisterEvent("PET_BAR_CHANGED", "CheckPet");
|
|
self:RegisterEvent("UNIT_PET", "CheckPet");
|
|
|
|
self:RegisterEvent("UNIT_HEALTH", "Update")
|
|
self:RegisterEvent("UNIT_MAXHEALTH", "Update")
|
|
|
|
self:RegisterEvent("UNIT_HAPPINESS", "PetHappiness")
|
|
|
|
self:CheckPet()
|
|
end
|
|
|
|
function PetHealth.prototype:PetHappiness(unit)
|
|
if (unit and (unit ~= self.unit)) then
|
|
return
|
|
end
|
|
|
|
self.happiness = GetPetHappiness()
|
|
self.happiness = self.happiness or 3 -- '3' means happy
|
|
self:Update(unit)
|
|
end
|
|
|
|
|
|
function PetHealth.prototype:CheckPet()
|
|
if (UnitExists(self.unit)) then
|
|
self.frame:Show()
|
|
self:PetHappiness(self.unit)
|
|
self:Update(self.unit)
|
|
else
|
|
self.frame:Hide()
|
|
end
|
|
end
|
|
|
|
|
|
function PetHealth.prototype:Update(unit)
|
|
PetHealth.super.prototype.Update(self)
|
|
if (unit and (unit ~= self.unit)) then
|
|
return
|
|
end
|
|
|
|
local color = "PetHealthHappy"
|
|
if (self.happiness == 2) then
|
|
color = "PetHealthContent"
|
|
elseif(self.happiness == 1) then
|
|
color = "PetHealthUnhappy"
|
|
end
|
|
|
|
if (self.moduleSettings.scaleColor) then
|
|
color = "ScaledHealthColor"
|
|
end
|
|
|
|
if not (self.alive) then
|
|
color = "Dead"
|
|
end
|
|
|
|
|
|
self:UpdateBar(self.health/self.maxHealth, color)
|
|
|
|
if not AceLibrary:HasInstance("LibDogTag-2.0") then
|
|
self:SetBottomText1(math.floor(self.healthPercentage * 100))
|
|
end
|
|
end
|
|
|
|
|
|
-- OVERRIDE
|
|
function PetHealth.prototype:GetOptions()
|
|
local opts = PetHealth.super.prototype.GetOptions(self)
|
|
|
|
opts["scaleHealthColor"] = {
|
|
type = "toggle",
|
|
name = "Color bar by health %",
|
|
desc = "Colors the health bar from MaxHealthColor to MinHealthColor based on current health %",
|
|
get = function()
|
|
return self.moduleSettings.scaleHealthColor
|
|
end,
|
|
set = function(value)
|
|
self.moduleSettings.scaleHealthColor = value
|
|
self:Redraw()
|
|
end,
|
|
disabled = function()
|
|
return not self.moduleSettings.enabled
|
|
end,
|
|
order = 41
|
|
}
|
|
|
|
return opts
|
|
end
|
|
|
|
|
|
|
|
-- Load us up
|
|
IceHUD.PetHealth = PetHealth:new()
|