CustomKeys: unit_khaos.lua

--------------------------------------------------------------------------------
--------------------------------------------------------------------------------

include("keysym.h.lua")


function widget:GetInfo()
  return {
    name      = "Khaos group",
    desc      = "description",
    author    = "author",
    date      = "31.1.2008",
    license   = "GNU GPL, v2 or later",
    layer     = 0,
    enabled   = false  --  loaded by default?
  }
end

--------------------------------------------------------------------------------
--------------------------------------------------------------------------------

local myTeam 

function widget:Initialize() 
        local _, _, spec, team = Spring.GetPlayerInfo(Spring.GetMyPlayerID())
        if spec then -- you are spectator, disable this widget
                widgetHandler:RemoveWidget()
                return false
        end
  myTeam = team
end




function widget:KeyPress(key, modifier, isRepeat)
  local filter = nil
  if (modifier.ctrl) then
                if (key == KEYSYMS.J) then 
      filter = {armjanus = 1, armstump = 1}
    end
    if (key == KEYSYMS.N_1) then
      filter = {armflash = 1}
    end
  end
  
  if (filter ~= nil) then 
    local toSelect = {}
    for _, id in pairs(Spring.GetVisibleUnits(myTeam)) do 
      if (filter[UnitDefs[Spring.GetUnitDefID(id)].name] ~= nil) then
        table.insert(toSelect, id)
      end
    end
    Spring.SelectUnitArray(toSelect)
  end
end


--------------------------------------------------------------------------------