package info.u_team.u_team_core.item.armor;

import net.minecraftforge.fml.RegistryObject;

public class ArmorSet {
	
	private final RegistryObject<UHelmetItem> helmet;
	private final RegistryObject<UChestplateItem> chestplate;
	private final RegistryObject<ULeggingsItem> leggings;
	private final RegistryObject<UBootsItem> boots;
	
	public ArmorSet(RegistryObject<UHelmetItem> helmet, RegistryObject<UChestplateItem> chestplate, RegistryObject<ULeggingsItem> leggings, RegistryObject<UBootsItem> boots) {
		this.helmet = helmet;
		this.chestplate = chestplate;
		this.leggings = leggings;
		this.boots = boots;
	}
	
	public RegistryObject<UHelmetItem> getHelmet() {
		return helmet;
	}
	
	public RegistryObject<UChestplateItem> getChestplate() {
		return chestplate;
	}
	
	public RegistryObject<ULeggingsItem> getLeggings() {
		return leggings;
	}
	
	public RegistryObject<UBootsItem> getBoots() {
		return boots;
	}
}

@@ -38,16 +38,89 @@
 armor;%0A%0A
+import java.util.Iterator;%0A%0Aimport com.google.common.collect.Iterators;%0A%0A
 import n
@@ -179,16 +179,74 @@
 ArmorSet
+ implements Iterable%3CRegistryObject%3C? extends UArmorItem%3E%3E
  %7B%0A%09%0A%09pr
@@ -1043,10 +1043,160 @@
 ots;%0A%09%7D%0A
+%09%0A%09@Override%0A%09public Iterator%3CRegistryObject%3C? extends UArmorItem%3E%3E iterator() %7B%0A%09%09return Iterators.forArray(helmet, chestplate, leggings, boots);%0A%09%7D%0A
 %7D%0A

Add iterator to the armor set for convenience