BaseTools: Use absolute import in GenFds

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

## @file\r
# process APRIORI file data and generate PEI/DXE APRIORI file\r
#\r
#  Copyright (c) 2007 - 2018, Intel Corporation. All rights reserved.<BR>\r
#\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
# Import Modules\r
#\r
from __future__ import absolute_import\r
from struct import *\r
import Common.LongFilePathOs as os\r
from io import BytesIO\r
from . import FfsFileStatement\r
from .GenFdsGlobalVariable import GenFdsGlobalVariable\r
from CommonDataClass.FdfClass import AprioriSectionClassObject\r
from Common.StringUtils import *\r
from Common.Misc import SaveFileOnChange, PathClass\r
from Common import EdkLogger\r
from Common.BuildToolError import *\r
from Common.DataType import TAB_COMMON\r
\r
## process APRIORI file data and generate PEI/DXE APRIORI file\r
#\r
#\r
class AprioriSection (AprioriSectionClassObject):\r
    ## The constructor\r
    #\r
    #   @param  self        The object pointer\r
    #\r
    def __init__(self):\r
        AprioriSectionClassObject.__init__(self)\r
        self.AprioriType = ""\r
\r
    ## GenFfs() method\r
    #\r
    #   Generate FFS for APRIORI file\r
    #\r
    #   @param  self        The object pointer\r
    #   @param  FvName      for whom apriori file generated\r
    #   @param  Dict        dictionary contains macro and its value\r
    #   @retval string      Generated file name\r
    #\r
    def GenFfs (self, FvName, Dict = {}, IsMakefile = False):\r
        DXE_GUID = "FC510EE7-FFDC-11D4-BD41-0080C73C8881"\r
        PEI_GUID = "1B45CC0A-156A-428A-AF62-49864DA0E6E6"\r
        Buffer = BytesIO('')\r
        AprioriFileGuid = DXE_GUID\r
        if self.AprioriType == "PEI":\r
            AprioriFileGuid = PEI_GUID\r
        OutputAprFilePath = os.path.join (GenFdsGlobalVariable.WorkSpaceDir, \\r
                                   GenFdsGlobalVariable.FfsDir,\\r
                                   AprioriFileGuid + FvName)\r
        if not os.path.exists(OutputAprFilePath) :\r
            os.makedirs(OutputAprFilePath)\r
\r
        OutputAprFileName = os.path.join( OutputAprFilePath, \\r
                                       AprioriFileGuid + FvName + '.Apri' )\r
        AprFfsFileName = os.path.join (OutputAprFilePath,\\r
                                    AprioriFileGuid + FvName + '.Ffs')\r
\r
        Dict.update(self.DefineVarDict)\r
        InfFileName = None\r
        for FfsObj in self.FfsList :\r
            Guid = ""\r
            if isinstance(FfsObj, FfsFileStatement.FileStatement):\r
                Guid = FfsObj.NameGuid\r
            else:\r
                InfFileName = NormPath(FfsObj.InfFileName)\r
                Arch = FfsObj.GetCurrentArch()\r
\r
                if Arch is not None:\r
                    Dict['$(ARCH)'] = Arch\r
                InfFileName = GenFdsGlobalVariable.MacroExtend(InfFileName, Dict, Arch)\r
\r
                if Arch is not None:\r
                    Inf = GenFdsGlobalVariable.WorkSpace.BuildObject[PathClass(InfFileName, GenFdsGlobalVariable.WorkSpaceDir), Arch, GenFdsGlobalVariable.TargetName, GenFdsGlobalVariable.ToolChainTag]\r
                    Guid = Inf.Guid\r
\r
                else:\r
                    Inf = GenFdsGlobalVariable.WorkSpace.BuildObject[PathClass(InfFileName, GenFdsGlobalVariable.WorkSpaceDir), TAB_COMMON, GenFdsGlobalVariable.TargetName, GenFdsGlobalVariable.ToolChainTag]\r
                    Guid = Inf.Guid\r
\r
                    self.BinFileList = Inf.Module.Binaries\r
                    if self.BinFileList == []:\r
                        EdkLogger.error("GenFds", RESOURCE_NOT_AVAILABLE,\r
                                        "INF %s not found in build ARCH %s!" \\r
                                        % (InfFileName, GenFdsGlobalVariable.ArchList))\r
\r
\r
            GuidPart = Guid.split('-')\r
            Buffer.write(pack('I', long(GuidPart[0], 16)))\r
            Buffer.write(pack('H', int(GuidPart[1], 16)))\r
            Buffer.write(pack('H', int(GuidPart[2], 16)))\r
\r
            for Num in range(2):\r
                Char = GuidPart[3][Num*2:Num*2+2]\r
                Buffer.write(pack('B', int(Char, 16)))\r
\r
            for Num in range(6):\r
                Char = GuidPart[4][Num*2:Num*2+2]\r
                Buffer.write(pack('B', int(Char, 16)))\r
\r
        SaveFileOnChange(OutputAprFileName, Buffer.getvalue())\r
\r
        RawSectionFileName = os.path.join( OutputAprFilePath, \\r
                                       AprioriFileGuid + FvName + '.raw' )\r
        MakefilePath = None\r
        if IsMakefile:\r
            if not InfFileName:\r
                return None\r
            MakefilePath = InfFileName, Arch\r
        GenFdsGlobalVariable.GenerateSection(RawSectionFileName, [OutputAprFileName], 'EFI_SECTION_RAW', IsMakefile=IsMakefile)\r
        GenFdsGlobalVariable.GenerateFfs(AprFfsFileName, [RawSectionFileName],\r
                                        'EFI_FV_FILETYPE_FREEFORM', AprioriFileGuid, MakefilePath=MakefilePath)\r
\r
        return AprFfsFileName\r
\r