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