WoWInterface SVN PowerAurasClassic

Compare Revisions

  • This comparison shows the changes necessary to convert path 
    /
    from Rev 9 to Rev 10
    ↔ Reverse comparison
  •

Rev 9 → Rev 10

trunk/PowerAuras/Libs/LibBanzai-2.0/LibBanzai-2.0.lua New file
0,0 → 1,247
--[[
Name: LibBanzai-2.0
Revision: $Revision: 72249 $
Author(s): Rabbit (rabbit.magtheridon@gmail.com), maia
Documentation: http://www.wowace.com/index.php/Banzai-2.0_API_Documentation
SVN: http://svn.wowace.com/wowace/trunk/BanzaiLib/Banzai-2.0
Description: Aggro notification library.
Dependencies: LibStub
]]
 
-------------------------------------------------------------------------------
-- Locals
-------------------------------------------------------------------------------
 
local MAJOR_VERSION = "LibBanzai-2.0"
local MINOR_VERSION = "$Revision: 72249 $"
 
if not LibStub then error(MAJOR_VERSION .. " requires LibStub.") end
local lib = LibStub:NewLibrary(MAJOR_VERSION, MINOR_VERSION)
if not lib then return end
 
lib.callbacks = lib.callbacks or {}
local callbacks = lib.callbacks
lib.frame = lib.frame or CreateFrame("Frame")
local frame = lib.frame
 
local _G = _G
local table_insert = table.insert
local UnitExists = _G.UnitExists
local UnitName = _G.UnitName
local UnitCanAttack = _G.UnitCanAttack
local GetNumRaidMembers = _G.GetNumRaidMembers
local GetNumPartyMembers = _G.GetNumPartyMembers
local unpack = _G.unpack
local type = _G.type
local assert = _G.assert
 
-------------------------------------------------------------------------------
-- Local heap
-------------------------------------------------------------------------------
 
local new, del
do
local cache = setmetatable({},{__mode='k'})
function new()
local t = next(cache)
if t then
cache[t] = nil
return t
else
return {}
end
end
function del(t)
for k in pairs(t) do
t[k] = nil
end
cache[t] = true
return nil
end
end
 
-------------------------------------------------------------------------------
-- Roster
-------------------------------------------------------------------------------
 
local raidUnits = setmetatable({}, {__index =
function(self, key)
self[key] = ("raid%d"):format(key)
return self[key]
end
})
local raidPetUnits = setmetatable({}, {__index =
function(self, key)
self[key] = ("raidpet%d"):format(key)
return self[key]
end
})
local partyUnits = {"party1","party2","party3","party4"}
local partyPetUnits = {"partypet1","partypet2","partypet3","partypet4"}
local roster = {}
local needsUpdate = nil
 
-- If some pet has the same name as a person in the raid, they'll end up being
-- the same unit for the purposes of banzai, but we won't care right now.
local function addUnit(unit)
if not UnitExists(unit) then return end
local name = UnitName(unit)
if not roster[name] then roster[name] = new() end
table_insert(roster[name], unit)
end
 
local function actuallyUpdateRoster()
for k in pairs(roster) do roster[k] = del(roster[k]) end
addUnit("player")
addUnit("pet")
addUnit("focus")
for i = 1, GetNumRaidMembers() do
addUnit(raidUnits[i])
addUnit(raidPetUnits[i])
end
for i = 1, GetNumPartyMembers() do
addUnit(partyUnits[i])
addUnit(partyPetUnits[i])
end
needsUpdate = nil
end
 
local function updateRoster()
needsUpdate = true
end
 
-------------------------------------------------------------------------------
-- Banzai
-------------------------------------------------------------------------------
 
local targets = setmetatable({}, {__index =
function(self, key)
self[key] = key .. "target"
return self[key]
end
})
 
local aggro = {}
local banzai = {}
 
local total = 0
local function updateBanzai(_, elapsed)
total = total + elapsed
if total > 0.2 then
if needsUpdate then actuallyUpdateRoster() end
for name, units in pairs(roster) do
local unit = units[1]
local targetId = targets[unit]
if UnitExists(targetId) then
local ttId = targets[targetId]
if unit == "focus" and UnitIsEnemy("focus", "player") then
ttId = "focustarget"
targetId = "focus"
end
if UnitExists(ttId) and UnitCanAttack(ttId, targetId) then
for n, u in pairs(roster) do
if UnitIsUnit(u[1], ttId) then
banzai[n] = (banzai[n] or 0) + 10
break
end
end
end
end
if banzai[name] then
if banzai[name] >= 5 then banzai[name] = banzai[name] - 5 end
if banzai[name] > 25 then banzai[name] = 25 end
end
end
for name, units in pairs(roster) do
if banzai[name] and banzai[name] > 15 then
if not aggro[name] then
aggro[name] = true
for i, v in ipairs(callbacks) do
v(1, name, unpack(units))
end
end
elseif aggro[name] then
aggro[name] = nil
for i, v in ipairs(callbacks) do
v(0, name, unpack(units))
end
end
end
total = 0
end
end
 
-------------------------------------------------------------------------------
-- Starting and stopping
-------------------------------------------------------------------------------
 
local running = nil
local function start()
if running then return end
updateRoster()
frame:SetScript("OnUpdate", updateBanzai)
frame:SetScript("OnEvent", updateRoster)
frame:RegisterEvent("RAID_ROSTER_UPDATE")
frame:RegisterEvent("PARTY_MEMBERS_CHANGED")
frame:RegisterEvent("UNIT_PET")
frame:RegisterEvent("PLAYER_FOCUS_CHANGED")
running = true
end
 
local function stop()
if not running then return end
frame:SetScript("OnUpdate", nil)
frame:SetScript("OnEvent", nil)
frame:UnregisterAllEvents()
running = nil
end
 
-------------------------------------------------------------------------------
-- API
-------------------------------------------------------------------------------
 
function lib:IsRunning() return running end
function lib:GetUnitAggroByUnitName(name) return aggro[name] end
function lib:GetUnitAggroByUnitId(unit)
if not UnitExists(unit) then return end
return aggro[UnitName(unit)]
end
 
function lib:RegisterCallback(func)
if type(func) ~= "function" then
error(("Bad argument to :RegisterCallback, function expected, got %q."):format(type(func)), 2)
end
 
table_insert(callbacks, func)
start()
end
 
function lib:UnregisterCallback(func)
if type(func) ~= "function" then
error(("Bad argument to :UnregisterCallback, function expected, got %q."):format(type(func)), 2)
end
 
local found = nil
for i, v in ipairs(callbacks) do
if v == func then
table.remove(callbacks, i)
found = true
break
end
end
if #callbacks == 0 then stop() end
 
if not found then
error("Bad argument to :UnregisterCallback, the provided function was not registered.", 2)
end
end
 
-------------------------------------------------------------------------------
-- Initialization
-------------------------------------------------------------------------------
 
frame:SetScript("OnUpdate", nil)
frame:SetScript("OnEvent", nil)
frame:UnregisterAllEvents()
if #callbacks > 0 then start() end
 
trunk/PowerAuras/Libs/LibBanzai-2.0/LibBanzai-2.0.toc New file
0,0 → 1,14
## Interface: 30200
## LoadOnDemand: 1
## Title: Lib: Banzai-2.0
## Notes: Aggro notification library.
## Author: Rabbit
## X-Website: http://www.wowace.com/wiki/BanzaiLib
## X-Email: rabbit.magtheridon@gmail.com
## X-Category: Library
## X-Credits: maia
## Version: 2.0
## X-AceLibrary-Banzai-2.0: true
 
lib.xml
 
trunk/PowerAuras/Libs/LibBanzai-2.0/lib.xml New file
0,0 → 1,5
<Ui xmlns="http://www.blizzard.com/wow/ui/" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.blizzard.com/wow/ui/
..\FrameXML\UI.xsd">
<Script file="LibStub\LibStub.lua" />
<Script file="LibBanzai-2.0.lua" />
</Ui>