diff --git a/Assembly.version b/Assembly.version index 881bd75..92ec459 100644 --- a/Assembly.version +++ b/Assembly.version @@ -1,7 +1,7 @@ - 1 - 2 - 7 + 2 + 0 + 0 \ No newline at end of file diff --git a/TweaksAndThings/Commands/CrewUpdateCommand.cs b/TweaksAndThings/Commands/CrewUpdateCommand.cs index b24bd0b..a949ac9 100644 --- a/TweaksAndThings/Commands/CrewUpdateCommand.cs +++ b/TweaksAndThings/Commands/CrewUpdateCommand.cs @@ -1,10 +1,14 @@ -using Game.State; +using Game.Notices; +using Game.State; using Helpers; using Model.Ops; using Network; using RMROC451.TweaksAndThings.Extensions; +using RMROC451.TweaksAndThings.Patches; using System.Linq; +using Track; using UI.Console; +using UnityEngine; namespace RMROC451.TweaksAndThings.Commands; @@ -37,8 +41,15 @@ public class EchoCommand : IConsoleCommand return "+ to include area or - to leave it out"; } - if (comps[2] == "+") message += $" {OpsController.Shared.ClosestArea(car)?.name ?? "???"}"; - Multiplayer.Broadcast($"{StateManager.Shared._playersManager.LocalPlayer} {Hyperlink.To(car)}: \"{message}\""); + var gamePoint = car.GetCenterPosition(Graph._graph); + EntityReference entityReference = new EntityReference(EntityType.Position, new Vector4( gamePoint.x, gamePoint.y, gamePoint.z, 0)); + EntityReference loco = new EntityReference(EntityType.Car, car.id); + if (comps[2] == "+") message = new Hyperlink(entityReference.URI(), string.Format(message, OpsController.Shared.ClosestArea(car)?.name ?? "???")); + + car.PostNotice(nameof(EchoCommand), $"{message} :{StateManager.Shared._playersManager.LocalPlayer}"); + ExpandedConsole_Add_Patch.SendMs(null, $"{Hyperlink.To(car)} {message}"); + Multiplayer.Broadcast($"{StateManager.Shared._playersManager.LocalPlayer} {Hyperlink.To(car)}: \"{message}\""); + return string.Empty; } } \ No newline at end of file diff --git a/TweaksAndThings/Enums/CrewHourLoadMethod.cs b/TweaksAndThings/Enums/CrewHourLoadMethod.cs new file mode 100644 index 0000000..51a7ff8 --- /dev/null +++ b/TweaksAndThings/Enums/CrewHourLoadMethod.cs @@ -0,0 +1,7 @@ +namespace RMROC451.TweaksAndThings.Enums; + +public enum CrewHourLoadMethod +{ + Tracks, + Daily +} diff --git a/TweaksAndThings/Enums/MrocHelperType.cs b/TweaksAndThings/Enums/MrocHelperType.cs index abd6b64..2843b14 100644 --- a/TweaksAndThings/Enums/MrocHelperType.cs +++ b/TweaksAndThings/Enums/MrocHelperType.cs @@ -4,5 +4,6 @@ public enum MrocHelperType { Handbrake, GladhandAndAnglecock, - BleedAirSystem + BleedAirSystem, + Oil } diff --git a/TweaksAndThings/Extensions/AutoEngineer_Extensions.cs b/TweaksAndThings/Extensions/AutoEngineer_Extensions.cs index edfab78..1fc94ad 100644 --- a/TweaksAndThings/Extensions/AutoEngineer_Extensions.cs +++ b/TweaksAndThings/Extensions/AutoEngineer_Extensions.cs @@ -1,6 +1,8 @@ using Model.AI; +using System; using System.Collections; using UnityEngine; +using Random = UnityEngine.Random; namespace RMROC451.TweaksAndThings.Extensions { @@ -15,7 +17,7 @@ namespace RMROC451.TweaksAndThings.Extensions public static IEnumerator MrocAutoOilerLoop(this AutoOiler oiler, Serilog.ILogger _log, bool cabooseRequired) { int originIndex = oiler.FindOriginIndex(); - Model.Car? foundCaboose = oiler._cars.CabooseInConsist(); + Model.Car? foundCaboose = oiler._originCar.FindMyCaboose(0.0f, false); if (originIndex < 0) { _log.Error("Couldn't find origin car {car}", oiler._originCar); @@ -37,6 +39,7 @@ namespace RMROC451.TweaksAndThings.Extensions while (true) { yield return new WaitForSeconds(AutoOiler.StartDelay.CabooseHalvedFloat(foundCaboose)); + foundCaboose = oiler._originCar.FindMyCaboose(0.0f,false); int carIndex = originIndex; float adjustedTimeToWalk = AutoOiler.TimeToWalkCar.CabooseHalvedFloat(foundCaboose); do @@ -58,7 +61,7 @@ namespace RMROC451.TweaksAndThings.Extensions if (car.HasHotbox && car.Oiled == 1f && cabooseRequired && foundCaboose) { _log.Information("AutoOiler {name}: {foundCaboose} repaired hotbox {car}", oiler.name, foundCaboose, car); - car.AdjustHotboxValue(0f); + car.AdjustHotboxValue(); } num += adjustedTimeToWalk; oiler._pendingRunDuration += adjustedTimeToWalk; @@ -74,18 +77,18 @@ namespace RMROC451.TweaksAndThings.Extensions public static IEnumerator MrocAutoHotboxSpotterLoop(this AutoHotboxSpotter spotter, Serilog.ILogger _log, bool cabooseRequired) { + Func foundCaboose = () => spotter._locomotive.FindMyCaboose(0.0f, false); while (true) { - Model.Car? foundCaboose = spotter._cars.CabooseInConsist(); if (!spotter.HasCars) { yield return new WaitForSeconds(1f); continue; } + var fc = foundCaboose(); _log.Information("AutoHotboxSpotter {name}: Hotbox Spotter Running, Found Caboose => {hasCaboose}; Has Cars {hasCars}; Requires Caboose {requiresCaboose}", - spotter.name, foundCaboose, spotter.HasCars, cabooseRequired); - foundCaboose = spotter._cars.CabooseInConsist(); - if (CabooseRequirementChecker(string.Format("{0} {1}", spotter.GetType().Name, spotter.name), cabooseRequired, foundCaboose, _log)) + spotter.name, fc, spotter.HasCars, cabooseRequired); + if (CabooseRequirementChecker(string.Format("{0} {1}", spotter.GetType().Name, spotter.name), cabooseRequired, fc, _log)) { yield break; } @@ -93,12 +96,12 @@ namespace RMROC451.TweaksAndThings.Extensions while (spotter.HasCars) { int num = Random.Range(60, 300); - foundCaboose = spotter._cars.CabooseInConsist(); - if (foundCaboose) + fc = foundCaboose(); + if (fc) { var numOrig = num; num = Random.Range(15, 30); - _log.Information("AutoHotboxSpotter {name}: Next check went from num(60,300) => {numOrig}; to num(15,30) => {hasCaboose}; Requires Caboose {requiresCaboose}", spotter.name, numOrig, num, foundCaboose, cabooseRequired); + _log.Information("AutoHotboxSpotter {name}: Next check went from num(60,300) => {numOrig}; to num(15,30) => {hasCaboose}; Requires Caboose {requiresCaboose}", spotter.name, numOrig, num, fc, cabooseRequired); } yield return new WaitForSeconds(num); spotter.CheckForHotbox(); diff --git a/TweaksAndThings/Extensions/Car_Extensions.cs b/TweaksAndThings/Extensions/Car_Extensions.cs index bce43ce..5bf77fc 100644 --- a/TweaksAndThings/Extensions/Car_Extensions.cs +++ b/TweaksAndThings/Extensions/Car_Extensions.cs @@ -1,15 +1,15 @@ using Game.Messages; -using Game.State; using Helpers; using Model; using Model.Definition; using Model.Definition.Data; using Model.Ops; +using Model.Ops.Timetable; +using Railloader; using Serilog; using System; using System.Collections.Generic; using System.Linq; -using System.Security.AccessControl; using UnityEngine; namespace RMROC451.TweaksAndThings.Extensions; @@ -53,7 +53,7 @@ public static class Car_Extensions return car; } - public static bool NotMotivePower(this Car car) => car is not BaseLocomotive && car.Archetype != Model.Definition.CarArchetype.Tender; + public static bool MotivePower(this Car car) => car is BaseLocomotive || car.Archetype == Model.Definition.CarArchetype.Tender; /// /// For every car in the consist, cost 1 minute of AI Engineer time. @@ -66,18 +66,42 @@ public static class Car_Extensions public static bool IsCabooseAndStoppedForLoadRefresh(this Car car, bool isFull) => car.IsCaboose() && car.IsStopped(30f) && !isFull; - public static Car? CabooseInConsist(this IEnumerable input) => input.FirstOrDefault(c => c.IsCaboose()); + public static Car? CabooseInConsist(this IEnumerable input) => input.FirstOrDefault(IsCaboose); public static bool ConsistNoFreight(this IEnumerable input) => - input.Where(c => !c.IsLocomotive).Any() && + input.Where(c => !c.MotivePower()).Any() && input - .Where(c => !c.IsLocomotive) - .All(c => !c.Archetype.IsFreight()); + .Where(c => !c.MotivePower()) + .All(c => !c.MrocIsFreight()); - public static Car? CabooseWithSufficientCrewHours(this Car car, float timeNeeded, HashSet carIdsCheckedAlready, bool decrement = false) + public static bool ConsistFreight(this IEnumerable input) => + input.Where(c => !c.MotivePower()).Any() && + input + .Where(c => !c.MotivePower()) + .All(c => c.MrocIsFreight()); + + public static bool MrocIsFreight(this Car c) => + c.Archetype.IsFreight() || c.Archetype switch + { + CarArchetype.LocomotiveDiesel => false, + CarArchetype.LocomotiveSteam => false, + CarArchetype.Coach => false, + CarArchetype.Baggage => false, + _ => true + }; + + + public static bool SelectedEngineExpress(this TrainController input) => + input.SelectedLocomotive.TryGetTimetableTrain(out Timetable.Train t) && + t.TrainClass == Timetable.TrainClass.First; + + public static Car? FindMyCaboose(this Car car, float timeNeeded, bool decrement = false) => + car.CarsNearCurrentCar(timeNeeded, decrement).FindNearestCabooseFromNearbyCars(); + + public static Car? CabooseWithSufficientCrewHours(this Car car, float timeNeeded, bool decrement = false) { Car? output = null; - if (carIdsCheckedAlready.Contains(car.id) || !car.IsCaboose()) return null; + if (!car.IsCaboose()) return null; List loadSlots = car.Definition.LoadSlots; for (int i = 0; i < loadSlots.Count; i++) @@ -94,51 +118,49 @@ public static class Car_Extensions return output; } - public static Car? HuntingForCabeeseNearCar(this Car car, float timeNeeded, TrainController tc, HashSet carIdsCheckedAlready, bool decrement = false) + private static Car FindNearestCabooseFromNearbyCars(this List<(Car car, bool crewCar, float distance)> source) => + source + .OrderBy(c => c.crewCar ? 0 : 1) + .ThenBy(c => c.distance) + .Select(c => c.car) + .FirstOrDefault(); + + private static List<(Car car, bool crewCar, float distance)> CarsNearCurrentCar(this Car car, float timeNeeded, bool decrement) { - List<(string carId, float distance)> source = - CarsNearCurrentCar(car, timeNeeded, tc, carIdsCheckedAlready, decrement); - Car output = FindNearestCabooseFromNearbyCars(tc, source); - if (output != null) output.CabooseWithSufficientCrewHours(timeNeeded, carIdsCheckedAlready, decrement); - - return output; - } - - private static Car FindNearestCabooseFromNearbyCars(TrainController tc, List<(string carId, float distance)> source) => - ( - from t in source - where t.distance < 21f //todo: add setting slider for catchment - orderby t.distance ascending - select tc.CarForId(t.carId) - ).FirstOrDefault(); - - private static List<(string carId, float distance)> CarsNearCurrentCar(Car car, float timeNeeded, TrainController tc, HashSet carIdsCheckedAlready, bool decrement) - { - Vector3 position = car.GetMotionSnapshot().Position; - Vector3 center = WorldTransformer.WorldToGame(position); - var cars = tc.CarIdsInRadius(center, 60f); - Log.Information($"{nameof(HuntingForCabeeseNearCar)} => {cars.Count()}"); - List<(string carId, float distance)> source = - cars - .Select(carId => + var cabeese = + car.EnumerateCoupled().SelectMany(consistCar => { - Car car = tc.CarForId(carId); - if (car == null || !car.CabooseWithSufficientCrewHours(timeNeeded, carIdsCheckedAlready)) - { - return (carId: carId, distance: 1000f); - } - Vector3 a = WorldTransformer.WorldToGame(car.GetMotionSnapshot().Position); - return (carId: carId, distance: Vector3.Distance(a, center)); - }).ToList(); + Vector3 position = consistCar.GetMotionSnapshot().Position; + Vector3 center = WorldTransformer.WorldToGame(position); + var o = TrainController.Shared + .CarIdsInRadius(center, SingletonPluginBase.Shared.CabeeseSearchRadiusInMeters()) + .Where(c => TrainController.Shared.CarForId(c).IsCaboose()); + return o; + }).Distinct().Select(c => TrainController.Shared.CarForId(c)); + + + Log.Information($"{nameof(CarsNearCurrentCar)} => {cabeese.Count()}"); + + List<(Car car, bool crewCar, float distance)> source = + cabeese.Select(c => (car: c, crewCar: c.IsCrewCar(), distance: car.Distance(c))).ToList(); + return source; } - public static void AdjustHotboxValue(this Car car, float hotboxValue) => - StateManager.ApplyLocal( - new PropertyChange( - car.id, PropertyChange.KeyForControl(PropertyChange.Control.Hotbox), - new FloatPropertyValue(hotboxValue) - ) - ); + public static float Distance(this Car car1, Car car2) + { + Vector3 position = car1.GetMotionSnapshot().Position; + Vector3 center = WorldTransformer.WorldToGame(position); + Vector3 a = WorldTransformer.WorldToGame(car2.GetMotionSnapshot().Position); + + return Vector3.Distance(a, center); + } + + public static bool IsCrewCar(this Car car) => + !string.IsNullOrEmpty(TrainController.Shared.SelectedLocomotive.trainCrewId) && + car.trainCrewId == TrainController.Shared.SelectedLocomotive.trainCrewId; + + + public static void AdjustHotboxValue(this Car car) => car.ControlProperties[PropertyChange.Control.Hotbox] = null; } diff --git a/TweaksAndThings/Extensions/TextSprite_Extensions.cs b/TweaksAndThings/Extensions/TextSprite_Extensions.cs index 7e58bc0..7c4dc24 100644 --- a/TweaksAndThings/Extensions/TextSprite_Extensions.cs +++ b/TweaksAndThings/Extensions/TextSprite_Extensions.cs @@ -7,7 +7,7 @@ namespace RMROC451.TweaksAndThings.Extensions; public static class TextSprite_Extensions { - public static string TriColorPiePercent(this float quantity, float capacity) + public static string TriColorPiePercent(this float quantity, float capacity, string name = "") { int num; if (capacity <= 0f) @@ -29,6 +29,8 @@ public static class TextSprite_Extensions color = "#D53427"; //Red } - return $""; + string sprite = string.IsNullOrEmpty(name) ? $"Pie{num:D2}" : name; + + return $""; } } diff --git a/TweaksAndThings/Patches/AutoEngineerControlSetBase_UpdateStatusLabel_Patch.cs b/TweaksAndThings/Patches/AutoEngineerControlSetBase_UpdateStatusLabel_Patch.cs new file mode 100644 index 0000000..d48abda --- /dev/null +++ b/TweaksAndThings/Patches/AutoEngineerControlSetBase_UpdateStatusLabel_Patch.cs @@ -0,0 +1,22 @@ +using HarmonyLib; +using Railloader; +using UI.Builder; +using UI.EngineControls; + +namespace RMROC451.TweaksAndThings.Patches; + +[HarmonyPatch(typeof(AutoEngineerControlSetBase))] +[HarmonyPatch(nameof(AutoEngineerControlSetBase.UpdateStatusLabel))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class AutoEngineerControlSetBase_UpdateStatusLabel_Patch +{ + static void Postfix(AutoEngineerControlSetBase __instance) + { + TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; + if (!tweaksAndThings.IsEnabled() || !AutoEngineerOrdersHelper_SendAutoEngineerCommand_Patch.SafetyFirstGoverningApplies()) return; + + string orig = __instance.statusLabel.text; + __instance.statusLabel.text = $"{orig}; Safety"; + __instance.statusLabel.rectTransform.Tooltip("Status", $"Safety First, Speed Limited\n\n{orig}"); + } +} diff --git a/TweaksAndThings/Patches/AutoEngineerOrdersHelper_SendAutoEngineerCommand_Patch.cs b/TweaksAndThings/Patches/AutoEngineerOrdersHelper_SendAutoEngineerCommand_Patch.cs new file mode 100644 index 0000000..623a66f --- /dev/null +++ b/TweaksAndThings/Patches/AutoEngineerOrdersHelper_SendAutoEngineerCommand_Patch.cs @@ -0,0 +1,92 @@ +using Game.Messages; +using HarmonyLib; +using Model.AI; +using Model.Definition; +using Railloader; +using RMROC451.TweaksAndThings.Extensions; +using Serilog; +using System; +using System.Linq; +using UI.EngineControls; + +namespace RMROC451.TweaksAndThings.Patches; + + +[HarmonyPatch(typeof(AutoEngineerOrdersHelper))] +[HarmonyPatch(nameof(AutoEngineerOrdersHelper.SendAutoEngineerCommand), typeof(AutoEngineerMode), typeof(bool), typeof(int), typeof(float), typeof(OrderWaypoint))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class AutoEngineerOrdersHelper_SendAutoEngineerCommand_Patch +{ + + private static Serilog.ILogger _log => Log.ForContext(); + + static bool Prefix(AutoEngineerMode mode, ref int maxSpeedMph) + { + TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; + if (!tweaksAndThings.IsEnabled() || !tweaksAndThings.SafetyFirst()) return true; + + if (SafetyFirstGoverningApplies()) + { + int orig = maxSpeedMph; + int limitedSpeed = Math.Min(maxSpeedMph, 20); + maxSpeedMph = mode switch + { + AutoEngineerMode.Road => limitedSpeed, + AutoEngineerMode.Waypoint => limitedSpeed, + _ => maxSpeedMph, + }; + + if (orig != maxSpeedMph) + { + _log.Information($"{Enum.GetName(typeof(AutoEngineerMode), mode)}[{TrainController.Shared.SelectedLocomotive.DisplayName}] {nameof(AutoEngineerOrdersExtensions.MaxSpeedMph)} limited to {limitedSpeed} from {orig}; No Caboose in Consist;"); + } + } + + return true; + + } + + internal static bool SafetyFirstGoverningApplies() + { + var _persistence = new AutoEngineerPersistence(TrainController.Shared.SelectedLocomotive.KeyValueObject); + var OrdersHelper = new AutoEngineerOrdersHelper(TrainController.Shared.SelectedLocomotive, _persistence); + + bool cabooseReq = SingletonPluginBase.Shared.RequireConsistCabooseForOilerAndHotboxSpotter(); + string logMessage = $"\n{nameof(SafetyFirstGoverningApplies)}:{Enum.GetName(typeof(AutoEngineerMode), OrdersHelper.Mode)}[{TrainController.Shared.SelectedLocomotive.DisplayName}] "; + Func firstClass = () => + { + var output = TrainController.Shared.SelectedEngineExpress(); + logMessage += $"\nfirst class {output}"; + return output; + }; + + Func FreightConsist = () => + { + bool output = !TrainController.Shared.SelectedLocomotive.EnumerateCoupled().ConsistNoFreight(); + logMessage += $"\nFreightConsist? {output}"; + logMessage += " " + string.Join(" / ", TrainController.Shared.SelectedLocomotive.EnumerateCoupled().Where(c => !c.MotivePower()).Select(c => $"{c.id} {Enum.GetName(typeof(CarArchetype), c.Archetype)}")); + return output; + }; + + Func noCaboose = () => + { + var output = TrainController.Shared.SelectedLocomotive.FindMyCaboose(0.0f, false) == null; + logMessage += $"\ncaboose? {!output}"; + return output; + }; + + logMessage += $"\nCaboose Required {cabooseReq}"; + + bool output = + cabooseReq && + !firstClass() && + FreightConsist() && + noCaboose(); + + logMessage += $"\nGovern AE? {output}"; + + _log.Information(logMessage); + + return output; + } +} diff --git a/TweaksAndThings/Patches/AutoEngineerOrdersHelper_SetWaypoint_patch.cs b/TweaksAndThings/Patches/AutoEngineerOrdersHelper_SetWaypoint_patch.cs new file mode 100644 index 0000000..a512189 --- /dev/null +++ b/TweaksAndThings/Patches/AutoEngineerOrdersHelper_SetWaypoint_patch.cs @@ -0,0 +1,26 @@ +using Game.Notices; +using HarmonyLib; +using Model; +using Serilog; +using UI.EngineControls; +using UnityEngine; + +namespace RMROC451.TweaksAndThings.Patches; + + +[HarmonyPatch(typeof(AutoEngineerOrdersHelper))] +[HarmonyPatch(nameof(AutoEngineerOrdersHelper.SetWaypoint), typeof(Track.Location), typeof(string))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class AutoEngineerOrdersHelper_SetWaypoint_patch +{ + private static Serilog.ILogger _log => Log.ForContext(); + static void Postfix(AutoEngineerOrdersHelper __instance, Track.Location location, string coupleToCarId) + { + _log.Information($"start setWP"); + Car selectedLoco = __instance._locomotive; + _log.Information($"{selectedLoco?.DisplayName ?? ""} set WP"); + Vector3 gamePoint = location.GetPosition(); + EntityReference entityReference = new EntityReference(EntityType.Position, new Vector4(gamePoint.x, gamePoint.y, gamePoint.z, 0)); + selectedLoco.PostNotice("ai-wpt-rmroc451", new Hyperlink(entityReference.URI(), $"WP SET")); + } +} diff --git a/TweaksAndThings/Patches/AutoHotboxSpotter_SpotterLoop_Patch.cs b/TweaksAndThings/Patches/AutoHotboxSpotter_SpotterLoop_Patch.cs index f0fd551..41970c9 100644 --- a/TweaksAndThings/Patches/AutoHotboxSpotter_SpotterLoop_Patch.cs +++ b/TweaksAndThings/Patches/AutoHotboxSpotter_SpotterLoop_Patch.cs @@ -17,7 +17,7 @@ internal class AutoHotboxSpotter_SpotterLoop_Patch public static bool Prefix(AutoHotboxSpotter __instance, ref IEnumerator __result) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return true; + if (!tweaksAndThings.IsEnabled()) return true; bool buttonsHaveCost = tweaksAndThings.EndGearHelpersRequirePayment(); bool cabooseRequired = tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter(); diff --git a/TweaksAndThings/Patches/AutoOiler_Loop_Patch.cs b/TweaksAndThings/Patches/AutoOiler_Loop_Patch.cs index afc2602..b4261f5 100644 --- a/TweaksAndThings/Patches/AutoOiler_Loop_Patch.cs +++ b/TweaksAndThings/Patches/AutoOiler_Loop_Patch.cs @@ -17,9 +17,9 @@ internal class AutoOiler_Loop_Patch public static bool Prefix(AutoOiler __instance, ref IEnumerator __result) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return true; + if (!tweaksAndThings.IsEnabled()) return true; bool buttonsHaveCost = tweaksAndThings.EndGearHelpersRequirePayment(); - bool cabooseRequired = tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter(); + bool cabooseRequired = tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter() && !__instance._cars.ConsistNoFreight(); if (buttonsHaveCost) __result = __instance.MrocAutoOilerLoop(_log, cabooseRequired); return !buttonsHaveCost; //only hit this if !buttonsHaveCost, since Loop is a coroutine diff --git a/TweaksAndThings/Patches/CarInspector_PopulateCarPanel_Patch.cs b/TweaksAndThings/Patches/CarInspector_PopulateCarPanel_Patch.cs index 341a70a..ecd6ae8 100644 --- a/TweaksAndThings/Patches/CarInspector_PopulateCarPanel_Patch.cs +++ b/TweaksAndThings/Patches/CarInspector_PopulateCarPanel_Patch.cs @@ -40,7 +40,7 @@ internal class CarInspector_PopulateCarPanel_Patch { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return true; + if (!tweaksAndThings.IsEnabled()) return true; bool buttonsHaveCost = tweaksAndThings.EndGearHelpersRequirePayment(); var consist = __instance._car.EnumerateCoupled(); @@ -82,6 +82,7 @@ internal class CarInspector_PopulateCarPanel_Patch { builder.HStack(delegate (UIPanelBuilder hstack) { + hstack = AddCarConsistRebuildObservers(hstack, consist, all: false); hstack.AddField("Consist Info", hstack.HStack(delegate (UIPanelBuilder field) { int consistLength = consist.Count(); @@ -96,7 +97,7 @@ internal class CarInspector_PopulateCarPanel_Patch } } - private static UIPanelBuilder AddCarConsistRebuildObservers(UIPanelBuilder builder, IEnumerable consist) + private static UIPanelBuilder AddCarConsistRebuildObservers(UIPanelBuilder builder, IEnumerable consist, bool all = true) { TagController tagController = UnityEngine.Object.FindFirstObjectByType(); foreach (Model.Car car in consist.Where(c => c.Archetype != Model.Definition.CarArchetype.Tender)) @@ -105,8 +106,8 @@ internal class CarInspector_PopulateCarPanel_Patch foreach (LogicalEnd logicalEnd in ends) { builder = AddObserver(builder, car, KeyValueKeyFor(EndGearStateKey.IsCoupled, car.LogicalToEnd(logicalEnd)), tagController); - builder = AddObserver(builder, car, KeyValueKeyFor(EndGearStateKey.IsAirConnected, car.LogicalToEnd(logicalEnd)), tagController); - builder = AddObserver(builder, car, KeyValueKeyFor(EndGearStateKey.Anglecock, car.LogicalToEnd(logicalEnd)), tagController); + if (all) builder = AddObserver(builder, car, KeyValueKeyFor(EndGearStateKey.IsAirConnected, car.LogicalToEnd(logicalEnd)), tagController); + if (all) builder = AddObserver(builder, car, KeyValueKeyFor(EndGearStateKey.Anglecock, car.LogicalToEnd(logicalEnd)), tagController); } } @@ -148,8 +149,9 @@ internal class CarInspector_PopulateCarPanel_Patch // } //} - public static void MrocConsistHelper(Model.Car car, MrocHelperType mrocHelperType, bool buttonsHaveCost) + public static int MrocConsistHelper(Model.Car car, MrocHelperType mrocHelperType, bool buttonsHaveCost) { + int output = 0; TrainController tc = UnityEngine.Object.FindObjectOfType(); IEnumerable consist = car.EnumerateCoupled(); _log.ForContext("car", car).Verbose($"{car} => {mrocHelperType} => {string.Join("/", consist.Select(c => c.ToString()))}"); @@ -178,14 +180,26 @@ internal class CarInspector_PopulateCarPanel_Patch break; case MrocHelperType.BleedAirSystem: - consist = consist.Where(c => c.NotMotivePower()); + consist = consist.Where(c => !c.MotivePower()); _log.ForContext("car", car).Information($"{car} => {mrocHelperType} => {string.Join("/", consist.Select(c => c.ToString()))}"); foreach (Model.Car bleed in consist) { StateManager.ApplyLocal(new PropertyChange(bleed.id, PropertyChange.Control.Bleed, 1)); } break; + + case MrocHelperType.Oil: + consist = consist.Where(c => c.NeedsOiling || c.HasHotbox); + _log.ForContext("car", car).Information($"{car} => {mrocHelperType} => {string.Join("/", consist.Select(c => c.ToString()))}"); + foreach (Model.Car oil in consist) + { + StateManager.ApplyLocal(new PropertyChange(oil.id, nameof(Car.Oiled).ToLower(), new FloatPropertyValue(1))); + output += car.HasHotbox ? 1 : 0; + if (car.HasHotbox) car.AdjustHotboxValue(); + } + break; } + return output; } internal static void CarEndAirUpdate(Car c) @@ -214,7 +228,7 @@ internal class CarInspector_PopulateCarPanel_Patch float timeCost = originalTimeCost; float crewCost = timeCost / 3600; //hours of time deducted from caboose. var tsString = crewCost.FormatCrewHours(OpsController_AnnounceCoalescedPayments_Patch.CrewHoursLoad().description); - Car? cabooseWithAvailCrew = NearbyCabooseWithAvailableCrew(car, crewCost, buttonsHaveCost); + Car? cabooseWithAvailCrew = car.FindMyCaboose(crewCost, buttonsHaveCost); if (cabooseWithAvailCrew == null) timeCost *= 1.5f; var cabooseFoundDisplay = cabooseWithAvailCrew?.DisplayName ?? "No caboose"; @@ -226,30 +240,4 @@ internal class CarInspector_PopulateCarPanel_Patch StateManager_OnDayDidChange_Patch.UnbilledAutoBrakeCrewRunDuration += timeCost; } } - - public static Car? NearbyCabooseWithAvailableCrew(Car car, float timeNeeded, bool decrement = false) - { - HashSet carIdsCheckedAlready = new(); - - //check current car. - Car? output = car.CabooseWithSufficientCrewHours(timeNeeded, carIdsCheckedAlready, decrement); - if (output != null) return output; //short out if we are good - carIdsCheckedAlready.Add(car.id); - - //check consist, for cabeese - IEnumerable consist = car.EnumerateCoupled(); - output = consist.FirstOrDefault(c => c.CabooseWithSufficientCrewHours(timeNeeded, carIdsCheckedAlready, decrement)); - if (output != null) return output; //short out if we are good - carIdsCheckedAlready.UnionWith(consist.Select(c => c.id)); - - //then check near consist cars for cabeese - TrainController tc = UnityEngine.Object.FindObjectOfType(); - foreach (var c in consist) - { - output = c.HuntingForCabeeseNearCar(timeNeeded, tc, carIdsCheckedAlready, decrement); - if (output != null) return output; //short out if we are good - } - - return output; - } } diff --git a/TweaksAndThings/Patches/CarPickable_Activate_Patch.cs b/TweaksAndThings/Patches/CarPickable_Activate_Patch.cs index 192e93f..f20795a 100644 --- a/TweaksAndThings/Patches/CarPickable_Activate_Patch.cs +++ b/TweaksAndThings/Patches/CarPickable_Activate_Patch.cs @@ -1,15 +1,17 @@ using Core; using HarmonyLib; using Model; +using Model.Ops; using Network; using Railloader; using RMROC451.TweaksAndThings.Enums; using RMROC451.TweaksAndThings.Extensions; using RollingStock; using Serilog; +using System; using System.Linq; -using System.Runtime.Remoting.Messaging; using UI; +using UI.Tags; using UnityEngine; @@ -37,12 +39,11 @@ internal class CarPickable_Activate_Patch private static bool Prefix(CarPickable __instance, PickableActivateEvent evt) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return true; + if (!tweaksAndThings.IsEnabled()) return true; bool bCtrlAltHeld = GameInput.IsControlDown && GameInput.IsAltDown; _log.ForContext("car", __instance.car).Information($"{GameInput.IsShiftDown} {GameInput.IsControlDown} {GameInput.IsAltDown} {bCtrlAltHeld} "); - //On Double click of a car: follow if (OnPointerDown(evt, PickableActivation.Primary)) { CameraSelector.shared.FollowCar(__instance.car); @@ -57,27 +58,29 @@ internal class CarPickable_Activate_Patch var consist = __instance.car.EnumerateCoupled(); bool handbrakesApplied = consist.Any(c => c.HandbrakeApplied()); bool airSystemIssues = consist.Any(c => c.EndAirSystemIssue()); - bool needsOiling = GameInput.IsShiftDown && consist.All(c => c.IsStopped()) && consist.Any(c => c.NeedsOiling || c.HasHotbox) && (consist.CabooseInConsist() || !tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter()); + Func cabooseNear = () => (bool)__instance.car.FindMyCaboose(0.0f, false); + bool needsOiling = GameInput.IsShiftDown && consist.All(c => c.IsStopped()) && consist.Any(c => c.NeedsOiling || c.HasHotbox) && (!tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter() || cabooseNear()); var chargeIt = handbrakesApplied || airSystemIssues || needsOiling; //CTRL + ALT + SHIFT : BrakesAngleCocksAndOiling //CTRL + ALT : Release Consist Brakes and Check AngleCocks - //CTRL + SHIFT : Toggle Consist Brakes - //CTRL : Toggle Car Brakes & Airup Car - //ALT + SHIFT : Check Consist AngleCocks - //ALT : Open Inspector to Car + //ALT + SHIFT : toggle consist brakes + //CTRL + SHIFT : Check Consist Angle Cocks + //ALT : Toggle car brakes and & air up cars + //CTRL : NOTHING; BASE CAR INSPECTOR + if (bCtrlAltHeld) { BrakesAngleCocksAndOiling(__instance, tweaksAndThings, GameInput.IsShiftDown, consist, handbrakesApplied, airSystemIssues, needsOiling, chargeIt); _log.ForContext("car", __instance.car).Information($"ctrlAlt{(GameInput.IsShiftDown ? "shift" : string.Empty)}Held!"); output = false; } - else if (GameInput.IsControlDown && GameInput.IsShiftDown) + else if (GameInput.IsAltDown && GameInput.IsShiftDown) { CarInspector_PopulateCarPanel_Patch.MrocConsistHelper(__instance.car, MrocHelperType.Handbrake, tweaksAndThings.EndGearHelpersRequirePayment()); _log.ForContext("car", __instance.car).Information("ctrlShiftHeld!"); output = false; } - else if (GameInput.IsAltDown && GameInput.IsShiftDown) + else if (GameInput.IsControlDown && GameInput.IsShiftDown) { if (airSystemIssues) CarInspector_PopulateCarPanel_Patch.MrocConsistHelper(__instance.car, MrocHelperType.GladhandAndAnglecock, tweaksAndThings.EndGearHelpersRequirePayment()); @@ -85,16 +88,14 @@ internal class CarPickable_Activate_Patch output = false; } else if (GameInput.IsAltDown) - { - UI.CarInspector.CarInspector.Show(__instance.car); - _log.ForContext("car", __instance.car).Information("altHeld!"); - output = false; - } - else if (GameInput.IsControlDown) { __instance.car.SetHandbrake(!__instance.car.HandbrakeApplied()); CarInspector_PopulateCarPanel_Patch.CarEndAirUpdate(__instance.car); + if (__instance.car.TryGetAdjacentCar(Car.LogicalEnd.A, out Model.Car cA)) CarInspector_PopulateCarPanel_Patch.CarEndAirUpdate(cA); + if (__instance.car.TryGetAdjacentCar(Car.LogicalEnd.B, out Model.Car cB)) CarInspector_PopulateCarPanel_Patch.CarEndAirUpdate(cB); _log.ForContext("car", __instance.car).Information("ctrlHeld!"); + TagController.Shared.UpdateTag(__instance.car, __instance.car.TagCallout, OpsController.Shared); + __instance.car.TagCallout.Update(); output = false; } return output; @@ -120,18 +121,9 @@ internal class CarPickable_Activate_Patch if (airSystemIssues) CarInspector_PopulateCarPanel_Patch.MrocConsistHelper(__instance.car, MrocHelperType.GladhandAndAnglecock, false); if (needsOiling) - { - foreach (var car in consist.Where(c => c.NeedsOiling || c.HasHotbox)) - { - float num2 = 1f - car.Oiled; - car.OffsetOiled(num2); - hbFix += car.HasHotbox ? 1 : 0; - if (car.HasHotbox) car.AdjustHotboxValue(0f); - } + hbFix = CarInspector_PopulateCarPanel_Patch.MrocConsistHelper(__instance.car, MrocHelperType.Oil, false); if (hbFix > 0) Multiplayer.Broadcast($"Near {Hyperlink.To(__instance.car)}: \"{hbFix.Pluralize("hotbox") + " repaired!"}\""); - } - if (chargeIt) CarInspector_PopulateCarPanel_Patch.CalculateCostIfEnabled(__instance.car, MrocHelperType.Handbrake, tweaksAndThings.EndGearHelpersRequirePayment(), consist); } diff --git a/TweaksAndThings/Patches/CarPickable_HandleShowContextMenu_Patch.cs b/TweaksAndThings/Patches/CarPickable_HandleShowContextMenu_Patch.cs index 5ceff11..fc174b0 100644 --- a/TweaksAndThings/Patches/CarPickable_HandleShowContextMenu_Patch.cs +++ b/TweaksAndThings/Patches/CarPickable_HandleShowContextMenu_Patch.cs @@ -15,31 +15,58 @@ namespace RMROC451.TweaksAndThings.Patches; [HarmonyPatchCategory("RMROC451TweaksAndThings")] internal class CarPickable_HandleShowContextMenu_Patch { - private static void Postfix(Car car) + private static bool Prefix(Car car) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return; + if (!tweaksAndThings.IsEnabled()) return true; bool buttonsHaveCost = tweaksAndThings.EndGearHelpersRequirePayment(); + TrainController trainController = TrainController.Shared; ContextMenu shared = ContextMenu.Shared; - shared.AddButton(ContextMenuQuadrant.Unused1, $"{(car.EnumerateCoupled().Any(c => c.HandbrakeApplied()) ? "Release " : "Set ")} Consist", SpriteName.Handbrake, delegate + if (ContextMenu.IsShown) + { + shared.Hide(); + } + shared.Clear(); + + shared.AddButton(ContextMenuQuadrant.General, (trainController.SelectedCar == car) ? "Deselect" : "Select", SpriteName.Select, delegate + { + trainController.SelectedCar = ((trainController.SelectedCar == car) ? null : car); + }); + if (GameInput.IsShiftDown) + { + + + if (!car.EnumerateCoupled().Any(c => !c.SupportsBleed())) + { + shared.AddButton(ContextMenuQuadrant.Brakes, $"Bleed Consist", SpriteName.Bleed, delegate + { + CarInspector_PopulateCarPanel_Patch.MrocConsistHelper(car, MrocHelperType.BleedAirSystem, buttonsHaveCost); + }); + } + + shared.AddButton(ContextMenuQuadrant.Brakes, $"{(car.EnumerateCoupled().Any(c => c.HandbrakeApplied()) ? "Release " : "Set ")} Consist", SpriteName.Handbrake, delegate { CarInspector_PopulateCarPanel_Patch.MrocConsistHelper(car, MrocHelperType.Handbrake, buttonsHaveCost); }); if (car.EnumerateCoupled().Any(c => c.EndAirSystemIssue())) { - shared.AddButton(ContextMenuQuadrant.Unused1, $"Air Up Consist", SpriteName.Select, delegate + shared.AddButton(ContextMenuQuadrant.Unused2, $"Air Up Consist", SpriteName.Select, delegate { CarInspector_PopulateCarPanel_Patch.MrocConsistHelper(car, MrocHelperType.GladhandAndAnglecock, buttonsHaveCost); }); } - - if (!car.EnumerateCoupled().Any(c => !c.SupportsBleed())) + } + else { + if (car.SupportsBleed()) { - shared.AddButton(ContextMenuQuadrant.Unused2, $"Bleed Consist", SpriteName.Bleed, delegate + shared.AddButton(ContextMenuQuadrant.Brakes, "Bleed", SpriteName.Bleed, car.SetBleed); + } + shared.AddButton(ContextMenuQuadrant.Brakes, car.air.handbrakeApplied ? "Release Handbrake" : "Apply Handbrake", SpriteName.Handbrake, delegate { - CarInspector_PopulateCarPanel_Patch.MrocConsistHelper(car, MrocHelperType.BleedAirSystem, buttonsHaveCost); + bool apply = !car.air.handbrakeApplied; + car.SetHandbrake(apply); }); } @@ -47,7 +74,10 @@ internal class CarPickable_HandleShowContextMenu_Patch { CameraSelector.shared.FollowCar(car); }); + + shared.Show(car.DisplayName); shared.BuildItemAngles(); shared.StartCoroutine(shared.AnimateButtonsShown()); + return false; } } diff --git a/TweaksAndThings/Patches/ContextMenu_PositionForItem_Patch.cs b/TweaksAndThings/Patches/ContextMenu_PositionForItem_Patch.cs deleted file mode 100644 index 1bef1ed..0000000 --- a/TweaksAndThings/Patches/ContextMenu_PositionForItem_Patch.cs +++ /dev/null @@ -1,25 +0,0 @@ -using HarmonyLib; -using Railloader; -using System; -using UI.ContextMenu; -using UnityEngine; - - -namespace RMROC451.TweaksAndThings.Patches; - -[HarmonyPatch(typeof(UI.ContextMenu.ContextMenu))] -//ContextMenuQuadrant quadrant, int index, float normalizedRadius = 1f -[HarmonyPatch(nameof(UI.ContextMenu.ContextMenu.PositionForItem), typeof(ContextMenuQuadrant), typeof(int), typeof(float))] -[HarmonyPatchCategory("RMROC451TweaksAndThings")] -internal class ContextMenu_PositionForItem_Patch -{ - static void Postfix(UI.ContextMenu.ContextMenu __instance, ref Vector2 __result, ContextMenuQuadrant quadrant, int index, float normalizedRadius = 1f) - { - TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return; - - float num = __instance._itemAngles[(quadrant, index)] * ((float)Math.PI / 180f); - float num2 = __instance.radius * normalizedRadius; - __result = new Vector2(1.3f * Mathf.Sin(num) * num2, Mathf.Cos(num) * num2); - } -} diff --git a/TweaksAndThings/Patches/ContextMenu_Show_Patch.cs b/TweaksAndThings/Patches/ContextMenu_Show_Patch.cs index 94f4406..007c018 100644 --- a/TweaksAndThings/Patches/ContextMenu_Show_Patch.cs +++ b/TweaksAndThings/Patches/ContextMenu_Show_Patch.cs @@ -19,7 +19,7 @@ internal class ContextMenu_Show_Patch static bool Prefix(UI.ContextMenu.ContextMenu __instance, string centerText) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return true; + if (!tweaksAndThings.IsEnabled()) return true; if (!__instance.GetRootCanvas(out var rootCanvas)) { @@ -77,7 +77,7 @@ internal class ContextMenu_DefaultAngleForItem_Patch static bool Prefix(UI.ContextMenu.ContextMenu __instance, ref float __result, ContextMenuQuadrant quadrant, int index, int quadrantItemCount) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return true; + if (!tweaksAndThings.IsEnabled()) return true; int num = quadrant switch @@ -108,7 +108,7 @@ internal class ContextMenu_AddButton_Patch static bool Prefix(UI.ContextMenu.ContextMenu __instance, ContextMenuQuadrant quadrant, string title, Sprite sprite, Action action) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return true; + if (!tweaksAndThings.IsEnabled()) return true; List list = __instance._quadrants[(int)quadrant]; diff --git a/TweaksAndThings/Patches/EngineRosterRow_Refresh_Patch.cs b/TweaksAndThings/Patches/EngineRosterRow_Refresh_Patch.cs index 7c629ec..5d21336 100644 --- a/TweaksAndThings/Patches/EngineRosterRow_Refresh_Patch.cs +++ b/TweaksAndThings/Patches/EngineRosterRow_Refresh_Patch.cs @@ -34,7 +34,7 @@ internal class EngineRosterRow_Refresh_Patch if (tweaksAndThings == null || rosterFuelColumnSettings == null || - !tweaksAndThings.IsEnabled || + !tweaksAndThings.IsEnabled() || rosterFuelColumnSettings.EngineRosterFuelStatusColumn == EngineRosterFuelDisplayColumn.None || (!GameInput.IsAltDown && !rosterFuelColumnSettings.EngineRosterShowsFuelStatusAlways)) { return; @@ -43,7 +43,10 @@ internal class EngineRosterRow_Refresh_Patch try { IEnumerable consist = __instance._engine.EnumerateCoupled().Where(c => c.EnableOiling); - bool cabooseRequirementFulfilled = consist.CabooseInConsist() || !tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter() || consist.ConsistNoFreight(); + bool cabooseRequirementFulfilled = + !tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter() + || consist.ConsistNoFreight() + || (bool)__instance._engine.FindMyCaboose(0.0f, false); float offendingPercentage = 0; Car engineOrTender = __instance._engine; List loadSlots = __instance._engine.Definition.LoadSlots; diff --git a/TweaksAndThings/Patches/EntityReference_Text_Patch.cs b/TweaksAndThings/Patches/EntityReference_Text_Patch.cs new file mode 100644 index 0000000..96c3222 --- /dev/null +++ b/TweaksAndThings/Patches/EntityReference_Text_Patch.cs @@ -0,0 +1,15 @@ +using HarmonyLib; + +namespace RMROC451.TweaksAndThings.Patches; + +[HarmonyPatch(typeof(EntityReference))] +[HarmonyPatch(nameof(EntityReference.Text))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class EntityReference_Text_Patch +{ + static void Postfix(ref string __result) + { + if (__result == "Unknown") + __result = string.Empty; + } +} diff --git a/TweaksAndThings/Patches/ExpandedConsole_Patch.cs b/TweaksAndThings/Patches/ExpandedConsole_Patch.cs index d651960..50b2461 100644 --- a/TweaksAndThings/Patches/ExpandedConsole_Patch.cs +++ b/TweaksAndThings/Patches/ExpandedConsole_Patch.cs @@ -25,13 +25,15 @@ internal class ExpandedConsole_Add_Patch { entry.Text = $"{entry.Timestamp} : {entry.Text}"; entry.Timestamp = RealNow(); - SendMs(ref entry); + SendMs((UI.Console.Console.Entry?)entry); } - static void SendMs(ref UI.Console.Console.Entry entry) + internal static void SendMs(UI.Console.Console.Entry? entry, string? text = null) { try { + if (entry is null && !String.IsNullOrEmpty(text)) entry = new() { Text = text }; + var msgText = entry?.Text ?? string.Empty; TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; StateManager shared = StateManager.Shared; GameStorage gameStorage = shared.Storage; @@ -46,7 +48,7 @@ internal class ExpandedConsole_Add_Patch { var t = new Regex("car:(.*)\""); - var carId = t.IsMatch(entry.Text) ? Regex.Match(entry.Text, "car:(.*?)\"").Groups[1].Captures[0].ToString() : string.Empty; + var carId = t.IsMatch(msgText) ? Regex.Match(msgText, "car:(.*?)\"").Groups[1].Captures[0].ToString() : string.Empty; Model.Car? car = TrainController.Shared.CarForString(carId); bool engineInMessage = car?.IsLocomotive ?? false; var image = engineInMessage ? @@ -81,7 +83,7 @@ internal class ExpandedConsole_Add_Patch { new { - description= Regex.Replace(entry.Text, "<.*?>", "").Replace(": ", "\n"), + description= Regex.Replace(msgText, "<.*?>", "").Replace(": ", "\n"), timestamp=DateTime.UtcNow, image }, diff --git a/TweaksAndThings/Patches/InterchangedIndustryLoader_ServiceInterchange_Patch.cs b/TweaksAndThings/Patches/InterchangedIndustryLoader_ServiceInterchange_Patch.cs new file mode 100644 index 0000000..14660a5 --- /dev/null +++ b/TweaksAndThings/Patches/InterchangedIndustryLoader_ServiceInterchange_Patch.cs @@ -0,0 +1,104 @@ +using Core; +using Game; +using Game.State; +using HarmonyLib; +using Model.Ops; +using Network; +using Railloader; +using Serilog; +using System; +using System.Collections.Generic; +using System.Linq; +using UnityEngine; + +namespace RMROC451.TweaksAndThings.Patches; + +[HarmonyPatch(typeof(InterchangedIndustryLoader))] +[HarmonyPatch(nameof(InterchangedIndustryLoader.ServeInterchange), typeof(IIndustryContext), typeof(Interchange))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class InterchangedIndustryLoader_ServiceInterchange_Patch +{ + private static Serilog.ILogger _log => Log.ForContext(); + /// + /// Allow Interchange service for buying parts/fuel when negative balance, adding an additional 20% penalty + /// + /// + /// + /// + /// + public static bool Prefix(InterchangedIndustryLoader __instance, IIndustryContext ctx, Interchange interchange) + { + TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; + if (!StateManager.IsHost || !tweaksAndThings.IsEnabled() || !tweaksAndThings.ServiceFundPenalties()) return true; + + StateManager shared = StateManager.Shared; + List list = (from car in EnumerateCars(__instance, ctx, requireWaybill: true) + where car.IsEmptyOrContains(__instance.load) + select car).ToList(); + int num2 = 0; + int num3 = 0; + int penalty = 0; + GameDateTime returnTime = ctx.Now.AddingDays(23f / 24f); + foreach (IOpsCar item3 in list) + { + (float quantity, float capacity) tuple = item3.QuantityOfLoad(__instance.load); + float item = tuple.quantity; + float item2 = tuple.capacity; + int num4 = Mathf.RoundToInt((item2 - item) * __instance.load.costPerUnit); + if (num4 > 0) + { + var canAfford = shared.CanAfford(num4); + penalty = Mathf.CeilToInt(!canAfford ? num4 * 0.2f : 0); + __instance.Industry.ApplyToBalance(-num4, __instance.ledgerCategory, null, 0, quiet: true); + + num2 += num4; + num3++; + } + item3.Load(__instance.load, item2); + item3.SetWaybill(null, __instance, "Full"); + ctx.MoveToBardo(item3); + __instance.ScheduleReturnFromBardo(item3, returnTime); + } + if (num2 > 0) + { + __instance.Industry.ApplyToBalance(-num2, __instance.ledgerCategory, null, num3, quiet: true); + string penaltyText = penalty > 0 ? $"; Overdraft fee of {penalty:C0}" : string.Empty; + Multiplayer.Broadcast(string.Format("{6}: Ordered {0} of {1} at {2} for {3:C0}{5}. Expected return: {4}.", num3.Pluralize("car"), __instance.load.description, __instance.HyperlinkToThis, num2, 1.Pluralize("day"), penaltyText, __instance.HyperlinkToThis)); + if (penalty > 0) + StateManager.Shared.ApplyToBalance(-penalty, __instance.ledgerCategory, null, $"Overdraft: {__instance.Industry.name}", 0, quiet: true); + } + + return false; + } + + private static IEnumerable EnumerateCars(InterchangedIndustryLoader __instance, IIndustryContext ctx, bool requireWaybill = false) + { + foreach (IOpsCar item in ctx.CarsAtPosition()) + { + if (!CarTypeMatches(__instance.Interchange, item)) continue; + + if (requireWaybill) + { + try + { + Waybill? waybill = item.Waybill; + string destId = waybill?.Destination.Identifier ?? string.Empty; + if (!waybill.HasValue || !destId.Equals(__instance.Identifier)) continue; + + } catch (Exception ex) + { + _log.Error(ex, $"{item.DisplayName} issue detecting waybill"); + continue; + } + } + yield return item; + } + } + + private static bool CarTypeMatches(Interchange interchange, IOpsCar car) + { + string carType = car.CarType; + return interchange.carTypeFilter.Matches(carType); + } + +} diff --git a/TweaksAndThings/Patches/MapWindow_OnClick_Patch.cs b/TweaksAndThings/Patches/MapWindow_OnClick_Patch.cs new file mode 100644 index 0000000..5619df3 --- /dev/null +++ b/TweaksAndThings/Patches/MapWindow_OnClick_Patch.cs @@ -0,0 +1,183 @@ +using HarmonyLib; +using Helpers; +using Map.Runtime; +using Model; +using Model.AI; +using Railloader; +using Serilog; +using System.Collections.Generic; +using System.IO; +using System.Linq; +using Track; +using UI; +using UI.EngineControls; +using UI.Map; +using UnityEngine; +using UnityEngine.Pool; +using UnityEngine.UI; +using static UI.AutoEngineerDestinationPicker; +using Vector2 = UnityEngine.Vector2; +using Vector3 = UnityEngine.Vector3; + +namespace RMROC451.TweaksAndThings.Patches; + +[HarmonyPatch(typeof(MapWindow))] +[HarmonyPatch(nameof(MapWindow.OnClick), typeof(Vector2))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class MapWindow_OnClick_Patch +{ + private static Serilog.ILogger _log => Log.ForContext(); + private static GameObject _prefabHolder; + private static MapIcon _waypointPrefab; + private static Dictionary wps = new Dictionary(); + + public static Sprite? LoadTexture(string fileName, string name) + { + string path = Path.Combine(SingletonPluginBase.Shared.ModDirectory, fileName); + Texture2D texture2D = new Texture2D(128, 128, TextureFormat.DXT5, mipChain: false); + texture2D.name = name; + texture2D.wrapMode = TextureWrapMode.Clamp; + if (!ImageConversion.LoadImage(texture2D, File.ReadAllBytes(path))) + { + _log.Information($"Unable to load {name} icon!"); + return null; + } + Sprite sprite = Sprite.Create(texture2D, new Rect(0f, 0f, texture2D.width, texture2D.height), new Vector2(0.5f, 0.5f)); + sprite.name = name; + + //if (!SpriteLibrary.Shared.entries.Any(s => s.name.ToString() == name)) SpriteLibrary.Shared.entries.Add(new() { name = SpriteName.Meh, sprite = sprite }); + return sprite; + } + public static MapIcon? waypointPrefab + { + get + { + if (_waypointPrefab == null) + { + CreateWaypointPrefab(); + } + return _waypointPrefab; + } + } + + internal static GameObject prefabHolder + { + get + { + if (_prefabHolder == null) + { + _prefabHolder = new GameObject("Prefab Holder"); + _prefabHolder.hideFlags = HideFlags.HideAndDontSave; + _prefabHolder.SetActive(value: false); + } + return _prefabHolder; + } + } + + private static void CreateWaypointPrefab() + { + float num = 0.6f; //come back + Sprite sprite = LoadTexture("Map_pin_icon.png", "MapPin"); + _waypointPrefab = Object.Instantiate(TrainController.Shared.locomotiveMapIconPrefab, prefabHolder.transform); + GameObject obj = _waypointPrefab.gameObject; + obj.hideFlags = HideFlags.HideAndDontSave; + obj.name = "Map Waypoint Icon"; + if ((bool)_waypointPrefab.Text) + { + Object.DestroyImmediate(_waypointPrefab.Text.gameObject); + } + Image componentInChildren = _waypointPrefab.GetComponentInChildren(); + componentInChildren.sprite = sprite; + componentInChildren.transform.localScale = Vector3.one * num; + } + + static bool Prefix(MapWindow __instance, Vector2 viewportNormalizedPoint) + { + if (GameInput.IsControlDown && GameInput.IsAltDown) + { + Ray ray = RayForViewportNormalizedPoint(__instance, viewportNormalizedPoint); + Vector3 gamePoint = MapManager.Instance.FindTerrainPointForXZ(WorldTransformer.WorldToGame(ray.origin)); + var selectedLoco = TrainController.Shared?.SelectedLocomotive; + if (selectedLoco != null) + { + Hit? valueOrDefault = null; + Camera _camera = null; + if (MainCameraHelper.TryGetIfNeeded(ref _camera)) + { + float rad = 200f; + if (Graph.Shared.TryGetLocationFromGamePoint(gamePoint, rad, out Location location)) + { + Hit? hit = HitLocation(location, selectedLoco); + if (hit.HasValue) + { + valueOrDefault = hit.GetValueOrDefault(); + location = valueOrDefault.Value.Location; + } + var aeoh = new AutoEngineerOrdersHelper(persistence: new AutoEngineerPersistence(selectedLoco.KeyValueObject), locomotive: selectedLoco); + var mw = (location: (Location)location, carId: valueOrDefault?.CarInfo?.car?.id ?? string.Empty); + + aeoh.SetWaypoint(mw.location, mw.carId); + aeoh.SetOrdersValue(maybeWaypoint: mw); + AutoEngineerDestinationPicker.Shared.Cancel(); + } + } + } + return false; + } + return true; + } + + public static Ray RayForViewportNormalizedPoint(MapWindow i, Vector2 v2) => + i.mapBuilder.mapCamera.ViewportPointToRay(new Vector3(v2.x, v2.y, 0f)); + + private static Hit? HitLocation(Location? location, BaseLocomotive selectedLoco) + { + var _graph = TrainController.Shared.graph; + if (location.HasValue) + { + Location valueOrDefault = location.GetValueOrDefault(); + TrainController shared = TrainController.Shared; + Vector3 position = _graph.GetPosition(valueOrDefault); + float num = 2f; + Hit? result = null; + HashSet value; + using (CollectionPool, Car>.Get(out value)) + { + shared.CheckForCarsAtPoint(position, 2f, value, valueOrDefault); + foreach (Car item in value) + { + if (selectedLoco.EnumerateCoupled().ToHashSet().Contains(item)) + { + continue; + } + if (!item[item.EndToLogical(Car.End.F)].IsCoupled) + { + Location location2 = _graph.LocationByMoving(item.LocationF, 0.5f, checkSwitchAgainstMovement: false, stopAtEndOfTrack: true); + float distanceBetweenClose = _graph.GetDistanceBetweenClose(valueOrDefault, location2); + if (distanceBetweenClose < num) + { + num = distanceBetweenClose; + result = new Hit(location2, (item, Car.End.F)); + } + } + if (!item[item.EndToLogical(Car.End.R)].IsCoupled) + { + Location location3 = _graph.LocationByMoving(item.LocationR, -0.5f, checkSwitchAgainstMovement: false, stopAtEndOfTrack: true).Flipped(); + float distanceBetweenClose2 = _graph.GetDistanceBetweenClose(valueOrDefault, location3); + if (distanceBetweenClose2 < num) + { + num = distanceBetweenClose2; + result = new Hit(location3, (item, Car.End.R)); + } + } + } + if (value.Count > 0) + { + return result; + } + } + return new Hit(valueOrDefault, null); + } + return null; + } +} \ No newline at end of file diff --git a/TweaksAndThings/Patches/NoticeExtensions_PostNotice_Patch.cs b/TweaksAndThings/Patches/NoticeExtensions_PostNotice_Patch.cs new file mode 100644 index 0000000..27fe48c --- /dev/null +++ b/TweaksAndThings/Patches/NoticeExtensions_PostNotice_Patch.cs @@ -0,0 +1,34 @@ +using Game.Notices; +using HarmonyLib; +using Model; +using Serilog; +using System; + +namespace RMROC451.TweaksAndThings.Patches; + +[HarmonyPatch(typeof(NoticeExtensions))] +[HarmonyPatch(nameof(NoticeExtensions.PostNotice), typeof(Car), typeof(string), typeof(string))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class NoticeExtensions_PostNotice_Patch +{ + private static ILogger _log => Log.ForContext(); + static void Postfix(Car car, string key, string content) + { + try + { + + //Log.Information($"{car.DisplayName} patch PostNotice"); + if (!string.IsNullOrEmpty(content) && + key.Equals("ai-wpt") && + content.ToLower().Contains("Arrived at Waypoint".ToLower()) + ) + { + car.PostNotice("ai-wpt-rmroc451", null); + } + } catch (Exception ex) + { + _log.ForContext("car", car).Error(ex, "woops"); + } + } + +} diff --git a/TweaksAndThings/Patches/OpsController_AnnounceCoalescedPayments_Patch.cs b/TweaksAndThings/Patches/OpsController_AnnounceCoalescedPayments_Patch.cs index ff2c016..337045a 100644 --- a/TweaksAndThings/Patches/OpsController_AnnounceCoalescedPayments_Patch.cs +++ b/TweaksAndThings/Patches/OpsController_AnnounceCoalescedPayments_Patch.cs @@ -85,13 +85,11 @@ internal class OpsController_AnnounceCoalescedPayments_Patch float rate2 = 24 * 2 * 8;// carLoadRate = 8 crew hours in 30 min loading; (24h * 2 to get half hour chunks * 8 hours to load in those chunks) float num2 = 99999999; //QuantityInStorage for crew-hours (infinite where crew can be shuffling about) float quantityToLoad = Mathf.Min(num2, IndustryComponent.RateToValue(rate2, deltaTime)); - OpsCarAdapter? oca = car.IsCaboose() ? new OpsCarAdapter(car, OpsController.Shared) : null; - bool isFull = !car.IsCaboose() ? true : (oca?.IsFull(CrewHoursLoad()) ?? true); if (car.IsCaboose() && !CrewCarStatus(car).spotted) { - CrewCarDict[car.id] = (true, !isFull); + CrewCarDict[car.id] = (true, CrewCarDict[car.id].filling); } - if (car.IsCabooseAndStoppedForLoadRefresh(isFull)) + if (car.IsCabooseAndStoppedForLoadRefresh(new OpsCarAdapter(car, OpsController.Shared).IsFull(CrewHoursLoad()))) { if (!CrewCarDict[car.id].filling) Multiplayer.Broadcast($"{Hyperlink.To(car)}: \"Topping off caboose crew.\""); CrewCarDict[car.id] = (CrewCarDict[car.id].spotted, true); @@ -101,15 +99,14 @@ internal class OpsController_AnnounceCoalescedPayments_Patch Multiplayer.Broadcast($"{Hyperlink.To(car)}: \"Caboose crew topped off.\""); CrewCarDict[car.id] = (CrewCarDict[car.id].spotted, false); } - (oca ?? new OpsCarAdapter(car, OpsController.Shared)).Load(CrewHoursLoad(), quantityToLoad); + new OpsCarAdapter(car, OpsController.Shared).Load(CrewHoursLoad(), quantityToLoad); } } public static bool Prefix(IndustryComponent __instance) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!StateManager.IsHost || !tweaksAndThings.IsEnabled || !(tweaksAndThings?.settings?.EndGearHelpersRequirePayment ?? false)) return true; - + if (!StateManager.IsHost || !tweaksAndThings.IsEnabled() || !tweaksAndThings.EndGearHelpersRequirePayment() || tweaksAndThings.DayLoadCrewHours()) return true; TrainController tc = UnityEngine.Object.FindAnyObjectByType(); try { diff --git a/TweaksAndThings/Patches/RepairTrack_DailyPayables_Patch.cs b/TweaksAndThings/Patches/RepairTrack_DailyPayables_Patch.cs new file mode 100644 index 0000000..91c2deb --- /dev/null +++ b/TweaksAndThings/Patches/RepairTrack_DailyPayables_Patch.cs @@ -0,0 +1,169 @@ +using Game; +using Game.State; +using HarmonyLib; +using Model; +using Model.Ops; +using Network; +using Railloader; +using Serilog; +using System; +using System.Collections.Generic; +using System.Linq; +using TMPro; +using UI.Builder; +using UnityEngine; +using static Model.Ops.RepairTrack; + +namespace RMROC451.TweaksAndThings.Patches; + +[HarmonyPatch(typeof(RepairTrack))] +[HarmonyPatch(nameof(RepairTrack.DailyPayables), typeof(GameDateTime), typeof(IIndustryContext))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class RepairTrack_DailyPayables_Patch +{ + private static Serilog.ILogger _log => Log.ForContext(); + private static Serilog.ILogger ContextualLogger(RepairTrack __instance) => + _log.ForContext("RepairTrackIdentifier", __instance?.Identifier ?? "???") + .ForContext("RepairTrackName", __instance?.name ?? "???") + .ForContext("RepairTrackDisplayName", __instance?.DisplayName ?? "???"); + + public static bool Prefix(RepairTrack __instance, GameDateTime now, IIndustryContext ctx) + { + TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; + if (!tweaksAndThings.IsEnabled() || !tweaksAndThings.ServiceFundPenalties()) return true; + + + RepairRateState rateState = __instance.RateState; + StateManager shared = StateManager.Shared; + + if (rateState.PayDue < 1E-06f) + { + rateState.PayDue = 0f; + rateState.PaidCurrent = true; + } + else + { + int num = Mathf.CeilToInt(rateState.PayDue); + rateState.PaidCurrent = shared.CanAfford(num); + int penalty = Mathf.CeilToInt(!rateState.PaidCurrent ? num * 0.2f : 0); + Multiplayer.Broadcast($"{Hyperlink.To(__instance.Industry)}: Paid {num:C0} wages for shop crew${(penalty > 0 ? $"; Overdraft fee of {penalty:C0}" : ".")}"); + + __instance.Industry.ApplyToBalance(-num, Ledger.Category.WagesRepair, null, 0, quiet: true); + if (penalty > 0) + StateManager.Shared.ApplyToBalance(-penalty, Ledger.Category.WagesRepair, null, $"Overdraft: {__instance.Industry.name}", 0, quiet: true); + rateState.PayDue = 0f; + rateState.PaidCurrent = true; + } + __instance.RateState = rateState; + return false; + } +} + +[HarmonyPatch(typeof(RepairTrack))] +[HarmonyPatch(nameof(RepairTrack.EnumerateCarsActual), typeof(IIndustryContext))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class RepairTrack_NeedsRepair_Patch +{ + private static Serilog.ILogger _log => Log.ForContext(); + public static void Postfix(RepairTrack __instance, ref IEnumerable __result) + { + TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; + if (!tweaksAndThings.IsEnabled()) return; + + __result = EnumerateCarsActualPatched(__result); + } + + internal static IEnumerable EnumerateCarsActualPatched(IEnumerable __result, bool usePatchedVersion = true) => + usePatchedVersion ? __result.Where(NeedsRepairPatched) : __result; + + internal static bool NeedsRepairPatched(Car car) + { + bool result = TryGetRepairDestination(car, out var overrideTag); + return result; + } +} + +[HarmonyPatch(typeof(RepairTrack))] +[HarmonyPatch(nameof(RepairTrack.BuildCars), typeof(UIPanelBuilder))] +[HarmonyPatchCategory("RMROC451TweaksAndThings")] +internal class RepairTrack_BuildCars_Patch +{ + public static bool Prefix(RepairTrack __instance, UIPanelBuilder builder) + { + TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; + if (!tweaksAndThings.IsEnabled()) return true; + + BuildCarsPatched(__instance, builder); + + return false; + } + + public static IEnumerable EnumerateCarsActualOrig(RepairTrack __instance, IIndustryContext ctx) + { + TrainController trainController = TrainController.Shared; + return from car in __instance.EnumerateCars(ctx) + select trainController.CarForId(car.Id); + } + + + public static void BuildCarsPatched(RepairTrack __instance, UIPanelBuilder builder) + { + IndustryContext industryContext = __instance.CreateContext(TimeWeather.Now, 0f); + IOrderedEnumerable> carGroups = from @group in EnumerateCarsActualOrig(__instance, industryContext).GroupBy(delegate (Car car) + { + if (InForOverhaul(car)) + { + return RepairGroup.Overhaul; + } + return RepairTrack_NeedsRepair_Patch.NeedsRepairPatched(car) ? RepairGroup.NeedsRepair : RepairGroup.None; + }) + orderby @group.Key + select @group; + builder.VStack(delegate (UIPanelBuilder uIPanelBuilder) + { + float repairRateMultiplier = __instance.RateState.PayRateMultiplier; + foreach (IGrouping item in carGroups) + { + RepairGroup repairGroup = item.Key; + uIPanelBuilder.AddSection(repairGroup switch + { + RepairGroup.Overhaul => "Overhauling", + RepairGroup.NeedsRepair => "Repairing", + RepairGroup.None => "No Work Order Assigned", + _ => throw new ArgumentOutOfRangeException(), + }); + foreach (Car car in item.OrderBy((Car car2) => car2.SortName)) + { + uIPanelBuilder.HStack(delegate (UIPanelBuilder uIPanelBuilder2) + { + uIPanelBuilder2.AddLabel(Hyperlink.To(car)).Width(130f); + uIPanelBuilder2.AddLabel($"{Mathf.RoundToInt(car.Condition * 100f)}%").Width(60f).HorizontalTextAlignment(HorizontalAlignmentOptions.Right); + string text = ""; + string title = ""; + if (repairGroup == RepairGroup.Overhaul) + { + float overhaulProgress = car.OverhaulProgress; + if (overhaulProgress > 0f) + { + text = $"{(int)(overhaulProgress * 100f)}%"; + title = "Overhaul Progress"; + } + } + uIPanelBuilder2.AddLabel(text).Width(60f).HorizontalTextAlignment(HorizontalAlignmentOptions.Right) + .Tooltip(title, null); + if (repairGroup != RepairGroup.None) + { + float num = CalculateRepairWorkOverall(car); + float num2 = num * 12000f * 0.0005f; + string text2 = ((!(repairRateMultiplier > 0f)) ? "Never" : GameDateTimeInterval.DeltaStringMinutes((int)(num / repairRateMultiplier * 60f * 24f), GameDateTimeInterval.Style.Short)); + uIPanelBuilder2.AddLabel(text2).Width(80f).HorizontalTextAlignment(HorizontalAlignmentOptions.Right) + .Tooltip("Time Remaining", "D:HH:MM or H:MM"); + uIPanelBuilder2.AddLabel($"{num2:F1}T").Width(80f).HorizontalTextAlignment(HorizontalAlignmentOptions.Right) + .Tooltip("Repair Parts Needed", null); + } + }); + } + } + }).Padding(new RectOffset(20, 0, 0, 0)); + } +} diff --git a/TweaksAndThings/Patches/StateManager_OnDayDidChange_Patch.cs b/TweaksAndThings/Patches/StateManager_OnDayDidChange_Patch.cs index 1e0ad3f..a760448 100644 --- a/TweaksAndThings/Patches/StateManager_OnDayDidChange_Patch.cs +++ b/TweaksAndThings/Patches/StateManager_OnDayDidChange_Patch.cs @@ -2,8 +2,11 @@ using Game.State; using HarmonyLib; using KeyValue.Runtime; +using Model.Ops; using Network; using Railloader; +using RMROC451.TweaksAndThings.Extensions; +using System.Linq; using UnityEngine; namespace RMROC451.TweaksAndThings.Patches; @@ -18,8 +21,28 @@ internal class StateManager_OnDayDidChange_Patch private static void Postfix(StateManager __instance) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled || !(tweaksAndThings?.settings?.EndGearHelpersRequirePayment ?? false)) return; - if (StateManager.IsHost) PayAutoBrakeCrewWages(__instance); + if (!tweaksAndThings.IsEnabled()) return; + + if (StateManager.IsHost) DoNewDayActivites(tweaksAndThings, __instance); + } + + private static void DoNewDayActivites(TweaksAndThingsPlugin tweaksAndThings, StateManager __instance) + { + if (tweaksAndThings.EndGearHelpersRequirePayment() && tweaksAndThings.DayLoadCrewHours()) LoadCabeese(__instance); + if (tweaksAndThings.EndGearHelpersRequirePayment()) PayAutoBrakeCrewWages(__instance); + } + + private static void LoadCabeese(StateManager __instance) + { + foreach (var car in TrainController.Shared.Cars.Where(Car_Extensions.IsCaboose)) + { + var data = car.QuantityCapacityOfLoad(OpsController_AnnounceCoalescedPayments_Patch.CrewHoursLoad()); + if (data.quantity < data.capacity) + { + Multiplayer.Broadcast($"{Hyperlink.To(car)}: \"Caboose crew topped off.\""); + new OpsCarAdapter(car, OpsController.Shared).Load(OpsController_AnnounceCoalescedPayments_Patch.CrewHoursLoad(), data.capacity - data.quantity); + } + } } private static void PayAutoBrakeCrewWages(StateManager __instance) diff --git a/TweaksAndThings/Patches/TagController_UpdateTag_Patch.cs b/TweaksAndThings/Patches/TagController_UpdateTag_Patch.cs index f4fdd0b..724e5b1 100644 --- a/TweaksAndThings/Patches/TagController_UpdateTag_Patch.cs +++ b/TweaksAndThings/Patches/TagController_UpdateTag_Patch.cs @@ -4,7 +4,7 @@ using Model; using Model.Ops; using Railloader; using RMROC451.TweaksAndThings.Extensions; -using Serilog; +using System; using System.Collections.Generic; using System.Linq; using UI.Tags; @@ -23,28 +23,34 @@ internal class TagController_UpdateTag_Patch { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled || !tweaksAndThings.settings.HandBrakeAndAirTagModifiers) + if (!tweaksAndThings.IsEnabled() || !tweaksAndThings.settings.HandBrakeAndAirTagModifiers) { return; } - ProceedWithPostFix(car, tagCallout, tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter()); + ProceedWithPostFix(car, tagCallout, tweaksAndThings.RequireConsistCabooseForOilerAndHotboxSpotter() || !tweaksAndThings.CabooseRequiredForLocoOilIndicator()); return; } private static void ProceedWithPostFix(Car car, TagCallout tagCallout, bool cabooseRequired) { - tagCallout.callout.Title = string.Format(tagTitleFormat, "{0}", car.DisplayName); + tagCallout.callout.Title = string.Format(tagTitleFormat, "{0}", Hyperlink.To(car)); List tags = new(); + string oilSpriteName = string.Empty;// "OilCan"; if (OpsController_AnnounceCoalescedPayments_Patch.CrewCarStatus(car).spotted) tags.Add("+"); //if (car.EnableOiling) tags.Add(car.HasHotbox ? TextSprites.Hotbox : $"{TextSprites.Warning}{car.Oiled.TriColorPiePercent(1)}"); - if (car.EnableOiling) tags.Add(car.HasHotbox ? TextSprites.Hotbox : car.Oiled.TriColorPiePercent(1)); + if (car.EnableOiling) tags.Add(car.HasHotbox ? TextSprites.Hotbox : car.Oiled.TriColorPiePercent(1, oilSpriteName)); IEnumerable consist = car.EnumerateCoupled().Where(c => c.EnableOiling); - bool cabooseRequirementFulfilled = consist.CabooseInConsist() || !cabooseRequired || consist.ConsistNoFreight(); - if (cabooseRequirementFulfilled && car.IsLocomotive && !car.NeedsOiling && StateManager.Shared.Storage.OilFeature && consist.Any(c => c.NeedsOiling)) - tags.Add(consist.OrderBy(c => c.Oiled).FirstOrDefault().Oiled.TriColorPiePercent(1)); + Func cabooseRequirementFulfilled = () => (!cabooseRequired || consist.ConsistNoFreight() || car.FindMyCaboose(0.0f, false)); + if (StateManager.Shared.Storage.OilFeature + && car.IsLocomotive + && !car.NeedsOiling + && (consist.Any(c => c.NeedsOiling) || consist.Any(c => c.HasHotbox) + && cabooseRequirementFulfilled()) + ) + tags.Add(consist.Any(c => c.HasHotbox) ? TextSprites.Hotbox : consist.OrderBy(c => c.Oiled).FirstOrDefault().Oiled.TriColorPiePercent(1, oilSpriteName)); if (car.EndAirSystemIssue()) tags.Add(TextSprites.CycleWaybills); if (car.HandbrakeApplied()) tags.Add(TextSprites.HandbrakeWheel); diff --git a/TweaksAndThings/Patches/WedgeImage_OnPopulateMesh.cs b/TweaksAndThings/Patches/WedgeImage_OnPopulateMesh.cs index c9904c3..dcc7121 100644 --- a/TweaksAndThings/Patches/WedgeImage_OnPopulateMesh.cs +++ b/TweaksAndThings/Patches/WedgeImage_OnPopulateMesh.cs @@ -13,7 +13,7 @@ internal class WedgeImage_OnPopulateMesh_Patch private static void Postfix(VertexHelper vh) { TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase.Shared; - if (!tweaksAndThings.IsEnabled) return; + if (!tweaksAndThings.IsEnabled()) return; vh.Clear(); //clear the image backgrounds for now. } diff --git a/TweaksAndThings/RMROC451.TweaksAndThings.csproj b/TweaksAndThings/RMROC451.TweaksAndThings.csproj index ca09fd7..d35475b 100644 --- a/TweaksAndThings/RMROC451.TweaksAndThings.csproj +++ b/TweaksAndThings/RMROC451.TweaksAndThings.csproj @@ -32,6 +32,10 @@ + + + + diff --git a/TweaksAndThings/Settings/Settings.cs b/TweaksAndThings/Settings/Settings.cs index 4d51ffa..de4b2a1 100644 --- a/TweaksAndThings/Settings/Settings.cs +++ b/TweaksAndThings/Settings/Settings.cs @@ -5,6 +5,7 @@ using RMROC451.TweaksAndThings.Enums; using UI.Builder; using Model; using RMROC451.TweaksAndThings.Extensions; +using UnityEngine.InputSystem; namespace RMROC451.TweaksAndThings; @@ -23,7 +24,12 @@ public class Settings RosterFuelColumnSettings engineRosterFuelColumnSettings, bool endGearHelpersRequirePayment, bool requireConsistCabooseForOilerAndHotboxSpotter, - bool cabooseAllowsConsistInfo + bool cabooseAllowsConsistInfo, + bool cabooseRequiredForLocoTagOilIndication, + bool servicingFundPenalty, + bool safetyFirst, + CrewHourLoadMethod loadCrewHoursMethod, + float cabeeseSearchRadiusFtInMeters ) { WebhookSettingsList = webhookSettingsList; @@ -32,6 +38,11 @@ public class Settings EndGearHelpersRequirePayment = endGearHelpersRequirePayment; RequireConsistCabooseForOilerAndHotboxSpotter = requireConsistCabooseForOilerAndHotboxSpotter; CabooseAllowsConsistInfo = cabooseAllowsConsistInfo; + CabooseRequiredForLocoTagOilIndication = cabooseRequiredForLocoTagOilIndication; + ServicingFundPenalty = servicingFundPenalty; + SafetyFirst = safetyFirst; + LoadCrewHoursMethod = loadCrewHoursMethod; + CabeeseSearchRadiusFtInMeters = cabeeseSearchRadiusFtInMeters; } public readonly UIState _selectedTabState = new UIState(null); @@ -41,6 +52,11 @@ public class Settings public bool EndGearHelpersRequirePayment; public bool RequireConsistCabooseForOilerAndHotboxSpotter; public bool CabooseAllowsConsistInfo; + public bool CabooseRequiredForLocoTagOilIndication; + public bool ServicingFundPenalty; + public bool SafetyFirst; + public CrewHourLoadMethod LoadCrewHoursMethod; + public float CabeeseSearchRadiusFtInMeters; internal void AddAnotherRow() { @@ -101,6 +117,8 @@ public static class SettingsExtensions return output; } + public static bool IsEnabled(this TweaksAndThingsPlugin input) => + input?.IsEnabled ?? false; public static bool CabooseAllowsConsistInfo(this TweaksAndThingsPlugin input) => input?.settings?.CabooseAllowsConsistInfo ?? false; public static bool EndGearHelpersRequirePayment(this TweaksAndThingsPlugin input) => @@ -108,6 +126,16 @@ public static class SettingsExtensions public static bool RequireConsistCabooseForOilerAndHotboxSpotter(this TweaksAndThingsPlugin input) => input?.settings?.RequireConsistCabooseForOilerAndHotboxSpotter ?? false; public static bool CabooseNonMotiveAllowedSetting(this TweaksAndThingsPlugin input, Car car) => - input.EndGearHelpersRequirePayment() && car.set.Cars.CabooseInConsist() && car.NotMotivePower(); + input.EndGearHelpersRequirePayment() && !car.MotivePower() && (bool)car.FindMyCaboose(0.0f, false); + public static bool CabooseRequiredForLocoOilIndicator(this TweaksAndThingsPlugin input) => + input?.settings?.CabooseRequiredForLocoTagOilIndication ?? false; + public static bool ServiceFundPenalties(this TweaksAndThingsPlugin input) => + input?.settings?.ServicingFundPenalty ?? false; + public static bool SafetyFirst(this TweaksAndThingsPlugin input) => + input?.settings?.SafetyFirst ?? false; + public static bool DayLoadCrewHours(this TweaksAndThingsPlugin input) => + (input?.settings?.LoadCrewHoursMethod ?? CrewHourLoadMethod.Tracks) == CrewHourLoadMethod.Daily; + public static float CabeeseSearchRadiusInMeters(this TweaksAndThingsPlugin input) => + input?.settings?.CabeeseSearchRadiusFtInMeters ?? 21f; } \ No newline at end of file diff --git a/TweaksAndThings/TweaksAndThingsPlugin.cs b/TweaksAndThings/TweaksAndThingsPlugin.cs index 2cadcc9..271ed19 100644 --- a/TweaksAndThings/TweaksAndThingsPlugin.cs +++ b/TweaksAndThings/TweaksAndThingsPlugin.cs @@ -1,17 +1,19 @@ // Ignore Spelling: RMROC using GalaSoft.MvvmLight.Messaging; +using Game.Messages; using Game.State; using HarmonyLib; using Railloader; +using RMROC451.TweaksAndThings.Commands; +using RMROC451.TweaksAndThings.Enums; using Serilog; using System; using System.Linq; using System.Net.Http; using UI.Builder; - -using RMROC451.TweaksAndThings.Enums; -using RMROC451.TweaksAndThings.Commands; +using UnityEngine; +using ILogger = Serilog.ILogger; namespace RMROC451.TweaksAndThings; @@ -33,6 +35,8 @@ public class TweaksAndThingsPlugin : SingletonPluginBase, IModdingContext moddingContext { get; set; } IModDefinition modDefinition { get; set; } + public string ModDirectory => modDefinition.Directory; + static TweaksAndThingsPlugin() { Log.Information("Hello! Static Constructor was called!"); @@ -48,7 +52,7 @@ public class TweaksAndThingsPlugin : SingletonPluginBase, moddingContext.RegisterConsoleCommand(new EchoCommand()); - settings = moddingContext.LoadSettingsData(self.Id); + settings = moddingContext.LoadSettingsData(self.Id) ?? new(); } public override void OnEnable() @@ -82,6 +86,10 @@ public class TweaksAndThingsPlugin : SingletonPluginBase, settings.WebhookSettingsList = settings?.WebhookSettingsList.SanitizeEmptySettings(); + builder.AddSection("Adjustments To Base Game", (UIPanelBuilder builder) => { + builder.AddLabel("Repair tracks now require cars to be waybilled, or they will not be serviced/overhauled.\nThey will report on the company window's location section as 'No Work Order Assigned'."); + }); + builder.Spacer(spacing * spacing); builder.AddTabbedPanels(settings._selectedTabState, delegate (UITabbedPanelBuilder tabBuilder) { tabBuilder.AddTab("Caboose Mods", "cabooseUpdates", CabooseMods); @@ -91,21 +99,37 @@ public class TweaksAndThingsPlugin : SingletonPluginBase, } private static string cabooseUse => "Caboose Use"; - private static string autoAiRequirment => "AutoAI\nRequirement"; + private static string autoAiRequirment => "AutoAI Requirement"; + private static string locoConsistOilIndication => "Consist Oil Indication"; + private static float spacing => 2f; private void CabooseMods(UIPanelBuilder builder) { - builder.AddField( + //UI.GameInput + //builder.AddField("Meow", + // builder.AddInputBindingControl( + + // ) + //) + + // InputAction a = new InputAction("connectCarsAndGladhands", InputActionType.Button) + + //var connectCarsAndGladhands = new InputAction("connectCarsAndGladhands"); + //connectCarsAndGladhands.AddCompositeBinding("connectCarsAndGladhandsComposite") + // .With("modifier", "/leftCtrl") + // .With("modifier", "/leftAlt"); + + //builder.AddField("meow", builder.AddInputBindingControl(connectCarsAndGladhands, conflict: true, ()=>)) + #region EndGearHelperCost + builder.AddFieldToggle( cabooseUse, - builder.AddToggle( - () => settings?.EndGearHelpersRequirePayment ?? false, + () => this.EndGearHelpersRequirePayment(), delegate (bool enabled) { if (settings == null) settings = new(); settings.EndGearHelpersRequirePayment = enabled; builder.Rebuild(); } - ) ).Tooltip("Enable End Gear Helper Cost", @$"Will cost 1 minute of AI Brake Crew & Caboose Crew time per car in the consist when the new inspector buttons are utilized. 1.5x multiplier penalty to AI Brake Crew cost if no sufficiently crewed caboose nearby. @@ -117,26 +141,94 @@ AutoOiler Update: Increases limit that crew will oiling a car from 75% -> 99%, a AutoOiler Update: if `{cabooseUse}` & `{autoAiRequirment.Replace("\n", " ")}` checked, then when a caboose is present, the AutoOiler will repair hotboxes afer oiling them to 100%. AutoHotboxSpotter Update: decrease the random wait from 30 - 300 seconds to 15 - 30 seconds (Safety Is Everyone's Job)"); + #endregion - builder.AddField( + #region CabeeseLoadOptions + if (this.EndGearHelpersRequirePayment()) + { + var columns = Enum.GetValues(typeof(CrewHourLoadMethod)).Cast().Select(i => i.ToString()).ToList(); + builder.Spacer(spacing); + builder.AddField("Refill Option", + builder.AddDropdown( + columns, + (int)(settings?.LoadCrewHoursMethod ?? CrewHourLoadMethod.Tracks), + delegate (int column) + { + if (settings == null) settings = new(); + settings.LoadCrewHoursMethod = (CrewHourLoadMethod)column; + builder.Rebuild(); + } + ) + ).Tooltip("Crew Hours Load Option", "Select whether you want to manually reload cabeese via:\n\ntrack method - (team/repair/passenger stop/interchange)\n\ndaily caboose top off - refill to 8h at new day."); + } + #endregion + + #region RequireCabeeseForOiler/HotboxDetection + builder.Spacer(spacing); + builder.AddFieldToggle( autoAiRequirment, - builder.AddToggle( - () => settings?.RequireConsistCabooseForOilerAndHotboxSpotter ?? false, + () => this.RequireConsistCabooseForOilerAndHotboxSpotter(), delegate (bool enabled) { if (settings == null) settings = new(); settings.RequireConsistCabooseForOilerAndHotboxSpotter = enabled; builder.Rebuild(); } + ).Tooltip("AI Hotbox\\Oiler Requires Caboose", $@"A caboose is required in the consist to check for Hotboxes and perform Auto Oiler, if checked."); + #endregion + + #region ShowLocomotiveConsistOilIndicator + builder.Spacer(spacing); + builder.AddFieldToggle( + locoConsistOilIndication, + () => settings?.CabooseRequiredForLocoTagOilIndication ?? false, + delegate (bool enabled) + { + if (settings == null) settings = new(); + settings.CabooseRequiredForLocoTagOilIndication = enabled; + builder.Rebuild(); + } + ).Tooltip(locoConsistOilIndication, $@"A caboose is required in the consist to report the lowest oil level in the consist in the locomotive's tag & roster entry."); + #endregion + + #region SafetyFirst + builder.Spacer(spacing); + builder.AddFieldToggle( + "Safety First!", + () => settings?.SafetyFirst ?? false, + delegate (bool enabled) + { + if (settings == null) settings = new(); + settings.SafetyFirst = enabled; + builder.Rebuild(); + } + ).Tooltip("Safety First", $@"On non-express timetabled consists, a caboose is required in the consist increase AE max speed > 20 in {Enum.GetName(typeof(AutoEngineerMode), AutoEngineerMode.Road)}/{Enum.GetName(typeof(AutoEngineerMode), AutoEngineerMode.Waypoint)} mode."); + #endregion + + #region CabeeseSearchRadius + builder.Spacer(spacing); + builder.AddField( + "Cabeese Search Radius", + builder.AddSlider( + () => this.CabeeseSearchRadiusInMeters(), + () => $"{string.Format(Mathf.CeilToInt(this.CabeeseSearchRadiusInMeters() * 3.28084f).ToString(), "N0")}ft", + delegate (float input) { + settings = settings ?? new(); + settings.CabeeseSearchRadiusFtInMeters = Mathf.CeilToInt(input); + builder.Rebuild(); + }, + minValue: 1f, + maxValue: Mathf.CeilToInt(5280f / 2f / 3.28084f), + wholeNumbers: true ) - ).Tooltip("AI Engineer Requires Caboose", $@"A caboose is required in the consist to check for Hotboxes and perform Auto Oiler, if checked."); + ).Tooltip("Cabeese Catchment Area", "How far should the cabeese hunting logic look away from the cars in the area to find a caboose?"); + #endregion } private void UiUpdates(UIPanelBuilder builder) { - builder.AddField( + builder.AddFieldToggle( "Enable Tag Updates", - builder.AddToggle( () => settings?.HandBrakeAndAirTagModifiers ?? false, delegate (bool enabled) { @@ -144,12 +236,24 @@ AutoHotboxSpotter Update: decrease the random wait from 30 - 300 seconds to 15 - settings.HandBrakeAndAirTagModifiers = enabled; builder.Rebuild(); } - ) ).Tooltip("Enable Tag Updates", $@"Will suffix tag title with: {TextSprites.CycleWaybills} if Air System issue. {TextSprites.HandbrakeWheel} if there is a handbrake set. {TextSprites.Hotbox} if a hotbox."); + builder.Spacer(spacing); + builder.AddFieldToggle( + "Debt Allowance", + () => settings?.ServicingFundPenalty ?? false, + delegate (bool enabled) + { + if (settings == null) settings = new(); + settings.ServicingFundPenalty = enabled; + builder.Rebuild(); + } + ).Tooltip("Allow Insufficient Funds", $@"Will allow interchange service and repair shops to still function when you are insolvent, at a 20% overdraft fee."); + + builder.Spacer(spacing); EngineRosterShowsFuelStatusUISection(builder); } @@ -158,6 +262,7 @@ AutoHotboxSpotter Update: decrease the random wait from 30 - 300 seconds to 15 - var columns = Enum.GetValues(typeof(EngineRosterFuelDisplayColumn)).Cast().Select(i => i.ToString()).ToList(); builder.AddSection("Fuel Display in Engine Roster", delegate (UIPanelBuilder builder) { + builder.Spacer(spacing); builder.AddField( "Enable", builder.AddDropdown(columns, (int)(settings?.EngineRosterFuelColumnSettings?.EngineRosterFuelStatusColumn ?? EngineRosterFuelDisplayColumn.None), @@ -170,9 +275,9 @@ AutoHotboxSpotter Update: decrease the random wait from 30 - 300 seconds to 15 - ) ).Tooltip("Enable Fuel Display in Engine Roster", $"Will add reaming fuel indication to Engine Roster (with details in roster row tool tip), Examples : {string.Join(" ", Enumerable.Range(0, 4).Select(i => TextSprites.PiePercent(i, 4)))}"); - builder.AddField( + builder.Spacer(spacing); + builder.AddFieldToggle( "Always Visible?", - builder.AddToggle( () => settings?.EngineRosterFuelColumnSettings?.EngineRosterShowsFuelStatusAlways ?? false, delegate (bool enabled) { @@ -180,7 +285,6 @@ AutoHotboxSpotter Update: decrease the random wait from 30 - 300 seconds to 15 - settings.EngineRosterFuelColumnSettings.EngineRosterShowsFuelStatusAlways = enabled; builder.Rebuild(); } - ) ).Tooltip("Fuel Display in Engine Roster Always Visible", $"Always displayed, if you want it hidden and only shown when you care to see, uncheck this, and then you can press ALT for it to populate on the next UI refresh cycle."); }); }