BaseTools: Refactor python print statements

[mirror_edk2.git] / BaseTools / Source / Python / Eot / c.py

## @file\r
# preprocess source file\r
#\r
#  Copyright (c) 2007 - 2014, 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 print_function\r
import sys\r
import Common.LongFilePathOs as os\r
import re\r
import CodeFragmentCollector\r
import FileProfile\r
from CommonDataClass import DataClass\r
from Common import EdkLogger\r
from EotToolError import *\r
import EotGlobalData\r
\r
# Global Dicts\r
IncludeFileListDict = {}\r
IncludePathListDict = {}\r
ComplexTypeDict = {}\r
SUDict = {}\r
\r
## GetFuncDeclPattern() method\r
#\r
#  Get the pattern of function declaration\r
#\r
#  @return p:    the pattern of function declaration\r
#\r
def GetFuncDeclPattern():\r
    p = re.compile(r'(EFIAPI|EFI_BOOT_SERVICE|EFI_RUNTIME_SERVICE)?\s*[_\w]+\s*\(.*\).*', re.DOTALL)\r
    return p\r
\r
## GetArrayPattern() method\r
#\r
#  Get the pattern of array\r
#\r
#  @return p:    the pattern of array\r
#\r
def GetArrayPattern():\r
    p = re.compile(r'[_\w]*\s*[\[.*\]]+')\r
    return p\r
\r
## GetTypedefFuncPointerPattern() method\r
#\r
#  Get the pattern of function pointer\r
#\r
#  @return p:    the pattern of function pointer\r
#\r
def GetTypedefFuncPointerPattern():\r
    p = re.compile('[_\w\s]*\([\w\s]*\*+\s*[_\w]+\s*\)\s*\(.*\)', re.DOTALL)\r
    return p\r
\r
## GetDB() method\r
#\r
#  Get global database instance\r
#\r
#  @return EotGlobalData.gDb:    the global database instance\r
#\r
def GetDB():\r
    return EotGlobalData.gDb\r
\r
## PrintErrorMsg() method\r
#\r
#  print error message\r
#\r
#  @param ErrorType: Type of error\r
#  @param Msg: Error message\r
#  @param TableName: table name of error found\r
#  @param ItemId: id of item\r
#\r
def PrintErrorMsg(ErrorType, Msg, TableName, ItemId):\r
    Msg = Msg.replace('\n', '').replace('\r', '')\r
    MsgPartList = Msg.split()\r
    Msg = ''\r
    for Part in MsgPartList:\r
        Msg += Part\r
        Msg += ' '\r
    GetDB().TblReport.Insert(ErrorType, OtherMsg = Msg, BelongsToTable = TableName, BelongsToItem = ItemId)\r
\r
## GetIdType() method\r
#\r
#  Find type of input string\r
#\r
#  @param Str: String to be parsed\r
#\r
#  @return Type: The type of the string\r
#\r
def GetIdType(Str):\r
    Type = DataClass.MODEL_UNKNOWN\r
    Str = Str.replace('#', '# ')\r
    List = Str.split()\r
    if List[1] == 'include':\r
        Type = DataClass.MODEL_IDENTIFIER_INCLUDE\r
    elif List[1] == 'define':\r
        Type = DataClass.MODEL_IDENTIFIER_MACRO_DEFINE\r
    elif List[1] == 'ifdef':\r
        Type = DataClass.MODEL_IDENTIFIER_MACRO_IFDEF\r
    elif List[1] == 'ifndef':\r
        Type = DataClass.MODEL_IDENTIFIER_MACRO_IFNDEF\r
    elif List[1] == 'endif':\r
        Type = DataClass.MODEL_IDENTIFIER_MACRO_ENDIF\r
    elif List[1] == 'pragma':\r
        Type = DataClass.MODEL_IDENTIFIER_MACRO_PROGMA\r
    else:\r
        Type = DataClass.MODEL_UNKNOWN\r
    return Type\r
\r
## GetIdentifierList() method\r
#\r
#  Get id of all files\r
#\r
#  @return IdList: The list of all id of files\r
#\r
def GetIdentifierList():\r
    IdList = []\r
\r
    for pp in FileProfile.PPDirectiveList:\r
        Type = GetIdType(pp.Content)\r
        IdPP = DataClass.IdentifierClass(-1, '', '', '', pp.Content, Type, -1, -1, pp.StartPos[0],pp.StartPos[1],pp.EndPos[0],pp.EndPos[1])\r
        IdList.append(IdPP)\r
\r
    for ae in FileProfile.AssignmentExpressionList:\r
        IdAE = DataClass.IdentifierClass(-1, ae.Operator, '', ae.Name, ae.Value, DataClass.MODEL_IDENTIFIER_ASSIGNMENT_EXPRESSION, -1, -1, ae.StartPos[0],ae.StartPos[1],ae.EndPos[0],ae.EndPos[1])\r
        IdList.append(IdAE)\r
\r
    FuncDeclPattern = GetFuncDeclPattern()\r
    ArrayPattern = GetArrayPattern()\r
    for var in FileProfile.VariableDeclarationList:\r
        DeclText = var.Declarator.strip()\r
        while DeclText.startswith('*'):\r
            var.Modifier += '*'\r
            DeclText = DeclText.lstrip('*').strip()\r
        var.Declarator = DeclText\r
        if FuncDeclPattern.match(var.Declarator):\r
            DeclSplitList = var.Declarator.split('(')\r
            FuncName = DeclSplitList[0]\r
            FuncNamePartList = FuncName.split()\r
            if len(FuncNamePartList) > 1:\r
                FuncName = FuncNamePartList[-1]\r
                Index = 0\r
                while Index < len(FuncNamePartList) - 1:\r
                    var.Modifier += ' ' + FuncNamePartList[Index]\r
                    var.Declarator = var.Declarator.lstrip().lstrip(FuncNamePartList[Index])\r
                    Index += 1\r
            IdVar = DataClass.IdentifierClass(-1, var.Modifier, '', var.Declarator, '', DataClass.MODEL_IDENTIFIER_FUNCTION_DECLARATION, -1, -1, var.StartPos[0],var.StartPos[1],var.EndPos[0],var.EndPos[1])\r
            IdList.append(IdVar)\r
            continue\r
\r
        if var.Declarator.find('{') == -1:\r
            for decl in var.Declarator.split(','):\r
                DeclList = decl.split('=')\r
                Name = DeclList[0].strip()\r
                if ArrayPattern.match(Name):\r
                    LSBPos = var.Declarator.find('[')\r
                    var.Modifier += ' ' + Name[LSBPos:]\r
                    Name = Name[0:LSBPos]\r
\r
                IdVar = DataClass.IdentifierClass(-1, var.Modifier, '', Name, (len(DeclList) > 1 and [DeclList[1]]or [''])[0], DataClass.MODEL_IDENTIFIER_VARIABLE, -1, -1, var.StartPos[0],var.StartPos[1],var.EndPos[0],var.EndPos[1])\r
                IdList.append(IdVar)\r
        else:\r
            DeclList = var.Declarator.split('=')\r
            Name = DeclList[0].strip()\r
            if ArrayPattern.match(Name):\r
                LSBPos = var.Declarator.find('[')\r
                var.Modifier += ' ' + Name[LSBPos:]\r
                Name = Name[0:LSBPos]\r
            IdVar = DataClass.IdentifierClass(-1, var.Modifier, '', Name, (len(DeclList) > 1 and [DeclList[1]]or [''])[0], DataClass.MODEL_IDENTIFIER_VARIABLE, -1, -1, var.StartPos[0],var.StartPos[1],var.EndPos[0],var.EndPos[1])\r
            IdList.append(IdVar)\r
\r
    for enum in FileProfile.EnumerationDefinitionList:\r
        LBPos = enum.Content.find('{')\r
        RBPos = enum.Content.find('}')\r
        Name = enum.Content[4:LBPos].strip()\r
        Value = enum.Content[LBPos+1:RBPos]\r
        IdEnum = DataClass.IdentifierClass(-1, '', '', Name, Value, DataClass.MODEL_IDENTIFIER_ENUMERATE, -1, -1, enum.StartPos[0],enum.StartPos[1],enum.EndPos[0],enum.EndPos[1])\r
        IdList.append(IdEnum)\r
\r
    for su in FileProfile.StructUnionDefinitionList:\r
        Type = DataClass.MODEL_IDENTIFIER_STRUCTURE\r
        SkipLen = 6\r
        if su.Content.startswith('union'):\r
            Type = DataClass.MODEL_IDENTIFIER_UNION\r
            SkipLen = 5\r
        LBPos = su.Content.find('{')\r
        RBPos = su.Content.find('}')\r
        if LBPos == -1 or RBPos == -1:\r
            Name = su.Content[SkipLen:].strip()\r
            Value = ''\r
        else:\r
            Name = su.Content[SkipLen:LBPos].strip()\r
            Value = su.Content[LBPos+1:RBPos]\r
        IdPE = DataClass.IdentifierClass(-1, '', '', Name, Value, Type, -1, -1, su.StartPos[0],su.StartPos[1],su.EndPos[0],su.EndPos[1])\r
        IdList.append(IdPE)\r
\r
    TdFuncPointerPattern = GetTypedefFuncPointerPattern()\r
    for td in FileProfile.TypedefDefinitionList:\r
        Modifier = ''\r
        Name = td.ToType\r
        Value = td.FromType\r
        if TdFuncPointerPattern.match(td.ToType):\r
            Modifier = td.FromType\r
            LBPos = td.ToType.find('(')\r
            TmpStr = td.ToType[LBPos+1:].strip()\r
            StarPos = TmpStr.find('*')\r
            if StarPos != -1:\r
                Modifier += ' ' + TmpStr[0:StarPos]\r
            while TmpStr[StarPos] == '*':\r
                Modifier += ' ' + '*'\r
                StarPos += 1\r
            TmpStr = TmpStr[StarPos:].strip()\r
            RBPos = TmpStr.find(')')\r
            Name = TmpStr[0:RBPos]\r
            Value = 'FP' + TmpStr[RBPos + 1:]\r
\r
        IdTd = DataClass.IdentifierClass(-1, Modifier, '', Name, Value, DataClass.MODEL_IDENTIFIER_TYPEDEF, -1, -1, td.StartPos[0],td.StartPos[1],td.EndPos[0],td.EndPos[1])\r
        IdList.append(IdTd)\r
\r
    for funcCall in FileProfile.FunctionCallingList:\r
        IdFC = DataClass.IdentifierClass(-1, '', '', funcCall.FuncName, funcCall.ParamList, DataClass.MODEL_IDENTIFIER_FUNCTION_CALLING, -1, -1, funcCall.StartPos[0],funcCall.StartPos[1],funcCall.EndPos[0],funcCall.EndPos[1])\r
        IdList.append(IdFC)\r
    return IdList\r
\r
## GetParamList() method\r
#\r
#  Get a list of parameters\r
#\r
#  @param FuncDeclarator: Function declarator\r
#  @param FuncNameLine: Line number of function name\r
#  @param FuncNameOffset: Offset of function name\r
#\r
#  @return ParamIdList: A list of parameters\r
#\r
def GetParamList(FuncDeclarator, FuncNameLine = 0, FuncNameOffset = 0):\r
    ParamIdList = []\r
    DeclSplitList = FuncDeclarator.split('(')\r
    if len(DeclSplitList) < 2:\r
        return ParamIdList\r
    FuncName = DeclSplitList[0]\r
    ParamStr = DeclSplitList[1].rstrip(')')\r
    LineSkipped = 0\r
    OffsetSkipped = 0\r
    Start = 0\r
    while FuncName.find('\n', Start) != -1:\r
        LineSkipped += 1\r
        OffsetSkipped = 0\r
        Start += FuncName.find('\n', Start)\r
        Start += 1\r
    OffsetSkipped += len(FuncName[Start:])\r
    OffsetSkipped += 1 #skip '('\r
    ParamBeginLine = FuncNameLine + LineSkipped\r
    ParamBeginOffset = OffsetSkipped\r
    for p in ParamStr.split(','):\r
        ListP = p.split()\r
        if len(ListP) == 0:\r
            continue\r
        ParamName = ListP[-1]\r
        DeclText = ParamName.strip()\r
        RightSpacePos = p.rfind(ParamName)\r
        ParamModifier = p[0:RightSpacePos]\r
        if ParamName == 'OPTIONAL':\r
            if ParamModifier == '':\r
                ParamModifier += ' ' + 'OPTIONAL'\r
                DeclText = ''\r
            else:\r
                ParamName = ListP[-2]\r
                DeclText = ParamName.strip()\r
                RightSpacePos = p.rfind(ParamName)\r
                ParamModifier = p[0:RightSpacePos]\r
                ParamModifier += 'OPTIONAL'\r
        while DeclText.startswith('*'):\r
            ParamModifier += ' ' + '*'\r
            DeclText = DeclText.lstrip('*').strip()\r
        ParamName = DeclText\r
\r
        Start = 0\r
        while p.find('\n', Start) != -1:\r
            LineSkipped += 1\r
            OffsetSkipped = 0\r
            Start += p.find('\n', Start)\r
            Start += 1\r
        OffsetSkipped += len(p[Start:])\r
\r
        ParamEndLine = ParamBeginLine + LineSkipped\r
        ParamEndOffset = OffsetSkipped\r
        IdParam = DataClass.IdentifierClass(-1, ParamModifier, '', ParamName, '', DataClass.MODEL_IDENTIFIER_PARAMETER, -1, -1, ParamBeginLine, ParamBeginOffset, ParamEndLine, ParamEndOffset)\r
        ParamIdList.append(IdParam)\r
        ParamBeginLine = ParamEndLine\r
        ParamBeginOffset = OffsetSkipped + 1 #skip ','\r
\r
    return ParamIdList\r
\r
## GetFunctionList()\r
#\r
#  Get a list of functions\r
#\r
#  @return FuncObjList: A list of function objects\r
#\r
def GetFunctionList():\r
    FuncObjList = []\r
    for FuncDef in FileProfile.FunctionDefinitionList:\r
        ParamIdList = []\r
        DeclText = FuncDef.Declarator.strip()\r
        while DeclText.startswith('*'):\r
            FuncDef.Modifier += '*'\r
            DeclText = DeclText.lstrip('*').strip()\r
\r
        FuncDef.Declarator = FuncDef.Declarator.lstrip('*')\r
        DeclSplitList = FuncDef.Declarator.split('(')\r
        if len(DeclSplitList) < 2:\r
            continue\r
\r
        FuncName = DeclSplitList[0]\r
        FuncNamePartList = FuncName.split()\r
        if len(FuncNamePartList) > 1:\r
            FuncName = FuncNamePartList[-1]\r
            Index = 0\r
            while Index < len(FuncNamePartList) - 1:\r
                FuncDef.Modifier += ' ' + FuncNamePartList[Index]\r
                Index += 1\r
\r
        FuncObj = DataClass.FunctionClass(-1, FuncDef.Declarator, FuncDef.Modifier, FuncName.strip(), '', FuncDef.StartPos[0],FuncDef.StartPos[1],FuncDef.EndPos[0],FuncDef.EndPos[1], FuncDef.LeftBracePos[0], FuncDef.LeftBracePos[1], -1, ParamIdList, [])\r
        FuncObjList.append(FuncObj)\r
\r
    return FuncObjList\r
\r
## CreateCCodeDB() method\r
#\r
#  Create database for all c code\r
#\r
#  @param FileNameList: A list of all c code file names\r
#\r
def CreateCCodeDB(FileNameList):\r
    FileObjList = []\r
    ParseErrorFileList = []\r
    ParsedFiles = {}\r
    for FullName in FileNameList:\r
        if os.path.splitext(FullName)[1] in ('.h', '.c'):\r
            if FullName.lower() in ParsedFiles:\r
                continue\r
            ParsedFiles[FullName.lower()] = 1\r
            EdkLogger.info("Parsing " + FullName)\r
            model = FullName.endswith('c') and DataClass.MODEL_FILE_C or DataClass.MODEL_FILE_H\r
            collector = CodeFragmentCollector.CodeFragmentCollector(FullName)\r
            try:\r
                collector.ParseFile()\r
            except:\r
                ParseErrorFileList.append(FullName)\r
            BaseName = os.path.basename(FullName)\r
            DirName = os.path.dirname(FullName)\r
            Ext = os.path.splitext(BaseName)[1].lstrip('.')\r
            ModifiedTime = os.path.getmtime(FullName)\r
            FileObj = DataClass.FileClass(-1, BaseName, Ext, DirName, FullName, model, ModifiedTime, GetFunctionList(), GetIdentifierList(), [])\r
            FileObjList.append(FileObj)\r
            collector.CleanFileProfileBuffer()\r
\r
    if len(ParseErrorFileList) > 0:\r
        EdkLogger.info("Found unrecoverable error during parsing:\n\t%s\n" % "\n\t".join(ParseErrorFileList))\r
\r
    Db = EotGlobalData.gDb\r
    for file in FileObjList:\r
        Db.InsertOneFile(file)\r
\r
    Db.UpdateIdentifierBelongsToFunction()\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
\r
    EdkLogger.Initialize()\r
    EdkLogger.SetLevel(EdkLogger.QUIET)\r
    CollectSourceCodeDataIntoDB(sys.argv[1])\r
\r
    print('Done!')\r