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 428f82b659e44b0584f69e44f6b0919499020c81..d53c7b93f746ce4092c7cf4417a2e1a880ee1ac3 100644 (file)
--- a/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
+++ b/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
@@ -47,14 +47,22 @@ 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
                                filepath.contains(".H") || filepath.contains(".dxs") || filepath.contains(".uni")) {\r
-                       localmodulesources.add(filepath.replace(modulepath + "\\", ""));\r
+                       localmodulesources.add(filepath.replace(modulepath + File.separator, ""));\r
                } else if (filepath.contains(".inf") || filepath.contains(".msa")) {\r
-                       temp = filepath.replace(modulepath + "\\", "");\r
-                       if (!temp.contains(File.separator)) {                                                           // .inf in subdirectory is not regarded\r
+                       temp = filepath.replace(modulepath + File.separator, "");\r
+                       if (!temp.contains(File.separator)) {                                                   // .inf in subdirectory is not regarded\r
                                msaorinf.add(temp);\r
                        }\r
                }\r