OmaUF/UnitFrames.lua

   1 -- RaidFrame.lua
   2 local _;
   3 local unpack, pairs = unpack, pairs;
   4 local format = string.format;
   5 local UnitHealthMax, UnitPowerMax = UnitHealthMax, UnitPowerMax;
   6 local CreateFrame, RegisterStateDriver, RegisterUnitWatch = CreateFrame, RegisterStateDriver, RegisterUnitWatch;
   7 local STANDARD_TEXT_FONT = STANDARD_TEXT_FONT;
   8 local GameTooltip = nil;
   9 local GameTooltip_SetDefaultAnchor = nil;
  10
  11 local registerEvents = OmaUFEvents.RegisterEvents;
  12 local unitEvent = OmaUFEvents.UnitEvent;
  13 local createAuraFrame = OmaUFAuras.CreateAuraFrame;
  14
  15 local Settings = OmaUFSettings;
  16 local indSize = Settings.IndSize;
  17 local baseColor = Settings.BaseColor;
  18 local bgColor = Settings.BgColor;
  19 local healthColor = Settings.HealthColor;
  20 local shieldColor = Settings.ShieldColor;
  21 local shieldhlColor = Settings.ShieldhlColor;
  22 local healpredColor = Settings.HealpredColor;
  23 local healabsorbColor = Settings.HealabsorbColor;
  24 -- placeholders with visible values when error happens
  25 local width, height = 10, 10;
  26 local anchorX, anchorY = 10, 10;
  27 local attributes = {};
  28
  29 local UnitFrames = CreateFrame("Frame", "OmaUnitFrames");
  30 local inheritedFrames = "SecureUnitButtonTemplate,SecureHandlerStateTemplate";
  31
  32 local events = {
  33     ["player"] = {
  34         "UNIT_ENTERED_VEHICLE",
  35         "UNIT_EXITED_VEHICLE",
  36         "UNIT_PET",
  37         "PLAYER_ROLES_ASSIGNED",
  38         "PLAYER_ENTERING_WORLD",
  39         "PLAYER_LEVEL_UP",
  40         "PLAYER_REGEN_DISABLED",
  41         "PLAYER_REGEN_ENABLED",
  42         "PLAYER_UPDATE_RESTING",
  43     },
  44     ["target"] = {
  45         "UNIT_ENTERED_VEHICLE",
  46         "UNIT_EXITED_VEHICLE",
  47         "UNIT_PET",
  48         "UNIT_LEVEL",
  49         "GROUP_ROSTER_UPDATE",
  50         "PLAYER_ROLES_ASSIGNED",
  51         "PLAYER_ENTERING_WORLD",
  52         "PLAYER_TARGET_CHANGED",
  53     },
  54 };
  55
  56 local M = {};
  57 OmaUnitFrames = M;
  58
  59 local function frameShow(frame)
  60     for _, ev in pairs(events[frame.unit]) do
  61         frame:RegisterEvent(ev);
  62     end
  63     registerEvents(frame);
  64     unitEvent(frame, "UPDATE_ALL_BARS");
  65 end
  66
  67 local function frameHide(frame)
  68     frame:UnregisterAllEvents();
  69 end
  70
  71 local function showTooltip(frame)
  72     GameTooltip_SetDefaultAnchor(GameTooltip, PlayerFrame);
  73     GameTooltip:SetUnit(frame:GetAttribute("unit"));
  74 end
  75
  76 local function hideTooltip(frame)
  77     GameTooltip:FadeOut();
  78 end
  79
  80 local function setupFrame(frame, secure, unit)
  81     secure:SetAttribute("unit", unit);
  82     frame:SetAttribute("unit", unit);
  83     frame.unit = unit;
  84     frame.displayed = unit;
  85     -- hide frame to get initial frameShow call
  86     frame:Hide();
  87     if unit == "player" then
  88         frame.vehicle = "vehicle";
  89     else
  90         frame.vehicle = unit.."pet";
  91     end
  92     -- create visuals
  93     frame.base = frame:CreateTexture(nil, "BACKGROUND");
  94     frame.base:SetAllPoints();
  95     frame.base:SetColorTexture(1, 1, 1);
  96     frame.base:SetVertexColor(unpack(baseColor));
  97     frame.healthback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
  98     frame.healthback:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
  99     frame.healthback:SetPoint("BOTTOMRIGHT", frame, "RIGHT", -1, -5);
 100     frame.healthback:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
 101     frame.healthback:SetVertexColor(unpack(bgColor));
 102     frame.health = frame:CreateTexture(nil, "BORDER");
 103     frame.health:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
 104     frame.health:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT");
 105     frame.health:SetPoint("BOTTOMLEFT", frame.healthback, "BOTTOMLEFT");
 106     frame.health:SetVertexColor(unpack(healthColor));
 107     frame.health:SetWidth(width);
 108     frame.health.max = UnitHealthMax(unit);
 109     frame.healthText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlightLarge");
 110     frame.healthText:SetPoint("RIGHT", frame.healthback, "RIGHT", -2, 0);
 111     frame.healthText:Hide();
 112     frame.manaback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
 113     frame.manaback:SetPoint("TOPLEFT", frame, "LEFT", 1, -5);
 114     frame.manaback:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
 115     frame.manaback:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
 116     frame.manaback:SetVertexColor(unpack(bgColor));
 117     frame.mana = frame:CreateTexture(nil, "BORDER");
 118     frame.mana:SetPoint("TOPLEFT", frame.manaback, "TOPLEFT");
 119     frame.mana:SetPoint("BOTTOMLEFT", frame.manaback, "BOTTOMLEFT");
 120     frame.mana:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
 121     frame.mana:SetWidth(width);
 122     frame.mana.max = UnitPowerMax(unit);
 123     frame.manaText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlight");
 124     frame.manaText:SetPoint("RIGHT", frame.manaback, "RIGHT", -2, 0);
 125     frame.manaText:Hide();
 126     frame.shield = frame:CreateTexture(nil, "BORDER");
 127     frame.shield:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
 128     frame.shield:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
 129     frame.shield:SetTexture("Interface\\RaidFrame\\Shield-Fill");
 130     frame.shield:SetVertexColor(unpack(shieldColor));
 131     frame.shield:Hide();
 132     frame.shieldhl = frame:CreateTexture(nil, "ARTWORK");
 133     frame.shieldhl:SetPoint("TOPLEFT", frame.healthback, "TOPRIGHT", -1, 0);
 134     frame.shieldhl:SetPoint("BOTTOMRIGHT", frame.healthback, "BOTTOMRIGHT", 1, 0);
 135     frame.shieldhl:SetColorTexture(unpack(shieldhlColor));
 136     frame.shieldhl:Hide();
 137     frame.healpred = frame:CreateTexture(nil, "ARTWORK");
 138     frame.healpred:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
 139     frame.healpred:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
 140     frame.healpred:SetColorTexture(unpack(healpredColor));
 141     frame.healpred:Hide();
 142     frame.healabsorb = frame:CreateTexture(nil, "ARTWORK");
 143     frame.healabsorb:SetPoint("TOPRIGHT", frame.health, "TOPRIGHT");
 144     frame.healabsorb:SetPoint("BOTTOMRIGHT", frame.health, "BOTTOMRIGHT");
 145     frame.healabsorb:SetColorTexture(unpack(healabsorbColor));
 146     frame.healabsorb:Hide();
 147     frame.overlay = frame:CreateTexture(nil, "ARTWORK", nil, 1);
 148     frame.overlay:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT");
 149     frame.overlay:SetPoint("BOTTOMRIGHT", frame.healthback, "BOTTOMRIGHT");
 150     frame.overlay:SetColorTexture(1, 1, 1);
 151     frame.overlay:Hide();
 152     frame.role = frame:CreateTexture(nil, "OVERLAY");
 153     frame.role:SetPoint("TOPLEFT", frame.healthback, "TOPRIGHT", -8, 8);
 154     frame.role:SetPoint("BOTTOMRIGHT", frame.healthback, "TOPRIGHT", 8, -8);
 155     frame.role:SetTexture("Interface\\LFGFRAME\\UI-LFG-ICON-PORTRAITROLES");
 156     frame.role:Hide();
 157     frame.status = frame:CreateTexture(nil, "OVERLAY");
 158     frame.status:SetPoint("TOPLEFT", frame.manaback, "BOTTOMLEFT", -8, 8);
 159     frame.status:SetPoint("BOTTOMRIGHT", frame.manaback, "BOTTOMLEFT", 8, -8);
 160     frame.status:SetTexture("Interface\\CHARACTERFRAME\\UI-StateIcon");
 161     frame.status:Hide();
 162     frame.pvp = frame:CreateTexture(nil, "OVERLAY");
 163     frame.pvp:SetPoint("TOPLEFT", frame.manaback, "BOTTOMLEFT", 8, 6);
 164     frame.pvp:SetPoint("BOTTOMRIGHT", frame.manaback, "BOTTOMLEFT", 32, -18);
 165     frame.pvp:SetTexture("Interface\\TARGETINGFRAME\\UI-PVP-Horde");
 166     frame.pvp:Hide();
 167     frame.leader = frame:CreateTexture(nil, "OVERLAY");
 168     frame.leader:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT", -6, 6);
 169     frame.leader:SetPoint("BOTTOMRIGHT", frame.healthback, "TOPLEFT", 6, -6);
 170     frame.leader:Hide();
 171     frame.name = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightLarge");
 172     frame.name:SetPoint("LEFT", frame.healthback, "LEFT", 2, 1);
 173     frame.level = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
 174     frame.level:SetPoint("LEFT", frame.manaback, "LEFT", 2, 1);
 175     -- set scripts
 176     frame:SetScript("OnShow", frameShow);
 177     frame:SetScript("OnHide", frameHide);
 178     frame:SetScript("OnEvent", unitEvent);
 179     secure:SetScript("OnEnter", showTooltip);
 180     secure:SetScript("OnLeave", hideTooltip);
 181     -- set attributes
 182     secure:RegisterForClicks("AnyDown");
 183     for attr, val in pairs(attributes) do
 184         secure:SetAttribute(attr, val);
 185     end
 186     -- rest give target and menu
 187     secure:SetAttribute("*type1", "target");
 188     secure:SetAttribute("*type2", "togglemenu");
 189 end
 190
 191 -- vehicle toggle from Shadowed Unit Frames
 192 local vehicletoggle = [=[
 193     local unit = self:GetAttribute("unit");
 194     if unit and newstate == "vehicle" and not UnitTargetsVehicleInRaidUI(unit) then
 195         self:SetAttribute("toggleForVehicle", false);
 196     else
 197         self:SetAttribute("toggleForVehicle", true);
 198     end
 199 ]=]
 200
 201 local function initializePlayer(parent)
 202     local secure = CreateFrame("Button", "OmaPlayerSecure", parent, inheritedFrames);
 203     local frame = CreateFrame("Frame", "OmaPlayer", parent);
 204     local unit = "player";
 205     secure:SetPoint("CENTER", parent, "CENTER", -300, -175);
 206     secure:SetWidth(width+2);
 207     secure:SetHeight(height+2);
 208     frame:SetPoint("CENTER", parent, "CENTER", -300, -175);
 209     frame:SetWidth(width+2);
 210     frame:SetHeight(height+2);
 211     setupFrame(frame, secure, unit);
 212     RegisterUnitWatch(frame);
 213     RegisterUnitWatch(secure);
 214     RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
 215     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 216 end
 217
 218 local function initializeTarget(parent)
 219     local secure = CreateFrame("Button", "OmaTargetSecure", parent, inheritedFrames);
 220     local frame = CreateFrame("Frame", "OmaTarget", parent);
 221     local unit = "target";
 222     secure:SetPoint("CENTER", parent, "CENTER", 300, -175);
 223     secure:SetWidth(width+2);
 224     secure:SetHeight(height+2);
 225     frame:SetPoint("CENTER", parent, "CENTER", 300, -175);
 226     frame:SetWidth(width+2);
 227     frame:SetHeight(height+2);
 228     setupFrame(frame, secure, unit);
 229     createAuraFrame(frame, unit);
 230     RegisterUnitWatch(frame);
 231     RegisterUnitWatch(secure);
 232     RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
 233     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 234 end
 235
 236 local function loadCharSettings()
 237     width, height = Settings.Character.Width, Settings.Character.Height;
 238     anchorX, anchorY = Settings.Character.AnchorX, Settings.Character.AnchorY;
 239     attributes = Settings.Character.Clickheal;
 240 end
 241
 242 local function initialize()
 243     loadCharSettings();
 244     -- let other addons hook these to anchor tooltip elsewhere
 245     GameTooltip = _G["GameTooltip"];
 246     GameTooltip_SetDefaultAnchor = _G["GameTooltip_SetDefaultAnchor"];
 247     initializePlayer(UIParent);
 248     initializeTarget(UIParent);
 249     -- TODO boss frames, pet frame, (arena frames)
 250 end
 251
 252 local hidden = false;
 253 local function hideBlizzardFrames()
 254     if hidden then return end
 255     hidden = true;
 256
 257     for _, frame in pairs({PlayerFrame, TargetFrame, TargetFrameToT}) do
 258         frame:UnregisterAllEvents();
 259         frame.healthbar:UnregisterAllEvents();
 260         frame.manabar:UnregisterAllEvents();
 261         if frame.spellbar then frame.spellbar:UnregisterAllEvents() end
 262         if frame.powerBarAlt then frame.powerBarAlt:UnregisterAllEvents() end
 263         frame:Hide();
 264     end
 265
 266     -- TODO create frames for class powers, currently using Simple Holy Power
 267     for _, frame in pairs({PlayerFrameAlternateManaBar, ComboFrame,
 268             PriestBarFrame, RuneFrame, WarlockPowerFrame, MonkHarmonyBarFrame,
 269             PaladinPowerBarFrame, MageArcaneChargesFrame}) do
 270         frame:UnregisterAllEvents();
 271         frame:Hide();
 272     end
 273
 274     -- from ShadowedUF, re-register vehicle events for default auras
 275     PlayerFrame:RegisterEvent("PLAYER_ENTERING_WORLD");
 276     PlayerFrame:RegisterEvent("UNIT_ENTERING_VEHICLE");
 277     PlayerFrame:RegisterEvent("UNIT_ENTERED_VEHICLE");
 278     PlayerFrame:RegisterEvent("UNIT_EXITING_VEHICLE");
 279     PlayerFrame:RegisterEvent("UNIT_EXITED_VEHICLE");
 280     PlayerFrame:SetMovable(true);
 281     PlayerFrame:SetUserPlaced(true);
 282     PlayerFrame:SetDontSavePosition(true);
 283 end
 284
 285 UnitFrames:RegisterEvent("PLAYER_LOGIN");
 286 UnitFrames:SetScript("OnEvent", function(self, event)
 287     if event == "PLAYER_LOGIN" then
 288         OmaUFLoadChar();
 289         OmaUFEvents.LoadChar();
 290         hideBlizzardFrames();
 291         initialize();
 292     end
 293 end);