-- Karos Graveyard
--   Homeworld 2 Scripting Wiki
--
-- Created by Thought (http://hw2.tproc.org)

-- Translated: ai/classdef.lua

aitrace("CPU: CLASSDEF LOADED") 
-- Do NOT CHANGE THIS 

squadclass = {} 
--Do NOT CHANGE THIS 

-- This next section is what ships belong to each class add ships in as they are needed
squadclass[eMotherShip] = {HGN_MOTHERSHIP, VGR_MOTHERSHIP, VGR_MOTHERSHIP_MAKAAN}
squadclass[eCollector] = {HGN_RESOURCECOLLECTOR, VGR_RESOURCECOLLECTOR}
squadclass[eScout] = {HGN_SCOUT, HGN_SCOUT2, HGN_PROBE, HGN_PROXIMITYSENSOR, HGN_ECMPROBE, VGR_SCOUT, VGR_PROBE, VGR_PROBE_PROX, VGR_PROBE_ECM}
squadclass[eRefinery] = {HGN_RESOURCECONTROLLER, VGR_RESOURCECONTROLLER}
squadclass[eBuilder] = {HGN_MOTHERSHIP, HGN_CARRIER, HGN_ESCORTCARRIER, HGN_SHIPYARD, VGR_MOTHERSHIP, VGR_MOTHERSHIP_MAKAAN, VGR_CARRIER, VGR_SHIPYARD}
squadclass[eDropOff] = {HGN_MOTHERSHIP, HGN_CARRIER, HGN_ESCORTCARRIER, HGN_SHIPYARD, HGN_RESOURCECONTROLLER, VGR_MOTHERSHIP, VGR_MOTHERSHIP_MAKAAN, VGR_CARRIER, VGR_SHIPYARD, VGR_RESOURCECONTROLLER}
squadclass[eSalvageDropOff] = {HGN_MOTHERSHIP, HGN_CARRIER, HGN_ESCORTCARRIER, HGN_SHIPYARD, VGR_MOTHERSHIP, VGR_MOTHERSHIP_MAKAAN, VGR_CARRIER, VGR_SHIPYARD}
squadclass[eFighter] = {HGN_INTERCEPTOR, HGN_INTERCEPTORADV, HGN_ADVANCEDFIGHTERTECH, HGN_ATTACKBOMBER, HGN_ATTACKBOMBER2, HGN_ATTACKBOMBERADV, HGN_ATTACKBOMBERELITE, VGR_INTERCEPTOR, VGR_BOMBER, VGR_LANCEFIGHTER}
squadclass[eCorvette] = {HGN_ASSAULTCORVETTE, HGN_PULSARCORVETTE, HGN_MINELAYERCORVETTE, VGR_MISSILECORVETTE, VGR_LASERCORVETTE, VGR_MINELAYERCORVETTE, VGR_COMMANDCORVETTE}
squadclass[eFrigate] = {HGN_ASSAULTFRIGATE, HGN_DEFENSEFIELDFRIGATE, HGN_IONCANNONFRIGATE, HGN_MARINEFRIGATE, HGN_MARINEFRIGATE_SOBAN, HGN_TORPEDOFRIGATE, HGN_TORPEDOFRIGATEELITE, VGR_ASSAULTFRIGATE, VGR_HEAVYMISSILEFRIGATE, VGR_INFILTRATORFRIGATE}
squadclass[eCapture] = {HGN_MARINEFRIGATE, HGN_MARINEFRIGATE_SOBAN, VGR_INFILTRATORFRIGATE}
squadclass[eShield] = {HGN_DEFENSEFIELDFRIGATE}
squadclass[ePlatform] = {HGN_GUNTURRET, HGN_IONTURRET, VGR_WEAPONPLATFORM_GUN, VGR_WEAPONPLATFORM_MISSILE, VGR_HYPERSPACE_PLATFORM}
squadclass[eAntiFighter] = {HGN_INTERCEPTOR, HGN_INTERCEPTORADV, HGN_ASSAULTCORVETTE, HGN_ASSAULTFRIGATE, HGN_GUNTURRET, VGR_INTERCEPTOR, VGR_ASSAULTFRIGATE, VGR_WEAPONPLATFORM_GUN}
squadclass[eAntiCorvette] = {HGN_PULSARCORVETTE, HGN_INTERCEPTORADV, HGN_TORPEDOFRIGATE, HGN_TORPEDOFRIGATEELITE, HGN_DESTROYER, VGR_LANCEFIGHTER, VGR_LAZERCORVETTE, VGR_DESTROYER}
squadclass[eAntiFrigate] = {HGN_IONTURRET, HGN_ATTACKBOMBER, HGN_ATTACKBOMBERELITE, HGN_IONCANNONFRIGATE, HGN_MARINEFRIGATE, HGN_MARINEFRIGATE_SOBAN, HGN_DEFENSEFIELDFRIGATE, HGN_DESTROYER, HGN_BATTLECRUISER, VGR_BOMBER, VGR_HEAVYMISSILEFRIGATE, VGR_INFILTRATORFRIGATE, VGR_DESTROYER, VGR_BATTLECRUISER, VGR_WEAPONPLATFORM_MISSILE}
squadclass[eCapital] = {HGN_CARRIER, HGN_ESCORTCARRIER, HGN_MOTHERSHIP, HGN_SHIPYARD, HGN_DESTROYER, HGN_BATTLECRUISER, HGN_DREADNAUGHT, VGR_CARRIER, VGR_MOTHERSHIP, VGR_MOTHERSHIP_MAKAAN, VGR_SHIPYARD, VGR_DESTROYER, VGR_BATTLECRUISER}
squadclass[eNonThreat] = {HGN_RESOURCECOLLECTOR, VGR_RESOURCECOLLECTOR, HGN_RESOURCECONTROLLER, VGR_RESOURCECONTROLLER, HGN_CARRIER, HGN_MOTHERSHIP, HGN_SHIPYARD, VGR_CARRIER, VGR_MOTHERSHIP, VGR_MOTHERSHIP_MAKAAN, VGR_SHIPYARD, HGN_SCOUT, HGN_PROBE, HGN_PROXIMITYSENSOR, HGN_ECMPROBE, VGR_SCOUT, VGR_PROBE, VGR_PROBE_PROX, VGR_PROBE_ECM, VGR_HYPERSPACE_PLATFORM}
squadclass[eHyperspaceGate] = {VGR_HYPERSPACE_PLATFORM}
squadclass[eSubSystemAttackers] = {HGN_ATTACKBOMBER, HGN_ATTACKBOMBER, HGN_ATTACKBOMBER2, HGN_ATTACKBOMBERADV, HGN_ATTACKBOMBERELITE, VGR_BOMBER}
squadclass[eNonCriticalSubSys] = {CLOAKGENERATOR, FIRECONTROLTOWER, HYPERSPACEINHIBITOR, ADVANCEDARRAY, CLOAKSENSOR}
squadclass[eGoodRepairAttackers] = {HGN_INTERCEPTOR, HGN_ASSAULTFRIGATE, HGN_IONCANNONFRIGATE, HGN_DESTROYER, HGN_BATTLECRUISER, VGR_INTERCEPTOR, VGR_MISSILECORVETTE, VGR_ASSAULTFRIGATE, VGR_DESTROYER, VGR_BATTLECRUISER}

-- Play if you want at your own risk with the next 3 lines
eUselessShips = eMaxCount
eBattleCruiser = eMaxCount + 1
sg_maxClasses = eBattleCruiser + 1
-- Think this next line are those ships which are considered useless for attacks etc. I've noticed the CPU rarely builds minelayers etc.
squadclass[eUselessShips] = {HGN_MINELAYERCORVETTE, VGR_MINELAYERCORVETTE, VGR_COMMANDCORVETTE}
-- Fairly Obvious, which vessels are counted as Battlecruisers
squadclass[eBattleCruiser] = {HGN_BATTLECRUISER, VGR_BATTLECRUISER}

-- Mess with these lines at your own risk
function FastAddToClass(tbl, classid)
  for a, b in tbl do
    AddToClass(b, classid)
  end
end

function ClassInitialize()
  for i = 0, sg_maxClasses do
    if squadclass[i] then
      FastAddToClass(squadclass[i], i)
    end
  end

-- Now lets those etags there real names....
 AddClassName(eMotherShip, "Motherships")
  AddClassName(eCollector, "Collectors")
  AddClassName(eDropOff, "DropOffs")
  AddClassName(eFighter, "Fighters")
  AddClassName(eFrigate, "Frigates")
  AddClassName(eCorvette, "Corvettes")
  AddClassName(eCapital, "Capital")
  AddClassName(eAntiFighter, "AntiFighter")
  AddClassName(eAntiCorvette, "AntiCorvette")
  AddClassName(eAntiFrigate, "AntiFrigate")
  AddClassName(ePlatform, "Platform")
  AddClassName(eRefinery, "Refinery")
  AddClassName(eHyperspaceGate, "HypGates")
  AddClassName(eShield, "Shields")
  AddClassName(eCapture, "Capture")
  AddClassName(eSubSystemAttackers, "SubSysKillas")
  AddClassName(eBattleCruiser, "BattleCruiser")
end

    24	[109]	SETGLOBAL  	3	; cp_attackPercent
    25	[110]	GETGLOBAL  	6	; aitrace
    26	[110]	PUSHSTRING 	7	; "I'm Losing!! Go on defence"
    27	[110]	CALL       	0 0