Files
ice-hud/modules/TargetInfo.lua
iceroth b8956a45fe Version 0.6
- Improved TargetInfo and ToT module
- RoundBar texture looks better than ever
- PlayerHealth and TargetHealth modules have an option to turn off Blizzard default frames
- Fixed MH3 rounding bug
2006-09-02 20:20:01 +00:00

622 lines
16 KiB
Lua

local AceOO = AceLibrary("AceOO-2.0")
local TargetInfo = AceOO.Class(IceElement)
local target = "target"
local internal = "internal"
TargetInfo.prototype.buffSize = nil
TargetInfo.prototype.width = nil
TargetInfo.prototype.name = nil
TargetInfo.prototype.guild = nil
TargetInfo.prototype.realm = nil
TargetInfo.prototype.rank = nil
TargetInfo.prototype.classLocale = nil
TargetInfo.prototype.classEnglish = nil
TargetInfo.prototype.leader = nil
TargetInfo.prototype.combat = nil
TargetInfo.prototype.pvp = nil
TargetInfo.prototype.level = nil
TargetInfo.prototype.classification = nil
TargetInfo.prototype.reaction = nil
TargetInfo.prototype.tapped = nil
TargetInfo.prototype.isPlayer = nil
-- Constructor --
function TargetInfo.prototype:init()
TargetInfo.super.prototype.init(self, "TargetInfo")
self.scalingEnabled = true
end
-- 'Public' methods -----------------------------------------------------------
-- OVERRIDE
function TargetInfo.prototype:Enable(core)
TargetInfo.super.prototype.Enable(self, core)
self:RegisterEvent("PLAYER_TARGET_CHANGED", "TargetChanged")
self:RegisterEvent("UNIT_AURA", "AuraChanged")
self:RegisterEvent("UNIT_FACTION", "TargetFaction")
self:RegisterEvent("UNIT_LEVEL", "TargetLevel")
self:RegisterEvent("UNIT_FLAGS", "TargetFlags")
self:RegisterEvent("UNIT_DYNAMIC_FLAGS", "TargetFlags")
self:RegisterEvent("RAID_TARGET_UPDATE", "UpdateRaidTargetIcon")
end
-- OVERRIDE
function TargetInfo.prototype:GetOptions()
local opts = TargetInfo.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["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:Redraw()
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:Redraw()
end,
min = 8,
max = 20,
step = 1,
disabled = function()
return not self.moduleSettings.enabled
end,
order = 34
}
return opts
end
-- OVERRIDE
function TargetInfo.prototype:GetDefaultSettings()
local defaults = TargetInfo.super.prototype.GetDefaultSettings(self)
defaults["fontSize"] = 13
defaults["vpos"] = -50
defaults["zoom"] = 0.2
defaults["buffSize"] = 13
return defaults
end
-- OVERRIDE
function TargetInfo.prototype:Redraw()
TargetInfo.super.prototype.Redraw(self)
if (self.moduleSettings.enabled) then
self:CreateFrame(true)
self:TargetChanged()
end
end
-- 'Protected' methods --------------------------------------------------------
-- OVERRIDE
function TargetInfo.prototype:CreateFrame()
TargetInfo.super.prototype.CreateFrame(self)
self.width = self.settings.gap + 50
self.frame:SetFrameStrata("BACKGROUND")
self.frame:SetWidth(self.width)
self.frame:SetHeight(42)
self.frame:ClearAllPoints()
self.frame:SetPoint("TOP", self.parent, "BOTTOM", 0, self.moduleSettings.vpos)
self.frame:SetScale(self.moduleSettings.scale)
self.frame:Show()
self:CreateTextFrame()
self:CreateInfoTextFrame()
self:CreateGuildTextFrame()
self:CreateBuffFrame()
self:CreateDebuffFrame()
self:CreateRaidIconFrame()
end
function TargetInfo.prototype:CreateTextFrame()
if (not self.frame.target) then
self.frame.target = CreateFrame("Button", nil, self.frame)
end
self.frame.target.unit = target -- for blizz default tooltip handling
self.frame.target:RegisterForClicks("LeftButtonUp", "RightButtonUp")
self.frame.target:SetScript("OnClick", function() self:TargetOnClick(arg1) end)
self.frame.target:SetScript("OnEnter", function() UnitFrame_OnEnter() end)
self.frame.target:SetScript("OnLeave", function() UnitFrame_OnLeave() end)
self.frame.target:SetWidth(self.width)
self.frame.target:SetHeight(14)
self.frame.target:SetPoint("TOP", self.frame, "TOP", 0, -2)
self.frame.targetName = self:FontFactory("Bold", self.moduleSettings.fontSize+1, nil, self.frame.targetName)
self.frame.targetName:SetJustifyH("CENTER")
self.frame.targetName:SetJustifyV("TOP")
self.frame.targetName:SetAllPoints(self.frame.target)
self.frame.target:Show()
end
function TargetInfo.prototype:CreateInfoTextFrame()
self.frame.targetInfo = self:FontFactory(nil, self.moduleSettings.fontSize, nil, self.frame.targetInfo)
self.frame.targetInfo:SetWidth(self.width)
self.frame.targetInfo:SetHeight(14)
self.frame.targetInfo:SetJustifyH("CENTER")
self.frame.targetInfo:SetJustifyV("TOP")
self.frame.targetInfo:SetPoint("TOP", self.frame, "TOP", 0, -16)
self.frame.targetInfo:Show()
end
function TargetInfo.prototype:CreateGuildTextFrame()
self.frame.targetGuild = self:FontFactory(nil, self.moduleSettings.fontSize, nil, self.frame.targetGuild)
self.frame.targetGuild:SetHeight(14)
self.frame.targetGuild:SetJustifyH("CENTER")
self.frame.targetGuild:SetJustifyV("TOP")
self.frame.targetGuild:SetAlpha(0.6)
self.frame.targetGuild:SetPoint("TOP", self.frame, "TOP", 0, -30)
self.frame.targetGuild:Show()
end
function TargetInfo.prototype:CreateRaidIconFrame()
if (not self.frame.raidIcon) then
self.frame.raidIcon = CreateFrame("Frame", nil, self.frame)
end
if (not self.frame.raidIcon.icon) then
self.frame.raidIcon.icon = self.frame.raidIcon:CreateTexture(nil, "BACKGROUND")
self.frame.raidIcon.icon:SetTexture("Interface\\TargetingFrame\\UI-RaidTargetingIcons")
end
self.frame.raidIcon:SetPoint("BOTTOM", self.frame, "TOP", 0, 1)
self.frame.raidIcon:SetWidth(16)
self.frame.raidIcon:SetHeight(16)
self.frame.raidIcon.icon:SetAllPoints(self.frame.raidIcon)
SetRaidTargetIconTexture(self.frame.raidIcon.icon, 0)
self.frame.raidIcon:Hide()
end
function TargetInfo.prototype:CreateBuffFrame()
if (not self.frame.buffFrame) then
self.frame.buffFrame = CreateFrame("Frame", nil, self.frame)
end
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", -5, 0)
self.frame.buffFrame:Show()
if (not self.frame.buffFrame.buffs) then
self.frame.buffFrame.buffs = {}
end
self.frame.buffFrame.buffs = self:CreateIconFrames(self.frame.buffFrame, -1, self.frame.buffFrame.buffs, "buff")
end
function TargetInfo.prototype:CreateDebuffFrame()
if (not self.frame.debuffFrame) then
self.frame.debuffFrame = CreateFrame("Frame", nil, self.frame)
end
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", 5, 0)
self.frame.debuffFrame:Show()
if (not self.frame.debuffFrame.buffs) then
self.frame.debuffFrame.buffs = {}
end
self.frame.debuffFrame.buffs = self:CreateIconFrames(self.frame.debuffFrame, 1, self.frame.debuffFrame.buffs, "debuff")
end
function TargetInfo.prototype:CreateIconFrames(parent, direction, buffs, type)
for i = 1, 16 do
if (not buffs[i]) then
buffs[i] = CreateFrame("Frame", nil, parent)
buffs[i].icon = CreateFrame("Frame", nil, buffs[i])
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)
local pos = (i > 8) and i-8 or i
local x = (((pos-1) * self.moduleSettings.buffSize) + (pos-0)) * direction
local y = (i > 8) and -self.moduleSettings.buffSize-0 or 0
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)
buffs[i].icon.stack = self:FontFactory("Bold", 11, buffs[i].icon)
buffs[i].icon.stack:ClearAllPoints()
buffs[i].icon.stack:SetPoint("BOTTOMRIGHT" , buffs[i].icon, "BOTTOMRIGHT", 1, -1)
end
buffs[i]:EnableMouse(true)
buffs[i].id = i
buffs[i]:SetScript("OnEnter", function() self:BuffOnEnter(type) end)
buffs[i]:SetScript("OnLeave", function() GameTooltip:Hide() end)
end
return buffs
end
function TargetInfo.prototype:UpdateBuffs()
local zoom = self.moduleSettings.zoom
for i = 1, 16 do
local buffTexture, buffApplications = UnitBuff("target", i)
--buffTexture = buffTexture or "Interface\\Icons\\Spell_Nature_Regeneration"
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.3 or 0
self.frame.buffFrame.buffs[i].texture:SetTexture(0, 0, 0, alpha)
self.frame.buffFrame.buffs[i].texture:SetVertexColor(color.r, color.g, color.b)
--buffApplications = 2
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
if (buffTexture) then
self.frame.buffFrame.buffs[i]:Show()
else
self.frame.buffFrame.buffs[i]:Hide()
end
end
for i = 1, 16 do
local buffTexture, buffApplications, debuffDispelType = UnitDebuff("target", i)
--buffTexture = buffTexture or "Interface\\Icons\\Ability_Creature_Disease_04"
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)
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
if (buffTexture) then
self.frame.debuffFrame.buffs[i]:Show()
else
self.frame.debuffFrame.buffs[i]:Hide()
end
end
end
function TargetInfo.prototype:AuraChanged(unit)
if (unit == target) then
self:UpdateBuffs()
end
end
function TargetInfo.prototype:UpdateRaidTargetIcon()
if not (UnitExists(target)) then
self.frame.raidIcon:Hide()
return
end
local index = GetRaidTargetIndex(target);
if (index and (index > 0)) then
SetRaidTargetIconTexture(self.frame.raidIcon.icon, index)
self.frame.raidIcon:Show()
else
self.frame.raidIcon:Hide()
end
end
function TargetInfo.prototype:TargetChanged()
if (not UnitExists(target)) then
self.frame.targetName:SetText()
self.frame.targetInfo:SetText()
self.frame.targetGuild:SetText()
self:UpdateBuffs()
self:UpdateRaidTargetIcon()
return
end
self.name, self.realm = UnitName(target)
self.classLocale, self.classEnglish = UnitClass(target)
self.isPlayer = UnitIsPlayer(target)
local guildName, guildRankName, guildRankIndex = GetGuildInfo(target);
self.guild = guildName and "<" .. guildName .. ">" or ""
if (self.classLocale and self.isPlayer) then
self.classLocale = "|c" .. self:GetHexColor(self.classEnglish) .. self.classLocale .. "|r"
else
self.classLocale = UnitCreatureType(target)
end
self.leader = UnitIsPartyLeader(target) and " Leader" or ""
-- pass "internal" as a paramater so event handler code doesn't execute
-- self:Update() unnecassarily
self:TargetLevel(internal)
self:TargetReaction(internal)
self:TargetFaction(internal)
self:TargetFlags(internal)
self:UpdateBuffs()
self:UpdateRaidTargetIcon()
self:Update(target)
end
function TargetInfo.prototype:TargetLevel(unit)
if (unit == target or unit == internal) then
self.level = UnitLevel(target)
local color = GetDifficultyColor((self.level > 0) and self.level or 100)
if (self.level > 0) then
if (UnitIsPlusMob(target)) then
self.level = self.level .. "+"
end
else
self.level = "??"
end
self.level = "|c" .. self:ConvertToHex(color) .. self.level .. "|r"
self:Update(unit)
end
end
function TargetInfo.prototype:TargetReaction(unit)
if (unit == target or unit == internal) then
self.reaction = UnitReaction(target, "player")
-- if we don't get reaction, unit is out of range - has to be friendly
-- to be targettable (party/raid)
if (not self.reaction) then
self.reaction = 5
end
self:Update(unit)
end
end
-- PVP status
function TargetInfo.prototype:TargetFaction(unit)
if (unit == target or unit == internal) then
if (self.isPlayer) then
if (UnitIsPVP(target)) then
local color = "ff10ff10" -- friendly
if (UnitFactionGroup(target) ~= UnitFactionGroup("player")) then
color = "ffff1010" -- hostile
end
self.pvp = " |c" .. color .. "PvP|r"
else
self.pvp = " |cff1010ffPvE|r"
end
else
self.pvp = ""
end
self:Update(unit)
end
end
function TargetInfo.prototype:TargetFlags(unit)
if (unit == target or unit == internal) then
self.tapped = UnitIsTapped(target) and (not UnitIsTappedByPlayer(target))
self.combat = UnitAffectingCombat(target) and " |cffee4030Combat|r" or ""
self:Update(unit)
end
end
function TargetInfo.prototype:Update(unit)
if (unit ~= target) then
return
end
local reactionColor = self:ConvertToHex(UnitReactionColor[self.reaction])
if (self.tapped) then
reactionColor = self:GetHexColor("tapped")
end
local line1 = string.format("|c%s%s|r", reactionColor, self.name or '')
self.frame.targetName:SetText(line1)
local line2 = string.format("%s %s%s%s%s",
self.level or '', self.classLocale or '', self.pvp or '', self.leader or '', self.combat or '')
self.frame.targetInfo:SetText(line2)
local realm = self.realm and " " .. self.realm or ""
local line3 = string.format("%s%s", self.guild or '', realm)
self.frame.targetGuild:SetText(line3)
end
function TargetInfo.prototype:TargetOnClick(button)
-- copy&paste from blizz code, it better work ;)
if (SpellIsTargeting() and button == "RightButton") then
SpellStopTargeting()
return
end
if (button == "LeftButton") then
if (SpellIsTargeting()) then
SpellTargetUnit(target)
elseif (CursorHasItem()) then
DropItemOnUnit(target)
end
else
ToggleDropDownMenu(1, nil, TargetFrameDropDown, "cursor")
end
end
function TargetInfo.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
TargetInfo:new()