2N/A

2N/A

2N/Aimport os.path

2N/A

2N/Afrom Library.Misc import Sdict

2N/Afrom Library.Misc import GetNonMetaDataFiles

2N/Afrom PomAdapter.InfPomAlignment import InfPomAlignment

2N/Afrom PomAdapter.DecPomAlignment import DecPomAlignment

2N/Aimport Logger.Log as Logger

2N/Afrom Logger import StringTable as ST

2N/Afrom Logger.ToolError import OPTION_VALUE_INVALID

2N/Afrom Logger.ToolError import FatalError

2N/Afrom Logger.ToolError import EDK1_INF_ERROR

2N/Afrom Object.POM.CommonObject import IdentificationObject

2N/Afrom Object.POM.CommonObject import CommonHeaderObject

2N/Afrom Object.POM.CommonObject import MiscFileObject

2N/A

class DistributionPackageHeaderObject(IdentificationObject, \

CommonHeaderObject):

def __init__(self):

IdentificationObject.__init__(self)

CommonHeaderObject.__init__(self)

self.ReadOnly = ''

self.RePackage = ''

self.Vendor = ''

self.Date = ''

self.Signature = 'Md5Sum'

self.XmlSpecification = ''

def GetReadOnly(self):

return self.ReadOnly

def SetReadOnly(self, ReadOnly):

self.ReadOnly = ReadOnly

def GetRePackage(self):

return self.RePackage

def SetRePackage(self, RePackage):

self.RePackage = RePackage

def GetVendor(self):

return self.Vendor

def SetDate(self, Date):

self.Date = Date

def GetDate(self):

return self.Date

def SetSignature(self, Signature):

self.Signature = Signature

def GetSignature(self):

return self.Signature

def SetXmlSpecification(self, XmlSpecification):

self.XmlSpecification = XmlSpecification

def GetXmlSpecification(self):

return self.XmlSpecification

class DistributionPackageClass(object):

def __init__(self):

self.Header = DistributionPackageHeaderObject()

#

# {(Guid, Version, Path) : PackageObj}

#

self.PackageSurfaceArea = Sdict()

#

# {(Guid, Version, Name, Path) : ModuleObj}

#

self.ModuleSurfaceArea = Sdict()

self.Tools = MiscFileObject()

self.MiscellaneousFiles = MiscFileObject()

self.UserExtensions = []

self.FileList = []

## Get all included packages and modules for a distribution package

#

# @param WorkspaceDir: WorkspaceDir

# @param PackageList: A list of all packages

# @param ModuleList: A list of all modules

#

def GetDistributionPackage(self, WorkspaceDir, PackageList, ModuleList):

#

# Get Packages

#

if PackageList:

for PackageFile in PackageList:

PackageFileFullPath = \

os.path.normpath(os.path.join(WorkspaceDir, PackageFile))

DecObj = DecPomAlignment(PackageFileFullPath, WorkspaceDir, CheckMulDec = True)

PackageObj = DecObj

#

# Parser inf file one bye one

#

ModuleInfFileList = PackageObj.GetModuleFileList()

for File in ModuleInfFileList:

WsRelPath = os.path.join(PackageObj.GetPackagePath(), File)

WsRelPath = os.path.normpath(WsRelPath)

if ModuleList and WsRelPath in ModuleList:

Logger.Error("UPT",

OPTION_VALUE_INVALID,

ST.ERR_NOT_STANDALONE_MODULE_ERROR%\

(WsRelPath, PackageFile))

Filename = os.path.normpath\

(os.path.join(PackageObj.GetRelaPath(), File))

os.path.splitext(Filename)

#

# Call INF parser to generate Inf Object.

# Actually, this call is not directly call, but wrapped by

# Inf class in InfPomAlignment.

#

try:

ModuleObj = InfPomAlignment(Filename, WorkspaceDir, \

PackageObj.GetPackagePath())

#

# Add module to package

#

ModuleDict = PackageObj.GetModuleDict()

ModuleDict[(ModuleObj.GetGuid(), \

ModuleObj.GetVersion(), \

ModuleObj.GetName(), \

ModuleObj.GetCombinePath())] = ModuleObj

PackageObj.SetModuleDict(ModuleDict)

except FatalError, ErrCode:

if ErrCode.message == EDK1_INF_ERROR:

Logger.Warn("UPT",

ST.WRN_EDK1_INF_FOUND%Filename)

else:

raise

self.PackageSurfaceArea\

[(PackageObj.GetGuid(), PackageObj.GetVersion(), \

PackageObj.GetCombinePath())] = PackageObj

#

# Get Modules

#

if ModuleList:

for ModuleFile in ModuleList:

ModuleFileFullPath = \

os.path.normpath(os.path.join(WorkspaceDir, ModuleFile))

try:

ModuleObj = InfPomAlignment(ModuleFileFullPath,

WorkspaceDir)

ModuleKey = (ModuleObj.GetGuid(),

ModuleObj.GetVersion(),

ModuleObj.GetName(),

ModuleObj.GetCombinePath())

self.ModuleSurfaceArea[ModuleKey] = ModuleObj

except FatalError, ErrCode:

if ErrCode.message == EDK1_INF_ERROR:

Logger.Error("UPT",

EDK1_INF_ERROR,

ST.WRN_EDK1_INF_FOUND%ModuleFileFullPath,

ExtraData=ST.ERR_NOT_SUPPORTED_SA_MODULE)

else:

raise

## Get all files included for a distribution package, except tool/misc of

# distribution level

#

# @retval DistFileList A list of filepath for NonMetaDataFile, relative to workspace

# @retval MetaDataFileList A list of filepath for MetaDataFile, relative to workspace

#

def GetDistributionFileList(self):

MetaDataFileList = []

for Guid, Version, Path in self.PackageSurfaceArea:

Package = self.PackageSurfaceArea[Guid, Version, Path]

PackagePath = Package.GetPackagePath()

FullPath = Package.GetFullPath()

MetaDataFileList.append(Path)

IncludePathList = Package.GetIncludePathList()

for IncludePath in IncludePathList:

SearchPath = os.path.normpath(os.path.join(os.path.dirname(FullPath), IncludePath))

AddPath = os.path.normpath(os.path.join(PackagePath, IncludePath))

self.FileList += GetNonMetaDataFiles(SearchPath, ['CVS', '.svn'], False, AddPath)

Module = None

ModuleDict = Package.GetModuleDict()

for Guid, Version, Name, Path in ModuleDict:

Module = ModuleDict[Guid, Version, Name, Path]

ModulePath = Module.GetModulePath()

FullPath = Module.GetFullPath()

PkgRelPath = os.path.normpath(os.path.join(PackagePath, ModulePath))

MetaDataFileList.append(Path)

self.FileList += GetNonMetaDataFiles(os.path.dirname(FullPath), ['CVS', '.svn'], False, PkgRelPath)

for Guid, Version, Name, Path in self.ModuleSurfaceArea:

Module = self.ModuleSurfaceArea[Guid, Version, Name, Path]

ModulePath = Module.GetModulePath()

FullPath = Module.GetFullPath()

MetaDataFileList.append(Path)

self.FileList += GetNonMetaDataFiles(os.path.dirname(FullPath), ['CVS', '.svn'], False, ModulePath)

return self.FileList, MetaDataFileList