tick window only for energy

rgd87 · Sep 10, 2019 · aacb1f1 · aacb1f1
1 parent f107d89
commit aacb1f1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/NugEnergy.lua b/NugEnergy.lua
@@ -283,10 +283,12 @@ function NugEnergy.Initialize(self)
         self.initialized = true
     end
 
+    twEnabled = false
 
     if class == "ROGUE" and NugEnergyDB.energy then
         PowerFilter = "ENERGY"
         PowerTypeIndex = Enum.PowerType.Energy
+        twEnabled = NugEnergyDB.twEnabled
         shouldBeFull = true
         self:RegisterEvent("UPDATE_STEALTH")
         self:SetScript("OnUpdate",self.UpdateEnergy)
@@ -327,12 +329,14 @@ function NugEnergy.Initialize(self)
         self.UNIT_DISPLAYPOWER = function(self)
             local newPowerType = select(2,UnitPowerType("player"))
             shouldBeFull = false
+            twEnabled = false
 
             -- restore to original MAXPOWER in case it was switched for classic energy
             NugEnergy.UNIT_MAXPOWER = NugEnergy.NORMAL_UNIT_MAXPOWER
             if newPowerType == "ENERGY" and NugEnergyDB.energy then
                 PowerFilter = "ENERGY"
                 PowerTypeIndex = Enum.PowerType.Energy
+                twEnabled = NugEnergyDB.twEnabled
                 shouldBeFull = true
                 self:RegisterEvent("UNIT_POWER_UPDATE")
                 self:RegisterEvent("UNIT_MAXPOWER")