OmaRF/RaidFrame.lua

   1 -- RaidFrame.lua
   2 local _;
   3 local unpack, pairs = unpack, pairs;
   4 local format = string.format;
   5 local UnitInRange = UnitInRange;
   6 local InCombatLockdown = InCombatLockdown;
   7 local CreateFrame, RegisterStateDriver, RegisterUnitWatch = CreateFrame, RegisterStateDriver, RegisterUnitWatch;
   8 local STANDARD_TEXT_FONT = STANDARD_TEXT_FONT;
   9 local GameTooltip = nil;
  10 local GameTooltip_SetDefaultAnchor = nil;
  11
  12 local registerEvents = OmaRFEvents.RegisterEvents;
  13 local registerUnitEvents = OmaRFEvents.RegisterUnitEvents;
  14 local unitEvent = OmaRFEvents.UnitEvent;
  15 local setupIndicators = OmaRFIndicators.SetupIndicators;
  16
  17 local Settings = OmaRFSettings;
  18 local baseColor = Settings.BaseColor;
  19 local bgColor = Settings.BgColor;
  20 local healthColor = Settings.HealthColor;
  21 local shieldColor = Settings.ShieldColor;
  22 local shieldhlColor = Settings.ShieldhlColor;
  23 local healpredColor = Settings.HealpredColor;
  24 local healabsorbColor = Settings.HealabsorbColor;
  25 local width, height = Settings.Width, Settings.Height;
  26 -- placeholders with visible values when error happens
  27 local anchorX, anchorY = 10, 10;
  28 local attributes = {};
  29
  30 local CFrame = CreateFrame("Frame", "OmaRFFrame", UIParent);
  31 local inheritedFrames = "SecureUnitButtonTemplate,SecureHandlerStateTemplate";
  32 local class = nil;
  33 local party = {};
  34 local raid = {};
  35
  36 local M = {};
  37 OmaRaidFrame = M;
  38
  39 local function unitUpdate(frame)
  40     -- there's no in/out of range event, have to check each frame
  41     -- from FrameXML/CompactUnitFrame.lua
  42     local inRange, checked = UnitInRange(frame.displayed);
  43     if checked and not inRange then
  44         frame:SetAlpha(0.55);
  45     else
  46         frame:SetAlpha(1);
  47     end
  48 end
  49
  50 local function frameShow(frame)
  51     registerEvents(frame);
  52     registerUnitEvents(frame);
  53     frame:SetScript("OnUpdate", unitUpdate);
  54     unitEvent(frame, "UPDATE_ALL_BARS");
  55 end
  56
  57 local function frameHide(frame)
  58     frame:UnregisterAllEvents();
  59     frame:SetScript("OnUpdate", nil);
  60 end
  61
  62 local function showTooltip(secure)
  63     GameTooltip_SetDefaultAnchor(GameTooltip, secure);
  64     GameTooltip:SetUnit(secure:GetAttribute("unit"));
  65 end
  66
  67 local function hideTooltip(secure)
  68     GameTooltip:FadeOut();
  69 end
  70
  71 local function setupFrame(frame, secure, unit)
  72     secure:SetAttribute("unit", unit);
  73     frame:SetAttribute("unit", unit);
  74     frame.unit = unit;
  75     frame.displayed = unit;
  76     -- hide frame to get initial frameShow call
  77     frame:Hide();
  78     if unit == "player" then frame.vehicle = "vehicle"
  79     else frame.vehicle = unit.."pet" end
  80     -- create visuals
  81     secure:SetWidth(width+2);
  82     secure:SetHeight(height+2);
  83     frame:SetWidth(width+2);
  84     frame:SetHeight(height+2);
  85     frame.base = frame:CreateTexture(nil, "BACKGROUND");
  86     frame.base:SetAllPoints();
  87     frame.base:SetColorTexture(1, 1, 1);
  88     frame.base:SetVertexColor(unpack(baseColor));
  89     frame.background = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
  90     frame.background:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
  91     frame.background:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
  92     frame.background:SetColorTexture(unpack(bgColor));
  93     frame.health = frame:CreateTexture(nil, "BORDER");
  94     frame.health:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
  95     frame.health:SetPoint("TOPLEFT", frame.background, "TOPLEFT");
  96     frame.health:SetPoint("BOTTOMLEFT", frame.background, "BOTTOMLEFT");
  97     frame.health:SetVertexColor(unpack(healthColor));
  98     frame.mana = frame:CreateTexture(nil, "BORDER");
  99     frame.mana:SetPoint("TOPLEFT", frame.background, "BOTTOMLEFT", 0, 2);
 100     frame.mana:SetPoint("BOTTOMLEFT", frame.background, "BOTTOMLEFT");
 101     frame.mana:SetColorTexture(1, 1, 1);
 102     frame.mana:Hide();
 103     frame.shield = frame:CreateTexture(nil, "BORDER");
 104     frame.shield:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
 105     frame.shield:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
 106     frame.shield:SetColorTexture(unpack(shieldColor));
 107     frame.shield:Hide();
 108     frame.shieldhl = frame:CreateTexture(nil, "ARTWORK");
 109     frame.shieldhl:SetPoint("TOPLEFT", frame.background, "TOPRIGHT", -1, 0);
 110     frame.shieldhl:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", 1, 0);
 111     frame.shieldhl:SetColorTexture(unpack(shieldhlColor));
 112     frame.shieldhl:Hide();
 113     frame.healpred = frame:CreateTexture(nil, "ARTWORK");
 114     frame.healpred:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
 115     frame.healpred:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
 116     frame.healpred:SetColorTexture(unpack(healpredColor));
 117     frame.healpred:Hide();
 118     frame.healabsorb = frame:CreateTexture(nil, "ARTWORK");
 119     frame.healabsorb:SetPoint("TOPRIGHT", frame.health, "TOPRIGHT");
 120     frame.healabsorb:SetPoint("BOTTOMRIGHT", frame.health, "BOTTOMRIGHT");
 121     frame.healabsorb:SetColorTexture(unpack(healabsorbColor));
 122     frame.healabsorb:Hide();
 123     frame.role = frame:CreateTexture(nil, "ARTWORK");
 124     frame.role:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", -2, 2);
 125     frame.role:SetPoint("TOPLEFT", frame.background, "BOTTOMRIGHT", -14, 14);
 126     frame.role:SetTexture("Interface\\LFGFRAME\\LFGROLE");
 127     frame.role:Hide();
 128     frame.overlay = frame:CreateTexture(nil, "ARTWORK", nil, 1);
 129     frame.overlay:SetPoint("TOPLEFT", frame.background, "TOPLEFT");
 130     frame.overlay:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT");
 131     frame.overlay:SetColorTexture(1, 1, 1);
 132     frame.overlay:Hide();
 133     frame.name = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
 134     frame.name:SetPoint("CENTER", frame.background, "CENTER", 0, 11);
 135     frame.text = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
 136     frame.text:SetFont(STANDARD_TEXT_FONT, 13);
 137     frame.text:SetPoint("CENTER", frame.background, "CENTER", 0, -1);
 138     frame.text:Hide();
 139     frame.rez = frame:CreateTexture(nil, "OVERLAY");
 140     frame.rez:SetPoint("TOPLEFT", frame.background, "CENTER", -12, 12);
 141     frame.rez:SetPoint("BOTTOMRIGHT", frame.background, "CENTER", 12, -12);
 142     frame.rez:SetTexture("Interface\\RaidFrame\\Raid-Icon-Rez");
 143     frame.rez:Hide();
 144     frame.ready = frame:CreateTexture(nil, "OVERLAY");
 145     frame.ready:SetPoint("TOPLEFT", frame.background, "BOTTOMLEFT", 0, 14);
 146     frame.ready:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMLEFT", 14, 0);
 147     frame.ready:Hide();
 148     frame.targeticon = frame:CreateTexture(nil, "OVERLAY");
 149     frame.targeticon:SetPoint("CENTER", frame.background, "TOP");
 150     frame.targeticon:SetWidth(12);
 151     frame.targeticon:SetHeight(12);
 152     frame.targeticon:SetTexture("Interface\\TARGETINGFRAME\\UI-RaidTargetingIcons");
 153     frame.targeticon:Hide();
 154     -- set up indicators
 155     setupIndicators(frame, class);
 156     -- set scripts
 157     frame:SetScript("OnShow", frameShow);
 158     frame:SetScript("OnHide", frameHide);
 159     frame:SetScript("OnEvent", unitEvent);
 160     secure:SetScript("OnEnter", showTooltip);
 161     secure:SetScript("OnLeave", hideTooltip);
 162     -- set attributes
 163     secure:RegisterForClicks("AnyDown");
 164     for attr, val in pairs(attributes) do
 165         secure:SetAttribute(attr, val);
 166     end
 167     -- rest give target and menu
 168     secure:SetAttribute("*type1", "target");
 169     secure:SetAttribute("*type2", "togglemenu");
 170 end
 171
 172 -- vehicle toggle from Shadowed Unit Frames
 173 local vehicletoggle = [=[
 174     local unit = self:GetAttribute("unit");
 175     if unit and newstate == "vehicle" and not UnitTargetsVehicleInRaidUI(unit) then
 176         self:SetAttribute("toggleForVehicle", false);
 177     else
 178         self:SetAttribute("toggleForVehicle", true);
 179     end
 180 ]=]
 181
 182 local function initializeParty(parent)
 183     local secure = CreateFrame("Button", "OmaPartySecure0", parent, inheritedFrames);
 184     local frame = CreateFrame("Frame", "OmaParty0", parent);
 185     local unit = "player";
 186     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
 187     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
 188     setupFrame(frame, secure, unit);
 189     RegisterStateDriver(frame, "visibility", "[@player,group:raid] hide; show");
 190     RegisterStateDriver(secure, "visibility", "[@player,group:raid] hide; show");
 191     RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
 192     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 193     party[0] = {secure=secure, frame=frame};
 194     for i = 1,4 do
 195         local secure = CreateFrame("Button", "OmaPartySecure"..i, parent, inheritedFrames);
 196         local frame = CreateFrame("Frame", "OmaParty"..i, parent);
 197         local unit = "party"..i;
 198         secure:SetPoint("TOPLEFT", party[i-1].secure, "TOPRIGHT");
 199         frame:SetPoint("TOPLEFT", party[i-1].frame, "TOPRIGHT");
 200         setupFrame(frame, secure, unit);
 201         local visible = format("[@%s,exists,group:raid] hide; [@%s,exists] show; hide", unit, unit);
 202         local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 203         RegisterStateDriver(frame, "visibility", visible);
 204         RegisterStateDriver(secure, "visibility", visible);
 205         RegisterStateDriver(secure, "vehicleui", vehicle);
 206         secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 207         party[i] = {secure=secure, frame=frame};
 208     end
 209 end
 210
 211 local function initializeRaid(parent)
 212     local secure = CreateFrame("Button", "OmaRaidSecure1", parent, inheritedFrames);
 213     local frame = CreateFrame("Frame", "OmaRaid1", parent);
 214     local unit = "raid1";
 215     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
 216     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
 217     setupFrame(frame, secure, unit);
 218     RegisterUnitWatch(frame);
 219     RegisterUnitWatch(secure);
 220     local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 221     RegisterStateDriver(secure, "vehicleui", vehicle);
 222     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 223     raid[1] = {secure=secure, frame=frame};
 224     for y = 1,7 do
 225         local i = y*5+1;
 226         local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, inheritedFrames);
 227         local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
 228         local unit = "raid"..i;
 229         secure:SetPoint("TOPLEFT", raid[i-5].secure, "BOTTOMLEFT");
 230         frame:SetPoint("TOPLEFT", raid[i-5].frame, "BOTTOMLEFT");
 231         setupFrame(frame, secure, unit);
 232         RegisterUnitWatch(frame);
 233         RegisterUnitWatch(secure);
 234         local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 235         RegisterStateDriver(secure, "vehicleui", vehicle);
 236         secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 237         raid[i] = {secure=secure, frame=frame};
 238     end
 239     for y = 0,7 do
 240         for x = 1,4 do
 241             local i = y*5+x+1;
 242             local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, inheritedFrames);
 243             local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
 244             local unit = "raid"..i;
 245             secure:SetPoint("TOPLEFT", raid[i-1].secure, "TOPRIGHT");
 246             frame:SetPoint("TOPLEFT", raid[i-1].frame, "TOPRIGHT");
 247             setupFrame(frame, secure, unit);
 248             RegisterUnitWatch(frame);
 249             RegisterUnitWatch(secure);
 250             local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 251             RegisterStateDriver(secure, "vehicleui", vehicle);
 252             secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 253             raid[i] = {secure=secure, frame=frame};
 254         end
 255     end
 256 end
 257
 258 local function initializeFocus(parent)
 259     local secure = CreateFrame("Button", "OmaFocusSecure", parent, inheritedFrames);
 260     local frame = CreateFrame("Frame", "OmaFocus", parent);
 261     local unit = "focus";
 262     secure:SetPoint("BOTTOMLEFT", parent, "TOPLEFT");
 263     frame:SetPoint("BOTTOMLEFT", parent, "TOPLEFT");
 264     setupFrame(frame, secure, unit);
 265     RegisterUnitWatch(frame);
 266     RegisterUnitWatch(secure);
 267     local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 268     RegisterStateDriver(secure, "vehicleui", vehicle);
 269     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 270 end
 271
 272 local function initialize()
 273     -- let other addons hook these to anchor tooltip elsewhere
 274     GameTooltip = _G["GameTooltip"];
 275     GameTooltip_SetDefaultAnchor = _G["GameTooltip_SetDefaultAnchor"];
 276     _, class = UnitClass("player");
 277     anchorX, anchorY = Settings.Character.AnchorX, Settings.Character.AnchorY;
 278     attributes = Settings.Character.Clickheal;
 279     CFrame:SetPoint("CENTER", nil, "CENTER", anchorX, anchorY);
 280     CFrame:SetHeight((height+2)*8);
 281     CFrame:SetWidth((width+2)*5);
 282     initializeParty(CFrame);
 283     initializeRaid(CFrame);
 284     initializeFocus(CFrame);
 285 end
 286
 287 -- hiding Blizzard frames somewhat based on ShadowedUF
 288 local function hideBlizzardRaidButton()
 289     if InCombatLockdown() then return end
 290     local manager = CompactRaidFrameManager;
 291     -- remove show and unlock buttons from manager
 292     manager.displayFrame.hiddenModeToggle:Hide();
 293     manager.displayFrame.lockedModeToggle:Hide();
 294 end
 295
 296 local hidden = false;
 297 local function hideBlizzardRaid()
 298     if hidden then return end
 299     hidden = true;
 300     local function hide()
 301         if InCombatLockdown() then return end
 302         CompactRaidFrameContainer:Hide();
 303         local shown = CompactRaidFrameManager_GetSetting("IsShown");
 304         if shown and shown ~= "0" then
 305             CompactRaidFrameManager_SetSetting("IsShown", "0");
 306         end
 307     end
 308
 309     CompactRaidFrameContainer:UnregisterAllEvents();
 310     -- there may still be frames created at startup getting events
 311     for i=1,MAX_RAID_MEMBERS do
 312         local frame = _G["CompactRaidFrame"..i];
 313         if not frame then break end
 314         frame:UnregisterAllEvents();
 315     end
 316     hooksecurefunc("CompactRaidFrameManager_UpdateShown", hide);
 317     hide();
 318     CompactRaidFrameContainer:HookScript("OnShow", hide);
 319     hooksecurefunc("CompactRaidFrameManager_UpdateOptionsFlowContainer", hideBlizzardRaidButton);
 320     hideBlizzardRaidButton();
 321
 322     -- hide focus frame
 323     for _, frame in pairs({FocusFrame, FocusFrameToT}) do
 324         frame:UnregisterAllEvents();
 325         frame.healthbar:UnregisterAllEvents();
 326         frame.manabar:UnregisterAllEvents();
 327         if frame.spellbar then frame.spellbar:UnregisterAllEvents() end
 328         if frame.powerBarAlt then frame.powerBarAlt:UnregisterAllEvents() end
 329         frame:Hide();
 330     end
 331 end
 332
 333 CFrame:RegisterEvent("PLAYER_LOGIN");
 334 CFrame:RegisterEvent("PLAYER_REGEN_ENABLED");
 335 CFrame:SetScript("OnEvent", function(self, event, addon)
 336     if event == "PLAYER_LOGIN" then
 337         OmaRFLoadChar();
 338         hideBlizzardRaid();
 339         initialize();
 340     elseif event == "PLAYER_REGEN_ENABLED" then
 341         -- if CompactRaidFrameManager_UpdateOptionsFlowContainer was called in combat
 342         -- couldn't hide the button
 343         hideBlizzardRaidButton();
 344     end
 345 end);