BaseTools/UPT: Fix UNI file name issue

[mirror_edk2.git] / BaseTools / Source / Python / UPT / Object / POM / ModuleObject.py

diff --git a/BaseTools/Source/Python/UPT/Object/POM/ModuleObject.py b/BaseTools/Source/Python/UPT/Object/POM/ModuleObject.py
index 620bbb411f0e406f74307401171930885a0bb941..4052d2883596d01971984428f491441a7fb6db7c 100644 (file)
--- a/BaseTools/Source/Python/UPT/Object/POM/ModuleObject.py
+++ b/BaseTools/Source/Python/UPT/Object/POM/ModuleObject.py
@@ -1,7 +1,7 @@
 ## @file\r
 # This file is used to define a class object to describe a module\r
 #\r
-# Copyright (c) 2011, Intel Corporation. All rights reserved.<BR>\r
+# Copyright (c) 2011 - 2017, 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
@@ -21,6 +21,7 @@ ModuleObject
 from Object.POM.CommonObject import CommonPropertiesObject\r
 from Object.POM.CommonObject import IdentificationObject\r
 from Object.POM.CommonObject import CommonHeaderObject\r
+from Object.POM.CommonObject import BinaryHeaderObject\r
 from Object.POM.CommonObject import HelpTextListObject\r
 from Object.POM.CommonObject import GuidVersionObject\r
                          \r
@@ -94,7 +95,7 @@ class SpecObject(object):
 #\r
 # This class defined header items used in Module file\r
 # \r
-class ModuleHeaderObject(IdentificationObject, CommonHeaderObject):\r
+class ModuleHeaderObject(IdentificationObject, CommonHeaderObject, BinaryHeaderObject):\r
     def __init__(self):\r
         self.IsLibrary = False\r
         self.IsLibraryModList = []\r
@@ -103,6 +104,8 @@ class ModuleHeaderObject(IdentificationObject, CommonHeaderObject):
         self.PcdIsDriver = ''\r
         self.PiSpecificationVersion = ''\r
         self.UefiSpecificationVersion = ''\r
+        self.UNIFlag = False\r
+        self.ModuleUniFile = ''\r
         #\r
         # SpecObject\r
         #\r
@@ -126,6 +129,7 @@ class ModuleHeaderObject(IdentificationObject, CommonHeaderObject):
         self.SupArchList = []\r
         IdentificationObject.__init__(self)\r
         CommonHeaderObject.__init__(self)\r
+        BinaryHeaderObject.__init__(self)\r
     \r
     def SetIsLibrary(self, IsLibrary):\r
         self.IsLibrary = IsLibrary\r
@@ -205,6 +209,11 @@ class ModuleHeaderObject(IdentificationObject, CommonHeaderObject):
     def GetSupArchList(self):\r
         return self.SupArchList\r
 \r
+    def SetModuleUniFile(self, ModuleUniFile):\r
+        self.ModuleUniFile = ModuleUniFile\r
+\r
+    def GetModuleUniFile(self):\r
+        return self.ModuleUniFile\r
 ##\r
 # SourceFileObject\r
 #\r
@@ -277,6 +286,7 @@ class AsBuildLibraryClassObject(object):
     def __init__(self):\r
         self.LibGuid = ''\r
         self.LibVersion = ''\r
+        self.SupArchList = []\r
     \r
     def SetLibGuid(self, LibGuid):\r
         self.LibGuid = LibGuid\r
@@ -288,6 +298,11 @@ class AsBuildLibraryClassObject(object):
     def GetLibVersion(self):\r
         return self.LibVersion\r
 \r
+    def SetSupArchList(self, SupArchList):\r
+        self.SupArchList = SupArchList\r
+    def GetSupArchList(self):\r
+        return self.SupArchList\r
+\r
 ##\r
 # AsBuiltObject\r
 #\r
@@ -308,7 +323,7 @@ class AsBuiltObject(object):
         #\r
         # List of BinaryBuildFlag object\r
         #\r
-        self.BinaryBuildFlagList = ''\r
+        self.BinaryBuildFlagList = []\r
 \r
     def SetPatchPcdList(self, PatchPcdList):\r
         self.PatchPcdList = PatchPcdList\r