Shaker/Avalonia/ShakerApp/ViewModels/ShakerDataViewModel.cs

using OxyPlot;
using Shaker.Models;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace ShakerApp.ViewModels
{
    internal class ShakerDataViewModel:ViewModelBase<IModel>
    {
        private NationalInstruments.Tdms.File file;
        private float[,] _Data = new float[0,0];
        private object _datalocker = new object();
        private Dictionary<Shaker.Models.AnalogType, List<(List<DataPoint>,Models.StatisticsModel)>> AnalogDataCache = new Dictionary<AnalogType, List<(List<DataPoint>, Models.StatisticsModel)>>();
        private ShakerDataViewModel()
        {
            GetEvent<AllConfig>().Subscrip((sender, args) =>
            {
                CommunicationViewModel.Instance.LocalCommunication.GetEvent(Topic.DATA)?.Subscrip((sender, args) =>
                {
                    if(args.Data.Length>0 && args.Data[0] is float[,] v)
                    {
                        _Data = v;
                        switch(ViewModels.ShakerStatusViewModel.Instance.RTStatus)
                        {
                            case RTStatus.SignalGen:
                                {
                                    if(file ==null)
                                    {
                                        file = new NationalInstruments.Tdms.File(System.IO.Path.Combine(ViewModels.ShakerSettingViewModel.Instance.DataDirectory, $"{DateTime.Now:yyyy-MM-dd HH-mm-ss}.sdat"), System.IO.FileMode.OpenOrCreate, System.IO.FileAccess.ReadWrite).Open();
                                    }
                                    //file.
                                }
                                break;
                            default:
                                if(file!=null)
                                {
                                    file?.Dispose();
                                    file = null;
                                }
                                break;
                        }
                        CalcAnalog();
                        CommunicationViewModel.Instance.ServiceCommunication?.GetEvent(Topic.DATA).Publish(this, null,v);
                    }
                });
            });
        }
        static ShakerDataViewModel()
        {

        }
        public List<(List<DataPoint>, Models.StatisticsModel)> GetAnalogData(AnalogType analogType)
        {
            lock(_datalocker)
            {
                if(AnalogDataCache.TryGetValue(analogType,out var list))
                {
                    if (list == null) return new List<(List<DataPoint>, Models.StatisticsModel)>();
                    return list;
                }
                return new List<(List<DataPoint>, Models.StatisticsModel)>();
            }
        }
        private void CalcAnalog()
        {
            if (_Data.Length == 0) return;
            lock (_datalocker)
            {
                AnalogDataCache.Clear();
                for (int i = 0; i < ShakerConfigViewModel.Instance.AnalogSignals.Count; i++)
                {
                    if (i >= _Data.GetLength(0)) break;
                    List<DataPoint> dataPoints = new List<DataPoint>();
                    float max = float.MinValue;
                    float min = float.MaxValue;
                    float v = 0;
                    for (int j= 0; j < _Data.GetLength(1); j++)
                    {
                        dataPoints.Add(new DataPoint(1.0 / ShakerConfigViewModel.Instance.SampleRate * j, _Data[i, j]));
                        max = _Data[i, j] > max ? _Data[i, j] : max;
                        min = _Data[i, j] < min ? _Data[i, j] : min;
                        v += _Data[i, j];
                    }
                    Models.StatisticsModel model = new Models.StatisticsModel()
                    {
                        Name = ShakerConfigViewModel.Instance.Model.AnalogSignalConfigs[i].Name,
                        Max = max,
                        Min = min,
                        RMS = MainViewModel.Default.Calc.Sum.Rms(ref _Data[i, 0], (uint)_Data.GetLength(1)),
                        Average = v/ _Data.GetLength(1),
                        Unit = ShakerConfigViewModel.Instance.Model.AnalogSignalConfigs[i].Unit,
                    };
                    if (AnalogDataCache.TryGetValue(ShakerConfigViewModel.Instance.Model.AnalogSignalConfigs[i].AnalogType,out var list))
                    {
                        if (list == null) list = new List<(List<DataPoint>, Models.StatisticsModel)>();
                        list.Add((dataPoints,model));
                    }
                    else
                    {
                        AnalogDataCache[ShakerConfigViewModel.Instance.Model.AnalogSignalConfigs[i].AnalogType] = new List<(List<DataPoint>, Models.StatisticsModel)> { (dataPoints,model) };
                    }
                }
            }
            GetEvent(Topic.DATA).Publish(this,null);
        }
        public static ShakerDataViewModel Instance { get; } = new ShakerDataViewModel();
    }
}