87236a4 - Refactor OmaCD, create a more robust inspect scheme
[wowui.git] / OmaRF / Events.lua
index af49cd4..3098918 100644 (file)
@@ -5,7 +5,6 @@ local ssub = string.sub;
 local min = math.min;
 local UnitName, UnitClass, UnitExists = UnitName, UnitClass, UnitExists;
 local UnitDebuff, UnitIsCharmed = UnitDebuff, UnitIsCharmed;
-local UnitPower, UnitPowerMax, UnitPowerType = UnitPower, UnitPowerMax, UnitPowerType;
 local UnitHealth, UnitHealthMax = UnitHealth, UnitHealthMax;
 local UnitIsAFK, UnitIsDND = UnitIsAFK, UnitIsDND;
 local UnitGetIncomingHeals, UnitGetTotalAbsorbs = UnitGetIncomingHeals, UnitGetTotalAbsorbs;
@@ -28,7 +27,6 @@ local baseColor = Settings.BaseColor;
 local overlayColorDispel = Settings.OverlayColorDispel;
 local overlayColorCharm = Settings.OverlayColorCharm;
 local overlayColorAlert = Settings.OverlayColorAlert;
-local powerColors = Settings.PowerColors;
 local width = Settings.Width;
 
 local M = {};
@@ -43,32 +41,6 @@ function M.RegisterEvents(frame)
     if frame.unit == "focus" then frame:RegisterEvent("PLAYER_FOCUS_CHANGED") end
 end
 
-local function unregisterPower(frame)
-    frame.health:SetPoint("BOTTOMLEFT", frame.background, "BOTTOMLEFT");
-    frame.overlay:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT");
-    frame.mana:Hide();
-    frame:UnregisterEvent("PARTY_MEMBER_ENABLE");
-    frame:UnregisterEvent("PARTY_MEMBER_DISABLE");
-    frame:UnregisterEvent("UNIT_POWER");
-    frame:UnregisterEvent("UNIT_MAXPOWER");
-    frame:UnregisterEvent("UNIT_DISPLAYPOWER");
-    frame:UnregisterEvent("UNIT_POWER_BAR_SHOW");
-    frame:UnregisterEvent("UNIT_POWER_BAR_HIDE");
-end
-
-local function registerPower(frame)
-    frame.health:SetPoint("BOTTOMLEFT", frame.background, "BOTTOMLEFT", 0, 2);
-    frame.overlay:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", 0, 2);
-    frame.mana:Show();
-    frame:RegisterEvent("PARTY_MEMBER_ENABLE");
-    frame:RegisterEvent("PARTY_MEMBER_DISABLE");
-    frame:RegisterUnitEvent("UNIT_POWER", frame.unit);
-    frame:RegisterUnitEvent("UNIT_MAXPOWER", frame.unit);
-    frame:RegisterUnitEvent("UNIT_DISPLAYPOWER", frame.unit);
-    frame:RegisterUnitEvent("UNIT_POWER_BAR_SHOW", frame.unit);
-    frame:RegisterUnitEvent("UNIT_POWER_BAR_HIDE", frame.unit);
-end
-
 function M.RegisterUnitEvents(frame)
     -- events are taken from FrameXML/CompactUnitFrame.lua
     local displayed = frame.unit ~= frame.displayed and frame.displayed or nil;
@@ -90,28 +62,7 @@ function M.RegisterUnitEvents(frame)
 end
 local registerUnitEvents = M.RegisterUnitEvents;
 
-local function updateMaxHealth(frame, unit)
-    frame.health.max = UnitHealthMax(unit);
-end
-
-local function updateHealth(frame, unit)
-    local current, max = UnitHealth(unit), frame.health.max;
-    if current > max or max <= 0 then
-        -- somehow current health has gone over the maximum (missed maxhealth event possibly)
-        -- just put health bar full and update max health for next event
-        frame.health:SetWidth(width);
-        updateMaxHealth(frame, unit);
-        frame.health:Show();
-    elseif current <= 0 or UnitIsDeadOrGhost(unit) then
-        frame.health:Hide();
-    else
-        frame.health:SetWidth(current/max*width);
-        frame.health:Show();
-    end
-end
-
 local function updateText(frame, unit)
-    local healthLost = frame.health.max - UnitHealth(unit);
     if UnitIsDeadOrGhost(unit) then
         frame.text:SetText("Dead");
         frame.text:Show();
@@ -121,46 +72,43 @@ local function updateText(frame, unit)
     elseif UnitIsAFK(unit) then
         frame.text:SetText("afk");
         frame.text:Show();
-    elseif UnitIsDND(unit) and healthLost == 0 then
+    elseif UnitIsDND(unit) then
         frame.text:SetText("dnd");
         frame.text:Show();
-    elseif healthLost > 0 then
-        if healthLost > 1200000000 then -- 1.2B
-            frame.text:SetFormattedText("-%.1fB", healthLost / 1000000000);
-        elseif healthLost > 1200000 then -- 1.2M
-            frame.text:SetFormattedText("-%.1fM", healthLost / 1000000);
-        elseif healthLost > 1000 then -- 1K
-            frame.text:SetFormattedText("-%dK", healthLost / 1000);
-        else
-            frame.text:SetFormattedText("-%d", healthLost)
-        end
-        frame.text:Show();
     else
         frame.text:Hide();
     end
 end
+M.UpdateText = updateText;
 
-local function updateMaxPower(frame, unit)
-    frame.mana.max = UnitPowerMax(unit);
+local function updateMaxHealth(frame, unit)
+    frame.health.max = UnitHealthMax(unit);
 end
+M.UpdateMaxHealth = updateMaxHealth;
 
-local function updatePower(frame, unit)
-    local current, max = UnitPower(unit), frame.mana.max;
-    if current <= 0 then
-        frame.mana:Hide();
-    elseif current > max or max <= 0 then
-        frame.mana:SetWidth(width);
-        updateMaxPower(frame, unit);
-        frame.mana:Show();
+local function updateHealth(frame, unit)
+    local current, max = UnitHealth(unit), frame.health.max;
+    if current > max or max <= 0 then
+        -- somehow current health has gone over the maximum (missed maxhealth event possibly)
+        -- just put health bar full and update max health for next event
+        frame.health:SetWidth(width);
+        updateMaxHealth(frame, unit);
+        frame.health:Show();
+    elseif current <= 0 or UnitIsDeadOrGhost(unit) then
+        frame.dead = true;
+        frame.health:Hide();
+        updateText(frame, unit); -- update death
     else
-        frame.mana:SetWidth(current/max*width);
-        frame.mana:Show();
+        frame.health:SetWidth(current/max*width);
+        frame.health:Show();
     end
-end
 
-local function updatePowerColor(frame, unit)
-    frame.mana:SetVertexColor(unpack(powerColors[UnitPowerType(unit)]));
+    if frame.dead and current > 0 then
+        frame.dead = nil;
+        updateText(frame, unit); -- update revive
+    end
 end
+M.UpdateHealth = updateHealth;
 
 -- TODO maybe add a prefix when in vehicle
 local function updateName(frame, unit)
@@ -172,6 +120,7 @@ local function updateName(frame, unit)
     local color = RAID_CLASS_COLORS[class];
     if color then frame.name:SetVertexColor(color.r, color.g, color.b) end
 end
+M.UpdateName = updateName;
 
 local function updateHealPred(frame, unit)
     local incoming = UnitGetIncomingHeals(unit) or 0;
@@ -185,6 +134,7 @@ local function updateHealPred(frame, unit)
         frame.healpred:Hide();
     end
 end
+M.UpdateHealPred = updateHealPred;
 
 local function updateShield(frame, unit)
     local shield = UnitGetTotalAbsorbs(unit) or 0;
@@ -208,6 +158,7 @@ local function updateShield(frame, unit)
         frame.shieldhl:Hide();
     end
 end
+M.UpdateShield = updateShield;
 
 local function updateHealAbsorb(frame, unit)
     local absorb = UnitGetTotalHealAbsorbs(unit) or 0;
@@ -220,6 +171,7 @@ local function updateHealAbsorb(frame, unit)
         frame.healabsorb:Hide();
     end
 end
+M.UpdateHealAbsorb = updateHealAbsorb;
 
 local function updateAuras(frame, unit)
     local alert = checkIndicators(frame, unit);
@@ -250,6 +202,7 @@ local function updateAuras(frame, unit)
         end
     end
 end
+M.UpdateAuras = updateAuras;
 
 local function updateAggro(frame, unit)
     local status = UnitThreatSituation(unit);
@@ -259,6 +212,7 @@ local function updateAggro(frame, unit)
         frame.base:SetVertexColor(unpack(baseColor));
     end
 end
+M.UpdateAggro = updateAggro;
 
 local function updateVehicle(frame)
     local shouldTargetVehicle = UnitHasVehicleUI(frame.unit) and
@@ -275,29 +229,21 @@ local function updateVehicle(frame)
         registerUnitEvents(frame);
     end
 end
+M.UpdateVehicle = updateVehicle;
 
 local function updateRole(frame, unit)
     local role = UnitGroupRolesAssigned(unit);
     if role == "HEALER" then
         frame.role:SetTexCoord(0.75, 1, 0, 1);
         frame.role:Show();
-        registerPower(frame);
-        frame.role.healer = true;
     elseif role == "TANK" then
         frame.role:SetTexCoord(0.5, 0.75, 0, 1);
         frame.role:Show();
-        if frame.role.healer then
-            frame.role.healer = false;
-            unregisterPower(frame);
-        end
     else
         frame.role:Hide();
-        if frame.role.healer then
-            frame.role.healer = false;
-            unregisterPower(frame);
-        end
     end
 end
+M.UpdateRole = updateRole;
 
 local function updateReadyCheck(frame, unit)
     local status = GetReadyCheckStatus(unit);
@@ -314,6 +260,7 @@ local function updateReadyCheck(frame, unit)
         frame.ready:Hide()
     end
 end
+M.UpdateReadyCheck = updateReadyCheck;
 
 local function updateRaidMarker(frame, unit)
     local index = GetRaidTargetIndex(unit);
@@ -324,11 +271,11 @@ local function updateRaidMarker(frame, unit)
         frame.targeticon:Hide();
     end
 end
+M.UpdateRaidMarker = updateRaidMarker;
 
 local eventFuncs = {
     ["UNIT_HEALTH"] = function(frame)
         updateHealth(frame, frame.displayed);
-        updateText(frame, frame.displayed);
         updateShield(frame, frame.displayed);
         updateHealAbsorb(frame, frame.displayed);
         -- no heal prediction update, that doesn't overflow too much
@@ -337,9 +284,6 @@ local eventFuncs = {
         -- without a RAID_TARGET_UPDATE event
         --updateRaidMarker(frame, frame.unit);
     end,
-    ["UNIT_POWER"] = function(frame)
-        updatePower(frame, frame.displayed);
-    end,
     ["UNIT_AURA"] = function(frame)
         updateAuras(frame, frame.displayed);
     end,
@@ -358,30 +302,15 @@ local eventFuncs = {
     ["UNIT_MAXHEALTH"] = function(frame)
         updateMaxHealth(frame, frame.displayed);
         updateHealth(frame, frame.displayed);
-        updateText(frame, frame.displayed);
         updateShield(frame, frame.displayed);
         updateHealAbsorb(frame, frame.displayed);
     end,
-    ["UNIT_MAXPOWER"] = function(frame)
-        updateMaxPower(frame, frame.displayed);
-        updatePower(frame, frame.displayed);
-    end,
-    ["UNIT_DISPLAYPOWER"] = function(frame)
-        updatePowerColor(frame, frame.displayed);
-        updateMaxPower(frame, frame.displayed);
-        updatePower(frame, frame.displayed);
-    end,
     ["UNIT_NAME_UPDATE"] = function(frame)
         updateName(frame, frame.unit);
     end,
     ["UNIT_CONNECTION"] = function(frame)
         updateText(frame, frame.displayed);
     end,
-    ["PARTY_MEMBER_ENABLE"] = function(frame)
-        -- new power info possibly (FrameXML/CompactUnitFrame.lua)
-        updateMaxPower(frame, frame.displayed);
-        updatePowerColor(frame, frame.displayed);
-    end,
     ["PLAYER_ROLES_ASSIGNED"] = function(frame)
         updateRole(frame, frame.unit);
     end,
@@ -397,11 +326,6 @@ local eventFuncs = {
         updateMaxHealth(frame, frame.displayed);
         updateHealth(frame, frame.displayed);
         updateText(frame, frame.displayed);
-        if frame.role.healer then
-            updateMaxPower(frame, frame.displayed);
-            updatePower(frame, frame.displayed);
-            updatePowerColor(frame, frame.displayed);
-        end
         updateAuras(frame, frame.displayed);
         updateShield(frame, frame.displayed);
         updateHealPred(frame, frame.displayed);
@@ -413,10 +337,7 @@ local eventFuncs = {
     end,
 };
 eventFuncs["UNIT_HEALTH_FREQUENT"] = eventFuncs["UNIT_HEALTH"];
-eventFuncs["PARTY_MEMBER_DISABLE"] = eventFuncs["PARTY_MEMBER_ENABLE"];
 eventFuncs["PLAYER_FLAGS_CHANGED"] = eventFuncs["UNIT_CONNECTION"];
-eventFuncs["UNIT_POWER_BAR_SHOW"] = eventFuncs["UNIT_DISPLAYPOWER"];
-eventFuncs["UNIT_POWER_BAR_HIDE"] = eventFuncs["UNIT_DISPLAYPOWER"];
 eventFuncs["READY_CHECK_CONFIRM"] = eventFuncs["READY_CHECK"];
 eventFuncs["READY_CHECK_FINISHED"] = eventFuncs["READY_CHECK"];
 eventFuncs["UNIT_ENTERED_VEHICLE"] = eventFuncs["UPDATE_ALL_BARS"];