path: root/src/main/java/gregtech/api/util/GT_AssemblyLineUtils.java

package gregtech.api.util;

import static gregtech.GT_Mod.GT_FML_LOGGER;

import cpw.mods.fml.common.FMLCommonHandler;
import gregtech.api.enums.GT_Values;
import gregtech.api.enums.ItemList;
import gregtech.api.util.GT_Recipe.GT_Recipe_AssemblyLine;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.nbt.NBTTagString;
import net.minecraftforge.fluids.FluidStack;
import scala.actors.threadpool.Arrays;

public class GT_AssemblyLineUtils {

	/**
	 * Checks the DataStick for deprecated/invalid recipes, updating them as required.
	 * @param aDataStick - The DataStick to process 
	 * @return Is this DataStick now valid with a current recipe?
	 */
	public static boolean processDataStick(ItemStack aDataStick) {
		if (!isItemDataStick(aDataStick)) {
			return false;
		}
		if (doesDataStickNeedUpdate(aDataStick)) {
			ItemStack aStickOutput = getDataStickOutput(aDataStick);
			if (aStickOutput != null) {
				GT_Recipe_AssemblyLine aIntendedRecipe = findAssemblyLineRecipeByOutput(aStickOutput);
				if (aIntendedRecipe != null) {
					return setAssemblyLineRecipeOnDataStick(aDataStick, aIntendedRecipe);
				}
			}
			return false;			
		}
		return true;
	}

	/**
	 * Finds an Assembly Line recipe from a DataStick.
	 * @param aDataStick - The DataStick to check.
	 * @return The GT_Recipe_AssemblyLine recipe contained on the DataStick, if any.
	 */
	public static GT_Recipe_AssemblyLine findAssemblyLineRecipeFromDataStick(ItemStack aDataStick) {	
		if (!isItemDataStick(aDataStick)) {
			return null;
		}
		ItemStack[] aInputs = new ItemStack[15];
		ItemStack[] aOutputs = new ItemStack[1];
		FluidStack[] aFluidInputs = new FluidStack[4];
		boolean aFoundRecipe = false;

		NBTTagCompound aTag = aDataStick.getTagCompound();
		if (aTag == null) {
			return null;
		}

		for (int i = 0; i < 15; i++) {
			int count = aTag.getInteger("a" + i);
			if (!aTag.hasKey("" + i) && count <= 0) {
				continue;
			}

			boolean flag = true;
			if (count > 0) {
				for (int j = 0; j < count; j++) {
					aInputs[i] = GT_Utility.loadItem(aTag, "a" + i + ":" + j);
					if (aInputs[i] == null) {
						continue;
					}
					if (GT_Values.D1) {
						GT_FML_LOGGER.info("Item " + i + " : " + aInputs[i].getUnlocalizedName());
					}
					flag = false;
					break;
				}
			}
			if (flag) {
				aInputs[i] = GT_Utility.loadItem(aTag, "" + i);
				if (aInputs[i] == null) {
					flag = false;
					continue;
				}
				if (GT_Values.D1) {
					GT_FML_LOGGER.info("Item " + i + " : " + aInputs[i].getUnlocalizedName());
				}
				flag = false;
			}
			if (GT_Values.D1) {
				GT_FML_LOGGER.info(i + (flag ? " not accepted" : " accepted"));
			}
		}

		if (GT_Values.D1) GT_FML_LOGGER.info("All Items done, start fluid check");
		for (int i = 0; i < 4; i++) {
			if (!aTag.hasKey("f" + i)) continue;
			aFluidInputs[i] = GT_Utility.loadFluid(aTag, "f" + i);
			if (aFluidInputs[i] == null) continue;
			if (GT_Values.D1) {
				GT_FML_LOGGER.info("Fluid " + i + " " + aFluidInputs[i].getUnlocalizedName());
			}
			if (GT_Values.D1) {
				GT_FML_LOGGER.info(i + " accepted");
			}
		}

		if (GT_Values.D1) {
			GT_FML_LOGGER.info("Input accepted, check other values");
		}
		aOutputs = new ItemStack[]{GT_Utility.loadItem(aTag, "output")};
		if (!aTag.hasKey("output") || !aTag.hasKey("time") || aTag.getInteger("time") <= 0 || !aTag.hasKey("eu") || aOutputs[0] == null || !GT_Utility.isStackValid(aOutputs[0])) {
			return null;
		}
		if (GT_Values.D1) {
			GT_FML_LOGGER.info("Find Data Stick recipe");
		}
		aFoundRecipe = true;

		if (aFoundRecipe) {
			int aTime = aTag.getInteger("time");
			int aEU = aTag.getInteger("eu");			
			for (GT_Recipe_AssemblyLine aRecipe : GT_Recipe.GT_Recipe_AssemblyLine.sAssemblylineRecipes) {
				if (aRecipe.mEUt == aEU && aRecipe.mDuration == aTime) {
					if (GT_Utility.areStacksEqual(aOutputs[0], aRecipe.mOutput, true)) {
						if (Arrays.equals(aRecipe.mInputs, aInputs) && Arrays.equals(aRecipe.mFluidInputs, aFluidInputs)) {
							return aRecipe;
						}
					}
				}
			}			
		}
		return null;
	}


	/**
	 * Finds a GT_Recipe_AssemblyLine based on the expected output ItemStack.
	 * @param aOutput - The Output of a GT_Recipe_AssemblyLine.
	 * @return First found GT_Recipe_AssemblyLine with matching output.
	 */
	public static GT_Recipe_AssemblyLine findAssemblyLineRecipeByOutput(ItemStack aOutput) {
		for (GT_Recipe_AssemblyLine aRecipe : GT_Recipe.GT_Recipe_AssemblyLine.sAssemblylineRecipes) {
			ItemStack aRecipeOutput = aRecipe.mOutput;
			if (GT_Utility.areStacksEqual(aRecipeOutput, aOutput)) {
				return aRecipe;
			}
		}
		return null;
	}


	/**
	 * Get the Output ItemStack from a Data Stick.
	 * @param aDataStick - The Data Stick to check.
	 * @return Output ItemStack contained on the Data Stick.
	 */
	public static ItemStack getDataStickOutput(ItemStack aDataStick) {
		if (doesDataStickHaveOutput(aDataStick)) {
			ItemStack aOutput = GT_Utility.loadItem(aDataStick.getTagCompound(), "output");
			return aOutput;
		}
		return null;
	}

	/**
	 * @param aDataStick - The Data Stick to check.
	 * @return Does this Data Stick have a valid output ItemStack?
	 */
	public static boolean doesDataStickHaveOutput(ItemStack aDataStick) {
		if (isItemDataStick(aDataStick) && aDataStick.hasTagCompound() && aDataStick.getTagCompound().hasKey("output")) {
			return true;
		}
		return false;
	}

	/**
	 * @param aDataStick - The Data Stick to check.
	 * @return Does this Data Stick need recipe data updated.
	 */
	public static boolean doesDataStickNeedUpdate(ItemStack aDataStick) {
		if (isItemDataStick(aDataStick) && doesDataStickHaveRecipeHash(aDataStick)) {
			String aStickHash = getHashFromDataStack(aDataStick);
			if (isValidHash(aStickHash) && doesDataStickHaveOutput(aDataStick)) {
				ItemStack aStickOutput = getDataStickOutput(aDataStick);
				GT_Recipe_AssemblyLine aIntendedRecipe = findAssemblyLineRecipeByOutput(aStickOutput);
				if (aStickHash.equals(generateRecipeHash(aIntendedRecipe))) {
					return false;
				}				
			}
		}	
		return true;
	}

	/**
	 * @param aDataStick - The Data Stick to check.
	 * @return Does this have a Recipe Hash String at all?
	 */
	public static boolean doesDataStickHaveRecipeHash(ItemStack aDataStick) {
		if (isItemDataStick(aDataStick) && aDataStick.hasTagCompound()) {
			NBTTagCompound aNBT = aDataStick.getTagCompound();
			if (aNBT.hasKey("Data.Recipe.Hash")) {
				return true;
			}
		}
		return false;
	}

	/**
	 * @param aDataStick - The Data Stick to procces.
	 * @return The stored Recipe Hash String on the Data Stick, will return an invalid Hash if one is not found. <p>
	 * Check with isValidHash(). <p> 
	 * Will not return Null.
	 */
	public static String getHashFromDataStack(ItemStack aDataStick) {
		if (isItemDataStick(aDataStick) && aDataStick.hasTagCompound()) {
			NBTTagCompound aNBT = aDataStick.getTagCompound();
			if (aNBT.hasKey("Data.Recipe.Hash")) {
				return aNBT.getString("Data.Recipe.Hash");

			}
		}
		return "Invalid.Recipe.Hash";
	}

	/**
	 * 
	 * @param aDataStick - The Data Stick to update.
	 * @param aRecipeHash - The Recipe Hash String to update with.
	 * @return Did we update the Recipe Hash String on the Data Stick?
	 */
	public static boolean writeRecipeHashToDataStick(ItemStack aDataStick, String aRecipeHash) {
		if (isItemDataStick(aDataStick) && aDataStick.hasTagCompound()) {
			NBTTagCompound aNBT = aDataStick.getTagCompound();
			aNBT.setString("Data.Recipe.Hash", aRecipeHash);
			aDataStick.setTagCompound(aNBT);
			return true;
		}
		return false;
	}

	/**
	 * @param aRecipe - The recipe to generate a Recipe Hash String from.
	 * @return The Recipe Hash String.
	 */
	public static String generateRecipeHash(GT_Recipe_AssemblyLine aRecipe) {
		String aHash = "Invalid.Recipe.Hash";		
		if (aRecipe != null) {
			aHash = "Hash."+aRecipe.hashCode();
		}		
		return aHash;
	}

	/**
	 * @param aHash - Recipe hash String, may be null but will just be treated as invalid.
	 * @return Is this Recipe Hash String valid?
	 */
	public static boolean isValidHash(String aHash) {
		if (aHash != null && aHash.length() > 0) {
			if (!aHash.equals("Invalid.Recipe.Hash") && aHash.contains("Hash.")) {
				return true;
			}
		}
		return false;
	}

	/**
	 * @param aStack - The ItemStack to check.
	 * @return Is this ItemStack a Data Stick?
	 */
	public static boolean isItemDataStick(ItemStack aStack) {
		return GT_Utility.isStackValid(aStack) && ItemList.Tool_DataStick.isStackEqual(aStack, false, true);
	}

	/**
	 * 
	 * @param aDataStick - The Data Stick to update.
	 * @param aNewRecipe - The New GT_Recipe_AssemblyLine recipe to update it with.
	 * @return Did we set the new recipe data & Recipe Hash String on the Data Stick?
	 */
	public static boolean setAssemblyLineRecipeOnDataStick(ItemStack aDataStick, GT_Recipe_AssemblyLine aNewRecipe) {
		if (isItemDataStick(aDataStick)) {
			String s = aNewRecipe.mOutput.getDisplayName();
			if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
				s = GT_Assemblyline_Server.lServerNames.get(aNewRecipe.mOutput.getDisplayName());
				if (s == null)
					s = aNewRecipe.mOutput.getDisplayName();
			}

			//remove possible old NBTTagCompound
			aDataStick.setTagCompound(new NBTTagCompound());
			GT_Utility.ItemNBT.setBookTitle(aDataStick, s + " Construction Data");

			NBTTagCompound tNBT = aDataStick.getTagCompound();
			if (tNBT == null) {
				tNBT = new NBTTagCompound();
			}

			tNBT.setTag("output", aNewRecipe.mOutput.writeToNBT(new NBTTagCompound()));
			tNBT.setInteger("time", aNewRecipe.mDuration);
			tNBT.setInteger("eu", aNewRecipe.mEUt);
			for (int i = 0; i < aNewRecipe.mInputs.length; i++) {
				tNBT.setTag("" + i, aNewRecipe.mInputs[i].writeToNBT(new NBTTagCompound()));
			}
			for (int i = 0; i < aNewRecipe.mOreDictAlt.length; i++) {
				if (aNewRecipe.mOreDictAlt[i] != null && aNewRecipe.mOreDictAlt[i].length > 0) {
					tNBT.setInteger("a" + i, aNewRecipe.mOreDictAlt[i].length);
					for (int j = 0; j < aNewRecipe.mOreDictAlt[i].length; j++) {
						tNBT.setTag("a" + i + ":" + j, aNewRecipe.mOreDictAlt[i][j].writeToNBT(new NBTTagCompound()));
					}
				}
			}
			for (int i = 0; i < aNewRecipe.mFluidInputs.length; i++) {
				tNBT.setTag("f" + i, aNewRecipe.mFluidInputs[i].writeToNBT(new NBTTagCompound()));
			}
			tNBT.setString("author", "Assembling Line Recipe Generator");
			NBTTagList tNBTList = new NBTTagList();
			s = aNewRecipe.mOutput.getDisplayName();
			if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
				s = GT_Assemblyline_Server.lServerNames.get(aNewRecipe.mOutput.getDisplayName());
				if (s == null)
					s = aNewRecipe.mOutput.getDisplayName();
			}
			tNBTList.appendTag(new NBTTagString("Construction plan for " + aNewRecipe.mOutput.stackSize + " " + s + ". Needed EU/t: " + aNewRecipe.mEUt + " Production time: " + (aNewRecipe.mDuration / 20)));
			for (int i = 0; i < aNewRecipe.mInputs.length; i++) {
				if (aNewRecipe.mOreDictAlt[i] != null) {
					int count = 0;
					StringBuilder tBuilder = new StringBuilder("Input Bus " + (i + 1) + ": ");
					for (ItemStack tStack : aNewRecipe.mOreDictAlt[i]) {
						if (tStack != null) {
							s = tStack.getDisplayName();
							if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
								s = GT_Assemblyline_Server.lServerNames.get(tStack.getDisplayName());
								if (s == null)
									s = tStack.getDisplayName();
							}


							tBuilder.append(count == 0 ? "" : "\nOr ").append(tStack.stackSize).append(" ").append(s);
							count++;
						}
					}
					if (count > 0) tNBTList.appendTag(new NBTTagString(tBuilder.toString()));
				} else if (aNewRecipe.mInputs[i] != null) {
					s = aNewRecipe.mInputs[i].getDisplayName();
					if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
						s = GT_Assemblyline_Server.lServerNames.get(aNewRecipe.mInputs[i].getDisplayName());
						if (s == null)
							s = aNewRecipe.mInputs[i].getDisplayName();
					}
					tNBTList.appendTag(new NBTTagString("Input Bus " + (i + 1) + ": " + aNewRecipe.mInputs[i].stackSize + " " + s));
				}
			}
			for (int i = 0; i < aNewRecipe.mFluidInputs.length; i++) {
				if (aNewRecipe.mFluidInputs[i] != null) {
					s = aNewRecipe.mFluidInputs[i].getLocalizedName();
					if (FMLCommonHandler.instance().getEffectiveSide().isServer()) {
						s = GT_Assemblyline_Server.lServerNames.get(aNewRecipe.mFluidInputs[i].getLocalizedName());
						if (s == null)
							s = aNewRecipe.mFluidInputs[i].getLocalizedName();
					}
					tNBTList.appendTag(new NBTTagString("Input Hatch " + (i + 1) + ": " + aNewRecipe.mFluidInputs[i].amount + "L " + s));
				}
			}
			tNBT.setTag("pages", tNBTList);		
			aDataStick.setTagCompound(tNBT);	
			// Set recipe hash
			writeRecipeHashToDataStick(aDataStick, generateRecipeHash(aNewRecipe));
			return true;
		}
		return false;
	}

}