diff --git a/OpenRA.Mods.RA/Widgets/Logic/CheatsLogic.cs b/OpenRA.Mods.RA/Widgets/Logic/CheatsLogic.cs index b5d4c3b299..68f1d6bea1 100644 --- a/OpenRA.Mods.RA/Widgets/Logic/CheatsLogic.cs +++ b/OpenRA.Mods.RA/Widgets/Logic/CheatsLogic.cs @@ -28,101 +28,126 @@ namespace OpenRA.Mods.RA.Widgets.Logic { var devTrait = world.LocalPlayer.PlayerActor.Trait(); - var shroudCheckbox = widget.Get("DISABLE_SHROUD"); - shroudCheckbox.IsChecked = () => devTrait.DisableShroud; - shroudCheckbox.OnClick = () => Order(world, "DevShroudDisable"); + var shroudCheckbox = widget.GetOrNull("DISABLE_SHROUD"); + if (shroudCheckbox != null) + { + shroudCheckbox.IsChecked = () => devTrait.DisableShroud; + shroudCheckbox.OnClick = () => Order(world, "DevShroudDisable"); + } - var pathCheckbox = widget.Get("SHOW_UNIT_PATHS"); - pathCheckbox.IsChecked = () => devTrait.PathDebug; - pathCheckbox.OnClick = () => Order(world, "DevPathDebug"); + var pathCheckbox = widget.GetOrNull("SHOW_UNIT_PATHS"); + if (pathCheckbox != null) + { + pathCheckbox.IsChecked = () => devTrait.PathDebug; + pathCheckbox.OnClick = () => Order(world, "DevPathDebug"); + } - widget.Get("GIVE_CASH").OnClick = () => + var cashButton = widget.GetOrNull("GIVE_CASH"); + if (cashButton != null) + cashButton.OnClick = () => world.IssueOrder(new Order("DevGiveCash", world.LocalPlayer.PlayerActor, false)); - var fastBuildCheckbox = widget.Get("INSTANT_BUILD"); - fastBuildCheckbox.IsChecked = () => devTrait.FastBuild; - fastBuildCheckbox.OnClick = () => Order(world, "DevFastBuild"); + var fastBuildCheckbox = widget.GetOrNull("INSTANT_BUILD"); + if (fastBuildCheckbox != null) + { + fastBuildCheckbox.IsChecked = () => devTrait.FastBuild; + fastBuildCheckbox.OnClick = () => Order(world, "DevFastBuild"); + } - var fastChargeCheckbox = widget.Get("INSTANT_CHARGE"); - fastChargeCheckbox.IsChecked = () => devTrait.FastCharge; - fastChargeCheckbox.OnClick = () => Order(world, "DevFastCharge"); + var fastChargeCheckbox = widget.GetOrNull("INSTANT_CHARGE"); + if (fastChargeCheckbox != null) + { + fastChargeCheckbox.IsChecked = () => devTrait.FastCharge; + fastChargeCheckbox.OnClick = () => Order(world, "DevFastCharge"); + } - var showMuzzlesCheckbox = widget.Get("SHOW_MUZZLES"); - showMuzzlesCheckbox.IsChecked = () => devTrait.ShowMuzzles; - showMuzzlesCheckbox.OnClick = () => devTrait.ShowMuzzles ^= true; + var showMuzzlesCheckbox = widget.GetOrNull("SHOW_MUZZLES"); + if (showMuzzlesCheckbox != null) + { + showMuzzlesCheckbox.IsChecked = () => devTrait.ShowMuzzles; + showMuzzlesCheckbox.OnClick = () => devTrait.ShowMuzzles ^= true; + } - var allTechCheckbox = widget.Get("ENABLE_TECH"); - allTechCheckbox.IsChecked = () => devTrait.AllTech; - allTechCheckbox.OnClick = () => Order(world, "DevEnableTech"); + var allTechCheckbox = widget.GetOrNull("ENABLE_TECH"); + if (allTechCheckbox != null) + { + allTechCheckbox.IsChecked = () => devTrait.AllTech; + allTechCheckbox.OnClick = () => Order(world, "DevEnableTech"); + } - var powerCheckbox = widget.Get("UNLIMITED_POWER"); - powerCheckbox.IsChecked = () => devTrait.UnlimitedPower; - powerCheckbox.OnClick = () => Order(world, "DevUnlimitedPower"); + var powerCheckbox = widget.GetOrNull("UNLIMITED_POWER"); + if (powerCheckbox != null) + { + powerCheckbox.IsChecked = () => devTrait.UnlimitedPower; + powerCheckbox.OnClick = () => Order(world, "DevUnlimitedPower"); + } - var buildAnywhereCheckbox = widget.Get("BUILD_ANYWHERE"); - buildAnywhereCheckbox.IsChecked = () => devTrait.BuildAnywhere; - buildAnywhereCheckbox.OnClick = () => Order(world, "DevBuildAnywhere"); + var buildAnywhereCheckbox = widget.GetOrNull("BUILD_ANYWHERE"); + if (buildAnywhereCheckbox != null) + { + buildAnywhereCheckbox.IsChecked = () => devTrait.BuildAnywhere; + buildAnywhereCheckbox.OnClick = () => Order(world, "DevBuildAnywhere"); + } - widget.Get("GIVE_EXPLORATION").OnClick = () => + var explorationButton = widget.GetOrNull("GIVE_EXPLORATION"); + if (explorationButton != null) + explorationButton.OnClick = () => world.IssueOrder(new Order("DevGiveExploration", world.LocalPlayer.PlayerActor, false)); - widget.Get("RESET_EXPLORATION").OnClick = () => + var noexplorationButton = widget.GetOrNull("RESET_EXPLORATION"); + if (noexplorationButton != null) + noexplorationButton.OnClick = () => world.IssueOrder(new Order("DevResetExploration", world.LocalPlayer.PlayerActor, false)); var dbgOverlay = world.WorldActor.TraitOrDefault(); - var showAstarCostCheckbox = widget.Get("SHOW_ASTAR"); - showAstarCostCheckbox.IsChecked = () => dbgOverlay != null ? dbgOverlay.Visible : false; - showAstarCostCheckbox.OnClick = () => { if (dbgOverlay != null) dbgOverlay.Visible ^= true; }; - - - var desync = widget.Get("DESYNC"); - var desyncEnabled = widget.Get("DESYNC_ARMED"); - desyncEnabled.IsChecked = () => !desync.Disabled; - desyncEnabled.OnClick = () => desync.Disabled ^= true; - - desync.Disabled = true; - desync.OnClick = () => + var showAstarCostCheckbox = widget.GetOrNull("SHOW_ASTAR"); + if (showAstarCostCheckbox != null) { - var trait = world.ActorsWithTrait().Random(CosmeticRandom).Trait; - var t = trait.GetType(); - string s; - const BindingFlags bf = BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance; + showAstarCostCheckbox.IsChecked = () => dbgOverlay != null ? dbgOverlay.Visible : false; + showAstarCostCheckbox.OnClick = () => { if (dbgOverlay != null) dbgOverlay.Visible ^= true; }; + } + var desync = widget.GetOrNull("DESYNC"); + var desyncEnabled = widget.GetOrNull("DESYNC_ARMED"); + if (desync != null && desyncEnabled != null) + { + desyncEnabled.IsChecked = () => !desync.Disabled; + desyncEnabled.OnClick = () => desync.Disabled ^= true; - var fields = t.GetFields(bf).Where(x => x.HasAttribute()).ToArray(); - if (fields.Length > 0) - { + desync.Disabled = true; + desync.OnClick = () => TriggerDesync(world); + } - var f = fields[CosmeticRandom.Next(fields.Length)]; - var before = f.GetValue(trait); + var close = widget.GetOrNull("CLOSE"); + if (close != null) + close.OnClick = () => { Ui.CloseWindow(); onExit(); }; + } - if (f.FieldType == typeof(Boolean)) - f.SetValue(trait, !(Boolean) f.GetValue(trait)); - else if (f.FieldType == typeof(Int32)) - f.SetValue(trait, CosmeticRandom.Next(Int32.MaxValue)); - else - { - s = "Sorry, Field-Type not implemented. Try again!"; - Game.AddChatLine(Color.White, "Debug", s); - Console.WriteLine(s); - } + void TriggerDesync(World world) + { + var trait = world.ActorsWithTrait().Random(CosmeticRandom).Trait; + var t = trait.GetType(); + const BindingFlags bf = BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance; + var fields = t.GetFields(bf).Where(x => x.HasAttribute()).ToArray(); + if (fields.Length > 0) + { + var f = fields[CosmeticRandom.Next(fields.Length)]; + var before = f.GetValue(trait); - var after = f.GetValue(trait); - - s = "Type: {0}\nField: ({1}) {2}\nBefore: {3}\nAfter: {4}".F(t.Name, f.FieldType.Name, f.Name, before, after); - Game.AddChatLine(Color.White, "Debug", s); - Console.WriteLine(s); - } + if (f.FieldType == typeof(Boolean)) + f.SetValue(trait, !(Boolean) f.GetValue(trait)); + else if (f.FieldType == typeof(Int32)) + f.SetValue(trait, CosmeticRandom.Next(Int32.MaxValue)); else - { - s = "Bad random choice. This trait has no fields. Try again!"; - Game.AddChatLine(Color.White, "Debug", s); - Console.WriteLine(s); - } - }; + Game.AddChatLine(Color.White, "Debug", "Sorry, Field-Type not implemented. Try again!"); - widget.Get("CLOSE").OnClick = () => { Ui.CloseWindow(); onExit(); }; + var after = f.GetValue(trait); + Game.AddChatLine(Color.White, "Debug", "Type: {0}\nField: ({1}) {2}\nBefore: {3}\nAfter: {4}" + .F(t.Name, f.FieldType.Name, f.Name, before, after)); + } + else + Game.AddChatLine(Color.White, "Debug", "Bad random choice. This trait has no fields. Try again!"); } public void Order(World world, string order)