Pet.lua 4.26 KB
Newer Older
Azilroka's avatar
Azilroka committed
1
local E, L, V, P, G = unpack(select(2, ...)); --Import: Engine, Locales, PrivateDB, ProfileDB, GlobalDB
Simpy's avatar
Simpy committed
2 3
local UF = E:GetModule('UnitFrames')

4 5
local _, ns = ...
local ElvUF = ns.oUF
Simpy's avatar
Simpy committed
6
assert(ElvUF, 'ElvUI was unable to locate oUF.')
7 8

local _G = _G
Simpy's avatar
Simpy committed
9
local tinsert = tinsert
10 11 12
-- GLOBALS: ElvUF_Player

function UF:Construct_PetFrame(frame)
Azilroka's avatar
Azilroka committed
13 14 15 16 17 18 19 20 21
	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"])
22 23
	frame.Castbar.SafeZone = nil
	frame.Castbar.LatencyTexture:Hide()
Azilroka's avatar
Azilroka committed
24 25 26 27 28 29 30 31 32 33
	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)
34
	frame.customTexts = {}
Simpy's avatar
Simpy committed
35

Simpy's avatar
Simpy committed
36
	frame:Point('BOTTOM', E.UIParent, 'BOTTOM', -342, 100)
Simpy's avatar
Simpy committed
37
	E:CreateMover(frame, frame:GetName()..'Mover', L["Pet Frame"], nil, nil, nil, 'ALL,SOLO', nil, 'unitframe,individualUnits,pet,generalGroup')
38

Simpy's avatar
Simpy committed
39
	frame.unitframeType = 'pet'
40 41 42 43 44 45 46 47 48 49 50 51 52
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)
53
		frame.USE_POWERBAR_OFFSET = (db.power.width == 'offset' and db.power.offset ~= 0) and frame.USE_POWERBAR and not frame.POWERBAR_DETACHED
54 55
		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
56
		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)))
57
		frame.USE_PORTRAIT = db.portrait and db.portrait.enable
Simpy's avatar
Simpy committed
58
		frame.USE_PORTRAIT_OVERLAY = frame.USE_PORTRAIT and (db.portrait.overlay or frame.ORIENTATION == 'MIDDLE')
59 60 61 62 63 64
		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

65 66 67 68 69 70 71 72
	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

Azilroka's avatar
Azilroka committed
73
	frame.Health.colorPetByUnitClass = db.health.colorPetByUnitClass
74 75
	frame.colors = ElvUF.colors
	frame:RegisterForClicks(self.db.targetOnMouseDown and 'AnyDown' or 'AnyUp')
Simpy's avatar
Simpy committed
76 77
	frame:Size(frame.UNIT_WIDTH, frame.UNIT_HEIGHT)
	_G[frame:GetName()..'Mover']:Size(frame:GetSize())
78 79 80 81 82

	UF:Configure_InfoPanel(frame)
	UF:Configure_HealthBar(frame)
	UF:UpdateNameSettings(frame)
	UF:Configure_Power(frame)
Azilroka's avatar
Azilroka committed
83
	UF:Configure_PowerPrediction(frame)
84 85 86
	UF:Configure_Portrait(frame)
	UF:Configure_Threat(frame)
	UF:EnableDisable_Auras(frame)
87
	UF:Configure_AllAuras(frame)
Simpy's avatar
Simpy committed
88
	UF:Configure_Fader(frame)
89 90
	UF:Configure_Castbar(frame)
	UF:Configure_HealComm(frame)
Azilroka's avatar
Azilroka committed
91
	UF:Configure_AuraBars(frame)
92
	UF:Configure_Cutaway(frame)
93
	UF:Configure_CustomTexts(frame)
Azilroka's avatar
Azilroka committed
94
	UF:Configure_AuraWatch(frame, true)
95

Simpy's avatar
Simpy committed
96
	frame:UpdateAllElements('ElvUI_UpdateAllElements')
97 98
end

Simpy's avatar
Simpy committed
99
tinsert(UF.unitstoload, 'pet')