keep firework velocity when moving from still

Author: beanbag44

src/main/kotlin/com/lambda/module/modules/movement/elytrafly/ElytraFly.kt

@@ -105,6 +105,19 @@ object ElytraFly : Module(
         }
     }
 
+    @JvmStatic
+    fun getFireworkTargetVelocity(rotPitch: Float, rotYaw: Float): Vec3d {
+        var targetSpeed = 1.5 * rocketSpeed
+        if (angledRocketBoost) {
+            val scale = sin(Math.toRadians(abs(rotPitch) * 2.0))
+            targetSpeed = (1.5 * rocketSpeed) + (maxAngledBoost * scale)
+        }
+        val vec = Vec3d.fromPolar(rotPitch, rotYaw)
+        val d = targetSpeed
+        val e = 0.1 * rocketSpeed
+        return vec.multiply(d + e * 2)
+    }
+
     @JvmStatic
     fun boostRocket() =
         runSafe {

src/main/kotlin/com/lambda/module/modules/movement/elytrafly/modes/GrimControlElytraFly.kt

@@ -27,6 +27,7 @@ import com.lambda.interaction.managers.rotating.IRotationRequest.Companion.rotat
 import com.lambda.interaction.managers.rotating.RotationManager
 import com.lambda.interaction.material.StackSelection.Companion.select
 import com.lambda.module.modules.movement.BetterFirework.startFirework
+import com.lambda.module.modules.movement.elytrafly.ElytraFly
 import com.lambda.module.modules.movement.elytrafly.ElytraFly.FlyMode
 import com.lambda.module.modules.movement.elytrafly.ElytraFlyMode
 import com.lambda.module.modules.render.Freecam
@@ -121,7 +122,10 @@ class GrimControlElytraFly(
 			rotationRequest { rotation(rot.y, rot.x) }.submit()
 			if (hasFirework) {
 				moving = true
-				if (prevStill) player.velocity = Vec3d.ZERO
+				if (prevStill) {
+					val activeRot = RotationManager.activeRotation
+					player.velocity = ElytraFly.getFireworkTargetVelocity(activeRot.pitchF, activeRot.yawF)
+				}
 			}
 		}