-- FrameInit.lua
local _;
local unpack, pairs, rawget = unpack, pairs, rawget;
+local format = string.format;
local UnitHealthMax, UnitPowerMax = UnitHealthMax, UnitPowerMax;
+local UnitInRange = UnitInRange;
local CreateFrame, RegisterStateDriver, RegisterUnitWatch = CreateFrame, RegisterStateDriver, RegisterUnitWatch;
-local SPELL_POWER_MANA = SPELL_POWER_MANA;
+local PowerTypeMana = Enum.PowerType.Mana;
+local PowerTypeRage = Enum.PowerType.Rage;
+local PowerTypeFocus = Enum.PowerType.Focus;
+local PowerTypeEnergy = Enum.PowerType.Energy;
+local PowerTypeRunic = Enum.PowerType.RunicPower;
+
+local unitEvent = nil;
local M = {};
OmaFrames = M;
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.OverlayColorCharm = {0.8, 0, 1, 0.5};
M.OverlayColorAlert = {1, 0, 0, 0.5};
M.PowerColors = {
- [SPELL_POWER_MANA] = {0, 0.5, 1},
- [SPELL_POWER_RAGE] = {1, 0, 0},
- [SPELL_POWER_FOCUS] = {1, 0.5, 0},
- [SPELL_POWER_ENERGY] = {1, 0.8, 0},
- [SPELL_POWER_RUNIC_POWER] = {0.9, 0, 0.1},
+ [PowerTypeMana] = {0, 0.5, 1},
+ [PowerTypeRage] = {1, 0, 0},
+ [PowerTypeFocus] = {1, 0.5, 0},
+ [PowerTypeEnergy] = {1, 0.8, 0},
+ [PowerTypeRunic] = {0.9, 0, 0.1},
};
-- watch to not remove mana entry
-setmetatable(M.PowerColors, {__index = function(t) return rawget(t, SPELL_POWER_MANA) end});
+setmetatable(M.PowerColors, {__index = function(t) return rawget(t, PowerTypeMana) end});
local positions = M.Positions;
local width, height = M.Width, M.Height;
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();
- 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.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.text:SetPoint("CENTER", frame.background, "CENTER", 0, -1);
frame.text:Hide();
setupIndicators(frame);
+ frame:SetScript("OnShow", frameShow);
+ frame:SetScript("OnHide", frameHide);
-- set attributes
secure:RegisterForClicks("AnyDown");
secure:SetAttribute("type1", "spell"); -- left click
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)
- 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;
+ 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);
+ RegisterStateDriver(frame, "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
- 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;
+ 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);
- 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)
- 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;
+ 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);
+ RegisterUnitWatch(frame);
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;
- 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;
+ 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);
+ RegisterUnitWatch(frame);
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;
- 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;
+ 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);
+ RegisterUnitWatch(frame);
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