BaseTools/UPT: Porting UPT Tool from Python2 to Python3

[mirror_edk2.git] / BaseTools / Source / Python / UPT / Library / CommentGenerating.py

## @file\r
# This file is used to define comment generating interface\r
#\r
# Copyright (c) 2011 - 2018, Intel Corporation. All rights reserved.<BR>\r
#\r
# This program and the accompanying materials are licensed and made available\r
# under the terms and conditions of the BSD License which accompanies this\r
# 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
CommentGenerating\r
'''\r
\r
##\r
# Import Modules\r
#\r
from Library.StringUtils import GetSplitValueList\r
from Library.DataType import TAB_SPACE_SPLIT\r
from Library.DataType import TAB_INF_GUIDTYPE_VAR\r
from Library.DataType import USAGE_ITEM_NOTIFY\r
from Library.DataType import ITEM_UNDEFINED\r
from Library.DataType import TAB_HEADER_COMMENT\r
from Library.DataType import TAB_BINARY_HEADER_COMMENT\r
from Library.DataType import TAB_COMMENT_SPLIT\r
from Library.DataType import TAB_SPECIAL_COMMENT\r
from Library.DataType import END_OF_LINE\r
from Library.DataType import TAB_COMMENT_EDK1_SPLIT\r
from Library.DataType import TAB_COMMENT_EDK1_START\r
from Library.DataType import TAB_COMMENT_EDK1_END\r
from Library.DataType import TAB_STAR\r
from Library.DataType import TAB_PCD_PROMPT\r
from Library.UniClassObject import ConvertSpecialUnicodes\r
from Library.Misc import GetLocalValue\r
## GenTailCommentLines\r
#\r
# @param TailCommentLines:  the tail comment lines that need to be generated\r
# @param LeadingSpaceNum:   the number of leading space needed for non-first\r
#                            line tail comment\r
#\r
def GenTailCommentLines (TailCommentLines, LeadingSpaceNum = 0):\r
    TailCommentLines = TailCommentLines.rstrip(END_OF_LINE)\r
    CommentStr = TAB_SPACE_SPLIT*2 + TAB_SPECIAL_COMMENT + TAB_SPACE_SPLIT + \\r
    (END_OF_LINE + LeadingSpaceNum * TAB_SPACE_SPLIT + TAB_SPACE_SPLIT*2 + TAB_SPECIAL_COMMENT + \\r
     TAB_SPACE_SPLIT).join(GetSplitValueList(TailCommentLines, END_OF_LINE))\r
\r
    return CommentStr\r
\r
## GenGenericComment\r
#\r
# @param CommentLines:   Generic comment Text, maybe Multiple Lines\r
#\r
def GenGenericComment (CommentLines):\r
    if not CommentLines:\r
        return ''\r
    CommentLines = CommentLines.rstrip(END_OF_LINE)\r
    CommentStr = TAB_SPECIAL_COMMENT + TAB_SPACE_SPLIT + (END_OF_LINE + TAB_COMMENT_SPLIT + TAB_SPACE_SPLIT).join\\r
    (GetSplitValueList(CommentLines, END_OF_LINE)) + END_OF_LINE\r
    return CommentStr\r
\r
## GenGenericCommentF\r
#\r
#  similar to GenGenericComment but will remove <EOL> at end of comment once,\r
#  and for line with only <EOL>, '#\n' will be generated instead of '# \n'\r
#\r
# @param CommentLines:   Generic comment Text, maybe Multiple Lines\r
# @return CommentStr:    Generated comment line\r
#\r
def GenGenericCommentF (CommentLines, NumOfPound=1, IsPrompt=False, IsInfLibraryClass=False):\r
    if not CommentLines:\r
        return ''\r
    #\r
    # if comment end with '\n', then remove it to prevent one extra line\r
    # generate later on\r
    #\r
    if CommentLines.endswith(END_OF_LINE):\r
        CommentLines = CommentLines[:-1]\r
    CommentStr = ''\r
    if IsPrompt:\r
        CommentStr += TAB_COMMENT_SPLIT * NumOfPound + TAB_SPACE_SPLIT + TAB_PCD_PROMPT + TAB_SPACE_SPLIT + \\r
        CommentLines.replace(END_OF_LINE, '') + END_OF_LINE\r
    else:\r
        CommentLineList = GetSplitValueList(CommentLines, END_OF_LINE)\r
        FindLibraryClass = False\r
        for Line in CommentLineList:\r
            # If this comment is for @libraryclass and it has multiple lines\r
            # make sure the second lines align to the first line after @libraryclass as below\r
            #\r
            # ## @libraryclass XYZ FIRST_LINE\r
            # ##               ABC SECOND_LINE\r
            #\r
            if IsInfLibraryClass and Line.find(u'@libraryclass ') > -1:\r
                FindLibraryClass = True\r
            if Line == '':\r
                CommentStr += TAB_COMMENT_SPLIT * NumOfPound + END_OF_LINE\r
            else:\r
                if FindLibraryClass and Line.find(u'@libraryclass ') > -1:\r
                    CommentStr += TAB_COMMENT_SPLIT * NumOfPound + TAB_SPACE_SPLIT + Line + END_OF_LINE\r
                elif FindLibraryClass:\r
                    CommentStr += TAB_COMMENT_SPLIT * NumOfPound + TAB_SPACE_SPLIT * 16 + Line + END_OF_LINE\r
                else:\r
                    CommentStr += TAB_COMMENT_SPLIT * NumOfPound + TAB_SPACE_SPLIT + Line + END_OF_LINE\r
\r
    return CommentStr\r
\r
\r
## GenHeaderCommentSection\r
#\r
# Generate Header comment sections\r
#\r
# @param Abstract      One line of abstract\r
# @param Description   multiple lines of Description\r
# @param Copyright     possible multiple copyright lines\r
# @param License       possible multiple license lines\r
#\r
def GenHeaderCommentSection(Abstract, Description, Copyright, License, IsBinaryHeader=False, \\r
                            CommChar=TAB_COMMENT_SPLIT):\r
    Content = ''\r
\r
    #\r
    # Convert special character to (c), (r) and (tm).\r
    #\r
    if IsBinaryHeader:\r
        Content += CommChar * 2 + TAB_SPACE_SPLIT + TAB_BINARY_HEADER_COMMENT + '\n'\r
    elif CommChar == TAB_COMMENT_EDK1_SPLIT:\r
        Content += CommChar + TAB_SPACE_SPLIT + TAB_COMMENT_EDK1_START + TAB_STAR + TAB_SPACE_SPLIT +\\r
         TAB_HEADER_COMMENT + '\n'\r
    else:\r
        Content += CommChar * 2 + TAB_SPACE_SPLIT + TAB_HEADER_COMMENT + '\n'\r
    if Abstract:\r
        Abstract = Abstract.rstrip('\n')\r
        Content += CommChar + TAB_SPACE_SPLIT + ('\n' + CommChar + TAB_SPACE_SPLIT).join(GetSplitValueList\\r
                                                                                                (Abstract, '\n'))\r
        Content += '\n' + CommChar + '\n'\r
    else:\r
        Content += CommChar + '\n'\r
\r
    if Description:\r
        Description = Description.rstrip('\n')\r
        Content += CommChar + TAB_SPACE_SPLIT + ('\n' + CommChar + TAB_SPACE_SPLIT).join(GetSplitValueList\\r
                                                  (Description, '\n'))\r
        Content += '\n' + CommChar + '\n'\r
\r
    #\r
    # There is no '#\n' line to separate multiple copyright lines in code base\r
    #\r
    if Copyright:\r
        Copyright = Copyright.rstrip('\n')\r
        Content += CommChar + TAB_SPACE_SPLIT + ('\n' + CommChar + TAB_SPACE_SPLIT).join\\r
        (GetSplitValueList(Copyright, '\n'))\r
        Content += '\n' + CommChar + '\n'\r
\r
    if License:\r
        License = License.rstrip('\n')\r
        Content += CommChar + TAB_SPACE_SPLIT + ('\n' + CommChar + TAB_SPACE_SPLIT).join(GetSplitValueList\\r
                                                  (License, '\n'))\r
        Content += '\n' + CommChar + '\n'\r
\r
    if CommChar == TAB_COMMENT_EDK1_SPLIT:\r
        Content += CommChar + TAB_SPACE_SPLIT + TAB_STAR + TAB_COMMENT_EDK1_END + '\n'\r
    else:\r
        Content += CommChar * 2 + '\n'\r
\r
    return Content\r
\r
\r
## GenInfPcdTailComment\r
#  Generate Pcd tail comment for Inf, this would be one line comment\r
#\r
# @param Usage:            Usage type\r
# @param TailCommentText:  Comment text for tail comment\r
#\r
def GenInfPcdTailComment (Usage, TailCommentText):\r
    if (Usage == ITEM_UNDEFINED) and (not TailCommentText):\r
        return ''\r
\r
    CommentLine = TAB_SPACE_SPLIT.join([Usage, TailCommentText])\r
    return GenTailCommentLines(CommentLine)\r
\r
## GenInfProtocolPPITailComment\r
#  Generate Protocol/PPI tail comment for Inf\r
#\r
# @param Usage:            Usage type\r
# @param TailCommentText:  Comment text for tail comment\r
#\r
def GenInfProtocolPPITailComment (Usage, Notify, TailCommentText):\r
    if (not Notify) and (Usage == ITEM_UNDEFINED) and (not TailCommentText):\r
        return ''\r
\r
    if Notify:\r
        CommentLine = USAGE_ITEM_NOTIFY + " ## "\r
    else:\r
        CommentLine = ''\r
\r
    CommentLine += TAB_SPACE_SPLIT.join([Usage, TailCommentText])\r
    return GenTailCommentLines(CommentLine)\r
\r
## GenInfGuidTailComment\r
#  Generate Guid tail comment for Inf\r
#\r
# @param Usage:            Usage type\r
# @param TailCommentText:  Comment text for tail comment\r
#\r
def GenInfGuidTailComment (Usage, GuidTypeList, VariableName, TailCommentText):\r
    GuidType = GuidTypeList[0]\r
    if (Usage == ITEM_UNDEFINED) and (GuidType == ITEM_UNDEFINED) and \\r
        (not TailCommentText):\r
        return ''\r
\r
    FirstLine = Usage + " ## " + GuidType\r
    if GuidType == TAB_INF_GUIDTYPE_VAR:\r
        FirstLine += ":" + VariableName\r
\r
    CommentLine = TAB_SPACE_SPLIT.join([FirstLine, TailCommentText])\r
    return GenTailCommentLines(CommentLine)\r
\r
## GenDecGuidTailComment\r
#\r
# @param SupModuleList:  Supported module type list\r
#\r
def GenDecTailComment (SupModuleList):\r
    CommentLine = TAB_SPACE_SPLIT.join(SupModuleList)\r
    return GenTailCommentLines(CommentLine)\r
\r
\r
## _GetHelpStr\r
#  get HelpString from a list of HelpTextObject, the priority refer to\r
#  related HLD\r
#\r
#  @param HelpTextObjList: List of HelpTextObject\r
#\r
#  @return HelpStr: the help text string found, '' means no help text found\r
#\r
def _GetHelpStr(HelpTextObjList):\r
    ValueList = []\r
    for HelpObj in HelpTextObjList:\r
        ValueList.append((HelpObj.GetLang(), HelpObj.GetString()))\r
    return GetLocalValue(ValueList, True)\r