VisionEdit/VisionJobFactory/VisionToolFactory.cs

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 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);
                        }
                    }
                }
            }
        }
    }
}