Andrew6810
312 lines
11 KiB
Lua
312 lines
11 KiB
Lua
local E, L, V, P, G = unpack(select(2, ...)); --Import: Engine, Locales, PrivateDB, ProfileDB, GlobalDB
|
|
local UF = E:GetModule("UnitFrames")
|
|
|
|
--Lua functions
|
|
local random = random
|
|
--WoW API / Variables
|
|
local CreateFrame = CreateFrame
|
|
|
|
local _, ns = ...
|
|
local ElvUF = ns.oUF
|
|
assert(ElvUF, "ElvUI was unable to locate oUF.")
|
|
|
|
function UF:Construct_EnergyBar(frame, bg, text, textPos)
|
|
local energy = CreateFrame("StatusBar", nil, frame)
|
|
UF.statusbars[energy] = true
|
|
|
|
energy.RaisedElementParent = CreateFrame("Frame", nil, energy)
|
|
energy.RaisedElementParent:SetFrameLevel(energy:GetFrameLevel() + 100)
|
|
energy.RaisedElementParent:SetAllPoints()
|
|
|
|
energy.PostUpdate = self.PostUpdateEnergy
|
|
energy.PostUpdateColor = self.PostUpdateEnergyColor
|
|
|
|
if bg then
|
|
energy.BG = energy:CreateTexture(nil, "BORDER")
|
|
energy.BG:SetAllPoints()
|
|
energy.BG:SetTexture(E.media.blankTex)
|
|
end
|
|
|
|
if text then
|
|
energy.value = frame.RaisedElementParent:CreateFontString(nil, "OVERLAY")
|
|
UF:Configure_FontString(energy.value)
|
|
|
|
local x = -2
|
|
if textPos == "LEFT" then
|
|
x = 2
|
|
end
|
|
|
|
energy.value:Point(textPos, frame.Health, textPos, x, 0)
|
|
|
|
energy.value.frequentUpdates = true
|
|
end
|
|
|
|
energy.colorDisconnected = false
|
|
energy.colorTapping = false
|
|
energy:CreateBackdrop("Default", nil, nil, self.thinBorders, true)
|
|
|
|
local clipFrame = CreateFrame('Frame', nil, energy)
|
|
clipFrame:SetAllPoints()
|
|
clipFrame:EnableMouse(false)
|
|
clipFrame.__frame = frame
|
|
energy.ClipFrame = clipFrame
|
|
|
|
return energy
|
|
end
|
|
|
|
function UF:Configure_Energy(frame)
|
|
if not frame.VARIABLES_SET then return end
|
|
local db = frame.db
|
|
local energy = frame.Energy
|
|
energy.origParent = frame
|
|
|
|
if frame.USE_ENERGYBAR then
|
|
if not frame:IsElementEnabled("Energy") then
|
|
frame:EnableElement("Energy")
|
|
energy:Show()
|
|
end
|
|
|
|
E:SetSmoothing(energy, self.db.smoothbars)
|
|
|
|
--Text
|
|
local attachPoint = self:GetObjectAnchorPoint(frame, db.energy.attachTextTo)
|
|
energy.value:ClearAllPoints()
|
|
energy.value:Point(db.energy.position, attachPoint, db.energy.position, db.energy.xOffset, db.energy.yOffset)
|
|
frame:Tag(energy.value, db.energy.text_format)
|
|
|
|
if db.energy.attachTextTo == "Energy" then
|
|
energy.value:SetParent(energy.RaisedElementParent)
|
|
else
|
|
energy.value:SetParent(frame.RaisedElementParent)
|
|
end
|
|
|
|
--Colors
|
|
energy.colorClass = nil
|
|
energy.colorReaction = nil
|
|
energy.colorEnergy = nil
|
|
|
|
if self.db.colors.energyclass then
|
|
energy.colorClass = true
|
|
energy.colorReaction = true
|
|
else
|
|
energy.colorEnergy = true
|
|
end
|
|
|
|
--Fix height in case it is lower than the theme allows
|
|
local heightChanged = false
|
|
if (not self.thinBorders and not E.PixelMode) and frame.ENERGYBAR_HEIGHT < 7 then --A height of 7 means 6px for borders and just 1px for the actual energy statusbar
|
|
frame.ENERGYBAR_HEIGHT = 7
|
|
if db.energy then db.energy.height = 7 end
|
|
heightChanged = true
|
|
elseif (self.thinBorders or E.PixelMode) and frame.ENERGYBAR_HEIGHT < 3 then --A height of 3 means 2px for borders and just 1px for the actual energy statusbar
|
|
frame.ENERGYBAR_HEIGHT = 3
|
|
if db.energy then db.energy.height = 3 end
|
|
heightChanged = true
|
|
end
|
|
if heightChanged then
|
|
--Update health size
|
|
frame.BOTTOM_OFFSET = UF:GetHealthBottomOffset(frame)
|
|
UF:Configure_HealthBar(frame)
|
|
end
|
|
|
|
--Position
|
|
energy:ClearAllPoints()
|
|
if frame.ENERGYBAR_DETACHED then
|
|
energy:Width(frame.ENERGYBAR_WIDTH - ((frame.BORDER + frame.SPACING)*2))
|
|
energy:Height(frame.ENERGYBAR_HEIGHT - ((frame.BORDER + frame.SPACING)*2))
|
|
if not energy.Holder or (energy.Holder and not energy.Holder.mover) then
|
|
energy.Holder = CreateFrame("Frame", nil, energy)
|
|
energy.Holder:Size(frame.ENERGYBAR_WIDTH, frame.ENERGYBAR_HEIGHT)
|
|
energy.Holder:Point("BOTTOM", frame, "BOTTOM", 0, -20)
|
|
energy:ClearAllPoints()
|
|
energy:Point("BOTTOMLEFT", energy.Holder, "BOTTOMLEFT", frame.BORDER+frame.SPACING, frame.BORDER+frame.SPACING)
|
|
--Currently only Player and Target can detach energy bars, so doing it this way is okay for now
|
|
if frame.unitframeType and frame.unitframeType == "player" then
|
|
E:CreateMover(energy.Holder, "PlayerEnergyBarMover", L["Player Energybar"], nil, nil, nil, "ALL,SOLO", nil, "unitframe,player,energy")
|
|
elseif frame.unitframeType and frame.unitframeType == "target" then
|
|
E:CreateMover(energy.Holder, "TargetEnergyBarMover", L["Target Energybar"], nil, nil, nil, "ALL,SOLO", nil, "unitframe,target,energy")
|
|
end
|
|
else
|
|
energy.Holder:Size(frame.ENERGYBAR_WIDTH, frame.ENERGYBAR_HEIGHT)
|
|
energy:ClearAllPoints()
|
|
energy:Point("BOTTOMLEFT", energy.Holder, "BOTTOMLEFT", frame.BORDER+frame.SPACING, frame.BORDER+frame.SPACING)
|
|
energy.Holder.mover:SetScale(1)
|
|
energy.Holder.mover:SetAlpha(1)
|
|
end
|
|
|
|
energy:SetFrameLevel(50) --RaisedElementParent uses 100, we want lower value to allow certain icons and texts to appear above energy
|
|
elseif frame.USE_ENERGYBAR_OFFSET then
|
|
local anchor = frame.Health
|
|
if frame.USE_POWERBAR and frame.USE_POWERBAR_OFFSET then
|
|
anchor = frame.Power
|
|
end
|
|
if frame.ORIENTATION == "LEFT" then
|
|
energy:Point("TOPRIGHT", anchor, "TOPRIGHT", frame.ENERGYBAR_OFFSET + (frame.HAPPINESS_WIDTH or 0), -frame.ENERGYBAR_OFFSET)
|
|
energy:Point("BOTTOMLEFT", anchor, "BOTTOMLEFT", frame.ENERGYBAR_OFFSET, -frame.ENERGYBAR_OFFSET)
|
|
elseif frame.ORIENTATION == "MIDDLE" then
|
|
local preOffset = 0
|
|
if frame.USE_POWERBAR and frame.USE_POWERBAR_OFFSET then
|
|
preOffset = preOffset + frame.POWERBAR_OFFSET
|
|
end
|
|
local postOffset = 0
|
|
if frame.USE_RAGEBAR and frame.USE_RAGEBAR_OFFSET then
|
|
postOffset = postOffset + frame.RAGEBAR_OFFSET
|
|
end
|
|
|
|
energy:Point("TOPLEFT", frame, "TOPLEFT",
|
|
frame.BORDER + frame.SPACING + postOffset,
|
|
-(preOffset + frame.ENERGYBAR_OFFSET + frame.CLASSBAR_YOFFSET) --+ frame.BORDER - frame.SPACING)
|
|
)
|
|
energy:Point("BOTTOMRIGHT", frame, "BOTTOMRIGHT",
|
|
-(frame.BORDER + frame.SPACING + postOffset),
|
|
frame.BORDER + frame.SPACING + postOffset
|
|
)
|
|
else
|
|
energy:Point("TOPLEFT", anchor, "TOPLEFT", -frame.ENERGYBAR_OFFSET - (frame.HAPPINESS_WIDTH or 0), -frame.ENERGYBAR_OFFSET)
|
|
energy:Point("BOTTOMRIGHT", anchor, "BOTTOMRIGHT", -frame.ENERGYBAR_OFFSET, -frame.ENERGYBAR_OFFSET)
|
|
end
|
|
energy:SetFrameLevel(frame.Health:GetFrameLevel() - 6) --Health uses 10
|
|
elseif frame.USE_INSET_ENERGYBAR then
|
|
if frame.USE_INSET_RAGEBAR and frame.USE_RAGEBAR then
|
|
energy:Point("BOTTOMLEFT", frame.Rage, "TOPLEFT",
|
|
0,
|
|
frame.BORDER * 2
|
|
)
|
|
energy:Point("BOTTOMRIGHT", frame.Rage, "TOPRIGHT",
|
|
0,
|
|
frame.BORDER * 2
|
|
)
|
|
else
|
|
energy:Point("BOTTOMLEFT", frame.Health, "BOTTOMLEFT",
|
|
frame.BORDER + frame.BORDER * 2,
|
|
frame.BORDER + frame.BORDER * 2
|
|
)
|
|
energy:Point("BOTTOMRIGHT", frame.Health, "BOTTOMRIGHT",
|
|
-(frame.BORDER + frame.BORDER * 2),
|
|
frame.BORDER + frame.BORDER * 2
|
|
)
|
|
end
|
|
|
|
energy:Height(frame.ENERGYBAR_HEIGHT - (frame.BORDER + frame.SPACING) * 2)
|
|
energy:SetFrameLevel(50)
|
|
elseif frame.USE_MINI_ENERGYBAR then
|
|
local totalHeight = frame.ENERGYBAR_HEIGHT - frame.BORDER
|
|
if frame.USE_POWERBAR and frame.USE_MINI_POWERBAR then
|
|
totalHeight = totalHeight + (frame.POWERBAR_HEIGHT - frame.BORDER)
|
|
end
|
|
if frame.USE_RAGEBAR and frame.USE_MINI_RAGEBAR then
|
|
totalHeight = totalHeight + (frame.RAGEBAR_HEIGHT - frame.BORDER)
|
|
end
|
|
local yPos = (totalHeight / 2) - (frame.POWERBAR_HEIGHT - frame.BORDER)
|
|
|
|
if frame.ORIENTATION == "LEFT" then
|
|
energy:Width(frame.ENERGYBAR_WIDTH - frame.BORDER * 2)
|
|
energy:Point("TOPRIGHT", frame.Health, "BOTTOMRIGHT",
|
|
-(frame.BORDER * 2 + 4) - (frame.HAPPINESS_WIDTH or 0),
|
|
yPos
|
|
)
|
|
elseif frame.ORIENTATION == "RIGHT" then
|
|
energy:Width(frame.ENERGYBAR_WIDTH - frame.BORDER*2)
|
|
energy:Point("TOPLEFT", frame.Health, "BOTTOMLEFT",
|
|
frame.BORDER * 2 + 4 + (frame.HAPPINESS_WIDTH or 0),
|
|
yPos
|
|
)
|
|
else
|
|
energy:Point("TOPLEFT", frame.Health, "BOTTOMLEFT",
|
|
frame.BORDER * 2 + 4,
|
|
yPos
|
|
)
|
|
energy:Point("TOPRIGHT", frame.Health, "BOTTOMRIGHT",
|
|
-(frame.BORDER * 2 + 4) - (frame.HAPPINESS_WIDTH or 0),
|
|
yPos
|
|
)
|
|
end
|
|
|
|
energy:Height(frame.ENERGYBAR_HEIGHT - (frame.BORDER + frame.SPACING) * 2)
|
|
energy:SetFrameLevel(50)
|
|
else -- Filled
|
|
local anchor = frame.Power.backdrop
|
|
if not frame.USE_POWERBAR or frame.USE_POWERBAR_DETACHED or frame.USE_INSET_POWERBAR or frame.USE_MINI_POWERBAR then
|
|
anchor = frame.Health.backdrop
|
|
end
|
|
energy:Point("TOPLEFT", anchor, "BOTTOMLEFT",
|
|
frame.BORDER,
|
|
-frame.SPACING * 3
|
|
)
|
|
energy:Point("TOPRIGHT", anchor, "BOTTOMRIGHT",
|
|
-frame.BORDER,
|
|
-frame.SPACING * 3
|
|
)
|
|
energy:Height(frame.ENERGYBAR_HEIGHT - (frame.BORDER + frame.SPACING) * 2)
|
|
energy:SetFrameLevel(frame.Health:GetFrameLevel() - 5)
|
|
end
|
|
|
|
--Hide mover until we detach again
|
|
if not frame.ENERGYBAR_DETACHED then
|
|
if energy.Holder and energy.Holder.mover then
|
|
energy.Holder.mover:SetScale(0.0001)
|
|
energy.Holder.mover:SetAlpha(0)
|
|
end
|
|
end
|
|
|
|
if db.energy.strataAndLevel and db.energy.strataAndLevel.useCustomStrata then
|
|
energy:SetFrameStrata(db.energy.strataAndLevel.frameStrata)
|
|
else
|
|
energy:SetFrameStrata("LOW")
|
|
end
|
|
if db.energy.strataAndLevel and db.energy.strataAndLevel.useCustomLevel then
|
|
energy:SetFrameLevel(db.energy.strataAndLevel.frameLevel)
|
|
energy.backdrop:SetFrameLevel(energy:GetFrameLevel() - 1)
|
|
end
|
|
|
|
if frame.ENERGYBAR_DETACHED and db.energy.parent == "UIPARENT" then
|
|
energy:SetParent(E.UIParent)
|
|
else
|
|
energy:SetParent(frame)
|
|
end
|
|
elseif frame:IsElementEnabled("Energy") then
|
|
frame:DisableElement("Energy")
|
|
energy:Hide()
|
|
frame:Tag(energy.value, "")
|
|
end
|
|
|
|
energy.custom_backdrop = UF.db.colors.customenergybackdrop and UF.db.colors.energy_backdrop
|
|
|
|
--Transparency Settings
|
|
UF:ToggleTransparentStatusBar(UF.db.colors.transparentEnergy, energy, energy.BG, nil, UF.db.colors.invertEnergy)
|
|
end
|
|
|
|
local tokens = {[0] = "MANA", "RAGE", "FOCUS", "ENERGY", "RUNIC_POWER"}
|
|
function UF:PostUpdateEnergyColor()
|
|
local parent = self.origParent or self:GetParent()
|
|
|
|
if parent.isForced then
|
|
local color = ElvUF.colors.energy[tokens[random(0, 4)]]
|
|
self:SetValue(random(1, self.max))
|
|
|
|
if not self.colorClass then
|
|
self:SetStatusBarColor(color[1], color[2], color[3])
|
|
|
|
if self.BG then
|
|
UF:UpdateBackdropTextureColor(self.BG, color[1], color[2], color[3])
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function UF:PostUpdateEnergy(unit)
|
|
local parent = self.origParent or self:GetParent()
|
|
if parent.isForced then
|
|
self:SetValue(random(1, self.max))
|
|
end
|
|
|
|
if parent.db and parent.db.energy and parent.db.energy.hideonnpc then
|
|
UF:PostNamePosition(parent, unit)
|
|
end
|
|
|
|
--Force update to AdditionalPower in order to reposition text if necessary
|
|
if parent:IsElementEnabled("AdditionalPower") then
|
|
E:Delay(0.01, parent.AdditionalPower.ForceUpdate, parent.AdditionalPower) --Delay it slightly so Power text has a chance to clear itself first
|
|
end
|
|
end
|