mirror of
https://github.com/rmroc451/TweaksAndThings.git
synced 2025-12-16 17:29:37 -06:00
225 lines
10 KiB
C#
225 lines
10 KiB
C#
using Game.Messages;
|
|
using Game.State;
|
|
using HarmonyLib;
|
|
using KeyValue.Runtime;
|
|
using Model;
|
|
using Model.OpsNew;
|
|
using Network;
|
|
using Railloader;
|
|
using RMROC451.TweaksAndThings.Enums;
|
|
using RMROC451.TweaksAndThings.Extensions;
|
|
using RollingStock;
|
|
using Serilog;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using UI.Builder;
|
|
using UI.CarInspector;
|
|
using UI.ContextMenu;
|
|
using UI.Tags;
|
|
using static Model.Car;
|
|
|
|
namespace RMROC451.TweaksAndThings.Patches;
|
|
|
|
[HarmonyPatch(typeof(CarInspector))]
|
|
[HarmonyPatch(nameof(CarInspector.PopulateCarPanel), typeof(UIPanelBuilder))]
|
|
[HarmonyPatchCategory("RMROC451TweaksAndThings")]
|
|
internal class CarInspector_PopulateCarPanel_Patch
|
|
{
|
|
private static ILogger _log => Log.ForContext<CarInspector_PopulateCarPanel_Patch>();
|
|
private static IEnumerable<LogicalEnd> ends = Enum.GetValues(typeof(LogicalEnd)).Cast<LogicalEnd>();
|
|
|
|
/// <summary>
|
|
/// If a caboose inspector is opened, it will auto set Anglecocks, gladhands and hand brakes
|
|
/// </summary>
|
|
/// <param name="__instance"></param>
|
|
/// <param name="builder"></param>
|
|
/// <returns></returns>
|
|
private static bool Prefix(CarInspector __instance, UIPanelBuilder builder)
|
|
{
|
|
|
|
TweaksAndThingsPlugin tweaksAndThings = SingletonPluginBase<TweaksAndThingsPlugin>.Shared;
|
|
if (!tweaksAndThings.IsEnabled) return true;
|
|
bool buttonsHaveCost = tweaksAndThings?.settings?.EndGearHelpersRequirePayment ?? false;
|
|
|
|
var consist = __instance._car.EnumerateCoupled(LogicalEnd.A);
|
|
builder = AddCarConsistRebuildObservers(builder, consist);
|
|
|
|
builder.HStack(delegate (UIPanelBuilder hstack)
|
|
{
|
|
var buttonName = $"{(consist.Any(c => c.HandbrakeApplied()) ? "Release " : "Set ")} {TextSprites.HandbrakeWheel}";
|
|
hstack.AddButtonCompact(buttonName, delegate {
|
|
MrocConsistHelper(__instance._car, MrocHelperType.Handbrake, buttonsHaveCost);
|
|
hstack.Rebuild();
|
|
}).Tooltip(buttonName, $"Iterates over cars in this consist and {(consist.Any(c => c.HandbrakeApplied()) ? "releases" : "sets")} {TextSprites.HandbrakeWheel}.");
|
|
|
|
if (consist.Any(c => c.EndAirSystemIssue()))
|
|
{
|
|
hstack.AddButtonCompact("Connect Air", delegate {
|
|
MrocConsistHelper(__instance._car, MrocHelperType.GladhandAndAnglecock, buttonsHaveCost);
|
|
hstack.Rebuild();
|
|
}).Tooltip("Connect Consist Air", "Iterates over each car in this consist and connects gladhands and opens anglecocks.");
|
|
}
|
|
|
|
hstack.AddButtonCompact("Bleed Consist", delegate {
|
|
MrocConsistHelper(__instance._car, MrocHelperType.BleedAirSystem, buttonsHaveCost);
|
|
hstack.Rebuild();
|
|
}).Tooltip("Bleed Air Lines", "Iterates over each car in this consist and bleeds the air out of the lines.");
|
|
});
|
|
|
|
return true;
|
|
}
|
|
|
|
private static UIPanelBuilder AddCarConsistRebuildObservers(UIPanelBuilder builder, IEnumerable<Model.Car> consist)
|
|
{
|
|
TagController tagController = UnityEngine.Object.FindFirstObjectByType<TagController>();
|
|
foreach (Model.Car car in consist.Where(c => c.Archetype != Model.Definition.CarArchetype.Tender))
|
|
{
|
|
builder = AddObserver(builder, car, PropertyChange.KeyForControl(PropertyChange.Control.Handbrake), tagController);
|
|
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);
|
|
}
|
|
}
|
|
|
|
return builder;
|
|
}
|
|
|
|
private static UIPanelBuilder AddObserver(UIPanelBuilder builder, Model.Car car, string key, TagController tagController)
|
|
{
|
|
builder.AddObserver(
|
|
car.KeyValueObject.Observe(
|
|
key,
|
|
delegate (Value value)
|
|
{
|
|
try
|
|
{
|
|
builder.Rebuild();
|
|
if (car.TagCallout != null) tagController.UpdateTags(CameraSelector.shared._currentCamera.GroundPosition, true); //tagController.UpdateTag(car, car.TagCallout, OpsController.Shared);
|
|
if (ContextMenu.IsShown && ContextMenu.Shared.centerLabel.text == car.DisplayName) CarPickable.HandleShowContextMenu(car);
|
|
}
|
|
catch(Exception ex)
|
|
{
|
|
_log.ForContext("car", car).Warning(ex, $"{nameof(AddObserver)} {car} Exception logged for {key}");
|
|
}
|
|
},
|
|
false
|
|
)
|
|
);
|
|
|
|
return builder;
|
|
}
|
|
|
|
//var dh = new DownloadHandlerAudioClip($"file://{cacheFileName}", AudioType.MPEG);
|
|
//dh.compressed = true; // This
|
|
|
|
//using (UnityWebRequest wr = new UnityWebRequest($"file://{cacheFileName}", "GET", dh, null)) {
|
|
// yield return wr.SendWebRequest();
|
|
// if (wr.responseCode == 200) {
|
|
// audioSource.clip = dh.audioClip;
|
|
// }
|
|
//}
|
|
|
|
public static void MrocConsistHelper(Model.Car car, MrocHelperType mrocHelperType, bool buttonsHaveCost)
|
|
{
|
|
TrainController tc = UnityEngine.Object.FindObjectOfType<TrainController>();
|
|
IEnumerable<Model.Car> consist = car.EnumerateCoupled(LogicalEnd.A);
|
|
_log.ForContext("car", car).Verbose($"{car} => {mrocHelperType} => {string.Join("/", consist.Select(c => c.ToString()))}");
|
|
|
|
CalculateCostIfEnabled(car, mrocHelperType, buttonsHaveCost, consist);
|
|
|
|
switch (mrocHelperType)
|
|
{
|
|
case MrocHelperType.Handbrake:
|
|
if (consist.Any(c => c.HandbrakeApplied()))
|
|
{
|
|
consist.Do(c => c.SetHandbrake(false));
|
|
}
|
|
else
|
|
{
|
|
consist = consist.Where(c => c is not BaseLocomotive && c.Archetype != Model.Definition.CarArchetype.Tender);
|
|
//when ApplyHandbrakesAsNeeded is called, and the consist contains an engine, it stops applying brakes.
|
|
tc.ApplyHandbrakesAsNeeded(consist.ToList(), PlaceTrainHandbrakes.Automatic);
|
|
}
|
|
break;
|
|
|
|
case MrocHelperType.GladhandAndAnglecock:
|
|
consist.Do(c =>
|
|
ends.Do(end =>
|
|
{
|
|
EndGear endGear = c[end];
|
|
|
|
StateManager.ApplyLocal(
|
|
new PropertyChange(
|
|
c.id,
|
|
KeyValueKeyFor(EndGearStateKey.Anglecock, c.LogicalToEnd(end)),
|
|
new FloatPropertyValue(endGear.IsCoupled ? 1f : 0f)
|
|
)
|
|
);
|
|
|
|
if (c.TryGetAdjacentCar(end, out Model.Car c2)) StateManager.ApplyLocal(new SetGladhandsConnected(c.id, c2.id, true));
|
|
})
|
|
);
|
|
break;
|
|
|
|
case MrocHelperType.BleedAirSystem:
|
|
consist = consist.Where(c => c.NotMotivePower());
|
|
_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;
|
|
}
|
|
}
|
|
|
|
private static void CalculateCostIfEnabled(Car car, MrocHelperType mrocHelperType, bool buttonsHaveCost, IEnumerable<Car> consist)
|
|
{
|
|
if (buttonsHaveCost)
|
|
{
|
|
float originalTimeCost = consist.CalculateCostForAutoEngineerEndGearSetting();
|
|
float timeCost = originalTimeCost;
|
|
float crewCost = timeCost / 3600; //hours of time deducted from caboose.
|
|
var tsString = crewCost.FormatCrewHours(IndustryComponent_Service_Patch.CrewHoursLoad().description);
|
|
Car? cabooseWithAvailCrew = NearbyCabooseWithAvailableCrew(car, crewCost, buttonsHaveCost);
|
|
if (cabooseWithAvailCrew == null) timeCost *= 1.5f;
|
|
var cabooseFoundDisplay = cabooseWithAvailCrew?.DisplayName ?? "No caboose";
|
|
|
|
_log.ForContext("car", car).Information($"{nameof(MrocConsistHelper)} {mrocHelperType} : [VACINITY CABEESE FOUND:{cabooseWithAvailCrew?.ToString() ?? "NONE"}] => Consist Length {consist.Count()} => costs {timeCost / 60} minutes of AI Engineer time, $5 per hour = ~${Math.Ceiling((decimal)(timeCost / 3600) * 5)} (*2 if no caboose nearby)");
|
|
|
|
|
|
Multiplayer.SendError(StateManager.Shared._playersManager.LocalPlayer, $"{(cabooseWithAvailCrew != null ? $"{cabooseWithAvailCrew.DisplayName} Hours Adjusted: ({tsString})\n" : string.Empty)}Wages: ~(${Math.Ceiling((decimal)(timeCost / 3600) * 5)})");
|
|
|
|
StateManager_OnDayDidChange_Patch.UnbilledAutoBrakeCrewRunDuration += timeCost;
|
|
}
|
|
}
|
|
|
|
public static Car? NearbyCabooseWithAvailableCrew(Car car, float timeNeeded, bool decrement = false)
|
|
{
|
|
HashSet<string> 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<Car> consist = car.EnumerateCoupled(LogicalEnd.A);
|
|
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<TrainController>();
|
|
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;
|
|
}
|
|
}
|