123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124 |
- using System;
- using System.Collections.Generic;
- using System.Data;
- using System.Threading.Tasks;
- using ANT_Managed_Library;
- using JetBrains.Annotations;
- using Sensors.ANT;
- using Sensors.Polar;
- using UniRx;
- using UnityEngine;
- namespace Sensors
- {
- public sealed class BikeSensorData
- {
- private readonly Subject<object> polarReceiverAvailable = new Subject<object>();
- [CanBeNull] private HrReceiver hrReceiver;
- [CanBeNull] private PolarReceiver polarReceiver;
- [CanBeNull] private PowerMeterReceiver powerMeterReceiver;
- [CanBeNull] private SpeedSensorReceiver speedSensorReceiver;
- private BikeSensorData()
- {
- PolarReceiverAvailable = polarReceiverAvailable.ToTask();
- }
- public SpeedSensorData? SpeedData =>
- speedSensorReceiver?.Connected ?? false ? speedSensorReceiver?.SensorData : null;
- public PowermeterSensorData? PowermeterData =>
- powerMeterReceiver?.Connected ?? false ? powerMeterReceiver?.SensorData : null;
- public HrSensorData? HrData => hrReceiver?.Connected ?? false ? hrReceiver?.SensorData : null;
- public PolarSensorData? PolarData => polarReceiver?.SensorData;
- [CanBeNull] public IObservable<EcgData> RawEcgData => polarReceiver?.RawEcgData;
- [CanBeNull] public IObservable<AccData> RawAccData => polarReceiver?.RawAccData;
- public PolarSensorConfig? PolarConfig => polarReceiver?.SensorConfig;
- public Task PolarReceiverAvailable { get; }
- public void StartListening(SpeedSensorConfig? speedSensorConfig = null,
- PolarSensorConfig? polarSensorConfig = null, int? powermeterId = null, int? hrAntId = null)
- {
- var awaitDevices = new List<AwaitDevice>();
- if (speedSensorConfig != null)
- {
- if (speedSensorReceiver != null)
- throw new InvalidConstraintException(
- "BikeSensorData: Already listening to Speed Sensor");
- speedSensorReceiver = new SpeedSensorReceiver(speedSensorConfig.Value);
- awaitDevices.Add(speedSensorReceiver);
- }
- if (hrAntId != null)
- {
- if (hrReceiver != null)
- throw new InvalidConstraintException(
- "BikeSensorData: Already listening to HR Sensor");
- hrReceiver = new HrReceiver(hrAntId.Value);
- awaitDevices.Add(hrReceiver);
- }
- if (powermeterId != null)
- {
- if (powerMeterReceiver != null)
- throw new InvalidConstraintException(
- "BikeSensorData: Already listening to Power Sensor");
- powerMeterReceiver = new PowerMeterReceiver(powermeterId.Value);
- //awaitDevices.Add(powerMeterReceiver);
- }
- if (awaitDevices.Count > 0)
- {
- try
- {
- AntDevices.Instance.StartScan(awaitDevices);
- }
- catch (Exception e)
- {
- Debug.Log($"Could not initialize Ant Devices: {e}");
- hrReceiver = null;
- powerMeterReceiver = null;
- speedSensorReceiver = null;
- }
- }
- if (polarSensorConfig != null)
- {
- if (polarReceiver != null)
- throw new InvalidConstraintException(
- "BikeSensorData: Already listening to Polar Sensor");
- polarReceiver = new PolarReceiver(polarSensorConfig.Value);
- polarReceiver.StartListening();
- polarReceiverAvailable.OnNext(null);
- polarReceiverAvailable.OnCompleted();
- }
- }
- public void Dispose()
- {
- polarReceiver?.Dispose();
- //TODO: also dispose ANT sensors?
- polarReceiverAvailable.Dispose();
- polarReceiver = null;
- speedSensorReceiver = null;
- powerMeterReceiver = null;
- hrReceiver = null;
- }
- #region singleton
- private static readonly Lazy<BikeSensorData>
- lazy =
- new Lazy<BikeSensorData>
- (() => new BikeSensorData());
- public static BikeSensorData Instance => lazy.Value;
- #endregion
- }
- }
|