BaseTools: Move OverrideAttribs to OptRomInfStatement.py

[mirror_edk2.git] / BaseTools / Source / Python / GenFds / OptRomInfStatement.py

diff --git a/BaseTools/Source/Python/GenFds/OptRomInfStatement.py b/BaseTools/Source/Python/GenFds/OptRomInfStatement.py
index dff8235ef755800055c4e6c192b6a9a6067039eb..e416b838d1e45b354bbdd0a60f354d0ed96a7dd4 100644 (file)
--- a/BaseTools/Source/Python/GenFds/OptRomInfStatement.py
+++ b/BaseTools/Source/Python/GenFds/OptRomInfStatement.py
@@ -15,16 +15,16 @@
 ##\r
 # Import Modules\r
 #\r
-import RuleSimpleFile\r
-import RuleComplexFile\r
-import Section\r
-import OptionRom\r
+from __future__ import absolute_import\r
+from . import RuleSimpleFile\r
+from . import RuleComplexFile\r
+from . import Section\r
 import Common.GlobalData as GlobalData\r
 \r
 from Common.DataType import *\r
 from Common.StringUtils import *\r
-from FfsInfStatement import FfsInfStatement\r
-from GenFdsGlobalVariable import GenFdsGlobalVariable\r
+from .FfsInfStatement import FfsInfStatement\r
+from .GenFdsGlobalVariable import GenFdsGlobalVariable\r
 \r
 ##\r
 #\r
@@ -45,9 +45,8 @@ class OptRomInfStatement (FfsInfStatement):
     #   @param  self        The object pointer\r
     #\r
     def __GetOptRomParams(self):\r
-\r
         if self.OverrideAttribs is None:\r
-            self.OverrideAttribs = OptionRom.OverrideAttribs()\r
+            self.OverrideAttribs = OverrideAttribs()\r
 \r
         if self.OverrideAttribs.NeedCompress is None:\r
             self.OverrideAttribs.NeedCompress = self.OptRomDefs.get ('PCI_COMPRESS')\r
@@ -151,4 +150,16 @@ class OptRomInfStatement (FfsInfStatement):
 \r
         return OutputFileList\r
 \r
+class OverrideAttribs:\r
+\r
+    ## The constructor\r
+    #\r
+    #   @param  self        The object pointer\r
+    #\r
+    def __init__(self):\r
 \r
+        self.PciVendorId = None\r
+        self.PciClassCode = None\r
+        self.PciDeviceId = None\r
+        self.PciRevision = None\r
+        self.NeedCompress = None\r