more nuker qol options from epic nuker

Author: beanbag44

src/main/kotlin/com/lambda/module/modules/player/Nuker.kt

@@ -18,7 +18,7 @@
 package com.lambda.module.modules.player
 
 import com.lambda.config.AutomationConfig.Companion.setDefaultAutomationConfig
-import com.lambda.config.applyEdits
+import com.lambda.context.SafeContext
 import com.lambda.interaction.BaritoneManager
 import com.lambda.interaction.construction.blueprint.TickingBlueprint.Companion.tickingBlueprint
 import com.lambda.interaction.construction.verify.TargetState
@@ -28,8 +28,12 @@ import com.lambda.task.RootTask.run
 import com.lambda.task.Task
 import com.lambda.task.tasks.BuildTask.Companion.build
 import com.lambda.util.BlockUtils.blockPos
+import com.lambda.util.BlockUtils.blockState
+import com.lambda.util.BlockUtils.isNotEmpty
+import com.lambda.util.math.MathUtils.ceilToInt
 import net.minecraft.block.Blocks
 import net.minecraft.util.math.BlockPos
+import net.minecraft.util.math.Direction
 
 object Nuker : Module(
 	name = "Nuker",
@@ -38,11 +42,14 @@ object Nuker : Module(
 ) {
 	private val height by setting("Height", 6, 1..8, 1)
 	private val width by setting("Width", 6, 1..8, 1)
-	private val flatten by setting("Flatten", true)
+	private val flattenMode by setting("Flatten Mode", FlattenMode.Standard)
+	private val directionalDig by setting("Directional Dig", DigDirection.None)
 	private val onGround by setting("On Ground", false, "Only break blocks when the player is standing on ground")
 	private val fillFluids by setting("Fill Fluids", false, "Removes liquids by filling them in before breaking")
 	private val fillFloor by setting("Fill Floor", false)
 	private val baritoneSelection by setting("Baritone Selection", false, "Restricts nuker to your baritone selection")
+	private val inverseSelection by setting("Inverse Selection", false, "Breaks blocks outside of the baritone selection and ignores blocks inside") { baritoneSelection }
+	private val sneakLowersFlatten by setting("Sneak Lowers Flatten", false)
 
 	private var task: Task<*>? = null
 
@@ -57,17 +64,9 @@ object Nuker : Module(
 					.asSequence()
 					.map { it.blockPos }
 					.filter { !world.isAir(it) }
-					.filter { !flatten || it.y >= player.blockPos.y }
-					.filter { pos ->
-						if (!baritoneSelection) true
-						else BaritoneManager.primary?.selectionManager?.selections?.any {
-							val min = it.min()
-							val max = it.max()
-							pos.x >= min.x && pos.x <= max.x
-									&& pos.y >= min.y && pos.y <= max.y
-									&& pos.z >= min.z && pos.z <= max.z
-						} ?: false
-					}
+					.filter { flattenMode == FlattenMode.None || isInFlatten(it) }
+					.filter { isWithinDigDirection(it) }
+					.filter { isInBaritoneSelection(it) == !inverseSelection }
 					.associateWith { if (fillFluids) TargetState.Air else TargetState.Empty }
 
 				if (fillFloor) {
@@ -86,4 +85,81 @@ object Nuker : Module(
 			task?.cancel()
 		}
 	}
-}
+
+	private fun SafeContext.isInFlatten(pos: BlockPos): Boolean {
+		if (flattenMode == FlattenMode.Staircase) {
+			val up = pos.up()
+			if ((blockState(up).isNotEmpty && (!baritoneSelection || isInBaritoneSelection(up)))
+				|| (blockState(up.east()).isNotEmpty && (!baritoneSelection || isInBaritoneSelection(up.east())))
+				|| (blockState(up.south()).isNotEmpty && (!baritoneSelection || isInBaritoneSelection(up.south())))
+				|| (blockState(up.west()).isNotEmpty && (!baritoneSelection || isInBaritoneSelection(up.west())))
+				|| (blockState(up.north()).isNotEmpty && (!baritoneSelection || isInBaritoneSelection(up.north())))
+			)  { return false }
+		}
+
+		val flattenY = player.y.ceilToInt()
+		val playerPos = player.blockPos
+		val flattenLevel =
+			if (sneakLowersFlatten && player.isSneaking) flattenY - 1
+			else flattenY
+
+		if (!flattenMode.isSmart && pos.y < flattenLevel)
+			return false
+
+		if (pos == player.supportingBlockPos) return false
+
+		val playerLookDir = player.horizontalFacing
+		val smartFlattenDir =
+			if (flattenMode == FlattenMode.Smart) playerLookDir
+			else playerLookDir?.opposite
+
+		if (pos.y >= flattenLevel) return true
+
+		val zeroedPos = pos.add(-playerPos.x, -flattenY, -playerPos.z)
+
+		return (zeroedPos.x < 0 && smartFlattenDir == Direction.EAST)
+				|| (zeroedPos.z < 0 && smartFlattenDir == Direction.SOUTH)
+				|| (zeroedPos.x > 0 && smartFlattenDir == Direction.WEST)
+				|| (zeroedPos.z > 0 && smartFlattenDir == Direction.NORTH)
+	}
+
+	private fun SafeContext.isWithinDigDirection(pos: BlockPos): Boolean {
+		val playerPos = player.blockPos
+		return when (directionalDig) {
+			DigDirection.None -> true
+			DigDirection.East -> playerPos.x <= pos.x
+			DigDirection.West -> playerPos.x >= pos.x
+			DigDirection.North -> playerPos.z >= pos.z
+			DigDirection.South -> playerPos.z <= pos.z
+		}
+	}
+
+	private fun isInBaritoneSelection(pos: BlockPos) =
+		if (!baritoneSelection) true
+		else BaritoneManager.primary?.selectionManager?.selections?.any {
+			val min = it.min()
+			val max = it.max()
+			pos.x >= min.x && pos.x <= max.x
+					&& pos.y >= min.y && pos.y <= max.y
+					&& pos.z >= min.z && pos.z <= max.z
+		} ?: false
+
+	private enum class FlattenMode {
+		None,
+		Standard,
+		Smart,
+		ReverseSmart,
+		Staircase;
+
+		val isSmart
+			get() = this == Smart || this == ReverseSmart
+	}
+
+	private enum class DigDirection {
+		None,
+		East,
+		South,
+		West,
+		North
+	}
+}