OmaUF/PetFrame.lua

   1 -- PetFrame.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 unitEvent = OmaUFEvents.UnitEvent;
  10
  11 local Settings = OmaUFSettings;
  12 local baseColor = Settings.BaseColor;
  13 local bgColor = Settings.BgColor;
  14 local healthColor = Settings.HealthColor;
  15 local shieldColor = Settings.ShieldColor;
  16 local shieldhlColor = Settings.ShieldhlColor;
  17 local healpredColor = Settings.HealpredColor;
  18 local healabsorbColor = Settings.HealabsorbColor;
  19 local width, height = Settings.Pet.Width, Settings.Pet.Height;
  20 local anchorX, anchorY = Settings.Pet.AnchorX, Settings.Pet.AnchorY;
  21 -- placeholders with visible values when error happens
  22 local attributes = {};
  23
  24 local inheritedFrames = "SecureUnitButtonTemplate,SecureHandlerStateTemplate";
  25 local barTexture = "Interface\\AddOns\\OmaRF\\images\\minimalist";
  26
  27 local function frameShow(frame)
  28     frame:RegisterEvent("UNIT_ENTERED_VEHICLE");
  29     frame:RegisterEvent("UNIT_EXITED_VEHICLE");
  30     frame:RegisterEvent("UNIT_PET");
  31     frame:RegisterEvent("PLAYER_ENTERING_WORLD");
  32     registerUnitEvents(frame);
  33     unitEvent(frame, "UPDATE_ALL_BARS");
  34 end
  35
  36 local function frameHide(frame)
  37     frame:UnregisterAllEvents();
  38 end
  39
  40 local function showTooltip(secure)
  41     GameTooltip_SetDefaultAnchor(GameTooltip, secure);
  42     GameTooltip:SetUnit(secure:GetAttribute("displayed"));
  43 end
  44
  45 local function hideTooltip(secure)
  46     GameTooltip:FadeOut();
  47 end
  48
  49 local vehicletoggle = [=[
  50     if newstate == "vehicle" then
  51         self:SetAttribute("displayed", self:GetAttribute("vehicle"));
  52     else
  53         self:SetAttribute("displayed", self:GetAttribute("unit"));
  54     end
  55 ]=]
  56
  57 function OmaUnitFrames.UpdatePetTooltips()
  58     GameTooltip = _G["GameTooltip"];
  59     GameTooltip_SetDefaultAnchor = _G["GameTooltip_SetDefaultAnchor"];
  60 end
  61
  62 function OmaUnitFrames.InitializePet(parent)
  63     attributes = Settings.Character.Clickheal; -- TODO pet clickheal separate with Mend Pet etc.
  64
  65     local secure = CreateFrame("Button", "OmaPetSecure", parent, inheritedFrames);
  66     local frame = CreateFrame("Frame", "OmaPet", parent);
  67     local unit = "pet";
  68     secure:SetPoint("TOPRIGHT", parent, "TOPLEFT", anchorX, anchorY);
  69     secure:SetAttribute("unit", unit);
  70     secure:SetAttribute("displayed", unit);
  71     secure:SetAttribute("vehicle", "player");
  72     frame:SetPoint("TOPRIGHT", parent, "TOPLEFT", anchorX, anchorY);
  73     frame:SetAttribute("unit", unit);
  74     frame.unit = unit;
  75     frame.displayed = unit;
  76     frame.vehicle = "player";
  77     -- hide frame to get initial frameShow call
  78     frame:Hide();
  79     -- create visuals
  80     secure:SetWidth(width+2);
  81     secure:SetHeight(height+2);
  82     frame:SetWidth(width+2);
  83     frame:SetHeight(height+2);
  84     frame.width = width;
  85     frame.base = frame:CreateTexture(nil, "BACKGROUND");
  86     frame.base:SetAllPoints();
  87     frame.base:SetColorTexture(1, 1, 1);
  88     frame.base:SetVertexColor(unpack(baseColor));
  89     frame.healthback = frame:CreateTexture(nil, "BACKGROUND", nil, 1);
  90     frame.healthback:SetPoint("TOPLEFT", frame, "TOPLEFT", 1, -1);
  91     frame.healthback:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -1, 1);
  92     frame.healthback:SetTexture(barTexture);
  93     frame.healthback:SetVertexColor(unpack(bgColor));
  94     frame.health = frame:CreateTexture(nil, "BORDER");
  95     frame.health:SetPoint("TOPLEFT", frame.healthback, "TOPLEFT");
  96     frame.health:SetPoint("BOTTOMLEFT", frame.healthback, "BOTTOMLEFT");
  97     frame.health:SetTexture(barTexture);
  98     frame.health:SetVertexColor(unpack(healthColor));
  99     frame.healthText = frame:CreateFontString(nil, "ARTWORK", "GameFontHighlight");
 100     frame.healthText:SetPoint("RIGHT", frame.healthback, "RIGHT", -2, 0);
 101     frame.targeticon = frame:CreateTexture(nil, "OVERLAY");
 102     frame.targeticon:SetPoint("CENTER", frame.healthback, "TOP");
 103     frame.targeticon:SetWidth(18);
 104     frame.targeticon:SetHeight(18);
 105     frame.targeticon:SetTexture("Interface\\TARGETINGFRAME\\UI-RaidTargetingIcons");
 106     frame.targeticon:Hide();
 107     -- set scripts
 108     frame:SetScript("OnShow", frameShow);
 109     frame:SetScript("OnHide", frameHide);
 110     frame:SetScript("OnEvent", unitEvent);
 111     secure:SetScript("OnEnter", showTooltip);
 112     secure:SetScript("OnLeave", hideTooltip);
 113     -- set attributes
 114     secure:RegisterForClicks("AnyDown");
 115     for attr, val in pairs(attributes) do
 116         secure:SetAttribute(attr, val);
 117     end
 118     -- rest give target and menu
 119     secure:SetAttribute("*type1", "target");
 120     secure:SetAttribute("*type2", "togglemenu");
 121     secure:SetAttribute("toggleForVehicle", true);
 122     RegisterUnitWatch(frame);
 123     RegisterUnitWatch(secure);
 124     RegisterStateDriver(secure, "vehicleui", "[vehicleui] vehicle; no");
 125     secure:SetAttribute("_onstate-vehicleui", vehicletoggle);
 126     return frame;
 127 end