BaseTools: DataType - cleanup list constants

[mirror_edk2.git] / BaseTools / Source / Python / AutoGen / ValidCheckingInfoObject.py

diff --git a/BaseTools/Source/Python/AutoGen/ValidCheckingInfoObject.py b/BaseTools/Source/Python/AutoGen/ValidCheckingInfoObject.py
index 92ede7a82324f59a82028368da2075487005e7b2..df7a9b889aa4f74c0d5a058d2e16571b18094020 100644 (file)
--- a/BaseTools/Source/Python/AutoGen/ValidCheckingInfoObject.py
+++ b/BaseTools/Source/Python/AutoGen/ValidCheckingInfoObject.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2015, Intel Corporation. All rights reserved.<BR>\r
+# Copyright (c) 2015 - 2018, Intel Corporation. All rights reserved.<BR>\r
 # This program and the accompanying materials\r
 # are licensed and made available under the terms and conditions of the BSD License\r
 # which accompanies this distribution.  The full text of the license may be found at\r
@@ -19,6 +19,7 @@ from Common.RangeExpression import RangeExpression
 from Common.Misc import *\r
 from StringIO import StringIO\r
 from struct import pack\r
+from Common.DataType import *\r
 \r
 class VAR_CHECK_PCD_VARIABLE_TAB_CONTAINER(object):\r
     def __init__(self):\r
@@ -250,13 +251,13 @@ class VAR_CHECK_PCD_VALID_OBJ(object):
         self.ValidData = True\r
         self.updateStorageWidth()\r
     def updateStorageWidth(self):\r
-        if self.PcdDataType == "UINT8" or self.PcdDataType == "BOOLEAN":\r
+        if self.PcdDataType == TAB_UINT8 or self.PcdDataType == "BOOLEAN":\r
             self.StorageWidth = 1\r
-        elif self.PcdDataType == "UINT16":\r
+        elif self.PcdDataType == TAB_UINT16:\r
             self.StorageWidth = 2\r
-        elif self.PcdDataType == "UINT32":\r
+        elif self.PcdDataType == TAB_UINT32:\r
             self.StorageWidth = 4\r
-        elif self.PcdDataType == "UINT64":\r
+        elif self.PcdDataType == TAB_UINT64:\r
             self.StorageWidth = 8\r
         else:\r
             self.StorageWidth = 0\r