[GEN][ZH] Add support for selecting structures in teams in World Builder (#1316)

Author: DayV-git

Generals/Code/Tools/WorldBuilder/src/TeamIdentity.cpp

@@ -253,7 +253,10 @@ void TeamIdentity::loadUnitsInfo(int idcMinUnit, NameKeyType keyMinUnit,
 
 		// next tier uses the editor sorting bits that design can specify in the INI
 		EditorSortingType sort = tTemplate->getEditorSorting();
-		if (( sort != ES_VEHICLE ) && (sort != ES_INFANTRY)) continue;
+		// TheSuperHackers @tweak DayV 20/07/2025 Allow structures in teams.
+		if (sort != ES_VEHICLE && sort != ES_INFANTRY && sort != ES_STRUCTURE) {
+			continue;
+		}
 
 		Int ndx = pCombo->AddString(tTemplate->getName().str());
 		if (type == tTemplate->getName()) {
@@ -437,6 +440,8 @@ void TeamIdentity::OnUnitTypeButton(Int idcUnitType)
 	PickUnitDialog dlg;
 	dlg.SetAllowableType(ES_VEHICLE);
 	dlg.SetAllowableType(ES_INFANTRY);
+	// TheSuperHackers @tweak DayV 20/07/2025 Allow structures in teams.
+	dlg.SetAllowableType(ES_STRUCTURE);
 	if (dlg.DoModal() == IDOK) {
 		AsciiString unit = dlg.getPickedUnit();
 		NameKeyType keyUnitType;

GeneralsMD/Code/Tools/WorldBuilder/src/TeamIdentity.cpp

@@ -253,7 +253,10 @@ void TeamIdentity::loadUnitsInfo(int idcMinUnit, NameKeyType keyMinUnit,
 
 		// next tier uses the editor sorting bits that design can specify in the INI
 		EditorSortingType sort = tTemplate->getEditorSorting();
-		if (( sort != ES_VEHICLE ) && (sort != ES_INFANTRY)) continue;
+		// TheSuperHackers @tweak DayV 20/07/2025 Allow structures in teams.
+		if (sort != ES_VEHICLE && sort != ES_INFANTRY && sort != ES_STRUCTURE) {
+			continue;
+		}
 
 		Int ndx = pCombo->AddString(tTemplate->getName().str());
 		if (type == tTemplate->getName()) {
@@ -437,6 +440,8 @@ void TeamIdentity::OnUnitTypeButton(Int idcUnitType)
 	PickUnitDialog dlg;
 	dlg.SetAllowableType(ES_VEHICLE);
 	dlg.SetAllowableType(ES_INFANTRY);
+	// TheSuperHackers @tweak DayV 20/07/2025 Allow structures in teams.
+	dlg.SetAllowableType(ES_STRUCTURE);  
 	if (dlg.DoModal() == IDOK) {
 		AsciiString unit = dlg.getPickedUnit();
 		NameKeyType keyUnitType;