WcsFramework/WCS.WorkEngineering/Extensions/DeviceExtension.cs

using System.Collections.Concurrent;
using WCS.Core;
using WCS.Entity;
using WCS.Entity.Protocol.Station;

namespace WCS.WorkEngineering.Extensions
{
    /// <summary>
    ///  设备扩展
    /// </summary>
    public static class DeviceExtension
    {
        /// <summary>
        ///  是否是输送线
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <returns></returns>
        public static bool IsConv(this Device source)
        {
            return source.HasProtocol(typeof(IStation521)) || source.HasProtocol(typeof(IStation520)) || source.HasProtocol(typeof(IStation523));
        }

        /// <summary>
        ///  是否是巷道
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <returns></returns>
        public static bool IsTunnel(this WCS_DeviceInfo source)
        {
            return source.Code.Contains("TY");
        }

        /// <summary>
        ///  获取下一个地址
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <param name="addto">目标地址</param>
        /// <returns></returns>
        public static string GetNextAdd(this WCS_DeviceInfo source, string addto)
        {
            var path = source.Paths.First(p => p.EndCode == addto).Path;
            return path.Split('-')[0];
        }

        /// <summary>
        ///  获取到达目标地址需要的路径信息
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <param name="addto">目标地址</param>
        /// <returns></returns>
        public static string GetAddtoPath(this WCS_DeviceInfo source, string addto)
        {
            return source.Paths.First(p => p.EndCode == addto).Path;
        }

        #region WCS_DEVICE扩展数据

        /// <summary>
        ///  设备标识字典
        /// </summary>
        private static ConcurrentDictionary<string, object> DeviceValues = new ConcurrentDictionary<string, object>();

        /// <summary>
        /// 添加设备标识
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <param name="flag">标识</param>
        public static void AddFlag(this Device source, DF flag)
        {
            var df = source.Get<DF>("DeviceFlag");
            df = df | flag;
            source.Set("DeviceFlag", df);
        }

        /// <summary>
        ///  当前设备是否包含标识
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <param name="flag">标识</param>
        /// <returns></returns>
        public static bool Is(this Device source, DF flag)
        {
            var df = source.Get<DF>("DeviceFlag");
            return (df & flag) == flag;
        }

        /// <summary>
        ///  为设备写入标识
        /// </summary>
        /// <typeparam name="T">写入值类型</typeparam>
        /// <param name="source">设备信息</param>
        /// <param name="key">标识类型</param>
        /// <param name="value">写入值</param>
        public static void Set<T>(this Device source, string key, T value)
        {
            DeviceValues[source.Code + key] = value;
        }

        /// <summary>
        ///  获取设备标识
        /// </summary>
        /// <typeparam name="T">写入值类型</typeparam>
        /// <param name="source">设备信息</param>
        /// <param name="key">标识类型</param>
        /// <returns></returns>
        public static T Get<T>(this Device source, string key)
        {
            if (!DeviceValues.ContainsKey(source.Code + key))
                return default(T);
            return (T)DeviceValues[source.Code + key];
        }

        /// <summary>
        ///  获取short类型的设备号
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <returns></returns>
        public static short Code(this Device source)
        {
            return short.Parse(source.Code);
        }

        /// <summary>
        ///  获取巷道
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <returns>返回巷道Code</returns>
        public static string Tunnel(this Device source)
        {
            return source.Get<string>("Tunnel");
        }

        /// <summary>
        ///  获取巷道号
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <returns>返回巷道号</returns>
        public static int TunnelNum(this Device source)
        {
            return int.Parse(source.Tunnel().Last().ToString());
        }

        /// <summary>
        ///  获取设备楼层信息
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <returns></returns>
        public static int Floor(this Device source)
        {
            return source.Get<int>("Floor");
        }

        /// <summary>
        ///  获取堆垛机
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <returns></returns>
        public static Device SRM(this Device source)
        {
            return source.Get<Device>("SRM");
        }

        /// <summary>
        ///  获取RGV
        /// </summary>
        /// <param name="source">设备信息</param>
        /// <returns></returns>
        public static Device RGV(this Device source)
        {
            return source.Get<Device>("RGV");
        }

        //public static void AddFlag(DF flag, params string[] devices)
        //{
        //    var arr = LogicHandler.AllObjects.OfType<WCS_DEVICE>().Where(v => devices.Contains(v.CODE)).ToArray();
        //    Parallel.ForEach(arr, v =>
        //    {
        //        v.AddFlag(flag);
        //    });
        //}

        #endregion WCS_DEVICE扩展数据
    }

    /// <summary>
    /// 设备配置
    /// </summary>
    [Flags]
    public enum DF
    {
        无 = 0,
    }
}