From 36b96ffcafde06601e3458eb8a41781abfe6f0b1 Mon Sep 17 00:00:00 2001
From: GlodBlock <60341015+GlodBlock@users.noreply.github.com>
Date: Thu, 23 Sep 2021 13:02:52 +0800
Subject: allow NEI to search fluid for cell

---
 src/main/java/gregtech/nei/GT_NEI_DefaultHandler.java | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'src/main/java/gregtech/nei/GT_NEI_DefaultHandler.java')

diff --git a/src/main/java/gregtech/nei/GT_NEI_DefaultHandler.java b/src/main/java/gregtech/nei/GT_NEI_DefaultHandler.java
index 802fcdbc6e..628620b4f7 100644
--- a/src/main/java/gregtech/nei/GT_NEI_DefaultHandler.java
+++ b/src/main/java/gregtech/nei/GT_NEI_DefaultHandler.java
@@ -12,6 +12,7 @@ import codechicken.nei.recipe.TemplateRecipeHandler;
 import cpw.mods.fml.common.event.FMLInterModComms;
 import gregtech.GT_Mod;
 import gregtech.api.enums.GT_Values;
+import gregtech.api.enums.ItemList;
 import gregtech.api.enums.OrePrefixes;
 import gregtech.api.gui.GT_GUIContainer_BasicMachine;
 import gregtech.api.objects.ItemData;
@@ -106,6 +107,14 @@ public class GT_NEI_DefaultHandler extends TemplateRecipeHandler {
                 }
             }
         }
+        FluidStack tFluidStack = GT_Utility.getFluidFromDisplayStack(aResult);
+        if (tFluidStack != null) {
+            for (FluidContainerRegistry.FluidContainerData tData : FluidContainerRegistry.getRegisteredFluidContainerData()) {
+                if (tData.fluid.isFluidEqual(tFluidStack)) {
+                    tResults.add(GT_Utility.copyOrNull(tData.filledContainer));
+                }
+            }
+        }
         for (GT_Recipe tRecipe : getSortedRecipes()) {
             if (!tRecipe.mHidden) {
                 CachedDefaultRecipe tNEIRecipe = new CachedDefaultRecipe(tRecipe);
@@ -140,6 +149,14 @@ public class GT_NEI_DefaultHandler extends TemplateRecipeHandler {
                 }
             }
         }
+        FluidStack tFluidStack = GT_Utility.getFluidFromDisplayStack(aInput);
+        if (tFluidStack != null) {
+            for (FluidContainerRegistry.FluidContainerData tData : FluidContainerRegistry.getRegisteredFluidContainerData()) {
+                if (tData.fluid.isFluidEqual(tFluidStack)) {
+                    tInputs.add(GT_Utility.copyOrNull(tData.filledContainer));
+                }
+            }
+        }
         for (GT_Recipe tRecipe : getSortedRecipes()) {
             if (!tRecipe.mHidden) {
                 CachedDefaultRecipe tNEIRecipe = new CachedDefaultRecipe(tRecipe);
-- 
cgit