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     -- set up indicators
 149     setupIndicators(frame, class);
 150     -- set scripts
 151     frame:SetScript("OnShow", frameShow);
 152     frame:SetScript("OnHide", frameHide);
 153     frame:SetScript("OnEvent", unitEvent);
 154     secure:SetScript("OnEnter", showTooltip);
 155     secure:SetScript("OnLeave", hideTooltip);
 156     -- set attributes
 157     secure:RegisterForClicks("AnyDown");
 158     for attr, val in pairs(attributes) do
 159         secure:SetAttribute(attr, val);
 160     end
 161     -- rest give target and menu
 162     secure:SetAttribute("*type1", "target");
 163     secure:SetAttribute("*type2", "togglemenu");
 164 end
 165
 166 -- vehicle toggle from Shadowed Unit Frames
 167 local vehicletoggle = [=[
 168     local unit = self:GetAttribute("unit");
 169     if unit and newstate == "vehicle" and not UnitTargetsVehicleInRaidUI(unit) then
 170         self:SetAttribute("toggleForVehicle", false);
 171     else
 172         self:SetAttribute("toggleForVehicle", true);
 173     end
 174 ]=]
 175
 176 local function initializeParty(parent)
 177     local secure = CreateFrame("Button", "OmaPartySecure0", parent, inheritedFrames);
 178     local frame = CreateFrame("Frame", "OmaParty0", parent);
 179     local unit = "player";
 180     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
 181     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
 182     setupFrame(frame, secure, unit);
 183     RegisterStateDriver(frame, "visibility", "[@player,group:raid] hide; show");
 184     RegisterStateDriver(secure, "visibility", "[@player,group:raid] hide; show");
 185     RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
 186     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 187     party[0] = {secure=secure, frame=frame};
 188     for i = 1,4 do
 189         local secure = CreateFrame("Button", "OmaPartySecure"..i, parent, inheritedFrames);
 190         local frame = CreateFrame("Frame", "OmaParty"..i, parent);
 191         local unit = "party"..i;
 192         secure:SetPoint("TOPLEFT", party[i-1].secure, "TOPRIGHT");
 193         frame:SetPoint("TOPLEFT", party[i-1].frame, "TOPRIGHT");
 194         setupFrame(frame, secure, unit);
 195         local visible = format("[@%s,exists,group:raid] hide; [@%s,exists] show; hide", unit, unit);
 196         local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 197         RegisterStateDriver(frame, "visibility", visible);
 198         RegisterStateDriver(secure, "visibility", visible);
 199         RegisterStateDriver(secure, "vehicleui", vehicle);
 200         secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 201         party[i] = {secure=secure, frame=frame};
 202     end
 203 end
 204
 205 local function initializeRaid(parent)
 206     local secure = CreateFrame("Button", "OmaRaidSecure1", parent, inheritedFrames);
 207     local frame = CreateFrame("Frame", "OmaRaid1", parent);
 208     local unit = "raid1";
 209     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
 210     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
 211     setupFrame(frame, secure, unit);
 212     RegisterUnitWatch(frame);
 213     RegisterUnitWatch(secure);
 214     local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 215     RegisterStateDriver(secure, "vehicleui", vehicle);
 216     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 217     raid[1] = {secure=secure, frame=frame};
 218     for y = 1,7 do
 219         local i = y*5+1;
 220         local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, inheritedFrames);
 221         local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
 222         local unit = "raid"..i;
 223         secure:SetPoint("TOPLEFT", raid[i-5].secure, "BOTTOMLEFT");
 224         frame:SetPoint("TOPLEFT", raid[i-5].frame, "BOTTOMLEFT");
 225         setupFrame(frame, secure, unit);
 226         RegisterUnitWatch(frame);
 227         RegisterUnitWatch(secure);
 228         local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 229         RegisterStateDriver(secure, "vehicleui", vehicle);
 230         secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 231         raid[i] = {secure=secure, frame=frame};
 232     end
 233     for y = 0,7 do
 234         for x = 1,4 do
 235             local i = y*5+x+1;
 236             local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, inheritedFrames);
 237             local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
 238             local unit = "raid"..i;
 239             secure:SetPoint("TOPLEFT", raid[i-1].secure, "TOPRIGHT");
 240             frame:SetPoint("TOPLEFT", raid[i-1].frame, "TOPRIGHT");
 241             setupFrame(frame, secure, unit);
 242             RegisterUnitWatch(frame);
 243             RegisterUnitWatch(secure);
 244             local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 245             RegisterStateDriver(secure, "vehicleui", vehicle);
 246             secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 247             raid[i] = {secure=secure, frame=frame};
 248         end
 249     end
 250 end
 251
 252 local function initializeFocus(parent)
 253     local secure = CreateFrame("Button", "OmaFocusSecure", parent, inheritedFrames);
 254     local frame = CreateFrame("Frame", "OmaFocus", parent);
 255     local unit = "focus";
 256     secure:SetPoint("BOTTOMLEFT", parent, "TOPLEFT");
 257     frame:SetPoint("BOTTOMLEFT", parent, "TOPLEFT");
 258     setupFrame(frame, secure, unit);
 259     RegisterUnitWatch(frame);
 260     RegisterUnitWatch(secure);
 261     local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
 262     RegisterStateDriver(secure, "vehicleui", vehicle);
 263     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 264 end
 265
 266 local function initialize()
 267     -- let other addons hook these to anchor tooltip elsewhere
 268     GameTooltip = _G["GameTooltip"];
 269     GameTooltip_SetDefaultAnchor = _G["GameTooltip_SetDefaultAnchor"];
 270     _, class = UnitClass("player");
 271     anchorX, anchorY = Settings.Character.AnchorX, Settings.Character.AnchorY;
 272     attributes = Settings.Character.Clickheal;
 273     CFrame:SetPoint("CENTER", nil, "CENTER", anchorX, anchorY);
 274     CFrame:SetHeight((height+2)*8);
 275     CFrame:SetWidth((width+2)*5);
 276     initializeParty(CFrame);
 277     initializeRaid(CFrame);
 278     initializeFocus(CFrame);
 279 end
 280
 281 -- hiding Blizzard frames somewhat based on ShadowedUF
 282 local function hideBlizzardRaidButton()
 283     if InCombatLockdown() then return end
 284     local manager = CompactRaidFrameManager;
 285     -- remove show and unlock buttons from manager
 286     manager.displayFrame.hiddenModeToggle:Hide();
 287     manager.displayFrame.lockedModeToggle:Hide();
 288 end
 289
 290 local hidden = false;
 291 local function hideBlizzardRaid()
 292     if hidden then return end
 293     hidden = true;
 294     local function hide()
 295         if InCombatLockdown() then return end
 296         CompactRaidFrameContainer:Hide();
 297         local shown = CompactRaidFrameManager_GetSetting("IsShown");
 298         if shown and shown ~= "0" then
 299             CompactRaidFrameManager_SetSetting("IsShown", "0");
 300         end
 301     end
 302
 303     CompactRaidFrameContainer:UnregisterAllEvents();
 304     -- there may still be frames created at startup getting events
 305     for i=1,MAX_RAID_MEMBERS do
 306         local frame = _G["CompactRaidFrame"..i];
 307         if not frame then break end
 308         frame:UnregisterAllEvents();
 309     end
 310     hooksecurefunc("CompactRaidFrameManager_UpdateShown", hide);
 311     hide();
 312     CompactRaidFrameContainer:HookScript("OnShow", hide);
 313     hooksecurefunc("CompactRaidFrameManager_UpdateOptionsFlowContainer", hideBlizzardRaidButton);
 314     hideBlizzardRaidButton();
 315
 316     -- hide focus frame
 317     for _, frame in pairs({FocusFrame, FocusFrameToT}) do
 318         frame:UnregisterAllEvents();
 319         frame.healthbar:UnregisterAllEvents();
 320         frame.manabar:UnregisterAllEvents();
 321         if frame.spellbar then frame.spellbar:UnregisterAllEvents() end
 322         if frame.powerBarAlt then frame.powerBarAlt:UnregisterAllEvents() end
 323         frame:Hide();
 324     end
 325 end
 326
 327 CFrame:RegisterEvent("PLAYER_LOGIN");
 328 CFrame:RegisterEvent("PLAYER_REGEN_ENABLED");
 329 CFrame:SetScript("OnEvent", function(self, event, addon)
 330     if event == "PLAYER_LOGIN" then
 331         OmaRFLoadChar();
 332         hideBlizzardRaid();
 333         initialize();
 334     elseif event == "PLAYER_REGEN_ENABLED" then
 335         -- if CompactRaidFrameManager_UpdateOptionsFlowContainer was called in combat
 336         -- couldn't hide the button
 337         hideBlizzardRaidButton();
 338     end
 339 end);