RecipeBannerAdvanced.java

package blusunrize.immersiveengineering.common.crafting;

import blusunrize.immersiveengineering.api.crafting.IngredientStack;
import blusunrize.immersiveengineering.common.util.Utils;
import net.minecraft.init.Items;
import net.minecraft.inventory.InventoryCrafting;
import net.minecraft.item.ItemStack;
import net.minecraft.item.crafting.RecipesBanners;
import net.minecraft.tileentity.TileEntityBanner;
import net.minecraft.tileentity.TileEntityBanner.EnumBannerPattern;

import javax.annotation.Nullable;
import java.util.LinkedHashMap;

/**
 * @author BluSunrize - 18.09.2016
 */
public class RecipeBannerAdvanced extends RecipesBanners.RecipeAddPattern
{
	public static LinkedHashMap<EnumBannerPattern, RecipeReference> advancedPatterns = new LinkedHashMap<>();

	public static void addAdvancedPatternRecipe(TileEntityBanner.EnumBannerPattern pattern, IngredientStack stack, int... offset)
	{
		if(offset!=null && offset.length>0)
		{
			int[] slotOffset = new int[2];
			slotOffset[0] = offset[0];
			if(offset.length>1)
				slotOffset[1] = offset[1];
			advancedPatterns.put(pattern, new RecipeReference(stack, slotOffset));
		}
		else
			advancedPatterns.put(pattern, new RecipeReference(stack));

	}

	public RecipeBannerAdvanced()
	{
		super();
	}

	@Override
	@Nullable
	protected TileEntityBanner.EnumBannerPattern matchPatterns(InventoryCrafting invCrafting)
	{
		for(TileEntityBanner.EnumBannerPattern pattern : TileEntityBanner.EnumBannerPattern.values())
			if(advancedPatterns.containsKey(pattern))
			{
				RecipeReference ref = advancedPatterns.get(pattern);
				boolean matchesPattern = true;
				boolean hasIngr = false;
				boolean hasColour = false;

				for(int i=0; i<invCrafting.getSizeInventory()&&matchesPattern; i++)
				{
					ItemStack itemstack = invCrafting.getStackInSlot(i);
					if(itemstack!=null && itemstack.getItem()!=Items.BANNER)
					{
						if(Utils.isDye(itemstack))
						{
							if(hasColour)
							{
								matchesPattern = false;
								break;
							}
							hasColour = true;
						}
						else
						{
							if(hasIngr || !ref.ingredient.matchesItemStack(itemstack))
							{
								matchesPattern = false;
								break;
							}
							if(ref.offsetToBanner[0]!=0 || ref.offsetToBanner[1]!=0)
							{
								int w = invCrafting.getWidth();
								int h = invCrafting.getHeight();
								int bannerSlot = i-ref.offsetToBanner[0]-ref.offsetToBanner[1]*w;
								if((i%w==0 && ref.offsetToBanner[0]>0)//banner needs to be left, but ingr is in leftmost column
										||(i%w==w-1 && ref.offsetToBanner[0]<0)//banner needs to be right but ingr is in rightmost column
										||(i/h==0 && ref.offsetToBanner[1]>0)//banner needs to be above but ingr is in topmost row
										||(i/h==h-1 && ref.offsetToBanner[1]<0)//banner needs to be below but ingr is in bottommost row
										||(bannerSlot<0||bannerSlot>=invCrafting.getSizeInventory()))//bannerslot is outside grid
								{
									matchesPattern = false;
									break;
								}
								ItemStack bannerCheck = invCrafting.getStackInSlot(bannerSlot);
								if(bannerCheck==null || bannerCheck.getItem()!=Items.BANNER)
								{
									matchesPattern = false;
									break;
								}
							}
							hasIngr = true;
						}
					}
				}
				if(!hasIngr)
					matchesPattern = false;
				if(matchesPattern)
					return pattern;
			}
		return null;
	}

	private static class RecipeReference
	{
		IngredientStack ingredient;
		int[] offsetToBanner = new int[2];

		public RecipeReference(IngredientStack ingredient)
		{
			this.ingredient = ingredient;
		}
		public RecipeReference(IngredientStack ingredient, int[] offsetToBanner)
		{
			this.ingredient = ingredient;
			this.offsetToBanner = offsetToBanner;
		}
	}
}