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

using System;
using System.Collections.Generic;
using Controller.Bicycle;
using Controller.Lean;
using Study;
using Sensors;
using Sensors.ANT;
using Sensors.Bluetooth;
using Tracking;
using System.Linq;
using UnityEngine;

// Declare outisde for public visibility
//public enum SteeringMode { frontWheel, Leaning, HMD };

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

        public float AdjustedRotation => frontWheelTracker.SteerRotation * multiplicator;
    }

    [Serializable]
    public struct HMDTrackerConfig
    {
        public HMDTracker cameraTracker;
        public float multiplicator;
        public float steeringOffset;

        public float AdjustedRotation => cameraTracker.SteerRotation * multiplicator;
        public float AdjustedLean => cameraTracker.LeanRotation * multiplicator;
    }

    [RequireComponent(typeof(IBicycleController))]
    public class SensorBikeController : MonoBehaviour
    {
        public PolarRotationMapping polarRotationMapping;
        public FrontWheelTrackerConfig frontWheelTrackerConfig;
        public HMDTrackerConfig hmdTrackerConfig;
        private float leanFactor;

        public bool steer = true;
        public bool accelerate = true;
        public bool lean = true;
        public Study.SteeringMode steeringSelection;

        private IBicycleController bicycleController;
        private bool isFrontWheelTrackerNotNull;
        private bool isHMDTrackerNotNull;
        private BikeSensorData sensorData;

        private Queue<float> previousLeanValues;
        public int framesToConsider = 10;
        public float tolerance = 5;

        private GameObject bikePlayer;
        private void Start()
        {
            isFrontWheelTrackerNotNull = frontWheelTrackerConfig.frontWheelTracker != null;
            isHMDTrackerNotNull = hmdTrackerConfig.cameraTracker != null;
            bicycleController = GetComponent<IBicycleController>();
            sensorData = BikeSensorData.Instance;
            //leanFactor = 90f / (polarRotationMapping.maxRight - polarRotationMapping.center);
            leanFactor = 45f / polarRotationMapping.maxRight;
            if (framesToConsider > 1) previousLeanValues = new Queue<float>(framesToConsider);

            // Sort of callibration for polar Sensor
            var polarData = sensorData.BleData;
            if (steeringSelection == Study.SteeringMode.Leaning && polarData != null)
            {
                Debug.Log("Calibration of Polar Sensor");
                polarRotationMapping.center = polarData.Value.Acc.y;
            }

            bikePlayer = GameObject.Find("bike");
        }

        private void Update()
        {
            Debug.Log("Bike Sensor Controller called");
            var speedData = sensorData.SpeedData;

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

            if (steer) SetSteer();

            if (lean) SetLeaningAngle();

            var polarData = sensorData.BleData;
            if (polarData != null) 
            {
                var playerStats = bikePlayer.GetComponent<PlayerStats>();
                playerStats.rotationPolarX = polarData.Value.Acc.x;
                playerStats.rotationPolarY = polarData.Value.Acc.y;
                playerStats.rotationPolarZ = polarData.Value.Acc.z;
            }
        }

        public void SetCondition(Study.SteeringMode mode)
        {
            Debug.Log("Setting Condition");
            steeringSelection = mode;
        }

        private void SetSteer()
        {
            switch (steeringSelection)
            {
                case Study.SteeringMode.frontWheel:
                    if (isFrontWheelTrackerNotNull)
                    {
                        bicycleController.CurrentSteerAngle = frontWheelTrackerConfig.AdjustedRotation * 2f;
                    }
                    break;
                case Study.SteeringMode.Leaning:
                    var polarData = sensorData.BleData;
                    if (polarData != null)
                    {
                        Debug.Log("Updating Polar Data");
                        Debug.Log("Polar Y: " + polarData.Value.Acc);
                        var polarLean = CalculateLeanRotationMultiFrame(polarData);
                        Debug.Log("Ploar Post-Processed: " + polarLean);
                        bicycleController.CurrentSteerAngle = polarLean;
                        // Activate below in case we also need the steering angle
                        //bicycleController.CurrentSteerAngle = (polarData.Value.Acc.y - polarRotationMapping.center) * leanFactor;
                    }
                    break;
                case Study.SteeringMode.HMD:
                    if (isHMDTrackerNotNull)
                    {
                        // We emprirically observed a right-drift
                        // We subtract a constant to counteract this
                        bicycleController.CurrentSteerAngle = hmdTrackerConfig.AdjustedRotation;
                    }
                    break;
            }
            Debug.Log("Updating Steering Angle to " + bicycleController.CurrentSteerAngle);
        }

        private void SetLeaningAngle()
        {
            //bicycleController.CurrentLeaningAngle = 
        }

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

        // Helpers for Lean Steering
        private float CalculateLeanRotationSingleFrame(BleSensorData? polarData)
        {
            float steerUpdate = polarData.Value.Acc.y - polarRotationMapping.center;
            //steerUpdate = Math.Abs(steerUpdate - polarRotationMapping.center) < tolerance ? 0 : steerUpdate;
            return steerUpdate * leanFactor;
        }

        private float CalculateLeanRotationMultiFrame(BleSensorData? polarData)
        {
            var angleThisFrame = CalculateLeanRotationSingleFrame(polarData);
            if (previousLeanValues.Count > framesToConsider) previousLeanValues.Dequeue();

            previousLeanValues.Enqueue(angleThisFrame);

            // Use exponential moving average
            float alpha = 2f / (framesToConsider + 1);
            float steerUpdate = previousLeanValues.DefaultIfEmpty()
             .Aggregate((ema, nextQuote) => alpha * nextQuote + (1 - alpha) * ema);

            // Do Global Thresholding, to avoid noisy steering
            return Math.Abs(steerUpdate - polarRotationMapping.center) < tolerance ? 0 : steerUpdate;
            //return steerUpdate;
        }
    }
}