HoHCI-BikeSteering/Assets/Scripts/Controller/BicycleController.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using UnityEngine.EventSystems;
using UnityEngine.Serialization;
using Valve.VR.InteractionSystem;

namespace Controller
{
    public class BicycleController : MonoBehaviour
    {
        #region Variables

        [Header("Visible Game Objects")] public GameObject rearWheel;
        public GameObject frontWheel;

        [Header("Game Objects for Physics")] public Transform centerOfMass;
        [FormerlySerializedAs("rb")] public Rigidbody rigidBody;
        public WheelConfig wheelConfig;

        [Header("Values")] public float offsetCollidersFromWheel = 0.25f;
        public float maxLeaningAngle = 35f;
        public float maxSteeringAngle = 80f;
        public float maxMotorTorque = 2000f;
        public float maxBreakTorque = 2000f;

        public float CurrentSteerAngle
        {
            get => currentSteerAngle;
            set => currentSteerAngle = Mathf.Clamp(value, -maxSteeringAngle, maxSteeringAngle);
        }

        public float CurrentMotorTorque
        {
            get => currentMotorTorque;
            set => currentMotorTorque = Mathf.Clamp(value, -maxMotorTorque, maxMotorTorque);
        }

        public float CurrentBrakeTorque
        {
            get => currentBrakeTorque;
            set => currentBrakeTorque = Mathf.Clamp(value, -maxBreakTorque, maxBreakTorque);
        }

        public float CurrentLeaningAngle
        {
            get => currentLeaningAngle;
            set => currentLeaningAngle = Mathf.Clamp(value, -maxLeaningAngle, maxLeaningAngle);
        }

        private WheelCollider[] allWheelColliders;
        private float initialWheelColliderY;
        private float currentSteerAngle = 0f;
        private float currentMotorTorque = 0f;
        private float currentBrakeTorque = 0f;
        private float currentLeaningAngle = 0f;

        #endregion

        // Start is called before the first frame update
        void Start()
        {
            rigidBody.centerOfMass = centerOfMass.localPosition;
            allWheelColliders = wheelConfig.AllWheels;
            wheelConfig.AdjustToGameObjects(frontWheel.transform, rearWheel.transform, offsetCollidersFromWheel);
            initialWheelColliderY = allWheelColliders[0].transform.localPosition.y;
        }


        void FixedUpdate()
        {
            ApplyColliderForces();
            Lean();
            //RotateStraight();
        }

        private void ApplyColliderForces()
        {
            //offsetCollidersFromWheel = Mathf.Max(0.05f, 0.45f - rigidBody.velocity.magnitude * 0.07f);
            //wheelConfig.AdjustToGameObjects(frontWheel.transform, rearWheel.transform, offsetCollidersFromWheel);
            ControlSteer(new[] {wheelConfig.frontLeft, wheelConfig.frontRight});
            ControlTorque(new[] {wheelConfig.rearLeft, wheelConfig.rearRight});
        }

        private void ControlSteer(IEnumerable<WheelCollider> colliders)
        {
            //float steering = maxSteeringAngle * CurrentSteerAngle * 0.2f;
            //leftWheels.localPosition = -Vector3.up * (relativeLeanAmount * CurrentSteerAngle * rigidBody.velocity.magnitude * 0.2f);
            //rightWheels.localPosition = Vector3.up * (relativeLeanAmount * CurrentSteerAngle * rigidBody.velocity.magnitude * 0.2f);
            colliders.ForEach(c => c.steerAngle = CurrentSteerAngle);
        }

        private void ControlTorque(IEnumerable<WheelCollider> colliders)
        {
            foreach (var c in colliders)
            {
                c.motorTorque = CurrentMotorTorque;
                c.brakeTorque = CurrentBrakeTorque;
            }
        }

        private void Lean()
        {
            //reset all wheels to being centered
            if (CurrentLeaningAngle == 0) //TODO: maybe add a threshold for leaning, e.g. < -0.05 and > 0.05 
            {
                //leaning left, left wheels up, right wheels down
                ApplyOffsetToTransform(wheelConfig.frontLeft.transform, initialWheelColliderY);
                ApplyOffsetToTransform(wheelConfig.rearLeft.transform, initialWheelColliderY);
                ApplyOffsetToTransform(wheelConfig.frontRight.transform, initialWheelColliderY);
                ApplyOffsetToTransform(wheelConfig.rearRight.transform, initialWheelColliderY);
            }

            //CurrentLeaningAngle < 0 -> leaning left, > 0 -> right
            var leaningAbs = Mathf.Abs(CurrentLeaningAngle * Mathf.Deg2Rad);

            //calculate offset for wheels; description Docu folder
            //1.57079633 rad = 90 deg
            var verticalOffset = offsetCollidersFromWheel * Mathf.Sin(leaningAbs) / Mathf.Sin(1.57079633f - leaningAbs);
            var yPlusOffset = initialWheelColliderY + verticalOffset;
            var yMinusOffset = initialWheelColliderY - verticalOffset;

            if (CurrentLeaningAngle < 0) //TODO: maybe add a threshold for leaning, e.g. < -0.05 and > 0.05 
            {
                //leaning left, left wheels up, right wheels down
                ApplyOffsetToTransform(wheelConfig.frontLeft.transform, yPlusOffset);
                ApplyOffsetToTransform(wheelConfig.rearLeft.transform, yPlusOffset);
                ApplyOffsetToTransform(wheelConfig.frontRight.transform, yMinusOffset);
                ApplyOffsetToTransform(wheelConfig.rearRight.transform, yMinusOffset);
            }
            else if (CurrentLeaningAngle > 0)
            {
                //leaning right, right wheels up, left wheels down
                ApplyOffsetToTransform(wheelConfig.frontLeft.transform, yMinusOffset);
                ApplyOffsetToTransform(wheelConfig.rearLeft.transform, yMinusOffset);
                ApplyOffsetToTransform(wheelConfig.frontRight.transform, yPlusOffset);
                ApplyOffsetToTransform(wheelConfig.rearRight.transform, yPlusOffset);
            }
        }

        private void ApplyOffsetToTransform(Transform t, float newY)
        {
            var oldPos = t.localPosition;
            t.localPosition = new Vector3(oldPos.x, newY, oldPos.z);
        }
    }
}