Only make source blocks valid for bottomless pools

Creators-of-Create · Sep 19, 2024 · 6b1531f · 6b1531f
1 parent aa15182
commit 6b1531f
Showing 1 changed file with 8 additions and 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
@@ -279,7 +279,14 @@ private void continueSearch() {
 		}
 
 		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();