Performance-Fish/Source/PerformanceFish/FunctionPointers/IndexGetter.cs

// 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 System.Linq;
using JetBrains.Annotations;
using RimWorld.Planet;

namespace PerformanceFish.FunctionPointers;

public static class IndexGetter<T> where T : notnull
{
	public static readonly unsafe delegate*<T, int> Default = IndexGetter.GetDefaultIndexFunctionPointer<T>();
}

public static class IndexGetter
{
	[UsedImplicitly(ImplicitUseTargetFlags.Members)]
	public static class Methods
	{
		public static int HediffSet(HediffSet set) => set.pawn.thingIDNumber;
		public static int HediffDef(HediffDef def) => def.shortHash;
		public static int Bill(Bill bill) => bill.loadID;
		public static int Pawn(Pawn pawn) => pawn.thingIDNumber;
		public static int Thing(Thing thing) => thing.thingIDNumber;
		public static int CompAssignableToPawn_Bed(CompAssignableToPawn_Bed comp) => comp.parent.thingIDNumber;
		public static int Map(Map map) => map.uniqueID;
		public static int DesignationManager(DesignationManager manager) => manager.map.uniqueID;
		public static int DesignationDef(DesignationDef def) => def.shortHash;
		public static int Room(Room room) => room.ID;
		public static int Ideo(Ideo ideo) => ideo.id;
		public static int RecipeDef(RecipeDef def) => def.shortHash;
		public static int RuntimeTypeHandle(RuntimeTypeHandle handle) => (int)handle.Value;
		public static int WorldObject(WorldObject worldObject) => worldObject.ID;
		public static int Hediff(Hediff hediff) => hediff.loadID;
	}

	internal static Cache.IndexGetter<T> GetDefaultIndexGetter<T>() where T : notnull
		=> _indexFuncs.TryGetValue(typeof(T), out var value)
			? (Cache.IndexGetter<T>)value
			: Fallback<T>();

	internal static unsafe delegate*<T, int> GetDefaultIndexFunctionPointer<T>() where T : notnull
		=> (delegate*<T, int>)(_getterMethods.TryGetValue(typeof(T))
			?? Fallback<T>().Method).MethodHandle.GetFunctionPointer();

	private static IEnumerable<MethodInfo> GetGetterMethods()
		=> typeof(Methods).GetMethods(BindingFlags.Static | BindingFlags.Public | BindingFlags.DeclaredOnly)
			.Where(static m => m.ReturnType == typeof(int) && m.GetParameters().Length == 1);

	private static Cache.IndexGetter<T> Fallback<T>() where T : notnull
	{
		var isDef = typeof(Def).IsAssignableFrom(typeof(T));
		var isThing = false; //typeof(Thing).IsAssignableFrom(typeof(T)) && !typeof(Mote).IsAssignableFrom(typeof(T));
		//// generally always safe in vanilla, but logging mods won't hurt either

		var message
			= $"Performance Fish failed to find type {
				typeof(T).FullName} in indexFuncs dictionary. Using fallback instead";

		if (isDef)
			Log.Error($"{message} to prevent further issues, but this likely won't work correctly.");
		else if (!isThing)
			Log.Warning($"{message}.");

		var type = typeof(T).BaseType;
		Delegate func;
		while (!_indexFuncs.TryGetValue(type, out func!))
		{
			type = type.BaseType
				?? throw new($"Performance Fish failed to find a fallback method to handle type {
					typeof(T).FullName} from mod {typeof(T).Assembly.GetName().Name}");
		}

		if (!isThing)
		{
			Log.Message($"Found fallback of type {type.FullName}.{
				(!isDef ? " This likely means we can safely continue from here." : "")}");
		}

		return (Cache.IndexGetter<T>)func;
	}

	private static Dictionary<Type, MethodInfo> _getterMethods
		= GetGetterMethods().ToDictionary(static m => m.GetParameters()[0].ParameterType, static m => m);

	private static Dictionary<Type, Delegate> _indexFuncs
		= _getterMethods.ToDictionary(static m => m.Key,
			static m => m.Value.CreateDelegate(typeof(Cache.IndexGetter<>).MakeGenericType(m.Key)));
}