Added gas grid optimization, various despawn optimizations, new hauling optimizations utilizing events and regions, bugfixes for old patches and for vanilla functions too, as well as smaller additions. 1.3 files removed.
This commit is contained in:
bbradson
@@ -0,0 +1,995 @@
|
||||
// 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/.
|
||||
|
||||
// #define GAS_DEBUG_L1 // erroneous gas changes
|
||||
// #define GAS_DEBUG_L2 // total gas grid changes
|
||||
// #define GAS_DEBUG_L3 // individual cell changes
|
||||
|
||||
using System.Diagnostics;
|
||||
using System.Linq;
|
||||
using Mono.Cecil;
|
||||
using Mono.Cecil.Cil;
|
||||
using MonoMod.Utils;
|
||||
using PerformanceFish.Prepatching;
|
||||
|
||||
namespace PerformanceFish;
|
||||
|
||||
public sealed class GasGridOptimization : ClassWithFishPrepatches
|
||||
{
|
||||
// public static readonly List<ParallelGasGrid> ParallelGasGrids = new();
|
||||
// public static GasGrid[] RegisteredGasGrids = Array.Empty<GasGrid>();
|
||||
|
||||
public sealed class SetDirectPatch : FishPrepatch
|
||||
{
|
||||
public override bool ShowSettings => false;
|
||||
|
||||
public override string? Description { get; }
|
||||
= "Required by the gas grid optimization";
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.SetDirect));
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
||||
|
||||
public static void ReplacementBody(GasGrid __instance, int index, byte smoke, byte toxic, byte rotStink)
|
||||
{
|
||||
if (!ModsConfig.BiotechActive)
|
||||
toxic = 0;
|
||||
|
||||
var gasGrids = __instance.ParallelGasGrids();
|
||||
|
||||
gasGrids[0].SetDirect(index, smoke);
|
||||
gasGrids[1].SetDirect(index, toxic);
|
||||
gasGrids[2].SetDirect(index, rotStink);
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class TickPatch : FishPrepatch
|
||||
{
|
||||
public override List<Type> LinkedPatches { get; } =
|
||||
[
|
||||
typeof(SetDirectPatch), typeof(ColorAtPatch), typeof(AddGasPatch), typeof(AnyGasAtPatch),
|
||||
typeof(DensityAtPatch), typeof(Debug_ClearAllPatch), typeof(Debug_FillAllPatch),
|
||||
typeof(Notify_ThingSpawnedPatch), typeof(ExposeDataPatch), typeof(MouseOverReadOutOnGUI),
|
||||
typeof(DebugToolsGeneralPushGas)
|
||||
];
|
||||
|
||||
// public static EventWaitHandle? WaitHandle { get; private set; }
|
||||
|
||||
public static object? MonitorObject { get; private set; }
|
||||
|
||||
public override string? Description { get; }
|
||||
= "Optimizes the gas grid to utilize highly performant bit manipulation for much faster processing of "
|
||||
+ "cells, usually in buckets of 64 simultaneously checked cells, and threading to process the different "
|
||||
+ "gas types in parallel. Additionally exposes gas to xml through the PerformanceFish.GasDef type. "
|
||||
+ "Performance impact scales with map size, but is generally large. Currently only utilizing single core "
|
||||
+ "parallelism, threading is planned";
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.Tick));
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
||||
|
||||
public static void ReplacementBody(GasGrid __instance)
|
||||
{
|
||||
// Parallel.For(0, ParallelGasGrids.Length, static i => ParallelGasGrids[i].Tick());
|
||||
// (WaitHandle ?? Setup()).Set();
|
||||
|
||||
// if (!RegisteredGasGrids.Contains(__instance))
|
||||
// Setup(__instance);
|
||||
|
||||
// ParallelNoAlloc.PulseRegisteredBackgroundWorkers(__instance.MonitorObject());
|
||||
|
||||
var gasGrids = __instance.ParallelGasGrids();
|
||||
|
||||
for (var i = 1; i < gasGrids.Length; i++)
|
||||
gasGrids[i].Tick();
|
||||
|
||||
if (gasGrids.Length < 1)
|
||||
return;
|
||||
|
||||
var firstGasGrid = gasGrids[0];
|
||||
firstGasGrid.Tick();
|
||||
|
||||
__instance.cycleIndexDiffusion = firstGasGrid.CycleIndexDiffusion;
|
||||
__instance.cycleIndexDissipation = firstGasGrid.CycleIndexDissipation;
|
||||
}
|
||||
|
||||
// [MemberNotNull(nameof(WaitHandle))]
|
||||
// public static EventWaitHandle Setup()
|
||||
// {
|
||||
// WaitHandle = new ManualResetEvent(false);
|
||||
// for (var i = ParallelGasGrids.Length; i-- > 1;)
|
||||
// {
|
||||
// var gasGrid = ParallelGasGrids[i];
|
||||
// ThreadPool.RegisterWaitForSingleObject(WaitHandle, (_, _) => gasGrid.Tick(), null, -1, false);
|
||||
// }
|
||||
//
|
||||
// return WaitHandle;
|
||||
// }
|
||||
|
||||
// public static ParallelGasGrid[] Setup(GasGrid instance)
|
||||
// {
|
||||
// // lock (ParallelGasGrids)
|
||||
// // {
|
||||
// // if (RegisteredGasGrids.Contains(instance))
|
||||
// // {
|
||||
// // // only possible if 2 threads try registering the same gas grid instance. Rimthreaded? Maybe throw?
|
||||
// // SpinWait.SpinUntil(static () => MonitorObject != null);
|
||||
// // #pragma warning disable CS8774
|
||||
// // return;
|
||||
// // #pragma warning restore CS8774
|
||||
// // }
|
||||
// //
|
||||
// // RegisteredGasGrids = RegisteredGasGrids.Add(instance);
|
||||
// // }
|
||||
//
|
||||
// return new ParallelGasGrid[DefDatabase<GasDef>.DefCount];
|
||||
//
|
||||
// var gasDefsList = DefDatabase<GasDef>.AllDefsListForReading;
|
||||
//
|
||||
// var newGasGrids = new ParallelGasGrid[gasDefsList.Count];
|
||||
// try
|
||||
// {
|
||||
// for (var i = 0; i < newGasGrids.Length; i++)
|
||||
// newGasGrids[i] = new(instance.map, gasDefsList[i]);
|
||||
//
|
||||
// // ParallelGasGrids.AddRangeFast(newGasGrids);
|
||||
//
|
||||
// instance.MonitorObject() = ParallelNoAlloc.RegisterBackgroundWaitingWorkers(MakeTickActions(newGasGrids,
|
||||
// Math.Max(newGasGrids.Length - 1, 0), 1)); // 1st grid ticked on main thread
|
||||
// }
|
||||
// catch (Exception e)
|
||||
// {
|
||||
// Log.Error($"Caught exception in map with ID {instance.map.uniqueID} and hashCode {instance.map.GetHashCode()}: {e}");
|
||||
// return Array.Empty<ParallelGasGrid>();
|
||||
// }
|
||||
//
|
||||
// return newGasGrids;
|
||||
// }
|
||||
|
||||
private static Action[] MakeTickActions<T>(T gasGrids, int count, int startIndex)
|
||||
where T : IList<ParallelGasGrid>
|
||||
{
|
||||
var actions = new Action[count];
|
||||
count += startIndex;
|
||||
for (var i = startIndex; i < count; i++)
|
||||
{
|
||||
var gasGrid = gasGrids[i];
|
||||
actions[i - startIndex] = () => gasGrid.Tick();
|
||||
}
|
||||
|
||||
return actions;
|
||||
}
|
||||
|
||||
// for save-game compatibility. The two cycle indices get scribed with ExposeData
|
||||
public static void AdvanceCycleIndex(ref int cycleIndex, int area, int increment)
|
||||
{
|
||||
cycleIndex += increment;
|
||||
if (cycleIndex >= area)
|
||||
cycleIndex = 0;
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class ColorAtPatch : FishPrepatch
|
||||
{
|
||||
public override bool ShowSettings => false;
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.ColorAt));
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
||||
|
||||
public static Color ReplacementBody(GasGrid __instance, IntVec3 cell)
|
||||
{
|
||||
var cellIndex = cell.CellToIndex(__instance.map);
|
||||
|
||||
var gasGrids = __instance.ParallelGasGrids();
|
||||
var gasGridCount = gasGrids.Length;
|
||||
|
||||
Span<float> densities = stackalloc float[gasGridCount];
|
||||
var sum = 0f;
|
||||
|
||||
for (var i = 0; i < gasGridCount; i++)
|
||||
sum += densities[i] = gasGrids[i].DensityAt(cellIndex);
|
||||
|
||||
var resultColor = default(Color);
|
||||
|
||||
for (var i = 0; i < gasGridCount; i++)
|
||||
resultColor += gasGrids[i].Color * (densities[i] / sum);
|
||||
|
||||
resultColor.a = Mathf.Lerp(GasGrid.AlphaRange.min, GasGrid.AlphaRange.max,
|
||||
sum / (/*gasGridCount*/ 3 * byte.MaxValue)); // maintain vanilla transparency
|
||||
|
||||
return resultColor;
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class AddGasPatch : FishPrepatch
|
||||
{
|
||||
public override bool ShowSettings => false;
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.AddGas));
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
||||
|
||||
public static void ReplacementBody(GasGrid __instance, IntVec3 cell, GasType gasType, int amount,
|
||||
bool canOverflow = true)
|
||||
=> __instance.ParallelGasGrids()[(int)gasType >> 3].AddGas(cell, amount, canOverflow);
|
||||
}
|
||||
|
||||
public sealed class AnyGasAtPatch : FishPrepatch
|
||||
{
|
||||
public override bool ShowSettings => false;
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.AnyGasAt), [typeof(int)]);
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
||||
|
||||
public static bool ReplacementBody(GasGrid __instance, int idx)
|
||||
{
|
||||
var gasGrids = __instance.ParallelGasGrids();
|
||||
for (var i = 0; i < gasGrids.Length; i++)
|
||||
{
|
||||
if (gasGrids[i].AnyGasAt(idx))
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class DensityAtPatch : FishPrepatch
|
||||
{
|
||||
public override bool ShowSettings => false;
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.DensityAt),
|
||||
[typeof(int), typeof(GasType)]);
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
||||
|
||||
public static byte ReplacementBody(GasGrid __instance, int index, GasType gasType)
|
||||
=> __instance.ParallelGasGrids()[(int)gasType >> 3].DensityAt(index);
|
||||
}
|
||||
|
||||
public sealed class Debug_ClearAllPatch : FishPrepatch
|
||||
{
|
||||
public override bool ShowSettings => false;
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.Debug_ClearAll));
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
||||
|
||||
public static void ReplacementBody(GasGrid __instance)
|
||||
{
|
||||
var gasGrids = __instance.ParallelGasGrids();
|
||||
|
||||
for (var i = 0; i < gasGrids.Length; i++)
|
||||
gasGrids[i].Clear();
|
||||
|
||||
__instance.map.mapDrawer.WholeMapChanged(MapMeshFlag.Gas);
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class Debug_FillAllPatch : FishPrepatch
|
||||
{
|
||||
public override string? Description { get; } = "Adds custom gases to the debug fill all action";
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.Debug_FillAll));
|
||||
|
||||
public static void Postfix(GasGrid __instance)
|
||||
{
|
||||
var gasGrids = __instance.ParallelGasGrids();
|
||||
if (gasGrids.Length < 3)
|
||||
return;
|
||||
|
||||
var cellCount = __instance.gasDensity.Length;
|
||||
var cellIndices = __instance.map.cellIndices;
|
||||
|
||||
for (var i = 0; i < cellCount; i++)
|
||||
{
|
||||
if (!__instance.GasCanMoveTo(cellIndices.IndexToCell(i)))
|
||||
continue;
|
||||
|
||||
for (var j = 3; j < gasGrids.Length; j++)
|
||||
gasGrids[j].SetDirect(i, 255);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class Notify_ThingSpawnedPatch : FishPrepatch
|
||||
{
|
||||
public override bool ShowSettings => false;
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.Notify_ThingSpawned));
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
=> ilProcessor.ReplaceBodyWith(ReplacementBody);
|
||||
|
||||
public static void ReplacementBody(GasGrid __instance, Thing thing)
|
||||
{
|
||||
if (!thing.IsSpawned() || thing.def.Fillage != FillCategory.Full)
|
||||
return;
|
||||
|
||||
var grids = __instance.ParallelGasGrids();
|
||||
var gridCount = grids.Length;
|
||||
var map = __instance.map;
|
||||
var mapDrawer = map.mapDrawer;
|
||||
var mapSizeX = GetMapSizeX(map);
|
||||
|
||||
foreach (var occupiedCell in thing.OccupiedRect())
|
||||
{
|
||||
var anyGridDirty = false;
|
||||
|
||||
for (var i = gridCount; i-- > 0;)
|
||||
{
|
||||
var grid = grids[i];
|
||||
var cellIndex = occupiedCell.CellToIndex(mapSizeX);
|
||||
|
||||
if (!grid.AnyGasAt(cellIndex))
|
||||
continue;
|
||||
|
||||
grid.SetDirect(cellIndex, 0);
|
||||
anyGridDirty = true;
|
||||
}
|
||||
|
||||
if (anyGridDirty)
|
||||
mapDrawer.MapMeshDirty(occupiedCell, MapMeshFlag.Gas);
|
||||
}
|
||||
}
|
||||
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public static int GetMapSizeX(Map map) => map.cellIndices.mapSizeX;
|
||||
}
|
||||
|
||||
public sealed class ExposeDataPatch : FishPrepatch
|
||||
{
|
||||
public override bool ShowSettings => false;
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(GasGrid), nameof(GasGrid.ExposeData));
|
||||
|
||||
public static void Prefix(GasGrid __instance)
|
||||
{
|
||||
if (Scribe.mode != LoadSaveMode.Saving)
|
||||
return;
|
||||
|
||||
var grids = __instance.ParallelGasGrids();
|
||||
var smokeGrid = grids[0];
|
||||
var toxicGrid = grids[1];
|
||||
var rotStinkGrid = grids[2];
|
||||
|
||||
for (var i = smokeGrid.CellCount; i-- > 0;)
|
||||
{
|
||||
__instance.gasDensity[i] = (rotStinkGrid.DensityAt(i) << 16)
|
||||
| (toxicGrid.DensityAt(i) << 8)
|
||||
| smokeGrid.DensityAt(i);
|
||||
}
|
||||
}
|
||||
|
||||
public static void Postfix(GasGrid __instance)
|
||||
{
|
||||
if (Scribe.mode != LoadSaveMode.LoadingVars)
|
||||
return;
|
||||
|
||||
var grids = __instance.ParallelGasGrids();
|
||||
var smokeGrid = grids[0];
|
||||
var toxicGrid = grids[1];
|
||||
var rotStinkGrid = grids[2];
|
||||
|
||||
for (var i = smokeGrid.CellCount; i-- > 0;)
|
||||
{
|
||||
var gasDensityAtCell = __instance.gasDensity[i];
|
||||
|
||||
smokeGrid.SetDirect(i, (byte)gasDensityAtCell);
|
||||
toxicGrid.SetDirect(i, (byte)((uint)gasDensityAtCell >> 8));
|
||||
rotStinkGrid.SetDirect(i, (byte)((uint)gasDensityAtCell >> 16));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class DebugToolsGeneralPushGas : FishPrepatch
|
||||
{
|
||||
public override string? Description { get; } = "Adds a debug action for adding custom gas defs";
|
||||
|
||||
public override MethodBase TargetMethodBase { get; } = methodof(DebugToolsGeneral.PushGas);
|
||||
|
||||
public static List<DebugActionNode> Postfix(List<DebugActionNode> __result)
|
||||
{
|
||||
var gasDefs = DefDatabase<GasDef>.AllDefsListForReading;
|
||||
|
||||
for (var i = 3; i < gasDefs.Count; i++)
|
||||
{
|
||||
var gasDef = gasDefs[i];
|
||||
|
||||
__result.Add(new(gasDef.LabelCap, DebugActionType.ToolMap,
|
||||
() => Find.CurrentMap.gasGrid.AddGas(UI.MouseCell(), gasDef, 5f)));
|
||||
}
|
||||
|
||||
return __result;
|
||||
}
|
||||
}
|
||||
|
||||
public sealed class MouseOverReadOutOnGUI : FishPrepatch
|
||||
{
|
||||
public override string? Description { get; }
|
||||
= "Adds custom gas defs to the mouse over readout on the bottom left";
|
||||
|
||||
public override MethodBase TargetMethodBase { get; }
|
||||
= AccessTools.DeclaredMethod(typeof(MouseoverReadout), nameof(MouseoverReadout.MouseoverReadoutOnGUI));
|
||||
|
||||
public override void Transpiler(ILProcessor ilProcessor, ModuleDefinition module)
|
||||
{
|
||||
var drawGasMethod = AccessTools.DeclaredMethod(typeof(MouseoverReadout), nameof(MouseoverReadout.DrawGas));
|
||||
|
||||
var instructions = ilProcessor.instructions;
|
||||
|
||||
var lastDrawGasIndex = instructions.Count - instructions.Reverse().FirstIndexOf(instruction
|
||||
=> instruction.Operand is MethodReference method && method.Is(drawGasMethod)) - 1;
|
||||
|
||||
var heightVariable = instructions[lastDrawGasIndex - 1];
|
||||
if (heightVariable.OpCode != OpCodes.Ldloca_S && heightVariable.OpCode != OpCodes.Ldloca)
|
||||
{
|
||||
ThrowHelper.ThrowInvalidOperationException(
|
||||
$"Failed to find height variable for MouseOverReadOutOnGUI patch. Expected Ldloca_S or "
|
||||
+ $"Ldloca, got '{heightVariable}' instead.");
|
||||
}
|
||||
|
||||
var cellVariable = instructions[lastDrawGasIndex - 4];
|
||||
if (!cellVariable.OpCode.Name.Contains("ldloc", StringComparison.OrdinalIgnoreCase))
|
||||
{
|
||||
ThrowHelper.ThrowInvalidOperationException(
|
||||
$"Failed to cell variable for MouseOverReadOutOnGUI patch. Expected Ldloc, got {
|
||||
cellVariable} instead.");
|
||||
}
|
||||
|
||||
ilProcessor.InsertRange(lastDrawGasIndex + 1,
|
||||
cellVariable,
|
||||
heightVariable,
|
||||
(OpCodes.Call, methodof(DrawCustomGases)));
|
||||
}
|
||||
|
||||
public static void DrawCustomGases(IntVec3 cell, ref float curYOffset)
|
||||
{
|
||||
var map = Find.CurrentMap;
|
||||
var gasGrids = map.gasGrid.ParallelGasGrids();
|
||||
var cellIndex = new CellIndex(cell, map);
|
||||
|
||||
for (var i = 3; i < gasGrids.Length; i++)
|
||||
{
|
||||
var gasGrid = gasGrids[i];
|
||||
if (!gasGrid.AnyGasAt(cellIndex))
|
||||
continue;
|
||||
|
||||
Widgets.Label(
|
||||
new(MouseoverReadout.BotLeft.x, UI.screenHeight - MouseoverReadout.BotLeft.y - curYOffset, 999f,
|
||||
999f),
|
||||
string.Concat(gasGrid.GasDef.LabelCap, " ",
|
||||
(gasGrid.DensityAt(cellIndex) / 255f).ToStringPercent("F0")));
|
||||
|
||||
curYOffset += 19f;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public class ParallelGasGrid
|
||||
{
|
||||
private ulong[] _gasCoverageForIndicesInRandomOrder;
|
||||
private byte[] _gasDensity;
|
||||
|
||||
public event Action<ParallelGasGrid>? Ticked;
|
||||
|
||||
public GasDef GasDef { get; private set; }
|
||||
|
||||
public int DissipationRate => GasDef.dissipationRate;
|
||||
|
||||
public bool Diffuses => GasDef.diffuses;
|
||||
|
||||
public Color Color => GasDef.color;
|
||||
|
||||
public bool CalculateGasEffects => anyGasEverAdded;
|
||||
|
||||
public Map Map;
|
||||
|
||||
public int
|
||||
CycleIndexDiffusion,
|
||||
CycleIndexDissipation;
|
||||
|
||||
public IntVec3[] CardinalDirections;
|
||||
|
||||
public List<IntVec3> CellsInRandomOrder = null!;
|
||||
|
||||
public int[]
|
||||
CellIndicesInRandomOrder = null!,
|
||||
SourceIndicesOfRandomCells = null!;
|
||||
|
||||
public bool anyGasEverAdded;
|
||||
|
||||
private DissipationTicker _dissipationTicker;
|
||||
private DiffusionTicker _diffusionTicker;
|
||||
|
||||
public int CellCount { get; private set; }
|
||||
|
||||
#pragma warning disable CS8618
|
||||
public ParallelGasGrid(Map map, GasDef gasDef)
|
||||
#pragma warning restore CS8618
|
||||
=> map.InvokeWhenCellIndicesReady(m => Initialize(m, gasDef));
|
||||
|
||||
private void Initialize(Map map, GasDef gasDef)
|
||||
{
|
||||
GasDef = gasDef;
|
||||
Map = map;
|
||||
CellCount = map.cellIndices.NumGridCells;
|
||||
_gasDensity = new byte[CellCount];
|
||||
_gasCoverageForIndicesInRandomOrder = new ulong[((_gasDensity.Length - 1) >> 6) + 1];
|
||||
CellIndicesInRandomOrder = new int[CellCount];
|
||||
SourceIndicesOfRandomCells = new int[CellCount];
|
||||
CardinalDirections = new IntVec3[GenAdj.CardinalDirections.Length];
|
||||
Array.Copy(GenAdj.CardinalDirections, CardinalDirections, GenAdj.CardinalDirections.Length);
|
||||
CycleIndexDiffusion = Rand.Range(0, map.Area >> 1);
|
||||
|
||||
_dissipationTicker = new() { GasGrid = this };
|
||||
_diffusionTicker = new() { GasGrid = this };
|
||||
|
||||
Ticked += _onFirstTickAction;
|
||||
}
|
||||
|
||||
private static Action<ParallelGasGrid> _onFirstTickAction = TickInitialize;
|
||||
|
||||
private static void TickInitialize(ParallelGasGrid grid)
|
||||
{
|
||||
// map.info is somehow null during ExposeData, which itself runs after CreateComponents and initializes a
|
||||
// 2nd copy of every component for each map instance, of which apparently three get created per map, two
|
||||
// of those thrown away, so cellsInRandomOrder.GetAll fails and throws when accessed from either
|
||||
// CreateComponents or ExposeData while loading a game. A total of 6 instances of each component get
|
||||
// created per map while loading it's stupid
|
||||
// Also cellsInRandomOrder itself initializes after gasGrid within CreateComponents, but that's easy to
|
||||
// solve
|
||||
grid.CellsInRandomOrder = grid.Map.cellsInRandomOrder.GetAll();
|
||||
grid.UpdateCellIndicesInRandomOrder();
|
||||
grid.Ticked -= _onFirstTickAction;
|
||||
}
|
||||
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public int CellToIndex(in IntVec3 cell) => cell.CellToIndex(Map);
|
||||
|
||||
public void UpdateCellIndicesInRandomOrder()
|
||||
{
|
||||
if (Map.cellIndices.NumGridCells != CellCount)
|
||||
{
|
||||
Log.Error($"Map size changed from {CellCount} to {
|
||||
Map.cellIndices.NumGridCells}. This is not supported.");
|
||||
return;
|
||||
}
|
||||
|
||||
var mapSizeX = Map.Size.x;
|
||||
for (var i = CellIndicesInRandomOrder.Length; i-- > 0;)
|
||||
{
|
||||
SourceIndicesOfRandomCells[
|
||||
CellIndicesInRandomOrder[i] = CellIndicesUtility.CellToIndex(CellsInRandomOrder[i], mapSizeX)] = i;
|
||||
}
|
||||
|
||||
_gasCoverageForIndicesInRandomOrder.Clear();
|
||||
|
||||
for (var i = _gasDensity.Length; i-- > 0;)
|
||||
{
|
||||
if (_gasDensity[i] > 0)
|
||||
SetDirect(i, _gasDensity[i]);
|
||||
}
|
||||
}
|
||||
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public bool AnyGasAt(int idx) => _gasDensity[idx] > 0;
|
||||
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public bool AnyGasAt(in IntVec3 cell) => AnyGasAt(CellToIndex(cell));
|
||||
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public bool AnyGasAt(CellIndex cellIndex) => AnyGasAt(cellIndex.Value);
|
||||
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public byte DensityAt(int index) => _gasDensity[index];
|
||||
|
||||
public byte DensityAt(in IntVec3 cell) => DensityAt(CellToIndex(cell));
|
||||
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public byte DensityAt(CellIndex cellIndex) => DensityAt(cellIndex.Value);
|
||||
|
||||
public void SetDirect(in IntVec3 cell, byte density) => SetDirect(CellToIndex(cell), density);
|
||||
|
||||
public void SetDirect(CellIndex cellIndex, byte density) => SetDirect(cellIndex.Value, density);
|
||||
|
||||
public void SetDirect(int index, byte density)
|
||||
{
|
||||
_gasDensity[index] = density;
|
||||
|
||||
index = SourceIndicesOfRandomCells[index];
|
||||
|
||||
ref var gasCoverageBucket = ref _gasCoverageForIndicesInRandomOrder[index >> 6];
|
||||
var hasGas = density > 0;
|
||||
gasCoverageBucket ^= ((ulong)-(long)hasGas.AsInt() ^ gasCoverageBucket) & (1UL << (index & 63));
|
||||
anyGasEverAdded |= hasGas;
|
||||
}
|
||||
|
||||
public void AddGas(CellIndex cellIndex, int amount, bool canOverflow = true)
|
||||
=> AddGas(IndexToCell(cellIndex.Value), amount, canOverflow);
|
||||
|
||||
public void AddGas(in IntVec3 cell, int amount, bool canOverflow = true)
|
||||
{
|
||||
if (amount <= 0 || !GasCanMoveTo(cell))
|
||||
return;
|
||||
|
||||
anyGasEverAdded = true;
|
||||
var index = CellToIndex(cell);
|
||||
|
||||
SetDirect(index, AdjustedDensity(_gasDensity[index] + amount, out var overflow));
|
||||
|
||||
Map.mapDrawer.MapMeshDirty(cell, MapMeshFlag.Gas);
|
||||
|
||||
if (canOverflow && overflow > 0)
|
||||
Overflow(cell, overflow);
|
||||
}
|
||||
|
||||
private static byte AdjustedDensity(int newDensity, out int overflow)
|
||||
{
|
||||
if (newDensity > 255)
|
||||
{
|
||||
overflow = newDensity - 255;
|
||||
return byte.MaxValue;
|
||||
}
|
||||
|
||||
overflow = 0;
|
||||
if (newDensity < 0)
|
||||
return 0;
|
||||
|
||||
return (byte)newDensity;
|
||||
}
|
||||
|
||||
private void Overflow(in IntVec3 cell, int amount)
|
||||
{
|
||||
if (amount <= 0)
|
||||
return;
|
||||
|
||||
var remainingAmount = amount;
|
||||
|
||||
Map.floodFiller.FloodFill(cell, GasCanMoveTo, c =>
|
||||
{
|
||||
var num = Mathf.Min(remainingAmount, 255 - DensityAt(c));
|
||||
if (num > 0)
|
||||
{
|
||||
AddGas(c, num, false);
|
||||
remainingAmount -= num;
|
||||
}
|
||||
|
||||
return remainingAmount <= 0;
|
||||
}, GenRadial.NumCellsInRadius(40f), true);
|
||||
}
|
||||
|
||||
public void Clear()
|
||||
{
|
||||
_gasCoverageForIndicesInRandomOrder.Clear();
|
||||
_gasDensity.Clear();
|
||||
anyGasEverAdded = false;
|
||||
}
|
||||
|
||||
private abstract class BitwiseGasTicker
|
||||
{
|
||||
public required ParallelGasGrid GasGrid;
|
||||
public abstract ref int CellCycleIndex { get; }
|
||||
|
||||
public void TickCells(int area, int cellCountToTick)
|
||||
{
|
||||
var cellCycleIndex = CellCycleIndex;
|
||||
|
||||
if ((cellCycleIndex & 63) != 0)
|
||||
TickOddCells(area, ref cellCycleIndex);
|
||||
|
||||
var gasCoverageArray = GasGrid._gasCoverageForIndicesInRandomOrder;
|
||||
|
||||
cellCountToTick >>= 6;
|
||||
cellCountToTick += 1;
|
||||
for (var i = 0; i < cellCountToTick; i++)
|
||||
{
|
||||
if (area - cellCycleIndex < 64)
|
||||
TickOddCells(area, ref cellCycleIndex);
|
||||
|
||||
var gasCoverage = gasCoverageArray[cellCycleIndex >> 6];
|
||||
if (gasCoverage != default)
|
||||
TickBucket(gasCoverage, cellCycleIndex);
|
||||
|
||||
cellCycleIndex += 64;
|
||||
}
|
||||
|
||||
CellCycleIndex = cellCycleIndex;
|
||||
}
|
||||
|
||||
private void TickBucket(ulong gasCoverage, int cellCycleIndex)
|
||||
{
|
||||
for (var i = 0; i < 64;)
|
||||
{
|
||||
var gasCoverageSlice = (gasCoverage >> i) & 0xFF;
|
||||
if (gasCoverageSlice == default)
|
||||
{
|
||||
i += 8;
|
||||
continue;
|
||||
}
|
||||
|
||||
for (var j = 0; j < 8; j++)
|
||||
{
|
||||
if (((gasCoverageSlice >> j) & 1UL) != default)
|
||||
TickingAction(cellCycleIndex + i);
|
||||
|
||||
i++;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public abstract void TickingAction(int cellCycleIndex);
|
||||
|
||||
[MethodImpl(MethodImplOptions.NoInlining)]
|
||||
private void TickOddCells(int area, ref int cellCycleIndex)
|
||||
{
|
||||
if (cellCycleIndex < area)
|
||||
{
|
||||
var gasCoverage = GasGrid._gasCoverageForIndicesInRandomOrder[cellCycleIndex >> 6];
|
||||
|
||||
do
|
||||
{
|
||||
if ((cellCycleIndex & 63) == 0)
|
||||
return;
|
||||
|
||||
if (((gasCoverage >> cellCycleIndex) & 1UL) != default)
|
||||
TickingAction(cellCycleIndex);
|
||||
|
||||
cellCycleIndex++;
|
||||
}
|
||||
while (cellCycleIndex < area);
|
||||
}
|
||||
|
||||
cellCycleIndex = 0;
|
||||
}
|
||||
}
|
||||
|
||||
private sealed class DissipationTicker : BitwiseGasTicker
|
||||
{
|
||||
public override ref int CellCycleIndex => ref GasGrid.CycleIndexDissipation;
|
||||
|
||||
public override void TickingAction(int cellCycleIndex)
|
||||
=> GasGrid.DissipateGasAt(GasGrid.CellIndicesInRandomOrder[cellCycleIndex]);
|
||||
}
|
||||
|
||||
private sealed class DiffusionTicker : BitwiseGasTicker
|
||||
{
|
||||
public override ref int CellCycleIndex => ref GasGrid.CycleIndexDiffusion;
|
||||
|
||||
public override void TickingAction(int cellCycleIndex)
|
||||
=> GasGrid.DiffuseGasAt(GasGrid.CellsInRandomOrder[cellCycleIndex]);
|
||||
}
|
||||
|
||||
public void Tick()
|
||||
{
|
||||
if (!CalculateGasEffects)
|
||||
{
|
||||
#if GAS_DEBUG_L1
|
||||
if (CheckCurrentlyContainsGas())
|
||||
{
|
||||
anyGasEverAdded = true;
|
||||
Debug.LogIncorrectAnyGasEverAdded(this);
|
||||
}
|
||||
#endif
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
Ticked?.Invoke(this);
|
||||
|
||||
var area = Map.Area;
|
||||
// CellsInRandomOrder = Map.cellsInRandomOrder.GetAll();
|
||||
// UpdateCellIndicesInRandomOrder();
|
||||
// Assigned in PostInitialize instead. Ludeon was likely doing this due to component construction order in
|
||||
// Map.ConstructComponents
|
||||
|
||||
#if GAS_DEBUG_L2
|
||||
using var previousDensity = new PooledArray<byte>(_gasDensity.Length);
|
||||
Array.Copy(_gasDensity, 0, previousDensity.BackingArray, 0, _gasDensity.Length);
|
||||
#endif
|
||||
|
||||
_dissipationTicker.TickCells(area, (area + 63) >> 6); // Mathf.CeilToInt(area * (1f / 64f))
|
||||
|
||||
if (Diffuses)
|
||||
_diffusionTicker.TickCells(area, (area + 31) >> 5); // Mathf.CeilToInt(area * (1f / 32f))
|
||||
|
||||
#if GAS_DEBUG_L2
|
||||
Debug.LogTotalGasChange(this, previousDensity);
|
||||
#endif
|
||||
|
||||
if (Map.IsHashIntervalTick(600))
|
||||
RecalculateEverHadGas();
|
||||
}
|
||||
|
||||
[MethodImpl(MethodImplOptions.NoInlining)]
|
||||
public void RecalculateEverHadGas() => anyGasEverAdded = CheckCurrentlyContainsGas();
|
||||
|
||||
private bool CheckCurrentlyContainsGas()
|
||||
{
|
||||
for (var i = _gasCoverageForIndicesInRandomOrder.Length; i-- > 0;)
|
||||
{
|
||||
if (_gasCoverageForIndicesInRandomOrder[i] != default)
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
private void DissipateGasAt(int index)
|
||||
{
|
||||
#if GAS_DEBUG_L1
|
||||
var previousDensity = _gasDensity[index];
|
||||
if (previousDensity == 0)
|
||||
Debug.LogInvalidDissipationAttempt(this, index);
|
||||
#endif
|
||||
|
||||
var densityAfterDissipation = Math.Max(_gasDensity[index] - DissipationRate, 0);
|
||||
|
||||
Debug.LogDissipationRate(this, index, densityAfterDissipation);
|
||||
SetDirect(index, (byte)densityAfterDissipation);
|
||||
|
||||
#if GAS_DEBUG_L1
|
||||
if (_gasDensity[index] >= previousDensity)
|
||||
Debug.LogFailedDissipation(this, index, previousDensity);
|
||||
#endif
|
||||
|
||||
if (densityAfterDissipation == 0)
|
||||
{
|
||||
Debug.LogMapMeshDirty(this, index);
|
||||
Map.mapDrawer.MapMeshDirty(IndexToCell(index), MapMeshFlag.Gas);
|
||||
}
|
||||
}
|
||||
|
||||
private void DiffuseGasAt(IntVec3 cell)
|
||||
{
|
||||
var originCellIndex = CellToIndex(cell);
|
||||
int densityAtOriginCell = _gasDensity[originCellIndex];
|
||||
if (densityAtOriginCell < 17)
|
||||
return;
|
||||
|
||||
var diffusedToAnyCell = false;
|
||||
CardinalDirections.Shuffle();
|
||||
|
||||
for (var i = 0; i < CardinalDirections.Length; i++)
|
||||
{
|
||||
var otherCell = cell + CardinalDirections[i];
|
||||
if (!GasCanMoveTo(otherCell))
|
||||
continue;
|
||||
|
||||
var otherCellIndex = CellToIndex(otherCell);
|
||||
int densityOtOtherCell = _gasDensity[otherCellIndex];
|
||||
|
||||
if (!TryDiffuseIndividualGas(ref densityAtOriginCell, ref densityOtOtherCell))
|
||||
continue;
|
||||
|
||||
SetDirect(otherCellIndex, (byte)densityOtOtherCell);
|
||||
Map.mapDrawer.MapMeshDirty(otherCell, MapMeshFlag.Gas);
|
||||
diffusedToAnyCell = true;
|
||||
if (densityAtOriginCell < 17)
|
||||
break;
|
||||
}
|
||||
|
||||
if (!diffusedToAnyCell)
|
||||
return;
|
||||
|
||||
SetDirect(originCellIndex, (byte)densityAtOriginCell);
|
||||
Map.mapDrawer.MapMeshDirty(cell, MapMeshFlag.Gas);
|
||||
}
|
||||
|
||||
private static bool TryDiffuseIndividualGas(ref int gasA, ref int gasB)
|
||||
{
|
||||
if (gasA < 17)
|
||||
return false;
|
||||
|
||||
var num = Mathf.Abs(gasA - gasB) >> 1;
|
||||
|
||||
if (gasA <= gasB || num < 17)
|
||||
return false;
|
||||
|
||||
Debug.LogDiffusionRate(gasA, gasB, num);
|
||||
|
||||
gasA -= num;
|
||||
gasB += num;
|
||||
|
||||
#if GAS_DEBUG_L1
|
||||
if (gasA < 0 || gasB < 0)
|
||||
Log.Error("Diffused below 0!");
|
||||
#endif
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public bool GasCanMoveTo(IntVec3 cell)
|
||||
{
|
||||
if (!cell.InBounds(Map))
|
||||
return false;
|
||||
if (cell.Filled(Map))
|
||||
return cell.GetDoor(Map)?.Open ?? false;
|
||||
return true;
|
||||
}
|
||||
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
private IntVec3 IndexToCell(int index)
|
||||
=> CellIndicesUtility.IndexToCell(index, Map.cellIndices.mapSizeX);
|
||||
|
||||
/// <summary>
|
||||
/// For modders. Performance Fish doesn't automatically scribe added custom defs.
|
||||
/// </summary>
|
||||
public void Scribe(string label)
|
||||
=> MapExposeUtility.ExposeUshort(Map,
|
||||
cell => DensityAt(cell),
|
||||
(cell, value) => SetDirect(cell, (byte)value),
|
||||
label);
|
||||
|
||||
// ReSharper disable UnusedMember.Local
|
||||
private static class Debug
|
||||
{
|
||||
[MethodImpl(MethodImplOptions.NoInlining)]
|
||||
[Conditional("GAS_DEBUG_L1")]
|
||||
internal static void LogIncorrectAnyGasEverAdded(ParallelGasGrid grid)
|
||||
{
|
||||
var gasDensities = grid._gasDensity;
|
||||
var firstCellIndexWithGas = Array.FindIndex(gasDensities, static g => g > 0);
|
||||
|
||||
Log.Error($"GasGrid of type '{grid.GasDef.defName}' contains gas at cell '{
|
||||
grid.IndexToCell(firstCellIndexWithGas)}' with index {
|
||||
firstCellIndexWithGas}, but anyGasEverAdded=false. This index had a source index of {
|
||||
grid.SourceIndicesOfRandomCells[firstCellIndexWithGas]} before randomization");
|
||||
}
|
||||
|
||||
[MethodImpl(MethodImplOptions.NoInlining)]
|
||||
[Conditional("GAS_DEBUG_L1")]
|
||||
internal static void LogInvalidDissipationAttempt(ParallelGasGrid grid, int index)
|
||||
=> Log.Error($"Tried to dissipate gas at cell '{
|
||||
grid.IndexToCell(index)}' with density of 0");
|
||||
|
||||
[MethodImpl(MethodImplOptions.NoInlining)]
|
||||
[Conditional("GAS_DEBUG_L1")]
|
||||
internal static void LogFailedDissipation(ParallelGasGrid grid, int index, byte previousDensity)
|
||||
=> Log.Error($"GasDensity went from '{previousDensity}' to '{grid._gasDensity[index]}' at cell '{
|
||||
grid.IndexToCell(index)}' while trying to dissipate. It should've decreased");
|
||||
|
||||
[MethodImpl(MethodImplOptions.NoInlining)]
|
||||
[Conditional("GAS_DEBUG_L2")]
|
||||
internal static void LogTotalGasChange(ParallelGasGrid grid, PooledArray<byte> previousDensity)
|
||||
=> Log.Message($"Total density for {grid.GasDef.defName} changed from {
|
||||
previousDensity.Sum(static b => b)} to {grid._gasDensity.Sum(static b => b)}");
|
||||
|
||||
[Conditional("GAS_DEBUG_L3")]
|
||||
internal static void LogMapMeshDirty(ParallelGasGrid grid, int index)
|
||||
=> Log.Message($"Dirtied MapMesh for '{grid.GasDef.defName}' at '{index}'");
|
||||
|
||||
[Conditional("GAS_DEBUG_L3")]
|
||||
internal static void LogDissipationRate(ParallelGasGrid grid, int index, int desiredResult)
|
||||
=> Log.Message($"Dissipating {grid._gasDensity[index] - desiredResult} gas of type '{
|
||||
grid.GasDef.defName}' at cell '{grid.IndexToCell(index)}' with target rate {
|
||||
grid.DissipationRate}. Was {grid._gasDensity[index]}, becomes {desiredResult}");
|
||||
|
||||
[Conditional("GAS_DEBUG_L3")]
|
||||
internal static void LogDiffusionRate(int gasA, int gasB, int num)
|
||||
=> Log.Message($"Diffusing {num} gas from cell with density {gasA} to cell with density {gasB}");
|
||||
}
|
||||
// ReSharper restore UnusedMember.Local
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user