little modify

[mirror_edk2.git] / Tools / Source / MigrationTools / org / tianocore / migration / ModuleReader.java

diff --git a/Tools/Source/MigrationTools/org/tianocore/migration/ModuleReader.java b/Tools/Source/MigrationTools/org/tianocore/migration/ModuleReader.java
index 2dcd6d2d6150b503e7ec95562e42e198cd7eb53d..cf1cabe22a6506640ab4f41985acaa27fa36bc03 100644 (file)
--- a/Tools/Source/MigrationTools/org/tianocore/migration/ModuleReader.java
+++ b/Tools/Source/MigrationTools/org/tianocore/migration/ModuleReader.java
@@ -27,22 +27,26 @@ public final class ModuleReader {
        \r
        public static final void ModuleScan(ModuleInfo m) throws Exception {\r
                mi = m;\r
-               \r
+\r
                Common.toDoAll(mi.modulepath, ModuleInfo.class.getMethod("enroll", String.class), mi, null, Common.FILE);\r
-               \r
+\r
                String filename = null;\r
                if (mi.msaorinf.isEmpty()) {\r
                        ModuleInfo.ui.println("No INF nor MSA file found!");\r
                        System.exit(0);\r
                } else {\r
-                       filename = ModuleInfo.ui.choose("Found .inf or .msa file for module\n" + mi.modulepath + "\nChoose one Please", mi.msaorinf.toArray());\r
+                       if (mi.msaorinf.size() == 1) {\r
+                               filename = (String)mi.msaorinf.toArray()[0];\r
+                       } else {\r
+                               filename = ModuleInfo.ui.choose("Found .inf or .msa file for module\n" + mi.modulepath + "\nChoose one Please", mi.msaorinf.toArray());\r
+                       }\r
                }\r
                if (filename.contains(".inf")) {\r
                        readInf(filename);\r
                } else if (filename.contains(".msa")) {\r
                        readMsa(filename);\r
                }\r
-               \r
+\r
                CommentOutNonLocalHFile();\r
                parsePreProcessedSourceCode();\r
 \r