X-Git-Url: https://www.aleksib.fi/git/wowui.git/blobdiff_plain/c6843f31e3e56b92d55a84fbd4da40b555809d31..c809e5da99427a93b5ad5eb010a1cb1cba507ec2:/OmaRF/Indicators.lua

diff --git a/OmaRF/Indicators.lua b/OmaRF/Indicators.lua
index 9557258..d37383a 100644
--- a/OmaRF/Indicators.lua
+++ b/OmaRF/Indicators.lua
@@ -1,232 +1,175 @@
-local f = OmaRF.frames;
-local majorFrames = OmaRF.majorFrames;
-local positions = OmaRF.positions;
-local pad = 2;
-local paddings = {
-    TOPLEFT = {pad, -pad},
-    TOPRIGHT = {-pad, -pad},
-    CENTER = {0, 0},
-    BOTTOMLEFT = {pad, pad},
-    BOTTOMRIGHT = {-pad, pad}
-};
-local watchedAuras;
-local majorAuras;
-local majorMax;
-local auraFilters = {"HELPFUL", "HARMFUL"};
-local DEFAULT_ICON = "Interface\\AddOns\\OmaRF\\images\\rhomb";
-local _;
-
--- global functions used every update
-local C_TimerAfter = C_Timer.After;
+-- Indicators.lua
+local pairs, ipairs = pairs, ipairs;
+local floor = math.floor;
 local GetTime = GetTime;
 local UnitAura = UnitAura;
-local UnitIsPlayer = UnitIsPlayer;
-local UnitIsConnected = UnitIsConnected;
-local UnitIsDeadOrGhost = UnitIsDeadOrGhost;
-local CompactRaidFrameContainer_ApplyToFrames = CompactRaidFrameContainer_ApplyToFrames;
-local format = string.format;
-local unpack = unpack;
-local floor = floor;
-local ceil = ceil;
-
--- update current auras
-hooksecurefunc("CompactUnitFrame_UpdateAuras", function(frame)
-    local frameName = frame:GetName();
-    if f[frameName] then
-        for _, ind in pairs(f[frameName]) do ind.expires = nil end
-        for _, ind in ipairs(majorFrames[frameName]) do ind.expires = nil end
-
-        local name, icon, count, expires, caster, id;
-        local unit = frame.displayedUnit;
-        local majorI = 1;
-        for _, filter in ipairs(auraFilters) do
-            local i = 1;
-            while true do
-                name, _, icon, count, _, _, expires, caster, _, _, id = UnitAura(unit, i, filter);
-                if not id then break end
-                local pos = watchedAuras[id] or watchedAuras[name];
-                if pos then
-                    local ind = f[frameName][pos];
-                    local config = OmaRF.db.profile.indicators[pos];
-                    if not config.mine or UnitIsPlayer(caster) then
-                        if config.useDefaultIcon then
-                            ind.icon:SetTexture(DEFAULT_ICON);
-                        else
-                            ind.icon:SetTexture(icon);
-                        end
-                        ind.expires = expires;
-                    end
-                end
-
-                if (majorAuras[id] or majorAuras[name]) and majorI <= majorMax then
-                    local ind = majorFrames[frameName][majorI];
-                    ind.icon:SetTexture(icon);
-                    ind.expires = expires;
-                    if count > 1 then
-                        ind.stackText:SetText(count);
-                    end
-                    majorI = majorI + 1;
-                end
-                i = i + 1;
-            end
-        end
+local CreateFrame = CreateFrame;
+local UnitIsDeadOrGhost, UnitIsConnected = UnitIsDeadOrGhost, UnitIsConnected;
+local CTimerAfter = C_Timer.After;
+
+local Settings = OmaRFSettings;
+local majorAuras = Settings.MajorAuras;
+local watchedAuras = {};
+
+local updaters = {};
+local updating = {};
+
+local M = {};
+OmaRFIndicators = M;
+M.Class = {};
+
+function M.SetupIndicators(frame, class)
+    frame.indBase = CreateFrame("Frame", nil, frame);
+    frame.indBase:SetAllPoints();
+    frame.indBase:Hide();
+    if M.Class[class] then
+        watchedAuras = M.Class[class].Auras;
+        frame.inds = M.Class[class].Setup(frame.indBase);
+    else
+        frame.inds = {};
     end
-end);
-
-local function configureIndicators(frame, name)
-    local frameName = name or frame:GetName();
-    if not f[frameName] then return end
 
-    local config = OmaRF.db.profile.indicators;
-    for pos, ind in pairs(f[frameName]) do
-        ind.text:SetFont(STANDARD_TEXT_FONT, config[pos]["textSize"]);
-        ind.text:SetTextColor(unpack(config[pos]["textColor"]));
-        ind.icon:SetWidth(config[pos]["iconSize"]);
-        ind.icon:SetHeight(config[pos]["iconSize"]);
-        ind.icon:SetTexture(DEFAULT_ICON);
-        ind.icon:SetVertexColor(unpack(config[pos]["iconColor"]));
+    frame.majorBase = CreateFrame("Frame", nil, frame);
+    frame.majorBase:SetPoint("TOPLEFT", frame, "TOPLEFT", 4, -10);
+    frame.majorBase:SetPoint("BOTTOMRIGHT");
+    frame.majors = {};
+    for i = 1,3 do
+        local tex = frame.majorBase:CreateTexture(nil, "OVERLAY");
+        tex = frame.majorBase:CreateTexture(nil, "OVERLAY");
+        if i == 1 then tex:SetPoint("TOPLEFT", frame.majorBase, "TOPLEFT");
+        else tex:SetPoint("TOPLEFT", frame.majors[i-1], "TOPRIGHT"); end
+        tex:SetWidth(20);
+        tex:SetHeight(20);
+        tex:Hide();
+        tex.text = frame.majorBase:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
+        tex.text:SetPoint("CENTER", tex, "BOTTOMRIGHT", -2, 2);
+        tex.text:Hide();
+        tex.stack = frame.majorBase:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
+        tex.stack:SetPoint("CENTER", tex, "TOPLEFT", 1, 0);
+        tex.stack:Hide();
+        tex.icon = true;
+        frame.majors[i] = tex;
     end
+end
 
-    config = OmaRF.db.profile.majorAuras;
-    for i, ind in ipairs(majorFrames[frameName]) do
-        if i == 1 then
-            ind.icon:ClearAllPoints();
-            ind.icon:SetPoint("CENTER", frame, "CENTER", -config.iconSize, 0);
-        end
-        ind.icon:SetWidth(config.iconSize);
-        ind.icon:SetHeight(config.iconSize);
-        ind.expireText:SetFont(STANDARD_TEXT_FONT, config["textSize"], "OUTLINE");
-        ind.stackText:SetFont(STANDARD_TEXT_FONT, config["textSize"], "OUTLINE");
+local function remaining(text, expires, current)
+    if expires == 0 then
+        text:SetText("");
+        return false;
+    end
+    local remain = expires - current;
+    if remain > 8 then
+        text:SetText("");
+    else
+        text:SetText(floor(remain+0.5));
     end
+    return true;
 end
 
--- Create the FontStrings used for indicators
-local function setupCompactUnitFrame(frame, name)
-    f[name] = {};
-    for _, pos in ipairs(positions) do
-        f[name][pos] = {};
-        f[name][pos].text = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall");
-        f[name][pos].text:SetPoint(pos, frame, pos, paddings[pos][1], paddings[pos][2]);
-        f[name][pos].icon = frame:CreateTexture(nil, "OVERLAY");
-        f[name][pos].icon:SetPoint(pos, frame, pos, paddings[pos][1], paddings[pos][2]);
+local function updateIndicators(frame)
+    local unit = frame.displayed;
+    if not frame:IsShown() or not UnitIsConnected(unit) or UnitIsDeadOrGhost(unit) then
+        updating[frame] = nil;
+        return;
     end
 
-    local config = OmaRF.db.profile.majorAuras;
-    majorFrames[name] = {};
-    for i = 1, config.max do
-        majorFrames[name][i] = {};
-        majorFrames[name][i].icon = frame:CreateTexture(nil, "OVERLAY");
-        if i > 1 then
-            majorFrames[name][i].icon:SetPoint("TOPLEFT", majorFrames[name][i-1].icon, "TOPRIGHT");
+    local needUpdate = false;
+    local current = GetTime();
+    for _, ind in pairs(frame.inds) do
+        if ind.text and ind.text.expires ~= nil then
+            needUpdate = remaining(ind.text, ind.text.expires, current) or needUpdate;
         end
-        majorFrames[name][i].expireText = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall");
-        majorFrames[name][i].expireText:SetPoint("BOTTOMRIGHT", majorFrames[name][i].icon, "BOTTOMRIGHT");
-        majorFrames[name][i].stackText = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall");
-        majorFrames[name][i].stackText:SetPoint("TOPLEFT", majorFrames[name][i].icon, "TOPLEFT");
     end
-
-    configureIndicators(frame, name);
+    for _, ind in pairs(frame.majors) do
+        if ind.text and ind.text.expires ~= nil then
+            needUpdate = remaining(ind.text, ind.text.expires, current) or needUpdate;
+        end
+    end
+    if needUpdate then
+        CTimerAfter(0.20, updaters[frame]);
+    else
+        updating[frame] = nil;
+    end
 end
 
-local function remaining(expires, current)
-    if expires == 0 then return "" end
-    local remain = expires - current;
-    if remain > 60 then
-        return format("%dm", ceil(remain/60));
+local function showInd(ind, expires, current, count, icon)
+    local needUpdate = false;
+    if ind.icon then
+        ind:SetTexture(icon);
+    end
+    if ind.text then
+        needUpdate = remaining(ind.text, expires, current);
+        ind.text.expires = expires;
+        ind.text:Show();
     end
-    return floor(remain+0.5);
+    if ind.stack and count > 1 then
+        ind.stack:SetText(count);
+        ind.stack:Show();
+    end
+    ind:Show();
+    return needUpdate;
 end
 
-local function hide(frameName)
-    for _, ind in pairs(f[frameName]) do
+local function hideInd(ind)
+    if ind.text then
+        ind.text.expires = nil;
         ind.text:Hide();
-        ind.icon:Hide();
-    end
-    for _, ind in ipairs(majorFrames[frameName]) do
-        ind.icon:Hide();
-        ind.expireText:Hide();
-        ind.stackText:Hide();
     end
+    if ind.stack then ind.stack:Hide() end
+    ind:Hide();
 end
 
--- Check the indicators on a frame and update the times on them
-local function updateIndicators(frame)
-    local frameName = frame:GetName();
-    local unit = frame.unit;
-    if not unit then return end -- possible if the frame is just being hidden
-
-    -- Create indicators if needed, out of combat if forbidden
-    if not f[frameName] then
-        if frame:IsForbidden() then
-            if not OmaRF.ooc_queue[frameName] then
-                OmaRF.ooc_queue[frameName] = {
-                    func = setupCompactUnitFrame,
-                    args = { frame, frameName }
-                };
-            end
-            return;
-        else
-            setupCompactUnitFrame(frame, frameName);
-        end
+function M.CheckIndicators(frame, unit)
+    for _, ind in pairs(frame.inds) do
+        hideInd(ind);
     end
-    -- Reset current
-    hide(frameName);
-    -- Hide if unit is dead/disconnected
-    if (not UnitIsConnected(unit)) or UnitIsDeadOrGhost(frame.displayedUnit) then
-        return;
+    for _, ind in pairs(frame.majors) do
+        hideInd(ind);
     end
-
+    local name, icon, count, expires, caster, id;
+    local showInds, showMajors, needUpdate = false, false, false;
+    local majorPos = 1;
+    local alert = false; -- color the whole bar
     local current = GetTime();
-    for pos, ind in pairs(f[frameName]) do
-        if ind.expires ~= nil then
-            local config = OmaRF.db.profile.indicators[pos];
-            if config.showIcon then
-                ind.icon:Show();
-            end
-            if config.showText then
-                ind.text:SetText(remaining(ind.expires, current));
-                ind.text:Show();
-            end
+    for spell, pos in pairs(watchedAuras) do
+        name, _, icon, count, _, _, expires = UnitAura(unit, spell, nil, "PLAYER HELPFUL");
+        if name then
+            needUpdate = showInd(frame.inds[pos], expires, current, count, icon) or needUpdate;
+            showInds = true;
         end
     end
-    for _, ind in ipairs(majorFrames[frameName]) do
-        if ind.expires ~= nil then
-            ind.icon:Show();
-            ind.expireText:SetText(remaining(ind.expires, current));
-            ind.expireText:Show();
-            ind.stackText:Show();
-        end
-    end
-end
 
--- Update all indicators
-function OmaRF:UpdateAllIndicators()
-    CompactRaidFrameContainer_ApplyToFrames(CompactRaidFrameContainer, "normal", updateIndicators);
-    if OmaRF.running then
-        C_TimerAfter(0.15, OmaRF.UpdateAllIndicators);
+    local i = 1;
+    while true do
+        name, _, icon, count, _, _, expires, caster, _, _, id = UnitAura(unit, i, "HARMFUL");
+        if not id or majorPos > 3 then break end
+        local major = majorAuras[id] or majorAuras[name];
+        if major then
+            needUpdate = showInd(frame.majors[majorPos], expires, current, count, icon) or needUpdate;
+            if major.bar then alert = true end
+            showMajors = true;
+            majorPos = majorPos + 1;
+        end
+        i = i + 1;
     end
-end
 
--- Used to update everything that is affected by the configuration
-function OmaRF:RefreshConfig()
-    self:OnDisable(); -- clear everything
-    if self.db.profile.enabled then
-        CompactRaidFrameContainer_ApplyToFrames(CompactRaidFrameContainer, "normal", configureIndicators);
-        watchedAuras = {};
-        for _, pos in ipairs(positions) do
-            for _, aura in ipairs(self.db.profile.indicators[pos]["auras"]) do
-                watchedAuras[aura] = pos; -- TODO single aura only in one position
+    if showInds or showMajors then
+        frame.indBase:Show();
+        frame.majorBase:Show();
+        if needUpdate and not updating[frame] then
+            updating[frame] = true; -- race?
+            -- create a function for updating the indicator
+            local func = updaters[frame];
+            if not func then
+                func = function() updateIndicators(frame) end;
+                updaters[frame] = func;
             end
+            CTimerAfter(0.20, func);
         end
-        majorAuras = {};
-        for _, aura in ipairs(self.db.profile.majorAuras["auras"]) do
-            majorAuras[aura] = true;
-        end
-        majorMax = OmaRF.db.profile.majorAuras["max"];
-
-        if next(watchedAuras) ~= nil or next(majorAuras) ~= nil then
-            self.running = true;
-            C_TimerAfter(0.15, self.UpdateAllIndicators);
-        end
+    else
+        frame.indBase:Hide();
+        frame.majorBase:Hide();
     end
+
+    return alert;
 end