Merged in AlkWork (pull request #4)

AlkWork

1 files changed, 9 insertions, 2 deletions

diff --git a/src/Java/gtPlusPlus/nei/NEI_GT_Config.java b/src/Java/gtPlusPlus/nei/NEI_GT_Config.java
index 45d406aa96..438ada00ba 100644
--- a/src/Java/gtPlusPlus/nei/NEI_GT_Config.java
+++ b/src/Java/gtPlusPlus/nei/NEI_GT_Config.java
@@ -4,15 +4,21 @@ import codechicken.nei.api.API;
 import codechicken.nei.api.IConfigureNEI;
 import gregtech.api.util.CustomRecipeMap;
 import gregtech.api.util.Recipe_GT.Gregtech_Recipe_Map;
-import gtPlusPlus.xmod.gregtech.common.tileentities.machines.multi.production.chemplant.GregtechMTE_ChemicalPlant;
+import gtPlusPlus.api.objects.data.AutoMap;
 
 public class NEI_GT_Config
 implements IConfigureNEI {
 	public static boolean sIsAdded = true;
+	
+	private static final AutoMap<String> mUniqueRecipeMapHandling = new AutoMap<String>();
 
 	@Override
 	public synchronized void loadConfig() {
 		sIsAdded = false;
+
+		mUniqueRecipeMapHandling.add(Gregtech_Recipe_Map.sChemicalPlantRecipes.mUnlocalizedName);
+		mUniqueRecipeMapHandling.add(Gregtech_Recipe_Map.sOreMillRecipes.mUnlocalizedName);
+		
 		for (final CustomRecipeMap tMap : gregtech.api.util.CustomRecipeMap.sMappings) {
 			if (tMap.mNEIAllowed) {
 				new GT_NEI_DefaultHandler(tMap);
@@ -20,12 +26,13 @@ implements IConfigureNEI {
 		}
 		for (final Gregtech_Recipe_Map tMap : gregtech.api.util.Recipe_GT.Gregtech_Recipe_Map.sMappings) {
 			if (tMap.mNEIAllowed) {				
-				if (!tMap.mUnlocalizedName.equals(Gregtech_Recipe_Map.sChemicalPlantRecipes.mUnlocalizedName)) {
+				if (!mUniqueRecipeMapHandling.contains(tMap.mUnlocalizedName)) {
 					new GT_NEI_MultiBlockHandler(tMap);					
 				}				
 			}
 		}
 		new GT_NEI_FluidReactor();
+		new GT_NEI_MillingMachine();
 		sIsAdded = true;
 		API.registerRecipeHandler(new DecayableRecipeHandler());
 		API.registerUsageHandler(new DecayableRecipeHandler());