BaseTool/Upt: Add support for Private

[mirror_edk2.git] / BaseTools / Source / Python / UPT / PomAdapter / DecPomAlignment.py

diff --git a/BaseTools/Source/Python/UPT/PomAdapter/DecPomAlignment.py b/BaseTools/Source/Python/UPT/PomAdapter/DecPomAlignment.py
index 11b0359176c60bc284d6f37b1efa15f60a4da973..436dc90e6dd3a48880e53cfd6157262cc1abb70c 100644 (file)
--- a/BaseTools/Source/Python/UPT/PomAdapter/DecPomAlignment.py
+++ b/BaseTools/Source/Python/UPT/PomAdapter/DecPomAlignment.py
@@ -1,7 +1,7 @@
 ## @file DecPomAlignment.py\r
 # This file contained the adapter for convert INF parser object to POM Object\r
 #\r
-# Copyright (c) 2011 - 2014, Intel Corporation. All rights reserved.<BR>\r
+# Copyright (c) 2011 - 2016, Intel Corporation. All rights reserved.<BR>\r
 #\r
 # This program and the accompanying materials are licensed and made available \r
 # under the terms and conditions of the BSD License which accompanies this \r
@@ -63,6 +63,7 @@ from Library.DataType import TAB_STR_TOKENHELP
 from Library.DataType import TAB_STR_TOKENERR\r
 from Library.DataType import TAB_HEX_START\r
 from Library.DataType import TAB_SPLIT\r
+import Library.DataType as DT\r
 from Library.CommentParsing import ParseHeaderCommentSection\r
 from Library.CommentParsing import ParseGenericComment\r
 from Library.CommentParsing import ParseDecPcdGenericComment\r
@@ -221,6 +222,14 @@ class DecPomAlignment(PackageObject):
             self.SetUserExtensionList(\r
                 self.GetUserExtensionList() + [UserExtension]\r
             )\r
+\r
+        # Add Private sections to UserExtension\r
+        if self.DecParser.GetPrivateSections():\r
+            PrivateUserExtension = UserExtensionObject()\r
+            PrivateUserExtension.SetStatement(self.DecParser.GetPrivateSections())\r
+            PrivateUserExtension.SetIdentifier(DT.TAB_PRIVATE)\r
+            PrivateUserExtension.SetUserID(DT.TAB_INTEL)\r
+            self.SetUserExtensionList(self.GetUserExtensionList() + [PrivateUserExtension])\r
             \r
     ## Generate miscellaneous files on DEC file\r
     #\r