using System.Collections;

using System.Collections.Generic;

using UnityEngine;

​

namespace Drone

{

    [RequireComponent(typeof(BoxCollider))]

    public class DroneEngine : MonoBehaviour, IEngine

    {

        #region Variables

        [Header("Engine Properties")]

        [SerializeField] private float maxPower = 4f;

        [SerializeField] private float speed = 1.25f;

​

        [Header("Propeller Properties")]

        [SerializeField] private Transform propeller;

        [SerializeField] private float propRotSpeed = 300f;

        #endregion

​

        #region Interface Methods

        public void InitEngine()

        {

        }

​

        public void UpdateEngine(Rigidbody rb, DroneInputs input)

        {

            // calculate difference between rotated up and upwards force needed to hover

            Vector3 upVec = transform.up;

            upVec.x = 0f;

            upVec.z = 0f;

            float diff = 1 - upVec.magnitude;

            float finalDiff = Physics.gravity.magnitude * diff;

​

            // Get forward vector in relation to world space

            Vector3 forVec = transform.forward;

            forVec.y = 0f;

​

            // Get right vector in relation to world space

            Vector3 rightVec = transform.right;

            rightVec.y = 0f;

​

            Vector3 engineForce = transform.up * ((rb.mass * Physics.gravity.magnitude + finalDiff / 1.9f) + (input.Throttle * maxPower)) / 4f;

            engineForce = engineForce + (forVec * input.Cyclic.y * speed) + (rightVec * input.Cyclic.x * speed); // add horizontal speed

​

            rb.AddForce(engineForce, ForceMode.Force);

            HandlePropellers();

        }

​

        void HandlePropellers()

        {

            if (!propeller) return;

​

            propeller.Rotate(Vector3.up, propRotSpeed);

        }

        #endregion

    }