Unit Groups and AI

I tried this, and it doesn't do what I want. I need to be able to check the design_id, or the general_type. This returns the unit type, as defined in VehicleSizes.txt. Unfortunatly, this means that it sees all satellites (for example, Recon Satellites) as exactly the same thing, Satellite.

Thanks for the suggestion, however.

Hrm, I'll give this a shot. I don't think it counts member units as cargo, but maybe it does.

That did it, thanks Kwok.

If anyone is interested, this bit of code will tell the AI to count objects by general type.

Count_Units_In_Cargo

Desired_cargo is the general_type of the desired object, as set in Script_AI_DesignCreation.txt and defined in Script_AI_GlobalConstants_DesignTypes.txt.
Object_id is the object id of the object holding the cargo.

//------------------------------------------------------------------------
// Count_Units_In_Cargo
//------------------------------------------------------------------------
function Count_Units_In_Cargo returns long
params
desired_cargo: string
object_id: long
vars
cargo_count: long
cargo_index: long
cargo_item_id: long
cargo_design_id: long
cargo_general_type: string
total_count: long
begin
set total_count := 0

// --- Count the ammount of cargo on the object.
set cargo_count := Sys_Get_Space_Object_Cargo_Item_Count(object_id)
if (cargo_count > 0) then
for cargo_index := 1 to cargo_count do

// --- Check each object against the desired type, and count the number present
set cargo_item_id := Sys_Get_Space_Object_Cargo_Item_ID(object_id, cargo_index)
set cargo_design_id := Sys_Get_Space_Object_Cargo_Item_Design_ID(object_id, cargo_item_id)
set cargo_general_type := Sys_Get_Vehicle_Design_General_Type(sys_long_Player_ID, cargo_design_id)

if (cargo_general_type = desired_cargo) then
set total_count := total_count + 1
endif

endfor
endif
return total_count
end

Returns total count in the target cargo.
----------------------------------------------------------------------------------------------------------------------------------------------------------------
Count_Units_In_Space_Sector counts the units in a space sector by general type.

Target_design_type is the general_type of the desired object, as set in Script_AI_DesignCreation.txt and defined in Script_AI_GlobalConstants_DesignTypes.txt.
Target_system is the system you want to check.
Target_sector is the sector in the above system you want to check.

If it encounters a unitgroup, it calls Count_Units_In_Cargo, with the id of the unit group and the general type of the desired object.

Returns total count in the target sector.

//------------------------------------------------------------------------
// Count_Units_In_Space_Sector
//------------------------------------------------------------------------
function Count_Units_In_Space_Sector returns long
params
target_design_type: string
target_system: long
target_sector: long
vars
object_list: longlist
object_count: long
object_index: long
object_id: long
object_sector: long
object_owner: long
object_design_id: long
object_general_type: string
total_count: long
begin
set total_count := 0

// --- List the objects in the system.
call Sys_Get_All_Visible_Space_Objects_In_System(sys_long_Player_ID, target_system, object_list)

// --- Fill up our object stats.
set object_count := object_list.count()
if (object_count > 0) then
for object_index := 1 to object_count do
set object_id := object_list.get(object_index)
set object_sector := Sys_Get_Space_Object_Sector_Location(object_id)
set object_owner := Sys_Get_Space_Object_Owner(object_id)
set object_design_id := Sys_Get_Space_Object_Vehicle_Design_ID(object_id)
set object_general_type := Sys_Get_Vehicle_Design_General_Type(sys_long_Player_ID, object_design_id)

// --- Check if it is in our desired sector, if we own it, and if it's the kind we want. If so, increment count.
if (object_sector = target_sector) and (object_owner = sys_long_Player_ID) and (object_general_type = target_design_type) then
set total_count := total_count + 1
endif

// --- If its a unit group, count all the units as cargo.
if (Sys_Get_Space_Object_Type(object_id) = SPACE_OBJECT_TYPE_UNITGROUP) then
set total_count := total_count + Count_Units_In_Cargo(target_design_type, object_id)
endif
endfor
endif

return total_count
end

Anyone is free to use any part of this code in their mods.

Cool, I'm using this as part of a separate build queue for units, so the AI builds them more often and efficiently.