diff --git a/src/main/java/net/minestom/server/collision/Aerodynamics.java b/src/main/java/net/minestom/server/collision/Aerodynamics.java index 954b8cdbc..d36a75cfc 100644 --- a/src/main/java/net/minestom/server/collision/Aerodynamics.java +++ b/src/main/java/net/minestom/server/collision/Aerodynamics.java @@ -19,6 +19,11 @@ public record Aerodynamics(double gravity, double horizontalAirResistance, doubl return new Aerodynamics(gravity, horizontalAirResistance, verticalAirResistance); } + @Contract(pure = true) + public @NotNull Aerodynamics withGravity(@NotNull DoubleUnaryOperator operator) { + return withHorizontalAirResistance(operator.apply(gravity)); + } + @Contract(pure = true) public @NotNull Aerodynamics withHorizontalAirResistance(double horizontalAirResistance) { return new Aerodynamics(gravity, horizontalAirResistance, verticalAirResistance); @@ -40,7 +45,7 @@ public record Aerodynamics(double gravity, double horizontalAirResistance, doubl } @Contract(pure = true) - public @NotNull Aerodynamics withAirResistance(double horizontal, double vertical) { + public @NotNull Aerodynamics withAirResistance(double horizontalAirResistance, double verticalAirResistance) { return new Aerodynamics(gravity, horizontalAirResistance, verticalAirResistance); } }