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
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