From 0c046e8c29eed33d31bc8a8e9a09a66b05d308c6 Mon Sep 17 00:00:00 2001
From: Electroblob77 <35599699+Electroblob77@users.noreply.github.com>
Date: Sat, 15 Aug 2020 23:09:57 +0100
Subject: [PATCH] Add elemental crystal and crystal block recipes to imbuement
 altar

---
 .../tileentity/TileEntityImbuementAltar.java        | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/main/java/electroblob/wizardry/tileentity/TileEntityImbuementAltar.java b/src/main/java/electroblob/wizardry/tileentity/TileEntityImbuementAltar.java
index c323ad9e..bd077f0a 100644
--- a/src/main/java/electroblob/wizardry/tileentity/TileEntityImbuementAltar.java
+++ b/src/main/java/electroblob/wizardry/tileentity/TileEntityImbuementAltar.java
@@ -4,11 +4,13 @@ import electroblob.wizardry.block.BlockReceptacle;
 import electroblob.wizardry.constants.Element;
 import electroblob.wizardry.item.IManaStoringItem;
 import electroblob.wizardry.item.ItemWizardArmour;
+import electroblob.wizardry.registry.WizardryBlocks;
 import electroblob.wizardry.registry.WizardryItems;
 import electroblob.wizardry.registry.WizardrySounds;
 import electroblob.wizardry.util.GeometryUtils;
 import electroblob.wizardry.util.ParticleBuilder;
 import electroblob.wizardry.util.ParticleBuilder.Type;
+import net.minecraft.item.Item;
 import net.minecraft.item.ItemStack;
 import net.minecraft.nbt.NBTTagCompound;
 import net.minecraft.network.NetworkManager;
@@ -131,6 +133,17 @@ public class TileEntityImbuementAltar extends TileEntity implements ITickable {
 			}
 		}
 
+		if((stack.getItem() == WizardryItems.magic_crystal || stack.getItem() == Item.getItemFromBlock(WizardryBlocks.crystal_block))
+				&& stack.getMetadata() == 0){
+
+			Element[] elements = getReceptacleElements();
+
+			if(Arrays.stream(elements).distinct().count() == 1 && elements[0] != null){ // All the same element
+				displayElement = elements[0];
+				return new ItemStack(stack.getItem(), elements[0].ordinal());
+			}
+		}
+
 		displayElement = null;
 		return ItemStack.EMPTY;
 	}