source: branches/Mainform refactoring/HeuristicLab.MainForm/3.2/MainFormManager.cs @ 2437

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2008 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using HeuristicLab.PluginInfrastructure;

namespace HeuristicLab.MainForm {
  public static class MainFormManager {
    private static object locker;
    private static IMainForm mainform;
    private static Dictionary<Type, List<Type>> views;
    private static Dictionary<Type, Type> defaultViews;

    static MainFormManager() {
      locker = new object();
      views = new Dictionary<Type, List<Type>>();
      defaultViews = new Dictionary<Type, Type>();
    }

    public static void RegisterMainForm(IMainForm mainform) {
      lock (locker) {
        if (MainFormManager.mainform == null) {
          MainFormManager.mainform = mainform;

          DiscoveryService ds = new DiscoveryService();
          Type[] types = ds.GetTypes(typeof(IView));

          foreach (Type t in types.Where(t => !t.IsAbstract && !t.IsInterface && !t.IsGenericType)) {
            foreach (Type viewableType in GetViewableType(t)) {
              if (viewableType != null) {
                if (!views.ContainsKey(viewableType))
                  views[viewableType] = new List<Type>();
                views[viewableType].Add(t);

                object[] attributes = t.GetCustomAttributes(typeof(DefaultView), false);
                if (attributes != null && attributes.Length == 1) {
                  if (defaultViews.ContainsKey(viewableType))
                    throw new ArgumentException("DefaultView for type " + viewableType + " is " + defaultViews[viewableType] +
                      ". Can't register additional DefaultView " + t + ".");
                  defaultViews[viewableType] = t;
                }
              }
            }
          }
        } else
          throw new ArgumentException("A mainform was already associated with the mainform manager.");
      }
    }

    private static IEnumerable<Type> GetViewableType(Type t) {
      foreach (Type interfaceType in t.GetInterfaces().Where(i => i.Namespace == "HeuristicLab.MainForm" && i.Name.StartsWith("IView"))) {
        if (interfaceType.IsGenericType && !interfaceType.IsGenericTypeDefinition)
          yield return interfaceType.GetGenericArguments()[0];
      }
    }

    public static IMainForm MainForm {
      get { return mainform; }
    }

    public static T GetMainForm<T>() where T : IMainForm {
      return (T)mainform;
    }

    public static IEnumerable<Type> GetViewTypes(Type viewableType) {
      List<Type> viewsForType = new List<Type>();
      foreach (KeyValuePair<Type, List<Type>> v in views) {
        if (v.Key.IsAssignableFrom(viewableType))
          viewsForType.AddRange(v.Value);
      }
      return viewsForType.Distinct();
    }

    public static bool ViewCanViewObject(IView view, object o) {
      return GetViewTypes(o.GetType()).Contains(view.GetType());
    }

    public static Type GetDefaultViewType(Type viewableType) {
      if (!defaultViews.ContainsKey(viewableType)) {
        List<Type> temp = defaultViews.Keys.Where(x => x.IsAssignableFrom(viewableType)).ToList();
        if (temp.Count == 0)
          return null;
        else if (temp.Count == 1)
          return defaultViews[temp[0]];
        else {
          temp.Sort(delegate(Type t1, Type t2) {
            if (t1.IsAssignableFrom(t2))
              return 1;
            return -1;
          }
          );
          if (!temp[0].IsAssignableFrom(temp[1]) && !temp[1].IsAssignableFrom(temp[0]))
            throw new Exception("Could not determine which is the default view for type " + viewableType.ToString() + ".");
          else
            return defaultViews[temp[0]];
        }
      } else
        return defaultViews[viewableType];
    }

    public static IView<T> CreateDefaultView<T>(T objectToView) {
      Type t = GetDefaultViewType(objectToView.GetType());
      if (t == null)
        return null;
      else
        return (IView<T>)Activator.CreateInstance(t);
    }
  }
}