Totem-module with statusbars

[Mischback]

The Totem-module of oUF handles totems by default with an icon a cooldown spiral.

For a current project I wanted the remaining duration to be displayed as statusbars, as it has been done with oUF_boring_totembar.

Here's my solution, thought I share it:

In your layout-file (in the section of the player-frame, for my layout), build a totem-layout (I stripped down mine a lot, this is poc-quality):

if ( class == 'SHAMAN' ) then
        local t = CreateFrame('Frame', nil, self)
 
        t[4] = CreateFrame('StatusBar', nil, t)
        t[4]:SetSize(width, height)
        t[4]:SetPoint('RIGHT', self, 'TOPRIGHT', -spacing, 0)
        t[4]:SetStatusBarTexture(tex)
        t[4]:SetStatusBarColor(unpack(self.colors.totems[AIR_TOTEM_SLOT]))
        t[3] = CreateFrame('StatusBar', nil, t)
        t[3]:SetSize(width, height)
        t[3]:SetPoint('TOPRIGHT', t[4], 'TOPLEFT', -spacing, 0)
        t[3]:SetStatusBarTexture(tex)
        t[3]:SetStatusBarColor(unpack(self.colors.totems[WATER_TOTEM_SLOT]))
        t[2] = CreateFrame('StatusBar', nil, t)
        t[2]:SetSize(width, height)
        t[2]:SetPoint('TOPRIGHT', t[3], 'TOPLEFT', -spacing, 0)
        t[2]:SetStatusBarTexture(tex)
        t[2]:SetStatusBarColor(unpack(self.colors.totems[EARTH_TOTEM_SLOT]))
        t[1] = CreateFrame('StatusBar', nil, t)
        t[1]:SetSize(width, height)
        t[1]:SetPoint('TOPRIGHT', t[2], 'TOPLEFT', -spacing, 0)
        t[1]:SetStatusBarTexture(tex)
        t[1]:SetStatusBarColor(unpack(self.colors.totems[FIRE_TOTEM_SLOT]))
 
        self.Totems = t
        self.Totems.Override = core.TotemOverride
    end

As you can see, I override the default-Update-function, so here it is:

core.TotemOverride = function(self, event, slot)
    local totems = self.Totems
 
    local totem = totems[slot]
    local haveTotem, name, start, duration, icon = GetTotemInfo(slot)
    if(duration > 0) then
        totem:SetMinMaxValues(0, duration)
        totem.start = start
        totem.duration = duration
        totem.TimeSinceLastUpdate = 0
        totem:SetScript('OnUpdate', function(self, elapsed)
            self.TimeSinceLastUpdate = self.TimeSinceLastUpdate +  elapsed
            if ( self.TimeSinceLastUpdate > 0.5 ) then
                self.TimeSinceLastUpdate = 0
                self:SetValue(self.start+self.duration-GetTime())
            end
        end)
        totem:Show()
    else
        totem:Hide()
    end
end

You may of course throttle the update as you want it to be.

[Rainrider]

@Mischback
FIRE_TOTEM_SLOT and so on translate to the corresponding totem id, so you could use a for loop for creating the totems. Here is what I use in my layout in case someone is interested.

This is what I do:

local AddTotems = function(self, width, height)
    local numTotems = MAX_TOTEMS
 
    self.Totems = CreateFrame("Frame", "oUF_Rain_Totems", self.Overlay)
    self.Totems:SetHeight(height)
    self.Totems:SetPoint("BOTTOMLEFT", self.Overlay, 0, 1)
    self.Totems:SetPoint("BOTTOMRIGHT", self.Overlay, 0, 1)
    
    for i = 1, numTotems do
        self.Totems[i] = CreateFrame("StatusBar", "oUF_Rain_Totem"..i, self.Totems)
        self.Totems[i]:SetStatusBarTexture(ns.media.TEXTURE)
        self.Totems[i]:SetMinMaxValues(0, 1)
        
        if (playerClass == "SHAMAN") then
            self.Totems[i]:SetSize((215 - numTotems - 1) / numTotems, height)
            self.Totems[i]:SetPoint("BOTTOMLEFT", self.Totems, (i - 1) * (214 / numTotems) + 1, 0)
            self.Totems[i]:SetStatusBarColor(unpack(ns.colors.totems[i]))
        elseif (playerClass == "DRUID") then -- Druid's mushrooms
            self.Totems[i]:SetSize(width, height)
            self.Totems[i]:SetStatusBarColor(unpack(ns.colors.class[playerClass]))
                if (i == 1) then
                    self.Totems[i]:SetPoint("BOTTOM", self.Overlay, "TOP", 0, 0)
                elseif (i == 2) then
                    self.Totems[i]:SetPoint("RIGHT", self.Totems[1], "LEFT", -1, 0)
                else
                    self.Totems[i]:SetPoint("LEFT", self.Totems[1], "RIGHT", 1, 0)
                end
        elseif (playerClass) == "DEATHKNIGHT" then -- Death knight's ghoul
            self.Totems[i]:SetSize(width, height)
            self.Totems[i]:SetStatusBarColor(unpack(ns.colors.class[playerClass]))
            self.Totems[i]:SetPoint("BOTTOM", self.Overlay, "TOP", 0, 0)
        end
        
        self.Totems[i]:SetBackdrop(ns.media.BACKDROP)
        self.Totems[i]:SetBackdropColor(0, 0, 0)
        
        self.Totems[i].Destroy = CreateFrame("Button", nil, self.Totems[i])
        self.Totems[i].Destroy:SetAllPoints()
        self.Totems[i].Destroy:RegisterForClicks("RightButtonUp")
        self.Totems[i].Destroy:SetScript("OnClick", function()
            if (IsShiftKeyDown()) then
                DestroyTotem(i)
            end
        end)
        
        self.Totems[i].Destroy:EnableMouse()
        self.Totems[i].Destroy:SetScript("OnEnter", function(self)
            GameTooltip:SetOwner(self, "ANCHOR_BOTTOMRIGHT")
            local totem = self:GetParent()
            GameTooltip:SetTotem(totem:GetID())
            GameTooltip:AddLine("|cffff0000"..GLYPH_SLOT_REMOVE_TOOLTIP.."|r") -- <Shift Right Click to Remove>
            GameTooltip:Show()
        end)
        self.Totems[i].Destroy:SetScript("OnLeave", function()
            GameTooltip:Hide()
        end)
    end
    
    self.Totems.PostUpdate = ns.PostUpdateTotems
end
ns.AddTotems = AddTotems

and the postupdate function:

local PostUpdateTotems = function(Totems, slot, haveTotem, name, start, duration, icon)
    local delay = 0.5
    local total = 0
    
    if (duration > 0) then
        local current = GetTime() - start
        if (current > 0) then
            Totems[slot]:SetValue(1 - (current / duration))
            Totems[slot]:SetScript("OnUpdate", function(self, elapsed)
                total = total + elapsed
                if (total >= delay) then
                    total = 0
                    self:SetValue(1 - ((GetTime() - start) / duration))
                end
            end)
        end
    end
end
ns.PostUpdateTotems = PostUpdateTotems

Probably using Override instead of PostUpdate makes more sense, I'll take a look at it later.