Sync EDKII BaseTools to BaseTools project r2065.

[mirror_edk2.git] / BaseTools / Source / Python / TargetTool / TargetTool.py

#\r
#  Copyright (c) 2007 - 2010, 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
import os\r
import sys\r
import traceback\r
from optparse import OptionParser\r
\r
import Common.EdkLogger as EdkLogger\r
import Common.BuildToolError as BuildToolError\r
from Common.DataType import *\r
\r
# To Do 1.set clean, 2. add item, if the line is disabled.\r
\r
class TargetTool():\r
    def __init__(self, opt, args):\r
        self.WorkSpace = os.path.normpath(os.getenv('WORKSPACE'))\r
        self.Opt       = opt\r
        self.Arg       = args[0]\r
        self.FileName  = os.path.normpath(os.path.join(self.WorkSpace, 'Conf', 'target.txt'))\r
        if os.path.isfile(self.FileName) == False:\r
            print "%s does not exist." % self.FileName\r
            sys.exit(1)\r
        self.TargetTxtDictionary = {\r
            TAB_TAT_DEFINES_ACTIVE_PLATFORM                            : None,\r
            TAB_TAT_DEFINES_TOOL_CHAIN_CONF                            : None,\r
            TAB_TAT_DEFINES_MAX_CONCURRENT_THREAD_NUMBER               : None,\r
            TAB_TAT_DEFINES_TARGET                                     : None,\r
            TAB_TAT_DEFINES_TOOL_CHAIN_TAG                             : None,\r
            TAB_TAT_DEFINES_TARGET_ARCH                                : None,\r
            TAB_TAT_DEFINES_BUILD_RULE_CONF                            : None,\r
        }\r
        self.LoadTargetTxtFile(self.FileName)\r
\r
    def LoadTargetTxtFile(self, filename):\r
        if os.path.exists(filename) and os.path.isfile(filename):\r
            return self.ConvertTextFileToDict(filename, '#', '=')\r
        else:\r
            raise ParseError('LoadTargetTxtFile() : No Target.txt file exists.')\r
            return 1\r
\r
#\r
# Convert a text file to a dictionary\r
#\r
    def ConvertTextFileToDict(self, FileName, CommentCharacter, KeySplitCharacter):\r
        """Convert a text file to a dictionary of (name:value) pairs."""\r
        try:\r
            f = open(FileName,'r')\r
            for Line in f:\r
                if Line.startswith(CommentCharacter) or Line.strip() == '':\r
                    continue\r
                LineList = Line.split(KeySplitCharacter,1)\r
                if len(LineList) >= 2:\r
                    Key = LineList[0].strip()\r
                    if Key.startswith(CommentCharacter) == False and Key in self.TargetTxtDictionary.keys():\r
                        if Key == TAB_TAT_DEFINES_ACTIVE_PLATFORM or Key == TAB_TAT_DEFINES_TOOL_CHAIN_CONF \\r
                          or Key == TAB_TAT_DEFINES_MAX_CONCURRENT_THREAD_NUMBER \\r
                          or Key == TAB_TAT_DEFINES_ACTIVE_MODULE:\r
                            self.TargetTxtDictionary[Key] = LineList[1].replace('\\', '/').strip()\r
                        elif Key == TAB_TAT_DEFINES_TARGET or Key == TAB_TAT_DEFINES_TARGET_ARCH \\r
                          or Key == TAB_TAT_DEFINES_TOOL_CHAIN_TAG or Key == TAB_TAT_DEFINES_BUILD_RULE_CONF:\r
                            self.TargetTxtDictionary[Key] = LineList[1].split()\r
            f.close()\r
            return 0\r
        except:\r
            last_type, last_value, last_tb = sys.exc_info()\r
            traceback.print_exception(last_type, last_value, last_tb)\r
\r
    def Print(self):\r
        KeyList = self.TargetTxtDictionary.keys()\r
        errMsg  = ''\r
        for Key in KeyList:\r
            if type(self.TargetTxtDictionary[Key]) == type([]):\r
                print "%-30s = %s" % (Key, ''.join(elem + ' ' for elem in self.TargetTxtDictionary[Key]))\r
            elif self.TargetTxtDictionary[Key] == None:\r
                errMsg += "  Missing %s configuration information, please use TargetTool to set value!" % Key + os.linesep \r
            else:\r
                print "%-30s = %s" % (Key, self.TargetTxtDictionary[Key])\r
        \r
        if errMsg != '':\r
            print os.linesep + 'Warning:' + os.linesep + errMsg\r
            \r
    def RWFile(self, CommentCharacter, KeySplitCharacter, Num):\r
        try:\r
            fr = open(self.FileName, 'r')\r
            fw = open(os.path.normpath(os.path.join(self.WorkSpace, 'Conf\\targetnew.txt')), 'w')\r
\r
            existKeys = []\r
            for Line in fr:\r
                if Line.startswith(CommentCharacter) or Line.strip() == '':\r
                    fw.write(Line)\r
                else:\r
                    LineList = Line.split(KeySplitCharacter,1)\r
                    if len(LineList) >= 2:\r
                        Key = LineList[0].strip()\r
                        if Key.startswith(CommentCharacter) == False and Key in self.TargetTxtDictionary.keys():\r
                            if Key not in existKeys:\r
                                existKeys.append(Key)\r
                            else:\r
                                print "Warning: Found duplicate key item in original configuration files!"\r
                                \r
                            if Num == 0:\r
                                Line = "%-30s = \n" % Key\r
                            else:\r
                                ret = GetConfigureKeyValue(self, Key)\r
                                if ret != None:\r
                                    Line = ret\r
                            fw.write(Line)\r
            for key in self.TargetTxtDictionary.keys():\r
                if key not in existKeys:\r
                    print "Warning: %s does not exist in original configuration file" % key\r
                    Line = GetConfigureKeyValue(self, key)\r
                    if Line == None:\r
                        Line = "%-30s = " % key\r
                    fw.write(Line)\r
                \r
            fr.close()\r
            fw.close()\r
            os.remove(self.FileName)\r
            os.rename(os.path.normpath(os.path.join(self.WorkSpace, 'Conf\\targetnew.txt')), self.FileName)\r
            \r
        except:\r
            last_type, last_value, last_tb = sys.exc_info()\r
            traceback.print_exception(last_type, last_value, last_tb)\r
\r
def GetConfigureKeyValue(self, Key):\r
    Line = None\r
    if Key == TAB_TAT_DEFINES_ACTIVE_PLATFORM and self.Opt.DSCFILE != None:\r
        dscFullPath = os.path.join(self.WorkSpace, self.Opt.DSCFILE)\r
        if os.path.exists(dscFullPath):\r
            Line = "%-30s = %s\n" % (Key, self.Opt.DSCFILE)\r
        else:\r
            EdkLogger.error("TagetTool", BuildToolError.FILE_NOT_FOUND, \r
                            "DSC file %s does not exist!" % self.Opt.DSCFILE, RaiseError=False)\r
    elif Key == TAB_TAT_DEFINES_TOOL_CHAIN_CONF and self.Opt.TOOL_DEFINITION_FILE != None:\r
        tooldefFullPath = os.path.join(self.WorkSpace, self.Opt.TOOL_DEFINITION_FILE)\r
        if os.path.exists(tooldefFullPath):\r
            Line = "%-30s = %s\n" % (Key, self.Opt.TOOL_DEFINITION_FILE)\r
        else:\r
            EdkLogger.error("TagetTool", BuildToolError.FILE_NOT_FOUND, \r
                            "Tooldef file %s does not exist!" % self.Opt.TOOL_DEFINITION_FILE, RaiseError=False)\r
\r
    elif self.Opt.NUM >= 2:\r
        Line = "%-30s = %s\n" % (Key, 'Enable')\r
    elif self.Opt.NUM <= 1:\r
        Line = "%-30s = %s\n" % (Key, 'Disable')        \r
    elif Key == TAB_TAT_DEFINES_MAX_CONCURRENT_THREAD_NUMBER and self.Opt.NUM != None:\r
        Line = "%-30s = %s\n" % (Key, str(self.Opt.NUM))\r
    elif Key == TAB_TAT_DEFINES_TARGET and self.Opt.TARGET != None:\r
        Line = "%-30s = %s\n" % (Key, ''.join(elem + ' ' for elem in self.Opt.TARGET))\r
    elif Key == TAB_TAT_DEFINES_TARGET_ARCH and self.Opt.TARGET_ARCH != None:\r
        Line = "%-30s = %s\n" % (Key, ''.join(elem + ' ' for elem in self.Opt.TARGET_ARCH))\r
    elif Key == TAB_TAT_DEFINES_TOOL_CHAIN_TAG and self.Opt.TOOL_CHAIN_TAG != None:\r
        Line = "%-30s = %s\n" % (Key, self.Opt.TOOL_CHAIN_TAG)\r
    elif Key == TAB_TAT_DEFINES_BUILD_RULE_CONF and self.Opt.BUILD_RULE_FILE != None:\r
        buildruleFullPath = os.path.join(self.WorkSpace, self.Opt.BUILD_RULE_FILE)\r
        if os.path.exists(buildruleFullPath):\r
            Line = "%-30s = %s\n" % (Key, self.Opt.BUILD_RULE_FILE)\r
        else:\r
            EdkLogger.error("TagetTool", BuildToolError.FILE_NOT_FOUND, \r
                            "Build rule file %s does not exist!" % self.Opt.BUILD_RULE_FILE, RaiseError=False)\r
    return Line\r
\r
VersionNumber = "0.01"\r
__version__ = "%prog Version " + VersionNumber\r
__copyright__ = "Copyright (c) 2007 - 2010, Intel Corporation  All rights reserved."\r
__usage__ = "%prog [options] {args} \\r
\nArgs:                                                  \\r
\n Clean  clean the all default configuration of target.txt. \\r
\n Print  print the all default configuration of target.txt. \\r
\n Set    replace the default configuration with expected value specified by option."\r
\r
gParamCheck = []\r
def SingleCheckCallback(option, opt_str, value, parser):\r
    if option not in gParamCheck:\r
        setattr(parser.values, option.dest, value)\r
        gParamCheck.append(option)\r
    else:\r
        parser.error("Option %s only allows one instance in command line!" % option)\r
\r
def RangeCheckCallback(option, opt_str, value, parser):\r
    if option not in gParamCheck:\r
        gParamCheck.append(option)\r
        if value < 1 or value > 8:\r
            parser.error("The count of multi-thread is not in valid range of 1 ~ 8.")\r
        else:\r
            setattr(parser.values, option.dest, value)\r
    else:\r
        parser.error("Option %s only allows one instance in command line!" % option)\r
        \r
def MyOptionParser():\r
    parser = OptionParser(version=__version__,prog="TargetTool.exe",usage=__usage__,description=__copyright__)\r
    parser.add_option("-a", "--arch", action="append", type="choice", choices=['IA32','X64','IPF','EBC', 'ARM','0'], dest="TARGET_ARCH",\r
        help="ARCHS is one of list: IA32, X64, IPF, ARM or EBC, which replaces target.txt's TARGET_ARCH definition. To specify more archs, please repeat this option. 0 will clear this setting in target.txt and can't combine with other value.")\r
    parser.add_option("-p", "--platform", action="callback", type="string", dest="DSCFILE", callback=SingleCheckCallback,\r
        help="Specify a DSC file, which replace target.txt's ACTIVE_PLATFORM definition. 0 will clear this setting in target.txt and can't combine with other value.")\r
    parser.add_option("-c", "--tooldef", action="callback", type="string", dest="TOOL_DEFINITION_FILE", callback=SingleCheckCallback,\r
        help="Specify the WORKSPACE relative path of tool_def.txt file, which replace target.txt's TOOL_CHAIN_CONF definition. 0 will clear this setting in target.txt and can't combine with other value.")\r
    parser.add_option("-t", "--target", action="append", type="choice", choices=['DEBUG','RELEASE','0'], dest="TARGET",\r
        help="TARGET is one of list: DEBUG, RELEASE, which replaces target.txt's TARGET definition. To specify more TARGET, please repeat this option. 0 will clear this setting in target.txt and can't combine with other value.")\r
    parser.add_option("-n", "--tagname", action="callback", type="string", dest="TOOL_CHAIN_TAG", callback=SingleCheckCallback,\r
        help="Specify the Tool Chain Tagname, which replaces target.txt's TOOL_CHAIN_TAG definition. 0 will clear this setting in target.txt and can't combine with other value.")\r
    parser.add_option("-r", "--buildrule", action="callback", type="string", dest="BUILD_RULE_FILE", callback=SingleCheckCallback,\r
        help="Specify the build rule configure file, which replaces target.txt's BUILD_RULE_CONF definition. If not specified, the default value Conf/build_rule.txt will be set.")\r
    parser.add_option("-m", "--multithreadnum", action="callback", type="int", dest="NUM", callback=RangeCheckCallback,\r
        help="Specify the multi-thread number which replace target.txt's MAX_CONCURRENT_THREAD_NUMBER. If the value is less than 2, MULTIPLE_THREAD will be disabled. If the value is larger than 1, MULTIPLE_THREAD will be enabled.")\r
    (opt, args)=parser.parse_args()\r
    return (opt, args)\r
\r
if __name__ == '__main__':\r
    EdkLogger.Initialize()\r
    EdkLogger.SetLevel(EdkLogger.QUIET)\r
    if os.getenv('WORKSPACE') == None:\r
        print "ERROR: WORKSPACE should be specified or edksetup script should be executed before run TargetTool"\r
        sys.exit(1)\r
        \r
    (opt, args) = MyOptionParser()\r
    if len(args) != 1 or (args[0].lower() != 'print' and args[0].lower() != 'clean' and args[0].lower() != 'set'):\r
        print "The number of args isn't 1 or the value of args is invalid."\r
        sys.exit(1)\r
    if opt.NUM != None and opt.NUM < 1:\r
        print "The MAX_CONCURRENT_THREAD_NUMBER must be larger than 0."\r
        sys.exit(1)\r
    if opt.TARGET != None and len(opt.TARGET) > 1:\r
        for elem in opt.TARGET:\r
            if elem == '0':\r
                print "0 will clear the TARGET setting in target.txt and can't combine with other value."\r
                sys.exit(1)\r
    if opt.TARGET_ARCH != None and len(opt.TARGET_ARCH) > 1:\r
        for elem in opt.TARGET_ARCH:\r
            if elem == '0':\r
                print "0 will clear the TARGET_ARCH setting in target.txt and can't combine with other value."\r
                sys.exit(1)\r
\r
    try:\r
        FileHandle = TargetTool(opt, args)\r
        if FileHandle.Arg.lower() == 'print':\r
            FileHandle.Print()\r
            sys.exit(0)\r
        elif FileHandle.Arg.lower() == 'clean':\r
            FileHandle.RWFile('#', '=', 0)\r
        else:\r
            FileHandle.RWFile('#', '=', 1)\r
    except Exception, e:\r
        last_type, last_value, last_tb = sys.exc_info()\r
        traceback.print_exception(last_type, last_value, last_tb)\r
\r