initial commit v.2.1.0
This commit is contained in:
Torsten Brendgen
404
Modules/Tracker/TrackerCore.lua
Normal file
404
Modules/Tracker/TrackerCore.lua
Normal file
@@ -0,0 +1,404 @@
|
||||
local ADDON_NAME = "HailMaryGuildTools"
|
||||
local HMGT = LibStub("AceAddon-3.0"):GetAddon(ADDON_NAME)
|
||||
if not HMGT then return end
|
||||
|
||||
HMGT.TrackerCore = HMGT.TrackerCore or {}
|
||||
|
||||
local function EntryNeedsVisualTicker(entry)
|
||||
if type(entry) ~= "table" then
|
||||
return false
|
||||
end
|
||||
|
||||
local remaining = tonumber(entry.remaining) or 0
|
||||
if remaining > 0 then
|
||||
return true
|
||||
end
|
||||
|
||||
local maxCharges = tonumber(entry.maxCharges) or 0
|
||||
local currentCharges = tonumber(entry.currentCharges)
|
||||
if maxCharges > 0 and currentCharges ~= nil and currentCharges < maxCharges then
|
||||
return true
|
||||
end
|
||||
|
||||
return false
|
||||
end
|
||||
|
||||
function HMGT:IsGroupTrackerConfig(tracker)
|
||||
return type(tracker) == "table" and tracker.trackerType == "group"
|
||||
end
|
||||
|
||||
function HMGT:GetTrackerSpellPool(categories)
|
||||
if HMGT_SpellData and type(HMGT_SpellData.GetSpellPoolForCategories) == "function" then
|
||||
return HMGT_SpellData.GetSpellPoolForCategories(categories)
|
||||
end
|
||||
return {}
|
||||
end
|
||||
|
||||
function HMGT:GetTrackerSpellsForPlayer(classToken, specIndex, categories)
|
||||
if HMGT_SpellData and type(HMGT_SpellData.GetSpellsForCategories) == "function" then
|
||||
return HMGT_SpellData.GetSpellsForCategories(classToken, specIndex, categories)
|
||||
end
|
||||
return {}
|
||||
end
|
||||
|
||||
function HMGT:GetTrackerPlayers(tracker)
|
||||
local players = {}
|
||||
|
||||
local ownName = self:NormalizePlayerName(UnitName("player"))
|
||||
local ownClass = select(2, UnitClass("player"))
|
||||
local includeOwnPlayer = true
|
||||
if self:IsGroupTrackerConfig(tracker) then
|
||||
includeOwnPlayer = tracker.includeSelfFrame == true
|
||||
end
|
||||
if includeOwnPlayer then
|
||||
players[#players + 1] = {
|
||||
name = ownName,
|
||||
class = ownClass,
|
||||
isOwn = true,
|
||||
unitId = "player",
|
||||
}
|
||||
end
|
||||
|
||||
if IsInRaid() then
|
||||
for i = 1, GetNumGroupMembers() do
|
||||
local unitId = "raid" .. i
|
||||
local name = self:NormalizePlayerName(UnitName(unitId))
|
||||
local class = select(2, UnitClass(unitId))
|
||||
if name and name ~= ownName then
|
||||
players[#players + 1] = {
|
||||
name = name,
|
||||
class = class,
|
||||
unitId = unitId,
|
||||
}
|
||||
end
|
||||
end
|
||||
elseif IsInGroup() then
|
||||
for i = 1, GetNumGroupMembers() - 1 do
|
||||
local unitId = "party" .. i
|
||||
local name = self:NormalizePlayerName(UnitName(unitId))
|
||||
local class = select(2, UnitClass(unitId))
|
||||
if name and name ~= ownName then
|
||||
players[#players + 1] = {
|
||||
name = name,
|
||||
class = class,
|
||||
unitId = unitId,
|
||||
}
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return players
|
||||
end
|
||||
|
||||
function HMGT:CollectTrackerEntriesForPlayer(tracker, playerInfo)
|
||||
local entries = {}
|
||||
if type(tracker) ~= "table" or type(playerInfo) ~= "table" then
|
||||
return entries
|
||||
end
|
||||
|
||||
local playerName = playerInfo.name
|
||||
if not playerName then
|
||||
return entries
|
||||
end
|
||||
|
||||
local pData = self.playerData[playerName]
|
||||
local classToken = pData and pData.class or playerInfo.class
|
||||
if not classToken then
|
||||
return entries
|
||||
end
|
||||
|
||||
local specIndex
|
||||
if playerInfo.isOwn then
|
||||
specIndex = GetSpecialization()
|
||||
if not specIndex or specIndex == 0 then
|
||||
return entries
|
||||
end
|
||||
else
|
||||
specIndex = pData and pData.specIndex or nil
|
||||
if not specIndex or tonumber(specIndex) <= 0 then
|
||||
return entries
|
||||
end
|
||||
end
|
||||
|
||||
local talents = pData and pData.talents or {}
|
||||
local spells = self:GetTrackerSpellsForPlayer(classToken, specIndex, tracker.categories)
|
||||
for _, spellEntry in ipairs(spells) do
|
||||
if tracker.enabledSpells[spellEntry.spellId] ~= false then
|
||||
local remaining, total, currentCharges, maxCharges = self:GetCooldownInfo(playerName, spellEntry.spellId)
|
||||
local effectiveCd = HMGT_SpellData.GetEffectiveCooldown(spellEntry, talents)
|
||||
local isAvailabilitySpell = self:IsAvailabilitySpell(spellEntry)
|
||||
local include = self:ShouldDisplayEntry(tracker, remaining, currentCharges, maxCharges, spellEntry)
|
||||
local spellKnown = self:IsTrackedSpellKnownForPlayer(playerName, spellEntry.spellId)
|
||||
local hasPartialCharges = (tonumber(maxCharges) or 0) > 0
|
||||
and (tonumber(currentCharges) or tonumber(maxCharges) or 0) < (tonumber(maxCharges) or 0)
|
||||
local hasActiveCd = ((remaining or 0) > 0) or hasPartialCharges
|
||||
|
||||
if not spellKnown and not hasActiveCd then
|
||||
include = false
|
||||
end
|
||||
if isAvailabilitySpell and not spellKnown then
|
||||
include = false
|
||||
end
|
||||
if not playerInfo.isOwn and isAvailabilitySpell and not self:HasAvailabilityState(playerName, spellEntry.spellId) then
|
||||
include = false
|
||||
end
|
||||
|
||||
if include then
|
||||
entries[#entries + 1] = {
|
||||
playerName = playerName,
|
||||
class = classToken,
|
||||
spellEntry = spellEntry,
|
||||
remaining = remaining,
|
||||
total = total > 0 and total or effectiveCd,
|
||||
currentCharges = currentCharges,
|
||||
maxCharges = maxCharges,
|
||||
}
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
return entries
|
||||
end
|
||||
|
||||
local function CopyEntriesForPreview(entries, playerName)
|
||||
local copies = {}
|
||||
for _, entry in ipairs(entries or {}) do
|
||||
local nextEntry = {}
|
||||
for key, value in pairs(entry) do
|
||||
nextEntry[key] = value
|
||||
end
|
||||
nextEntry.playerName = playerName
|
||||
copies[#copies + 1] = nextEntry
|
||||
end
|
||||
return copies
|
||||
end
|
||||
|
||||
function HMGT:BuildPartyPreviewEntries(entries, resolveUnitAnchorFrame)
|
||||
local byPlayer = {}
|
||||
local order = {}
|
||||
local unitByPlayer = {}
|
||||
for index = 1, 4 do
|
||||
local unitId = "party" .. index
|
||||
if not resolveUnitAnchorFrame or resolveUnitAnchorFrame(unitId) then
|
||||
local playerName = string.format("Party %d", index)
|
||||
local playerEntries = CopyEntriesForPreview(entries, playerName)
|
||||
if #playerEntries > 0 then
|
||||
byPlayer[playerName] = playerEntries
|
||||
order[#order + 1] = playerName
|
||||
unitByPlayer[playerName] = unitId
|
||||
end
|
||||
end
|
||||
end
|
||||
return byPlayer, order, unitByPlayer, #order > 0
|
||||
end
|
||||
|
||||
function HMGT:CollectTrackerEntries(tracker)
|
||||
local entries = {}
|
||||
local players = self:GetTrackerPlayers(tracker)
|
||||
for _, playerInfo in ipairs(players) do
|
||||
local playerEntries = self:CollectTrackerEntriesForPlayer(tracker, playerInfo)
|
||||
for _, entry in ipairs(playerEntries) do
|
||||
entries[#entries + 1] = entry
|
||||
end
|
||||
end
|
||||
return entries
|
||||
end
|
||||
|
||||
function HMGT:CollectTrackerTestEntries(tracker)
|
||||
local playerName = self:NormalizePlayerName(UnitName("player")) or "Player"
|
||||
local classToken = select(2, UnitClass("player"))
|
||||
if not classToken then
|
||||
return {}
|
||||
end
|
||||
|
||||
local entries = {}
|
||||
local pData = self.playerData[playerName]
|
||||
local talents = pData and pData.talents or {}
|
||||
local spells = self:GetTrackerSpellsForPlayer(classToken, GetSpecialization() or 0, tracker.categories)
|
||||
for _, spellEntry in ipairs(spells) do
|
||||
if tracker.enabledSpells[spellEntry.spellId] ~= false then
|
||||
local remaining, total, currentCharges, maxCharges = self:GetCooldownInfo(playerName, spellEntry.spellId)
|
||||
local effectiveCd = HMGT_SpellData.GetEffectiveCooldown(spellEntry, talents)
|
||||
local isAvailabilitySpell = self:IsAvailabilitySpell(spellEntry)
|
||||
local spellKnown = self:IsTrackedSpellKnownForPlayer(playerName, spellEntry.spellId)
|
||||
local hasPartialCharges = (tonumber(maxCharges) or 0) > 0
|
||||
and (tonumber(currentCharges) or tonumber(maxCharges) or 0) < (tonumber(maxCharges) or 0)
|
||||
local hasActiveCd = ((remaining or 0) > 0) or hasPartialCharges
|
||||
local hasAvailabilityState = isAvailabilitySpell and self:HasAvailabilityState(playerName, spellEntry.spellId)
|
||||
if spellKnown or hasActiveCd or hasAvailabilityState then
|
||||
entries[#entries + 1] = {
|
||||
playerName = playerName,
|
||||
class = classToken,
|
||||
spellEntry = spellEntry,
|
||||
remaining = remaining,
|
||||
total = total > 0 and total or effectiveCd,
|
||||
currentCharges = currentCharges,
|
||||
maxCharges = maxCharges,
|
||||
}
|
||||
end
|
||||
end
|
||||
end
|
||||
return entries
|
||||
end
|
||||
|
||||
function HMGT:BuildEntriesForTracker(tracker, trackerKey)
|
||||
local key = trackerKey or tostring(tonumber(tracker and tracker.id) or 0)
|
||||
if tracker and tracker.testMode then
|
||||
return self:CollectTrackerTestEntries(tracker), true
|
||||
end
|
||||
if tracker and tracker.demoMode then
|
||||
return self:GetDemoEntries(key, self:GetTrackerSpellPool(tracker.categories), tracker), true
|
||||
end
|
||||
if not tracker or not tracker.enabled or not self:IsVisibleForCurrentGroup(tracker) then
|
||||
return {}, false
|
||||
end
|
||||
return self:CollectTrackerEntries(tracker), true
|
||||
end
|
||||
|
||||
function HMGT:BuildEntriesByPlayerForTracker(tracker, trackerKey, resolveUnitAnchorFrame)
|
||||
local key = trackerKey or tostring(tonumber(tracker and tracker.id) or 0)
|
||||
local ownName = self:NormalizePlayerName(UnitName("player")) or "Player"
|
||||
if tracker.testMode then
|
||||
local entries = self:CollectTrackerTestEntries(tracker)
|
||||
if self:IsGroupTrackerConfig(tracker) and tracker.attachToPartyFrame == true then
|
||||
return self:BuildPartyPreviewEntries(entries, resolveUnitAnchorFrame)
|
||||
end
|
||||
local byPlayer, order, unitByPlayer = {}, {}, {}
|
||||
if #entries > 0 then
|
||||
byPlayer[ownName] = entries
|
||||
order[1] = ownName
|
||||
unitByPlayer[ownName] = "player"
|
||||
end
|
||||
return byPlayer, order, unitByPlayer, true
|
||||
end
|
||||
if tracker.demoMode then
|
||||
local entries = self:GetDemoEntries(key, self:GetTrackerSpellPool(tracker.categories), tracker)
|
||||
if self:IsGroupTrackerConfig(tracker) and tracker.attachToPartyFrame == true then
|
||||
return self:BuildPartyPreviewEntries(entries, resolveUnitAnchorFrame)
|
||||
end
|
||||
for _, entry in ipairs(entries) do
|
||||
entry.playerName = ownName
|
||||
end
|
||||
local byPlayer, order, unitByPlayer = {}, {}, {}
|
||||
if #entries > 0 then
|
||||
byPlayer[ownName] = entries
|
||||
order[1] = ownName
|
||||
unitByPlayer[ownName] = "player"
|
||||
end
|
||||
return byPlayer, order, unitByPlayer, true
|
||||
end
|
||||
if not tracker.enabled or not self:IsVisibleForCurrentGroup(tracker) then
|
||||
return {}, {}, {}, false
|
||||
end
|
||||
if IsInRaid() or not IsInGroup() then
|
||||
return {}, {}, {}, false
|
||||
end
|
||||
local byPlayer, order, unitByPlayer = {}, {}, {}
|
||||
for _, playerInfo in ipairs(self:GetTrackerPlayers(tracker)) do
|
||||
local entries = self:CollectTrackerEntriesForPlayer(tracker, playerInfo)
|
||||
if #entries > 0 then
|
||||
local playerName = playerInfo.name
|
||||
byPlayer[playerName] = entries
|
||||
order[#order + 1] = playerName
|
||||
unitByPlayer[playerName] = playerInfo.unitId
|
||||
end
|
||||
end
|
||||
return byPlayer, order, unitByPlayer, true
|
||||
end
|
||||
|
||||
function HMGT:FinalizeTrackerEntries(tracker, entries, trackerKey)
|
||||
local result = entries or {}
|
||||
if self.FilterDisplayEntries then
|
||||
result = self:FilterDisplayEntries(tracker, result) or result
|
||||
end
|
||||
if self.SortDisplayEntries then
|
||||
self:SortDisplayEntries(result, trackerKey)
|
||||
end
|
||||
|
||||
local shouldTick = false
|
||||
for _, entry in ipairs(result) do
|
||||
if EntryNeedsVisualTicker(entry) then
|
||||
shouldTick = true
|
||||
break
|
||||
end
|
||||
end
|
||||
|
||||
return result, shouldTick
|
||||
end
|
||||
|
||||
function HMGT:TriggerTrackerUpdate(reason)
|
||||
local function normalizeReason(value)
|
||||
if value == true then
|
||||
return "trackers"
|
||||
elseif value == "trackers" or value == "layout" or value == "visual" then
|
||||
return value
|
||||
end
|
||||
return "full"
|
||||
end
|
||||
|
||||
local function mergeReasons(current, incoming)
|
||||
local priority = {
|
||||
visual = 1,
|
||||
layout = 2,
|
||||
trackers = 3,
|
||||
full = 4,
|
||||
}
|
||||
current = normalizeReason(current)
|
||||
incoming = normalizeReason(incoming)
|
||||
if (priority[incoming] or 4) >= (priority[current] or 4) then
|
||||
return incoming
|
||||
end
|
||||
return current
|
||||
end
|
||||
|
||||
self._trackerUpdateMinDelay = self._trackerUpdateMinDelay or 0.08
|
||||
self._trackerUpdatePending = true
|
||||
self._trackerUpdateReason = mergeReasons(self._trackerUpdateReason, reason)
|
||||
if HMGT.TrackerManager then
|
||||
local normalizedReason = normalizeReason(reason)
|
||||
if normalizedReason == "trackers" then
|
||||
HMGT.TrackerManager:MarkTrackersDirty()
|
||||
elseif normalizedReason == "layout" then
|
||||
HMGT.TrackerManager:MarkLayoutDirty()
|
||||
end
|
||||
end
|
||||
if self._updateScheduled then return end
|
||||
|
||||
local now = GetTime()
|
||||
local last = self._lastTrackerUpdateAt or 0
|
||||
local delay = math.max(0, self._trackerUpdateMinDelay - (now - last))
|
||||
self._updateScheduled = true
|
||||
|
||||
self:ScheduleTimer(function()
|
||||
self._updateScheduled = nil
|
||||
if not self._trackerUpdatePending then return end
|
||||
self._trackerUpdatePending = nil
|
||||
self._lastTrackerUpdateAt = GetTime()
|
||||
local pendingReason = self._trackerUpdateReason
|
||||
self._trackerUpdateReason = nil
|
||||
|
||||
local function profileModule(name, fn)
|
||||
if not fn then return end
|
||||
local t0 = debugprofilestop and debugprofilestop() or nil
|
||||
fn()
|
||||
local t1 = debugprofilestop and debugprofilestop() or nil
|
||||
if t0 and t1 then
|
||||
local mod = HMGT[name]
|
||||
local count = mod and mod.lastEntryCount or 0
|
||||
self:Debug("verbose", "UIUpdate %s took %.2fms entries=%s", tostring(name), t1 - t0, tostring(count))
|
||||
end
|
||||
end
|
||||
|
||||
profileModule("TrackerManager", HMGT.TrackerManager and function()
|
||||
if pendingReason == "visual" and HMGT.TrackerManager.RefreshVisibleVisuals then
|
||||
HMGT.TrackerManager:RefreshVisibleVisuals()
|
||||
else
|
||||
HMGT.TrackerManager:UpdateDisplay()
|
||||
end
|
||||
end or nil)
|
||||
|
||||
if self._trackerUpdatePending then
|
||||
self:TriggerTrackerUpdate()
|
||||
end
|
||||
end, delay)
|
||||
end
|
||||
Reference in New Issue
Block a user