bbradson
38 lines
1.3 KiB
C#
38 lines
1.3 KiB
C#
// Copyright (c) 2023 bradson
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
using PerformanceFish.ModCompatibility;
|
|
|
|
namespace PerformanceFish.JobSystem;
|
|
|
|
public sealed class WorkGiver_HaulOptimization : ClassWithFishPatches
|
|
{
|
|
public sealed class PotentialWorkThingsGlobal_Patch : FishPatch
|
|
{
|
|
public override List<string> IncompatibleModIDs { get; } = [PackageIDs.MULTIPLAYER];
|
|
|
|
public override string Description { get; }
|
|
= "Sorts haulables by distance before running expensive hauling calculations on them to avoid checks on "
|
|
+ "far away items";
|
|
|
|
public override MethodBase TargetMethodInfo { get; }
|
|
= AccessTools.Method(typeof(WorkGiver_Haul), nameof(WorkGiver_Haul.PotentialWorkThingsGlobal));
|
|
|
|
public static IEnumerable<Thing> Postfix(IEnumerable<Thing> __result, Pawn? pawn)
|
|
{
|
|
if (pawn != null && __result is List<Thing> list)
|
|
Sort(list, pawn);
|
|
return __result;
|
|
}
|
|
|
|
private static void Sort(List<Thing> list, Pawn pawn)
|
|
{
|
|
_comparer.rootCell = pawn.Position;
|
|
list.Sort(_comparer);
|
|
}
|
|
|
|
private static WorkGiver_DoBillOptimization.ThingPositionComparer _comparer = new();
|
|
}
|
|
} |