Performance Fish

Source/PerformanceFish/JobSystem/WorkGiver_HaulOptimization.cs

@@ -0,0 +1,29 @@
+// Copyright (c) 2022 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/.
+
+namespace PerformanceFish.JobSystem;
+
+public class WorkGiver_HaulOptimization : ClassWithFishPatches
+{
+    public class PotentialWorkThingsGlobal_Patch : FishPatch
+    {
+        public override string Description => "Sorts haulables by distance before running expensive hauling calculations on them to avoid checks on far away items";
+        public override MethodBase TargetMethodInfo => 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);
+
+			//Log.ErrorOnce(list.Select(t => t.Position.DistanceToSquared(pawn.Position)).ToStringSafeEnumerable(), 84736285);
+        }
+        private static WorkGiver_DoBillOptimization.ThingPositionComparer _comparer = new();
+    }
+}