Add RadiusUtils#iterateFromCenter

MModding · Mar 29, 2024 · e5c3f2e · e5c3f2e
1 parent 9d138d6
commit e5c3f2e
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/src/main/java/com/mmodding/mmodding_lib/library/utils/RadiusUtils.java b/src/main/java/com/mmodding/mmodding_lib/library/utils/RadiusUtils.java
@@ -1,7 +1,10 @@
 package com.mmodding.mmodding_lib.library.utils;
 
 import net.minecraft.util.math.BlockPos;
+import net.minecraft.util.math.Direction;
 
+import java.util.HashSet;
+import java.util.Set;
 import java.util.function.Consumer;
 
 public class RadiusUtils {
@@ -16,4 +19,37 @@ public static void forBlockPosInCubicRadius(BlockPos pos, int radius, Consumer<?
 			pos.getZ() + radius
 		).forEach(execute);
 	}
+
+	public static void iterateFromCenterInSquare(BlockPos center, int radius, Consumer<BlockPos> action) {
+		RadiusUtils.iterateFromCenter(center, radius, radius, radius, action);
+	}
+
+	public static void iterateFromCenter(BlockPos center, int radiusX, int radiusY, int radiusZ, Consumer<BlockPos> action) {
+		Set<BlockPos> alreadyChecked = new HashSet<>();
+		BlockPos firstExtremis = center.add(radiusX, radiusY, radiusZ);
+		BlockPos lastExtremis = center.add(-radiusX, -radiusY, -radiusZ);
+		for (Direction direction : Direction.values()) {
+			RadiusUtils.checkActionNext(center.offset(direction), direction.getOpposite(), alreadyChecked, firstExtremis, lastExtremis, action);
+		}
+	}
+
+	private static void checkActionNext(BlockPos pos, Direction excluded, Set<BlockPos> alreadyChecked, BlockPos firstExtremis, BlockPos lastExtremis, Consumer<BlockPos> action) {
+		if (!alreadyChecked.contains(pos)) {
+			BlockPos firstSubtract = pos.subtract(firstExtremis);
+			BlockPos lastSubtract = pos.subtract(lastExtremis);
+			boolean firstCheck = firstSubtract.getX() <= 0 && firstSubtract.getY() <= 0 && firstSubtract.getZ() <= 0;
+			boolean lastCheck = -lastSubtract.getX() <= 0 && -lastSubtract.getY() <= 0 && -lastSubtract.getZ() <= 0;
+
+			if (firstCheck && lastCheck) {
+				alreadyChecked.add(pos);
+				action.accept(pos);
+				for (Direction direction : Direction.values()) {
+					if (direction != excluded) {
+						RadiusUtils.checkActionNext(pos.offset(direction), direction.getOpposite(), alreadyChecked, firstExtremis, lastExtremis, action);
+					}
+				}
+			}
+		}
+	}
 }
+