Toggle repair/sell button state when active.

Mailaender · Jul 15, 2011 · bc1a506 · bc1a506
1 parent 1c29c95
commit bc1a506
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/OpenRA.Mods.Cnc/Widgets/Logic/CncIngameChromeLogic.cs b/OpenRA.Mods.Cnc/Widgets/Logic/CncIngameChromeLogic.cs
@@ -107,14 +107,16 @@ public CncIngameChromeLogic([ObjectCreator.Param] Widget widget,
 
 				var sidebarRoot = playerWidgets.GetWidget("SIDEBAR_BACKGROUND");
 
-				var sellButton = sidebarRoot.GetWidget<ButtonWidget>("SELL_BUTTON");
+				var sellButton = sidebarRoot.GetWidget<ToggleButtonWidget>("SELL_BUTTON");
 				sellButton.OnClick = () => world.ToggleInputMode<SellOrderGenerator>();
+				sellButton.IsToggled = () => world.OrderGenerator is SellOrderGenerator;
 				var sellIcon = sellButton.GetWidget<ImageWidget>("ICON");
 				sellIcon.GetImageName = () => world.OrderGenerator is SellOrderGenerator ? "sell-active" : "sell";
 
-				var repairButton = sidebarRoot.GetWidget<ButtonWidget>("REPAIR_BUTTON");
+				var repairButton = sidebarRoot.GetWidget<ToggleButtonWidget>("REPAIR_BUTTON");
 				repairButton.IsDisabled = () => !RepairOrderGenerator.PlayerIsAllowedToRepair( world );
 				repairButton.OnClick = () => world.ToggleInputMode<RepairOrderGenerator>();
+				repairButton.IsToggled = () => world.OrderGenerator is RepairOrderGenerator;
 				var repairIcon = repairButton.GetWidget<ImageWidget>("ICON");
 				repairIcon.GetImageName = () => repairButton.IsDisabled() ? "repair-disabled" :
 					world.OrderGenerator is RepairOrderGenerator ? "repair-active" : "repair";