HoHCI-BikeSim/Assets/Scripts/Controller/SensorBikeController.cs

using System;
using Controller.Bicycle;
using Sensors;
using Sensors.ANT;
using Sensors.Polar;
using Tracking;
using UnityEngine;
using UnityEngine.Serialization;

namespace Controller
{
    [Serializable]
    public struct PolarRotationMapping
    {
        public float maxRight;
        public float center;
    }

    [Serializable]
    public struct FrontWheelTrackerConfig
    {
        public FrontWheelTracker frontWheelTracker;
        public float multiplicator;

        public Vector3 AdjustedRotation => frontWheelTracker.Rotation * multiplicator;
    }

    public class SensorBikeController : MonoBehaviour
    {
        public PolarRotationMapping polarRotationMapping;

        public SpeedSensorConfig speedSensorConfig;
        public PolarSensorConfig polarSensorConfig;
        public FrontWheelTrackerConfig frontWheelTrackerConfig;

        public bool steer = true;
        public bool accelerate = true;
        public bool lean = true;

        private WcBicycleController wcBicycleController;
        private BikeSensorData sensorData;
        private float leanFactor;
        private bool isFrontWheelTrackerNotNull;

        private async void Start()
        {
            isFrontWheelTrackerNotNull = frontWheelTrackerConfig.frontWheelTracker != null;
            wcBicycleController = GetComponent<WcBicycleController>();
            sensorData = BikeSensorData.Instance;
            await sensorData.StartListening(polarSensorConfig: polarSensorConfig, speedSensorConfig: speedSensorConfig);
            leanFactor = 90f / (polarRotationMapping.maxRight - polarRotationMapping.center);
        }

        private void Update()
        {
            var speedData = sensorData.SpeedData;
            var polarData = sensorData.PolarData;

            if (speedData != null && accelerate)
            {
                SetSpeed(speedData.Value);
            }

            if (isFrontWheelTrackerNotNull && steer)
            {
                SetSteer();
            }

            if (polarData != null && lean)
            {
                SetLeaningAngle(polarData.Value);
            }
        }

        private void SetSteer()
        {
            wcBicycleController.CurrentSteerAngle = frontWheelTrackerConfig.AdjustedRotation.y; //TODO: something a bit smarter
        }

        private void OnDestroy()
        {
            sensorData?.Dispose();
        }

        private void SetLeaningAngle(PolarSensorData polarData)
        {
            //don't lean while standing / walking to bike
            if (wcBicycleController.rigidBody.velocity.magnitude > .5f)
            {
                wcBicycleController.CurrentLeaningAngle = (polarData.Acc.y - polarRotationMapping.center) * leanFactor;
            }
        }

        private void SetSpeed(SpeedSensorData speedData)
        {
            wcBicycleController.CurrentSpeed = speedData.Speed;
            
            
        }
    }
}