local AceOO = AceLibrary("AceOO-2.0") local PetInfo = AceOO.Class(IceElement) local target = "playerpet" local internal = "internal" PetInfo.prototype.buffSize = nil PetInfo.prototype.width = nil PetInfo.prototype.name = nil PetInfo.prototype.guild = nil PetInfo.prototype.realm = nil PetInfo.prototype.classLocale = nil PetInfo.prototype.classEnglish = nil PetInfo.prototype.leader = nil PetInfo.prototype.combat = nil PetInfo.prototype.pvp = nil PetInfo.prototype.level = nil PetInfo.prototype.classification = nil PetInfo.prototype.reaction = nil PetInfo.prototype.tapped = nil PetInfo.prototype.isPlayer = nil -- Constructor -- function PetInfo.prototype:init() PetInfo.super.prototype.init(self, "PetInfo") self.scalingEnabled = true end -- 'Public' methods ----------------------------------------------------------- -- OVERRIDE function PetInfo.prototype:Enable(core) PetInfo.super.prototype.Enable(self, core) self:RegisterEvent("PLAYER_TARGET_CHANGED", "TargetChanged") self:RegisterEvent("UNIT_AURA", "AuraChanged") RegisterUnitWatch(self.frame) end -- OVERRIDE function PetInfo.prototype:Disable(core) PetInfo.super.prototype.Disable(self, core) UnregisterUnitWatch(self.frame) end -- OVERRIDE function PetInfo.prototype:GetOptions() local opts = PetInfo.super.prototype.GetOptions(self) opts["vpos"] = { type = "range", name = "Vertical Position", desc = "Vertical Position", get = function() return self.moduleSettings.vpos end, set = function(v) self.moduleSettings.vpos = v self:Redraw() end, min = -300, max = 300, step = 10, disabled = function() return not self.moduleSettings.enabled end, order = 31 } opts["fontSize"] = { type = 'range', name = 'Font Size', desc = 'Font Size', get = function() return self.moduleSettings.fontSize end, set = function(v) self.moduleSettings.fontSize = v self:Redraw() end, min = 8, max = 20, step = 1, disabled = function() return not self.moduleSettings.enabled end, order = 32 } opts["stackFontSize"] = { type = 'range', name = 'Stack Font Size', desc = 'Stack Font Size', get = function() return self.moduleSettings.stackFontSize end, set = function(v) self.moduleSettings.stackFontSize = v self:RedrawBuffs() end, min = 8, max = 20, step = 1, disabled = function() return not self.moduleSettings.enabled end, order = 32 } opts["zoom"] = { type = 'range', name = 'Buff zoom', desc = 'Buff/debuff icon zoom', get = function() return self.moduleSettings.zoom end, set = function(v) self.moduleSettings.zoom = v self:RedrawBuffs() end, min = 0, max = 0.2, step = 0.01, disabled = function() return not self.moduleSettings.enabled end, isPercent = true, order = 33 } opts["buffSize"] = { type = 'range', name = 'Buff size', desc = 'Buff/debuff icon size', get = function() return self.moduleSettings.buffSize end, set = function(v) self.moduleSettings.buffSize = v self:RedrawBuffs() end, min = 8, max = 30, step = 1, disabled = function() return not self.moduleSettings.enabled end, order = 34 } opts["filter"] = { type = 'text', name = 'Filter buffs/debuffs', desc = 'Toggles filtering buffs and debuffs (uses Blizzard default filter code)', get = function() return self.moduleSettings.filter end, set = function(v) self.moduleSettings.filter = v self:RedrawBuffs() end, disabled = function() return not self.moduleSettings.enabled end, validate = { "Never", "In Combat", "Always" }, order = 35 } opts["perRow"] = { type = 'range', name = 'Buffs / row', desc = 'How many buffs/debuffs is shown on each row', get = function() return self.moduleSettings.perRow end, set = function(v) self.moduleSettings.perRow = v self:RedrawBuffs() end, min = 5, max = 20, step = 1, disabled = function() return not self.moduleSettings.enabled end, order = 36 } opts["mouseTarget"] = { type = 'toggle', name = 'Mouseover for target', desc = 'Toggle mouseover on/off for target', get = function() return self.moduleSettings.mouseTarget end, set = function(v) self.moduleSettings.mouseTarget = v self:Redraw() end, disabled = function() return not self.moduleSettings.enabled end, order = 37 } opts["mouseBuff"] = { type = 'toggle', name = 'Mouseover for buffs', desc = 'Toggle mouseover on/off for buffs/debuffs', get = function() return self.moduleSettings.mouseBuff end, set = function(v) self.moduleSettings.mouseBuff = v self:RedrawBuffs() end, disabled = function() return not self.moduleSettings.enabled end, order = 38 } return opts end -- OVERRIDE function PetInfo.prototype:GetDefaultSettings() local defaults = PetInfo.super.prototype.GetDefaultSettings(self) defaults["fontSize"] = 13 defaults["stackFontSize"] = 11 defaults["vpos"] = -50 defaults["zoom"] = 0.08 defaults["buffSize"] = 20 defaults["mouseTarget"] = true defaults["mouseBuff"] = true defaults["filter"] = "Never" defaults["perRow"] = 10 return defaults end -- OVERRIDE function PetInfo.prototype:Redraw() PetInfo.super.prototype.Redraw(self) if (self.moduleSettings.enabled) then self:CreateFrame(true) self:TargetChanged() end end function PetInfo.prototype:RedrawBuffs() if (self.moduleSettings.enabled) then self:CreateBuffFrame(false) self:CreateDebuffFrame(false) self:TargetChanged() end end -- 'Protected' methods -------------------------------------------------------- -- OVERRIDE function PetInfo.prototype:CreateFrame(redraw) if not (self.frame) then self.frame = CreateFrame("Button", "IceHUD_"..self.elementName, self.parent, "SecureUnitButtonTemplate") end self.width = self.settings.gap + 50 self.frame:SetScale(self.moduleSettings.scale) -- set showing/hiding the frame depending on current target self.frame:SetAttribute("unit", target) self.frame:SetFrameStrata("BACKGROUND") self.frame:SetWidth(self.width) self.frame:SetHeight(32) self.frame:ClearAllPoints() self.frame:SetPoint("TOP", self.parent, "BOTTOM", 0, self.moduleSettings.vpos) self.frame:SetScale(self.moduleSettings.scale) if (self.moduleSettings.mouseTarget) then self.frame:EnableMouse(true) self.frame:RegisterForClicks("AnyUp") else self.frame:EnableMouse(false) self.frame:RegisterForClicks() end self.frame.unit = target -- set up stuff for clicking self.frame:SetAttribute("type1", "playerpet") self.frame:SetAttribute("type2", "menu") self.frame:SetAttribute("unit", target) self.frame.menu = function() ToggleDropDownMenu(1, nil, TargetFrameDropDown, "cursor") end -- create a fancy highlight frame for mouse over if (not self.frame.highLight) then self.frame.highLight = self.frame:CreateTexture(nil, "OVERLAY") self.frame.highLight:SetTexture("Interface\\QuestFrame\\UI-QuestTitleHighlight") self.frame.highLight:SetBlendMode("ADD") self.frame.highLight:SetAllPoints(self.frame) self.frame.highLight:SetVertexColor(1, 1, 1, 0.25) self.frame.highLight:Hide() end -- create rest of the frames self:CreateBuffFrame(redraw) self:CreateDebuffFrame(redraw) end function PetInfo.prototype:CreateBuffFrame(redraw) if (not self.frame.buffFrame) then self.frame.buffFrame = CreateFrame("Frame", nil, self.frame) self.frame.buffFrame:SetFrameStrata("BACKGROUND") self.frame.buffFrame:SetWidth(1) self.frame.buffFrame:SetHeight(1) self.frame.buffFrame:ClearAllPoints() self.frame.buffFrame:SetPoint("TOPRIGHT", self.frame, "TOPLEFT", -10, 0) self.frame.buffFrame:Show() self.frame.buffFrame.buffs = {} end if (not redraw) then self.frame.buffFrame.buffs = self:CreateIconFrames(self.frame.buffFrame, -1, self.frame.buffFrame.buffs, "buff") end end function PetInfo.prototype:CreateDebuffFrame(redraw) if (not self.frame.debuffFrame) then self.frame.debuffFrame = CreateFrame("Frame", nil, self.frame) self.frame.debuffFrame:SetFrameStrata("BACKGROUND") self.frame.debuffFrame:SetWidth(1) self.frame.debuffFrame:SetHeight(1) self.frame.debuffFrame:ClearAllPoints() self.frame.debuffFrame:SetPoint("TOPLEFT", self.frame, "TOPRIGHT", 10, 0) self.frame.debuffFrame:Show() self.frame.debuffFrame.buffs = {} end if (not redraw) then self.frame.debuffFrame.buffs = self:CreateIconFrames(self.frame.debuffFrame, 1, self.frame.debuffFrame.buffs, "debuff") end end function PetInfo.prototype:CreateIconFrames(parent, direction, buffs, type) for i = 1, IceCore.BuffLimit do if (not buffs[i]) then buffs[i] = CreateFrame("Frame", nil, parent) buffs[i].icon = CreateFrame("Frame", nil, buffs[i]) buffs[i].cd = CreateFrame("Cooldown", nil, buffs[i], "CooldownFrameTemplate") end buffs[i]:SetFrameStrata("BACKGROUND") buffs[i]:SetWidth(self.moduleSettings.buffSize) buffs[i]:SetHeight(self.moduleSettings.buffSize) buffs[i].icon:SetFrameStrata("BACKGROUND") buffs[i].icon:SetWidth(self.moduleSettings.buffSize-2) buffs[i].icon:SetHeight(self.moduleSettings.buffSize-2) buffs[i].cd:SetFrameStrata("BACKGROUND") buffs[i].cd:SetFrameLevel(buffs[i].icon:GetFrameLevel()+1) buffs[i].cd:SetReverse(true) buffs[i].cd:ClearAllPoints() buffs[i].cd:SetAllPoints(buffs[i]) local pos = math.fmod(i, self.moduleSettings.perRow) if (pos == 0) then pos = self.moduleSettings.perRow end local x = (((pos-1) * self.moduleSettings.buffSize) + pos) * direction local y = math.floor((i-1) / self.moduleSettings.perRow) * self.moduleSettings.buffSize * -1 buffs[i]:ClearAllPoints() buffs[i]:SetPoint("TOP", x, y) buffs[i].icon:ClearAllPoints() buffs[i].icon:SetPoint("CENTER", 0, 0) buffs[i]:Show() buffs[i].icon:Show() if (not buffs[i].texture) then buffs[i].texture = buffs[i]:CreateTexture() buffs[i].texture:ClearAllPoints() buffs[i].texture:SetAllPoints(buffs[i]) buffs[i].icon.texture = buffs[i].icon:CreateTexture() buffs[i].icon.texture:SetTexture(nil) buffs[i].icon.texture:ClearAllPoints() buffs[i].icon.texture:SetAllPoints(buffs[i].icon) end buffs[i].icon.stack = self:FontFactory(self.moduleSettings.stackFontSize, buffs[i].icon, buffs[i].icon.stack, "OUTLINE") buffs[i].icon.stack:ClearAllPoints() buffs[i].icon.stack:SetPoint("BOTTOMRIGHT" , buffs[i].icon, "BOTTOMRIGHT", 3, -1) buffs[i].id = i if (self.moduleSettings.mouseBuff) then buffs[i]:EnableMouse(true) buffs[i]:SetScript("OnEnter", function() self:BuffOnEnter(type) end) buffs[i]:SetScript("OnLeave", function() GameTooltip:Hide() end) else buffs[i]:EnableMouse(false) buffs[i]:SetScript("OnEnter", nil) buffs[i]:SetScript("OnLeave", nil) end end return buffs end function PetInfo.prototype:UpdateBuffs() local zoom = self.moduleSettings.zoom local filter = false if (self.moduleSettings.filter == "Always") then filter = true elseif (self.moduleSettings.filter == "In Combat") then if (UnitAffectingCombat("player")) then filter = true end end local hostile = UnitCanAttack("player", "playerpet") for i = 1, IceCore.BuffLimit do local buffName, buffRank, buffTexture, buffApplications, buffDuration, buffTimeLeft = UnitBuff("playerpet", i, filter and not hostile) if (buffTexture) then self.frame.buffFrame.buffs[i].icon.texture:SetTexture(buffTexture) self.frame.buffFrame.buffs[i].icon.texture:SetTexCoord(zoom, 1-zoom, zoom, 1-zoom) local alpha = buffTexture and 0.5 or 0 self.frame.buffFrame.buffs[i].texture:SetTexture(0, 0, 0, alpha) -- cooldown frame if (buffDuration and buffDuration > 0 and buffTimeLeft and buffTimeLeft > 0) then local start = GetTime() - buffDuration + buffTimeLeft self.frame.buffFrame.buffs[i].cd:SetCooldown(start, buffDuration) self.frame.buffFrame.buffs[i].cd:Show() else self.frame.buffFrame.buffs[i].cd:Hide() end if (buffApplications and (buffApplications > 1)) then self.frame.buffFrame.buffs[i].icon.stack:SetText(buffApplications) else self.frame.buffFrame.buffs[i].icon.stack:SetText(nil) end self.frame.buffFrame.buffs[i]:Show() else self.frame.buffFrame.buffs[i]:Hide() end end for i = 1, IceCore.BuffLimit do local buffName, buffRank, buffTexture, buffApplications, debuffDispelType, debuffDuration, debuffTimeLeft = UnitDebuff("playerpet", i, filter and not hostile) if (buffTexture and (not hostile or not filter or (filter and debuffDuration))) then local color = debuffDispelType and DebuffTypeColor[debuffDispelType] or DebuffTypeColor["none"] local alpha = buffTexture and 1 or 0 self.frame.debuffFrame.buffs[i].texture:SetTexture(1, 1, 1, alpha) self.frame.debuffFrame.buffs[i].texture:SetVertexColor(color.r, color.g, color.b) -- cooldown frame if (debuffDuration and debuffDuration > 0 and debuffTimeLeft and debuffTimeLeft > 0) then local start = GetTime() - debuffDuration + debuffTimeLeft self.frame.debuffFrame.buffs[i].cd:SetCooldown(start, debuffDuration) self.frame.debuffFrame.buffs[i].cd:Show() else self.frame.debuffFrame.buffs[i].cd:Hide() end self.frame.debuffFrame.buffs[i].icon.texture:SetTexture(buffTexture) self.frame.debuffFrame.buffs[i].icon.texture:SetTexCoord(zoom, 1-zoom, zoom, 1-zoom) if (buffApplications and (buffApplications > 1)) then self.frame.debuffFrame.buffs[i].icon.stack:SetText(buffApplications) else self.frame.debuffFrame.buffs[i].icon.stack:SetText(nil) end self.frame.debuffFrame.buffs[i]:Show() else self.frame.debuffFrame.buffs[i]:Hide() end end end function PetInfo.prototype:AuraChanged(unit) if (unit == "pet") then self:UpdateBuffs() end end function PetInfo.prototype:TargetChanged() if (not UnitExists("target")) then self.frame.buffFrame:Hide() self.frame.debuffFrame:Hide() return end self.frame.buffFrame:Show() self.frame.debuffFrame:Show() self:UpdateBuffs() end function PetInfo.prototype:BuffOnEnter(type) if (not this:IsVisible()) then return end GameTooltip:SetOwner(this, "ANCHOR_BOTTOMRIGHT") if (type == "buff") then GameTooltip:SetUnitBuff(target, this.id) else GameTooltip:SetUnitDebuff(target, this.id) end end -- Load us up IceHUD.PetInfo = PetInfo:new()