REMODEL OF SourceFileReplacer.java

[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..d53c7b93f746ce4092c7cf4417a2e1a880ee1ac3 100644 (file)
--- a/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
+++ b/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
@@ -47,6 +47,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 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