new-horizons/NewHorizons/Components/StellarRemnantController.cs

using NewHorizons.Components.SizeControllers;
using NewHorizons.External.Modules.VariableSize;
using System;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;

namespace NewHorizons.Components
{
    public class StellarRemnantController : MonoBehaviour
    {
        private StellarRemnantType _type = StellarRemnantType.Default;

        private bool _activated;

        private StarEvolutionController _starEvolutionController;

        private StellarRemnantProxy _proxy;

        private StarController _starController;

        private float _siderealPeriod = 0;
        private float _surfaceGravity = 0;
        private float _surfaceSize = 0;
        private float _sphereOfInfluence = 0;
        private float _alignmentRadius = 0;

        public StellarRemnantType GetRemnantType() => _type;
        public void SetRemnantType(StellarRemnantType type) => _type = type;

        public void SetSiderealPeriod(float siderealPeriod) => _siderealPeriod = siderealPeriod;
        public void SetSurfaceGravity(float surfaceGravity) => _surfaceGravity = surfaceGravity;
        public void SetSurfaceSize(float surfaceSize) => _surfaceSize = surfaceSize;
        public void SetAlignmentRadius(float alignmentRadius) => _alignmentRadius = alignmentRadius;
        public void SetSphereOfInfluence(float sphereOfInfluence) => _sphereOfInfluence = sphereOfInfluence;
        public void SetStarController(StarController starController) => _starController = starController;

        public void SetProxy(StellarRemnantProxy proxy) => _proxy = proxy;

        public void SetStarEvolutionController(StarEvolutionController controller) => _starEvolutionController = controller;

        public void PartiallyActivate()
        {
            if (!gameObject.activeSelf) gameObject.SetActive(true);
        }

        public void FullyActivate()
        {
            if (!gameObject.activeSelf) gameObject.SetActive(true);

            var owrb = this.GetAttachedOWRigidbody();
            if (owrb != null)
            {
                owrb.SetAngularVelocity(Vector3.up * (_siderealPeriod == 0 ? 0 : 2 * Mathf.PI / (_siderealPeriod * 60)));

                var gravityVolume = owrb.GetAttachedGravityVolume();
                if (gravityVolume != null)
                {
                    gravityVolume._alignmentRadius = _alignmentRadius;
                    gravityVolume._upperSurfaceRadius = _surfaceSize;
                    gravityVolume._surfaceAcceleration = _surfaceGravity;
                    gravityVolume._gravitationalMass = _surfaceGravity * Mathf.Pow(_surfaceSize, gravityVolume._falloffExponent) / (1f / 1000f);
                    if (gravityVolume._setMass) owrb.SetMass(gravityVolume._gravitationalMass);
                }

                var referenceFrameVolume = owrb._attachedRFVolume;
                if (referenceFrameVolume != null)
                {
                    referenceFrameVolume.GetComponent<SphereCollider>().radius = _sphereOfInfluence * 2;
                    referenceFrameVolume._maxColliderRadius = _sphereOfInfluence * 2;
                }
            }

            if (_starController != null) StarLightController.AddStar(_starController);

            _activated = true;
        }

        public bool IsActivated() => _activated;
    }
}