Free cookie consent management tool by TermsFeed Policy Generator

source: trunk/sources/HeuristicLab.Common/3.3/ObjectExtensions.cs @ 13494

Last change on this file since 13494 was 13494, checked in by pfleck, 8 years ago

#2557
Excluded all Types from System.Reflection.Emit from object graph traversal.

File size: 5.0 KB
RevLine 
[6103]1#region License Information
2/* HeuristicLab
[12012]3 * Copyright (C) 2002-2015 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
[6103]4 *
5 * This file is part of HeuristicLab.
6 *
7 * HeuristicLab is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation, either version 3 of the License, or
10 * (at your option) any later version.
11 *
12 * HeuristicLab is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
19 */
20#endregion
21
22using System;
[6205]23using System.Collections;
[6103]24using System.Collections.Generic;
[6205]25using System.Collections.Specialized;
[6380]26using System.Linq;
[6103]27using System.Reflection;
[7139]28using System.Security;
[6500]29using System.Threading;
[6103]30
31namespace HeuristicLab.Common {
32  public static class ObjectExtensions {
[8128]33    public static IEnumerable<T> ToEnumerable<T>(this T obj) {
34      yield return obj;
35    }
36
[8212]37    public static IEnumerable<object> GetObjectGraphObjects(this object obj, HashSet<object> excludedMembers = null, bool excludeStaticMembers = false) {
[6380]38      if (obj == null) return Enumerable.Empty<object>();
[8212]39      if (excludedMembers == null) excludedMembers = new HashSet<object>();
[9855]40      var fieldInfos = new Dictionary<Type, FieldInfo[]>();
[6380]41
[6103]42      var objects = new HashSet<object>();
[6380]43      var stack = new Stack<object>();
44
45      stack.Push(obj);
46      while (stack.Count > 0) {
47        object current = stack.Pop();
48        objects.Add(current);
49
[9855]50        foreach (object o in GetChildObjects(current, excludedMembers, excludeStaticMembers, fieldInfos)) {
[8310]51          if (o == null) continue;
52          if (ExcludeType(o.GetType())) continue;
53          if (objects.Contains(o)) continue;
54          stack.Push(o);
[6380]55        }
56      }
57
[6103]58      return objects;
59    }
[6891]60
[6192]61    /// <summary>
62    /// Types not collected:
63    ///   * System.Delegate
[6205]64    ///   * System.Reflection.Pointer
[6192]65    ///   * Primitives (Boolean, Byte, SByte, Int16, UInt16, Int32, UInt32, Int64, UInt64, IntPtr, UIntPtr, Char, Double, Single)
[7796]66    ///   * string, decimal, DateTime
[6205]67    ///   * Arrays of types not collected
[13494]68    ///   * All types from System.Reflection.Emit
[6192]69    /// </summary>
[6380]70    private static bool ExcludeType(Type type) {
71      return type.IsPrimitive ||
72             type == typeof(string) ||
[8310]73             type == typeof(string[]) ||
[6380]74             type == typeof(decimal) ||
[8310]75             type == typeof(decimal[]) ||
[7796]76             type == typeof(DateTime) ||
[8310]77             type == typeof(DateTime[]) ||
[6380]78             typeof(Delegate).IsAssignableFrom(type) ||
79             typeof(Pointer).IsAssignableFrom(type) ||
[13494]80             type.Namespace != "System.Reflection.Emit" ||
[6380]81             (type.HasElementType && ExcludeType(type.GetElementType()));
82    }
[8212]83
[9855]84    private static IEnumerable<object> GetChildObjects(object obj, HashSet<object> excludedMembers, bool excludeStaticMembers, Dictionary<Type, FieldInfo[]> fieldInfos) {
[6192]85      Type type = obj.GetType();
[6181]86
[6500]87      if (type.IsSubclassOfRawGeneric(typeof(ThreadLocal<>))) {
88        PropertyInfo info = type.GetProperty("Value");
89        object value = info.GetValue(obj, null);
[8310]90        if (value != null && !excludedMembers.Contains(value))
[8212]91          yield return value;
[6500]92      } else if (type.IsSubclassOfRawGeneric(typeof(Dictionary<,>)) ||
93           type.IsSubclassOfRawGeneric(typeof(SortedDictionary<,>)) ||
94           type.IsSubclassOfRawGeneric(typeof(SortedList<,>)) ||
95           obj is SortedList ||
96           obj is OrderedDictionary ||
97           obj is ListDictionary ||
98           obj is Hashtable) {
[6205]99        var dictionary = obj as IDictionary;
[8212]100        foreach (object value in dictionary.Keys) {
[8310]101          if (excludedMembers.Contains(value)) continue;
[6380]102          yield return value;
[8212]103        }
104        foreach (object value in dictionary.Values) {
[8310]105          if (excludedMembers.Contains(value)) continue;
[6380]106          yield return value;
[8212]107        }
[6205]108      } else if (type.IsArray || type.IsSubclassOfRawGeneric(typeof(HashSet<>))) {
109        var enumerable = obj as IEnumerable;
[8212]110        foreach (var value in enumerable) {
[8310]111          if (excludedMembers.Contains(value)) continue;
[6380]112          yield return value;
[8212]113        }
[6380]114      } else {
[9855]115        if (!fieldInfos.ContainsKey(type))
116          fieldInfos[type] = type.GetAllFields().ToArray();
117        foreach (FieldInfo f in fieldInfos[type]) {
[6891]118          if (excludeStaticMembers && f.IsStatic) continue;
[7139]119          object fieldValue;
120          try {
121            fieldValue = f.GetValue(obj);
[8310]122          }
123          catch (SecurityException) {
[7139]124            continue;
125          }
[8310]126          if (excludedMembers.Contains(fieldValue)) continue;
[7139]127          yield return fieldValue;
[6380]128        }
[6192]129      }
[6103]130    }
131  }
132}
Note: See TracBrowser for help on using the repository browser.