SphereOpt.cs

using System.Collections.Generic;
using BepInEx;
using BepInEx.Logging;
using HarmonyLib;
using System.IO;
using System.Reflection;
using UnityEngine;

namespace SphereOpt
{
    [BepInPlugin(PluginInfo.PLUGIN_GUID, PluginInfo.PLUGIN_NAME, PluginInfo.PLUGIN_VERSION)]
    public class SphereOpt : BaseUnityPlugin
    {
        public static ManualLogSource logger;
        private static AssetBundle bundle;
        private static readonly string AssemblyPath = Path.GetDirectoryName(Assembly.GetAssembly(typeof(SphereOpt)).Location);

        private static Dictionary<int, InstDysonShellRenderer> instRenderers = new Dictionary<int, InstDysonShellRenderer>();

        private static AssetBundle Bundle
        {
            get
            {
                if (bundle == null)
                {
                    var path = Path.Combine(AssemblyPath, "sphereopt-bundle");
                    if (File.Exists(path))
                    {
                        bundle = AssetBundle.LoadFromFile(path);
                    }
                    else
                    {
                        logger.LogError("Failed to load AssetBundle!".Translate());
                        return null;
                    }
                }
                return bundle;
            }
        }

        private void Awake()
        {
            logger = Logger;

            CustomShaderManager.InitWithBundle(Bundle);

            CustomShaderManager.AddCustomShaderDesc(
                "dysonshell-inst",
                "VF Shaders/Dyson Sphere/Dyson Shell Unlit Instanced"
            );

            CustomShaderManager.AddCustomShaderDesc(
                "dysonframe",
                "VF Shaders/Dyson Sphere/Frame Inst REPLACE",
                "VF Shaders/Dyson Sphere/Frame Inst"
            );

            CustomShaderManager.AddCustomShaderDesc(
                "instFrameLOD2",
                "VF Shaders/Dyson Sphere/Frame Inst REPLACE LOD2"
            );

            CustomShaderManager.AddCustomShaderDesc(
                "dysonnode",
                "VF Shaders/Dyson Sphere/Node Inst REPLACE",
                "VF Shaders/Dyson Sphere/Node Inst"
            );

            Harmony.CreateAndPatchAll(typeof(Patch_VFPreload));
            Harmony.CreateAndPatchAll(typeof(Patch_DysonShell));
            Harmony.CreateAndPatchAll(typeof(Patch_DysonSphereSegmentRenderer));
            Harmony.CreateAndPatchAll(typeof(Patch_DEBUG));
        }

        public static InstDysonShellRenderer getInstDysonShellRendererForSphere(DysonSphere ds)
        {
            if (!instRenderers.TryGetValue(ds.starData.id, out InstDysonShellRenderer renderer))
            {
                renderer = new InstDysonShellRenderer(ds);
                instRenderers[ds.starData.id] = renderer;
            }
            return renderer;
        }

        public static void RemoveRenderer(DysonSphere ds)
        {
            instRenderers.Remove(ds.starData.id);
        }

        public static void UpdateColor(DysonShell shell)
        {
            var dsRenderer = getInstDysonShellRendererForSphere(shell.dysonSphere);
            var layer = dsRenderer.getInstShellLayer(shell.layerId);
            layer.UpdateColor(shell.id, shell.color);
        }
    }
}