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

[mirror_edk2.git] / BaseTools / Source / Python / CommonDataClass / DistributionPackageClass.py

## @file\r
# This file is used to define a class object to describe a distribution package\r
#\r
# Copyright (c) 2008, Intel Corporation\r
# All rights reserved. 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
# Import Modules\r
#\r
import os.path\r
from CommonClass import *\r
from Common.Misc import sdict\r
from Common.Misc import GetFiles\r
from Common.DecClassObjectLight import Dec\r
from Common.InfClassObjectLight import Inf\r
from Common.XmlParser import *\r
\r
## DistributionPackageHeaderClass\r
#\r
class DistributionPackageHeaderClass(IdentificationClass, CommonHeaderClass):\r
    def __init__(self):\r
        IdentificationClass.__init__(self)\r
        CommonHeaderClass.__init__(self)\r
        self.ReadOnly = 'False'\r
        self.RePackage = 'True'\r
        self.Vendor = ''\r
        self.Date = ''\r
        self.Signature = 'Md5Sum'\r
        self.XmlSpecification = ''\r
\r
## DistributionPackageClass\r
#\r
#\r
class DistributionPackageClass(object):\r
    def __init__(self):\r
        self.Header = DistributionPackageHeaderClass()\r
        self.PackageSurfaceArea = sdict() # {(Guid, Version, Path) : PackageObj}\r
        self.ModuleSurfaceArea = sdict()  # {(Guid, Version, Path) : ModuleObj}\r
        self.Tools = MiscFileClass()\r
        self.MiscellaneousFiles = MiscFileClass()\r
        self.UserExtensions = []\r
    \r
    ## Get all included packages and modules for a distribution package\r
    # \r
    # @param WorkspaceDir:  WorkspaceDir\r
    # @param PackageList:   A list of all packages\r
    # @param ModuleList:    A list of all modules\r
    #\r
    def GetDistributionPackage(self, WorkspaceDir, PackageList, ModuleList):\r
        AllGuidVersionDict = {}\r
        # Get Packages\r
        if PackageList:\r
            for PackageFile in PackageList:\r
                PackageFileFullPath = os.path.normpath(os.path.join(WorkspaceDir, PackageFile))\r
                DecObj = Dec(PackageFileFullPath, True, WorkspaceDir)\r
                PackageObj = DecObj.Package\r
                AllGuidVersionDict[PackageFileFullPath] = [PackageObj.PackageHeader.Guid, PackageObj.PackageHeader.Version]\r
                \r
                # Parser inf file one bye one\r
                for File in PackageObj.MiscFiles.Files:\r
                    Filename = os.path.normpath(os.path.join(PackageObj.PackageHeader.RelaPath, File.Filename))\r
                    (Name, ExtName) = os.path.splitext(Filename)\r
                    if ExtName.upper() == '.INF':\r
                        InfObj = Inf(Filename, True, WorkspaceDir, DecObj.Identification.PackagePath)\r
                        ModuleObj = InfObj.Module\r
                        AllGuidVersionDict[File] = [ModuleObj.ModuleHeader.Guid, ModuleObj.ModuleHeader.Version]\r
                        # Find and update Guid/Version of LibraryClass\r
                        for Item in ModuleObj.LibraryClasses:\r
                            if Item.RecommendedInstance:\r
                                LibClassIns = os.path.normpath(os.path.join(WorkspaceDir, Item.RecommendedInstance))\r
                                Guid, Version = '', ''\r
                                if LibClassIns in AllGuidVersionDict:\r
                                    Guid = AllGuidVersionDict[LibClassIns][0]\r
                                    Version = AllGuidVersionDict[LibClassIns][1]\r
                                else:\r
                                    Lib = Inf(LibClassIns, True, WorkspaceDir)\r
                                    Guid = Lib.Module.ModuleHeader.Guid\r
                                    Version = Lib.Module.ModuleHeader.Version\r
                                    AllGuidVersionDict[LibClassIns] = [Guid, Version]\r
                                Item.RecommendedInstanceGuid = Guid\r
                                Item.RecommendedInstanceVersion = Version\r
                        # Find and update Guid/Version of \r
                        for Item in ModuleObj.PackageDependencies:\r
                            if Item.FilePath:\r
                                PackageFilePath = os.path.normpath(os.path.join(WorkspaceDir, Item.FilePath))\r
                                Guid, Version = '', ''\r
                                if PackageFilePath in AllGuidVersionDict:\r
                                    Guid = AllGuidVersionDict[PackageFilePath][0]\r
                                    Version = AllGuidVersionDict[PackageFilePath][1]\r
                                else:\r
                                    PackageDependencies = Dec(PackageFilePath, True, WorkspaceDir)\r
                                    Guid = PackageDependencies.Package.PackageHeader.Guid\r
                                    Version = PackageDependencies.Package.PackageHeader.Version\r
                                    AllGuidVersionDict[PackageFilePath] = [Guid, Version]\r
                                Item.PackageGuid = Guid\r
                                Item.PackageVersion = Version\r
\r
                        # Add module to package\r
                        PackageObj.Modules[(ModuleObj.ModuleHeader.Guid, ModuleObj.ModuleHeader.Version, ModuleObj.ModuleHeader.CombinePath)] = ModuleObj\r
                self.PackageSurfaceArea[(PackageObj.PackageHeader.Guid, PackageObj.PackageHeader.Version, PackageObj.PackageHeader.CombinePath)] = PackageObj\r
\r
        # Get Modules\r
        if ModuleList:\r
            for ModuleFile in ModuleList:\r
                ModuleFileFullPath = os.path.normpath(os.path.join(WorkspaceDir, ModuleFile))\r
                InfObj = Inf(ModuleFileFullPath, True, WorkspaceDir)\r
                ModuleObj = InfObj.Module\r
                AllGuidVersionDict[ModuleFileFullPath] = [ModuleObj.ModuleHeader.Guid, ModuleObj.ModuleHeader.Version]\r
                # Find and update Guid/Version of LibraryClass\r
                for Item in ModuleObj.LibraryClasses:\r
                    if Item.RecommendedInstance:\r
                        LibClassIns = os.path.normpath(os.path.join(WorkspaceDir, Item.RecommendedInstance))\r
                        Guid, Version = '', ''\r
                        if LibClassIns in AllGuidVersionDict:\r
                            Guid = AllGuidVersionDict[LibClassIns][0]\r
                            Version = AllGuidVersionDict[LibClassIns][1]\r
                        else:\r
                            Lib = Inf(LibClassIns, True, WorkspaceDir)\r
                            Guid = Lib.Module.ModuleHeader.Guid\r
                            Version = Lib.Module.ModuleHeader.Version\r
                            AllGuidVersionDict[LibClassIns] = [Guid, Version]\r
                        Item.RecommendedInstanceGuid = Guid\r
                        Item.RecommendedInstanceVersion = Version\r
                # Find and update Guid/Version of \r
                for Item in ModuleObj.PackageDependencies:\r
                    if Item.FilePath:\r
                        PackageFilePath = os.path.normpath(os.path.join(WorkspaceDir, Item.FilePath))\r
                        Guid, Version = '', ''\r
                        if PackageFilePath in AllGuidVersionDict:\r
                            Guid = AllGuidVersionDict[PackageFilePath][0]\r
                            Version = AllGuidVersionDict[PackageFilePath][1]\r
                        else:\r
                            PackageDependencies = Dec(PackageFilePath, True, WorkspaceDir)\r
                            Guid = PackageDependencies.Package.PackageHeader.Guid\r
                            Version = PackageDependencies.Package.PackageHeader.Version\r
                            AllGuidVersionDict[PackageFilePath] = [Guid, Version]\r
                        Item.PackageGuid = Guid\r
                        Item.PackageVersion = Version\r
                self.ModuleSurfaceArea[(ModuleObj.ModuleHeader.Guid, ModuleObj.ModuleHeader.Version, ModuleObj.ModuleHeader.CombinePath)] = ModuleObj\r
\r
##\r
#\r
# This acts like the main() function for the script, unless it is 'import'ed into another\r
# script.\r
#\r
if __name__ == '__main__':\r
    pass\r
    D = DistributionPackageClass()\r
    D.GetDistributionPackage(os.getenv('WORKSPACE'), ['MdePkg/MdePkg.dec', 'TianoModulePkg/TianoModulePkg.dec'], ['MdeModulePkg/Application/HelloWorld/HelloWorld.inf'])\r
    Xml = DistributionPackageXml()\r
    print Xml.ToXml(D)\r
    E = Xml.FromXml('C:\\2.xml')\r
    #print Xml.ToXml(E)\r