add dxsfile convert

[mirror_edk2.git] / Tools / Source / MigrationTools / org / tianocore / migration / ModuleInfo.java

diff --git a/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java b/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
index 8d384aa34477b7ba534bc7fe3623e34342c81983..5706488c94fdc965c06b99f4b3cfcf9a04ca37b1 100644 (file)
--- a/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
+++ b/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
@@ -15,6 +15,8 @@ package org.tianocore.migration;
 import java.io.*;\r
 import java.util.*;\r
 \r
 import java.io.*;\r
 import java.util.*;\r
 \r

+import org.tianocore.ModuleTypeDef;\r
+\r

 /*\r
        Class ModuleInfo is built for scanning the source files, it contains all the needed\r
 information and all the temporary data.\r
 /*\r
        Class ModuleInfo is built for scanning the source files, it contains all the needed\r
 information and all the temporary data.\r


@@ -47,6 +49,14 @@ public final class ModuleInfo {
        public final Set<String> protocol = new HashSet<String>();\r
        public final Set<String> ppi = new HashSet<String>();\r
 \r
        public final Set<String> protocol = new HashSet<String>();\r
        public final Set<String> ppi = new HashSet<String>();\r
 \r

+       public final String getModuleType() {\r
+               if (moduletype.contains("PEI")) {\r
+                       return "PEIM";\r
+               } else {\r
+                       return "DXE_DRIVER";\r
+               }\r
+       }\r
+       \r

        public final void enroll(String filepath) throws Exception {\r
                String temp = null;\r
                if (filepath.contains(".c") || filepath.contains(".C") || filepath.contains(".h") || \r
        public final void enroll(String filepath) throws Exception {\r
                String temp = null;\r
                if (filepath.contains(".c") || filepath.contains(".C") || filepath.contains(".h") || \r