From 461cc0051552bfbe5bf7fb61f241941e147e72d6 Mon Sep 17 00:00:00 2001
From: Micle <micle@micle.dev>
Date: Tue, 18 Jan 2022 20:01:00 +0000
Subject: [PATCH] Added all new items into model gen.

---
 .../data/client/ModItemModelProvider.java       | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java b/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java
index a26554c..e37fa0a 100755
--- a/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java
+++ b/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java
@@ -1,8 +1,14 @@
 package dev.micle.totemofreviving.data.client;
 
 import dev.micle.totemofreviving.TotemOfReviving;
-import dev.micle.totemofreviving.item.StrawChargeItem;
-import dev.micle.totemofreviving.item.StrawTotemItem;
+import dev.micle.totemofreviving.item.charge.DiamondChargeItem;
+import dev.micle.totemofreviving.item.charge.IronChargeItem;
+import dev.micle.totemofreviving.item.charge.NetheriteChargeItem;
+import dev.micle.totemofreviving.item.charge.StrawChargeItem;
+import dev.micle.totemofreviving.item.totem.DiamondTotemItem;
+import dev.micle.totemofreviving.item.totem.IronTotemItem;
+import dev.micle.totemofreviving.item.totem.NetheriteTotemItem;
+import dev.micle.totemofreviving.item.totem.StrawTotemItem;
 import net.minecraft.data.DataGenerator;
 import net.minecraftforge.client.model.generators.ItemModelBuilder;
 import net.minecraftforge.client.model.generators.ItemModelProvider;
@@ -19,7 +25,14 @@ public class ModItemModelProvider extends ItemModelProvider {
         ModelFile itemGenerated = getExistingFile(mcLoc("item/generated"));
 
         builder(itemGenerated, StrawTotemItem.getName());
+		builder(itemGenerated, IronTotemItem.getName());
+		builder(itemGenerated, DiamondTotemItem.getName());
+		builder(itemGenerated, NetheriteTotemItem.getName());
+		
 		builder(itemGenerated, StrawChargeItem.getName());
+		builder(itemGenerated, IronChargeItem.getName());
+		builder(itemGenerated, DiamondChargeItem.getName());
+		builder(itemGenerated, NetheriteChargeItem.getName());
     }
 
     private ItemModelBuilder builder(ModelFile itemGenerated, String name) {