[mirror_edk2.git] / BaseTools / Source / Python / Common / VpdInfoFile.py

## @file\r
# \r
# This package manage the VPD PCD information file which will be generated\r
# by build tool's autogen.\r
# The VPD PCD information file will be input for third-party BPDG tool which\r
# is pointed by *_*_*_VPD_TOOL_GUID in conf/tools_def.txt \r
#\r
#\r
# Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
# This program and the accompanying materials\r
# are licensed and made available under the terms and conditions of the BSD License\r
# which accompanies this distribution.  The full text of the license may be found at\r
# http://opensource.org/licenses/bsd-license.php\r
#\r
# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
#\r
import Common.LongFilePathOs as os\r
import re\r
import Common.EdkLogger as EdkLogger\r
import Common.BuildToolError as BuildToolError\r
import subprocess\r
from Common.LongFilePathSupport import OpenLongFilePath as open\r
\r
FILE_COMMENT_TEMPLATE = \\r
"""\r
## @file\r
#\r
#  THIS IS AUTO-GENERATED FILE BY BUILD TOOLS AND PLEASE DO NOT MAKE MODIFICATION.\r
#\r
#  This file lists all VPD informations for a platform collected by build.exe.\r
# \r
# Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
# This program and the accompanying materials\r
# are licensed and made available under the terms and conditions of the BSD License\r
# which accompanies this distribution.  The full text of the license may be found at\r
# http://opensource.org/licenses/bsd-license.php\r
#\r
# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
#\r
\r
"""\r
\r
## The class manage VpdInfoFile.\r
#\r
#  This file contains an ordered (based on position in the DSC file) list of the PCDs specified in the platform description file (DSC). The Value field that will be assigned to the PCD comes from the DSC file, INF file (if not defined in the DSC file) or the DEC file (if not defined in the INF file). This file is used as an input to the BPDG tool.\r
#  Format for this file (using EBNF notation) is:\r
#  <File>            :: = [<CommentBlock>]\r
#                         [<PcdEntry>]*\r
#  <CommentBlock>    ::=  ["#" <String> <EOL>]*\r
#  <PcdEntry>        ::=  <PcdName> "|" <Offset> "|" <Size> "|" <Value> <EOL>\r
#  <PcdName>         ::=  <TokenSpaceCName> "." <PcdCName>\r
#  <TokenSpaceCName> ::=  C Variable Name of the Token Space GUID\r
#  <PcdCName>        ::=  C Variable Name of the PCD\r
#  <Offset>          ::=  {"*"} {<HexNumber>}\r
#  <HexNumber>       ::=  "0x" (a-fA-F0-9){1,8}\r
#  <Size>            ::=  <HexNumber>\r
#  <Value>           ::=  {<HexNumber>} {<NonNegativeInt>} {<QString>} {<Array>}\r
#  <NonNegativeInt>  ::=  (0-9)+\r
#  <QString>         ::=  ["L"] <DblQuote> <String> <DblQuote>\r
#  <DblQuote>        ::=  0x22\r
#  <Array>           ::=  {<CArray>} {<NList>}\r
#  <CArray>          ::=  "{" <HexNumber> ["," <HexNumber>]* "}"\r
#  <NList>           ::=  <HexNumber> ["," <HexNumber>]*\r
#\r
class VpdInfoFile:\r
    \r
    ## The mapping dictionary from datum type to size string.\r
    _MAX_SIZE_TYPE = {"BOOLEAN":"1", "UINT8":"1", "UINT16":"2", "UINT32":"4", "UINT64":"8"}\r
    _rVpdPcdLine = None \r
    ## Constructor\r
    def __init__(self):\r
        ## Dictionary for VPD in following format\r
        #\r
        #  Key    : PcdClassObject instance. \r
        #           @see BuildClassObject.PcdClassObject\r
        #  Value  : offset in different SKU such as [sku1_offset, sku2_offset]\r
        self._VpdArray = {}\r
    \r
    ## Add a VPD PCD collected from platform's autogen when building.\r
    #\r
    #  @param vpds  The list of VPD PCD collected for a platform.\r
    #               @see BuildClassObject.PcdClassObject\r
    #\r
    #  @param offset integer value for VPD's offset in specific SKU.\r
    #\r
    def Add(self, Vpd, Offset):\r
        if (Vpd == None):\r
            EdkLogger.error("VpdInfoFile", BuildToolError.ATTRIBUTE_UNKNOWN_ERROR, "Invalid VPD PCD entry.")\r
        \r
        if not (Offset >= 0 or Offset == "*"):\r
            EdkLogger.error("VpdInfoFile", BuildToolError.PARAMETER_INVALID, "Invalid offset parameter: %s." % Offset)\r
        \r
        if Vpd.DatumType == "VOID*":\r
            if Vpd.MaxDatumSize <= 0:\r
                EdkLogger.error("VpdInfoFile", BuildToolError.PARAMETER_INVALID, \r
                                "Invalid max datum size for VPD PCD %s.%s" % (Vpd.TokenSpaceGuidCName, Vpd.TokenCName))\r
        elif Vpd.DatumType in ["BOOLEAN", "UINT8", "UINT16", "UINT32", "UINT64"]: \r
            if Vpd.MaxDatumSize == None or Vpd.MaxDatumSize == "":\r
                Vpd.MaxDatumSize = VpdInfoFile._MAX_SIZE_TYPE[Vpd.DatumType]\r
        else:\r
            EdkLogger.error("VpdInfoFile", BuildToolError.PARAMETER_INVALID,  \r
                            "Invalid DatumType %s for VPD PCD %s.%s" % (Vpd.DatumType, Vpd.TokenSpaceGuidCName, Vpd.TokenCName))\r
            \r
        if Vpd not in self._VpdArray.keys():\r
            #\r
            # If there is no Vpd instance in dict, that imply this offset for a given SKU is a new one \r
            #\r
            self._VpdArray[Vpd] = [Offset]\r
        else:\r
            #\r
            # If there is an offset for a specific SKU in dict, then append this offset for other sku to array.\r
            #\r
            self._VpdArray[Vpd].append(Offset)\r
            \r
        \r
    ## Generate VPD PCD information into a text file\r
    #  \r
    #  If parameter FilePath is invalid, then assert.\r
    #  If \r
    #  @param FilePath        The given file path which would hold VPD information\r
    def Write(self, FilePath):\r
        if not (FilePath != None or len(FilePath) != 0):\r
            EdkLogger.error("VpdInfoFile", BuildToolError.PARAMETER_INVALID,  \r
                            "Invalid parameter FilePath: %s." % FilePath)        \r
        try:\r
            fd = open(FilePath, "w")\r
        except:\r
            EdkLogger.error("VpdInfoFile", \r
                            BuildToolError.FILE_OPEN_FAILURE, \r
                            "Fail to open file %s for written." % FilePath)\r
        \r
        try:\r
            # write file header\r
            fd.write(FILE_COMMENT_TEMPLATE)\r
\r
            # write each of PCD in VPD type\r
            Pcds = self._VpdArray.keys()\r
            Pcds.sort()\r
            for Pcd in Pcds:\r
                i = 0\r
                for Offset in self._VpdArray[Pcd]:\r
                    PcdValue = str(Pcd.SkuInfoList[Pcd.SkuInfoList.keys()[i]].DefaultValue).strip()\r
                    if PcdValue == "" :\r
                        PcdValue  = Pcd.DefaultValue\r
                        \r
                    fd.write("%s.%s|%s|%s|%s|%s  \n" % (Pcd.TokenSpaceGuidCName, Pcd.TokenCName, str(Pcd.SkuInfoList.keys()[i]),str(Offset).strip(), str(Pcd.MaxDatumSize).strip(),PcdValue))\r
                    i += 1\r
        except:\r
            EdkLogger.error("VpdInfoFile",\r
                            BuildToolError.FILE_WRITE_FAILURE,\r
                            "Fail to write file %s" % FilePath) \r
        fd.close()\r
\r
    ## Read an existing VPD PCD info file.\r
    #\r
    #  This routine will read VPD PCD information from existing file and construct\r
    #  internal PcdClassObject array.\r
    #  This routine could be used by third-party tool to parse VPD info file content.\r
    #\r
    #  @param FilePath The full path string for existing VPD PCD info file.\r
    def Read(self, FilePath):\r
        try:\r
            fd = open(FilePath, "r")\r
        except:\r
            EdkLogger.error("VpdInfoFile", \r
                            BuildToolError.FILE_OPEN_FAILURE, \r
                            "Fail to open file %s for written." % FilePath)\r
        Lines = fd.readlines()\r
        for Line in Lines:\r
            Line = Line.strip()\r
            if len(Line) == 0 or Line.startswith("#"):\r
                continue\r
            \r
            #\r
            # the line must follow output format defined in BPDG spec.\r
            #\r
            try:\r
                PcdName, SkuId,Offset, Size, Value = Line.split("#")[0].split("|")\r
                PcdName, SkuId,Offset, Size, Value = PcdName.strip(), SkuId.strip(),Offset.strip(), Size.strip(), Value.strip()\r
                TokenSpaceName, PcdTokenName = PcdName.split(".")\r
            except:\r
                EdkLogger.error("BPDG", BuildToolError.PARSER_ERROR, "Fail to parse VPD information file %s" % FilePath)\r
            \r
            Found = False\r
            \r
            for VpdObject in self._VpdArray.keys():\r
                for sku in VpdObject.SkuInfoList.keys(): \r
                    if VpdObject.TokenSpaceGuidCName == TokenSpaceName and VpdObject.TokenCName == PcdTokenName.strip() and sku == SkuId:\r
                        if self._VpdArray[VpdObject][VpdObject.SkuInfoList.keys().index(sku)] == "*":\r
                            if Offset == "*":\r
                                EdkLogger.error("BPDG", BuildToolError.FORMAT_INVALID, "The offset of %s has not been fixed up by third-party BPDG tool." % PcdName)                              \r
                            self._VpdArray[VpdObject][VpdObject.SkuInfoList.keys().index(sku)] = Offset\r
                        Found = True\r
            if not Found:\r
                EdkLogger.error("BPDG", BuildToolError.PARSER_ERROR, "Can not find PCD defined in VPD guid file.")\r
                \r
    ## Get count of VPD PCD collected from platform's autogen when building.\r
    #\r
    #  @return The integer count value \r
    def GetCount(self):\r
        Count = 0\r
        for OffsetList in self._VpdArray.values():\r
            Count += len(OffsetList)\r
            \r
        return Count\r
    \r
    ## Get an offset value for a given VPD PCD\r
    #\r
    #  Because BPDG only support one Sku, so only return offset for SKU default.   \r
    #\r
    #  @param vpd    A given VPD PCD \r
    def GetOffset(self, vpd):\r
        if not self._VpdArray.has_key(vpd):\r
            return None\r
        \r
        if len(self._VpdArray[vpd]) == 0:\r
            return None\r
        \r
        return self._VpdArray[vpd]\r
    \r
## Call external BPDG tool to process VPD file\r
#    \r
#  @param ToolPath      The string path name for BPDG tool\r
#  @param VpdFileName   The string path name for VPD information guid.txt\r
# \r
def CallExtenalBPDGTool(ToolPath, VpdFileName):\r
    assert ToolPath != None, "Invalid parameter ToolPath"\r
    assert VpdFileName != None and os.path.exists(VpdFileName), "Invalid parameter VpdFileName"\r
    \r
    OutputDir = os.path.dirname(VpdFileName)\r
    FileName = os.path.basename(VpdFileName)\r
    BaseName, ext = os.path.splitext(FileName)\r
    OutputMapFileName = os.path.join(OutputDir, "%s.map" % BaseName)\r
    OutputBinFileName = os.path.join(OutputDir, "%s.bin" % BaseName)\r
          \r
    try:\r
        PopenObject = subprocess.Popen([ToolPath,\r
                                        '-o', OutputBinFileName, \r
                                        '-m', OutputMapFileName,\r
                                        '-q',\r
                                        '-f',\r
                                        VpdFileName],\r
                                        stdout=subprocess.PIPE, \r
                                        stderr= subprocess.PIPE)\r
    except Exception, X:\r
        EdkLogger.error("BPDG", BuildToolError.COMMAND_FAILURE, ExtraData="%s" % (str(X)))\r
    (out, error) = PopenObject.communicate()\r
    print out\r
    while PopenObject.returncode == None :\r
        PopenObject.wait()\r
    \r
    if PopenObject.returncode != 0:\r
        if PopenObject.returncode != 0:\r
            EdkLogger.debug(EdkLogger.DEBUG_1, "Fail to call BPDG tool", str(error))\r
            EdkLogger.error("BPDG", BuildToolError.COMMAND_FAILURE, "Fail to execute BPDG tool with exit code: %d, the error message is: \n %s" % \\r
                            (PopenObject.returncode, str(error)))\r
        \r
    return PopenObject.returncode\r
Commit	Line	Data
e56468c0	1	## @file\r
	2	# \r
	3	# This package manage the VPD PCD information file which will be generated\r
	4	# by build tool's autogen.\r
	5	# The VPD PCD information file will be input for third-party BPDG tool which\r
	6	# is pointed by __*_VPD_TOOL_GUID in conf/tools_def.txt \r
	7	#\r
	8	#\r
1be2ed90	9	# Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
e56468c0	10	# This program and the accompanying materials\r
	11	# are licensed and made available under the terms and conditions of the BSD License\r
	12	# which accompanies this distribution. The full text of the license may be found at\r
	13	# http://opensource.org/licenses/bsd-license.php\r
	14	#\r
	15	# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	16	# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	17	#\r
1be2ed90	18	import Common.LongFilePathOs as os\r
e56468c0	19	import re\r
	20	import Common.EdkLogger as EdkLogger\r
	21	import Common.BuildToolError as BuildToolError\r
	22	import subprocess\r
1be2ed90	23	from Common.LongFilePathSupport import OpenLongFilePath as open\r
e56468c0	24	\r
	25	FILE_COMMENT_TEMPLATE = \\r
	26	"""\r
	27	## @file\r
	28	#\r
	29	# THIS IS AUTO-GENERATED FILE BY BUILD TOOLS AND PLEASE DO NOT MAKE MODIFICATION.\r
	30	#\r
	31	# This file lists all VPD informations for a platform collected by build.exe.\r
	32	# \r
	33	# Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>\r
	34	# This program and the accompanying materials\r
	35	# are licensed and made available under the terms and conditions of the BSD License\r
	36	# which accompanies this distribution. The full text of the license may be found at\r
	37	# http://opensource.org/licenses/bsd-license.php\r
	38	#\r
	39	# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	40	# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	41	#\r
	42	\r
	43	"""\r
	44	\r
	45	## The class manage VpdInfoFile.\r
	46	#\r
	47	# This file contains an ordered (based on position in the DSC file) list of the PCDs specified in the platform description file (DSC). The Value field that will be assigned to the PCD comes from the DSC file, INF file (if not defined in the DSC file) or the DEC file (if not defined in the INF file). This file is used as an input to the BPDG tool.\r
	48	# Format for this file (using EBNF notation) is:\r
	49	# <File> :: = [<CommentBlock>]\r
	50	# [<PcdEntry>]*\r
	51	# <CommentBlock> ::= ["#" <String> <EOL>]*\r
	52	# <PcdEntry> ::= <PcdName> "\|" <Offset> "\|" <Size> "\|" <Value> <EOL>\r
	53	# <PcdName> ::= <TokenSpaceCName> "." <PcdCName>\r
	54	# <TokenSpaceCName> ::= C Variable Name of the Token Space GUID\r
	55	# <PcdCName> ::= C Variable Name of the PCD\r
	56	# <Offset> ::= {"*"} {<HexNumber>}\r
	57	# <HexNumber> ::= "0x" (a-fA-F0-9){1,8}\r
	58	# <Size> ::= <HexNumber>\r
	59	# <Value> ::= {<HexNumber>} {<NonNegativeInt>} {<QString>} {<Array>}\r
	60	# <NonNegativeInt> ::= (0-9)+\r
	61	# <QString> ::= ["L"] <DblQuote> <String> <DblQuote>\r
	62	# <DblQuote> ::= 0x22\r
	63	# <Array> ::= {<CArray>} {<NList>}\r
	64	# <CArray> ::= "{" <HexNumber> ["," <HexNumber>]* "}"\r
	65	# <NList> ::= <HexNumber> ["," <HexNumber>]*\r
	66	#\r
	67	class VpdInfoFile:\r
	68	\r
	69	## The mapping dictionary from datum type to size string.\r
	70	_MAX_SIZE_TYPE = {"BOOLEAN":"1", "UINT8":"1", "UINT16":"2", "UINT32":"4", "UINT64":"8"}\r
	71	_rVpdPcdLine = None \r
	72	## Constructor\r
	73	def __init__(self):\r
	74	## Dictionary for VPD in following format\r
	75	#\r
	76	# Key : PcdClassObject instance. \r
	77	# @see BuildClassObject.PcdClassObject\r
	78	# Value : offset in different SKU such as [sku1_offset, sku2_offset]\r
	79	self._VpdArray = {}\r
	80	\r
	81	## Add a VPD PCD collected from platform's autogen when building.\r
	82	#\r
	83	# @param vpds The list of VPD PCD collected for a platform.\r
	84	# @see BuildClassObject.PcdClassObject\r
	85	#\r
	86	# @param offset integer value for VPD's offset in specific SKU.\r
	87	#\r
88	def Add(self, Vpd, Offset):\r
89	if (Vpd == None):\r
90	EdkLogger.error("VpdInfoFile", BuildToolError.ATTRIBUTE_UNKNOWN_ERROR, "Invalid VPD PCD entry.")\r
91	\r
92	if not (Offset >= 0 or Offset == "*"):\r
93	EdkLogger.error("VpdInfoFile", BuildToolError.PARAMETER_INVALID, "Invalid offset parameter: %s." % Offset)\r
94	\r
95	if Vpd.DatumType == "VOID*":\r
96	if Vpd.MaxDatumSize <= 0:\r
97	EdkLogger.error("VpdInfoFile", BuildToolError.PARAMETER_INVALID, \r
98	"Invalid max datum size for VPD PCD %s.%s" % (Vpd.TokenSpaceGuidCName, Vpd.TokenCName))\r
99	elif Vpd.DatumType in ["BOOLEAN", "UINT8", "UINT16", "UINT32", "UINT64"]: \r
100	if Vpd.MaxDatumSize == None or Vpd.MaxDatumSize == "":\r
101	Vpd.MaxDatumSize = VpdInfoFile._MAX_SIZE_TYPE[Vpd.DatumType]\r
102	else:\r
103	EdkLogger.error("VpdInfoFile", BuildToolError.PARAMETER_INVALID, \r
104	"Invalid DatumType %s for VPD PCD %s.%s" % (Vpd.DatumType, Vpd.TokenSpaceGuidCName, Vpd.TokenCName))\r
105	\r
106	if Vpd not in self._VpdArray.keys():\r
107	#\r
108	# If there is no Vpd instance in dict, that imply this offset for a given SKU is a new one \r
109	#\r
110	self._VpdArray[Vpd] = [Offset]\r
111	else:\r
112	#\r
113	# If there is an offset for a specific SKU in dict, then append this offset for other sku to array.\r
114	#\r
115	self._VpdArray[Vpd].append(Offset)\r
116	\r
117	\r
118	## Generate VPD PCD information into a text file\r
119	# \r
120	# If parameter FilePath is invalid, then assert.\r
121	# If \r
122	# @param FilePath The given file path which would hold VPD information\r
123	def Write(self, FilePath):\r
124	if not (FilePath != None or len(FilePath) != 0):\r
125	EdkLogger.error("VpdInfoFile", BuildToolError.PARAMETER_INVALID, \r
126	"Invalid parameter FilePath: %s." % FilePath) \r
127	try:\r
128	fd = open(FilePath, "w")\r
129	except:\r
130	EdkLogger.error("VpdInfoFile", \r
131	BuildToolError.FILE_OPEN_FAILURE, \r
132	"Fail to open file %s for written." % FilePath)\r
133	\r
134	try:\r
135	# write file header\r
136	fd.write(FILE_COMMENT_TEMPLATE)\r
137	\r
138	# write each of PCD in VPD type\r
6780eef1 LG	139	Pcds = self._VpdArray.keys()\r
	140	Pcds.sort()\r
	141	for Pcd in Pcds:\r
e8a47801	142	i = 0\r
e56468c0	143	for Offset in self._VpdArray[Pcd]:\r
e8a47801	144	PcdValue = str(Pcd.SkuInfoList[Pcd.SkuInfoList.keys()[i]].DefaultValue).strip()\r
e56468c0	145	if PcdValue == "" :\r
	146	PcdValue = Pcd.DefaultValue\r
	147	\r
e8a47801 LG	148	fd.write("%s.%s\|%s\|%s\|%s\|%s \n" % (Pcd.TokenSpaceGuidCName, Pcd.TokenCName, str(Pcd.SkuInfoList.keys()[i]),str(Offset).strip(), str(Pcd.MaxDatumSize).strip(),PcdValue))\r
e8a47801 LG	149	i += 1\r
e56468c0	150	except:\r
	151	EdkLogger.error("VpdInfoFile",\r
	152	BuildToolError.FILE_WRITE_FAILURE,\r
	153	"Fail to write file %s" % FilePath) \r
	154	fd.close()\r
	155	\r
	156	## Read an existing VPD PCD info file.\r
	157	#\r
	158	# This routine will read VPD PCD information from existing file and construct\r
	159	# internal PcdClassObject array.\r
	160	# This routine could be used by third-party tool to parse VPD info file content.\r
	161	#\r
	162	# @param FilePath The full path string for existing VPD PCD info file.\r
	163	def Read(self, FilePath):\r
	164	try:\r
	165	fd = open(FilePath, "r")\r
	166	except:\r
	167	EdkLogger.error("VpdInfoFile", \r
	168	BuildToolError.FILE_OPEN_FAILURE, \r
	169	"Fail to open file %s for written." % FilePath)\r
	170	Lines = fd.readlines()\r
	171	for Line in Lines:\r
	172	Line = Line.strip()\r
	173	if len(Line) == 0 or Line.startswith("#"):\r
	174	continue\r
	175	\r
	176	#\r
	177	# the line must follow output format defined in BPDG spec.\r
	178	#\r
	179	try:\r
e8a47801 LG	180	PcdName, SkuId,Offset, Size, Value = Line.split("#")[0].split("\|")\r
e8a47801 LG	181	PcdName, SkuId,Offset, Size, Value = PcdName.strip(), SkuId.strip(),Offset.strip(), Size.strip(), Value.strip()\r
e56468c0	182	TokenSpaceName, PcdTokenName = PcdName.split(".")\r
	183	except:\r
	184	EdkLogger.error("BPDG", BuildToolError.PARSER_ERROR, "Fail to parse VPD information file %s" % FilePath)\r
	185	\r
	186	Found = False\r
e8a47801	187	\r
e56468c0	188	for VpdObject in self._VpdArray.keys():\r
e8a47801 LG	189	for sku in VpdObject.SkuInfoList.keys(): \r
	190	if VpdObject.TokenSpaceGuidCName == TokenSpaceName and VpdObject.TokenCName == PcdTokenName.strip() and sku == SkuId:\r
	191	if self._VpdArray[VpdObject][VpdObject.SkuInfoList.keys().index(sku)] == "*":\r
	192	if Offset == "*":\r
	193	EdkLogger.error("BPDG", BuildToolError.FORMAT_INVALID, "The offset of %s has not been fixed up by third-party BPDG tool." % PcdName) \r
	194	self._VpdArray[VpdObject][VpdObject.SkuInfoList.keys().index(sku)] = Offset\r
	195	Found = True\r
e56468c0	196	if not Found:\r
	197	EdkLogger.error("BPDG", BuildToolError.PARSER_ERROR, "Can not find PCD defined in VPD guid file.")\r
	198	\r
	199	## Get count of VPD PCD collected from platform's autogen when building.\r
	200	#\r
	201	# @return The integer count value \r
	202	def GetCount(self):\r
	203	Count = 0\r
	204	for OffsetList in self._VpdArray.values():\r
	205	Count += len(OffsetList)\r
	206	\r
	207	return Count\r
	208	\r
	209	## Get an offset value for a given VPD PCD\r
	210	#\r
	211	# Because BPDG only support one Sku, so only return offset for SKU default. \r
	212	#\r
	213	# @param vpd A given VPD PCD \r
	214	def GetOffset(self, vpd):\r
	215	if not self._VpdArray.has_key(vpd):\r
	216	return None\r
	217	\r
	218	if len(self._VpdArray[vpd]) == 0:\r
	219	return None\r
	220	\r
	221	return self._VpdArray[vpd]\r
	222	\r
	223	## Call external BPDG tool to process VPD file\r
	224	# \r
	225	# @param ToolPath The string path name for BPDG tool\r
	226	# @param VpdFileName The string path name for VPD information guid.txt\r
	227	# \r
08dd311f	228	def CallExtenalBPDGTool(ToolPath, VpdFileName):\r
e56468c0	229	assert ToolPath != None, "Invalid parameter ToolPath"\r
08dd311f	230	assert VpdFileName != None and os.path.exists(VpdFileName), "Invalid parameter VpdFileName"\r
e56468c0	231	\r
08dd311f LG	232	OutputDir = os.path.dirname(VpdFileName)\r
	233	FileName = os.path.basename(VpdFileName)\r
	234	BaseName, ext = os.path.splitext(FileName)\r
	235	OutputMapFileName = os.path.join(OutputDir, "%s.map" % BaseName)\r
	236	OutputBinFileName = os.path.join(OutputDir, "%s.bin" % BaseName)\r
e56468c0	237	\r
	238	try:\r
	239	PopenObject = subprocess.Popen([ToolPath,\r
	240	'-o', OutputBinFileName, \r
	241	'-m', OutputMapFileName,\r
08dd311f	242	'-q',\r
e56468c0	243	'-f',\r
08dd311f	244	VpdFileName],\r
e56468c0	245	stdout=subprocess.PIPE, \r
	246	stderr= subprocess.PIPE)\r
	247	except Exception, X:\r
	248	EdkLogger.error("BPDG", BuildToolError.COMMAND_FAILURE, ExtraData="%s" % (str(X)))\r
	249	(out, error) = PopenObject.communicate()\r
	250	print out\r
	251	while PopenObject.returncode == None :\r
	252	PopenObject.wait()\r
	253	\r
	254	if PopenObject.returncode != 0:\r
	255	if PopenObject.returncode != 0:\r
	256	EdkLogger.debug(EdkLogger.DEBUG_1, "Fail to call BPDG tool", str(error))\r
	257	EdkLogger.error("BPDG", BuildToolError.COMMAND_FAILURE, "Fail to execute BPDG tool with exit code: %d, the error message is: \n %s" % \\r
	258	(PopenObject.returncode, str(error)))\r
	259	\r
	260	return PopenObject.returncode\r