Changed GenFds to use database based module/package/platform information retrieval...
[people/mcb30/basetools.git] / Source / Python / GenFds / AprioriSection.py
1 ## @file\r
2 # process APRIORI file data and generate PEI/DXE APRIORI file\r
3 #\r
4 #  Copyright (c) 2007, Intel Corporation\r
5 #\r
6 #  All rights reserved. This program and the accompanying materials\r
7 #  are licensed and made available under the terms and conditions of the BSD License\r
8 #  which accompanies this distribution.  The full text of the license may be found at\r
9 #  http://opensource.org/licenses/bsd-license.php\r
10 #\r
11 #  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
12 #  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
13 #\r
14 \r
15 ##\r
16 # Import Modules\r
17 #\r
18 from struct import *\r
19 import os\r
20 import StringIO\r
21 import FfsFileStatement\r
22 from GenFdsGlobalVariable import GenFdsGlobalVariable\r
23 from CommonDataClass.FdfClass import AprioriSectionClassObject\r
24 from Common.String import *\r
25 \r
26 ## process APRIORI file data and generate PEI/DXE APRIORI file\r
27 #\r
28 #\r
29 class AprioriSection (AprioriSectionClassObject):\r
30     ## The constructor\r
31     #\r
32     #   @param  self        The object pointer\r
33     #\r
34     def __init__(self):\r
35         AprioriSectionClassObject.__init__(self)\r
36         self.AprioriType = ""\r
37     \r
38     ## GenFfs() method\r
39     #\r
40     #   Generate FFS for APRIORI file\r
41     #\r
42     #   @param  self        The object pointer\r
43     #   @param  FvName      for whom apriori file generated\r
44     #   @param  Dict        dictionary contains macro and its value\r
45     #   @retval string      Generated file name\r
46     #\r
47     def GenFfs (self, FvName, Dict = {}):\r
48         DXE_GUID = "FC510EE7-FFDC-11D4-BD41-0080C73C8881"\r
49         PEI_GUID = "1B45CC0A-156A-428A-AF62-49864DA0E6E6"\r
50         Buffer = StringIO.StringIO('')\r
51         AprioriFileGuid = DXE_GUID\r
52         if self.AprioriType == "PEI":\r
53             AprioriFileGuid = PEI_GUID\r
54         OutputAprFilePath = os.path.join (GenFdsGlobalVariable.WorkSpaceDir, \\r
55                                    GenFdsGlobalVariable.FfsDir,\\r
56                                    AprioriFileGuid + FvName)\r
57         if not os.path.exists(OutputAprFilePath) :\r
58             os.makedirs(OutputAprFilePath)\r
59             \r
60         OutputAprFileName = os.path.join( OutputAprFilePath, \\r
61                                        AprioriFileGuid + FvName + '.Apri' )\r
62         AprFfsFileName = os.path.join (OutputAprFilePath,\\r
63                                     AprioriFileGuid + FvName + '.Ffs')\r
64                                    \r
65         OutputAprFile = open(OutputAprFileName, 'w+b')\r
66         \r
67         Dict.update(self.DefineVarDict)\r
68         \r
69         for FfsObj in self.FfsList :\r
70             Guid = ""\r
71             if isinstance(FfsObj, FfsFileStatement.FileStatement):\r
72                 Guid = FfsObj.NameGuid\r
73             else:\r
74                 InfFileName = NormPath(FfsObj.InfFileName)\r
75                 Arch = FfsObj.GetCurrentArch()\r
76                 \r
77                 if Arch != None:\r
78                     Dict['$(ARCH)'] = Arch\r
79                 InfFileName = GenFdsGlobalVariable.MacroExtend(InfFileName, Dict, Arch)\r
80 \r
81                 if Arch != None:\r
82                     Inf = GenFdsGlobalVariable.WorkSpace.BuildObject[InfFileName, Arch]\r
83                     Guid = Inf.Guid\r
84         \r
85                 else:\r
86                     Inf = GenFdsGlobalVariable.WorkSpace.BuildObject[InfFileName, 'COMMON']\r
87                     Guid = Inf.Guid\r
88                     \r
89                     self.BinFileList = Inf.Module.Binaries\r
90                     if self.BinFileList == []:\r
91                         raise Exception ("INF %s not found in build ARCH %s!" % (InfFileName, GenFdsGlobalVariable.ArchList))\r
92                         sys.exit(1)\r
93             \r
94                 \r
95             GuidPart = Guid.split('-')\r
96             Buffer.write(pack('I', long(GuidPart[0], 16)))\r
97             Buffer.write(pack('H', int(GuidPart[1], 16)))\r
98             Buffer.write(pack('H', int(GuidPart[2], 16)))\r
99             \r
100             for Num in range(2):\r
101                 Char = GuidPart[3][Num*2:Num*2+2]\r
102                 Buffer.write(pack('B', int(Char, 16)))\r
103             \r
104             for Num in range(6):\r
105                 Char = GuidPart[4][Num*2:Num*2+2]\r
106                 Buffer.write(pack('B', int(Char, 16)))\r
107     \r
108         OutputAprFile.write(Buffer.getvalue())\r
109         OutputAprFile.close()\r
110         RawSectionFileName = os.path.join( OutputAprFilePath, \\r
111                                        AprioriFileGuid + FvName + '.raw' )\r
112         \r
113         GenSectionCmd = (\r
114             'GenSec',\r
115             '-o', RawSectionFileName,\r
116             '-s', 'EFI_SECTION_RAW',\r
117             OutputAprFileName,\r
118             )\r
119         \r
120         GenFdsGlobalVariable.CallExternalTool(GenSectionCmd, "GenSection Failed!")\r
121         \r
122         GenFfsCmd = (\r
123             'GenFfs',\r
124             '-t', 'EFI_FV_FILETYPE_FREEFORM',\r
125             '-g', AprioriFileGuid,\r
126             '-o', AprFfsFileName,\r
127             '-i', RawSectionFileName,\r
128             )\r
129         \r
130         GenFdsGlobalVariable.CallExternalTool(GenFfsCmd,"GenFfs Failed !")\r
131         return AprFfsFileName\r
132             \r