X-Git-Url: https://git.proxmox.com/?a=blobdiff_plain;f=Tools%2FSource%2FMigrationTools%2Forg%2Ftianocore%2Fmigration%2FModuleInfo.java;h=d53c7b93f746ce4092c7cf4417a2e1a880ee1ac3;hb=16b7eeef5489a5e16da1574c13df53990ffac2cf;hp=428f82b659e44b0584f69e44f6b0919499020c81;hpb=5f4eb6b6d5eb39012da23c16fdf3f9d988f9db06;p=mirror_edk2.git

diff --git a/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java b/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
index 428f82b659..d53c7b93f7 100644
--- 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>();
 	public final Set<String> ppi = new HashSet<String>();
 
+	public final String getModuleType() {
+		if (moduletype.contains("PEI")) {
+			return "PEIM";
+		} else {
+			return "DXE_DRIVER";
+		}
+	}
+	
 	public final void enroll(String filepath) throws Exception {
 		String temp = null;
 		if (filepath.contains(".c") || filepath.contains(".C") || filepath.contains(".h") || 
 				filepath.contains(".H") || filepath.contains(".dxs") || filepath.contains(".uni")) {
-			localmodulesources.add(filepath.replace(modulepath + "\\", ""));
+			localmodulesources.add(filepath.replace(modulepath + File.separator, ""));
 		} else if (filepath.contains(".inf") || filepath.contains(".msa")) {
-			temp = filepath.replace(modulepath + "\\", "");
-			if (!temp.contains(File.separator)) {								// .inf in subdirectory is not regarded
+			temp = filepath.replace(modulepath + File.separator, "");
+			if (!temp.contains(File.separator)) {							// .inf in subdirectory is not regarded
 				msaorinf.add(temp);
 			}
 		}