From b734f105b1cb49a7be4376537d9763b95423dd47 Mon Sep 17 00:00:00 2001
From: Parnic <parnic@parnic.com>
Date: Sun, 28 Sep 2008 04:43:06 +0000
Subject: [PATCH] - added Threat module (since the standalone one on WoWI has
 been abandoned. this is simply the code for that addon + fixes from acapela
 for Threat-1.0 to Threat-2.0); disabled by default

---
 IceHUD.toc         |   1 +
 embeds.xml         |   2 +
 modules/Threat.lua | 234 +++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 237 insertions(+)
 create mode 100644 modules/Threat.lua

diff --git a/IceHUD.toc b/IceHUD.toc
index 6cef915..f0a61ae 100644
--- a/IceHUD.toc
+++ b/IceHUD.toc
@@ -51,3 +51,4 @@ modules\LacerateCount.lua
 modules\Runes.lua
 modules\TargetOfTargetHealth.lua
 modules\TargetOfTargetMana.lua
+modules\Threat.lua
diff --git a/embeds.xml b/embeds.xml
index 02b93f4..67ad7c0 100644
--- a/embeds.xml
+++ b/embeds.xml
@@ -15,5 +15,7 @@
 <Include file="libs\LibDruidMana-1.0\lib.xml"/>
 <Include file="libs\LibSharedMedia-3.0\lib.xml"/>
 <Script file="libs\Waterfall-1.0\Waterfall-1.0.lua"/>
+<Include file="libs\Threat-2.0\embeds.xml"/>
+<Include file="libs\Threat-2.0\Threat-2.0\lib.xml"/>
 
 </Ui>
diff --git a/modules/Threat.lua b/modules/Threat.lua
new file mode 100644
index 0000000..17b2cc9
--- /dev/null
+++ b/modules/Threat.lua
@@ -0,0 +1,234 @@
+--[[
+Name: IHUD_Threat
+Version: 1.2
+Author: Caryna/Turalyon EU (Alliance) (updated for Threat-2.0 by 'acapela' of WoWI and merged into IceHUD by Parnic)
+Description: adds a threat bar to IceHUD
+]]
+
+local AceOO = AceLibrary("AceOO-2.0")
+local Threat = LibStub("Threat-2.0")
+
+local IHUD_Threat = AceOO.Class(IceUnitBar)
+
+IHUD_Threat.prototype.color = nil
+IHUD_Threat.aggroBar = nil
+IHUD_Threat.aggroBarMulti = nil
+
+-- constructor
+function IHUD_Threat.prototype:init()
+	IHUD_Threat.super.prototype.init(self, "Threat", "target")
+
+	self:SetDefaultColor("ThreatLow", 102, 204, 51)
+	self:SetDefaultColor("ThreatMedium", 0, 204, 204)
+	self:SetDefaultColor("ThreatHigh", 204, 0, 153)
+	self:SetDefaultColor("ThreatDanger", 255, 0, 0)
+	self:SetDefaultColor("ThreatCustom", 255, 255, 0)
+
+	self:OnCoreLoad()
+end
+
+-- default settings
+function IHUD_Threat.prototype:GetDefaultSettings()
+	local settings = IHUD_Threat.super.prototype.GetDefaultSettings(self)
+	settings["side"] = IceCore.Side.Left
+	settings["offset"] = 3
+	settings["enabled"] = false
+	return settings
+end
+
+-- options stuff
+function IHUD_Threat.prototype:GetOptions()
+	local opts = IHUD_Threat.super.prototype.GetOptions(self)
+
+	opts["enabled"] = {
+		type = "toggle",
+		name = "|c" .. self.configColor .. "Enabled|r",
+		desc = "Enable/disable module (requires Threat-2.0 library)",
+		get = function()
+			return self.moduleSettings.enabled
+		end,
+		set = function(value)
+			self.moduleSettings.enabled = value
+			if (value) then
+				self:Enable(true)
+			else
+				self:Disable()
+			end
+		end,
+		disabled = function()
+			return Threat == nil
+		end,
+		order = 20
+	}
+
+	return opts
+end
+
+-- enable plugin
+function IHUD_Threat.prototype:Enable(core)
+	IHUD_Threat.super.prototype.Enable(self, core)
+
+	self:RegisterEvent("PLAYER_TARGET_CHANGED", "Update")
+	self:ScheduleRepeatingEvent(self.elementName, self.Update, 0.2, self)
+
+	self:Update(self.unit)
+end
+
+-- disable plugin
+function IHUD_Threat.prototype:Disable(core)
+	IHUD_Threat.super.prototype.Disable(self, core)
+
+	self:CancelScheduledEvent(self.elementName)
+end
+
+-- OVERRIDE
+function IHUD_Threat.prototype:CreateFrame()
+	IHUD_Threat.super.prototype.CreateFrame(self)
+	
+	self:CreateAggroBar()
+end
+
+-- create the aggro range indicator bar
+function IHUD_Threat.prototype:CreateAggroBar()
+	if not (self.aggroBar) then
+		self.aggroBar = CreateFrame("StatusBar", nil, self.frame)
+	end
+	
+	self.aggroBar:SetFrameStrata("BACKGROUND")
+	self.aggroBar:SetWidth(self.settings.barWidth)
+	self.aggroBar:SetHeight(self.settings.barHeight)
+	
+	if not (self.aggroBar.bar) then
+		self.aggroBar.bar = self.aggroBar:CreateTexture(nil, "BACKGROUND")
+	end
+	
+	self.aggroBar.bar:SetTexture(IceElement.TexturePath .. self.settings.barTexture .. "BG")
+	self.aggroBar.bar:SetAllPoints(self.aggroBar)
+	
+	self.aggroBar:SetStatusBarTexture(self.aggroBar.bar)
+	
+	local r, g, b = self.settings.backgroundColor.r, self.settings.backgroundColor.g, self.settings.backgroundColor.b
+	if (self.settings.backgroundToggle) then
+		r, g, b = self:GetColor("CastCasting")
+	end
+	self.aggroBar:SetStatusBarColor(r, g, b, self.moduleSettings.lagAlpha)
+	
+	if (self.moduleSettings.side == IceCore.Side.Left) then
+		self.aggroBar.bar:SetTexCoord(1, 0, 0, 0)
+	else
+		self.aggroBar.bar:SetTexCoord(0, 1, 0, 0)
+	end
+	
+	self.aggroBar:ClearAllPoints()
+	self.aggroBar:SetPoint("BOTTOM", self.frame, "BOTTOM", 0, 0)
+end
+
+-- bar stuff
+function IHUD_Threat.prototype:Update(unit)
+	-- only show bar if the user has Threat-2.0 and there's any threat
+	if not Threat or not Threat:IsActive() then
+		self:Show(false)
+		return
+	end
+
+	IHUD_Threat.super.prototype.Update(self)
+
+	if (unit and (unit ~= self.unit)) then
+		return
+	end
+
+	if not unit then
+		unit = self.unit
+	end
+	
+	if not UnitExists("target") or not UnitCanAttack("player", "target") or UnitIsDead("target") or UnitIsFriend("player", "target") or UnitPlayerControlled("target")
+	then
+		self:Show(false)
+		return
+	else
+		self:Show(true)
+	end
+
+	-- get my threat and the tank threat
+	local threatMe = Threat:GetThreat( UnitGUID("player") , UnitGUID("target") )
+	local threatTank = 0
+	local threatTankName = ""
+	
+	if UnitExists("targettarget") then
+		threatTankName = UnitGUID("targettarget")
+		threatTank = Threat:GetThreat( UnitGUID("targettarget"), UnitGUID("target") )
+	else
+		threatTank, threatTankName = Threat:GetMaxThreatOnTarget( UnitGUID("target") )
+	end
+
+	-- adjust max threat to avoid divide by 0
+	if ( threatTank == 0 ) then
+		threatTank = 1
+	end
+
+	-- aggro gain limit
+	local threatMulti = 1.3
+	if ( Threat:UnitInMeleeRange( "target" ) ) then
+		threatMulti = 1.1
+--	elseif ( UnitExists("targettarget") and ( UnitName("targettarget") == UnitName("player") ) ) then
+--		threatMulti = 1
+	elseif ( threatTankName == UnitGUID("player") ) then
+		threatMulti = 1
+	end
+	
+	-- get my threat percentage
+	local threatPct = self:MathRound( (100/threatTank) * threatMe, 1 )
+
+	IceHUD:Debug( "threatMe = " .. threatMe .. ", threatTank = " .. threatTank .. ", threatPct = " .. threatPct )
+	
+	-- set percentage text
+	self:SetBottomText1( threatPct .. "%" )
+	self:SetBottomText2()
+
+	-- set bar color
+	if( threatMulti == 1 ) then
+		self.color = "ThreatDanger"
+	elseif( threatPct < 50 ) then
+		self.color = "ThreatLow"
+	elseif ( threatPct < 80 ) then
+		self.color = "ThreatMedium"
+	else
+		self.color = "ThreatHigh"
+	end
+
+--[[	local g = floor( (255 / (threatTank*threatMulti) ) *threatMe )
+	
+	if(g>255) then g=255 end
+	if(g<0) then g=0 end
+	
+	IceHUD.IceCore:SetColor("ThreatCustom", 255, 255, 0)	
+	self.color = "ThreatCustom"
+]]	
+	-- set the bar value
+	self:UpdateBar( threatMe / ( threatTank*threatMulti ), self.color )
+	
+	-- do the aggro indicator bar stuff, but only if it has changed
+	if ( self.aggroBarMulti ~= threatMulti ) then
+		self.aggroBarMulti = threatMulti
+
+		local pos = 1 - (1 / threatMulti)
+		local y = self.settings.barHeight - ( pos * self.settings.barHeight )
+		
+		if ( self.moduleSettings.side == IceCore.Side.Left ) then
+			self.aggroBar.bar:SetTexCoord(1, 0, 0, pos)
+		else
+			self.aggroBar.bar:SetTexCoord(0, 1, 0, pos)
+		end
+		
+		self.aggroBar:SetPoint("BOTTOM", self.frame, "BOTTOM", 0, y)
+	end
+end
+
+-- rounding stuff
+function IHUD_Threat.prototype:MathRound(num, idp)
+	local mult = 10^(idp or 0)
+	return math.floor(num  * mult + 0.5) / mult
+end
+
+-- Load us up
+IceHUD.IHUD_Threat = IHUD_Threat:new()