X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=Tools%2FSource%2FMigrationTools%2Forg%2Ftianocore%2Fmigration%2FModuleInfo.java;h=5706488c94fdc965c06b99f4b3cfcf9a04ca37b1;hp=8d384aa34477b7ba534bc7fe3623e34342c81983;hb=9481cdac188b126dfea3178be6ca84c8d6e0e005;hpb=1d5181824b2176e11389b7cd645da027a739a110

diff --git a/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java b/Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
index 8d384aa344..5706488c94 100644
--- 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.*;
 import java.util.*;
 
+import org.tianocore.ModuleTypeDef;
+
 /*
 	Class ModuleInfo is built for scanning the source files, it contains all the needed
 information and all the temporary data.
@@ -47,6 +49,14 @@ 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") ||