| ## @file | |
| # This file is used to collect all defined strings in multiple uni files | |
| # | |
| # | |
| # Copyright (c) 2014 Hewlett-Packard Development Company, L.P.<BR> | |
| # | |
| # Copyright (c) 2007 - 2016, Intel Corporation. All rights reserved.<BR> | |
| # This program and the accompanying materials | |
| # are licensed and made available under the terms and conditions of the BSD License | |
| # which accompanies this distribution. The full text of the license may be found at | |
| # http://opensource.org/licenses/bsd-license.php | |
| # | |
| # THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, | |
| # WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. | |
| ## | |
| # Import Modules | |
| # | |
| import Common.LongFilePathOs as os, codecs, re | |
| import distutils.util | |
| import Common.EdkLogger as EdkLogger | |
| import StringIO | |
| from Common.BuildToolError import * | |
| from Common.String import GetLineNo | |
| from Common.Misc import PathClass | |
| from Common.LongFilePathSupport import LongFilePath | |
| ## | |
| # Static definitions | |
| # | |
| UNICODE_WIDE_CHAR = u'\\wide' | |
| UNICODE_NARROW_CHAR = u'\\narrow' | |
| UNICODE_NON_BREAKING_CHAR = u'\\nbr' | |
| UNICODE_UNICODE_CR = '\r' | |
| UNICODE_UNICODE_LF = '\n' | |
| NARROW_CHAR = u'\uFFF0' | |
| WIDE_CHAR = u'\uFFF1' | |
| NON_BREAKING_CHAR = u'\uFFF2' | |
| CR = u'\u000D' | |
| LF = u'\u000A' | |
| NULL = u'\u0000' | |
| TAB = u'\t' | |
| BACK_SLASH_PLACEHOLDER = u'\u0006' | |
| gIncludePattern = re.compile("^#include +[\"<]+([^\"< >]+)[>\"]+$", re.MULTILINE | re.UNICODE) | |
| ## Convert a python unicode string to a normal string | |
| # | |
| # Convert a python unicode string to a normal string | |
| # UniToStr(u'I am a string') is 'I am a string' | |
| # | |
| # @param Uni: The python unicode string | |
| # | |
| # @retval: The formatted normal string | |
| # | |
| def UniToStr(Uni): | |
| return repr(Uni)[2:-1] | |
| ## Convert a unicode string to a Hex list | |
| # | |
| # Convert a unicode string to a Hex list | |
| # UniToHexList('ABC') is ['0x41', '0x00', '0x42', '0x00', '0x43', '0x00'] | |
| # | |
| # @param Uni: The python unicode string | |
| # | |
| # @retval List: The formatted hex list | |
| # | |
| def UniToHexList(Uni): | |
| List = [] | |
| for Item in Uni: | |
| Temp = '%04X' % ord(Item) | |
| List.append('0x' + Temp[2:4]) | |
| List.append('0x' + Temp[0:2]) | |
| return List | |
| LangConvTable = {'eng':'en', 'fra':'fr', \ | |
| 'aar':'aa', 'abk':'ab', 'ave':'ae', 'afr':'af', 'aka':'ak', 'amh':'am', \ | |
| 'arg':'an', 'ara':'ar', 'asm':'as', 'ava':'av', 'aym':'ay', 'aze':'az', \ | |
| 'bak':'ba', 'bel':'be', 'bul':'bg', 'bih':'bh', 'bis':'bi', 'bam':'bm', \ | |
| 'ben':'bn', 'bod':'bo', 'bre':'br', 'bos':'bs', 'cat':'ca', 'che':'ce', \ | |
| 'cha':'ch', 'cos':'co', 'cre':'cr', 'ces':'cs', 'chu':'cu', 'chv':'cv', \ | |
| 'cym':'cy', 'dan':'da', 'deu':'de', 'div':'dv', 'dzo':'dz', 'ewe':'ee', \ | |
| 'ell':'el', 'epo':'eo', 'spa':'es', 'est':'et', 'eus':'eu', 'fas':'fa', \ | |
| 'ful':'ff', 'fin':'fi', 'fij':'fj', 'fao':'fo', 'fry':'fy', 'gle':'ga', \ | |
| 'gla':'gd', 'glg':'gl', 'grn':'gn', 'guj':'gu', 'glv':'gv', 'hau':'ha', \ | |
| 'heb':'he', 'hin':'hi', 'hmo':'ho', 'hrv':'hr', 'hat':'ht', 'hun':'hu', \ | |
| 'hye':'hy', 'her':'hz', 'ina':'ia', 'ind':'id', 'ile':'ie', 'ibo':'ig', \ | |
| 'iii':'ii', 'ipk':'ik', 'ido':'io', 'isl':'is', 'ita':'it', 'iku':'iu', \ | |
| 'jpn':'ja', 'jav':'jv', 'kat':'ka', 'kon':'kg', 'kik':'ki', 'kua':'kj', \ | |
| 'kaz':'kk', 'kal':'kl', 'khm':'km', 'kan':'kn', 'kor':'ko', 'kau':'kr', \ | |
| 'kas':'ks', 'kur':'ku', 'kom':'kv', 'cor':'kw', 'kir':'ky', 'lat':'la', \ | |
| 'ltz':'lb', 'lug':'lg', 'lim':'li', 'lin':'ln', 'lao':'lo', 'lit':'lt', \ | |
| 'lub':'lu', 'lav':'lv', 'mlg':'mg', 'mah':'mh', 'mri':'mi', 'mkd':'mk', \ | |
| 'mal':'ml', 'mon':'mn', 'mar':'mr', 'msa':'ms', 'mlt':'mt', 'mya':'my', \ | |
| 'nau':'na', 'nob':'nb', 'nde':'nd', 'nep':'ne', 'ndo':'ng', 'nld':'nl', \ | |
| 'nno':'nn', 'nor':'no', 'nbl':'nr', 'nav':'nv', 'nya':'ny', 'oci':'oc', \ | |
| 'oji':'oj', 'orm':'om', 'ori':'or', 'oss':'os', 'pan':'pa', 'pli':'pi', \ | |
| 'pol':'pl', 'pus':'ps', 'por':'pt', 'que':'qu', 'roh':'rm', 'run':'rn', \ | |
| 'ron':'ro', 'rus':'ru', 'kin':'rw', 'san':'sa', 'srd':'sc', 'snd':'sd', \ | |
| 'sme':'se', 'sag':'sg', 'sin':'si', 'slk':'sk', 'slv':'sl', 'smo':'sm', \ | |
| 'sna':'sn', 'som':'so', 'sqi':'sq', 'srp':'sr', 'ssw':'ss', 'sot':'st', \ | |
| 'sun':'su', 'swe':'sv', 'swa':'sw', 'tam':'ta', 'tel':'te', 'tgk':'tg', \ | |
| 'tha':'th', 'tir':'ti', 'tuk':'tk', 'tgl':'tl', 'tsn':'tn', 'ton':'to', \ | |
| 'tur':'tr', 'tso':'ts', 'tat':'tt', 'twi':'tw', 'tah':'ty', 'uig':'ug', \ | |
| 'ukr':'uk', 'urd':'ur', 'uzb':'uz', 'ven':'ve', 'vie':'vi', 'vol':'vo', \ | |
| 'wln':'wa', 'wol':'wo', 'xho':'xh', 'yid':'yi', 'yor':'yo', 'zha':'za', \ | |
| 'zho':'zh', 'zul':'zu'} | |
| ## GetLanguageCode | |
| # | |
| # Check the language code read from .UNI file and convert ISO 639-2 codes to RFC 4646 codes if appropriate | |
| # ISO 639-2 language codes supported in compatiblity mode | |
| # RFC 4646 language codes supported in native mode | |
| # | |
| # @param LangName: Language codes read from .UNI file | |
| # | |
| # @retval LangName: Valid lanugage code in RFC 4646 format or None | |
| # | |
| def GetLanguageCode(LangName, IsCompatibleMode, File): | |
| global LangConvTable | |
| length = len(LangName) | |
| if IsCompatibleMode: | |
| if length == 3 and LangName.isalpha(): | |
| TempLangName = LangConvTable.get(LangName.lower()) | |
| if TempLangName != None: | |
| return TempLangName | |
| return LangName | |
| else: | |
| EdkLogger.error("Unicode File Parser", FORMAT_INVALID, "Invalid ISO 639-2 language code : %s" % LangName, File) | |
| if (LangName[0] == 'X' or LangName[0] == 'x') and LangName[1] == '-': | |
| return LangName | |
| if length == 2: | |
| if LangName.isalpha(): | |
| return LangName | |
| elif length == 3: | |
| if LangName.isalpha() and LangConvTable.get(LangName.lower()) == None: | |
| return LangName | |
| elif length == 5: | |
| if LangName[0:2].isalpha() and LangName[2] == '-': | |
| return LangName | |
| elif length >= 6: | |
| if LangName[0:2].isalpha() and LangName[2] == '-': | |
| return LangName | |
| if LangName[0:3].isalpha() and LangConvTable.get(LangName.lower()) == None and LangName[3] == '-': | |
| return LangName | |
| EdkLogger.error("Unicode File Parser", FORMAT_INVALID, "Invalid RFC 4646 language code : %s" % LangName, File) | |
| ## Ucs2Codec | |
| # | |
| # This is only a partial codec implementation. It only supports | |
| # encoding, and is primarily used to check that all the characters are | |
| # valid for UCS-2. | |
| # | |
| class Ucs2Codec(codecs.Codec): | |
| def __init__(self): | |
| self.__utf16 = codecs.lookup('utf-16') | |
| def encode(self, input, errors='strict'): | |
| for Char in input: | |
| CodePoint = ord(Char) | |
| if CodePoint >= 0xd800 and CodePoint <= 0xdfff: | |
| raise ValueError("Code Point is in range reserved for " + | |
| "UTF-16 surrogate pairs") | |
| elif CodePoint > 0xffff: | |
| raise ValueError("Code Point too large to encode in UCS-2") | |
| return self.__utf16.encode(input) | |
| TheUcs2Codec = Ucs2Codec() | |
| def Ucs2Search(name): | |
| if name == 'ucs-2': | |
| return codecs.CodecInfo( | |
| name=name, | |
| encode=TheUcs2Codec.encode, | |
| decode=TheUcs2Codec.decode) | |
| else: | |
| return None | |
| codecs.register(Ucs2Search) | |
| ## StringDefClassObject | |
| # | |
| # A structure for language definition | |
| # | |
| class StringDefClassObject(object): | |
| def __init__(self, Name = None, Value = None, Referenced = False, Token = None, UseOtherLangDef = ''): | |
| self.StringName = '' | |
| self.StringNameByteList = [] | |
| self.StringValue = '' | |
| self.StringValueByteList = '' | |
| self.Token = 0 | |
| self.Referenced = Referenced | |
| self.UseOtherLangDef = UseOtherLangDef | |
| self.Length = 0 | |
| if Name != None: | |
| self.StringName = Name | |
| self.StringNameByteList = UniToHexList(Name) | |
| if Value != None: | |
| self.StringValue = Value + u'\x00' # Add a NULL at string tail | |
| self.StringValueByteList = UniToHexList(self.StringValue) | |
| self.Length = len(self.StringValueByteList) | |
| if Token != None: | |
| self.Token = Token | |
| def __str__(self): | |
| return repr(self.StringName) + ' ' + \ | |
| repr(self.Token) + ' ' + \ | |
| repr(self.Referenced) + ' ' + \ | |
| repr(self.StringValue) + ' ' + \ | |
| repr(self.UseOtherLangDef) | |
| def UpdateValue(self, Value = None): | |
| if Value != None: | |
| self.StringValue = Value + u'\x00' # Add a NULL at string tail | |
| self.StringValueByteList = UniToHexList(self.StringValue) | |
| self.Length = len(self.StringValueByteList) | |
| ## UniFileClassObject | |
| # | |
| # A structure for .uni file definition | |
| # | |
| class UniFileClassObject(object): | |
| def __init__(self, FileList = [], IsCompatibleMode = False, IncludePathList = []): | |
| self.FileList = FileList | |
| self.Token = 2 | |
| self.LanguageDef = [] #[ [u'LanguageIdentifier', u'PrintableName'], ... ] | |
| self.OrderedStringList = {} #{ u'LanguageIdentifier' : [StringDefClassObject] } | |
| self.OrderedStringDict = {} #{ u'LanguageIdentifier' : {StringName:(IndexInList)} } | |
| self.OrderedStringListByToken = {} #{ u'LanguageIdentifier' : {Token: StringDefClassObject} } | |
| self.IsCompatibleMode = IsCompatibleMode | |
| self.IncludePathList = IncludePathList | |
| if len(self.FileList) > 0: | |
| self.LoadUniFiles(FileList) | |
| # | |
| # Get Language definition | |
| # | |
| def GetLangDef(self, File, Line): | |
| Lang = distutils.util.split_quoted((Line.split(u"//")[0])) | |
| if len(Lang) != 3: | |
| try: | |
| FileIn = self.OpenUniFile(LongFilePath(File.Path)) | |
| except UnicodeError, X: | |
| EdkLogger.error("build", FILE_READ_FAILURE, "File read failure: %s" % str(X), ExtraData=File); | |
| except: | |
| EdkLogger.error("build", FILE_OPEN_FAILURE, ExtraData=File); | |
| LineNo = GetLineNo(FileIn, Line, False) | |
| EdkLogger.error("Unicode File Parser", PARSER_ERROR, "Wrong language definition", | |
| ExtraData="""%s\n\t*Correct format is like '#langdef en-US "English"'""" % Line, File=File, Line=LineNo) | |
| else: | |
| LangName = GetLanguageCode(Lang[1], self.IsCompatibleMode, self.File) | |
| LangPrintName = Lang[2] | |
| IsLangInDef = False | |
| for Item in self.LanguageDef: | |
| if Item[0] == LangName: | |
| IsLangInDef = True | |
| break; | |
| if not IsLangInDef: | |
| self.LanguageDef.append([LangName, LangPrintName]) | |
| # | |
| # Add language string | |
| # | |
| self.AddStringToList(u'$LANGUAGE_NAME', LangName, LangName, 0, True, Index=0) | |
| self.AddStringToList(u'$PRINTABLE_LANGUAGE_NAME', LangName, LangPrintName, 1, True, Index=1) | |
| if not IsLangInDef: | |
| # | |
| # The found STRING tokens will be added into new language string list | |
| # so that the unique STRING identifier is reserved for all languages in the package list. | |
| # | |
| FirstLangName = self.LanguageDef[0][0] | |
| if LangName != FirstLangName: | |
| for Index in range (2, len (self.OrderedStringList[FirstLangName])): | |
| Item = self.OrderedStringList[FirstLangName][Index] | |
| if Item.UseOtherLangDef != '': | |
| OtherLang = Item.UseOtherLangDef | |
| else: | |
| OtherLang = FirstLangName | |
| self.OrderedStringList[LangName].append (StringDefClassObject(Item.StringName, '', Item.Referenced, Item.Token, OtherLang)) | |
| self.OrderedStringDict[LangName][Item.StringName] = len(self.OrderedStringList[LangName]) - 1 | |
| return True | |
| def OpenUniFile(self, FileName): | |
| # | |
| # Read file | |
| # | |
| try: | |
| UniFile = open(FileName, mode='rb') | |
| FileIn = UniFile.read() | |
| UniFile.close() | |
| except: | |
| EdkLogger.Error("build", FILE_OPEN_FAILURE, ExtraData=File) | |
| # | |
| # Detect Byte Order Mark at beginning of file. Default to UTF-8 | |
| # | |
| Encoding = 'utf-8' | |
| if (FileIn.startswith(codecs.BOM_UTF16_BE) or | |
| FileIn.startswith(codecs.BOM_UTF16_LE)): | |
| Encoding = 'utf-16' | |
| self.VerifyUcs2Data(FileIn, FileName, Encoding) | |
| UniFile = StringIO.StringIO(FileIn) | |
| Info = codecs.lookup(Encoding) | |
| (Reader, Writer) = (Info.streamreader, Info.streamwriter) | |
| return codecs.StreamReaderWriter(UniFile, Reader, Writer) | |
| def VerifyUcs2Data(self, FileIn, FileName, Encoding): | |
| Ucs2Info = codecs.lookup('ucs-2') | |
| # | |
| # Convert to unicode | |
| # | |
| try: | |
| FileDecoded = codecs.decode(FileIn, Encoding) | |
| Ucs2Info.encode(FileDecoded) | |
| except: | |
| UniFile = StringIO.StringIO(FileIn) | |
| Info = codecs.lookup(Encoding) | |
| (Reader, Writer) = (Info.streamreader, Info.streamwriter) | |
| File = codecs.StreamReaderWriter(UniFile, Reader, Writer) | |
| LineNumber = 0 | |
| ErrMsg = lambda Encoding, LineNumber: \ | |
| '%s contains invalid %s characters on line %d.' % \ | |
| (FileName, Encoding, LineNumber) | |
| while True: | |
| LineNumber = LineNumber + 1 | |
| try: | |
| Line = File.readline() | |
| if Line == '': | |
| EdkLogger.error('Unicode File Parser', PARSER_ERROR, | |
| ErrMsg(Encoding, LineNumber)) | |
| Ucs2Info.encode(Line) | |
| except: | |
| EdkLogger.error('Unicode File Parser', PARSER_ERROR, | |
| ErrMsg('UCS-2', LineNumber)) | |
| # | |
| # Get String name and value | |
| # | |
| def GetStringObject(self, Item): | |
| Language = '' | |
| Value = '' | |
| Name = Item.split()[1] | |
| # Check the string name | |
| if Name != '': | |
| MatchString = re.match('^[a-zA-Z][a-zA-Z0-9_]*$', Name, re.UNICODE) | |
| if MatchString == None or MatchString.end(0) != len(Name): | |
| EdkLogger.error('Unicode File Parser', FORMAT_INVALID, 'The string token name %s defined in UNI file %s contains the invalid character.' % (Name, self.File)) | |
| LanguageList = Item.split(u'#language ') | |
| for IndexI in range(len(LanguageList)): | |
| if IndexI == 0: | |
| continue | |
| else: | |
| Language = LanguageList[IndexI].split()[0] | |
| Value = LanguageList[IndexI][LanguageList[IndexI].find(u'\"') + len(u'\"') : LanguageList[IndexI].rfind(u'\"')] #.replace(u'\r\n', u'') | |
| Language = GetLanguageCode(Language, self.IsCompatibleMode, self.File) | |
| self.AddStringToList(Name, Language, Value) | |
| # | |
| # Get include file list and load them | |
| # | |
| def GetIncludeFile(self, Item, Dir): | |
| FileName = Item[Item.find(u'#include ') + len(u'#include ') :Item.find(u' ', len(u'#include '))][1:-1] | |
| self.LoadUniFile(FileName) | |
| def StripComments(self, Line): | |
| Comment = u'//' | |
| CommentPos = Line.find(Comment) | |
| while CommentPos >= 0: | |
| # if there are non matched quotes before the comment header | |
| # then we are in the middle of a string | |
| # but we need to ignore the escaped quotes and backslashes. | |
| if ((Line.count(u'"', 0, CommentPos) - Line.count(u'\\"', 0, CommentPos)) & 1) == 1: | |
| CommentPos = Line.find (Comment, CommentPos + 1) | |
| else: | |
| return Line[:CommentPos].strip() | |
| return Line.strip() | |
| # | |
| # Pre-process before parse .uni file | |
| # | |
| def PreProcess(self, File): | |
| if not os.path.exists(File.Path) or not os.path.isfile(File.Path): | |
| EdkLogger.error("Unicode File Parser", FILE_NOT_FOUND, ExtraData=File.Path) | |
| try: | |
| FileIn = self.OpenUniFile(LongFilePath(File.Path)) | |
| except UnicodeError, X: | |
| EdkLogger.error("build", FILE_READ_FAILURE, "File read failure: %s" % str(X), ExtraData=File.Path); | |
| except: | |
| EdkLogger.error("build", FILE_OPEN_FAILURE, ExtraData=File.Path); | |
| Lines = [] | |
| # | |
| # Use unique identifier | |
| # | |
| for Line in FileIn: | |
| Line = Line.strip() | |
| Line = Line.replace(u'\\\\', BACK_SLASH_PLACEHOLDER) | |
| Line = self.StripComments(Line) | |
| # | |
| # Ignore empty line | |
| # | |
| if len(Line) == 0: | |
| continue | |
| Line = Line.replace(u'/langdef', u'#langdef') | |
| Line = Line.replace(u'/string', u'#string') | |
| Line = Line.replace(u'/language', u'#language') | |
| Line = Line.replace(u'/include', u'#include') | |
| Line = Line.replace(UNICODE_WIDE_CHAR, WIDE_CHAR) | |
| Line = Line.replace(UNICODE_NARROW_CHAR, NARROW_CHAR) | |
| Line = Line.replace(UNICODE_NON_BREAKING_CHAR, NON_BREAKING_CHAR) | |
| Line = Line.replace(u'\\r\\n', CR + LF) | |
| Line = Line.replace(u'\\n', CR + LF) | |
| Line = Line.replace(u'\\r', CR) | |
| Line = Line.replace(u'\\t', u' ') | |
| Line = Line.replace(u'\t', u' ') | |
| Line = Line.replace(u'\\"', u'"') | |
| Line = Line.replace(u"\\'", u"'") | |
| Line = Line.replace(BACK_SLASH_PLACEHOLDER, u'\\') | |
| StartPos = Line.find(u'\\x') | |
| while (StartPos != -1): | |
| EndPos = Line.find(u'\\', StartPos + 1, StartPos + 7) | |
| if EndPos != -1 and EndPos - StartPos == 6 : | |
| if re.match('[a-fA-F0-9]{4}', Line[StartPos + 2 : EndPos], re.UNICODE): | |
| EndStr = Line[EndPos: ] | |
| UniStr = ('\u' + (Line[StartPos + 2 : EndPos])).decode('unicode_escape') | |
| if EndStr.startswith(u'\\x') and len(EndStr) >= 7: | |
| if EndStr[6] == u'\\' and re.match('[a-fA-F0-9]{4}', EndStr[2 : 6], re.UNICODE): | |
| Line = Line[0 : StartPos] + UniStr + EndStr | |
| else: | |
| Line = Line[0 : StartPos] + UniStr + EndStr[1:] | |
| StartPos = Line.find(u'\\x', StartPos + 1) | |
| IncList = gIncludePattern.findall(Line) | |
| if len(IncList) == 1: | |
| for Dir in [File.Dir] + self.IncludePathList: | |
| IncFile = PathClass(str(IncList[0]), Dir) | |
| if os.path.isfile(IncFile.Path): | |
| Lines.extend(self.PreProcess(IncFile)) | |
| break | |
| else: | |
| EdkLogger.error("Unicode File Parser", FILE_NOT_FOUND, Message="Cannot find include file", ExtraData=str(IncList[0])) | |
| continue | |
| Lines.append(Line) | |
| return Lines | |
| # | |
| # Load a .uni file | |
| # | |
| def LoadUniFile(self, File = None): | |
| if File == None: | |
| EdkLogger.error("Unicode File Parser", PARSER_ERROR, 'No unicode file is given') | |
| self.File = File | |
| # | |
| # Process special char in file | |
| # | |
| Lines = self.PreProcess(File) | |
| # | |
| # Get Unicode Information | |
| # | |
| for IndexI in range(len(Lines)): | |
| Line = Lines[IndexI] | |
| if (IndexI + 1) < len(Lines): | |
| SecondLine = Lines[IndexI + 1] | |
| if (IndexI + 2) < len(Lines): | |
| ThirdLine = Lines[IndexI + 2] | |
| # | |
| # Get Language def information | |
| # | |
| if Line.find(u'#langdef ') >= 0: | |
| self.GetLangDef(File, Line) | |
| continue | |
| Name = '' | |
| Language = '' | |
| Value = '' | |
| # | |
| # Get string def information format 1 as below | |
| # | |
| # #string MY_STRING_1 | |
| # #language eng | |
| # My first English string line 1 | |
| # My first English string line 2 | |
| # #string MY_STRING_1 | |
| # #language spa | |
| # Mi segunda secuencia 1 | |
| # Mi segunda secuencia 2 | |
| # | |
| if Line.find(u'#string ') >= 0 and Line.find(u'#language ') < 0 and \ | |
| SecondLine.find(u'#string ') < 0 and SecondLine.find(u'#language ') >= 0 and \ | |
| ThirdLine.find(u'#string ') < 0 and ThirdLine.find(u'#language ') < 0: | |
| Name = Line[Line.find(u'#string ') + len(u'#string ') : ].strip(' ') | |
| Language = SecondLine[SecondLine.find(u'#language ') + len(u'#language ') : ].strip(' ') | |
| for IndexJ in range(IndexI + 2, len(Lines)): | |
| if Lines[IndexJ].find(u'#string ') < 0 and Lines[IndexJ].find(u'#language ') < 0: | |
| Value = Value + Lines[IndexJ] | |
| else: | |
| IndexI = IndexJ | |
| break | |
| # Value = Value.replace(u'\r\n', u'') | |
| Language = GetLanguageCode(Language, self.IsCompatibleMode, self.File) | |
| # Check the string name | |
| if not self.IsCompatibleMode and Name != '': | |
| MatchString = re.match('^[a-zA-Z][a-zA-Z0-9_]*$', Name, re.UNICODE) | |
| if MatchString == None or MatchString.end(0) != len(Name): | |
| EdkLogger.error('Unicode File Parser', FORMAT_INVALID, 'The string token name %s defined in UNI file %s contains the invalid character.' % (Name, self.File)) | |
| self.AddStringToList(Name, Language, Value) | |
| continue | |
| # | |
| # Get string def information format 2 as below | |
| # | |
| # #string MY_STRING_1 #language eng "My first English string line 1" | |
| # "My first English string line 2" | |
| # #language spa "Mi segunda secuencia 1" | |
| # "Mi segunda secuencia 2" | |
| # #string MY_STRING_2 #language eng "My first English string line 1" | |
| # "My first English string line 2" | |
| # #string MY_STRING_2 #language spa "Mi segunda secuencia 1" | |
| # "Mi segunda secuencia 2" | |
| # | |
| if Line.find(u'#string ') >= 0 and Line.find(u'#language ') >= 0: | |
| StringItem = Line | |
| for IndexJ in range(IndexI + 1, len(Lines)): | |
| if Lines[IndexJ].find(u'#string ') >= 0 and Lines[IndexJ].find(u'#language ') >= 0: | |
| IndexI = IndexJ | |
| break | |
| elif Lines[IndexJ].find(u'#string ') < 0 and Lines[IndexJ].find(u'#language ') >= 0: | |
| StringItem = StringItem + Lines[IndexJ] | |
| elif Lines[IndexJ].count(u'\"') >= 2: | |
| StringItem = StringItem[ : StringItem.rfind(u'\"')] + Lines[IndexJ][Lines[IndexJ].find(u'\"') + len(u'\"') : ] | |
| self.GetStringObject(StringItem) | |
| continue | |
| # | |
| # Load multiple .uni files | |
| # | |
| def LoadUniFiles(self, FileList): | |
| if len(FileList) > 0: | |
| for File in FileList: | |
| self.LoadUniFile(File) | |
| # | |
| # Add a string to list | |
| # | |
| def AddStringToList(self, Name, Language, Value, Token = None, Referenced = False, UseOtherLangDef = '', Index = -1): | |
| for LangNameItem in self.LanguageDef: | |
| if Language == LangNameItem[0]: | |
| break | |
| else: | |
| EdkLogger.error('Unicode File Parser', FORMAT_NOT_SUPPORTED, "The language '%s' for %s is not defined in Unicode file %s." \ | |
| % (Language, Name, self.File)) | |
| if Language not in self.OrderedStringList: | |
| self.OrderedStringList[Language] = [] | |
| self.OrderedStringDict[Language] = {} | |
| IsAdded = True | |
| if Name in self.OrderedStringDict[Language]: | |
| IsAdded = False | |
| if Value != None: | |
| ItemIndexInList = self.OrderedStringDict[Language][Name] | |
| Item = self.OrderedStringList[Language][ItemIndexInList] | |
| Item.UpdateValue(Value) | |
| Item.UseOtherLangDef = '' | |
| if IsAdded: | |
| Token = len(self.OrderedStringList[Language]) | |
| if Index == -1: | |
| self.OrderedStringList[Language].append(StringDefClassObject(Name, Value, Referenced, Token, UseOtherLangDef)) | |
| self.OrderedStringDict[Language][Name] = Token | |
| for LangName in self.LanguageDef: | |
| # | |
| # New STRING token will be added into all language string lists. | |
| # so that the unique STRING identifier is reserved for all languages in the package list. | |
| # | |
| if LangName[0] != Language: | |
| if UseOtherLangDef != '': | |
| OtherLangDef = UseOtherLangDef | |
| else: | |
| OtherLangDef = Language | |
| self.OrderedStringList[LangName[0]].append(StringDefClassObject(Name, '', Referenced, Token, OtherLangDef)) | |
| self.OrderedStringDict[LangName[0]][Name] = len(self.OrderedStringList[LangName[0]]) - 1 | |
| else: | |
| self.OrderedStringList[Language].insert(Index, StringDefClassObject(Name, Value, Referenced, Token, UseOtherLangDef)) | |
| self.OrderedStringDict[Language][Name] = Index | |
| # | |
| # Set the string as referenced | |
| # | |
| def SetStringReferenced(self, Name): | |
| # | |
| # String stoken are added in the same order in all language string lists. | |
| # So, only update the status of string stoken in first language string list. | |
| # | |
| Lang = self.LanguageDef[0][0] | |
| if Name in self.OrderedStringDict[Lang]: | |
| ItemIndexInList = self.OrderedStringDict[Lang][Name] | |
| Item = self.OrderedStringList[Lang][ItemIndexInList] | |
| Item.Referenced = True | |
| # | |
| # Search the string in language definition by Name | |
| # | |
| def FindStringValue(self, Name, Lang): | |
| if Name in self.OrderedStringDict[Lang]: | |
| ItemIndexInList = self.OrderedStringDict[Lang][Name] | |
| return self.OrderedStringList[Lang][ItemIndexInList] | |
| return None | |
| # | |
| # Search the string in language definition by Token | |
| # | |
| def FindByToken(self, Token, Lang): | |
| for Item in self.OrderedStringList[Lang]: | |
| if Item.Token == Token: | |
| return Item | |
| return None | |
| # | |
| # Re-order strings and re-generate tokens | |
| # | |
| def ReToken(self): | |
| # | |
| # Retoken all language strings according to the status of string stoken in the first language string. | |
| # | |
| FirstLangName = self.LanguageDef[0][0] | |
| # Convert the OrderedStringList to be OrderedStringListByToken in order to faciliate future search by token | |
| for LangNameItem in self.LanguageDef: | |
| self.OrderedStringListByToken[LangNameItem[0]] = {} | |
| # | |
| # Use small token for all referred string stoken. | |
| # | |
| RefToken = 0 | |
| for Index in range (0, len (self.OrderedStringList[FirstLangName])): | |
| FirstLangItem = self.OrderedStringList[FirstLangName][Index] | |
| if FirstLangItem.Referenced == True: | |
| for LangNameItem in self.LanguageDef: | |
| LangName = LangNameItem[0] | |
| OtherLangItem = self.OrderedStringList[LangName][Index] | |
| OtherLangItem.Referenced = True | |
| OtherLangItem.Token = RefToken | |
| self.OrderedStringListByToken[LangName][OtherLangItem.Token] = OtherLangItem | |
| RefToken = RefToken + 1 | |
| # | |
| # Use big token for all unreferred string stoken. | |
| # | |
| UnRefToken = 0 | |
| for Index in range (0, len (self.OrderedStringList[FirstLangName])): | |
| FirstLangItem = self.OrderedStringList[FirstLangName][Index] | |
| if FirstLangItem.Referenced == False: | |
| for LangNameItem in self.LanguageDef: | |
| LangName = LangNameItem[0] | |
| OtherLangItem = self.OrderedStringList[LangName][Index] | |
| OtherLangItem.Token = RefToken + UnRefToken | |
| self.OrderedStringListByToken[LangName][OtherLangItem.Token] = OtherLangItem | |
| UnRefToken = UnRefToken + 1 | |
| # | |
| # Show the instance itself | |
| # | |
| def ShowMe(self): | |
| print self.LanguageDef | |
| #print self.OrderedStringList | |
| for Item in self.OrderedStringList: | |
| print Item | |
| for Member in self.OrderedStringList[Item]: | |
| print str(Member) | |
| # This acts like the main() function for the script, unless it is 'import'ed into another | |
| # script. | |
| if __name__ == '__main__': | |
| EdkLogger.Initialize() | |
| EdkLogger.SetLevel(EdkLogger.DEBUG_0) | |
| a = UniFileClassObject([PathClass("C:\\Edk\\Strings.uni"), PathClass("C:\\Edk\\Strings2.uni")]) | |
| a.ReToken() | |
| a.ShowMe() |