You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

136 lines
3.7 KiB

--[[
# Element: Portraits
Handles the updating of the unit's portrait.
## Widget
Portrait - A `PlayerModel` or a `Texture` used to represent the unit's portrait.
## Notes
A question mark model will be used if the widget is a PlayerModel and the client doesn't have the model information for
the unit.
## Examples
-- 3D Portrait
-- Position and size
local Portrait = CreateFrame('PlayerModel', nil, self)
Portrait:SetSize(32, 32)
Portrait:SetPoint('RIGHT', self, 'LEFT')
-- Register it with oUF
self.Portrait = Portrait
-- 2D Portrait
local Portrait = self:CreateTexture(nil, 'OVERLAY')
Portrait:SetSize(32, 32)
Portrait:SetPoint('RIGHT', self, 'LEFT')
-- Register it with oUF
self.PBPortrait = Portrait
--]]
local PA = _G.ProjectAzilroka
local oUF = PA.oUF
if not oUF then
return
end
local function Update(self, event, unit)
local petInfo = self.pbouf_petinfo
if not petInfo then
return
end
--[[ Callback: Portrait:PreUpdate(unit)
Called before the element has been updated.
* self - the Portrait element
* unit - the unit for which the update has been triggered (string)
--]]
local element = self.PBPortrait
if (element.PreUpdate) then
element:PreUpdate(unit)
end
local displayID = C_PetBattles.GetDisplayID(petInfo.petOwner, petInfo.petIndex)
local isAvailable = select(2, C_PetBattles.GetName(petInfo.petOwner, petInfo.petIndex)) ~= nil
element.stateChanged = event ~= "OnUpdate" or element.displayID ~= displayID or element.state ~= isAvailable
if element.stateChanged then -- ElvUI changed
element.playerModel = element:IsObjectType("PlayerModel")
element.state = isAvailable
element.displayID = displayID
if element.playerModel then
if not isAvailable then
element:SetCamDistanceScale(0.25)
element:SetPortraitZoom(0)
element:SetPosition(0, 0, 0.25)
element:ClearModel()
element:SetModel([[Interface\Buttons\TalkToMeQuestionMark.m2]])
else
element:SetCamDistanceScale(1)
element:SetPortraitZoom(1)
element:SetPosition(0, 0, 0)
element:ClearModel()
element:SetDisplayInfo(displayID)
end
elseif not element.customTexture then -- ElvUI changed
local icon = C_PetBattles.GetIcon(petInfo.petOwner, petInfo.petIndex)
element:SetTexture(icon)
end
end
--[[ Callback: Portrait:PostUpdate(unit)
Called after the element has been updated.
* self - the Portrait element
* unit - the unit for which the update has been triggered (string)
--]]
if (element.PostUpdate) then
return element:PostUpdate(unit, event)
end
end
local function Path(self, ...)
--[[ Override: Portrait.Override(self, event, unit)
Used to completely override the internal update function.
* self - the parent object
* event - the event triggering the update (string)
* unit - the unit accompanying the event (string)
--]]
return (self.PBPortrait.Override or Update)(self, ...)
end
local function ForceUpdate(element)
return Path(element.__owner, "ForceUpdate", element.__owner.unit)
end
local function Enable(self, unit)
local element = self.PBPortrait
if (element) then
element.__owner = self
element.ForceUpdate = ForceUpdate
self:RegisterEvent("PET_BATTLE_OPENING_START", Path, true)
self:RegisterEvent("PET_BATTLE_OPENING_DONE", Path, true)
element:Show()
return true
end
end
local function Disable(self)
local element = self.PBPortrait
if (element) then
element:Hide()
self:UnregisterEvent("PET_BATTLE_OPENING_START")
self:UnregisterEvent("PET_BATTLE_OPENING_DONE")
end
end
oUF:AddElement("PBPortrait", Path, Enable, Disable)