ElvUI/Modules/UnitFrames/Groups/Arena.lua

209 lines
8.4 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.')
local _G = _G
local unpack = unpack
local CreateFrame = CreateFrame
local IsAddOnLoaded = IsAddOnLoaded
local GetSpecializationInfoByID = GetSpecializationInfoByID
local LOCALIZED_CLASS_NAMES_MALE = LOCALIZED_CLASS_NAMES_MALE
local ArenaHeader = CreateFrame('Frame', 'ArenaHeader', E.UIParent)
function UF:ToggleArenaPreparationInfo(frame, show, specName, specTexture, specClass)
if not (frame and frame.ArenaPrepSpec and frame.ArenaPrepIcon) then return end
local specIcon = (frame.db and frame.db.pvpSpecIcon) and frame:IsElementEnabled('PVPSpecIcon')
frame.forceInRange = show -- used to force unitframe range
if show then -- during `PostUpdateArenaPreparation` this means spec class and name exist
frame.ArenaPrepSpec:SetText(specName..' - '..LOCALIZED_CLASS_NAMES_MALE[specClass])
frame.Health.value:Hide()
if specIcon then
frame.ArenaPrepIcon:SetTexture(specTexture or [[INTERFACE\ICONS\INV_MISC_QUESTIONMARK]])
frame.ArenaPrepIcon.bg:Show()
frame.ArenaPrepIcon:Show()
frame.PVPSpecIcon:Hide()
end
else -- mainly called from `PostUpdateArenaFrame` to hide them
frame.ArenaPrepSpec:SetText('')
frame.Health.value:Show()
if specIcon then
frame.ArenaPrepIcon.bg:Hide()
frame.ArenaPrepIcon:Hide()
frame.PVPSpecIcon:Show()
end
end
end
function UF:PostUpdateArenaFrame(event)
if self and event and (event ~= 'ARENA_PREP_OPPONENT_SPECIALIZATIONS' and event ~= 'PLAYER_ENTERING_WORLD') then
UF:ToggleArenaPreparationInfo(self)
end
end
function UF:PostUpdateArenaPreparation(_, specID)
local _, specName, specTexture, specClass
if specID and specID > 0 then
_, specName, _, specTexture, _, specClass = GetSpecializationInfoByID(specID)
end
UF:ToggleArenaPreparationInfo(self and self.__owner, specClass and specName, specName, specTexture, specClass)
end
function UF:Construct_ArenaFrames(frame)
frame.RaisedElementParent = CreateFrame('Frame', nil, frame)
frame.RaisedElementParent.TextureParent = CreateFrame('Frame', nil, frame.RaisedElementParent)
frame.RaisedElementParent:SetFrameLevel(frame:GetFrameLevel() + 100)
frame.Health = UF:Construct_HealthBar(frame, true, true, 'RIGHT')
frame.Name = UF:Construct_NameText(frame)
if not frame.isChild then
frame.Power = UF:Construct_PowerBar(frame, true, true, 'LEFT')
frame.PowerPrediction = UF:Construct_PowerPrediction(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)
frame.HealthPrediction = UF:Construct_HealComm(frame)
frame.MouseGlow = UF:Construct_MouseGlow(frame)
frame.TargetGlow = UF:Construct_TargetGlow(frame)
frame.FocusGlow = UF:Construct_FocusGlow(frame)
frame.Trinket = UF:Construct_Trinket(frame)
frame.PVPSpecIcon = UF:Construct_PVPSpecIcon(frame)
frame.PvPClassificationIndicator = UF:Construct_PvPClassificationIndicator(frame) -- Cart / Flag / Orb / Assassin Bounty
frame.Fader = UF:Construct_Fader()
frame:SetAttribute('type2', 'focus')
frame.customTexts = {}
frame.InfoPanel = UF:Construct_InfoPanel(frame)
frame.unitframeType = 'arena'
-- Arena Preparation
frame.ArenaPrepIcon = frame:CreateTexture(nil, 'OVERLAY')
frame.ArenaPrepIcon.bg = CreateFrame('Frame', nil, frame, 'BackdropTemplate')
frame.ArenaPrepIcon.bg:SetAllPoints(frame.PVPSpecIcon.bg)
frame.ArenaPrepIcon.bg:SetTemplate()
frame.ArenaPrepIcon:SetParent(frame.ArenaPrepIcon.bg)
frame.ArenaPrepIcon:SetTexCoord(unpack(E.TexCoords))
frame.ArenaPrepIcon:SetInside(frame.ArenaPrepIcon.bg)
frame.ArenaPrepIcon.bg:Hide()
frame.ArenaPrepIcon:Hide()
frame.ArenaPrepSpec = frame.Health:CreateFontString(nil, 'OVERLAY')
frame.ArenaPrepSpec:Point('CENTER')
UF:Configure_FontString(frame.ArenaPrepSpec)
frame.Health.PostUpdateArenaPreparation = self.PostUpdateArenaPreparation -- used to update arena prep info
frame.PostUpdate = self.PostUpdateArenaFrame -- used to hide arena prep info
end
frame.Cutaway = UF:Construct_Cutaway(frame)
ArenaHeader:Point('BOTTOMRIGHT', E.UIParent, 'RIGHT', -105, -165)
E:CreateMover(ArenaHeader, ArenaHeader:GetName()..'Mover', L["Arena Frames"], nil, nil, nil, 'ALL,ARENA', nil, 'unitframe,groupUnits,arena,generalGroup')
frame.mover = ArenaHeader.mover
end
function UF:Update_ArenaFrames(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
frame.PORTRAIT_WIDTH = (frame.USE_PORTRAIT_OVERLAY or not frame.USE_PORTRAIT) and 0 or db.portrait.width
frame.CLASSBAR_YOFFSET = 0
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)
frame.PVPINFO_WIDTH = db.pvpSpecIcon and frame.UNIT_HEIGHT or 0
end
if not IsAddOnLoaded('Clique') then
if db.middleClickFocus then
frame:SetAttribute('type3', 'focus')
elseif frame:GetAttribute('type3') == 'focus' then
frame:SetAttribute('type3', nil)
end
end
frame.colors = ElvUF.colors
frame:RegisterForClicks(self.db.targetOnMouseDown and 'AnyDown' or 'AnyUp')
frame:Size(frame.UNIT_WIDTH, frame.UNIT_HEIGHT)
UF:Configure_InfoPanel(frame)
UF:Configure_HealthBar(frame)
UF:UpdateNameSettings(frame)
UF:Configure_Power(frame)
UF:Configure_PowerPrediction(frame)
UF:Configure_Portrait(frame)
UF:EnableDisable_Auras(frame)
UF:Configure_AllAuras(frame)
UF:Configure_Castbar(frame)
UF:Configure_PVPSpecIcon(frame)
UF:Configure_Trinket(frame)
UF:Configure_Fader(frame)
UF:Configure_HealComm(frame)
UF:Configure_Cutaway(frame)
UF:Configure_CustomTexts(frame)
UF:Configure_PvPClassificationIndicator(frame)
frame:ClearAllPoints()
if frame.index == 1 then
local ArenaHeaderMover = _G.ArenaHeaderMover
if db.growthDirection == 'UP' then
frame:Point('BOTTOMRIGHT', ArenaHeaderMover, 'BOTTOMRIGHT')
elseif db.growthDirection == 'RIGHT' then
frame:Point('LEFT', ArenaHeaderMover, 'LEFT')
elseif db.growthDirection == 'LEFT' then
frame:Point('RIGHT', ArenaHeaderMover, 'RIGHT')
else --Down
frame:Point('TOPRIGHT', ArenaHeaderMover, 'TOPRIGHT')
end
else
if db.growthDirection == 'UP' then
frame:Point('BOTTOMRIGHT', _G['ElvUF_Arena'..frame.index-1], 'TOPRIGHT', 0, db.spacing)
elseif db.growthDirection == 'RIGHT' then
frame:Point('LEFT', _G['ElvUF_Arena'..frame.index-1], 'RIGHT', db.spacing, 0)
elseif db.growthDirection == 'LEFT' then
frame:Point('RIGHT', _G['ElvUF_Arena'..frame.index-1], 'LEFT', -db.spacing, 0)
else --Down
frame:Point('TOPRIGHT', _G['ElvUF_Arena'..frame.index-1], 'BOTTOMRIGHT', 0, -db.spacing)
end
end
if db.growthDirection == 'UP' or db.growthDirection == 'DOWN' then
ArenaHeader:Width(frame.UNIT_WIDTH)
ArenaHeader:Height(frame.UNIT_HEIGHT + ((frame.UNIT_HEIGHT + db.spacing) * 4))
elseif db.growthDirection == 'LEFT' or db.growthDirection == 'RIGHT' then
ArenaHeader:Width(frame.UNIT_WIDTH + ((frame.UNIT_WIDTH + db.spacing) * 4))
ArenaHeader:Height(frame.UNIT_HEIGHT)
end
frame:UpdateAllElements('ElvUI_UpdateAllElements')
end
UF.unitgroupstoload.arena = {5, 'ELVUI_UNITTARGET'}