305 lines
11 KiB
C#
305 lines
11 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 System.Linq;
|
|
using PerformanceFish.Events;
|
|
using PerformanceFish.Hauling;
|
|
using PerformanceFish.Planet;
|
|
using Prepatcher;
|
|
using RimWorld.Planet;
|
|
using PFMapEvents = PerformanceFish.Events.MapEvents;
|
|
|
|
namespace PerformanceFish;
|
|
|
|
public static class PrepatcherFields
|
|
{
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateObject))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern object LockObject(this ListerThings listerThings);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateMapEvents))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern PFMapEvents.Instanced Events(this Map map);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateThingEvents))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ThingEvents.Instanced Events(this Thing thing);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateGeneList))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern List<Gene> GenesToTick(this Pawn_GeneTracker geneTracker);
|
|
|
|
#if V1_6
|
|
[PrepatcherField]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref GeneTrackerOptimization.GeneListVersionState GeneListVersions(this Pawn_GeneTracker geneTracker);
|
|
|
|
[PrepatcherField]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref CompTickOptimization.ThingTickCache ThingTickCache(this ThingWithComps thing);
|
|
#endif
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateWorldPawnsCache))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref WorldPawnsOptimization.Cache Cache(this WorldPawns worldPawns);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateBedHashSet))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern HashSet<Building_Bed> UniqueContainedBeds(this Room room);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateIndexMap))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern FishTable<int, int> IndexMap<T>(this ThingOwner<T> thingOwner) where T : Thing;
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateIndexMap))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern FishTable<int, int> IndexMapByDef(this ListerThings listerThings);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateIndexMapByGroup))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern FishTable<GroupThingPair, int> IndexMapByGroup(this ListerThings listerThings);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateItemCountGrid))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern Cache.CellGrid<int> ItemCountGrid(this Map map);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateBitCellGrid))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern Cache.BitCellGrid StorageBlockerGrid(this Map map);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateStorageSettingsCache))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref StorageSettingsPatches.StorageSettingsCache Cache(this StorageSettings storageSettings);
|
|
|
|
[PrepatcherField]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref int SpawnedMemberCount(this StorageGroup storageGroup);
|
|
|
|
#if V1_4
|
|
[PrepatcherField]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref object? MonitorObject(this GasGrid gasGrid);
|
|
#endif
|
|
|
|
#if !V1_4
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateParallelGasJob))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern GasGridOptimization.ParallelJob ParallelJob(this GasGrid gasGrid);
|
|
#endif
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateParallelGasGridArray))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern GasGridOptimization.ParallelGasGrid[] ParallelGasGrids(this GasGrid gasGrid);
|
|
|
|
[PrepatcherField]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref Texture2D? ExpandingIconCache(this WorldObject worldObject);
|
|
|
|
[PrepatcherField]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref Color? ExpandingIconColorCache(this WorldObject worldObject);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(GetDefaultDistrictArray))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref StorageDistrict[] Districts(this SlotGroup slotGroup);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateStorageDistrictGrid))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern Cache.CellGrid<StorageDistrict?> StorageDistrictGrid(this Map map);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateListerBuildingsCache))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref Listers.Buildings.Cache Cache(this ListerBuildings listerBuildings);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateListerThingsCache))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref Listers.Things.Cache Cache(this ListerThings listerThings);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateListerHaulablesCache))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref Listers.Haulables.Cache Cache(this ListerHaulables listerHaulables);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateHaulDestinationManagerCache))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref HaulDestinationManagerCache Cache(this HaulDestinationManager manager);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateBaseStatsCache))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern float[] BaseStatsCache(this ThingDef thingDef);
|
|
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(GetEmptyArray))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern ref float[] StatsCache(this BuildableDef buildableDef);
|
|
|
|
// [PrepatcherField]
|
|
// [ValueInitializer(nameof(CreateHediffCompList))]
|
|
// [MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
// public static extern List<HediffComp> HediffCompsWithVisible(this HediffWithComps hediff);
|
|
//
|
|
// [PrepatcherField]
|
|
// [ValueInitializer(nameof(CreateHediffCompList))]
|
|
// [MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
// public static extern List<HediffComp> HediffCompsWithShouldRemove(this HediffWithComps hediff);
|
|
//
|
|
// [PrepatcherField]
|
|
// [ValueInitializer(nameof(CreateHediffCompList))]
|
|
// [MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
// public static extern List<HediffComp> HediffCompsWithPostTick(this HediffWithComps hediff);
|
|
|
|
#if false
|
|
[PrepatcherField]
|
|
[ValueInitializer(nameof(CreateHediffList))]
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public static extern List<Hediff> HediffsToTick(this Pawn_HealthTracker healthTracker);
|
|
|
|
public static List<Hediff> CreateHediffList() => new();
|
|
#endif
|
|
|
|
public static List<Gene> CreateGeneList() => [];
|
|
public static WorldPawnsOptimization.Cache CreateWorldPawnsCache() => new();
|
|
public static HashSet<Building_Bed> CreateBedHashSet() => [];
|
|
public static FishTable<int, int> CreateIndexMap() => new();
|
|
public static FishTable<GroupThingPair, int> CreateIndexMapByGroup() => new();
|
|
public static Cache.CellGrid<int> CreateItemCountGrid(Map map) => new(map);
|
|
public static Cache.BitCellGrid CreateBitCellGrid(Map map) => new(map);
|
|
public static PFMapEvents.Instanced CreateMapEvents() => new();
|
|
public static ThingEvents.Instanced CreateThingEvents() => new();
|
|
public static StorageSettingsPatches.StorageSettingsCache CreateStorageSettingsCache() => new();
|
|
public static StorageDistrict[] GetDefaultDistrictArray() => StorageDistrict.GetDefaultArray();
|
|
public static Cache.CellGrid<StorageDistrict> CreateStorageDistrictGrid(Map map) => new(map);
|
|
public static Listers.Buildings.Cache CreateListerBuildingsCache() => new();
|
|
public static Listers.Things.Cache CreateListerThingsCache() => new();
|
|
public static Listers.Haulables.Cache CreateListerHaulablesCache() => new();
|
|
public static HaulDestinationManagerCache CreateHaulDestinationManagerCache() => new();
|
|
public static float[] GetEmptyArray() => Array.Empty<float>();
|
|
public static object CreateObject() => new();
|
|
#if !V1_4
|
|
public static GasGridOptimization.ParallelJob CreateParallelGasJob() => new();
|
|
#endif
|
|
|
|
public static float[] CreateBaseStatsCache()
|
|
{
|
|
var array = new float[Enum.GetNames(typeof(Defs.ThingDefPatches.Stats)).Length];
|
|
StatCaching.ResetToDefaults(array);
|
|
return array;
|
|
}
|
|
|
|
public static object CreateGroupMonitorObject()
|
|
=> ParallelNoAlloc.RegisterBackgroundWaitingWorkers(Array.Empty<Action>());
|
|
|
|
public static GasGridOptimization.ParallelGasGrid[] CreateParallelGasGridArray(GasGrid gasGrid)
|
|
{
|
|
DefDatabase<GasDef>.SetIndices();
|
|
|
|
var map = gasGrid.map;
|
|
var gasDefs = DefDatabase<GasDef>.AllDefsListForReading;
|
|
|
|
if (gasDefs.Count
|
|
#if V1_4
|
|
< 3
|
|
#else
|
|
< 4
|
|
#endif
|
|
)
|
|
{
|
|
Log.Error($"GasDefs are missing! Expected are at least "
|
|
#if V1_4
|
|
+ $"3"
|
|
#else
|
|
+ $"4"
|
|
#endif
|
|
+ $", DefDatabase contains {
|
|
gasDefs.Count} instead. ParallelGasGrid won't work correctly in this state.");
|
|
}
|
|
else
|
|
{
|
|
if (gasDefs[0] != GasDefOf.BlindSmoke
|
|
|| gasDefs[1] != GasDefOf.ToxGas
|
|
|| gasDefs[2] != GasDefOf.RotStink
|
|
#if !V1_4
|
|
|| gasDefs[3] != GasDefOf.DeadLifeDust
|
|
#endif
|
|
)
|
|
{
|
|
Log.Warning("GasDefs out of order. Sorting now to fix.");
|
|
var allGasDefs = gasDefs.ToList();
|
|
|
|
allGasDefs.Remove(gasDefs[0] = GasDefOf.BlindSmoke);
|
|
allGasDefs.Remove(gasDefs[1] = GasDefOf.ToxGas);
|
|
allGasDefs.Remove(gasDefs[2] = GasDefOf.RotStink);
|
|
#if !V1_4
|
|
allGasDefs.Remove(gasDefs[3] = GasDefOf.DeadLifeDust);
|
|
#endif
|
|
|
|
for (var i = 0; i < allGasDefs.Count; i++)
|
|
{
|
|
gasDefs[i
|
|
#if V1_4
|
|
+ 3
|
|
#else
|
|
+ 4
|
|
#endif
|
|
] = allGasDefs[i];
|
|
}
|
|
|
|
DefDatabase<GasDef>.SetIndices();
|
|
}
|
|
}
|
|
|
|
var grids = new GasGridOptimization.ParallelGasGrid[gasDefs.Count];
|
|
for (var i = 0; i < grids.Length; i++)
|
|
grids[i] = new(map, gasDefs[i]);
|
|
|
|
#if V1_4
|
|
if (FishSettings.ThreadingEnabled)
|
|
{
|
|
var monitorObject = gasGrid.MonitorObject() ??= CreateGroupMonitorObject();
|
|
|
|
for (var i = 1; i < grids.Length; i++)
|
|
{
|
|
var grid = grids[i];
|
|
ParallelNoAlloc.RegisterBackgroundWaitingWorker(monitorObject, () => grid.Tick());
|
|
}
|
|
}
|
|
#else
|
|
var parallelJob = gasGrid.ParallelJob();
|
|
if (parallelJob != null!)
|
|
{
|
|
parallelJob.GasGrids = grids;
|
|
}
|
|
else
|
|
{
|
|
Log.Error(
|
|
"Failed to initialize GasGrid parallel job. Map components may have loaded in an incorrect order.");
|
|
}
|
|
#endif
|
|
return grids;
|
|
}
|
|
|
|
// public static List<HediffComp> CreateHediffCompList() => new();
|
|
}
|