let critic prepared

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

diff --git a/Tools/Source/MigrationTools/org/tianocore/migration/MsaWriter.java b/Tools/Source/MigrationTools/org/tianocore/migration/MsaWriter.java
index 7d907690e902b51cc9f28631f45a7aef8e3de674..aeed32d345673e84cf8ebdc1d87b05ba37dd54ae 100644 (file)
--- a/Tools/Source/MigrationTools/org/tianocore/migration/MsaWriter.java
+++ b/Tools/Source/MigrationTools/org/tianocore/migration/MsaWriter.java
@@ -20,15 +20,11 @@ import org.tianocore.SupportedArchitectures.Enum;
 import org.apache.xmlbeans.*;\r
 \r
 public class MsaWriter {\r
-       MsaWriter(String path, ModuleInfo moduleinfo, Database database) {\r
-               modulepath = path;\r
+       MsaWriter(ModuleInfo moduleinfo) {\r
                mi = moduleinfo;\r
-               db = database;\r
        }\r
 \r
-       private String modulepath;\r
        private ModuleInfo mi;\r
-       private Database db;\r
        \r
        private ModuleSurfaceAreaDocument msadoc = ModuleSurfaceAreaDocument.Factory.newInstance();\r
        \r
@@ -51,6 +47,31 @@ public class MsaWriter {
                return answer;\r
        }\r
        \r
+       private void addSourceFiles (String name) {                                             // furthur modification needed\r
+               List<Enum> arch = new ArrayList<Enum>();\r
+               FilenameDocument.Filename filename;\r
+               filename = sourcefiles.addNewFilename();\r
+               filename.setStringValue(name);\r
+               \r
+               if (name.contains("x64" + File.separator)) {                            // filename ???\r
+                       arch.add(SupportedArchitectures.X_64);\r
+                       System.out.println("x64" + File.separator);\r
+                       filename.setSupArchList(arch);\r
+               } else if (name.contains("Ia32" + File.separator)) {            // filename ???\r
+                       arch.add(SupportedArchitectures.IA_32);\r
+                       System.out.println("Ia32" + File.separator);\r
+                       filename.setSupArchList(arch);\r
+               } else if (name.contains("Ipf" + File.separator)) {                     // filename ???\r
+                       arch.add(SupportedArchitectures.IPF);\r
+                       System.out.println("Ipf" + File.separator);\r
+                       filename.setSupArchList(arch);\r
+               } else if (name.contains("Ebc" + File.separator)) {                     // filename ???\r
+                       arch.add(SupportedArchitectures.EBC);\r
+                       System.out.println("Ebc" + File.separator);\r
+                       filename.setSupArchList(arch);\r
+               }\r
+       }\r
+       \r
        private ModuleSurfaceAreaDocument fulfillMsadoc() throws Exception {\r
                Iterator<String> it;\r
                String temp;\r
@@ -78,7 +99,7 @@ public class MsaWriter {
                msaheader.setCopyright("Copyright (c) 2006, Intel Corporation");\r
                msaheader.setVersion("1.0");\r
                msaheader.setAbstract("Component name for module " + mi.modulename);\r
-               msaheader.setDescription("FIX ME!");                                                    //???\r
+               msaheader.setDescription("FIX ME!");\r
                msaheader.addNewLicense().setStringValue("All rights reserved.\n" +\r
                                "      This software and associated documentation (if any) is furnished\n" +\r
                                "      under a license and may only be used or copied in accordance\n" +\r
@@ -105,7 +126,7 @@ public class MsaWriter {
                \r
                it = mi.localmodulesources.iterator();\r
                while (it.hasNext()) {\r
-                       sourcefiles.addNewFilename().setStringValue(it.next());\r
+                       addSourceFiles(it.next());\r
                }\r
                if (!mi.protocol.isEmpty()) {\r
                        protocols = msa.addNewProtocols();\r
@@ -160,9 +181,29 @@ public class MsaWriter {
         options.setSavePrettyPrintIndent(2);\r
         options.setUseDefaultNamespace();\r
         \r
-               BufferedWriter bw = new BufferedWriter(new FileWriter(modulepath + File.separator + "result" + File.separator + mi.modulename + ".msa"));\r
+               BufferedWriter bw = new BufferedWriter(new FileWriter(MigrationTool.ModuleInfoMap.get(mi) + File.separator + "Migration_" + mi.modulename + File.separator + mi.modulename + ".msa"));\r
                fulfillMsadoc().save(bw, options);\r
+               //MsaTreeEditor.init(mi, ui, msadoc);\r
+               bw.flush();\r
+               bw.close();\r
+       }\r
+\r
+       private static void flush(String path, ModuleSurfaceAreaDocument msadoc) throws Exception {\r
+        XmlOptions options = new XmlOptions();\r
+\r
+        options.setCharacterEncoding("UTF-8");\r
+        options.setSavePrettyPrint();\r
+        options.setSavePrettyPrintIndent(2);\r
+        options.setUseDefaultNamespace();\r
+\r
+               BufferedWriter bw = new BufferedWriter(new FileWriter(path));\r
+               msadoc.save(bw, options);\r
                bw.flush();\r
                bw.close();\r
        }\r
+       \r
+       public static final void parse(String msafile) throws Exception {\r
+               ModuleSurfaceAreaDocument msadoc = ModuleSurfaceAreaDocument.Factory.parse(msafile);\r
+               flush("c:\\temp.msa", msadoc);\r
+       }\r
 }\r