From 32b6e9d75e8310bc0b2296ae5333cdfa0abd0d95 Mon Sep 17 00:00:00 2001
From: "Nick J. Connors" <nicholasconnors@hotmail.ca>
Date: Mon, 3 Jan 2022 13:01:27 -0500
Subject: [PATCH] Made asteroid belts support the proc gen module

---
 NewHorizons/Builder/Body/AsteroidBeltBuilder.cs | 2 ++
 NewHorizons/External/AsteroidBeltModule.cs      | 1 +
 2 files changed, 3 insertions(+)

diff --git a/NewHorizons/Builder/Body/AsteroidBeltBuilder.cs b/NewHorizons/Builder/Body/AsteroidBeltBuilder.cs
index 9e49a5d2..96ea1047 100644
--- a/NewHorizons/Builder/Body/AsteroidBeltBuilder.cs
+++ b/NewHorizons/Builder/Body/AsteroidBeltBuilder.cs
@@ -59,6 +59,8 @@ namespace NewHorizons.Builder.Body
                     }
                 };
 
+                var asteroidConfig = new PlanetConfig(config);
+                if (belt.ProcGen != null) asteroidConfig.ProcGen = belt.ProcGen;
                 var asteroid = new NewHorizonsBody(new PlanetConfig(config), assets, uniqueName);
                 Main.NextPassBodies.Add(asteroid);
             }
diff --git a/NewHorizons/External/AsteroidBeltModule.cs b/NewHorizons/External/AsteroidBeltModule.cs
index b2edb237..c275b376 100644
--- a/NewHorizons/External/AsteroidBeltModule.cs
+++ b/NewHorizons/External/AsteroidBeltModule.cs
@@ -13,5 +13,6 @@ namespace NewHorizons.External
         public float Inclination { get; set; }
         public float LongitudeOfAscendingNode { get; set; }
         public int RandomSeed { get; set; }
+        public ProcGenModule ProcGen { get; set; }
     }
 }