ba69982 - Fix raid aura throttling and throttle unit frame auras as well
[wowui.git] / OmaUF / TargetFrame.lua
1 -- TargetFrame.lua
2 local _;
3 local unpack, pairs = unpack, pairs;
4 local format = string.format;
5 local GameTooltip = GameTooltip;
6 local GameTooltip_SetDefaultAnchor = GameTooltip_SetDefaultAnchor;
7
8 local registerUnitEvents = OmaUFEvents.RegisterUnitEvents;
9 local registerCastEvents = OmaUFCastBar.RegisterCastEvents;
10 local unregisterCastEvents = OmaUFCastBar.UnregisterCastEvents;
11 local unitEvent = OmaUFEvents.UnitEvent;
12 local createAuraFrame = OmaUFAuras.CreateAuraFrame;
13
14 local Settings = OmaUFSettings;
15 local baseColor = Settings.BaseColor;
16 local bgColor = Settings.BgColor;
17 local healthColor = Settings.HealthColor;
18 local shieldColor = Settings.ShieldColor;
19 local shieldhlColor = Settings.ShieldhlColor;
20 local width, height = Settings.Target.Width, Settings.Target.Height;
21 local anchorX, anchorY = Settings.Target.AnchorX, Settings.Target.AnchorY;
22 -- placeholders with visible values when error happens
23 local attributes = {};
24
25 local inheritedFrames = "SecureUnitButtonTemplate,SecureHandlerStateTemplate";
26 local barTexture = "Interface\\AddOns\\OmaRF\\images\\minimalist";
27
28 local function frameShow(frame)
29     frame:RegisterEvent("GROUP_ROSTER_UPDATE");
30     frame:RegisterEvent("PLAYER_ENTERING_WORLD");
31     frame:RegisterEvent("PLAYER_TARGET_CHANGED");
32     frame:RegisterEvent("PARTY_LEADER_CHANGED");
33     frame:RegisterUnitEvent("UNIT_LEVEL", frame.unit);
34     registerUnitEvents(frame);
35     registerCastEvents(frame.castbar);
36     unitEvent(frame, "UPDATE_ALL_BARS");
37 end
38
39 local function frameHide(frame)
40     frame:UnregisterAllEvents();
41     unregisterCastEvents(frame.castbar);
42 end
43
44 local function showTooltip(secure)
45     GameTooltip_SetDefaultAnchor(GameTooltip, secure);
46     GameTooltip:SetUnit(secure:GetAttribute("unit"));
47 end
48
49 local function hideTooltip(secure)
50     GameTooltip:FadeOut();
51 end
52
53 function OmaUnitFrames.InitializeTarget(parent)
54     attributes = Settings.Character.Clickheal;
55
56     local secure = CreateFrame("Button", "OmaTargetSecure", parent, inheritedFrames);
57     local frame = CreateFrame("Frame", "OmaTarget", parent);
58     local unit = "target";
59     secure:SetPoint("CENTER", parent, "CENTER", anchorX, anchorY);
60     secure:SetAttribute("unit", unit);
61     frame:SetPoint("CENTER", parent, "CENTER", anchorX, anchorY);
62     frame:SetAttribute("unit", unit);
63     frame.unit = unit;
64     frame.displayed = unit;
65     -- hide frame to get initial frameShow call
66     frame:Hide();
67     -- create visuals
68     secure:SetWidth(width+2);
69     secure:SetHeight(height+2);
70     frame:SetWidth(width+2);
71     frame:SetHeight(height+2);
72     frame.width = width;
73     frame.base = frame:CreateTexture(nil, "BACKGROUND");
74     frame.base:SetAllPoints();
75     frame.base:SetColorTexture(1, 1, 1);
76     frame.base:SetVertexColor(unpack(baseColor));
77     frame.healthback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
78     frame.healthback:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
79     frame.healthback:SetPoint("BOTTOMRIGHT", frame, "RIGHT", -1, -5);
80     frame.healthback:SetTexture(barTexture);
81     frame.healthback:SetVertexColor(unpack(bgColor));
82     frame.health = frame:CreateTexture(nil, "BORDER");
83     frame.health:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT");
84     frame.health:SetPoint("BOTTOMLEFT", frame.healthback, "BOTTOMLEFT");
85     frame.health:SetTexture(barTexture);
86     frame.health:SetVertexColor(unpack(healthColor));
87     frame.healthText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlightLarge");
88     frame.healthText:SetPoint("RIGHT", frame.healthback, "RIGHT", -2, 1);
89     frame.manaback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
90     frame.manaback:SetPoint("TOPLEFT", frame, "LEFT", 1, -5);
91     frame.manaback:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
92     frame.manaback:SetTexture(barTexture);
93     frame.manaback:SetVertexColor(unpack(bgColor));
94     frame.mana = frame:CreateTexture(nil, "BORDER");
95     frame.mana:SetPoint("TOPLEFT", frame.manaback, "TOPLEFT");
96     frame.mana:SetPoint("BOTTOMLEFT", frame.manaback, "BOTTOMLEFT");
97     frame.mana:SetTexture(barTexture);
98     frame.manaText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlight");
99     frame.manaText:SetPoint("RIGHT", frame.manaback, "RIGHT", -2, 1);
100     frame.manaText:Hide();
101     frame.shield = frame:CreateTexture(nil, "BORDER");
102     frame.shield:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
103     frame.shield:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
104     frame.shield:SetTexture(barTexture);
105     frame.shield:SetVertexColor(unpack(shieldColor));
106     frame.shield:Hide();
107     frame.shieldhl = frame:CreateTexture(nil, "ARTWORK");
108     frame.shieldhl:SetPoint("TOPLEFT", frame.healthback, "TOPRIGHT", -1, 0);
109     frame.shieldhl:SetPoint("BOTTOMRIGHT", frame.healthback, "BOTTOMRIGHT", 1, 0);
110     frame.shieldhl:SetColorTexture(unpack(shieldhlColor));
111     frame.shieldhl:Hide();
112     frame.status = frame:CreateTexture(nil, "OVERLAY");
113     frame.status:SetPoint("TOPLEFT", frame.manaback, "BOTTOMLEFT", -8, 8);
114     frame.status:SetPoint("BOTTOMRIGHT", frame.manaback, "BOTTOMLEFT", 8, -8);
115     frame.status:SetTexture("Interface\\CHARACTERFRAME\\UI-StateIcon");
116     frame.status:Hide();
117     frame.pvp = frame:CreateTexture(nil, "OVERLAY");
118     frame.pvp:SetPoint("TOPLEFT", frame.manaback, "BOTTOMLEFT", 8, 6);
119     frame.pvp:SetPoint("BOTTOMRIGHT", frame.manaback, "BOTTOMLEFT", 32, -18);
120     frame.pvp:SetTexture("Interface\\TARGETINGFRAME\\UI-PVP-Horde");
121     frame.pvp:Hide();
122     frame.leader = frame:CreateTexture(nil, "OVERLAY");
123     frame.leader:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT", -6, 6);
124     frame.leader:SetPoint("BOTTOMRIGHT", frame.healthback, "TOPLEFT", 6, -6);
125     frame.leader:Hide();
126     frame.name = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightLarge");
127     frame.name:SetPoint("LEFT", frame.healthback, "LEFT", 2, 1);
128     frame.name.count = 8;
129     frame.level = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
130     frame.level:SetPoint("LEFT", frame.manaback, "LEFT", 2, 1);
131     frame.targeticon = frame:CreateTexture(nil, "OVERLAY");
132     frame.targeticon:SetPoint("CENTER", frame.healthback, "TOP");
133     frame.targeticon:SetWidth(18);
134     frame.targeticon:SetHeight(18);
135     frame.targeticon:SetTexture("Interface\\TARGETINGFRAME\\UI-RaidTargetingIcons");
136     frame.targeticon:Hide();
137     frame.castbar = OmaUFCastBar.CreateCastBar(frame, unit, 8);
138     createAuraFrame(frame, unit);
139     -- set scripts
140     frame:SetScript("OnShow", frameShow);
141     frame:SetScript("OnHide", frameHide);
142     frame:SetScript("OnEvent", unitEvent);
143     secure:SetScript("OnEnter", showTooltip);
144     secure:SetScript("OnLeave", hideTooltip);
145     -- set attributes
146     secure:RegisterForClicks("AnyDown");
147     for attr, val in pairs(attributes) do
148         secure:SetAttribute(attr, val);
149     end
150     -- rest give target and menu
151     secure:SetAttribute("*type1", "target");
152     secure:SetAttribute("*type2", "togglemenu");
153     secure:SetAttribute("toggleForVehicle", false);
154     RegisterUnitWatch(frame);
155     RegisterUnitWatch(secure);
156     return frame;
157 end