Shaker/Avalonia/ShakerApp/ViewModels/SignalPreview/AnalogSignalPreviewViewModel.cs

using Avalonia.Collections;
using Avalonia.Controls;
using OxyPlot;
using OxyPlot.Axes;
using Shaker.Models;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using ShakerApp.Tools;
using OxyPlot.Series;
using System.Diagnostics;
using System.Diagnostics.CodeAnalysis;

namespace ShakerApp.ViewModels
{
    public class AnalogSignalPreviewViewModel:DisplayViewModelBase<IModel>,IDataPreview
    {
        public string AttachTitle 
        { 
            get => attachTitle;
            set
            {
                if (attachTitle == value) return;
                attachTitle = value;
                UpdatePlotTitle(SelectedAnalog);
            }
        }
        List<OxyPlot.Series.LineSeries> lineSeries = new List<OxyPlot.Series.LineSeries>();
        private object locker = new object();
        public AvaloniaList<ViewModels.StatisticsViewModel> Statistics { get; } = new AvaloniaList<StatisticsViewModel>();
        private AnalogType selectedAnalog;
        private bool canChangedAnalog = true;
        public bool StatisticsVisibily { get => statisticsVisibily; set =>SetProperty(ref statisticsVisibily , value); }
        public AnalogSignalPreviewViewModel(bool createcontrol = false):base()
        {
            Content = typeof(Views.AnalogSignalPreviewView);
            if(createcontrol)
            {
                Control = (Control)Activator.CreateInstance(Content)!;
            }
            PlotModel.Axes.Add(new OxyPlot.Axes.LinearAxis()
            {
                MaximumPadding = 0,
                MinimumPadding = 0,
                Title = App.Current?.FindResource("Time") + "",
                Unit = "s",
                Key = "X",
                MajorGridlineStyle = OxyPlot.LineStyle.Solid,
                Position = OxyPlot.Axes.AxisPosition.Bottom
            });
            PlotModel.Axes.Add(new OxyPlot.Axes.LinearAxis()
            {
                Title = App.Current?.FindResource("Ampt") + "",
                Unit = "V",
                Key = "Y",
                MajorGridlineStyle = OxyPlot.LineStyle.Solid,
                Position = OxyPlot.Axes.AxisPosition.Left,
            });
            UpdatePlotTitle(lasttype);
            PlotModel.Legends.Add(new OxyPlot.Legends.Legend()
            {
                ShowInvisibleSeries = true,
            });
            PlotController.BindMouseWheel(OxyMouseWheelCommand);
            GetEvent(Topic.DATA).Subscrip((_, _) =>
            {
                if (!UpSignalData) return;
                lock(locker)
                {
                    if(SelectedAnalog == AnalogType.Driver)
                    {

                    }
                    var data = ShakerDataViewModel.Instance.GetAnalogData(SelectedAnalog);
                    PlotModel.InvalidatePlot(false);
                    for(int i=0;i<data.Count;i++)
                    {
                        lineSeries[i].ItemsSource = data[i].Item1;
                        Statistics[i].UpDateModel(data[i].Item2);
                        if (ShakerDataViewModel.Instance.AllowTHDChannels.Contains(data[i].Item2.Name))
                        {
                            THD = data[i].Item2.THD;
                        }
                    }
                    PlotModel.InvalidatePlot(true);
                }
            });
            GetEvent(Models.Topic.InitSeries).Subscrip((_, _) =>
            {
                lock(locker)
                {
                    Menu.Clear();
                    ShakerConfigViewModel.Instance.AnalogSignals
                    .DistinctBy(x=>x.Value.AnalogType)
                    .ToList()
                        .ForEach(x =>
                        {
                            Menu.Add(new TimeDomainMenuViewModel()
                            {
                                IsChecked = false,
                                IsEnabled = true,
                                AnalogType = x.Value.AnalogType,
                                Unit = x.Value.Unit,
                            });
                        });
                    if (CanChangedAnalog)
                    {
                        int index = Menu.ToList().FindIndex(x => x.AnalogType == lasttype);
                        if (index == -1)
                        {
                            SelectedAnalog = Menu.First().AnalogType;
                        }
                        else
                        {
                            SelectedAnalog = lasttype;
                        }
                    }
                    else
                    {
                        SelectedAnalog = lasttype;
                    }
                }
            });


            GetEvent(ViewModels.ShakerSettingViewModel.LANGUAGECHANGEDEVENT).Subscrip((_, _) =>
            {
                PlotModel.InvalidatePlot(false);
                PlotModel.Axes[0].Title = App.Current?.FindResource("Time") + "";
                PlotModel.Axes[1].Title = App.Current?.FindResource("Ampt") + "";
                UpdatePlotTitle(SelectedAnalog);
                var config = ShakerConfigViewModel.Instance.Model.AnalogSignalConfigs.First(x => x.AnalogType == SelectedAnalog);
                for(int i=0;i<PlotModel.Series.Count;i++)
                {
                    PlotModel.Series[i].Title = App.Current?.FindResource(config.Name) + "";
                }
            });
        }
        public string THDTitle { get => tHDTitle; set =>SetProperty(ref tHDTitle , value); }
        public float THD { get => tHD; set =>SetProperty(ref tHD , value); }
        public bool THDVisibily => SelectedAnalog == AnalogType.Acceleration || SelectedAnalog == AnalogType.Displacement;
        [AllowNull]
        public Control Control { get; }
        private AnalogType lasttype = AnalogType.Displacement;
        private bool statisticsVisibily = true;
        private string attachTitle = string.Empty;
        private float tHD = 0;
        private string tHDTitle = string.Empty;
        private bool tHDEnabled = true;

        private void UpdatePlotTitle(AnalogType analogType)
        {
            PlotModel.Title = string.IsNullOrEmpty(AttachTitle) ? App.Current?.FindResource(analogType.Description()) + "" : $"{App.Current?.FindResource(AttachTitle)}-{App.Current?.FindResource(analogType.Description())}";
        }
        public AvaloniaList<TimeDomainMenuViewModel> Menu { get; } = new AvaloniaList<TimeDomainMenuViewModel>();
        public AnalogSignalPreviewViewModel(Shaker.Models.AnalogType analogType,bool createcontrol = false):this(createcontrol)
        {
            lasttype = analogType;
        }
        public AnalogType SelectedAnalog
        {
            get => selectedAnalog;
            set
            {
                SetProperty(ref selectedAnalog, value);
                ChangeAnalogType(value);
                OnPropertyChanged(nameof(THDVisibily));
            }
        }
        public bool UpSignalData { get; set; } = true;
        private void ChangeAnalogType(AnalogType type)
        {
            lock (locker)
            {
                lineSeries.Clear();
                PlotModel.Series.Clear();
                Statistics.Clear();
                for(int i=0;i<Menu.Count;i++)
                {
                    Menu[i].IsChecked = Menu[i].AnalogType == type;
                }
                var config = ShakerConfigViewModel.Instance.Model.AnalogSignalConfigs.First(x => x.AnalogType == type);
                PlotModel.Axes[1].Unit = config.Unit;
                var allconfig = ShakerConfigViewModel.Instance.Model.AnalogSignalConfigs.Where(x => x.AnalogType == type).ToList();
                for (int i = 0; i < allconfig.Count; i++)
                {
                    LineSeries series = new LineSeries();
                    series.Title = App.Current?.FindResource(allconfig[i].Name) + "";
                    if (ShakerDataViewModel.Instance.AllowTHDChannels.Contains(allconfig[i].Name))
                    {
                        THDTitle = allconfig[i].Name;
                    }
                    series.XAxisKey = "X";
                    series.YAxisKey = "Y";
                    series.DataFieldX = nameof(DataPoint.X);
                    series.DataFieldY = nameof(DataPoint.Y);
                    Statistics.Add(new StatisticsViewModel(new Models.StatisticsModel()
                    {
                        Name = allconfig[i].Name,
                    }));
                    lineSeries.Add(series);
                }
                PlotModel.InvalidatePlot(false);
                UpdatePlotTitle(type);
                lineSeries.ForEach(x => PlotModel.Series.Add(x));
                PlotModel.InvalidatePlot(true);
            }
        }
        public bool CanChangedAnalog
        {
            get => canChangedAnalog;
            set =>SetProperty(ref canChangedAnalog , value); 
        }
        public bool THDEnabled { get => tHDEnabled; set =>SetProperty(ref tHDEnabled , value); }
        public OxyPlot.PlotModel PlotModel { get; } = new OxyPlot.PlotModel();
        public OxyPlot.PlotController PlotController { get; } = new OxyPlot.PlotController();
        public IViewCommand<OxyMouseWheelEventArgs> OxyMouseWheelCommand => new DelegatePlotCommand<OxyMouseWheelEventArgs>(OnOxyMouseWheel);
        private void OnOxyMouseWheel(IPlotView view, IController controller, OxyMouseWheelEventArgs args)
        {
            HandleZoomByWheel(view, args);
            if (view.ActualModel is PlotModel plotModel && plotModel.Axes.Count >= 1 && plotModel.Axes[0] is LinearAxis axis)
            {
                axis.MajorStep = (axis.Maximum - axis.Minimum) / 10;
            }
        }
        private void HandleZoomByWheel(IPlotView view, OxyMouseWheelEventArgs args, double factor = 1)
        {
            var m = new ZoomStepManipulator(view)
            {
                AxisPreference = AxisPreference.X,
                Step = args.Delta * 0.001 * factor,
                FineControl = args.IsControlDown,
            };
            m.Started(args);
        }

    }
}