Sync.UnitData[self:GetEntityId()] = false

local unitEnh = SimUnitEnhancements[entId]

                  --Holding alt builds the highest tech available mex instead of tech 1
                  if IsKeyDown('Menu') and depositCat == 'MASSEXTRACTION' then
                     for i=3, 1, -1 do
                        local buildableUnits = EntityCategoryGetUnitList(buildableCategories * ParseEntityCategory(depositCat) * ParseEntityCategory('TECH'..i))
                        if buildableUnits[1] then
                           buildBP = buildableUnits[1]
                           break
                        end
                     end
                  else
                     local buildableUnits = EntityCategoryGetUnitList(buildableCategories * ParseEntityCategory(depositCat) * categories.TECH1)
                     buildBP = buildableUnits[1]
                  end

local GetScore = import('/mods/TeamUtils/modules/mciscore.lua').GetScore

local lastFocusedArmy = 0
local emergencyResetProcedureStarted = false
local UpdateFlag = false
local UpdateCounter = 0
local allUnits = {}
local Delay = 3
local resetcount=0
local resetflag=0
###############################################################################

function UpdateControl(flag)
   #if flag == 1 then Reset() end
   UpdateCounter = UpdateCounter + flag
   if UpdateFlag and  UpdateCounter <= 0 then
      #print("Updater: Disabled")
      import('/lua/ui/game/gamemain.lua').RemoveBeatFunction(UpdateAllUnits)
      UpdateFlag = false
   elseif UpdateFlag == false and UpdateCounter > 0 then
      #print("Updater: Enabled")
      import('/lua/ui/game/gamemain.lua').AddBeatFunction(UpdateAllUnits)
      Reset()
      UpdateFlag = true
   end
end
###############################################################################
function AddSelection()
   for _, unit in (GetSelectedUnits() or {}) do
      allUnits[unit:GetEntityId()] = unit
   end
end

function Reset()
   import('/lua/ui/game/construction.lua').SaveCheckedTab()
   local currentlySelected = GetSelectedUnits() or {}
   local commandmode = import('/lua/ui/game/commandmode.lua')
   local currentCommand = commandmode.GetCommandMode()

   UISelectionByCategory("ALLUNITS", false, false, false, false)
   AddSelection()

   SelectUnits(currentlySelected)
   import('/lua/ui/game/construction.lua').RestoreCheckedTab()
   commandmode.StartCommandMode(currentCommand[1], currentCommand[2])

end

function Resetter(count)
   resetcount=count
   if resetcount>0 and resetflag==0 then
      ForkThread(ResetCounter)
      resetflag=1
   end
end

function ResetCounter()
   while (resetcount>0) do
      Reset()
      #print("Reset " .. resetcount)
      resetcount=resetcount-1
      WaitSeconds(4)
   end
   resetflag=0
end

function EmergencyReset(targetUnitCount)
   WaitSeconds(Delay)
   local newTargetUnitCount = GetScore()[GetFocusArmy()].general.currentunits.count
   -- if after 10 seconds we are not loosing units and the unit count is still larger we have to do a reset
   if newTargetUnitCount >= targetUnitCount and newTargetUnitCount > table.getsize(allUnits)  then
      #print("New units detected")
      Reset()
   end
   emergencyResetProcedureStarted = false
end

function UpdateAllUnits()
   if GetFocusArmy() != lastFocusedArmy then
      Reset()
      lastFocusedArmy = GetFocusArmy()
   end

   AddSelection()
   
   -- add focused (building or assisting)
   for _, unit in allUnits do
      if not unit:IsDead() and unit:GetFocus() and not unit:GetFocus():IsDead()then
         allUnits[unit:GetFocus():GetEntityId()] = unit:GetFocus()
      end
   end
   
   -- remove dead
   for entityid, unit in allUnits do
      if unit:IsDead() or unit:GetArmy() != GetFocusArmy() then
         allUnits[entityid] = nil
         continue
      end
   end
   
   -- emergency reset
   local targetNumOfUnits = GetScore()[GetFocusArmy()].general.currentunits.count
   if not emergencyResetProcedureStarted and targetNumOfUnits > table.getsize(allUnits) then
      emergencyResetProcedureStarted = true
      ForkThread(EmergencyReset, targetNumOfUnits)
   end
end


function GetAllUnits()
   return allUnits
end

function GetACU()
   local ACU = {}
   for entityid, unit in allUnits do
      if unit:IsInCategory("COMMAND") then
         ACU[entityid] = unit
      end
   end
   return ACU
end

warning: Error running HandleEvent script in CScriptObject at 317c5900: ...iance\gamedata\lua.scd\lua\ui\controls\worldview.lua(653): attempt to call method `GetArmy' (a nil value)
         stack traceback:
            ...iance\gamedata\lua.scd\lua\ui\controls\worldview.lua(653): in function <...iance\gamedata\lua.scd\lua\ui\controls\worldview.lua:616>