add to harvestables

DREDGE-Mods · Sep 11, 2024 · 2ed3b6b · 2ed3b6b
1 parent 6b14348
commit 2ed3b6b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Winch/Util/PoiUtil.cs b/Winch/Util/PoiUtil.cs
@@ -240,6 +240,7 @@ internal static GameObject CreateGameObjectFromCustomHarvestPoi(CustomHarvestPOI
         harvestPoiDataModel.nightItems = customHarvestPoi.NightItems;
 
         harvestPoiDataModel.id = customHarvestPoi.id;
+        Harvestables.Add(customHarvestPoi.id, harvestPoiDataModel);
 
         harvestPoi.HarvestPOIData = harvestPoiDataModel;
         harvestPoi.Harvestable = harvestPoiDataModel;
@@ -276,6 +277,7 @@ internal static GameObject CreateGameObjectFromCustomItemPoi(CustomItemPOI custo
         itemPoiDataModel.items = customItemPoi.Items;
 
         itemPoiDataModel.id = customItemPoi.id;
+        Harvestables.Add(customItemPoi.id, itemPoiDataModel);
 
         itemPoi.itemPOIData = itemPoiDataModel;
         itemPoi.Harvestable = itemPoiDataModel;
@@ -484,4 +486,9 @@ public static IHarvestable[] GetAllHarvestables()
     {
         return Harvestables.Values.ToArray();
     }
+
+    public static GameObject[] GetAllHarvestParticlePrefabs()
+    {
+        return HarvestParticlePrefabs.Values.Concat(ModdedHarvestParticlePrefabs.Values).ToArray();
+    }
 }