Files
coa-elvui/ElvUI/Modules/UnitFrames/Groups/RaidPets.lua
T
Andrew6810 60ef8a38af init
2022-10-21 07:09:01 -07:00

189 lines
5.5 KiB
Lua

local E, L, V, P, G = unpack(select(2, ...)); --Import: Engine, Locales, PrivateDB, ProfileDB, GlobalDB
local UF = E:GetModule("UnitFrames")
local _, ns = ...
local ElvUF = ns.oUF
assert(ElvUF, "ElvUI was unable to locate oUF.")
--Lua functions
--WoW API / Variables
local CreateFrame = CreateFrame
local GetInstanceInfo = GetInstanceInfo
local InCombatLockdown = InCombatLockdown
local RegisterStateDriver = RegisterStateDriver
local UnregisterStateDriver = UnregisterStateDriver
function UF:Construct_RaidpetFrames()
self:SetScript("OnEnter", UnitFrame_OnEnter)
self:SetScript("OnLeave", UnitFrame_OnLeave)
self.RaisedElementParent = CreateFrame("Frame", nil, self)
self.RaisedElementParent.TextureParent = CreateFrame("Frame", nil, self.RaisedElementParent)
self.RaisedElementParent:SetFrameLevel(self:GetFrameLevel() + 100)
self.Health = UF:Construct_HealthBar(self, true, true, "RIGHT")
self.Name = UF:Construct_NameText(self)
self.Portrait3D = UF:Construct_Portrait(self, "model")
self.Portrait2D = UF:Construct_Portrait(self, "texture")
self.Buffs = UF:Construct_Buffs(self)
self.Debuffs = UF:Construct_Debuffs(self)
self.AuraWatch = UF:Construct_AuraWatch(self)
self.RaidDebuffs = UF:Construct_RaidDebuffs(self)
self.DebuffHighlight = UF:Construct_DebuffHighlight(self)
self.TargetGlow = UF:Construct_TargetGlow(self)
self.MouseGlow = UF:Construct_MouseGlow(self)
self.ThreatIndicator = UF:Construct_Threat(self)
self.RaidTargetIndicator = UF:Construct_RaidIcon(self)
self.HealCommBar = UF:Construct_HealComm(self)
self.Fader = UF:Construct_Fader()
self.Cutaway = UF:Construct_Cutaway(self)
self.customTexts = {}
self.unitframeType = "raidpet"
UF:Update_StatusBars()
UF:Update_FontStrings()
self.db = UF.db.units.raidpet
self.PostCreate = UF.Update_RaidpetFrames
return self
end
--I don"t know if this function is needed or not? But the error I pm'ed you about was because of the missing OnEvent so I just added it.
function UF:RaidPetsSmartVisibility(event)
if not self.db or (self.db and not self.db.enable) or (UF.db and not UF.db.smartRaidFilter) or self.isForced then return end
if event == "PLAYER_REGEN_ENABLED" then self:UnregisterEvent("PLAYER_REGEN_ENABLED") end
if not InCombatLockdown() then
local _, instanceType = GetInstanceInfo()
if instanceType == "raid" then
UnregisterStateDriver(self, "visibility")
self:Show()
elseif self.db.visibility then
RegisterStateDriver(self, "visibility", self.db.visibility)
end
else
self:RegisterEvent("PLAYER_REGEN_ENABLED")
return
end
end
function UF:Update_RaidpetHeader(header, db)
header.db = db
local headerHolder = header:GetParent()
headerHolder.db = db
if not headerHolder.positioned then
headerHolder:ClearAllPoints()
headerHolder:Point("BOTTOMLEFT", E.UIParent, "BOTTOMLEFT", 4, 574)
E:CreateMover(headerHolder, headerHolder:GetName().."Mover", L["Raid Pet Frames"], nil, nil, nil, "ALL,RAID10,RAID25,RAID40", nil, "unitframe,raidpet,generalGroup")
headerHolder.positioned = true
headerHolder:RegisterEvent("PLAYER_LOGIN")
headerHolder:RegisterEvent("ZONE_CHANGED_NEW_AREA")
headerHolder:SetScript("OnEvent", UF.RaidPetsSmartVisibility)
end
UF.RaidPetsSmartVisibility(headerHolder)
end
function UF:Update_RaidpetFrames(frame, db)
if not db then
db = frame.db
else
frame.db = db
end
frame.Portrait = frame.Portrait or (db.portrait.style == "2D" and frame.Portrait2D or frame.Portrait3D)
frame.colors = ElvUF.colors
frame:RegisterForClicks(self.db.targetOnMouseDown and "AnyDown" or "AnyUp")
do
if self.thinBorders then
frame.SPACING = 0
frame.BORDER = E.mult
else
frame.BORDER = E.Border
frame.SPACING = E.Spacing
end
frame.SHADOW_SPACING = 3
frame.ORIENTATION = db.orientation --allow this value to change when unitframes position changes on screen?
frame.UNIT_WIDTH = db.width
frame.UNIT_HEIGHT = db.height
frame.USE_POWERBAR = false
frame.POWERBAR_DETACHED = false
frame.USE_INSET_POWERBAR = false
frame.USE_MINI_POWERBAR = false
frame.USE_POWERBAR_OFFSET = false
frame.POWERBAR_OFFSET = 0
frame.POWERBAR_HEIGHT = 0
frame.POWERBAR_WIDTH = 0
frame.USE_PORTRAIT = db.portrait and db.portrait.enable
frame.USE_PORTRAIT_OVERLAY = frame.USE_PORTRAIT and (db.portrait.overlay or frame.ORIENTATION == "MIDDLE")
frame.PORTRAIT_WIDTH = (frame.USE_PORTRAIT_OVERLAY or not frame.USE_PORTRAIT) and 0 or db.portrait.width
frame.CLASSBAR_YOFFSET = 0
frame.BOTTOM_OFFSET = 0
frame.VARIABLES_SET = true
end
if not InCombatLockdown() then
frame:Size(frame.UNIT_WIDTH, frame.UNIT_HEIGHT)
else
frame:SetAttribute("initial-width", frame.UNIT_WIDTH)
frame:SetAttribute("initial-height", frame.UNIT_HEIGHT)
end
--Health
UF:Configure_HealthBar(frame)
--Name
UF:UpdateNameSettings(frame)
--Portrait
UF:Configure_Portrait(frame)
--Threat
UF:Configure_Threat(frame)
--Auras
UF:EnableDisable_Auras(frame)
UF:Configure_Auras(frame, "Buffs")
UF:Configure_Auras(frame, "Debuffs")
--RaidDebuffs
UF:Configure_RaidDebuffs(frame)
--Raid Icon
UF:Configure_RaidIcon(frame)
--Debuff Highlight
UF:Configure_DebuffHighlight(frame)
--OverHealing
UF:Configure_HealComm(frame)
--Fader
UF:Configure_Fader(frame)
--Cutaway
UF:Configure_Cutaway(frame)
--BuffIndicator
UF:UpdateAuraWatch(frame, true) --2nd argument is the petOverride
--CustomTexts
UF:Configure_CustomTexts(frame)
frame:UpdateAllElements("ForceUpdate")
end
--Added an additional argument at the end, specifying the header Template we want to use
UF.headerstoload.raidpet = {nil, nil, "SecureGroupPetHeaderTemplate"}