bbradson
35 lines
1.1 KiB
C#
35 lines
1.1 KiB
C#
// 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/.
|
|
|
|
using Verse.AI;
|
|
|
|
namespace PerformanceFish.JobSystem;
|
|
|
|
public class WorkGiver_MergeOptimization : ClassWithFishPatches
|
|
{
|
|
public class JobOnThing_Patch : FishPatch
|
|
{
|
|
public override string Description => "Removes mergeables from ListerMergeables if they no longer qualify as mergeable when trying to check for a merging job on them. Vanilla RimWorld just fails and keeps them in";
|
|
public override MethodBase TargetMethodInfo => AccessTools.Method(typeof(WorkGiver_Merge), nameof(WorkGiver_Merge.JobOnThing));
|
|
|
|
public static bool Prefix(Thing t, ref Job? __result)
|
|
{
|
|
if (t.GetSlotGroup() is null)
|
|
{
|
|
RemoveFromListerMergeables(t);
|
|
__result = null;
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
private static void RemoveFromListerMergeables(Thing t)
|
|
{
|
|
var map = t.MapHeld;
|
|
map.listerMergeables.TryRemove(t);
|
|
map.listerHaulables.Check(t);
|
|
}
|
|
}
|
|
} |