BaseTools: Customize deepcopy function.

[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 15bb82291094937604f54a973bee8cd8c1212ba7..008eee1a1664b4ba1fe16318d34a638ab88ec70e 100644 (file)
--- a/BaseTools/Source/Python/Workspace/BuildClassObject.py
+++ b/BaseTools/Source/Python/Workspace/BuildClassObject.py
@@ -16,6 +16,8 @@ from Common.DataType import *
 import collections\r
 import re\r
 from collections import OrderedDict\r
+from Common.Misc import CopyDict\r
+import copy\r
 StructPattern = re.compile(r'[_a-zA-Z][0-9A-Za-z_\[\]]*$')\r
 ArrayIndex = re.compile("\[\s*\d{0,1}\s*\]")\r
 ## PcdClassObject\r
@@ -211,6 +213,37 @@ class PcdClassObject(object):
     def __hash__(self):\r
         return hash((self.TokenCName, self.TokenSpaceGuidCName))\r
 \r
+    def sharedcopy(self,new_pcd):\r
+        new_pcd.TokenCName = self.TokenCName\r
+        new_pcd.TokenSpaceGuidCName = self.TokenSpaceGuidCName\r
+        new_pcd.TokenSpaceGuidValue = self.TokenSpaceGuidValue\r
+        new_pcd.Type = self.Type\r
+        new_pcd.DatumType = self.DatumType\r
+        new_pcd.DefaultValue = self.DefaultValue\r
+        new_pcd.TokenValue = self.TokenValue\r
+        new_pcd.MaxDatumSize = self.MaxDatumSize\r
+\r
+        new_pcd.Phase = self.Phase\r
+        new_pcd.Pending = self.Pending\r
+        new_pcd.IsOverrided = self.IsOverrided\r
+        new_pcd.IsFromBinaryInf = self.IsFromBinaryInf\r
+        new_pcd.IsFromDsc = self.IsFromDsc\r
+        new_pcd.PcdValueFromComm = self.PcdValueFromComm\r
+        new_pcd.PcdValueFromFdf = self.PcdValueFromFdf\r
+        new_pcd.UserDefinedDefaultStoresFlag = self.UserDefinedDefaultStoresFlag\r
+        new_pcd.DscRawValue = self.DscRawValue\r
+        new_pcd.CustomAttribute = self.CustomAttribute\r
+        new_pcd.validateranges = [item for item in self.validateranges]\r
+        new_pcd.validlists = [item for item in self.validlists]\r
+        new_pcd.expressions = [item for item in self.expressions]\r
+        new_pcd.SkuInfoList = {key: copy.deepcopy(skuobj) for key,skuobj in self.SkuInfoList.items()}\r
+        return new_pcd\r
+\r
+    def __deepcopy__(self,memo):\r
+        new_pcd = PcdClassObject()\r
+        self.sharedcopy(new_pcd)\r
+        return new_pcd\r
+\r
 class StructurePcd(PcdClassObject):\r
     def __init__(self, StructuredPcdIncludeFile=None, Packages=None, Name=None, Guid=None, Type=None, DatumType=None, Value=None, Token=None, MaxDatumSize=None, SkuInfoList=None, IsOverrided=False, GuidValue=None, validateranges=None, validlists=None, expressions=None,default_store = TAB_DEFAULT_STORES_DEFAULT):\r
         if SkuInfoList is None:\r
@@ -303,6 +336,25 @@ class StructurePcd(PcdClassObject):
             self.PcdFieldValueFromComm = PcdObject.PcdFieldValueFromComm if PcdObject.PcdFieldValueFromComm else self.PcdFieldValueFromComm\r
             self.PcdFieldValueFromFdf = PcdObject.PcdFieldValueFromFdf if PcdObject.PcdFieldValueFromFdf else self.PcdFieldValueFromFdf\r
 \r
+    def __deepcopy__(self,memo):\r
+        new_pcd = StructurePcd()\r
+        self.sharedcopy(new_pcd)\r
+\r
+        new_pcd.DefaultValueFromDec = self.DefaultValueFromDec\r
+        new_pcd.PcdMode = self.PcdMode\r
+        new_pcd.StructName = self.DatumType\r
+        new_pcd.PcdDefineLineNo = self.PcdDefineLineNo\r
+        new_pcd.PkgPath = self.PkgPath\r
+        new_pcd.StructuredPcdIncludeFile = [item for item in self.StructuredPcdIncludeFile]\r
+        new_pcd.PackageDecs = [item for item in self.PackageDecs]\r
+        new_pcd.DefaultValues = CopyDict(self.DefaultValues)\r
+        new_pcd.DefaultFromDSC=CopyDict(self.DefaultFromDSC)\r
+        new_pcd.SkuOverrideValues = CopyDict(self.SkuOverrideValues)\r
+        new_pcd.PcdFieldValueFromComm = CopyDict(self.PcdFieldValueFromComm)\r
+        new_pcd.PcdFieldValueFromFdf = CopyDict(self.PcdFieldValueFromFdf)\r
+        new_pcd.ValueChain = {item for item in self.ValueChain}\r
+        return new_pcd\r
+\r
 LibraryClassObject = namedtuple('LibraryClassObject', ['LibraryClass','SupModList'], verbose=False)\r
 \r
 ## ModuleBuildClassObject\r