Improved functionality of favorite constructions

Warzone2100 · Jun 14, 2024 · 30ae267 · 30ae267
1 parent 221be39
commit 30ae267
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/hci/build.cpp b/src/hci/build.cpp
@@ -120,7 +120,15 @@ void BuildController::startBuildPosition(STRUCTURE_STATS *buildOption)
 
 void BuildController::toggleFavorites(STRUCTURE_STATS *buildOption)
 {
-	asStructureStats[buildOption->index].isFavorite = !shouldShowFavorites();
+	bool &isFavorite = asStructureStats[buildOption->index].isFavorite;
+	if (shouldShowFavorites())
+	{
+		isFavorite = false;
+	}
+	else
+	{
+		isFavorite = !isFavorite;
+	}
 	updateBuildOptionsList();
 }
 
@@ -449,6 +457,10 @@ class BuildOptionButton: public StatsFormButton
 
 		displayIMD(AtlasImage(), ImdObject::StructureStat(stat), xOffset, yOffset);
 		displayIfHighlight(xOffset, yOffset);
+		if (asStructureStats[stat->index].isFavorite && !controller->shouldShowFavorites())
+		{
+			iV_DrawImage(IntImages, IMAGE_ALLY_RESEARCH_TC, xOffset + x() + 44, yOffset + y() + 3);
+		}
 	}
 
 private: