From 61257611601dbbc90bfca7270d2bf16cfefef5df Mon Sep 17 00:00:00 2001
From: Noah Pilarski <mehmehmeh313@gmail.com>
Date: Sun, 21 Aug 2022 16:22:21 -0400
Subject: [PATCH] Set death type

---
 NewHorizons/Builder/Body/StarBuilder.cs                        | 2 ++
 .../Components/SizeControllers/StarEvolutionController.cs      | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/NewHorizons/Builder/Body/StarBuilder.cs b/NewHorizons/Builder/Body/StarBuilder.cs
index 463697a5..def3fa05 100644
--- a/NewHorizons/Builder/Body/StarBuilder.cs
+++ b/NewHorizons/Builder/Body/StarBuilder.cs
@@ -156,6 +156,7 @@ namespace NewHorizons.Builder.Body
                 if (starModule.curve != null) controller.SetScaleCurve(starModule.curve);
                 controller.size = starModule.size;
                 controller.supernovaSize = starModule.supernovaSize;
+                controller.deathType = starModule.stellarDeathType;
                 controller.atmosphere = sunAtmosphere;
                 controller.controller = starController;
                 controller.supernova = supernova;
@@ -217,6 +218,7 @@ namespace NewHorizons.Builder.Body
                 if (starModule.curve != null) controller.SetScaleCurve(starModule.curve);
                 controller.size = starModule.size;
                 controller.supernovaSize = starModule.supernovaSize;
+                controller.deathType = starModule.stellarDeathType;
                 controller.supernova = supernova;
                 controller.StartColour = starModule.tint;
                 controller.EndColour = starModule.endTint;
diff --git a/NewHorizons/Components/SizeControllers/StarEvolutionController.cs b/NewHorizons/Components/SizeControllers/StarEvolutionController.cs
index 6d07818c..e0a5b01d 100644
--- a/NewHorizons/Components/SizeControllers/StarEvolutionController.cs
+++ b/NewHorizons/Components/SizeControllers/StarEvolutionController.cs
@@ -1,4 +1,5 @@
 using NewHorizons.Builder.Body;
+using NewHorizons.External.Modules.VariableSize;
 using NewHorizons.Handlers;
 using NewHorizons.Utility;
 using System;
@@ -53,6 +54,8 @@ namespace NewHorizons.Components.SizeControllers
         public float lifespan = 22f; // minutes
         public float supernovaSize = 50000f;
 
+        public StellarDeathType deathType;
+
         private bool _isSupernova;
         private float _supernovaStartTime;