Check In tool source code based on Build tool project revision r1655.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@8964 6f19259b-4bc3-4df7-8a09-765794883524

BaseTools/Source/Python/GenFds/Capsule.py

@@ -0,0 +1,89 @@
+## @file
+# generate capsule
+#
+#  Copyright (c) 2007, Intel Corporation
+#
+#  All rights reserved. This program and the accompanying materials
+#  are licensed and made available under the terms and conditions of the BSD License
+#  which accompanies this distribution.  The full text of the license may be found at
+#  http://opensource.org/licenses/bsd-license.php
+#
+#  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+#
+
+##
+# Import Modules
+#
+from GenFdsGlobalVariable import GenFdsGlobalVariable
+from CommonDataClass.FdfClass import CapsuleClassObject
+import os
+import subprocess
+import StringIO
+from Common.Misc import SaveFileOnChange
+
+
+T_CHAR_LF = '\n'
+
+## create inf file describes what goes into capsule and call GenFv to generate capsule
+#
+#
+class Capsule (CapsuleClassObject) :
+    ## The constructor
+    #
+    #   @param  self        The object pointer
+    #
+    def __init__(self):
+        CapsuleClassObject.__init__(self)
+        # For GenFv
+        self.BlockSize = None
+        # For GenFv
+        self.BlockNum = None
+
+    ## Generate capsule
+    #
+    #   @param  self        The object pointer
+    #
+    def GenCapsule(self):
+        CapInfFile = self.GenCapInf()
+        CapInfFile.writelines("[files]" + T_CHAR_LF)
+
+        for CapsuleDataObj in self.CapsuleDataList :
+            FileName = CapsuleDataObj.GenCapsuleSubItem()
+            CapInfFile.writelines("EFI_FILE_NAME = " + \
+                                   FileName      + \
+                                   T_CHAR_LF)
+        SaveFileOnChange(self.CapInfFileName, CapInfFile.getvalue(), False)
+        CapInfFile.close()
+        #
+        # Call GenFv tool to generate capsule
+        #
+        CapOutputFile = os.path.join(GenFdsGlobalVariable.FvDir, self.UiCapsuleName)
+        CapOutputFile = CapOutputFile + '.Cap'
+        GenFdsGlobalVariable.GenerateFirmwareVolume(
+                                CapOutputFile,
+                                [self.CapInfFileName],
+                                Capsule=True
+                                )
+        GenFdsGlobalVariable.SharpCounter = 0
+
+    ## Generate inf file for capsule
+    #
+    #   @param  self        The object pointer
+    #   @retval file        inf file object
+    #
+    def GenCapInf(self):
+        self.CapInfFileName = os.path.join(GenFdsGlobalVariable.FvDir,
+                                   self.UiCapsuleName +  "_Cap" + '.inf')
+        CapInfFile = StringIO.StringIO() #open (self.CapInfFileName , 'w+')
+
+        CapInfFile.writelines("[options]" + T_CHAR_LF)
+
+        for Item in self.TokensDict.keys():
+            CapInfFile.writelines("EFI_"                    + \
+                                  Item                      + \
+                                  ' = '                     + \
+                                  self.TokensDict.get(Item) + \
+                                  T_CHAR_LF)
+
+        return CapInfFile