]> git.proxmox.com Git - mirror_edk2.git/blobdiff - Tools/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
Turn to Regex in ModuleReader.java
[mirror_edk2.git] / Tools / Source / MigrationTools / org / tianocore / migration / ModuleInfo.java
index b238551cac84ca18dc6e7af3094358c61bb515a1..5e3bbb50d02878f0ae17f47918d825942ebcb21c 100644 (file)
@@ -39,6 +39,7 @@ public class ModuleInfo {
        \r
        public Set<String> localmodulesources = new HashSet<String>();          //contains both .c and .h\r
        public Set<String> preprocessedccodes = new HashSet<String>();\r
+       public Set<String> msaorinf = new HashSet<String>();                            //only a little, hash may be too big for this\r
        \r
        public Set<String> hashfuncc = new HashSet<String>();\r
        public Set<String> hashfuncd = new HashSet<String>();\r
@@ -54,50 +55,42 @@ public class ModuleInfo {
        \r
        private static String migrationcomment = "//%$//";\r
        \r
-       private void moduleScan() throws Exception {\r
-               String[] list = new File(modulepath).list();\r
-               boolean hasInf = false;\r
-               String infname = null;\r
-               boolean hasMsa = false;\r
-               String msaname = null;\r
+       private void dirScan(String subpath) throws Exception {\r
+               String[] list = new File(modulepath + File.separator + subpath).list();                 // if no sub , separator need?\r
+               File test;\r
                \r
                for (int i = 0 ; i < list.length ; i++) {\r
-                       if (new File(list[i]).isDirectory()) {\r
-                               ;\r
+                       test = new File(modulepath + File.separator + subpath + list[i]);\r
+                       if (test.isDirectory()) {\r
+                               if (list[i].contains("result") || list[i].contains("temp")) {\r
+                               } else {\r
+                                       dirScan(subpath + list[i] + File.separator);\r
+                               }\r
                        } else {\r
-                               if (list[i].contains(".c") || list[i].contains(".C")) {\r
-                                       localmodulesources.add(list[i]);\r
-                               } else if (list[i].contains(".h") || list[i].contains(".H")) {\r
-                                       localmodulesources.add(list[i]);        //the case that several .inf or .msa found is not concerned\r
-                               } else if (list[i].contains(".dxs")) {\r
-                                       localmodulesources.add(list[i]);\r
-                               } else if (list[i].contains(".uni")) {\r
-                                       localmodulesources.add(list[i]);\r
-                               } else if (list[i].contains(".inf")) {\r
-                                       if (ui.yesOrNo("Found .inf file : " + list[i] + "\nDo you want to use this file as this module's .inf?")) {\r
-                                               hasInf = true;\r
-                                               infname = list[i];\r
-                                       } else {\r
-                                               continue;\r
-                                       }\r
-                               } else if (list[i].contains(".msa")) {\r
-                                       if (ui.yesOrNo("Found .msa file : " + list[i] + "\nDo you want to use this file as this module's .msa?")) {\r
-                                               hasMsa = true;\r
-                                               msaname = list[i];\r
-                                       } else {\r
-                                               continue;\r
-                                       }\r
+                               if (list[i].contains(".c") || list[i].contains(".C") || list[i].contains(".h") || \r
+                                       list[i].contains(".H") || list[i].contains(".dxs") || list[i].contains(".uni")) {\r
+                                       localmodulesources.add(subpath + list[i]);\r
+                               } else if (list[i].contains(".inf") || list[i].contains(".msa")) {\r
+                                       msaorinf.add(subpath + list[i]);\r
                                }\r
                        }\r
                }\r
-               \r
-               ModuleReader mr = new ModuleReader(modulepath, this, db);\r
-               if (hasInf) {                                                   // this sequence shows using .inf as default\r
-                       mr.readInf(infname);\r
-               } else if (hasMsa) {\r
-                       mr.readMsa(msaname);\r
-               } else {\r
+       }\r
+       \r
+       private void moduleScan() throws Exception {\r
+               dirScan("");\r
+               String filename = null;\r
+               if (msaorinf.isEmpty()) {\r
                        ui.println("No INF nor MSA file found!");\r
+                       System.exit(0);\r
+               } else {\r
+                       filename = ui.choose("Found .inf or .msa file in the module\nChoose one Please", msaorinf.toArray());\r
+               }\r
+               ModuleReader mr = new ModuleReader(modulepath, this, db, ui);\r
+               if (filename.contains(".inf")) {\r
+                       mr.readInf(filename);\r
+               } else if (filename.contains(".msa")) {\r
+                       mr.readMsa(filename);\r
                }\r
                \r
                CommentOutNonLocalHFile();\r
@@ -137,7 +130,7 @@ public class ModuleInfo {
                ui.println(show + hash.size());\r
                ui.println(hash);\r
        }\r
-\r
+       \r
        // add '//' to all non-local include lines\r
        private void CommentOutNonLocalHFile() throws IOException {\r
                BufferedReader rd;\r
@@ -146,20 +139,18 @@ public class ModuleInfo {
                PrintWriter outfile;\r
 \r
                Pattern ptninclude = Pattern.compile("[\"<](.*[.]h)[\">]");\r
-               Matcher mtcinclude;\r
-               \r
-               File tempdir = new File(modulepath + File.separator + "temp" + File.separator);\r
-               if (!tempdir.exists()) tempdir.mkdir();\r
+               Matcher mtrinclude;\r
                \r
                Iterator<String> ii = localmodulesources.iterator();\r
                while ( ii.hasNext() ) {\r
                        curFile = ii.next();\r
                        rd = new BufferedReader(new FileReader(modulepath + File.separator + curFile));\r
+                       Common.ensureDir(modulepath + File.separator + "temp" + File.separator + curFile);\r
                        outfile = new PrintWriter(new BufferedWriter(new FileWriter(modulepath + File.separator + "temp" + File.separator + curFile)));\r
                        while ((line = rd.readLine()) != null) {\r
                                if (line.contains("#include")) {\r
-                                       mtcinclude = ptninclude.matcher(line);\r
-                                       if (mtcinclude.find() && localmodulesources.contains(mtcinclude.group(1))) {\r
+                                       mtrinclude = ptninclude.matcher(line);\r
+                                       if (mtrinclude.find() && localmodulesources.contains(mtrinclude.group(1))) {\r
                                        } else {\r
                                                line = migrationcomment + line;\r
                                        }\r
@@ -170,15 +161,7 @@ public class ModuleInfo {
                        outfile.close();\r
                }\r
        }\r
-       /*\r
-       private void search(String line, Pattern ptn, Method md) {\r
-               matmacro = Func.ptntmacro.matcher(line);\r
-               while (matmacro.find()) {\r
-                       if ((temp = Func.registerMacro(matmacro, this, db)) != null) {\r
-                       }\r
-               }\r
-       }\r
-       */\r
+\r
        private void parsePreProcessedSourceCode() throws Exception {\r
                //Cl cl = new Cl(modulepath);\r
                //cl.execute("Fat.c");\r
@@ -187,19 +170,20 @@ public class ModuleInfo {
                //System.out.println("Note!!!! The CL is not implemented now , pls do it manually!!! RUN :");\r
                //System.out.println("cl " + modulepath + "\\temp\\*.c" + " -P");\r
                //String[] list = new File(modulepath + File.separator + "temp").list();        // without CL , add\r
-               String[] list = new File(modulepath).list();\r
-               for (int i = 0 ; i < list.length ; i++) {\r
-                       if (list[i].contains(".c")) {                                                                                   // without CL , change to .i\r
-                               preprocessedccodes.add(list[i]);\r
-                       }\r
-               }\r
-               //\r
-               Iterator<String> ii = preprocessedccodes.iterator();\r
                BufferedReader rd = null;\r
                String ifile = null;\r
                String line = null;\r
                String temp = null;\r
-               //StringBuffer result = new StringBuffer();\r
+               \r
+               Iterator<String> ii = localmodulesources.iterator();\r
+               while (ii.hasNext()) {\r
+                       temp = ii.next();\r
+                       if (temp.contains(".c")) {\r
+                               preprocessedccodes.add(temp);\r
+                       }\r
+               }\r
+               \r
+               ii = preprocessedccodes.iterator();\r
                \r
                Pattern patefifuncc = Pattern.compile("g?(BS|RT)\\s*->\\s*([a-zA-Z_]\\w*)",Pattern.MULTILINE);\r
                Pattern patentrypoint = Pattern.compile("EFI_([A-Z]*)_ENTRY_POINT\\s*\\(([^\\(\\)]*)\\)",Pattern.MULTILINE);\r
@@ -291,4 +275,4 @@ public class ModuleInfo {
        public static void main(String[] args) throws Exception {\r
                FirstPanel.init();\r
        }\r
-}\r
+}
\ No newline at end of file