1 files changed, 100 insertions, 0 deletions
diff --git a/src/main/java/gregtech/api/util/GTShapelessRecipe.java b/src/main/java/gregtech/api/util/GTShapelessRecipe.java
new file mode 100644
index 0000000000..20c1d361be
--- /dev/null
+++ b/src/main/java/gregtech/api/util/GTShapelessRecipe.java
@@ -0,0 +1,100 @@
+package gregtech.api.util;
+
+import net.minecraft.enchantment.Enchantment;
+import net.minecraft.enchantment.EnchantmentHelper;
+import net.minecraft.inventory.InventoryCrafting;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NBTTagCompound;
+import net.minecraft.world.World;
+import net.minecraftforge.oredict.ShapelessOreRecipe;
+
+import gregtech.api.interfaces.internal.IGTCraftingRecipe;
+
+public class GTShapelessRecipe extends ShapelessOreRecipe implements IGTCraftingRecipe {
+
+    public final boolean mRemovableByGT, mKeepingNBT;
+    private final Enchantment[] mEnchantmentsAdded;
+    private final int[] mEnchantmentLevelsAdded;
+
+    public GTShapelessRecipe(ItemStack aResult, boolean aDismantleAble, boolean aRemovableByGT, boolean aKeepingNBT,
+        Enchantment[] aEnchantmentsAdded, int[] aEnchantmentLevelsAdded, Object... aRecipe) {
+        super(aResult, aRecipe);
+        mEnchantmentsAdded = aEnchantmentsAdded;
+        mEnchantmentLevelsAdded = aEnchantmentLevelsAdded;
+        mRemovableByGT = aRemovableByGT;
+        mKeepingNBT = aKeepingNBT;
+    }
+
+    @Override
+    public boolean matches(InventoryCrafting aGrid, World aWorld) {
+        if (mKeepingNBT) {
+            ItemStack tStack = null;
+            for (int i = 0; i < aGrid.getSizeInventory(); i++) {
+                if (aGrid.getStackInSlot(i) != null) {
+                    if (tStack != null) {
+                        if ((tStack.hasTagCompound() != aGrid.getStackInSlot(i)
+                            .hasTagCompound()) || (tStack.hasTagCompound()
+                                && !tStack.getTagCompound()
+                                    .equals(
+                                        aGrid.getStackInSlot(i)
+                                            .getTagCompound())))
+                            return false;
+                    }
+                    tStack = aGrid.getStackInSlot(i);
+                }
+            }
+        }
+        return super.matches(aGrid, aWorld);
+    }
+
+    @Override
+    public ItemStack getCraftingResult(InventoryCrafting aGrid) {
+        ItemStack rStack = super.getCraftingResult(aGrid);
+        if (rStack != null) {
+            // Update the Stack
+            GTUtility.updateItemStack(rStack);
+
+            // Keeping NBT
+            if (mKeepingNBT) for (int i = 0; i < aGrid.getSizeInventory(); i++) {
+                if (aGrid.getStackInSlot(i) != null && aGrid.getStackInSlot(i)
+                    .hasTagCompound()) {
+                    rStack.setTagCompound(
+                        (NBTTagCompound) aGrid.getStackInSlot(i)
+                            .getTagCompound()
+                            .copy());
+                    break;
+                }
+            }
+
+            // Charge Values
+            if (GTModHandler.isElectricItem(rStack)) {
+                GTModHandler.dischargeElectricItem(rStack, Integer.MAX_VALUE, Integer.MAX_VALUE, true, false, true);
+                int tCharge = 0;
+                for (int i = 0; i < aGrid.getSizeInventory(); i++) tCharge += GTModHandler.dischargeElectricItem(
+                    aGrid.getStackInSlot(i),
+                    Integer.MAX_VALUE,
+                    Integer.MAX_VALUE,
+                    true,
+                    true,
+                    true);
+                if (tCharge > 0) GTModHandler.chargeElectricItem(rStack, tCharge, Integer.MAX_VALUE, true, false);
+            }
+
+            // Add Enchantments
+            for (int i = 0; i < mEnchantmentsAdded.length; i++) GTUtility.ItemNBT.addEnchantment(
+                rStack,
+                mEnchantmentsAdded[i],
+                EnchantmentHelper.getEnchantmentLevel(mEnchantmentsAdded[i].effectId, rStack)
+                    + mEnchantmentLevelsAdded[i]);
+
+            // Update the Stack again
+            GTUtility.updateItemStack(rStack);
+        }
+        return rStack;
+    }
+
+    @Override
+    public boolean isRemovable() {
+        return mRemovableByGT;
+    }
+}