Sync BaseTools Branch (version r2321) to EDKII main trunk.

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

diff --git a/BaseTools/Source/Python/GenFds/CapsuleData.py b/BaseTools/Source/Python/GenFds/CapsuleData.py
index 85307b90c647624dcd6d788d443c66783cfc8781..aef8df0e16bd08f3b7e947942d2b3f20feac557e 100644 (file)
--- a/BaseTools/Source/Python/GenFds/CapsuleData.py
+++ b/BaseTools/Source/Python/GenFds/CapsuleData.py
@@ -43,7 +43,7 @@ class CapsuleFfs (CapsuleData):
     #\r
     #   @param  self        The object pointer\r
     #\r
-    def __init_(self) :\r
+    def __init__(self) :\r
         self.Ffs = None\r
         self.FvName = None\r
 \r
@@ -87,3 +87,51 @@ class CapsuleFv (CapsuleData):
         else:\r
             FvFile = GenFdsGlobalVariable.ReplaceWorkspaceMacro(self.FvName)\r
             return FvFile\r
+\r
+## FD class for capsule data\r
+#\r
+#\r
+class CapsuleFd (CapsuleData):\r
+    ## The constructor\r
+    #\r
+    #   @param  self        The object pointer\r
+    #\r
+    def __init__(self) :\r
+        self.Ffs = None\r
+        self.FdName = None\r
+        self.CapsuleName = None\r
+\r
+    ## generate FD capsule data\r
+    #\r
+    #   @param  self        The object pointer\r
+    #   @retval string      Generated file name\r
+    #\r
+    def GenCapsuleSubItem(self):\r
+        if self.FdName.find('.fd') == -1:\r
+            if self.FdName.upper() in GenFdsGlobalVariable.FdfParser.Profile.FdDict.keys():\r
+                FdObj = GenFdsGlobalVariable.FdfParser.Profile.FdDict.get(self.FdName.upper())\r
+                FdFile = FdObj.GenFd()\r
+                return FdFile\r
+        else:\r
+            FdFile = GenFdsGlobalVariable.ReplaceWorkspaceMacro(self.FdName)\r
+            return FdFile\r
+        \r
+## AnyFile class for capsule data\r
+#\r
+#\r
+class CapsuleAnyFile (CapsuleData):\r
+    ## The constructor\r
+    #\r
+    #   @param  self        The object pointer\r
+    #\r
+    def __init__(self) :\r
+        self.Ffs = None\r
+        self.FileName = None\r
+\r
+    ## generate AnyFile capsule data\r
+    #\r
+    #   @param  self        The object pointer\r
+    #   @retval string      Generated file name\r
+    #\r
+    def GenCapsuleSubItem(self):\r
+        return self.FileName
\ No newline at end of file