X-Git-Url: https://www.aleksib.fi/git/wowui.git/blobdiff_plain/a0d7f3ac4eccb0300ead1d3c745b9b2587351c7e..92fe6f25185f79a8c40a382d1e420fbea08f12f5:/OmaRF/FrameInit.lua diff --git a/OmaRF/FrameInit.lua b/OmaRF/FrameInit.lua index 51c59e3..df34967 100644 --- a/OmaRF/FrameInit.lua +++ b/OmaRF/FrameInit.lua @@ -1,7 +1,9 @@ -- 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 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 unitEvent = nil; + 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.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}; @@ -86,13 +89,64 @@ local function setupIndicators(frame) 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); @@ -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:SetColorTexture(1, 1, 1); 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); - -- 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 @@ -181,87 +221,135 @@ local function setupFrame(frame, secure, unit) 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