using System;
using System.Diagnostics;
using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.Versioning;
using BepInEx;
using BepInEx.Configuration;
using BepInEx.Logging;
using HarmonyLib;
using UnityEngine;
using UnityEngine.Rendering;
[assembly: CompilationRelaxations(8)]
[assembly: RuntimeCompatibility(WrapNonExceptionThrows = true)]
[assembly: Debuggable(DebuggableAttribute.DebuggingModes.IgnoreSymbolStoreSequencePoints)]
[assembly: TargetFramework(".NETFramework,Version=v4.7.2", FrameworkDisplayName = ".NET Framework 4.7.2")]
[assembly: AssemblyVersion("0.0.0.0")]
namespace DSPSwarmDrawFix;
public enum NearSailsMode
{
Off,
Auto,
Always
}
[BepInPlugin("me.gp.dspswarmdrawfix", "DSPSwarmDrawFix", "1.2.0")]
public class SwarmDrawFixPlugin : BaseUnityPlugin
{
internal static ManualLogSource Log;
internal static ConfigEntry<NearSailsMode> NearMode;
internal static readonly FieldRef<DysonSwarm, ComputeBuffer> SwarmBuffer = AccessTools.FieldRefAccess<DysonSwarm, ComputeBuffer>("swarmBuffer");
internal static readonly FieldRef<DysonSwarm, ComputeBuffer> SwarmInfoBuffer = AccessTools.FieldRefAccess<DysonSwarm, ComputeBuffer>("swarmInfoBuffer");
internal static readonly FieldRef<DysonSwarm, ComputeBuffer> OrbitColorsBuffer = AccessTools.FieldRefAccess<DysonSwarm, ComputeBuffer>("sailOrbitColorsBuffer");
internal static readonly FieldRef<DysonSwarm, ComputeBuffer> BulletBuffer = AccessTools.FieldRefAccess<DysonSwarm, ComputeBuffer>("bulletBuffer");
internal static readonly FieldRef<DysonSwarm, float> SizeInMat = AccessTools.FieldRefAccess<DysonSwarm, float>("sizeInMat");
private void Awake()
{
//IL_0030: Unknown result type (might be due to invalid IL or missing references)
//IL_0035: Unknown result type (might be due to invalid IL or missing references)
Log = ((BaseUnityPlugin)this).Logger;
NearMode = ((BaseUnityPlugin)this).Config.Bind<NearSailsMode>("Rendering", "NearSails", NearSailsMode.Auto, "Near sails: the solid mesh drawn within 2500 units instead of a flat sprite.\nThe game culls them on the GPU through an append buffer and an indirect draw, which D3DMetal does not handle, so this mod instances the whole swarm instead — cost then scales with sail count.\nAuto: only draw when the camera is close enough to a sail orbit for anything to be in range. Off: never draw them. Always: 1.1.0 behaviour, costly on large swarms.");
Harmony val = new Harmony("me.gp.dspswarmdrawfix");
val.PatchAll(typeof(SwarmDrawPatch));
val.PatchAll(typeof(NearDrawPatch));
Log.LogInfo((object)$"DSPSwarmDrawFix active: far sails as indexed triangles, near sails {NearMode.Value}");
}
private void OnDestroy()
{
SwarmDrawPatch.ReleaseBuffers();
NearDrawPatch.ReleaseBuffers();
}
}
internal static class Prop
{
internal static readonly int SwarmBuffer = Shader.PropertyToID("_SwarmBuffer");
internal static readonly int SwarmInfoBuffer = Shader.PropertyToID("_SwarmInfoBuffer");
internal static readonly int NodeBuffer = Shader.PropertyToID("_NodeBuffer");
internal static readonly int OrbitColor = Shader.PropertyToID("_OrbitColor");
internal static readonly int NearIdBuffer = Shader.PropertyToID("_NearIdBuffer");
internal static readonly int BulletBuffer = Shader.PropertyToID("_BulletBuffer");
internal static readonly int SunPosition = Shader.PropertyToID("_SunPosition");
internal static readonly int SunPositionMap = Shader.PropertyToID("_SunPosition_Map");
internal static readonly int LocalRot = Shader.PropertyToID("_LocalRot");
internal static readonly int GameTick = Shader.PropertyToID("_GameTick");
internal static readonly int Stride = Shader.PropertyToID("_Stride");
internal static readonly int DistScalePoint = Shader.PropertyToID("_DistScalePoint");
internal static readonly int RenderPlace = Shader.PropertyToID("_Global_DS_RenderPlace");
internal static readonly int EditorMaskS = Shader.PropertyToID("_Global_DS_EditorMaskS");
internal static readonly int GameMaskS = Shader.PropertyToID("_Global_DS_GameMaskS");
}
internal struct SwarmFrame
{
internal Vector3 SunPos;
internal Vector3 SunPosMap;
internal Vector4 LocalRot;
internal uint Tick;
internal static SwarmFrame Build(DysonSwarm sw)
{
//IL_000a: Unknown result type (might be due to invalid IL or missing references)
//IL_000f: Unknown result type (might be due to invalid IL or missing references)
//IL_0016: Unknown result type (might be due to invalid IL or missing references)
//IL_001b: Unknown result type (might be due to invalid IL or missing references)
//IL_0036: Unknown result type (might be due to invalid IL or missing references)
//IL_003b: Unknown result type (might be due to invalid IL or missing references)
//IL_0083: Unknown result type (might be due to invalid IL or missing references)
//IL_0089: Unknown result type (might be due to invalid IL or missing references)
//IL_008e: Unknown result type (might be due to invalid IL or missing references)
//IL_0093: Unknown result type (might be due to invalid IL or missing references)
//IL_00f0: Unknown result type (might be due to invalid IL or missing references)
//IL_00f1: Unknown result type (might be due to invalid IL or missing references)
//IL_00f6: Unknown result type (might be due to invalid IL or missing references)
//IL_00fb: Unknown result type (might be due to invalid IL or missing references)
//IL_0101: Invalid comparison between Unknown and I4
//IL_0098: Unknown result type (might be due to invalid IL or missing references)
//IL_009d: Unknown result type (might be due to invalid IL or missing references)
//IL_009e: Unknown result type (might be due to invalid IL or missing references)
//IL_00a3: Unknown result type (might be due to invalid IL or missing references)
//IL_00a4: Unknown result type (might be due to invalid IL or missing references)
//IL_00a6: Unknown result type (might be due to invalid IL or missing references)
//IL_00ab: Unknown result type (might be due to invalid IL or missing references)
//IL_00b0: Unknown result type (might be due to invalid IL or missing references)
//IL_00b5: Unknown result type (might be due to invalid IL or missing references)
//IL_00e4: Unknown result type (might be due to invalid IL or missing references)
//IL_00e9: Unknown result type (might be due to invalid IL or missing references)
//IL_010b: Unknown result type (might be due to invalid IL or missing references)
//IL_0110: Unknown result type (might be due to invalid IL or missing references)
//IL_0115: Unknown result type (might be due to invalid IL or missing references)
//IL_0123: Unknown result type (might be due to invalid IL or missing references)
//IL_0128: Unknown result type (might be due to invalid IL or missing references)
//IL_012d: Unknown result type (might be due to invalid IL or missing references)
SwarmFrame result = new SwarmFrame
{
SunPos = Vector3.zero,
SunPosMap = Vector3.zero,
LocalRot = new Vector4(0f, 0f, 0f, 1f),
Tick = (uint)(GameMain.gameTick & 0xFFFFFFFFu)
};
if (sw.starData == null || sw.gameData == null)
{
return result;
}
PlanetData localPlanet = sw.gameData.localPlanet;
Player mainPlayer = sw.gameData.mainPlayer;
VectorLF3 val = sw.starData.uPosition - mainPlayer.uPosition;
if (localPlanet != null)
{
val = Maths.QInvRotateLF(localPlanet.runtimeRotation, val);
val += VectorLF3.op_Implicit(mainPlayer.position);
result.LocalRot = new Vector4(localPlanet.runtimeRotation.x, localPlanet.runtimeRotation.y, localPlanet.runtimeRotation.z, localPlanet.runtimeRotation.w);
}
result.SunPos = VectorLF3.op_Implicit(val);
if ((int)DysonSphere.renderPlace == 1)
{
result.SunPosMap = VectorLF3.op_Implicit((sw.starData.uPosition - UIStarmap.viewTargetStatic) * 0.00025);
}
return result;
}
internal void ApplyTo(Material mat)
{
//IL_0007: Unknown result type (might be due to invalid IL or missing references)
//IL_000c: Unknown result type (might be due to invalid IL or missing references)
//IL_001d: Unknown result type (might be due to invalid IL or missing references)
//IL_0022: Unknown result type (might be due to invalid IL or missing references)
//IL_0033: Unknown result type (might be due to invalid IL or missing references)
mat.SetVector(Prop.SunPosition, Vector4.op_Implicit(SunPos));
mat.SetVector(Prop.SunPositionMap, Vector4.op_Implicit(SunPosMap));
mat.SetVector(Prop.LocalRot, LocalRot);
mat.SetInt(Prop.GameTick, (int)Tick);
}
internal static int LayerForRenderPlace()
{
//IL_0000: Unknown result type (might be due to invalid IL or missing references)
//IL_0006: Invalid comparison between Unknown and I4
//IL_000b: Unknown result type (might be due to invalid IL or missing references)
//IL_0011: Invalid comparison between Unknown and I4
if ((int)DysonSphere.renderPlace == 1)
{
return 20;
}
if ((int)DysonSphere.renderPlace == 2)
{
return 21;
}
return 16;
}
}
internal static class SwarmDrawPatch
{
private const int ChunkSails = 5460;
private static GraphicsBuffer quadIndexBuffer;
private static bool failedOnce;
internal static void ReleaseBuffers()
{
GraphicsBuffer obj = quadIndexBuffer;
if (obj != null)
{
obj.Release();
}
quadIndexBuffer = null;
}
private static GraphicsBuffer GetIndexBuffer()
{
//IL_0072: Unknown result type (might be due to invalid IL or missing references)
//IL_007c: Expected O, but got Unknown
if (quadIndexBuffer != null && quadIndexBuffer.IsValid())
{
return quadIndexBuffer;
}
int num = 16380;
int[] array = new int[num * 6];
for (int i = 0; i < num; i++)
{
int num2 = i * 4;
int num3 = i * 6;
array[num3] = num2;
array[num3 + 1] = num2 + 1;
array[num3 + 2] = num2 + 2;
array[num3 + 3] = num2;
array[num3 + 4] = num2 + 2;
array[num3 + 5] = num2 + 3;
}
quadIndexBuffer = new GraphicsBuffer((Target)2, array.Length, 4);
quadIndexBuffer.SetData((Array)array);
return quadIndexBuffer;
}
[HarmonyPrefix]
[HarmonyPatch(typeof(DysonSwarm), "DrawPost")]
private static bool DrawPostPrefix(DysonSwarm __instance)
{
try
{
ReplacementDrawPost(__instance);
return false;
}
catch (Exception arg)
{
if (!failedOnce)
{
failedOnce = true;
SwarmDrawFixPlugin.Log.LogError((object)$"DrawPost replacement threw, falling back to vanilla: {arg}");
}
return true;
}
}
private static void ReplacementDrawPost(DysonSwarm sw)
{
//IL_000d: Unknown result type (might be due to invalid IL or missing references)
//IL_0013: Invalid comparison between Unknown and I4
//IL_003a: Unknown result type (might be due to invalid IL or missing references)
//IL_0040: Invalid comparison between Unknown and I4
SwarmFrame swarmFrame = SwarmFrame.Build(sw);
Camera val = Camera.main;
if ((int)DysonSphere.renderPlace == 1)
{
UIStarmap starmap = UIRoot.instance.uiGame.starmap;
if ((Object)(object)starmap != (Object)null)
{
val = starmap.screenCamera;
}
}
else if ((int)DysonSphere.renderPlace == 2)
{
UIDysonEditor dysonEditor = UIRoot.instance.uiGame.dysonEditor;
if ((Object)(object)dysonEditor != (Object)null)
{
val = dysonEditor.screenCamera;
}
}
if ((Object)(object)val != (Object)null)
{
float num = SwarmDrawFixPlugin.SizeInMat.Invoke(sw) * (float)Screen.height * 0.4f;
float num2 = 1f / Mathf.Tan(val.fieldOfView * ((float)Math.PI / 180f) * 0.5f) * num;
sw.sailFarMaterial.SetFloat(Prop.DistScalePoint, num2);
}
ComputeBuffer val2 = SwarmDrawFixPlugin.SwarmBuffer.Invoke(sw);
Material sailFarMaterial = sw.sailFarMaterial;
sailFarMaterial.SetInt(Prop.Stride, 5460);
sailFarMaterial.SetBuffer(Prop.SwarmBuffer, val2);
sailFarMaterial.SetBuffer(Prop.SwarmInfoBuffer, SwarmDrawFixPlugin.SwarmInfoBuffer.Invoke(sw));
sailFarMaterial.SetBuffer(Prop.NodeBuffer, sw.dysonSphere.nrdBuffer);
sailFarMaterial.SetBuffer(Prop.OrbitColor, SwarmDrawFixPlugin.OrbitColorsBuffer.Invoke(sw));
swarmFrame.ApplyTo(sailFarMaterial);
sailFarMaterial.SetPass(0);
if (sw.sailCursor > 0 && val2 != null)
{
int num3 = (sw.sailCursor + 5460 - 1) / 5460;
Graphics.DrawProceduralNow((MeshTopology)0, GetIndexBuffer(), 98280, num3);
}
ComputeBuffer val3 = SwarmDrawFixPlugin.BulletBuffer.Invoke(sw);
if (val3 != null)
{
sw.bulletMaterial.SetBuffer(Prop.BulletBuffer, val3);
sw.bulletMaterial.SetPass(0);
Graphics.DrawProceduralNow((MeshTopology)2, sw.bulletCursor * 8, 1);
}
}
}
internal static class NearDrawPatch
{
private const float NearThreshold = 2500f;
private static ComputeBuffer identityBuffer;
private static bool failedOnce;
internal static void ReleaseBuffers()
{
ComputeBuffer obj = identityBuffer;
if (obj != null)
{
obj.Release();
}
identityBuffer = null;
}
private static ComputeBuffer GetIdentityBuffer(int minCount)
{
//IL_0040: Unknown result type (might be due to invalid IL or missing references)
//IL_004a: Expected O, but got Unknown
if (identityBuffer != null && identityBuffer.IsValid() && identityBuffer.count >= minCount)
{
return identityBuffer;
}
int num = Mathf.NextPowerOfTwo(minCount);
ComputeBuffer obj = identityBuffer;
if (obj != null)
{
obj.Release();
}
identityBuffer = new ComputeBuffer(num, 4, (ComputeBufferType)0);
int[] array = new int[num];
for (int i = 0; i < num; i++)
{
array[i] = i;
}
identityBuffer.SetData((Array)array);
return identityBuffer;
}
[HarmonyPrefix]
[HarmonyPatch(typeof(DysonSwarm), "DrawModel")]
private static bool DrawModelPrefix(DysonSwarm __instance, ERenderPlace place, int editorMask, int gameMask)
{
try
{
ReplacementDrawModel(__instance, editorMask, gameMask);
return false;
}
catch (Exception arg)
{
if (!failedOnce)
{
failedOnce = true;
SwarmDrawFixPlugin.Log.LogError((object)$"DrawModel replacement threw, falling back to vanilla: {arg}");
}
return true;
}
}
private static bool AnySailCanBeNear(DysonSwarm sw, Camera camera)
{
//IL_0017: Unknown result type (might be due to invalid IL or missing references)
//IL_0022: Unknown result type (might be due to invalid IL or missing references)
//IL_0027: Unknown result type (might be due to invalid IL or missing references)
//IL_002c: Unknown result type (might be due to invalid IL or missing references)
//IL_0037: Unknown result type (might be due to invalid IL or missing references)
//IL_003c: Unknown result type (might be due to invalid IL or missing references)
//IL_0041: Unknown result type (might be due to invalid IL or missing references)
SailOrbit[] orbits = sw.orbits;
if (orbits == null)
{
return false;
}
VectorLF3 val = sw.gameData.mainPlayer.uPosition + VectorLF3.op_Implicit(((Component)camera).transform.position) - sw.starData.uPosition;
float num = (float)((VectorLF3)(ref val)).magnitude;
for (int i = 1; i < sw.orbitCursor; i++)
{
if (orbits[i].enabled && orbits[i].count > 0 && Mathf.Abs(num - orbits[i].radius) <= 2500f)
{
return true;
}
}
return false;
}
private static void ReplacementDrawModel(DysonSwarm sw, int editorMask, int gameMask)
{
//IL_00f6: Unknown result type (might be due to invalid IL or missing references)
//IL_0100: Expected I4, but got Unknown
//IL_0127: Unknown result type (might be due to invalid IL or missing references)
//IL_013b: Unknown result type (might be due to invalid IL or missing references)
//IL_0140: Unknown result type (might be due to invalid IL or missing references)
NearSailsMode value = SwarmDrawFixPlugin.NearMode.Value;
if (value == NearSailsMode.Off || (Object)(object)Configs.builtin.solarSailMesh == (Object)null || sw.starData == null || sw.gameData == null || sw.sailCursor <= 1)
{
return;
}
Camera main = Camera.main;
if (!((Object)(object)main == (Object)null) && (value != NearSailsMode.Auto || AnySailCanBeNear(sw, main)))
{
ComputeBuffer val = SwarmDrawFixPlugin.SwarmBuffer.Invoke(sw);
if (val != null)
{
SwarmFrame swarmFrame = SwarmFrame.Build(sw);
Material sailNearMaterial = sw.sailNearMaterial;
sailNearMaterial.SetBuffer(Prop.NearIdBuffer, GetIdentityBuffer(sw.sailCursor));
sailNearMaterial.SetBuffer(Prop.SwarmBuffer, val);
sailNearMaterial.SetBuffer(Prop.SwarmInfoBuffer, SwarmDrawFixPlugin.SwarmInfoBuffer.Invoke(sw));
sailNearMaterial.SetBuffer(Prop.NodeBuffer, sw.dysonSphere.nrdBuffer);
sailNearMaterial.SetBuffer(Prop.OrbitColor, SwarmDrawFixPlugin.OrbitColorsBuffer.Invoke(sw));
swarmFrame.ApplyTo(sailNearMaterial);
sailNearMaterial.SetInt(Prop.RenderPlace, (int)DysonSphere.renderPlace);
sailNearMaterial.SetInt(Prop.EditorMaskS, editorMask);
sailNearMaterial.SetInt(Prop.GameMaskS, gameMask);
Graphics.DrawMeshInstancedProcedural(Configs.builtin.solarSailMesh, 0, sailNearMaterial, new Bounds(Vector3.zero, new Vector3(100000f, 100000f, 100000f)), sw.sailCursor, (MaterialPropertyBlock)null, (ShadowCastingMode)0, false, SwarmFrame.LayerForRenderPlace(), (Camera)null, (LightProbeUsage)1, (LightProbeProxyVolume)null);
}
}
}
}
