Remove compensation for gravity (#13)

PisterLab · Oct 12, 2024 · 80a6d55 · 80a6d55
1 parent 2c49d34
commit 80a6d55
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 10 deletions.
diff --git a/Assets/Scripts/Agent.cs b/Assets/Scripts/Agent.cs
@@ -249,14 +249,8 @@ protected virtual void AlignWithVelocity() {
     }
   }
 
-  protected Vector3 CalculateAcceleration(Vector3 accelerationInput,
-                                          bool compensateForGravity = false) {
+  protected Vector3 CalculateAcceleration(Vector3 accelerationInput) {
     Vector3 gravity = Physics.gravity;
-    if (compensateForGravity) {
-      Vector3 gravityProjection = CalculateGravityProjectionOnPitchAndYaw();
-      accelerationInput -= gravityProjection;
-    }
-
     float airDrag = CalculateDrag();
     float liftInducedDrag = CalculateLiftInducedDrag(accelerationInput + gravity);
     float dragAcceleration = -(airDrag + liftInducedDrag);

diff --git a/Assets/Scripts/Interceptor.cs b/Assets/Scripts/Interceptor.cs
@@ -63,7 +63,7 @@ protected override void UpdateBoost(double deltaTime) {
     Vector3 accelerationInput = boostAcceleration * rollAxis;
 
     // Calculate the total acceleration
-    Vector3 acceleration = CalculateAcceleration(accelerationInput, compensateForGravity: false);
+    Vector3 acceleration = CalculateAcceleration(accelerationInput);
 
     // Apply the acceleration force
     GetComponent<Rigidbody>().AddForce(acceleration, ForceMode.Acceleration);
@@ -97,7 +97,7 @@ protected override void UpdateMidCourse(double deltaTime) {
     }
 
     // Calculate and set the total acceleration
-    Vector3 acceleration = CalculateAcceleration(accelerationInput, compensateForGravity: true);
+    Vector3 acceleration = CalculateAcceleration(accelerationInput);
     GetComponent<Rigidbody>().AddForce(acceleration, ForceMode.Acceleration);
   }
 

diff --git a/Assets/Scripts/Threats/FixedWingThreat.cs b/Assets/Scripts/Threats/FixedWingThreat.cs
@@ -52,7 +52,7 @@ protected override void UpdateMidCourse(double deltaTime) {
     }
 
     // Calculate and set the total acceleration
-    Vector3 acceleration = CalculateAcceleration(accelerationInput, compensateForGravity: true);
+    Vector3 acceleration = CalculateAcceleration(accelerationInput);
     GetComponent<Rigidbody>().AddForce(acceleration, ForceMode.Acceleration);
   }