OmaUF/TargetFrame.lua

   1 -- TargetFrame.lua
   2 local _;
   3 local unpack, pairs = unpack, pairs;
   4 local format = string.format;
   5 local CreateFrame, RegisterStateDriver, RegisterUnitWatch = CreateFrame, RegisterStateDriver, RegisterUnitWatch;
   6 local STANDARD_TEXT_FONT = STANDARD_TEXT_FONT;
   7 local GameTooltip = nil;
   8 local GameTooltip_SetDefaultAnchor = nil;
   9
  10 local registerUnitEvents = OmaUFEvents.RegisterUnitEvents;
  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 healpredColor = Settings.HealpredColor;
  21 local healabsorbColor = Settings.HealabsorbColor;
  22 local width, height = Settings.Target.Width, Settings.Target.Height;
  23 local anchorX, anchorY = Settings.Target.AnchorX, Settings.Target.AnchorY;
  24 -- placeholders with visible values when error happens
  25 local attributes = {};
  26
  27 local inheritedFrames = "SecureUnitButtonTemplate,SecureHandlerStateTemplate";
  28
  29 local function frameShow(frame)
  30     frame:RegisterEvent("UNIT_ENTERED_VEHICLE");
  31     frame:RegisterEvent("UNIT_EXITED_VEHICLE");
  32     frame:RegisterEvent("UNIT_PET");
  33     frame:RegisterEvent("GROUP_ROSTER_UPDATE"); -- TODO needed? won't be having group number in frames
  34     frame:RegisterEvent("PLAYER_ENTERING_WORLD");
  35     frame:RegisterEvent("PLAYER_TARGET_CHANGED");
  36     frame:RegisterEvent("PARTY_LEADER_CHANGED");
  37     frame:RegisterUnitEvent("UNIT_LEVEL", frame.unit);
  38     registerUnitEvents(frame);
  39     unitEvent(frame, "UPDATE_ALL_BARS");
  40 end
  41
  42 local function frameHide(frame)
  43     frame:UnregisterAllEvents();
  44 end
  45
  46 local function showTooltip(secure)
  47     GameTooltip_SetDefaultAnchor(GameTooltip, secure);
  48     GameTooltip:SetUnit(secure:GetAttribute("unit"));
  49 end
  50
  51 local function hideTooltip(secure)
  52     GameTooltip:FadeOut();
  53 end
  54
  55 function OmaUnitFrames.InitializeTarget(parent)
  56     GameTooltip = _G["GameTooltip"];
  57     GameTooltip_SetDefaultAnchor = _G["GameTooltip_SetDefaultAnchor"];
  58     attributes = Settings.Character.Clickheal;
  59
  60     local secure = CreateFrame("Button", "OmaTargetSecure", parent, inheritedFrames);
  61     local frame = CreateFrame("Frame", "OmaTarget", parent);
  62     local unit = "target";
  63     secure:SetPoint("CENTER", parent, "CENTER", anchorX, anchorY);
  64     secure:SetAttribute("unit", unit);
  65     frame:SetPoint("CENTER", parent, "CENTER", anchorX, anchorY);
  66     frame:SetAttribute("unit", unit);
  67     frame.unit = unit;
  68     frame.displayed = unit;
  69     --frame.vehicle = unit.."pet"; -- TODO nothing extra needed for target vehicles?
  70     -- hide frame to get initial frameShow call
  71     frame:Hide();
  72     -- create visuals
  73     secure:SetWidth(width+2);
  74     secure:SetHeight(height+2);
  75     frame:SetWidth(width+2);
  76     frame:SetHeight(height+2);
  77     frame.width = width;
  78     frame.base = frame:CreateTexture(nil, "BACKGROUND");
  79     frame.base:SetAllPoints();
  80     frame.base:SetColorTexture(1, 1, 1);
  81     frame.base:SetVertexColor(unpack(baseColor));
  82     frame.healthback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
  83     frame.healthback:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
  84     frame.healthback:SetPoint("BOTTOMRIGHT", frame, "RIGHT", -1, -5);
  85     frame.healthback:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
  86     frame.healthback:SetVertexColor(unpack(bgColor));
  87     frame.health = frame:CreateTexture(nil, "BORDER");
  88     frame.health:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT");
  89     frame.health:SetPoint("BOTTOMLEFT", frame.healthback, "BOTTOMLEFT");
  90     frame.health:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
  91     frame.health:SetVertexColor(unpack(healthColor));
  92     frame.healthText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlightLarge");
  93     frame.healthText:SetPoint("RIGHT", frame.healthback, "RIGHT", -2, 0);
  94     frame.manaback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
  95     frame.manaback:SetPoint("TOPLEFT", frame, "LEFT", 1, -5);
  96     frame.manaback:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
  97     frame.manaback:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
  98     frame.manaback:SetVertexColor(unpack(bgColor));
  99     frame.mana = frame:CreateTexture(nil, "BORDER");
 100     frame.mana:SetPoint("TOPLEFT", frame.manaback, "TOPLEFT");
 101     frame.mana:SetPoint("BOTTOMLEFT", frame.manaback, "BOTTOMLEFT");
 102     frame.mana:SetTexture("Interface\\RaidFrame\\Raid-Bar-Hp-Fill");
 103     frame.manaText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlight");
 104     frame.manaText:SetPoint("RIGHT", frame.manaback, "RIGHT", -2, 0);
 105     frame.manaText:Hide();
 106     frame.shield = frame:CreateTexture(nil, "BORDER");
 107     frame.shield:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
 108     frame.shield:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
 109     frame.shield:SetTexture("Interface\\RaidFrame\\Shield-Fill");
 110     frame.shield:SetVertexColor(unpack(shieldColor));
 111     frame.shield:Hide();
 112     frame.shieldhl = frame:CreateTexture(nil, "ARTWORK");
 113     frame.shieldhl:SetPoint("TOPLEFT", frame.healthback, "TOPRIGHT", -1, 0);
 114     frame.shieldhl:SetPoint("BOTTOMRIGHT", frame.healthback, "BOTTOMRIGHT", 1, 0);
 115     frame.shieldhl:SetColorTexture(unpack(shieldhlColor));
 116     frame.shieldhl:Hide();
 117     frame.healpred = frame:CreateTexture(nil, "ARTWORK");
 118     frame.healpred:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
 119     frame.healpred:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
 120     frame.healpred:SetColorTexture(unpack(healpredColor));
 121     frame.healpred:Hide();
 122     frame.healabsorb = frame:CreateTexture(nil, "ARTWORK");
 123     frame.healabsorb:SetPoint("TOPRIGHT", frame.health, "TOPRIGHT");
 124     frame.healabsorb:SetPoint("BOTTOMRIGHT", frame.health, "BOTTOMRIGHT");
 125     frame.healabsorb:SetColorTexture(unpack(healabsorbColor));
 126     frame.healabsorb:Hide();
 127     frame.status = frame:CreateTexture(nil, "OVERLAY");
 128     frame.status:SetPoint("TOPLEFT", frame.manaback, "BOTTOMLEFT", -8, 8);
 129     frame.status:SetPoint("BOTTOMRIGHT", frame.manaback, "BOTTOMLEFT", 8, -8);
 130     frame.status:SetTexture("Interface\\CHARACTERFRAME\\UI-StateIcon");
 131     frame.status:Hide();
 132     frame.pvp = frame:CreateTexture(nil, "OVERLAY");
 133     frame.pvp:SetPoint("TOPLEFT", frame.manaback, "BOTTOMLEFT", 8, 6);
 134     frame.pvp:SetPoint("BOTTOMRIGHT", frame.manaback, "BOTTOMLEFT", 32, -18);
 135     frame.pvp:SetTexture("Interface\\TARGETINGFRAME\\UI-PVP-Horde");
 136     frame.pvp:Hide();
 137     frame.leader = frame:CreateTexture(nil, "OVERLAY");
 138     frame.leader:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT", -6, 6);
 139     frame.leader:SetPoint("BOTTOMRIGHT", frame.healthback, "TOPLEFT", 6, -6);
 140     frame.leader:Hide();
 141     frame.name = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightLarge");
 142     frame.name:SetPoint("LEFT", frame.healthback, "LEFT", 2, 1);
 143     frame.level = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
 144     frame.level:SetPoint("LEFT", frame.manaback, "LEFT", 2, 1);
 145     -- set scripts
 146     frame:SetScript("OnShow", frameShow);
 147     frame:SetScript("OnHide", frameHide);
 148     frame:SetScript("OnEvent", unitEvent);
 149     secure:SetScript("OnEnter", showTooltip);
 150     secure:SetScript("OnLeave", hideTooltip);
 151     -- set attributes
 152     secure:RegisterForClicks("AnyDown");
 153     for attr, val in pairs(attributes) do
 154         secure:SetAttribute(attr, val);
 155     end
 156     -- rest give target and menu
 157     secure:SetAttribute("*type1", "target");
 158     secure:SetAttribute("*type2", "togglemenu");
 159     createAuraFrame(frame, unit);
 160     RegisterUnitWatch(frame);
 161     RegisterUnitWatch(secure);
 162     return frame;
 163 end