Make Pos & Vec proper classes (in the future record then primitive)

This commit is contained in:
TheMode 2021-07-03 03:07:18 +02:00
parent dd1bf74837
commit 061b6091e9
6 changed files with 178 additions and 315 deletions

View File

@ -1,14 +1,47 @@
package net.minestom.server.utils.incubator; package net.minestom.server.utils.incubator;
import org.jetbrains.annotations.Contract; import org.jetbrains.annotations.Contract;
import org.jetbrains.annotations.NotNull;
/**
* Represents a 3D point.
*/
public interface Point { public interface Point {
/**
* Gets the X coordinate.
*
* @return the X coordinate
*/
@Contract(pure = true) @Contract(pure = true)
double x(); double x();
/**
* Gets the Y coordinate.
*
* @return the Y coordinate
*/
@Contract(pure = true) @Contract(pure = true)
double y(); double y();
/**
* Gets the Z coordinate.
*
* @return the Z coordinate
*/
@Contract(pure = true) @Contract(pure = true)
double z(); double z();
/**
* Converts all coordinates to integers.
*
* @return a new point representing a block position
*/
@Contract(pure = true)
default @NotNull Point asBlockPosition() {
final int castedY = (int) y();
return Vec.vec((int) Math.floor(x()),
(y() == castedY) ? castedY : castedY + 1,
(int) Math.floor(z()));
}
} }

View File

@ -3,11 +3,21 @@ package net.minestom.server.utils.incubator;
import org.jetbrains.annotations.Contract; import org.jetbrains.annotations.Contract;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
public interface Pos extends Point { public final class Pos implements Point {
private final double x, y, z;
private final float yaw, pitch;
private Pos(double x, double y, double z, float yaw, float pitch) {
this.x = x;
this.y = y;
this.z = z;
this.yaw = yaw;
this.pitch = pitch;
}
@Contract(pure = true) @Contract(pure = true)
static @NotNull Pos pos(double x, double y, double z, float yaw, float pitch) { static @NotNull Pos pos(double x, double y, double z, float yaw, float pitch) {
return new PosImpl(x, y, z, yaw, pitch); return new Pos(x, y, z, yaw, pitch);
} }
@Contract(pure = true) @Contract(pure = true)
@ -17,7 +27,7 @@ public interface Pos extends Point {
@Contract(pure = true) @Contract(pure = true)
static @NotNull Pos pos(double x, double y, double z) { static @NotNull Pos pos(double x, double y, double z) {
return new PosImpl(x, y, z); return new Pos(x, y, z, 0, 0);
} }
@Contract(pure = true) @Contract(pure = true)
@ -26,24 +36,47 @@ public interface Pos extends Point {
} }
@Contract(pure = true) @Contract(pure = true)
@NotNull Pos withCoord(double x, double y, double z); public @NotNull Pos withCoord(double x, double y, double z) {
return new Pos(x, y, z, yaw, pitch);
}
@Contract(pure = true) @Contract(pure = true)
default @NotNull Pos withCoord(@NotNull Vec vec) { public @NotNull Pos withCoord(@NotNull Vec vec) {
return withCoord(vec.x(), vec.y(), vec.z()); return withCoord(vec.x(), vec.y(), vec.z());
} }
@Contract(pure = true) @Contract(pure = true)
@NotNull Pos withView(float yaw, float pitch); public @NotNull Pos withView(float yaw, float pitch) {
return new Pos(x, y, z, yaw, pitch);
}
@Override
public double x() {
return x;
}
@Override
public double y() {
return y;
}
@Override
public double z() {
return z;
}
@Contract(pure = true) @Contract(pure = true)
float yaw(); public float yaw() {
return yaw;
}
@Contract(pure = true) @Contract(pure = true)
float pitch(); public float pitch() {
return pitch;
}
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec asVec() { public @NotNull Vec asVec() {
return Vec.vec(x(), y(), z()); return Vec.vec(x(), y(), z());
} }
} }

View File

@ -1,55 +0,0 @@
package net.minestom.server.utils.incubator;
import org.jetbrains.annotations.NotNull;
class PosImpl implements Pos {
private final double x, y, z;
private final float yaw, pitch;
PosImpl(double x, double y, double z, float yaw, float pitch) {
this.x = x;
this.y = y;
this.z = z;
this.yaw = yaw;
this.pitch = pitch;
}
PosImpl(double x, double y, double z) {
this(x, y, z, 0, 0);
}
@Override
public double x() {
return x;
}
@Override
public double y() {
return y;
}
@Override
public double z() {
return z;
}
@Override
public @NotNull Pos withCoord(double x, double y, double z) {
return new PosImpl(x, y, z, yaw, pitch);
}
@Override
public @NotNull Pos withView(float yaw, float pitch) {
return new PosImpl(x, y, z, yaw, pitch);
}
@Override
public float yaw() {
return yaw;
}
@Override
public float pitch() {
return pitch;
}
}

View File

@ -4,15 +4,24 @@ import net.minestom.server.utils.MathUtils;
import org.jetbrains.annotations.Contract; import org.jetbrains.annotations.Contract;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import java.util.Objects;
import java.util.function.DoubleUnaryOperator; import java.util.function.DoubleUnaryOperator;
import java.util.function.UnaryOperator; import java.util.function.UnaryOperator;
/** /**
* Represents an immutable 3D vector. * Represents an immutable 3D vector.
*/ */
public interface Vec extends Point { public final class Vec implements Point {
Vec ZERO = vec(0); public static final Vec ZERO = vec(0);
Vec ONE = vec(1); public static final Vec ONE = vec(1);
private final double x, y, z;
private Vec(double x, double y, double z) {
this.x = x;
this.y = y;
this.z = z;
}
/** /**
* Creates a new vec with the 3 coordinates set. * Creates a new vec with the 3 coordinates set.
@ -23,8 +32,8 @@ public interface Vec extends Point {
* @return the created vec * @return the created vec
*/ */
@Contract(pure = true) @Contract(pure = true)
static @NotNull Vec vec(double x, double y, double z) { public static @NotNull Vec vec(double x, double y, double z) {
return new VecImpl.Vec3(x, y, z); return new Vec(x, y, z);
} }
/** /**
@ -35,8 +44,8 @@ public interface Vec extends Point {
* @return the created vec * @return the created vec
*/ */
@Contract(pure = true) @Contract(pure = true)
static @NotNull Vec vec(double x, double z) { public static @NotNull Vec vec(double x, double z) {
return new VecImpl.Tuple(x, z); return new Vec(x, 0, z);
} }
/** /**
@ -46,21 +55,10 @@ public interface Vec extends Point {
* @return the created vec * @return the created vec
*/ */
@Contract(pure = true) @Contract(pure = true)
static @NotNull Vec vec(double value) { public static @NotNull Vec vec(double value) {
return new VecImpl.Single(value); return new Vec(value, value, value);
} }
/**
* Creates a new vec of the same type with the specified coordinates.
*
* @param x the X coordinate
* @param y the Y coordinate
* @param z the Z coordinate
* @return the creates vec
*/
@Contract(pure = true)
@NotNull Vec with(double x, double y, double z);
/** /**
* Creates a new vec with coordinated depending on {@code this}. * Creates a new vec with coordinated depending on {@code this}.
* *
@ -68,135 +66,127 @@ public interface Vec extends Point {
* @return the created vec * @return the created vec
*/ */
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec with(@NotNull Operator operator) { public @NotNull Vec with(@NotNull Operator operator) {
return operator.apply(x(), y(), z()); return operator.apply(x(), y(), z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec withX(@NotNull DoubleUnaryOperator operator) { public @NotNull Vec withX(@NotNull DoubleUnaryOperator operator) {
return with(operator.applyAsDouble(x()), y(), z()); return new Vec(operator.applyAsDouble(x()), y(), z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec withX(double x) { public @NotNull Vec withX(double x) {
return with(x, y(), z()); return new Vec(x, y(), z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec withY(@NotNull DoubleUnaryOperator operator) { public @NotNull Vec withY(@NotNull DoubleUnaryOperator operator) {
return with(x(), operator.applyAsDouble(y()), z()); return new Vec(x(), operator.applyAsDouble(y()), z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec withY(double y) { public @NotNull Vec withY(double y) {
return with(x(), y, z()); return new Vec(x(), y, z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec withZ(@NotNull DoubleUnaryOperator operator) { public @NotNull Vec withZ(@NotNull DoubleUnaryOperator operator) {
return with(x(), y(), operator.applyAsDouble(z())); return new Vec(x(), y(), operator.applyAsDouble(z()));
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec withZ(double z) { public @NotNull Vec withZ(double z) {
return with(x(), y(), z); return new Vec(x(), y(), z);
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec add(@NotNull Vec vec) { public @NotNull Vec add(@NotNull Vec vec) {
return with(x() + vec.x(), y() + vec.y(), z() + vec.z()); return new Vec(x() + vec.x(), y() + vec.y(), z() + vec.z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec add(double value) { public @NotNull Vec add(double value) {
return with(x() + value, y() + value, z() + value); return new Vec(x() + value, y() + value, z() + value);
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec sub(@NotNull Vec vec) { public @NotNull Vec sub(@NotNull Vec vec) {
return with(x() - vec.x(), y() - vec.y(), z() - vec.z()); return new Vec(x() - vec.x(), y() - vec.y(), z() - vec.z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec sub(double value) { public @NotNull Vec sub(double value) {
return with(x() - value, y() - value, z() - value); return new Vec(x() - value, y() - value, z() - value);
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec mul(@NotNull Vec vec) { public @NotNull Vec mul(@NotNull Vec vec) {
return with(x() * vec.x(), y() * vec.y(), z() * vec.z()); return new Vec(x() * vec.x(), y() * vec.y(), z() * vec.z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec mul(double value) { public @NotNull Vec mul(double value) {
return with(x() * value, y() * value, z() * value); return new Vec(x() * value, y() * value, z() * value);
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec div(@NotNull Vec vec) { public @NotNull Vec div(@NotNull Vec vec) {
return with(x() / vec.x(), y() / vec.y(), z() / vec.z()); return new Vec(x() / vec.x(), y() / vec.y(), z() / vec.z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec div(double value) { public @NotNull Vec div(double value) {
return with(x() / value, y() / value, z() / value); return new Vec(x() / value, y() / value, z() / value);
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec neg() { public @NotNull Vec neg() {
return with(-x(), -y(), -z()); return new Vec(-x(), -y(), -z());
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec abs() { public @NotNull Vec abs() {
return with(Math.abs(x()), Math.abs(y()), Math.abs(z())); return new Vec(Math.abs(x()), Math.abs(y()), Math.abs(z()));
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec min(@NotNull Vec vec) { public @NotNull Vec min(@NotNull Vec vec) {
return with(Math.min(x(), vec.x()), Math.min(y(), vec.y()), Math.min(z(), vec.z())); return new Vec(Math.min(x(), vec.x()), Math.min(y(), vec.y()), Math.min(z(), vec.z()));
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec min(double value) { public @NotNull Vec min(double value) {
return with(Math.min(x(), value), Math.min(y(), value), Math.min(z(), value)); return new Vec(Math.min(x(), value), Math.min(y(), value), Math.min(z(), value));
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec max(@NotNull Vec vec) { public @NotNull Vec max(@NotNull Vec vec) {
return with(Math.max(x(), vec.x()), Math.max(y(), vec.y()), Math.max(z(), vec.z())); return new Vec(Math.max(x(), vec.x()), Math.max(y(), vec.y()), Math.max(z(), vec.z()));
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec max(double value) { public @NotNull Vec max(double value) {
return with(Math.max(x(), value), Math.max(y(), value), Math.max(z(), value)); return new Vec(Math.max(x(), value), Math.max(y(), value), Math.max(z(), value));
} }
@Contract(pure = true) @Contract(pure = true)
default Vec apply(@NotNull UnaryOperator<@NotNull Vec> operator) { public Vec apply(@NotNull UnaryOperator<@NotNull Vec> operator) {
return operator.apply(this); return operator.apply(this);
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Pos asPosition() { public @NotNull Pos asPosition() {
return Pos.pos(this); return Pos.pos(this);
} }
@Contract(pure = true)
default @NotNull Vec asBlockPosition() {
final int castedY = (int) y();
return with((int) Math.floor(x()),
(y() == castedY) ? castedY : castedY + 1,
(int) Math.floor(z()));
}
/** /**
* Gets the magnitude of the vector squared. * Gets the magnitude of the vector squared.
* *
* @return the magnitude * @return the magnitude
*/ */
@Contract(pure = true) @Contract(pure = true)
default double lengthSquared() { public double lengthSquared() {
return MathUtils.square(x()) + MathUtils.square(y()) + MathUtils.square(z()); return MathUtils.square(x()) + MathUtils.square(y()) + MathUtils.square(z());
} }
@ -210,7 +200,7 @@ public interface Vec extends Point {
* @return the magnitude * @return the magnitude
*/ */
@Contract(pure = true) @Contract(pure = true)
default double length() { public double length() {
return Math.sqrt(lengthSquared()); return Math.sqrt(lengthSquared());
} }
@ -220,9 +210,9 @@ public interface Vec extends Point {
* @return the same vector * @return the same vector
*/ */
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec normalize() { public @NotNull Vec normalize() {
final double length = length(); final double length = length();
return with(x() / length, y() / length, z() / length); return new Vec(x() / length, y() / length, z() / length);
} }
/** /**
@ -236,7 +226,7 @@ public interface Vec extends Point {
* @return the distance * @return the distance
*/ */
@Contract(pure = true) @Contract(pure = true)
default double distance(@NotNull Vec vec) { public double distance(@NotNull Vec vec) {
return Math.sqrt(MathUtils.square(x() - vec.x()) + return Math.sqrt(MathUtils.square(x() - vec.x()) +
MathUtils.square(y() - vec.y()) + MathUtils.square(y() - vec.y()) +
MathUtils.square(z() - vec.z())); MathUtils.square(z() - vec.z()));
@ -249,7 +239,7 @@ public interface Vec extends Point {
* @return the squared distance * @return the squared distance
*/ */
@Contract(pure = true) @Contract(pure = true)
default double distanceSquared(@NotNull Vec vec) { public double distanceSquared(@NotNull Vec vec) {
return MathUtils.square(x() - vec.x()) + return MathUtils.square(x() - vec.x()) +
MathUtils.square(y() - vec.y()) + MathUtils.square(y() - vec.y()) +
MathUtils.square(z() - vec.z()); MathUtils.square(z() - vec.z());
@ -262,7 +252,7 @@ public interface Vec extends Point {
* @return angle in radians * @return angle in radians
*/ */
@Contract(pure = true) @Contract(pure = true)
default double angle(@NotNull Vec vec) { public double angle(@NotNull Vec vec) {
final double dot = MathUtils.clamp(dot(vec) / (length() * vec.length()), -1.0, 1.0); final double dot = MathUtils.clamp(dot(vec) / (length() * vec.length()), -1.0, 1.0);
return Math.acos(dot); return Math.acos(dot);
} }
@ -275,7 +265,7 @@ public interface Vec extends Point {
* @return dot product * @return dot product
*/ */
@Contract(pure = true) @Contract(pure = true)
default double dot(@NotNull Vec vec) { public double dot(@NotNull Vec vec) {
return x() * vec.x() + y() * vec.y() + z() * vec.z(); return x() * vec.x() + y() * vec.y() + z() * vec.z();
} }
@ -292,8 +282,8 @@ public interface Vec extends Point {
* @return the same vector * @return the same vector
*/ */
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec cross(@NotNull Vec o) { public @NotNull Vec cross(@NotNull Vec o) {
return with(y() * o.z() - o.y() * z(), return new Vec(y() * o.z() - o.y() * z(),
z() * o.x() - o.z() * x(), z() * o.x() - o.z() * x(),
x() * o.y() - o.x() * y()); x() * o.y() - o.x() * y());
} }
@ -307,32 +297,55 @@ public interface Vec extends Point {
* @return Linear interpolated vector * @return Linear interpolated vector
*/ */
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec lerp(@NotNull Vec vec, double alpha) { public @NotNull Vec lerp(@NotNull Vec vec, double alpha) {
final double x = x(); final double x = x();
final double y = y(); final double y = y();
final double z = z(); final double z = z();
return with(x + (alpha * (vec.x() - x)), return new Vec(x + (alpha * (vec.x() - x)),
y + (alpha * (vec.y() - y)), y + (alpha * (vec.y() - y)),
z + (alpha * (vec.z() - z))); z + (alpha * (vec.z() - z)));
} }
@Contract(pure = true) @Contract(pure = true)
default @NotNull Vec interpolate(@NotNull Vec target, double alpha, @NotNull Interpolation interpolation) { public @NotNull Vec interpolate(@NotNull Vec target, double alpha, @NotNull Interpolation interpolation) {
return lerp(target, interpolation.apply(alpha)); return lerp(target, interpolation.apply(alpha));
} }
@Contract(pure = true) @Override
default boolean eq(@NotNull Vec vec) { public double x() {
return VecImpl.equals(this, vec); return x;
}
@Override
public double y() {
return y;
}
@Override
public double z() {
return z;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Vec vec = (Vec) o;
return Double.compare(vec.x, x) == 0 && Double.compare(vec.y, y) == 0 && Double.compare(vec.z, z) == 0;
}
@Override
public int hashCode() {
return Objects.hash(x, y, z);
} }
@FunctionalInterface @FunctionalInterface
interface Operator { public interface Operator {
@NotNull Vec apply(double x, double y, double z); @NotNull Vec apply(double x, double y, double z);
} }
@FunctionalInterface @FunctionalInterface
interface Interpolation { public interface Interpolation {
Interpolation LINEAR = a -> a; Interpolation LINEAR = a -> a;
Interpolation SMOOTH = a -> a * a * (3 - 2 * a); Interpolation SMOOTH = a -> a * a * (3 - 2 * a);

View File

@ -1,160 +0,0 @@
package net.minestom.server.utils.incubator;
import org.jetbrains.annotations.NotNull;
import java.util.Objects;
final class VecImpl {
static final class Vec3 implements Vec {
private final double x, y, z;
Vec3(double x, double y, double z) {
this.x = x;
this.y = y;
this.z = z;
}
@Override
public final @NotNull Vec with(double x, double y, double z) {
return new Vec3(x, y, z);
}
@Override
public final double x() {
return x;
}
@Override
public final double y() {
return y;
}
@Override
public final double z() {
return z;
}
@Override
public boolean equals(Object o) {
return VecImpl.equals(this, o);
}
@Override
public int hashCode() {
return VecImpl.hashCode(this);
}
@Override
public String toString() {
return VecImpl.toString(this);
}
}
static final class Tuple implements Vec {
private final double x, z;
Tuple(double x, double z) {
this.x = x;
this.z = z;
}
@Override
public final @NotNull Vec with(double x, double y, double z) {
return new Vec3(x, y, z);
}
@Override
public final double x() {
return x;
}
@Override
public final double y() {
return 0;
}
@Override
public final double z() {
return z;
}
@Override
public boolean equals(Object o) {
return VecImpl.equals(this, o);
}
@Override
public int hashCode() {
return VecImpl.hashCode(this);
}
@Override
public String toString() {
return VecImpl.toString(this);
}
}
static final class Single implements Vec {
private final double value;
Single(double value) {
this.value = value;
}
@Override
public final @NotNull Vec with(double x, double y, double z) {
return new Vec3(x, y, z);
}
@Override
public final double x() {
return value;
}
@Override
public final double y() {
return value;
}
@Override
public final double z() {
return value;
}
@Override
public boolean equals(Object o) {
return VecImpl.equals(this, o);
}
@Override
public int hashCode() {
return VecImpl.hashCode(this);
}
@Override
public String toString() {
return VecImpl.toString(this);
}
}
static boolean equals(@NotNull Vec vec1, Object o) {
if (vec1 == o) return true;
if (!(o instanceof Vec)) return false;
Vec vec2 = (Vec) o;
return Double.compare(vec1.x(), vec2.x()) == 0 &&
Double.compare(vec1.y(), vec2.y()) == 0 &&
Double.compare(vec1.z(), vec2.z()) == 0;
}
static int hashCode(@NotNull Vec vec) {
return Objects.hash(vec.x(), vec.y(), vec.z());
}
static @NotNull String toString(@NotNull Vec vec) {
return "Vec3{" +
"x=" + vec.x() +
", y=" + vec.y() +
", z=" + vec.z() +
'}';
}
}

View File

@ -158,8 +158,7 @@ public class PlayerInit {
.withZ(operand -> operand + 5) .withZ(operand -> operand + 5)
.withY(20) .withY(20)
.mul(Vec.vec(5, 5, 5)) .mul(Vec.vec(5, 5, 5))
.mul(5) .mul(5);
.asBlockPosition();
UnaryOperator<Vec> blockTransform2 = vec -> Vec.vec(vec.x() * 2, 20, vec.z() + 5); UnaryOperator<Vec> blockTransform2 = vec -> Vec.vec(vec.x() * 2, 20, vec.z() + 5);
var vec = Vec.ZERO.apply(blockTransform).add(Vec.vec(5)); var vec = Vec.ZERO.apply(blockTransform).add(Vec.vec(5));
var vec2 = Vec.ZERO.with((x, y, z) -> Vec.vec(x + 5, y / 5, z * 5)); var vec2 = Vec.ZERO.with((x, y, z) -> Vec.vec(x + 5, y / 5, z * 5));