GymCollection.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.gym
{
    public class GymCollection : IEnumerable<IXMyCaffeGym>
    {
        List<IXMyCaffeGym> m_rgGym = new List<IXMyCaffeGym>();
 
        public GymCollection()
        {
        }
 
        public List<Exception> Load()
        {
            m_rgGym.Add(new CartPoleGym());
            m_rgGym.Add(new AtariGym());
            m_rgGym.Add(new CurveGym());
            m_rgGym.Add(new DataGeneralGym());
            m_rgGym.Add(new ModelGym());
 
            return addCustomGyms();
        }
 
        private List<Exception> addCustomGyms()
        {
            string codeBase = Assembly.GetExecutingAssembly().CodeBase;
            UriBuilder uri = new UriBuilder(codeBase);
            string path = Uri.UnescapeDataString(uri.Path);
            string strPath = Path.GetDirectoryName(path);
            List<Exception> rgErr = new List<Exception>();
 
            strPath += "\\CustomGyms";
 
            if (Directory.Exists(strPath))
            {
                string[] rgstrFiles = Directory.GetFiles(strPath);
 
                foreach (string strFile in rgstrFiles)
                {
                    FileInfo fi = new FileInfo(strFile);
 
                    if (fi.Extension.ToLower() == ".dll")
                    {
                        Exception excpt;
                        IXMyCaffeGym igym = loadCustomGym(strFile, out excpt);
                        if (igym != null)
                        {
                            m_rgGym.Add(igym);
                        }
                        else if (excpt != null)
                        {
                            rgErr.Add(excpt);
                        }
                    }
                }
            }
 
            return rgErr;
        }
 
        private IXMyCaffeGym loadCustomGym(string strFile, out Exception err)
        {
            err = null;
 
            try
            {
                Assembly a = Assembly.LoadFile(strFile);
                AssemblyName aName = a.GetName();
 
                foreach (Type t in a.GetTypes())
                {
                    Type[] rgT = t.GetInterfaces();
 
                    foreach (Type iface in rgT)
                    {
                        string strIface = iface.ToString();
                        if (strIface.Contains("IXMyCaffeGym"))
                        {
                            object obj = Activator.CreateInstance(t);
                            return obj as IXMyCaffeGym;
                        }
                    }
                }
 
                return null;
            }
            catch (Exception excpt)
            {
                if (excpt is System.Reflection.ReflectionTypeLoadException)
                {
                    var typeLoadException = excpt as ReflectionTypeLoadException;
                    var loaderExceptions = typeLoadException.LoaderExceptions;
 
                    if (loaderExceptions != null && loaderExceptions.Length > 0)
                        excpt = new Exception("Gym '" + strFile + "' failed to load!", loaderExceptions[0]);
                }
 
                err = excpt;
 
                return null;
            }
        }
 
        public IXMyCaffeGym Find(string strName)
        {
            int nPos = strName.IndexOf(':');
            if (nPos > 0)
                strName = strName.Substring(0, nPos);
 
            foreach (IXMyCaffeGym igym in m_rgGym)
            {
                if (igym.Name == strName)
                    return igym;
            }
 
            return null;
        }
 
        public IEnumerator<IXMyCaffeGym> GetEnumerator()
        {
            return m_rgGym.GetEnumerator();
        }
 
        IEnumerator IEnumerable.GetEnumerator()
        {
            return m_rgGym.GetEnumerator();
        }
    }
}