7cd9b15 - Add pet frame, remove mana bars from non-healers

[wowui.git] / OmaRF / Indicators.lua

diff --git a/OmaRF/Indicators.lua b/OmaRF/Indicators.lua
index ba89735..3ecdcf6 100644 (file)
--- a/OmaRF/Indicators.lua
+++ b/OmaRF/Indicators.lua
@@ -1,149 +1,175 @@
-local media = LibStub:GetLibrary("LibSharedMedia-3.0");
-local f = OmaRF.frames;
-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; -- all watched auras
-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 GetTime = GetTime;
 local UnitAura = UnitAura;

-local UnitIsPlayer = UnitIsPlayer;
-local UnitIsConnected = UnitIsConnected;
-local UnitIsDeadOrGhost = UnitIsDeadOrGhost;
-local CompactRaidFrameContainer_ApplyToFrames = CompactRaidFrameContainer_ApplyToFrames;
+local UnitIsDeadOrGhost, UnitIsConnected = UnitIsDeadOrGhost, UnitIsConnected;
+local CTimerAfter = C_Timer.After;
+
+local Settings = OmaRFSettings;
+local majorAuras = Settings.MajorAuras;
+local indSize = Settings.IndSize;
+local positions = nil;
+local watchedAuras = nil;
+
+local updaters = {};
+local updating = {};
+local auraFilters = {"HELPFUL", "HARMFUL"};

 
 

--- list of important auras TODO try to use spellIDs
-local centerAuras = {
-    "Power Word: Shield"
-};
+local M = {};
+OmaRFIndicators = M;

 
 

-local function configureIndicators(frame, name)
-    local frameName = name or frame:GetName();
-    if not f[frameName] then return end
+function M.SetupIndicators(frame)
+    if not watchedAuras then
+        watchedAuras = Settings.Character["WatchedAuras"];
+        positions = Settings.Character.Positions;
+    end

 
 

-    local config = OmaRF.db.profile;
-    local font = media and media:Fetch('font', config.indicatorFont) or STANDARD_TEXT_FONT;
-    for pos, ind in pairs(f[frameName]) do
-        ind.text:SetFont(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"]));
-        if config[pos]["showIcon"] then
-            ind.icon:Show();
+    frame.inds = CreateFrame("Frame", nil, frame);
+    frame.inds:SetAllPoints();
+    frame.inds:Hide();
+    for _, pos in pairs(positions) do
+        frame.inds[pos] = frame.inds:CreateTexture(nil, "OVERLAY");
+        frame.inds[pos]:SetPoint(pos, frame.inds, pos);
+        frame.inds[pos]:SetWidth(indSize);
+        frame.inds[pos]:SetHeight(indSize);
+        frame.inds[pos]:SetTexture("Interface\\AddOns\\OmaRF\\images\\rhomb");
+        frame.inds[pos]:SetVertexColor(1, 0, 0);
+        frame.inds[pos]:Hide();
+        frame.inds[pos].text = frame.inds:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
+        frame.inds[pos].text:SetPoint("BOTTOMRIGHT", frame.inds[pos], "BOTTOMRIGHT");
+        frame.inds[pos].text:Hide();
+    end
+    frame.major = CreateFrame("Frame", nil, frame);
+    frame.major:SetPoint("TOPLEFT", frame, "TOPLEFT", 4, -indSize + 4);
+    frame.major:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT");
+    for i = 1,3 do
+        frame.major[i] = frame.major:CreateTexture(nil, "OVERLAY");
+        if i == 1 then
+            frame.major[i]:SetPoint("TOPLEFT", frame.major, "TOPLEFT");

         else
         else

-            ind.icon:Hide();
+            frame.major[i]:SetPoint("TOPLEFT", frame.major[i-1], "TOPRIGHT");

         end
         end

+        frame.major[i]:SetWidth(indSize*2);
+        frame.major[i]:SetHeight(indSize*2);
+        frame.major[i]:Hide();
+        frame.major[i].text = frame.major:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
+        frame.major[i].text:SetPoint("BOTTOMRIGHT", frame.major[i], "BOTTOMRIGHT");
+        frame.major[i].text:Hide();
+        frame.major[i].stack = frame.major:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
+        frame.major[i].stack:SetPoint("TOPLEFT", frame.major[i], "TOPLEFT");
+        frame.major[i].stack:Hide();

     end
 end
 
     end
 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 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
     end

-    configureIndicators(frame, name);
+    return true;

 end
 
 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;
+local function updateIndicators(frame, unit)
+    if not frame:IsShown() or not UnitIsConnected(unit) or UnitIsDeadOrGhost(unit) then
+        updating[frame] = nil;
+        return;
+    end

 
 

-    -- Create indicators if needed
-    if not f[frameName] then setupCompactUnitFrame(frame, frameName) end
-    -- Reset current
-    for _, ind in pairs(f[frameName]) do
-        ind.text:SetText("");
-        ind.icon:SetTexture("");
+    local needUpdate = false;
+    local current = GetTime();
+    for _, pos in pairs(positions) do
+        local ind = frame.inds[pos];
+        if ind.expires ~= nil then
+            needUpdate = remaining(ind.text, ind.expires, current) or needUpdate;
+        end

     end
     end

-    -- Hide if unit is dead/disconnected
-    if (not UnitIsConnected(unit)) or UnitIsDeadOrGhost(frame.displayedUnit) then
-        return;
+    for i = 1,3 do
+        local ind = frame.major[i];
+        if ind.expires ~= nil then
+            needUpdate = remaining(ind.text, ind.expires, current) or needUpdate;
+        end
+    end
+    if needUpdate then
+        CTimerAfter(0.16, updaters[frame]);
+    else
+        updating[frame] = nil;

     end
     end

+end

 
 

-    local name, icon, count, debuff, expires, caster, id;
+function M.CheckIndicators(frame, unit)
+    for _, pos in pairs(positions) do
+        frame.inds[pos].expires = nil;
+        frame.inds[pos]:Hide();
+        frame.inds[pos].text:Hide();
+    end
+    for i = 1,3 do
+        frame.major[i].expires = nil;
+        frame.major[i]:Hide();
+        frame.major[i].text:Hide();
+        frame.major[i].stack:Hide();
+    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 _, filter in ipairs(auraFilters) do
         local i = 1;
         while true do
     local current = GetTime();
     for _, filter in ipairs(auraFilters) do
         local i = 1;
         while true do

-            name, _, icon, count, debuff, _, expires, caster, _, _, id = UnitAura(unit, i, filter);
+            name, _, icon, count, _, _, expires, caster, _, _, id = UnitAura(unit, i, filter);

             if not id then break end
             if not id then break end

-            local pos = watchedAuras[name] or watchedAuras[id] or watchedAuras[debuff];
-            if pos then
-                local config = OmaRF.db.profile[pos];
-                if not config.mine or UnitIsPlayer(caster) then
-                    if config.showIcon and not config.useDefaultIcon then
-                        -- show icon
-                        ind.icon:SetTexture(icon);
-                    end
-                    if config.showText then
-                        -- show text
-                        local text;
-                        local remaining = expires - current;
-                        if remaining > 60 then
-                            text = string.format("%dm", ceil(remaining/60));
-                        else
-                            text = string.format("%d", floor(remaining+0.5));
-                        end
-                        if count > 1 and config.stack then
-                            if text then
-                                text = count.."-"..text;
-                            else
-                                text = count;
-                            end
-                        end
-
-                        ind.text:SetText(text);
-                    end
+            local pos = watchedAuras[id] or watchedAuras[name];
+            if pos and caster == "player" then
+                needUpdate = remaining(frame.inds[pos].text, expires, current);
+                frame.inds[pos].expires = expires;
+                frame.inds[pos]:Show();
+                frame.inds[pos].text:Show();
+                showInds = true;
+            end
+            local major = majorAuras[id] or majorAuras[name];
+            if major and majorPos <= 3 then
+                needUpdate = remaining(frame.major[majorPos].text, expires, current);
+                frame.major[majorPos].expires = expires;
+                frame.major[majorPos]:SetTexture(icon);
+                frame.major[majorPos]:Show();
+                frame.major[majorPos].text:Show();
+                if count > 1 then
+                    frame.major[majorPos].stack:SetText(count);
+                    frame.major[majorPos].stack:Show();
+                end
+                if major.bar then
+                    alert = true;

                 end
                 end

+                showMajors = true;
+                majorPos = majorPos + 1;

             end
             i = i + 1;
         end
     end
             end
             i = i + 1;
         end
     end

-end
-
--- Update all indicators
-function OmaRF:UpdateAllIndicators()
-    CompactRaidFrameContainer_ApplyToFrames(CompactRaidFrameContainer, "normal", updateIndicators);
-    if self.running then
-        C_TimerAfter(0.15, self:UpdateAllIndicators);
-    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);
-        -- Format aura strings
-        watchedAuras = {};
-        for _, pos in ipairs(positions) do
-            for _, aura in ipairs(self.db.profile[pos]["auras"]) do
-                watchedAuras[aura] = pos; -- TODO single aura only in one position
+    if showInds or showMajors then
+        frame.inds:Show();
+        frame.major: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, unit) end;
+                updaters[frame] = func;

             end
             end

+            CTimerAfter(0.16, func);

         end
         end

-
-        if next(watchedAuras) ~= nil then
-            self.running = true;
-            C_TimerAfter(0.15, self:UpdateAllIndicators);
-        end
+    else
+        frame.inds:Hide();
+        frame.major:Hide();

     end
     end

+
+    return alert;

 end
 end