added ChunkPos distance operations

Author: emyfops

src/main/kotlin/com/lambda/util/math/Vectors.kt

@@ -21,6 +21,7 @@ import com.lambda.util.math.MathUtils.floorToInt
 import com.lambda.util.math.MathUtils.sq
 import net.minecraft.entity.Entity
 import net.minecraft.util.math.BlockPos
+import net.minecraft.util.math.ChunkPos
 import net.minecraft.util.math.Direction
 import net.minecraft.util.math.EightWayDirection
 import net.minecraft.util.math.Vec2f
@@ -78,6 +79,19 @@ infix operator fun Vec2f.div(other: Int): Vec2f = times(1.0 / other)
 
 operator fun Vec2f.unaryMinus(): Vec2f = negate()
 
+/* Chunk */
+val ChunkPos.center: ChunkPos
+    get() = ChunkPos(centerX, centerZ)
+
+infix fun ChunkPos.dist(other: Vec3i): Double = other dist this
+infix fun ChunkPos.dist(other: Vec3d): Double = other dist this
+infix fun ChunkPos.dist(other: ChunkPos): Double = sqrt((this distSq other).toDouble())
+infix fun ChunkPos.distCenter(other: ChunkPos): Double = sqrt((this distSqCenter other).toDouble())
+infix fun ChunkPos.distSq(other: Vec3i): Int = other distSq this
+infix fun ChunkPos.distSq(other: Vec3d): Double = other distSq this
+infix fun ChunkPos.distSq(other: ChunkPos): Int = (other.x - x).sq + (other.z - z).sq
+infix fun ChunkPos.distSqCenter(other: ChunkPos): Int = (other.centerX - centerX).sq + (other.centerZ - centerZ).sq
+
 /* Vec3d */
 fun Vec3d.approximate(other: Vec3d, precision: Double = 2.0E-4): Boolean =
     (subtract(other) distSq Vec3d.ZERO) > precision.pow(2)
@@ -100,9 +114,10 @@ operator fun Vec3d.component3() = z
 
 infix fun Vec3d.dist(other: Vec3d): Double = sqrt(this distSq other)
 infix fun Vec3d.dist(other: Vec3i): Double = sqrt(this distSq other)
-infix fun Vec3d.distSq(other: Vec3d): Double = squaredDistanceTo(other)
-infix fun Vec3d.distSq(other: Vec3i): Double =
-    squaredDistanceTo(other.x.toDouble(), other.y.toDouble(), other.z.toDouble())
+infix fun Vec3d.dist(other: ChunkPos): Double = sqrt(this distSq other)
+infix fun Vec3d.distSq(other: Vec3d): Double = (other.x - x).sq + (other.y - y).sq + (other.z - z).sq
+infix fun Vec3d.distSq(other: Vec3i): Double = (other.x - x).sq + (other.y - y).sq + (other.z - z).sq
+infix fun Vec3d.distSq(other: ChunkPos): Double = (other.x - x).sq + (other.z - z).sq
 
 infix operator fun Vec3d.plus(other: Vec3d): Vec3d = add(other)
 infix operator fun Vec3d.plus(other: Vec3i): Vec3d = Vec3d(x + other.x, y + other.y, z + other.z)
@@ -148,8 +163,10 @@ val Vec3i.vec3d
 
 infix fun Vec3i.dist(other: Vec3d): Double = sqrt(this distSq other)
 infix fun Vec3i.dist(other: Vec3i): Double = sqrt((this distSq other).toDouble())
+infix fun Vec3i.dist(other: ChunkPos): Double = sqrt((this distSq other).toDouble())
 infix fun Vec3i.distSq(other: Vec3d): Double = getSquaredDistance(other)
 infix fun Vec3i.distSq(other: Vec3i): Int = (x - other.x).sq + (y - other.y).sq + (z - other.z).sq
+infix fun Vec3i.distSq(other: ChunkPos): Int = (other.x - x).sq + (other.z - z).sq
 
 infix operator fun Vec3i.plus(other: Vec3i): Vec3i = add(other)
 infix operator fun Vec3i.plus(other: Int): Vec3i = add(other, other, other)