增加查找线工具

2026-03-28 11:16:34 +08:00 · 2019-08-04 21:51:40 +08:00
parent 61de0d1236
commit 9d8944dcc3
256 changed files with 719371 additions and 778 deletions
--- a/VisionJobFactory/VisionToolFactory.cs
+++ b/VisionJobFactory/VisionToolFactory.cs
@@ -0,0 +1,136 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.Linq;
+using System.Reflection;
+using System.Text;
+using System.Threading.Tasks;
+using CommonMethods;
+
+namespace VisionJobFactory
+{
+
+    public class VisionToolFactory
+    {
+        private static Dictionary<ToolType, Type> animalTypeDic = new Dictionary<ToolType, Type>();
+        public static void InitVisionToolTypeDic()
+        {
+            animalTypeDic.Clear();
+
+            //读取所有带有VisionToolAttribute的类
+            var classEnumerator = new ClassEnumerator(typeof(VisionToolAttribute), null, typeof(VisionToolAttribute).Assembly);
+            var em = classEnumerator.Results.GetEnumerator();
+
+            while (em.MoveNext())
+            {
+                var classType = em.Current;
+                var atts = classType.GetCustomAttributes(typeof(VisionToolAttribute), true);
+                if (atts.Length > 0)
+                {
+                    var att = atts[0] as VisionToolAttribute;
+                    if (null != att)
+                    {
+                        animalTypeDic.Add(att.ToolType, classType);
+                    }
+                }
+            }
+        }
+
+        public static IToolInfo CreateToolVision(ToolType animalType, string toolName)
+        {
+            if (animalTypeDic.ContainsKey(animalType))
+            {
+                return (IToolInfo)Activator.CreateInstance(animalTypeDic[animalType], new object[] { toolName });
+            }
+            return null;
+        }
+        public static IToolInfo CreateToolVision(ToolType animalType)
+        {
+            if (animalTypeDic.ContainsKey(animalType))
+            {
+                return (IToolInfo)Activator.CreateInstance(animalTypeDic[animalType]);
+            }
+            return null;
+        }
+    }
+
+    //自定义Attribute
+    public sealed class VisionToolAttribute : Attribute
+    {
+        public ToolType ToolType { get; private set; }
+        public VisionToolAttribute(ToolType toolType)
+        {
+            ToolType = toolType;
+        }
+    }
+
+    //根据Attribute提取类
+    public class ClassEnumerator
+    {
+        protected List<Type> results = new List<Type>();
+
+        public List<Type> Results
+        {
+            get
+            {
+                return results;
+            }
+        }
+
+        private Type AttributeType;
+        private Type InterfaceType;
+
+        public ClassEnumerator(Type InAttributeType, Type InInterfaceType, Assembly InAssembly, bool bIgnoreAbstract = true, bool bInheritAttribute = false, bool bShouldCrossAssembly = false)
+        {
+            AttributeType = InAttributeType;
+            InterfaceType = InInterfaceType;
+
+            try
+            {
+                if (bShouldCrossAssembly)
+                {
+                    Assembly[] Assemblys = AppDomain.CurrentDomain.GetAssemblies();
+                    if (Assemblys != null)
+                    {
+                        for (int i = 0, len = Assemblys.Length; i < len; i++)
+                        {
+                            CheckInAssembly(Assemblys[i], bIgnoreAbstract, bInheritAttribute);
+                        }
+                    }
+                }
+                else
+                {
+                    CheckInAssembly(InAssembly, bIgnoreAbstract, bInheritAttribute);
+                }
+            }
+            catch (Exception e)
+            {
+                Debug.WriteLine("Error in enumerate classes: " + e.Message);
+            }
+        }
+
+        private void CheckInAssembly(Assembly InAssembly, bool bInIgnoreAbstract, bool bInInheritAttribute)
+        {
+            Type[] types = InAssembly.GetTypes();
+            if (null == types)
+            {
+                return;
+            }
+
+            for (int i = 0, len = types.Length; i < len; i++)
+            {
+                var type = types[i];
+                if (InterfaceType == null || InterfaceType.IsAssignableFrom(type))
+                {
+                    if (!bInIgnoreAbstract || (bInIgnoreAbstract && !type.IsAbstract))
+                    {
+                        if (type.GetCustomAttributes(AttributeType, bInInheritAttribute).Length > 0)
+                        {
+                            results.Add(type);
+                        }
+                    }
+                }
+            }
+        }
+    }
+}