OmaRF/RaidFrame.lua

   1 -- RaidFrame.lua
   2 local _;
   3 local unpack, pairs, rawget = unpack, pairs, rawget;
   4 local format = string.format;
   5 local UnitHealthMax, UnitPowerMax = UnitHealthMax, UnitPowerMax;
   6 local UnitInRange = UnitInRange;
   7 local InCombatLockdown = InCombatLockdown;
   8 local CreateFrame, RegisterStateDriver, RegisterUnitWatch = CreateFrame, RegisterStateDriver, RegisterUnitWatch;
   9 local STANDARD_TEXT_FONT = STANDARD_TEXT_FONT;
  10 local GameTooltip = nil;
  11 local GameTooltip_SetDefaultAnchor = nil;
  12
  13 local registerEvents = OmaRFEvents.RegisterEvents;
  14 local registerUnitEvents = OmaRFEvents.RegisterUnitEvents;
  15 local unitEvent = OmaRFEvents.UnitEvent;
  16 local setupIndicators = OmaRFIndicators.SetupIndicators;
  17
  18 local Settings = OmaRFSettings;
  19 local baseColor = Settings.BaseColor;
  20 local bgColor = Settings.BgColor;
  21 local healthColor = Settings.HealthColor;
  22 local shieldColor = Settings.ShieldColor;
  23 local shieldhlColor = Settings.ShieldhlColor;
  24 local healpredColor = Settings.HealpredColor;
  25 local healabsorbColor = Settings.HealabsorbColor;
  26 local width, height = Settings.Width, Settings.Height;
  27 -- placeholders with visible values when error happens
  28 local anchorX, anchorY = 10, 10;
  29 local attributes = {};
  30
  31 local CFrame = CreateFrame("Frame", "OmaRFFrame", UIParent);
  32 local inheritedFrames = "SecureUnitButtonTemplate,SecureHandlerStateTemplate";
  33 local party = {};
  34 local raid = {};
  35
  36 local M = {};
  37 OmaRaidFrame = M;
  38
  39 local function unitUpdate(self, elapsed)
  40     -- there's no in/out of range event, have to check each frame
  41     -- from FrameXML/CompactUnitFrame.lua
  42     local inRange, checked = UnitInRange(self.displayed);
  43     if checked and not inRange then
  44         self:SetAlpha(0.55);
  45     else
  46         self: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);
 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 loadCharSettings()
 267     anchorX, anchorY = Settings.Character.AnchorX, Settings.Character.AnchorY;
 268     attributes = Settings.Character.Clickheal;
 269 end
 270
 271 local function initialize()
 272     -- let other addons hook these to anchor tooltip elsewhere
 273     GameTooltip = _G["GameTooltip"];
 274     GameTooltip_SetDefaultAnchor = _G["GameTooltip_SetDefaultAnchor"];
 275     loadCharSettings();
 276     CFrame:SetPoint("CENTER", nil, "CENTER", anchorX, anchorY);
 277     CFrame:SetHeight((height+2)*8);
 278     CFrame:SetWidth((width+2)*5);
 279     initializeParty(CFrame);
 280     initializeRaid(CFrame);
 281     initializeFocus(CFrame);
 282 end
 283
 284 -- hiding Blizzard frames somewhat based on ShadowedUF
 285 local function hideBlizzardRaidButton()
 286     if InCombatLockdown() then return end
 287     local manager = CompactRaidFrameManager;
 288     -- remove show and unlock buttons from manager
 289     manager.displayFrame.hiddenModeToggle:Hide();
 290     manager.displayFrame.lockedModeToggle:Hide();
 291 end
 292
 293 local hidden = false;
 294 local function hideBlizzardRaid()
 295     if hidden then return end
 296     hidden = true;
 297     local function hide()
 298         if InCombatLockdown() then return end
 299         CompactRaidFrameContainer:Hide();
 300         local shown = CompactRaidFrameManager_GetSetting("IsShown");
 301         if shown and shown ~= "0" then
 302             CompactRaidFrameManager_SetSetting("IsShown", "0");
 303         end
 304     end
 305
 306     CompactRaidFrameContainer:UnregisterAllEvents();
 307     -- there may still be frames created at startup getting events
 308     for i=1,MAX_RAID_MEMBERS do
 309         local frame = _G["CompactRaidFrame"..i];
 310         if not frame then break end
 311         frame:UnregisterAllEvents();
 312     end
 313     hooksecurefunc("CompactRaidFrameManager_UpdateShown", hide);
 314     hide();
 315     CompactRaidFrameContainer:HookScript("OnShow", hide);
 316     hooksecurefunc("CompactRaidFrameManager_UpdateOptionsFlowContainer", hideBlizzardRaidButton);
 317     hideBlizzardRaidButton();
 318
 319     -- hide focus frame
 320     for _, frame in pairs({FocusFrame, FocusFrameToT}) do
 321         frame:UnregisterAllEvents();
 322         frame.healthbar:UnregisterAllEvents();
 323         frame.manabar:UnregisterAllEvents();
 324         if frame.spellbar then frame.spellbar:UnregisterAllEvents() end
 325         if frame.powerBarAlt then frame.powerBarAlt:UnregisterAllEvents() end
 326         frame:Hide();
 327     end
 328 end
 329
 330 CFrame:RegisterEvent("PLAYER_LOGIN");
 331 CFrame:RegisterEvent("PLAYER_REGEN_ENABLED");
 332 CFrame:SetScript("OnEvent", function(self, event, addon)
 333     if event == "PLAYER_LOGIN" then
 334         OmaRFLoadChar();
 335         hideBlizzardRaid();
 336         initialize();
 337     elseif event == "PLAYER_REGEN_ENABLED" then
 338         -- if CompactRaidFrameManager_UpdateOptionsFlowContainer was called in combat
 339         -- couldn't hide the button
 340         hideBlizzardRaidButton();
 341     end
 342 end);