92fe6f2 - Vehicle support

[wowui.git] / OmaRF / FrameInit.lua

diff --git a/OmaRF/FrameInit.lua b/OmaRF/FrameInit.lua
index 51c59e3..df34967 100644 (file)
--- a/OmaRF/FrameInit.lua
+++ b/OmaRF/FrameInit.lua
@@ -1,7 +1,9 @@
 -- FrameInit.lua
 local _;
 local unpack, pairs, rawget = unpack, pairs, rawget;
 -- FrameInit.lua
 local _;
 local unpack, pairs, rawget = unpack, pairs, rawget;

+local format = string.format;

 local UnitHealthMax, UnitPowerMax = UnitHealthMax, UnitPowerMax;
 local UnitHealthMax, UnitPowerMax = UnitHealthMax, UnitPowerMax;

+local UnitInRange = UnitInRange;

 local CreateFrame, RegisterStateDriver, RegisterUnitWatch = CreateFrame, RegisterStateDriver, RegisterUnitWatch;
 local PowerTypeMana = Enum.PowerType.Mana;
 local PowerTypeRage = Enum.PowerType.Rage;
 local CreateFrame, RegisterStateDriver, RegisterUnitWatch = CreateFrame, RegisterStateDriver, RegisterUnitWatch;
 local PowerTypeMana = Enum.PowerType.Mana;
 local PowerTypeRage = Enum.PowerType.Rage;


@@ -9,6 +11,8 @@ local PowerTypeFocus = Enum.PowerType.Focus;
 local PowerTypeEnergy = Enum.PowerType.Energy;
 local PowerTypeRunic = Enum.PowerType.RunicPower;
 
 local PowerTypeEnergy = Enum.PowerType.Energy;
 local PowerTypeRunic = Enum.PowerType.RunicPower;
 

+local unitEvent = nil;
+

 local M = {};
 OmaFrames = M;
 
 local M = {};
 OmaFrames = M;
 


@@ -16,8 +20,7 @@ OmaFrames = M;
 M.Positions = {"TOPRIGHT", "BOTTOMLEFT"};
 M.Width, M.Height = 80, 40;
 M.AnchorX, M.AnchorY = 0, -330;
 M.Positions = {"TOPRIGHT", "BOTTOMLEFT"};
 M.Width, M.Height = 80, 40;
 M.AnchorX, M.AnchorY = 0, -330;

-M.IndSize = 14;
-M.BaseColor = {0, 0, 0};
+M.IndSize = 14; M.BaseColor = {0, 0, 0};

 M.BgColor = {0.7, 0.7, 0.7};
 M.HealthColor = {0.3, 0.3, 0.3};
 M.ShieldColor = {0, 0.7, 1};
 M.BgColor = {0.7, 0.7, 0.7};
 M.HealthColor = {0.3, 0.3, 0.3};
 M.ShieldColor = {0, 0.7, 1};


@@ -86,13 +89,64 @@ local function setupIndicators(frame)
     end
 end
 
     end
 end
 

+local function unitUpdate(self, elapsed)
+    -- there's no in/out of range event, have to check each frame
+    -- from FrameXML/CompactUnitFrame.lua
+    local inRange, checked = UnitInRange(self.displayed);
+    if checked and not inRange then
+        self:SetAlpha(0.55);
+    else
+        self:SetAlpha(1);
+    end
+end
+
+function M.RegisterEvents(frame)
+    -- events are taken from FrameXML/CompactUnitFrame.lua
+    -- TODO vehicle support, ready check support, raid marker support,
+    -- player flags support (/afk, /dnd)
+    -- TODO only update for vehicle events here
+    local displayed = frame.unit ~= frame.displayed and frame.displayed or nil;
+    frame:RegisterEvent("PARTY_MEMBER_ENABLE");
+    frame:RegisterEvent("PARTY_MEMBER_DISABLE");
+    frame:RegisterEvent("UNIT_ENTERED_VEHICLE");
+    frame:RegisterEvent("UNIT_EXITED_VEHICLE");
+    frame:RegisterEvent("UNIT_PET");
+    frame:RegisterUnitEvent("UNIT_HEALTH", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_HEALTH_FREQUENT", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_MAXHEALTH", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_POWER", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_MAXPOWER", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_DISPLAYPOWER", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_NAME_UPDATE", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_AURA", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_HEAL_PREDICTION", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_ABSORB_AMOUNT_CHANGED", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_HEAL_ABSORB_AMOUNT_CHANGED", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_THREAT_SITUATION_UPDATE", frame.unit, displayed);
+    frame:RegisterUnitEvent("UNIT_CONNECTION", frame.unit, displayed);
+    frame:RegisterUnitEvent("INCOMING_RESURRECT_CHANGED", frame.unit, displayed);
+end
+local registerEvents = M.RegisterEvents;
+
+local function frameShow(frame)
+    registerEvents(frame);
+    frame:SetScript("OnUpdate", unitUpdate);
+    unitEvent(frame, "UPDATE_ALL_BARS", frame.displayed);
+end
+
+local function frameHide(frame)
+    frame:UnregisterAllEvents();
+    frame:SetScript("OnUpdate", nil);
+end
+

 local function setupFrame(frame, secure, unit)
     -- create visuals
     secure:SetWidth(width+2);
     frame:SetWidth(width+2);
     frame.base = frame:CreateTexture(nil, "BACKGROUND");
     frame.base:SetAllPoints();
 local function setupFrame(frame, secure, unit)
     -- create visuals
     secure:SetWidth(width+2);
     frame:SetWidth(width+2);
     frame.base = frame:CreateTexture(nil, "BACKGROUND");
     frame.base:SetAllPoints();

-    frame.base:SetColorTexture(unpack(baseColor));
+    frame.base:SetColorTexture(1, 1, 1);
+    frame.base:SetVertexColor(unpack(baseColor));

     frame.background = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
     frame.background:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
     frame.background:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
     frame.background = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
     frame.background:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
     frame.background:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);


@@ -133,6 +187,7 @@ local function setupFrame(frame, secure, unit)
     frame.overlay = frame:CreateTexture(nil, "ARTWORK", nil, 1);
     frame.overlay:SetPoint("TOPLEFT", frame.background, "TOPLEFT");
     frame.overlay:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", 0, 2);
     frame.overlay = frame:CreateTexture(nil, "ARTWORK", nil, 1);
     frame.overlay:SetPoint("TOPLEFT", frame.background, "TOPLEFT");
     frame.overlay:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", 0, 2);

+    frame.overlay:SetColorTexture(1, 1, 1);

     frame.overlay:Hide();
     frame.name = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
     frame.name:SetPoint("CENTER", frame.background, "CENTER", 0, 11);
     frame.overlay:Hide();
     frame.name = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
     frame.name:SetPoint("CENTER", frame.background, "CENTER", 0, 11);


@@ -141,23 +196,8 @@ local function setupFrame(frame, secure, unit)
     frame.text:SetPoint("CENTER", frame.background, "CENTER", 0, -1);
     frame.text:Hide();
     setupIndicators(frame);
     frame.text:SetPoint("CENTER", frame.background, "CENTER", 0, -1);
     frame.text:Hide();
     setupIndicators(frame);

-    -- setup events
-    frame:RegisterEvent("PARTY_MEMBER_ENABLE");
-    frame:RegisterEvent("PARTY_MEMBER_DISABLE");
-    frame:RegisterUnitEvent("UNIT_HEALTH", unit);
-    frame:RegisterUnitEvent("UNIT_HEALTH_FREQUENT", unit);
-    frame:RegisterUnitEvent("UNIT_MAXHEALTH", unit);
-    frame:RegisterUnitEvent("UNIT_POWER", unit);
-    frame:RegisterUnitEvent("UNIT_MAXPOWER", unit);
-    frame:RegisterUnitEvent("UNIT_DISPLAYPOWER", unit);
-    frame:RegisterUnitEvent("UNIT_NAME_UPDATE", unit);
-    frame:RegisterUnitEvent("UNIT_AURA", unit);
-    frame:RegisterUnitEvent("UNIT_HEAL_PREDICTION", unit);
-    frame:RegisterUnitEvent("UNIT_ABSORB_AMOUNT_CHANGED", unit);
-    frame:RegisterUnitEvent("UNIT_HEAL_ABSORB_AMOUNT_CHANGED", unit);
-    frame:RegisterUnitEvent("UNIT_THREAT_SITUATION_UPDATE", unit);
-    frame:RegisterUnitEvent("UNIT_CONNECTION", unit);
-    frame:RegisterUnitEvent("INCOMING_RESURRECT_CHANGED", unit);
+    frame:SetScript("OnShow", frameShow);
+    frame:SetScript("OnHide", frameHide);

     -- set attributes
     secure:RegisterForClicks("AnyDown");
     secure:SetAttribute("type1", "spell"); -- left click
     -- set attributes
     secure:RegisterForClicks("AnyDown");
     secure:SetAttribute("type1", "spell"); -- left click


@@ -181,87 +221,135 @@ local function setupFrame(frame, secure, unit)
     secure:SetAttribute("*type2", "togglemenu");
 end
 
     secure:SetAttribute("*type2", "togglemenu");
 end
 

+-- vehicle toggle from Shadowed Unit Frames
+local vehicletoggle = [=[
+    local unit = self:GetAttribute("unit");
+    if unit and newstate == "vehicle" and not UnitTargetsVehicleInRaidUI(unit) then
+        print(unit, "in vehicle, still target player");
+        self:SetAttribute("toggleForVehicle", false);
+    else
+        self:SetAttribute("toggleForVehicle", true);
+    end
+]=]
+
+-- TODO reorganizing needed for eventhandler stuff

 function M.InitializeParty(parent, party, eventhandler)
 function M.InitializeParty(parent, party, eventhandler)

-    local secure = CreateFrame("Button", "OmaPartySecure0", parent, "SecureUnitButtonTemplate");
+    unitEvent = eventhandler;
+    local secure = CreateFrame("Button", "OmaPartySecure0", parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");

     local frame = CreateFrame("Frame", "OmaParty0", parent);
     local unit = "player";
     secure:SetAttribute("unit", unit);
     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
     secure:SetHeight(height+2);
     frame.unit = unit;
     local frame = CreateFrame("Frame", "OmaParty0", parent);
     local unit = "player";
     secure:SetAttribute("unit", unit);
     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
     secure:SetHeight(height+2);
     frame.unit = unit;

+    frame.vehicle = "vehicle";
+    frame.displayed = unit;
+    frame:SetAttribute("unit", unit);

     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
     frame:SetHeight(height+2);
     frame:SetScript("OnEvent", eventhandler);
     frame:Hide();
     setupFrame(frame, secure, unit);
     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
     frame:SetHeight(height+2);
     frame:SetScript("OnEvent", eventhandler);
     frame:Hide();
     setupFrame(frame, secure, unit);

+    RegisterStateDriver(frame, "visibility", "[@player,group:raid] hide; show");

     RegisterStateDriver(secure, "visibility", "[@player,group:raid] hide; show");
     RegisterStateDriver(secure, "visibility", "[@player,group:raid] hide; show");

+    RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
+    secure:SetAttribute("_onstate-vehicleui", vehicletoggle);

     party[0] = {secure=secure, frame=frame};
     for i = 1,4 do
     party[0] = {secure=secure, frame=frame};
     for i = 1,4 do

-        local secure = CreateFrame("Button", "OmaPartySecure"..i, parent, "SecureUnitButtonTemplate");
+        local secure = CreateFrame("Button", "OmaPartySecure"..i, parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");

         local frame = CreateFrame("Frame", "OmaParty"..i, parent);
         local unit = "party"..i;
         secure:SetAttribute("unit", unit);
         secure:SetPoint("TOPLEFT", party[i-1].secure, "TOPRIGHT");
         secure:SetPoint("BOTTOMLEFT", party[i-1].secure, "BOTTOMRIGHT");
         frame.unit = unit;
         local frame = CreateFrame("Frame", "OmaParty"..i, parent);
         local unit = "party"..i;
         secure:SetAttribute("unit", unit);
         secure:SetPoint("TOPLEFT", party[i-1].secure, "TOPRIGHT");
         secure:SetPoint("BOTTOMLEFT", party[i-1].secure, "BOTTOMRIGHT");
         frame.unit = unit;

+        frame.vehicle = unit.."pet";
+        frame.displayed = unit;
+        frame:SetAttribute("unit", unit);

         frame:SetPoint("TOPLEFT", party[i-1].frame, "TOPRIGHT");
         frame:SetPoint("BOTTOMLEFT", party[i-1].frame, "BOTTOMRIGHT");
         frame:SetScript("OnEvent", unitEvent);
         frame:Hide();
         setupFrame(frame, secure, unit);
         frame:SetPoint("TOPLEFT", party[i-1].frame, "TOPRIGHT");
         frame:SetPoint("BOTTOMLEFT", party[i-1].frame, "BOTTOMRIGHT");
         frame:SetScript("OnEvent", unitEvent);
         frame:Hide();
         setupFrame(frame, secure, unit);

-        RegisterUnitWatch(secure);
+        local visible = format("[@%s,exists,group:raid] hide; [@%s,exists] show; hide", unit, unit);
+        local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
+        RegisterStateDriver(frame, "visibility", visible);
+        RegisterStateDriver(secure, "visibility", visible);
+        RegisterStateDriver(secure, "vehicleui", vehicle);
+        secure:SetAttribute("_onstate-vehicleui", vehicletoggle);

         party[i] = {secure=secure, frame=frame};
     end
 end
 
 function M.InitializeRaid(parent, raid, eventhandler)
         party[i] = {secure=secure, frame=frame};
     end
 end
 
 function M.InitializeRaid(parent, raid, eventhandler)

-    local secure = CreateFrame("Button", "OmaRaidSecure1", parent, "SecureUnitButtonTemplate");
+    local secure = CreateFrame("Button", "OmaRaidSecure1", parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");

     local frame = CreateFrame("Frame", "OmaRaid1", parent);
     local unit = "raid1";
     secure:SetAttribute("unit", unit);
     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
     secure:SetHeight(height+2);
     frame.unit = unit;
     local frame = CreateFrame("Frame", "OmaRaid1", parent);
     local unit = "raid1";
     secure:SetAttribute("unit", unit);
     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
     secure:SetHeight(height+2);
     frame.unit = unit;

+    frame.vehicle = unit.."pet";
+    frame.displayed = unit;
+    frame:SetAttribute("unit", unit);

     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
     frame:SetHeight(height+2);
     frame:SetScript("OnEvent", eventhandler);
     frame:Hide();
     setupFrame(frame, secure, unit);
     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
     frame:SetHeight(height+2);
     frame:SetScript("OnEvent", eventhandler);
     frame:Hide();
     setupFrame(frame, secure, unit);

+    RegisterUnitWatch(frame);

     RegisterUnitWatch(secure);
     RegisterUnitWatch(secure);

+    local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
+    RegisterStateDriver(secure, "vehicleui", vehicle);
+    secure:SetAttribute("_onstate-vehicleui", vehicletoggle);

     raid[1] = {secure=secure, frame=frame};
     for y = 1,7 do
         local i = y*5+1;
     raid[1] = {secure=secure, frame=frame};
     for y = 1,7 do
         local i = y*5+1;

-        local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, "SecureUnitButtonTemplate");
+        local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");

         local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
         local unit = "raid"..i;
         secure:SetAttribute("unit", unit);
         secure:SetPoint("TOPLEFT", raid[i-5].secure, "BOTTOMLEFT");
         secure:SetHeight(height+2);
         frame.unit = unit;
         local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
         local unit = "raid"..i;
         secure:SetAttribute("unit", unit);
         secure:SetPoint("TOPLEFT", raid[i-5].secure, "BOTTOMLEFT");
         secure:SetHeight(height+2);
         frame.unit = unit;

+        frame.vehicle = unit.."pet";
+        frame.displayed = unit;
+        frame:SetAttribute("unit", unit);

         frame:SetPoint("TOPLEFT", raid[i-5].frame, "BOTTOMLEFT");
         frame:SetHeight(height+2);
         frame:SetScript("OnEvent", unitEvent);
         frame:Hide();
         setupFrame(frame, secure, unit);
         frame:SetPoint("TOPLEFT", raid[i-5].frame, "BOTTOMLEFT");
         frame:SetHeight(height+2);
         frame:SetScript("OnEvent", unitEvent);
         frame:Hide();
         setupFrame(frame, secure, unit);

+        RegisterUnitWatch(frame);

         RegisterUnitWatch(secure);
         RegisterUnitWatch(secure);

+        local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
+        RegisterStateDriver(secure, "vehicleui", vehicle);
+        secure:SetAttribute("_onstate-vehicleui", vehicletoggle);

         raid[i] = {secure=secure, frame=frame};
     end
     for y = 0,7 do
         for x = 1,4 do
             local i = y*5+x+1;
         raid[i] = {secure=secure, frame=frame};
     end
     for y = 0,7 do
         for x = 1,4 do
             local i = y*5+x+1;

-            local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, "SecureUnitButtonTemplate");
+            local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, "SecureUnitButtonTemplate,SecureHandlerStateTemplate");

             local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
             local unit = "raid"..i;
             secure:SetAttribute("unit", unit);
             secure:SetPoint("TOPLEFT", raid[i-1].secure, "TOPRIGHT");
             secure:SetPoint("BOTTOMLEFT", raid[i-1].secure, "BOTTOMRIGHT");
             frame.unit = unit;
             local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
             local unit = "raid"..i;
             secure:SetAttribute("unit", unit);
             secure:SetPoint("TOPLEFT", raid[i-1].secure, "TOPRIGHT");
             secure:SetPoint("BOTTOMLEFT", raid[i-1].secure, "BOTTOMRIGHT");
             frame.unit = unit;

+            frame.vehicle = unit.."pet";
+            frame.displayed = unit;
+            frame:SetAttribute("unit", unit);

             frame:SetPoint("TOPLEFT", raid[i-1].frame, "TOPRIGHT");
             frame:SetPoint("BOTTOMLEFT", raid[i-1].frame, "BOTTOMRIGHT");
             frame:SetScript("OnEvent", unitEvent);
             frame:Hide();
             setupFrame(frame, secure, unit);
             frame:SetPoint("TOPLEFT", raid[i-1].frame, "TOPRIGHT");
             frame:SetPoint("BOTTOMLEFT", raid[i-1].frame, "BOTTOMRIGHT");
             frame:SetScript("OnEvent", unitEvent);
             frame:Hide();
             setupFrame(frame, secure, unit);

+            RegisterUnitWatch(frame);

             RegisterUnitWatch(secure);
             RegisterUnitWatch(secure);

+            local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
+            RegisterStateDriver(secure, "vehicleui", vehicle);
+            secure:SetAttribute("_onstate-vehicleui", vehicletoggle);

             raid[i] = {secure=secure, frame=frame};
         end
     end
             raid[i] = {secure=secure, frame=frame};
         end
     end