From a302178374dcc3f554e3ffa76089f7116d028593 Mon Sep 17 00:00:00 2001
From: IThundxr <me@ithundxr.dev>
Date: Thu, 19 Sep 2024 17:24:24 -0400
Subject: [PATCH] Only make source blocks valid for bottomless pools

---
 .../content/fluids/transfer/FluidDrainingBehaviour.java  | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/simibubi/create/content/fluids/transfer/FluidDrainingBehaviour.java b/src/main/java/com/simibubi/create/content/fluids/transfer/FluidDrainingBehaviour.java
index 6354642fa..2767563ac 100644
--- a/src/main/java/com/simibubi/create/content/fluids/transfer/FluidDrainingBehaviour.java
+++ b/src/main/java/com/simibubi/create/content/fluids/transfer/FluidDrainingBehaviour.java
@@ -279,7 +279,14 @@ public class FluidDrainingBehaviour extends FluidManipulationBehaviour {
 		}
 
 		int maxBlocks = maxBlocks();
-		if (visited.size() > maxBlocks && canDrainInfinitely(fluid) && !queue.isEmpty()) {
+		int sourceBlocks = 0;
+
+		for (BlockPos pos : visited) {
+			if (getWorld().getFluidState(pos).isSource())
+				sourceBlocks++;
+		}
+
+		if (sourceBlocks > maxBlocks && canDrainInfinitely(fluid) && !queue.isEmpty()) {
 			infinite = true;
 			BlockPos firstValid = queue.first()
 				.pos();