Fix several bugs & enhancement.

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

diff --git a/Tools/Java/Source/MigrationTools/org/tianocore/migration/MsaWriter.java b/Tools/Java/Source/MigrationTools/org/tianocore/migration/MsaWriter.java
index 3db25079057b4d98851bbd705448a9ce6693b4fd..132d6c3bc5a59b9d58da1fcdbd4c41d3f4830e90 100644 (file)
--- a/Tools/Java/Source/MigrationTools/org/tianocore/migration/MsaWriter.java
+++ b/Tools/Java/Source/MigrationTools/org/tianocore/migration/MsaWriter.java
@@ -70,11 +70,20 @@ public class MsaWriter {
             filename.setSupArchList(arch);\r
         }\r
     }\r
-    \r
+    private void addWrapper() {\r
+        XmlCursor cursor = msa.newCursor();\r
+        String uri = "http://www.TianoCore.org/2006/Edk2.0";\r
+        cursor.push();\r
+        cursor.toNextToken();\r
+        cursor.insertNamespace("", uri);\r
+        cursor.insertNamespace("xsi", "http://www.w3.org/2001/XMLSchema-instance");\r
+        cursor.pop();\r
+        msa = (ModuleSurfaceAreaDocument.ModuleSurfaceArea)cursor.getObject();\r
+    }\r
     private ModuleSurfaceAreaDocument fulfillMsadoc() throws Exception {\r
         Iterator<String> it;\r
         String temp;\r
-        \r
+       \r
         if (mi.modulename != null) {\r
             msaheader.setModuleName(mi.modulename);\r
         } else {\r
@@ -173,7 +182,8 @@ public class MsaWriter {
                 lc.setUsage(UsageTypes.ALWAYS_CONSUMED);\r
             }\r
         }\r
-        \r
+        addWrapper();\r
+        msadoc.setModuleSurfaceArea(msa);\r
         return msadoc;\r
     }\r
     \r
@@ -184,7 +194,7 @@ public class MsaWriter {
         options.setSavePrettyPrint();\r
         options.setSavePrettyPrintIndent(2);\r
         options.setUseDefaultNamespace();\r
-        \r
+\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