52e5175 - Fix power bars, remove overlay color for basic unit frames
[wowui.git] / 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 unitEvent = OmaRFEvents.UnitEvent;
15
16 local Settings = OmaRFSettings;
17 local indSize = Settings.IndSize;
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 positions = {};
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 setupIndicators(frame)
40     frame.inds = CreateFrame("Frame", nil, frame);
41     frame.inds:SetAllPoints();
42     frame.inds:Hide();
43     for _, pos in pairs(positions) do
44         frame.inds[pos] = frame.inds:CreateTexture(nil, "OVERLAY");
45         frame.inds[pos]:SetPoint(pos, frame.inds, pos);
46         frame.inds[pos]:SetWidth(indSize);
47         frame.inds[pos]:SetHeight(indSize);
48         frame.inds[pos]:SetTexture("Interface\\AddOns\\OmaRF\\images\\rhomb");
49         frame.inds[pos]:SetVertexColor(1, 0, 0);
50         frame.inds[pos]:Hide();
51         frame.inds[pos].text = frame.inds:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
52         frame.inds[pos].text:SetPoint("BOTTOMRIGHT", frame.inds[pos], "BOTTOMRIGHT");
53         frame.inds[pos].text:Hide();
54     end
55     frame.major = CreateFrame("Frame", nil, frame);
56     frame.major:SetPoint("TOPLEFT", frame, "TOPLEFT", 4, -indSize + 4);
57     frame.major:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT");
58     for i = 1,3 do
59         frame.major[i] = frame.major:CreateTexture(nil, "OVERLAY");
60         if i == 1 then
61             frame.major[i]:SetPoint("TOPLEFT", frame.major, "TOPLEFT");
62         else
63             frame.major[i]:SetPoint("TOPLEFT", frame.major[i-1], "TOPRIGHT");
64         end
65         frame.major[i]:SetWidth(indSize*2);
66         frame.major[i]:SetHeight(indSize*2);
67         frame.major[i]:Hide();
68         frame.major[i].text = frame.major:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
69         frame.major[i].text:SetPoint("BOTTOMRIGHT", frame.major[i], "BOTTOMRIGHT");
70         frame.major[i].text:Hide();
71         frame.major[i].stack = frame.major:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
72         frame.major[i].stack:SetPoint("TOPLEFT", frame.major[i], "TOPLEFT");
73         frame.major[i].stack:Hide();
74     end
75 end
76
77 local function unitUpdate(self, elapsed)
78     -- there's no in/out of range event, have to check each frame
79     -- from FrameXML/CompactUnitFrame.lua
80     local inRange, checked = UnitInRange(self.displayed);
81     if checked and not inRange then
82         self:SetAlpha(0.55);
83     else
84         self:SetAlpha(1);
85     end
86 end
87
88 local function frameShow(frame)
89     frame:RegisterEvent("PARTY_MEMBER_ENABLE");
90     frame:RegisterEvent("PARTY_MEMBER_DISABLE");
91     frame:RegisterEvent("UNIT_ENTERED_VEHICLE");
92     frame:RegisterEvent("UNIT_EXITED_VEHICLE");
93     frame:RegisterEvent("UNIT_PET");
94     frame:RegisterEvent("PLAYER_ROLES_ASSIGNED");
95     frame:RegisterEvent("GROUP_ROSTER_UPDATE");
96     frame:RegisterEvent("PLAYER_ENTERING_WORLD");
97     frame:RegisterEvent("READY_CHECK");
98     frame:RegisterEvent("READY_CHECK_CONFIRM");
99     frame:RegisterEvent("READY_CHECK_FINISHED");
100     if frame.unit == "focus" then frame:RegisterEvent("PLAYER_FOCUS_CHANGED") end
101     registerEvents(frame);
102     frame:SetScript("OnUpdate", unitUpdate);
103     unitEvent(frame, "UPDATE_ALL_BARS", frame.displayed);
104 end
105
106 local function frameHide(frame)
107     frame:UnregisterAllEvents();
108     frame:SetScript("OnUpdate", nil);
109 end
110
111 local function showTooltip(frame)
112     GameTooltip_SetDefaultAnchor(GameTooltip, frame);
113     GameTooltip:SetUnit(frame:GetAttribute("unit"));
114 end
115
116 local function hideTooltip(frame)
117     GameTooltip:FadeOut();
118 end
119
120 local function setupFrame(frame, secure, unit)
121     secure:SetAttribute("unit", unit);
122     frame:SetAttribute("unit", unit);
123     frame.unit = unit;
124     frame.displayed = unit;
125     -- hide frame to get initial frameShow call
126     frame:Hide();
127     if unit == "player" then
128         frame.vehicle = "vehicle";
129     else
130         frame.vehicle = unit.."pet";
131     end
132     -- create visuals
133     secure:SetWidth(width+2);
134     secure:SetHeight(height+2);
135     frame:SetWidth(width+2);
136     frame:SetHeight(height+2);
137     frame.base = frame:CreateTexture(nil, "BACKGROUND");
138     frame.base:SetAllPoints();
139     frame.base:SetColorTexture(1, 1, 1);
140     frame.base:SetVertexColor(unpack(baseColor));
141     frame.background = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
142     frame.background:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
143     frame.background:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
144     frame.background:SetColorTexture(unpack(bgColor));
145     frame.health = frame:CreateTexture(nil, "BORDER");
146     frame.health:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
147     frame.health:SetPoint("TOPLEFT", frame.background, "TOPLEFT");
148     frame.health:SetPoint("BOTTOMLEFT", frame.background, "BOTTOMLEFT", 0, 2);
149     frame.health:SetVertexColor(unpack(healthColor));
150     frame.health:SetWidth(width);
151     frame.health.max = UnitHealthMax(unit);
152     frame.mana = frame:CreateTexture(nil, "BORDER");
153     frame.mana:SetPoint("TOPLEFT", frame.background, "BOTTOMLEFT", 0, 2);
154     frame.mana:SetPoint("BOTTOMLEFT", frame.background, "BOTTOMLEFT");
155     frame.mana:SetColorTexture(1, 1, 1);
156     frame.mana:SetWidth(width);
157     frame.mana.max = UnitPowerMax(unit);
158     frame.shield = frame:CreateTexture(nil, "BORDER");
159     frame.shield:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
160     frame.shield:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
161     frame.shield:SetColorTexture(unpack(shieldColor));
162     frame.shield:Hide();
163     frame.shieldhl = frame:CreateTexture(nil, "ARTWORK");
164     frame.shieldhl:SetPoint("TOPLEFT", frame.background, "TOPRIGHT", -1, 0);
165     frame.shieldhl:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", 1, 0);
166     frame.shieldhl:SetColorTexture(unpack(shieldhlColor));
167     frame.shieldhl:Hide();
168     frame.healpred = frame:CreateTexture(nil, "ARTWORK");
169     frame.healpred:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
170     frame.healpred:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
171     frame.healpred:SetColorTexture(unpack(healpredColor));
172     frame.healpred:Hide();
173     frame.healabsorb = frame:CreateTexture(nil, "ARTWORK");
174     frame.healabsorb:SetPoint("TOPRIGHT", frame.health, "TOPRIGHT");
175     frame.healabsorb:SetPoint("BOTTOMRIGHT", frame.health, "BOTTOMRIGHT");
176     frame.healabsorb:SetColorTexture(unpack(healabsorbColor));
177     frame.healabsorb:Hide();
178     frame.role = frame:CreateTexture(nil, "ARTWORK");
179     frame.role:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", -2, 2);
180     frame.role:SetPoint("TOPLEFT", frame.background, "BOTTOMRIGHT", -14, 14);
181     frame.role:SetTexCoord(0.75, 1, 0, 1);
182     frame.role:SetTexture("Interface\\LFGFRAME\\LFGROLE");
183     frame.role:Hide();
184     frame.overlay = frame:CreateTexture(nil, "ARTWORK", nil, 1);
185     frame.overlay:SetPoint("TOPLEFT", frame.background, "TOPLEFT");
186     frame.overlay:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMRIGHT", 0, 2);
187     frame.overlay:SetColorTexture(1, 1, 1);
188     frame.overlay:Hide();
189     frame.name = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
190     frame.name:SetPoint("CENTER", frame.background, "CENTER", 0, 11);
191     frame.text = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
192     frame.text:SetFont(STANDARD_TEXT_FONT, 13);
193     frame.text:SetPoint("CENTER", frame.background, "CENTER", 0, -1);
194     frame.text:Hide();
195     frame.rez = frame:CreateTexture(nil, "OVERLAY");
196     frame.rez:SetPoint("TOPLEFT", frame.background, "CENTER", -12, 12);
197     frame.rez:SetPoint("BOTTOMRIGHT", frame.background, "CENTER", 12, -12);
198     frame.rez:SetTexture("Interface\\RaidFrame\\Raid-Icon-Rez");
199     frame.rez:Hide();
200     frame.ready = frame:CreateTexture(nil, "OVERLAY");
201     frame.ready:SetPoint("TOPLEFT", frame.background, "BOTTOMLEFT", 0, 12);
202     frame.ready:SetPoint("BOTTOMRIGHT", frame.background, "BOTTOMLEFT", 12, 0);
203     frame.ready:Hide();
204     -- set up indicators
205     setupIndicators(frame);
206     -- set scripts
207     frame:SetScript("OnShow", frameShow);
208     frame:SetScript("OnHide", frameHide);
209     frame:SetScript("OnEvent", unitEvent);
210     -- let other addons hook these to anchor tooltip elsewhere
211     GameTooltip = _G["GameTooltip"];
212     GameTooltip_SetDefaultAnchor = _G["GameTooltip_SetDefaultAnchor"];
213     secure:SetScript("OnEnter", showTooltip);
214     secure:SetScript("OnLeave", hideTooltip);
215     -- set attributes
216     secure:RegisterForClicks("AnyDown");
217     for attr, val in pairs(attributes) do
218         secure:SetAttribute(attr, val);
219     end
220     -- rest give target and menu
221     secure:SetAttribute("*type1", "target");
222     secure:SetAttribute("*type2", "togglemenu");
223 end
224
225 -- vehicle toggle from Shadowed Unit Frames
226 local vehicletoggle = [=[
227     local unit = self:GetAttribute("unit");
228     if unit and newstate == "vehicle" and not UnitTargetsVehicleInRaidUI(unit) then
229         self:SetAttribute("toggleForVehicle", false);
230     else
231         self:SetAttribute("toggleForVehicle", true);
232     end
233 ]=]
234
235 local function initializeParty(parent)
236     local secure = CreateFrame("Button", "OmaPartySecure0", parent, inheritedFrames);
237     local frame = CreateFrame("Frame", "OmaParty0", parent);
238     local unit = "player";
239     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
240     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
241     setupFrame(frame, secure, unit);
242     RegisterStateDriver(frame, "visibility", "[@player,group:raid] hide; show");
243     RegisterStateDriver(secure, "visibility", "[@player,group:raid] hide; show");
244     RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
245     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
246     party[0] = {secure=secure, frame=frame};
247     for i = 1,4 do
248         local secure = CreateFrame("Button", "OmaPartySecure"..i, parent, inheritedFrames);
249         local frame = CreateFrame("Frame", "OmaParty"..i, parent);
250         local unit = "party"..i;
251         secure:SetPoint("TOPLEFT", party[i-1].secure, "TOPRIGHT");
252         frame:SetPoint("TOPLEFT", party[i-1].frame, "TOPRIGHT");
253         setupFrame(frame, secure, unit);
254         local visible = format("[@%s,exists,group:raid] hide; [@%s,exists] show; hide", unit, unit);
255         local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
256         RegisterStateDriver(frame, "visibility", visible);
257         RegisterStateDriver(secure, "visibility", visible);
258         RegisterStateDriver(secure, "vehicleui", vehicle);
259         secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
260         party[i] = {secure=secure, frame=frame};
261     end
262 end
263
264 local function initializeRaid(parent)
265     local secure = CreateFrame("Button", "OmaRaidSecure1", parent, inheritedFrames);
266     local frame = CreateFrame("Frame", "OmaRaid1", parent);
267     local unit = "raid1";
268     secure:SetPoint("TOPLEFT", parent, "TOPLEFT");
269     frame:SetPoint("TOPLEFT", parent, "TOPLEFT");
270     setupFrame(frame, secure, unit);
271     RegisterUnitWatch(frame);
272     RegisterUnitWatch(secure);
273     local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
274     RegisterStateDriver(secure, "vehicleui", vehicle);
275     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
276     raid[1] = {secure=secure, frame=frame};
277     for y = 1,7 do
278         local i = y*5+1;
279         local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, inheritedFrames);
280         local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
281         local unit = "raid"..i;
282         secure:SetPoint("TOPLEFT", raid[i-5].secure, "BOTTOMLEFT");
283         frame:SetPoint("TOPLEFT", raid[i-5].frame, "BOTTOMLEFT");
284         setupFrame(frame, secure, unit);
285         RegisterUnitWatch(frame);
286         RegisterUnitWatch(secure);
287         local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
288         RegisterStateDriver(secure, "vehicleui", vehicle);
289         secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
290         raid[i] = {secure=secure, frame=frame};
291     end
292     for y = 0,7 do
293         for x = 1,4 do
294             local i = y*5+x+1;
295             local secure = CreateFrame("Button", "OmaRaidSecure"..i, parent, inheritedFrames);
296             local frame = CreateFrame("Frame", "OmaRaid"..i, parent);
297             local unit = "raid"..i;
298             secure:SetPoint("TOPLEFT", raid[i-1].secure, "TOPRIGHT");
299             frame:SetPoint("TOPLEFT", raid[i-1].frame, "TOPRIGHT");
300             setupFrame(frame, secure, unit);
301             RegisterUnitWatch(frame);
302             RegisterUnitWatch(secure);
303             local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
304             RegisterStateDriver(secure, "vehicleui", vehicle);
305             secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
306             raid[i] = {secure=secure, frame=frame};
307         end
308     end
309 end
310
311 local function initializeFocus(parent)
312     local secure = CreateFrame("Button", "OmaFocusSecure", parent, inheritedFrames);
313     local frame = CreateFrame("Frame", "OmaFocus", parent);
314     local unit = "focus";
315     secure:SetPoint("BOTTOMLEFT", parent, "TOPLEFT");
316     frame:SetPoint("BOTTOMLEFT", parent, "TOPLEFT");
317     setupFrame(frame, secure, unit);
318     RegisterUnitWatch(frame);
319     RegisterUnitWatch(secure);
320     local vehicle = format("[@%s,unithasvehicleui] vehicle; no", unit);
321     RegisterStateDriver(secure, "vehicleui", vehicle);
322     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
323 end
324
325 local function loadCharSettings()
326     anchorX, anchorY = Settings.Character.AnchorX, Settings.Character.AnchorY;
327     attributes = Settings.Character.Clickheal;
328     positions = Settings.Character.Positions;
329 end
330
331 local function initialize()
332     loadCharSettings();
333     CFrame:SetPoint("CENTER", nil, "CENTER", anchorX, anchorY);
334     CFrame:SetHeight((height+2)*8);
335     CFrame:SetWidth((width+2)*5+1);
336     initializeParty(CFrame);
337     initializeRaid(CFrame);
338     initializeFocus(CFrame);
339 end
340
341 -- hiding Blizzard frames somewhat based on ShadowedUF
342 local function hideBlizzardRaidButton()
343     if InCombatLockdown() then return end
344     local manager = CompactRaidFrameManager;
345     -- remove show and unlock buttons from manager
346     manager.displayFrame.hiddenModeToggle:Hide();
347     manager.displayFrame.lockedModeToggle:Hide();
348 end
349
350 local hidden = false;
351 local function hideBlizzardRaid()
352     if hidden then return end
353     hidden = true;
354     local function hide()
355         if InCombatLockdown() then return end
356         CompactRaidFrameContainer:Hide();
357         local shown = CompactRaidFrameManager_GetSetting("IsShown");
358         if shown and shown ~= "0" then
359             CompactRaidFrameManager_SetSetting("IsShown", "0");
360         end
361     end
362
363     CompactRaidFrameContainer:UnregisterAllEvents();
364     -- there may still be frames created at startup getting events
365     for i=1,MAX_RAID_MEMBERS do
366         local frame = _G["CompactRaidFrame"..i];
367         if not frame then break end
368         frame:UnregisterAllEvents();
369     end
370     hooksecurefunc("CompactRaidFrameManager_UpdateShown", hide);
371     hide();
372     CompactRaidFrameContainer:HookScript("OnShow", hide);
373     hooksecurefunc("CompactRaidFrameManager_UpdateOptionsFlowContainer", hideBlizzardRaidButton);
374     hideBlizzardRaidButton();
375
376     -- hide focus frame
377     for _, frame in pairs({FocusFrame, FocusFrameToT}) do
378         frame:UnregisterAllEvents();
379         frame.healthbar:UnregisterAllEvents();
380         frame.manabar:UnregisterAllEvents();
381         if frame.spellbar then frame.spellbar:UnregisterAllEvents() end
382         if frame.powerBarAlt then frame.powerBarAlt:UnregisterAllEvents() end
383         frame:Hide();
384     end
385 end
386
387 CFrame:RegisterEvent("PLAYER_LOGIN");
388 CFrame:RegisterEvent("PLAYER_REGEN_ENABLED");
389 CFrame:SetScript("OnEvent", function(self, event, addon)
390     if event == "PLAYER_LOGIN" then
391         OmaRFLoadChar();
392         OmaRFIndicators.LoadChar();
393         hideBlizzardRaid();
394         initialize();
395     elseif event == "PLAYER_REGEN_ENABLED" then
396         -- if CompactRaidFrameManager_UpdateOptionsFlowContainer was called in combat
397         -- couldn't hide the button
398         hideBlizzardRaidButton();
399     end
400 end);