OmaUF/PlayerFrame.lua

   1 -- PlayerFrame.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
  13 local Settings = OmaUFSettings;
  14 local baseColor = Settings.BaseColor;
  15 local bgColor = Settings.BgColor;
  16 local healthColor = Settings.HealthColor;
  17 local shieldColor = Settings.ShieldColor;
  18 local shieldhlColor = Settings.ShieldhlColor;
  19 local healpredColor = Settings.HealpredColor;
  20 local healabsorbColor = Settings.HealabsorbColor;
  21 local width, height = Settings.Player.Width, Settings.Player.Height;
  22 local anchorX, anchorY = Settings.Player.AnchorX, Settings.Player.AnchorY;
  23 -- placeholders with visible values when error happens
  24 local attributes = {};
  25
  26 local inheritedFrames = "SecureUnitButtonTemplate,SecureHandlerStateTemplate";
  27 local barTexture = "Interface\\AddOns\\OmaRF\\images\\minimalist";
  28
  29 local function frameShow(frame)
  30     frame:RegisterEvent("UNIT_ENTERED_VEHICLE");
  31     frame:RegisterEvent("UNIT_EXITED_VEHICLE");
  32     frame:RegisterEvent("GROUP_ROSTER_UPDATE");
  33     frame:RegisterEvent("PLAYER_ENTERING_WORLD");
  34     frame:RegisterEvent("PLAYER_LEVEL_UP");
  35     frame:RegisterEvent("PLAYER_REGEN_DISABLED");
  36     frame:RegisterEvent("PLAYER_REGEN_ENABLED");
  37     frame:RegisterEvent("PLAYER_UPDATE_RESTING");
  38     frame:RegisterEvent("PARTY_LEADER_CHANGED");
  39     registerUnitEvents(frame);
  40     registerCastEvents(frame.castbar);
  41     unitEvent(frame, "UPDATE_ALL_BARS");
  42 end
  43
  44 local function frameHide(frame)
  45     frame:UnregisterAllEvents();
  46     unregisterCastEvents(frame.castbar);
  47 end
  48
  49 local function showTooltip(secure)
  50     GameTooltip_SetDefaultAnchor(GameTooltip, secure);
  51     GameTooltip:SetUnit(secure:GetAttribute("displayed"));
  52 end
  53
  54 local function hideTooltip(secure)
  55     GameTooltip:FadeOut();
  56 end
  57
  58 local vehicletoggle = [=[
  59     if newstate == "vehicle" then
  60         self:SetAttribute("displayed", self:GetAttribute("vehicle"));
  61     else
  62         self:SetAttribute("displayed", self:GetAttribute("unit"));
  63     end
  64 ]=]
  65
  66 function OmaUnitFrames.UpdatePlayerTooltips()
  67     GameTooltip = _G["GameTooltip"];
  68     GameTooltip_SetDefaultAnchor = _G["GameTooltip_SetDefaultAnchor"];
  69 end
  70
  71 function OmaUnitFrames.InitializePlayer(parent)
  72     attributes = Settings.Character.Clickheal;
  73
  74     local secure = CreateFrame("Button", "OmaPlayerSecure", parent, inheritedFrames);
  75     local frame = CreateFrame("Frame", "OmaPlayer", parent);
  76     local unit = "player";
  77     secure:SetPoint("CENTER", parent, "CENTER", anchorX, anchorY);
  78     secure:SetAttribute("unit", unit);
  79     secure:SetAttribute("displayed", unit);
  80     secure:SetAttribute("vehicle", "vehicle");
  81     frame:SetPoint("CENTER", parent, "CENTER", anchorX, anchorY);
  82     frame:SetAttribute("unit", unit);
  83     frame.unit = unit;
  84     frame.displayed = unit;
  85     frame.vehicle = "vehicle";
  86     -- hide frame to get initial frameShow call
  87     frame:Hide();
  88     -- create visuals
  89     secure:SetWidth(width+2);
  90     secure:SetHeight(height+2);
  91     frame:SetWidth(width+2);
  92     frame:SetHeight(height+2);
  93     frame.width = width;
  94     frame.base = frame:CreateTexture(nil, "BACKGROUND");
  95     frame.base:SetAllPoints();
  96     frame.base:SetColorTexture(1, 1, 1);
  97     frame.base:SetVertexColor(unpack(baseColor));
  98     frame.healthback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
  99     frame.healthback:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
 100     frame.healthback:SetPoint("BOTTOMRIGHT", frame, "RIGHT", -1, -5);
 101     frame.healthback:SetTexture(barTexture);
 102     frame.healthback:SetVertexColor(unpack(bgColor));
 103     frame.health = frame:CreateTexture(nil, "BORDER");
 104     frame.health:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT");
 105     frame.health:SetPoint("BOTTOMLEFT", frame.healthback, "BOTTOMLEFT");
 106     frame.health:SetTexture(barTexture);
 107     frame.health:SetVertexColor(unpack(healthColor));
 108     frame.healthText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlightLarge");
 109     frame.healthText:SetPoint("RIGHT", frame.healthback, "RIGHT", -2, 1);
 110     frame.manaback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
 111     frame.manaback:SetPoint("TOPLEFT", frame, "LEFT", 1, -5);
 112     frame.manaback:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
 113     frame.manaback:SetTexture(barTexture);
 114     frame.manaback:SetVertexColor(unpack(bgColor));
 115     frame.mana = frame:CreateTexture(nil, "BORDER");
 116     frame.mana:SetPoint("TOPLEFT", frame.manaback, "TOPLEFT");
 117     frame.mana:SetPoint("BOTTOMLEFT", frame.manaback, "BOTTOMLEFT");
 118     frame.mana:SetTexture(barTexture);
 119     frame.manaText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlight");
 120     frame.manaText:SetPoint("RIGHT", frame.manaback, "RIGHT", -2, 1);
 121     frame.manaText:Hide();
 122     frame.shield = frame:CreateTexture(nil, "BORDER");
 123     frame.shield:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
 124     frame.shield:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
 125     frame.shield:SetTexture(barTexture);
 126     frame.shield:SetVertexColor(unpack(shieldColor));
 127     frame.shield:Hide();
 128     frame.shieldhl = frame:CreateTexture(nil, "ARTWORK");
 129     frame.shieldhl:SetPoint("TOPLEFT", frame.healthback, "TOPRIGHT", -1, 0);
 130     frame.shieldhl:SetPoint("BOTTOMRIGHT", frame.healthback, "BOTTOMRIGHT", 1, 0);
 131     frame.shieldhl:SetColorTexture(unpack(shieldhlColor));
 132     frame.shieldhl:Hide();
 133     frame.healpred = frame:CreateTexture(nil, "ARTWORK");
 134     frame.healpred:SetPoint("TOPLEFT", frame.health, "TOPRIGHT");
 135     frame.healpred:SetPoint("BOTTOMLEFT", frame.health, "BOTTOMRIGHT");
 136     frame.healpred:SetColorTexture(unpack(healpredColor));
 137     frame.healpred:Hide();
 138     frame.healabsorb = frame:CreateTexture(nil, "ARTWORK");
 139     frame.healabsorb:SetPoint("TOPRIGHT", frame.health, "TOPRIGHT");
 140     frame.healabsorb:SetPoint("BOTTOMRIGHT", frame.health, "BOTTOMRIGHT");
 141     frame.healabsorb:SetColorTexture(unpack(healabsorbColor));
 142     frame.healabsorb:Hide();
 143     frame.status = frame:CreateTexture(nil, "OVERLAY");
 144     frame.status:SetPoint("TOPLEFT", frame.manaback, "BOTTOMLEFT", -8, 8);
 145     frame.status:SetPoint("BOTTOMRIGHT", frame.manaback, "BOTTOMLEFT", 8, -8);
 146     frame.status:SetTexture("Interface\\CHARACTERFRAME\\UI-StateIcon");
 147     frame.status:Hide();
 148     frame.pvp = frame:CreateTexture(nil, "OVERLAY");
 149     frame.pvp:SetPoint("TOPLEFT", frame.manaback, "BOTTOMLEFT", 8, 6);
 150     frame.pvp:SetPoint("BOTTOMRIGHT", frame.manaback, "BOTTOMLEFT", 32, -18);
 151     frame.pvp:SetTexture("Interface\\TARGETINGFRAME\\UI-PVP-Horde");
 152     frame.pvp:Hide();
 153     frame.leader = frame:CreateTexture(nil, "OVERLAY");
 154     frame.leader:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT", -6, 6);
 155     frame.leader:SetPoint("BOTTOMRIGHT", frame.healthback, "TOPLEFT", 6, -6);
 156     frame.leader:Hide();
 157     frame.level = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight");
 158     frame.level:SetPoint("LEFT", frame.manaback, "LEFT", 2, 1);
 159     frame.targeticon = frame:CreateTexture(nil, "OVERLAY");
 160     frame.targeticon:SetPoint("CENTER", frame.healthback, "TOP");
 161     frame.targeticon:SetWidth(18);
 162     frame.targeticon:SetHeight(18);
 163     frame.targeticon:SetTexture("Interface\\TARGETINGFRAME\\UI-RaidTargetingIcons");
 164     frame.targeticon:Hide();
 165     frame.castbar = OmaUFCastBar.CreateCastBar(frame, unit, 8);
 166     -- set scripts
 167     frame:SetScript("OnShow", frameShow);
 168     frame:SetScript("OnHide", frameHide);
 169     frame:SetScript("OnEvent", unitEvent);
 170     secure:SetScript("OnEnter", showTooltip);
 171     secure:SetScript("OnLeave", hideTooltip);
 172     -- set attributes
 173     secure:RegisterForClicks("AnyDown");
 174     for attr, val in pairs(attributes) do
 175         secure:SetAttribute(attr, val);
 176     end
 177     -- rest give target and menu
 178     secure:SetAttribute("*type1", "target");
 179     secure:SetAttribute("*type2", "togglemenu");
 180     secure:SetAttribute("toggleForVehicle", true);
 181     RegisterUnitWatch(frame);
 182     RegisterUnitWatch(secure);
 183     RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
 184     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 185     return frame;
 186 end