WCS-Pedestal/Projects/永冠OPP/WCS.Service/DeviceExtentions.cs

using DBHelper_SqlSugar;
using System;
using System.Collections.Generic;
using System.Linq;
using WCS.Core;
using WCS.Entity;
using WCS.Entity.Protocol;
using WCS.Entity.Protocol.Station;

namespace WCS.Service
{
    public static class DeviceExtentions
    {
        public static IBCR80 BCR(this WCS_DEVICE source)
        {
            var dev = Device.Find("BCR" + source.CODE);
            return dev.DEVICEPROTOCOLS.First().Data<IBCR80>();
        }

        public static short PalletType(this WCS_DEVICE source)
        {
            return (short)source.Get<int>("PalletType");
        }

        public static short ProductLine(this WCS_DEVICE source)
        {
            return (short)source.Get<int>("ProductLine");
        }

        public static short WorkShop(this WCS_DEVICE source)
        {
            return (short)source.Get<int>("WorkShop");
        }

        /// <summary>
        /// 获取从source到endaddr的路径中的所有位置
        /// </summary>
        /// <param name="source">起始位置</param>
        /// <param name="endAddr">目标位置</param>
        /// <param name="condition">路径筛选条件</param>
        /// <returns></returns>
        public static List<WCS_DEVICE> GetPath(this WCS_DEVICE source, string endAddr, Func<List<WCS_DEVICE>, bool> condition = null)
        {
            var q = source.PATHS.Where(v => v.START == source & v.END.CODE.Contains(endAddr.ToUpper()))
                .Select(v => v.PATH.Split('-').Select(v => Device.Find(v)).ToList());

            if (condition != null)
                q = q.Where(condition);

            return q.FirstOrDefault();
        }

        /// <summary>
        /// 获取从source到endaddr的路径中的第一个位置
        /// </summary>
        /// <param name="source">起始位置</param>
        /// <param name="endAddr">目标位置</param>
        /// <param name="condition">路径筛选条件</param>
        /// <returns></returns>
        public static WCS_DEVICE GetPath(this WCS_DEVICE source, string endAddr)
        {
            var q = source.PATHS.Where(v => v.START == source && v.END.CODE.Contains(endAddr.ToUpper()))
                .Select(v => v.PATH.Split('-').Select(v => Device.Find(v)).ToList());

            return q.FirstOrDefault().FirstOrDefault();
        }

        public static WCS_DEVICE GetNext(this WCS_DEVICE source, string endAddr)
        {
            return source.GetNext(endAddr, null);
        }

        /// <summary>
        /// 获取从source到endaddr的路径中的下一个位置
        /// </summary>
        /// <param name="source">起始位置</param>
        /// <param name="endAddr">目标位置</param>
        /// <param name="condition">路径筛选条件</param>
        /// <returns></returns>
        public static WCS_DEVICE GetNext(this WCS_DEVICE source, string endAddr, Func<List<WCS_DEVICE>, bool> condition)
        {
            var path = source.GetPath(endAddr, condition);
            if (Ltc.Do(path, v => v == null || v.Count == 0))
                return null;

            if (source.IsConv())
            {
                var obj = path.FirstOrDefault(v => !v.IsConv());
                if (obj == null)
                {
                    return path.Last();
                }
                //else if (obj.IsSC() || obj.IsTunnel())
                //{
                //    var lastConv = path.TakeWhile(v => v.IsConv()).LastOrDefault();
                //    return lastConv;
                //}
                else if (obj.IsRGV())
                {
                    var target = path.Skip(path.IndexOf(obj) + 1).FirstOrDefault();
                    return target;
                }
                else
                    return null;
            }
            else
            {
                return path.First();
            }
        }

        public static bool IsRGV(this WCS_DEVICE source)
        {
            return source.DEVICEPROTOCOLS.Any(v => v.DB.PROTOCOL == typeof(IRGV520).AssemblyQualifiedName);
        }

        public static bool IsConv(this WCS_DEVICE source)
        {
            return source.DEVICEPROTOCOLS.Any(v => v.DB.PROTOCOL == typeof(IStation521).AssemblyQualifiedName);
        }

        /// <summary>
        /// 是否堆垛机
        /// </summary>
        /// <param name="source"></param>
        /// <returns></returns>
        public static bool IsSC(this WCS_DEVICE source)
        {
            return source.DEVICEPROTOCOLS.Any(v => v.DB.PROTOCOL == typeof(ISRM520).AssemblyQualifiedName);
        }

        public static bool IsTunnel(this WCS_DEVICE source)
        {
            return source.DEVICEPROTOCOLS.Count == 0 && source.CODE.StartsWith("T");
        }
    }

    public static class TaskExtentions
    {
        public static string AddrCurrent(this WCS_TASK source, Db db)
        {
            var q1 = db.Default.Queryable<WCS_Station520>().Where(v => v.ISLAST && v.Tasknum == source.ID).Select(v => new TemporaryModel
            {
                TYPE = 1,
                CODE = v.DEVICE.CODE
            });
            var q2 = db.Default.Queryable<WCS_RGV521>().Where(v => v.ISLAST && (v.TaskID_1 == source.ID || v.TaskID_2 == source.ID)).Select(v => new TemporaryModel
            {
                TYPE = 2,
                CODE = v.DEVICE.CODE
            });
            var q = db.Default.UnionAll(q1, q2);

            var res = q.OrderBy(v => v.TYPE).First();
            return res == null ? "" : res.CODE;
        }
    }

    public class TemporaryModel
    {
        public int TYPE { get; set; }

        public string CODE { get; set; }
    }
}