bbradson
59 lines
1.7 KiB
C#
59 lines
1.7 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 Mono.Cecil;
|
|
using Mono.Cecil.Cil;
|
|
using PerformanceFish.Prepatching;
|
|
|
|
namespace PerformanceFish;
|
|
|
|
public sealed class IdeoBuildingPresenceDemandOptimization : ClassWithFishPrepatches
|
|
{
|
|
public sealed class BuildingPresentPatch : FishPrepatch
|
|
{
|
|
public override string? Description { get; }
|
|
= "Optimization to use cached ListerBuildings data instead of scanning through all colonist buildings, "
|
|
+ "frequently";
|
|
|
|
public override MethodBase TargetMethodBase { get; }
|
|
= AccessTools.DeclaredMethod(typeof(IdeoBuildingPresenceDemand),
|
|
nameof(IdeoBuildingPresenceDemand.BuildingPresent));
|
|
|
|
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
|
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
|
|
|
public static bool ReplacementBody(IdeoBuildingPresenceDemand __instance, Map map)
|
|
{
|
|
var buildings = map.listerBuildings.AllBuildingsColonistOfDef(__instance.parent.ThingDef);
|
|
|
|
if (buildings is IndexedFishSet<Building> fishSet)
|
|
{
|
|
var list = fishSet.ReadOnlyList;
|
|
for (var i = list.Count; i-- > 0;)
|
|
{
|
|
if (list[i].StyleSourcePrecept == __instance.parent)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
return FallbackCheck(__instance, buildings);
|
|
}
|
|
}
|
|
|
|
public static bool FallbackCheck(IdeoBuildingPresenceDemand __instance, IEnumerable<Building> buildings)
|
|
{
|
|
foreach (var building in buildings)
|
|
{
|
|
if (building.StyleSourcePrecept == __instance.parent)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
} |