multi class

TypeTreeGenerator/Generator.cs

@@ -15,14 +15,20 @@ public void loadFolder(string assemblyFolder)
             this.assemblyLoader.Loaded = true;
         }
 
-        public IEnumerable<TypeDefinition> getTypeDefs(string m_AssemblyName, String m_ClassName, String m_Namespace)
+        public IEnumerable<TypeDefinition> getTypeDefs(string m_AssemblyName, List<String> m_ClassNames, String m_Namespace)
         {
-            IEnumerable<TypeDefinition> typeDefs = (m_ClassName.Length > 0) ?
-                new List<TypeDefinition>()
+            if (m_ClassNames.Count > 0)
+            {
+                List<TypeDefinition> typeDefs = new List<TypeDefinition>();
+                foreach (String m_ClassName in m_ClassNames)
                 {
-                    assemblyLoader.GetTypeDefinition(m_AssemblyName, string.IsNullOrEmpty(m_Namespace) ? m_ClassName : $"{m_Namespace}.{m_ClassName}")
-                } : assemblyLoader.GetTypeDefinitions(m_AssemblyName);
-            return typeDefs;
+                    typeDefs.Add(assemblyLoader.GetTypeDefinition(m_AssemblyName, string.IsNullOrEmpty(m_Namespace) ? m_ClassName : $"{m_Namespace}.{m_ClassName}"));
+                }
+                return typeDefs;
+            } else
+            {
+                return assemblyLoader.GetTypeDefinitions(m_AssemblyName);
+            }
         }
 
         public List<TypeTreeNode> convertToTypeTreeNodes(TypeDefinition typeDef, int[] version)
@@ -38,4 +44,4 @@ public List<TypeTreeNode> convertToTypeTreeNodes(TypeDefinition typeDef, int[] v
             return nodes;
         }
     }
-}
+}

TypeTreeGenerator/TypeTreeGenerator.csproj

@@ -1,7 +1,7 @@
 <Project Sdk="Microsoft.NET.Sdk">
 
 	<PropertyGroup>
-		<TargetFrameworks>net5.0;net6.0</TargetFrameworks>
+		<TargetFrameworks>net6.0</TargetFrameworks>
 		<OutputType>Library</OutputType>
 		<AutoGenerateBindingRedirects>true</AutoGenerateBindingRedirects>
 		<SelfContained>true</SelfContained>

TypeTreeGeneratorCLI/CLI.cs

@@ -36,7 +36,7 @@ static void Main(string[] args)
             }
             string assemblyFolder = "";
             string m_AssemblyName = "";
-            string m_ClassName = "";
+            List<string> m_ClassNames = new();
             string m_Namespace = "";
             int[] version = null;
             string dump = "simple";
@@ -55,7 +55,7 @@ static void Main(string[] args)
                         i++;
                         break;
                     case "-c":
-                        m_ClassName = args[i + 1];
+                        m_ClassNames.Add(args[i + 1]);
                         i++;
                         break;
                     case "-n":
@@ -91,7 +91,7 @@ static void Main(string[] args)
 
             // load typedefs
             gen.loadFolder(assemblyFolder);
-            IEnumerable<TypeDefinition> typeDefs = gen.getTypeDefs(m_AssemblyName, m_ClassName, m_Namespace);
+            IEnumerable<TypeDefinition> typeDefs = gen.getTypeDefs(m_AssemblyName, m_ClassNames, m_Namespace);
             if (typeDefs == null)
             {
                 Console.WriteLine("Failed to get type definitions.");

TypeTreeGeneratorCLI/TypeTreeGeneratorCLI.csproj

@@ -1,7 +1,7 @@
 <Project Sdk="Microsoft.NET.Sdk">
 
 	<PropertyGroup>
-		<TargetFrameworks>net5.0;net6.0</TargetFrameworks>
+		<TargetFrameworks>net6.0</TargetFrameworks>
 		<OutputType>Exe</OutputType>
 		<Platforms>AnyCPU;x86;x64</Platforms>
 		<GenerateRuntimeConfigurationFiles>true</GenerateRuntimeConfigurationFiles>