Virgo_Mask/VirgoUI/Controls/ClientBase/Handlers/WaferStatusHandler.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Threading;
using System.Threading;
using MECF.Framework.Common.DataCenter;
using OpenSEMI.Core.Msg;

namespace OpenSEMI.ClientBase.Handlers
{
    public class WaferStatusHandler : IHandler
    {
        public static readonly List<string> ReverseModules = new List<string>() { "LP1", "LP2", "LP3", "LP4", "Cassette", "Buffer" };

        public WaferStatusHandler(Func<string, List<WaferInfo>> funcGetWaferStatus, List<string> allModules)
        {
            this.WaferStatusDic = new Dictionary<string, ModuleWaferManager>();
            this.looper = new MsgPool(500, Do);

            var fromBottom = QueryDataClient.Instance.Service.GetConfig($"EFEM.LoadPort.SlotFromBottom");
            if (fromBottom != null && !((bool) fromBottom))
            {
                ReverseModules.Remove("LP1");
                ReverseModules.Remove("LP2");
                ReverseModules.Remove("LP3");
                ReverseModules.Remove("LP4");
            }

            this.GetWaferStatusByModule = funcGetWaferStatus;

            //init wafer status dictionary
            allModules.ForEach(key => { InitWafersByModule(key); });
        }

        public void Handle()
        {
            this.looper.Run();
        }

        private void InitWafersByModule(string moduleID)
        {
            lock (lockObj)
            {
                if (!this.WaferStatusDic.ContainsKey(moduleID))
                {
                    this.WaferStatusDic.Add(moduleID, new ModuleWaferManager(moduleID));
                    List<WaferInfo> mWafers = GetWaferStatusByModule(moduleID);
                    if (mWafers != null)
                    {
                        ModuleWaferManager modinfowithwafer = this.WaferStatusDic[moduleID];
                        for (int i = 0; i < mWafers.Count; i++)
                        {
                            modinfowithwafer.Wafers.Add(mWafers[i]);
                        }
                        if (ReverseModules.Contains(moduleID))
                        {
                            modinfowithwafer.Wafers.Reverse();
                        }
                    }
                }
            }
        }

        private void Do(MsgPool pool)
        {
            lock (lockObj)
            {
                if (Application.Current == null || Application.Current.Dispatcher == null)
                    return;

                foreach (KeyValuePair<string, ModuleWaferManager> pair in WaferStatusDic)
                {
                    List<WaferInfo> mWafers = GetWaferStatusByModule(pair.Key);
                    Application.Current.Dispatcher.BeginInvoke(DispatcherPriority.Normal, (ThreadStart)delegate
                    {
                        if (mWafers != null && mWafers.Count == pair.Value.Wafers.Count)
                        {
                            int index;
                            for (int i = 0; i < mWafers.Count; i++)
                            {
                                //status used in UI control
                                if (ReverseModules.Contains(pair.Key))
                                    index = mWafers.Count - i - 1;
                                else
                                    index = i;

                                pair.Value.Wafers[i].WaferStatus = mWafers[index].WaferStatus;
                                pair.Value.Wafers[i].WaferID = mWafers[index].WaferID;
                                pair.Value.Wafers[i].SourceName = mWafers[index].SourceName;
                                //pair.Value.Wafers[i].HasWafer = mWafers[index].HasWafer;
                                pair.Value.Wafers[i].IsReversed = mWafers[index].IsReversed;
                                pair.Value.Wafers[i].Orient = mWafers[index].Orient;
                            }
                        }

                    });

                }



            }
        }

        public Dictionary<string, ModuleWaferManager> WaferStatusDic { get; private set; }
        public MsgPool looper { get; private set; }
        private Func<string, List<WaferInfo>> GetWaferStatusByModule;
        private static object lockObj = new object();
    }
}