Performance Fish

2022-10-24 18:24:02 +02:00
parent 594b32b4ef
commit 2efa8e466c
69 changed files with 9147 additions and 3 deletions
@@ -0,0 +1,100 @@
+// 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 GearMassCache = PerformanceFish.Cache.ByIndex<Verse.Pawn, PerformanceFish.MassUtilityCaching.GearMassCacheValue>;
+using InventoryMassCache = PerformanceFish.Cache.ByIndex<Verse.Pawn, PerformanceFish.MassUtilityCaching.InventoryMassCacheValue>;
+
+namespace PerformanceFish;
+
+public class MassUtilityCaching : ClassWithFishPatches
+{
+	public class GearMass_Patch : FirstPriorityFishPatch
+	{
+		public override string Description => "Caches pawn gear mass. Hauling jobs check this quite a lot";
+		public override Delegate TargetMethodGroup => MassUtility.GearMass;
+
+		[MethodImpl(MethodImplOptions.AggressiveInlining)]
+		public static bool Prefix(Pawn p, ref float __result, out bool __state)
+		{
+			ref var cache = ref GearMassCache.Get.GetReference(p);
+			if (cache.ShouldRefreshNow(p))
+				return __state = true;
+
+			__result = cache.mass;
+			return __state = false;
+		}
+
+		[MethodImpl(MethodImplOptions.AggressiveInlining)]
+		public static void Postfix(Pawn p, bool __state, float __result)
+		{
+			if (!__state)
+				return;
+
+			GearMassCache.Get[p] = new(__result, p);
+		}
+	}
+
+	public class InventoryMass_Patch : FirstPriorityFishPatch
+	{
+		public override string Description => "Caches pawn inventory mass. Hauling jobs check this quite a lot";
+		public override Delegate TargetMethodGroup => MassUtility.InventoryMass;
+
+		[MethodImpl(MethodImplOptions.AggressiveInlining)]
+		public static bool Prefix(Pawn p, ref float __result, out bool __state)
+		{
+			ref var cache = ref InventoryMassCache.Get.GetReference(p);
+			if (cache.ShouldRefreshNow(p))
+				return __state = true;
+
+			__result = cache.mass;
+			return __state = false;
+		}
+
+		[MethodImpl(MethodImplOptions.AggressiveInlining)]
+		public static void Postfix(Pawn p, bool __state, float __result)
+		{
+			if (!__state)
+				return;
+
+			InventoryMassCache.Get[p] = new(__result, p);
+		}
+	}
+
+	public struct GearMassCacheValue
+	{
+		public float mass;
+		private int _nextRefreshTick;
+		private int _equipmentListsState;
+		public GearMassCacheValue(float mass, Pawn p)
+		{
+			this.mass = mass;
+			_nextRefreshTick = Current.Game.tickManager.TicksGame + 3072 + Math.Abs(Rand.Int % 2048);
+			_equipmentListsState = GetEquipmentListsState(p);
+		}
+		private static int GetEquipmentListsState(Pawn p)
+		{
+			var result = p.apparel != null ? p.apparel.WornApparel.Version() : 0;
+			if (p.equipment != null)
+				result += p.equipment.AllEquipmentListForReading.Version();
+			return result;
+		}
+		public bool ShouldRefreshNow(Pawn p) => _nextRefreshTick < Current.Game.tickManager.TicksGame || _equipmentListsState != GetEquipmentListsState(p);
+	}
+
+	public struct InventoryMassCacheValue
+	{
+		public float mass;
+		private int _nextRefreshTick;
+		private int _inventoryListsState;
+		public InventoryMassCacheValue(float mass, Pawn p)
+		{
+			this.mass = mass;
+			_nextRefreshTick = Current.Game.tickManager.TicksGame + 256 + Math.Abs(Rand.Int % 256);
+			_inventoryListsState = GetInventoryListsState(p); //shorter refresh interval because of stack sizes not being considered by the listVersion check
+		}
+		private static int GetInventoryListsState(Pawn p) => p.inventory?.innerContainer is { } container ? container.innerList.Version() : 0;
+		public bool ShouldRefreshNow(Pawn p) => _nextRefreshTick < Current.Game.tickManager.TicksGame || _inventoryListsState != GetInventoryListsState(p);
+	}
+}