BaseTools: Support Structure PCD value assignment in DEC/DSC

[mirror_edk2.git] / BaseTools / Source / Python / Workspace / BuildClassObject.py

diff --git a/BaseTools/Source/Python/Workspace/BuildClassObject.py b/BaseTools/Source/Python/Workspace/BuildClassObject.py
index 5fa497b803d4e93e1c5db85b310c8389432badd1..da265b68f1af510e844dc3d804e53e1c3ff2c446 100644 (file)
--- a/BaseTools/Source/Python/Workspace/BuildClassObject.py
+++ b/BaseTools/Source/Python/Workspace/BuildClassObject.py
@@ -16,6 +16,7 @@ import Common.LongFilePathOs as os
 from Common.Misc import sdict\r
 from Common.Misc import RealPath2\r
 from Common.BuildToolError import *\r
+import collections\r
 \r
 ## PcdClassObject\r
 #\r
@@ -106,6 +107,63 @@ class PcdClassObject(object):
     def __hash__(self):\r
         return hash((self.TokenCName, self.TokenSpaceGuidCName))\r
 \r
+class StructurePcd(PcdClassObject):\r
+    def __init__(self, StructuredPcdIncludeFile="", Packages=None, Name=None, Guid=None, Type=None, DatumType=None, Value=None, Token=None, MaxDatumSize=None, SkuInfoList={}, IsOverrided=False, GuidValue=None, validateranges=[], validlists=[], expressions=[]):\r
+        super(StructurePcd, self).__init__(Name, Guid, Type, DatumType, Value, Token, MaxDatumSize, SkuInfoList, IsOverrided, GuidValue, validateranges, validlists, expressions)\r
+        self.StructuredPcdIncludeFile = StructuredPcdIncludeFile\r
+        self.PackageDecs = Packages\r
+        self.DefaultStoreName = ['STANDARD']\r
+        self.DefaultValues = collections.OrderedDict({})\r
+        self.PcdMode = None\r
+        self.SkuOverrideValues = collections.OrderedDict({})\r
+        self.FlexibleFieldName = None\r
+\r
+    def __repr__(self):\r
+        return self.TypeName\r
+\r
+    def AddDefaultValue (self, FieldName, Value, FileName="", LineNo=0):\r
+        self.DefaultValues[FieldName] = [Value.strip(), FileName, LineNo]\r
+        return self.DefaultValues[FieldName]\r
+\r
+    def AddOverrideValue (self, FieldName, Value, SkuName, FileName="", LineNo=0):\r
+        if SkuName not in self.SkuOverrideValues:\r
+            self.SkuOverrideValues[SkuName] = collections.OrderedDict({})\r
+        self.SkuOverrideValues[SkuName][FieldName] = [Value.strip(), FileName, LineNo]\r
+        return self.SkuOverrideValues[SkuName][FieldName]\r
+\r
+    def SetPcdMode (self, PcdMode):\r
+        self.PcdMode = PcdMode\r
+\r
+    def SetFlexibleFieldName (self, FlexibleFieldName):\r
+        self.FlexibleFieldName = FlexibleFieldName\r
+\r
+    def copy(self, PcdObject):\r
+        self.TokenCName = PcdObject.TokenCName if PcdObject.TokenCName else self.TokenCName\r
+        self.TokenSpaceGuidCName = PcdObject.TokenSpaceGuidCName if PcdObject.TokenSpaceGuidCName else PcdObject.TokenSpaceGuidCName\r
+        self.TokenSpaceGuidValue = PcdObject.TokenSpaceGuidValue if PcdObject.TokenSpaceGuidValue else self.TokenSpaceGuidValue\r
+        self.Type = PcdObject.Type if PcdObject.Type else self.Type\r
+        self.DatumType = PcdObject.DatumType if PcdObject.DatumType else self.DatumType\r
+        self.DefaultValue = PcdObject.DefaultValue if  PcdObject.DefaultValue else self.DefaultValue\r
+        self.TokenValue = PcdObject.TokenValue if PcdObject.TokenValue else self.TokenValue\r
+        self.MaxDatumSize = PcdObject.MaxDatumSize if PcdObject.MaxDatumSize else self.MaxDatumSize\r
+        self.SkuInfoList = PcdObject.SkuInfoList if PcdObject.SkuInfoList else self.SkuInfoList\r
+        self.Phase = PcdObject.Phase if PcdObject.Phase else self.Phase\r
+        self.Pending = PcdObject.Pending if PcdObject.Pending else self.Pending\r
+        self.IsOverrided = PcdObject.IsOverrided if PcdObject.IsOverrided else self.IsOverrided\r
+        self.IsFromBinaryInf = PcdObject.IsFromBinaryInf if PcdObject.IsFromBinaryInf else self.IsFromBinaryInf\r
+        self.IsFromDsc = PcdObject.IsFromDsc if PcdObject.IsFromDsc else self.IsFromDsc\r
+        self.validateranges = PcdObject.validateranges if PcdObject.validateranges else self.validateranges\r
+        self.validlists = PcdObject.validlists if PcdObject.validlists else self.validlists\r
+        self.expressions = PcdObject.expressions if PcdObject.expressions else self.expressions\r
+        if type(PcdObject) is StructurePcd:\r
+            self.StructuredPcdIncludeFile = PcdObject.StructuredPcdIncludeFile if PcdObject.StructuredPcdIncludeFile else self.StructuredPcdIncludeFile\r
+            self.PackageDecs = PcdObject.PackageDecs if PcdObject.PackageDecs else self.PackageDecs\r
+            self.DefaultValues = PcdObject.DefaultValues if PcdObject.DefaultValues else self.DefaultValues\r
+            self.PcdMode = PcdObject.PcdMode if PcdObject.PcdMode else self.PcdMode\r
+            self.DefaultFromDSC=None\r
+            self.OverrideValues = PcdObject.SkuOverrideValues if PcdObject.SkuOverrideValues else self.SkuOverrideValues\r
+            self.FlexibleFieldName = PcdObject.FlexibleFieldName if PcdObject.FlexibleFieldName else self.FlexibleFieldName\r
+\r
 ## LibraryClassObject\r
 #\r
 # This Class defines LibraryClassObject used in BuildDatabase\r