Sort types in type hierarchy.

dotnet · Sep 27, 2019 · 6b0ff5b · 6b0ff5b
1 parent ebf12da
commit 6b0ff5b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/Documentation/SymbolDefinitionWriter.cs b/src/Documentation/SymbolDefinitionWriter.cs
@@ -361,7 +361,11 @@ private void WriteNamespaces(in OneOrMany<IAssemblySymbol> assemblies, Cancellat
 
         private void WriteTypeHierarchy(IEnumerable<INamedTypeSymbol> types, CancellationToken cancellationToken = default)
         {
-            TypeHierarchy hierarchy = TypeHierarchy.Create(types, SymbolDefinitionComparer.SystemFirst.TypeComparer);
+            IComparer<INamedTypeSymbol> comparer = (Format.Includes(SymbolDefinitionPartFilter.ContainingNamespaceInTypeHierarchy))
+                ? SymbolDefinitionComparer.SystemFirst.TypeComparer
+                : SymbolDefinitionComparer.OmitContainingNamespace.TypeComparer;
+
+            TypeHierarchy hierarchy = TypeHierarchy.Create(types, comparer);
 
             WriteTypeHierarchy(hierarchy, cancellationToken);
         }

diff --git a/src/Workspaces.Core/Comparers/SymbolDefinitionComparer.cs b/src/Workspaces.Core/Comparers/SymbolDefinitionComparer.cs
@@ -19,6 +19,8 @@ internal SymbolDefinitionComparer(SymbolDefinitionSortOptions options = SymbolDe
 
         public static SymbolDefinitionComparer SystemFirst { get; } = new DefaultSymbolDefinitionComparer(SymbolDefinitionSortOptions.SystemFirst);
 
+        public static SymbolDefinitionComparer OmitContainingNamespace { get; } = new DefaultSymbolDefinitionComparer(SymbolDefinitionSortOptions.OmitContainingNamespace);
+
         public static SymbolDefinitionComparer SystemFirstOmitContainingNamespace { get; } = new DefaultSymbolDefinitionComparer(SymbolDefinitionSortOptions.SystemFirst | SymbolDefinitionSortOptions.OmitContainingNamespace);
 
         public SymbolDefinitionSortOptions Options { get; }