ElvUI/Modules/UnitFrames/Units/Pet.lua

100 lines
4.3 KiB
Lua
Raw Permalink Normal View History

2020-11-13 14:27:50 -05:00
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.')
local _G = _G
local tinsert = tinsert
-- GLOBALS: ElvUF_Player
function UF:Construct_PetFrame(frame)
frame.Health = UF:Construct_HealthBar(frame, true, true, 'RIGHT')
frame.Power = UF:Construct_PowerBar(frame, true, true, 'LEFT')
frame.PowerPrediction = UF:Construct_PowerPrediction(frame)
frame.Name = UF:Construct_NameText(frame)
frame.Portrait3D = UF:Construct_Portrait(frame, 'model')
frame.Portrait2D = UF:Construct_Portrait(frame, 'texture')
frame.Buffs = UF:Construct_Buffs(frame)
frame.Debuffs = UF:Construct_Debuffs(frame)
frame.Castbar = UF:Construct_Castbar(frame, L["Pet Castbar"])
frame.Castbar.SafeZone = nil
frame.Castbar.LatencyTexture:Hide()
frame.ThreatIndicator = UF:Construct_Threat(frame)
frame.HealthPrediction = UF:Construct_HealComm(frame)
frame.AuraWatch = UF:Construct_AuraWatch(frame)
frame.AuraBars = UF:Construct_AuraBarHeader(frame)
frame.InfoPanel = UF:Construct_InfoPanel(frame)
frame.MouseGlow = UF:Construct_MouseGlow(frame)
frame.FocusGlow = UF:Construct_FocusGlow(frame)
frame.TargetGlow = UF:Construct_TargetGlow(frame)
frame.Fader = UF:Construct_Fader()
frame.Cutaway = UF:Construct_Cutaway(frame)
frame.customTexts = {}
frame:Point('BOTTOM', E.UIParent, 'BOTTOM', -342, 100)
E:CreateMover(frame, frame:GetName()..'Mover', L["Pet Frame"], nil, nil, nil, 'ALL,SOLO', nil, 'unitframe,individualUnits,pet,generalGroup')
frame.unitframeType = 'pet'
end
function UF:Update_PetFrame(frame, db)
frame.db = db
do
frame.ORIENTATION = db.orientation --allow this value to change when unitframes position changes on screen?
frame.UNIT_WIDTH = db.width
frame.UNIT_HEIGHT = db.infoPanel.enable and (db.height + db.infoPanel.height) or db.height
frame.USE_POWERBAR = db.power.enable
frame.POWERBAR_DETACHED = db.power.detachFromFrame
frame.USE_INSET_POWERBAR = not frame.POWERBAR_DETACHED and db.power.width == 'inset' and frame.USE_POWERBAR
frame.USE_MINI_POWERBAR = (not frame.POWERBAR_DETACHED and db.power.width == 'spaced' and frame.USE_POWERBAR)
frame.USE_POWERBAR_OFFSET = (db.power.width == 'offset' and db.power.offset ~= 0) and frame.USE_POWERBAR and not frame.POWERBAR_DETACHED
frame.POWERBAR_OFFSET = frame.USE_POWERBAR_OFFSET and db.power.offset or 0
frame.POWERBAR_HEIGHT = not frame.USE_POWERBAR and 0 or db.power.height
frame.POWERBAR_WIDTH = frame.USE_MINI_POWERBAR and (frame.UNIT_WIDTH - (UF.BORDER*2))/2 or (frame.POWERBAR_DETACHED and db.power.detachedWidth or (frame.UNIT_WIDTH - ((UF.BORDER+UF.SPACING)*2)))
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.USE_INFO_PANEL = not frame.USE_MINI_POWERBAR and not frame.USE_POWERBAR_OFFSET and db.infoPanel.enable
frame.INFO_PANEL_HEIGHT = frame.USE_INFO_PANEL and db.infoPanel.height or 0
frame.BOTTOM_OFFSET = UF:GetHealthBottomOffset(frame)
end
if db.strataAndLevel and db.strataAndLevel.useCustomStrata then
frame:SetFrameStrata(db.strataAndLevel.frameStrata)
end
if db.strataAndLevel and db.strataAndLevel.useCustomLevel then
frame:SetFrameLevel(db.strataAndLevel.frameLevel)
end
frame.Health.colorPetByUnitClass = db.health.colorPetByUnitClass
frame.colors = ElvUF.colors
frame:RegisterForClicks(self.db.targetOnMouseDown and 'AnyDown' or 'AnyUp')
frame:Size(frame.UNIT_WIDTH, frame.UNIT_HEIGHT)
_G[frame:GetName()..'Mover']:Size(frame:GetSize())
UF:Configure_InfoPanel(frame)
UF:Configure_HealthBar(frame)
UF:UpdateNameSettings(frame)
UF:Configure_Power(frame)
UF:Configure_PowerPrediction(frame)
UF:Configure_Portrait(frame)
UF:Configure_Threat(frame)
UF:EnableDisable_Auras(frame)
UF:Configure_AllAuras(frame)
UF:Configure_Fader(frame)
UF:Configure_Castbar(frame)
UF:Configure_HealComm(frame)
UF:Configure_AuraBars(frame)
UF:Configure_Cutaway(frame)
UF:Configure_CustomTexts(frame)
UF:Configure_AuraWatch(frame, true)
frame:UpdateAllElements('ElvUI_UpdateAllElements')
end
tinsert(UF.unitstoload, 'pet')