Added initialRepairDroidGroup

Warzone2100 · May 3, 2024 · 2a0e824 · 2a0e824
1 parent d631ed3
commit 2a0e824
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/cmddroid.cpp b/src/cmddroid.cpp
@@ -90,6 +90,8 @@ bool cmdDroidAddDroid(DROID *psCommander, DROID *psDroid)
 	ASSERT_OR_RETURN(false, psDroid != nullptr, "psDroid is null?");
 
 	auto initialDroidGroup = psDroid->group;
+	auto initialRepairDroidGroup = psDroid->repairGroup;
+	// select a value depending on whether the commander was assigned manually or after construction in the factory
 	auto fallbackDroidGroup = initialDroidGroup != UBYTE_MAX ? initialDroidGroup : psDroid->fallbackGroup;
 
 	if (psCommander->psGroup == nullptr)
@@ -131,6 +133,7 @@ bool cmdDroidAddDroid(DROID *psCommander, DROID *psDroid)
 			psDroid->fallbackGroup = UBYTE_MAX;
 			SelectGroupDroid(psDroid);
 		}
+		psDroid->repairGroup = initialRepairDroidGroup;
 	}
 
 	if (initialDroidGroup != psDroid->group)