]> git.proxmox.com Git - mirror_edk2.git/blame - BaseTools/Source/Python/AutoGen/UniClassObject.py
BaseTools: use new shared GUID regular expressions
[mirror_edk2.git] / BaseTools / Source / Python / AutoGen / UniClassObject.py
CommitLineData
97fa0ee9
YL
1## @file\r
2# This file is used to collect all defined strings in multiple uni files\r
3#\r
71f02911
CS
4#\r
5# Copyright (c) 2014 Hewlett-Packard Development Company, L.P.<BR>\r
6#\r
314e2fb1 7# Copyright (c) 2007 - 2016, Intel Corporation. All rights reserved.<BR>\r
40d841f6 8# This program and the accompanying materials\r
30fdf114
LG
9# are licensed and made available under the terms and conditions of the BSD License\r
10# which accompanies this distribution. The full text of the license may be found at\r
11# http://opensource.org/licenses/bsd-license.php\r
12#\r
13# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
14# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
15\r
30fdf114
LG
16##\r
17# Import Modules\r
18#\r
1be2ed90 19import Common.LongFilePathOs as os, codecs, re\r
40d841f6 20import distutils.util\r
30fdf114 21import Common.EdkLogger as EdkLogger\r
d80e451b 22import StringIO\r
30fdf114
LG
23from Common.BuildToolError import *\r
24from Common.String import GetLineNo\r
25from Common.Misc import PathClass\r
1be2ed90 26from Common.LongFilePathSupport import LongFilePath\r
30fdf114
LG
27##\r
28# Static definitions\r
29#\r
30UNICODE_WIDE_CHAR = u'\\wide'\r
31UNICODE_NARROW_CHAR = u'\\narrow'\r
32UNICODE_NON_BREAKING_CHAR = u'\\nbr'\r
33UNICODE_UNICODE_CR = '\r'\r
34UNICODE_UNICODE_LF = '\n'\r
35\r
36NARROW_CHAR = u'\uFFF0'\r
37WIDE_CHAR = u'\uFFF1'\r
38NON_BREAKING_CHAR = u'\uFFF2'\r
39CR = u'\u000D'\r
40LF = u'\u000A'\r
41NULL = u'\u0000'\r
42TAB = u'\t'\r
71f02911 43BACK_SLASH_PLACEHOLDER = u'\u0006'\r
30fdf114
LG
44\r
45gIncludePattern = re.compile("^#include +[\"<]+([^\"< >]+)[>\"]+$", re.MULTILINE | re.UNICODE)\r
46\r
47## Convert a python unicode string to a normal string\r
48#\r
49# Convert a python unicode string to a normal string\r
50# UniToStr(u'I am a string') is 'I am a string'\r
51#\r
52# @param Uni: The python unicode string\r
53#\r
54# @retval: The formatted normal string\r
55#\r
56def UniToStr(Uni):\r
57 return repr(Uni)[2:-1]\r
58\r
59## Convert a unicode string to a Hex list\r
60#\r
61# Convert a unicode string to a Hex list\r
62# UniToHexList('ABC') is ['0x41', '0x00', '0x42', '0x00', '0x43', '0x00']\r
63#\r
64# @param Uni: The python unicode string\r
65#\r
66# @retval List: The formatted hex list\r
67#\r
68def UniToHexList(Uni):\r
69 List = []\r
70 for Item in Uni:\r
71 Temp = '%04X' % ord(Item)\r
72 List.append('0x' + Temp[2:4])\r
73 List.append('0x' + Temp[0:2])\r
74 return List\r
75\r
76LangConvTable = {'eng':'en', 'fra':'fr', \\r
77 'aar':'aa', 'abk':'ab', 'ave':'ae', 'afr':'af', 'aka':'ak', 'amh':'am', \\r
78 'arg':'an', 'ara':'ar', 'asm':'as', 'ava':'av', 'aym':'ay', 'aze':'az', \\r
79 'bak':'ba', 'bel':'be', 'bul':'bg', 'bih':'bh', 'bis':'bi', 'bam':'bm', \\r
80 'ben':'bn', 'bod':'bo', 'bre':'br', 'bos':'bs', 'cat':'ca', 'che':'ce', \\r
81 'cha':'ch', 'cos':'co', 'cre':'cr', 'ces':'cs', 'chu':'cu', 'chv':'cv', \\r
82 'cym':'cy', 'dan':'da', 'deu':'de', 'div':'dv', 'dzo':'dz', 'ewe':'ee', \\r
83 'ell':'el', 'epo':'eo', 'spa':'es', 'est':'et', 'eus':'eu', 'fas':'fa', \\r
84 'ful':'ff', 'fin':'fi', 'fij':'fj', 'fao':'fo', 'fry':'fy', 'gle':'ga', \\r
85 'gla':'gd', 'glg':'gl', 'grn':'gn', 'guj':'gu', 'glv':'gv', 'hau':'ha', \\r
86 'heb':'he', 'hin':'hi', 'hmo':'ho', 'hrv':'hr', 'hat':'ht', 'hun':'hu', \\r
87 'hye':'hy', 'her':'hz', 'ina':'ia', 'ind':'id', 'ile':'ie', 'ibo':'ig', \\r
88 'iii':'ii', 'ipk':'ik', 'ido':'io', 'isl':'is', 'ita':'it', 'iku':'iu', \\r
89 'jpn':'ja', 'jav':'jv', 'kat':'ka', 'kon':'kg', 'kik':'ki', 'kua':'kj', \\r
90 'kaz':'kk', 'kal':'kl', 'khm':'km', 'kan':'kn', 'kor':'ko', 'kau':'kr', \\r
91 'kas':'ks', 'kur':'ku', 'kom':'kv', 'cor':'kw', 'kir':'ky', 'lat':'la', \\r
92 'ltz':'lb', 'lug':'lg', 'lim':'li', 'lin':'ln', 'lao':'lo', 'lit':'lt', \\r
93 'lub':'lu', 'lav':'lv', 'mlg':'mg', 'mah':'mh', 'mri':'mi', 'mkd':'mk', \\r
94 'mal':'ml', 'mon':'mn', 'mar':'mr', 'msa':'ms', 'mlt':'mt', 'mya':'my', \\r
95 'nau':'na', 'nob':'nb', 'nde':'nd', 'nep':'ne', 'ndo':'ng', 'nld':'nl', \\r
96 'nno':'nn', 'nor':'no', 'nbl':'nr', 'nav':'nv', 'nya':'ny', 'oci':'oc', \\r
97 'oji':'oj', 'orm':'om', 'ori':'or', 'oss':'os', 'pan':'pa', 'pli':'pi', \\r
98 'pol':'pl', 'pus':'ps', 'por':'pt', 'que':'qu', 'roh':'rm', 'run':'rn', \\r
99 'ron':'ro', 'rus':'ru', 'kin':'rw', 'san':'sa', 'srd':'sc', 'snd':'sd', \\r
100 'sme':'se', 'sag':'sg', 'sin':'si', 'slk':'sk', 'slv':'sl', 'smo':'sm', \\r
101 'sna':'sn', 'som':'so', 'sqi':'sq', 'srp':'sr', 'ssw':'ss', 'sot':'st', \\r
102 'sun':'su', 'swe':'sv', 'swa':'sw', 'tam':'ta', 'tel':'te', 'tgk':'tg', \\r
103 'tha':'th', 'tir':'ti', 'tuk':'tk', 'tgl':'tl', 'tsn':'tn', 'ton':'to', \\r
104 'tur':'tr', 'tso':'ts', 'tat':'tt', 'twi':'tw', 'tah':'ty', 'uig':'ug', \\r
105 'ukr':'uk', 'urd':'ur', 'uzb':'uz', 'ven':'ve', 'vie':'vi', 'vol':'vo', \\r
106 'wln':'wa', 'wol':'wo', 'xho':'xh', 'yid':'yi', 'yor':'yo', 'zha':'za', \\r
107 'zho':'zh', 'zul':'zu'}\r
108\r
109## GetLanguageCode\r
110#\r
111# Check the language code read from .UNI file and convert ISO 639-2 codes to RFC 4646 codes if appropriate\r
112# ISO 639-2 language codes supported in compatiblity mode\r
113# RFC 4646 language codes supported in native mode\r
114#\r
115# @param LangName: Language codes read from .UNI file\r
116#\r
117# @retval LangName: Valid lanugage code in RFC 4646 format or None\r
118#\r
119def GetLanguageCode(LangName, IsCompatibleMode, File):\r
120 global LangConvTable\r
121\r
122 length = len(LangName)\r
123 if IsCompatibleMode:\r
124 if length == 3 and LangName.isalpha():\r
125 TempLangName = LangConvTable.get(LangName.lower())\r
126 if TempLangName != None:\r
da92f276 127 return TempLangName\r
30fdf114
LG
128 return LangName\r
129 else:\r
130 EdkLogger.error("Unicode File Parser", FORMAT_INVALID, "Invalid ISO 639-2 language code : %s" % LangName, File)\r
131\r
40d841f6
LG
132 if (LangName[0] == 'X' or LangName[0] == 'x') and LangName[1] == '-':\r
133 return LangName\r
30fdf114
LG
134 if length == 2:\r
135 if LangName.isalpha():\r
136 return LangName\r
137 elif length == 3:\r
138 if LangName.isalpha() and LangConvTable.get(LangName.lower()) == None:\r
139 return LangName\r
140 elif length == 5:\r
141 if LangName[0:2].isalpha() and LangName[2] == '-':\r
142 return LangName\r
143 elif length >= 6:\r
144 if LangName[0:2].isalpha() and LangName[2] == '-':\r
145 return LangName\r
146 if LangName[0:3].isalpha() and LangConvTable.get(LangName.lower()) == None and LangName[3] == '-':\r
147 return LangName\r
148\r
149 EdkLogger.error("Unicode File Parser", FORMAT_INVALID, "Invalid RFC 4646 language code : %s" % LangName, File)\r
150\r
d80e451b
JJ
151## Ucs2Codec\r
152#\r
153# This is only a partial codec implementation. It only supports\r
154# encoding, and is primarily used to check that all the characters are\r
155# valid for UCS-2.\r
156#\r
157class Ucs2Codec(codecs.Codec):\r
158 def __init__(self):\r
159 self.__utf16 = codecs.lookup('utf-16')\r
160\r
161 def encode(self, input, errors='strict'):\r
162 for Char in input:\r
163 CodePoint = ord(Char)\r
164 if CodePoint >= 0xd800 and CodePoint <= 0xdfff:\r
165 raise ValueError("Code Point is in range reserved for " +\r
166 "UTF-16 surrogate pairs")\r
167 elif CodePoint > 0xffff:\r
168 raise ValueError("Code Point too large to encode in UCS-2")\r
169 return self.__utf16.encode(input)\r
170\r
171TheUcs2Codec = Ucs2Codec()\r
172def Ucs2Search(name):\r
173 if name == 'ucs-2':\r
174 return codecs.CodecInfo(\r
175 name=name,\r
176 encode=TheUcs2Codec.encode,\r
177 decode=TheUcs2Codec.decode)\r
178 else:\r
179 return None\r
180codecs.register(Ucs2Search)\r
181\r
30fdf114
LG
182## StringDefClassObject\r
183#\r
184# A structure for language definition\r
185#\r
186class StringDefClassObject(object):\r
187 def __init__(self, Name = None, Value = None, Referenced = False, Token = None, UseOtherLangDef = ''):\r
188 self.StringName = ''\r
189 self.StringNameByteList = []\r
190 self.StringValue = ''\r
191 self.StringValueByteList = ''\r
192 self.Token = 0\r
193 self.Referenced = Referenced\r
194 self.UseOtherLangDef = UseOtherLangDef\r
195 self.Length = 0\r
196\r
197 if Name != None:\r
198 self.StringName = Name\r
199 self.StringNameByteList = UniToHexList(Name)\r
200 if Value != None:\r
201 self.StringValue = Value + u'\x00' # Add a NULL at string tail\r
202 self.StringValueByteList = UniToHexList(self.StringValue)\r
203 self.Length = len(self.StringValueByteList)\r
204 if Token != None:\r
205 self.Token = Token\r
206\r
207 def __str__(self):\r
208 return repr(self.StringName) + ' ' + \\r
209 repr(self.Token) + ' ' + \\r
210 repr(self.Referenced) + ' ' + \\r
211 repr(self.StringValue) + ' ' + \\r
212 repr(self.UseOtherLangDef)\r
213\r
756ad8f8
LG
214 def UpdateValue(self, Value = None):\r
215 if Value != None:\r
216 self.StringValue = Value + u'\x00' # Add a NULL at string tail\r
217 self.StringValueByteList = UniToHexList(self.StringValue)\r
218 self.Length = len(self.StringValueByteList)\r
219\r
30fdf114
LG
220## UniFileClassObject\r
221#\r
222# A structure for .uni file definition\r
223#\r
224class UniFileClassObject(object):\r
756ad8f8 225 def __init__(self, FileList = [], IsCompatibleMode = False, IncludePathList = []):\r
30fdf114
LG
226 self.FileList = FileList\r
227 self.Token = 2\r
228 self.LanguageDef = [] #[ [u'LanguageIdentifier', u'PrintableName'], ... ]\r
229 self.OrderedStringList = {} #{ u'LanguageIdentifier' : [StringDefClassObject] }\r
79b74a03
LG
230 self.OrderedStringDict = {} #{ u'LanguageIdentifier' : {StringName:(IndexInList)} }\r
231 self.OrderedStringListByToken = {} #{ u'LanguageIdentifier' : {Token: StringDefClassObject} }\r
30fdf114 232 self.IsCompatibleMode = IsCompatibleMode\r
756ad8f8 233 self.IncludePathList = IncludePathList\r
30fdf114
LG
234 if len(self.FileList) > 0:\r
235 self.LoadUniFiles(FileList)\r
236\r
237 #\r
238 # Get Language definition\r
239 #\r
240 def GetLangDef(self, File, Line):\r
40d841f6 241 Lang = distutils.util.split_quoted((Line.split(u"//")[0]))\r
30fdf114
LG
242 if len(Lang) != 3:\r
243 try:\r
d80e451b 244 FileIn = self.OpenUniFile(LongFilePath(File.Path))\r
30fdf114
LG
245 except UnicodeError, X:\r
246 EdkLogger.error("build", FILE_READ_FAILURE, "File read failure: %s" % str(X), ExtraData=File);\r
247 except:\r
248 EdkLogger.error("build", FILE_OPEN_FAILURE, ExtraData=File);\r
249 LineNo = GetLineNo(FileIn, Line, False)\r
250 EdkLogger.error("Unicode File Parser", PARSER_ERROR, "Wrong language definition",\r
47fea6af 251 ExtraData="""%s\n\t*Correct format is like '#langdef en-US "English"'""" % Line, File=File, Line=LineNo)\r
30fdf114
LG
252 else:\r
253 LangName = GetLanguageCode(Lang[1], self.IsCompatibleMode, self.File)\r
40d841f6 254 LangPrintName = Lang[2]\r
30fdf114
LG
255\r
256 IsLangInDef = False\r
257 for Item in self.LanguageDef:\r
258 if Item[0] == LangName:\r
259 IsLangInDef = True\r
260 break;\r
261\r
262 if not IsLangInDef:\r
263 self.LanguageDef.append([LangName, LangPrintName])\r
264\r
265 #\r
266 # Add language string\r
267 #\r
268 self.AddStringToList(u'$LANGUAGE_NAME', LangName, LangName, 0, True, Index=0)\r
269 self.AddStringToList(u'$PRINTABLE_LANGUAGE_NAME', LangName, LangPrintName, 1, True, Index=1)\r
270\r
756ad8f8
LG
271 if not IsLangInDef:\r
272 #\r
273 # The found STRING tokens will be added into new language string list\r
274 # so that the unique STRING identifier is reserved for all languages in the package list. \r
275 #\r
276 FirstLangName = self.LanguageDef[0][0]\r
277 if LangName != FirstLangName:\r
278 for Index in range (2, len (self.OrderedStringList[FirstLangName])):\r
279 Item = self.OrderedStringList[FirstLangName][Index]\r
280 if Item.UseOtherLangDef != '':\r
281 OtherLang = Item.UseOtherLangDef\r
282 else:\r
283 OtherLang = FirstLangName\r
284 self.OrderedStringList[LangName].append (StringDefClassObject(Item.StringName, '', Item.Referenced, Item.Token, OtherLang))\r
79b74a03 285 self.OrderedStringDict[LangName][Item.StringName] = len(self.OrderedStringList[LangName]) - 1\r
30fdf114
LG
286 return True\r
287\r
d80e451b
JJ
288 def OpenUniFile(self, FileName):\r
289 #\r
290 # Read file\r
291 #\r
292 try:\r
293 UniFile = open(FileName, mode='rb')\r
294 FileIn = UniFile.read()\r
295 UniFile.close()\r
296 except:\r
297 EdkLogger.Error("build", FILE_OPEN_FAILURE, ExtraData=File)\r
298\r
299 #\r
be264422 300 # Detect Byte Order Mark at beginning of file. Default to UTF-8\r
d80e451b 301 #\r
be264422
JJ
302 Encoding = 'utf-8'\r
303 if (FileIn.startswith(codecs.BOM_UTF16_BE) or\r
304 FileIn.startswith(codecs.BOM_UTF16_LE)):\r
305 Encoding = 'utf-16'\r
d80e451b
JJ
306\r
307 self.VerifyUcs2Data(FileIn, FileName, Encoding)\r
308\r
309 UniFile = StringIO.StringIO(FileIn)\r
310 Info = codecs.lookup(Encoding)\r
311 (Reader, Writer) = (Info.streamreader, Info.streamwriter)\r
312 return codecs.StreamReaderWriter(UniFile, Reader, Writer)\r
313\r
314 def VerifyUcs2Data(self, FileIn, FileName, Encoding):\r
315 Ucs2Info = codecs.lookup('ucs-2')\r
316 #\r
317 # Convert to unicode\r
318 #\r
319 try:\r
320 FileDecoded = codecs.decode(FileIn, Encoding)\r
321 Ucs2Info.encode(FileDecoded)\r
322 except:\r
323 UniFile = StringIO.StringIO(FileIn)\r
324 Info = codecs.lookup(Encoding)\r
325 (Reader, Writer) = (Info.streamreader, Info.streamwriter)\r
326 File = codecs.StreamReaderWriter(UniFile, Reader, Writer)\r
327 LineNumber = 0\r
328 ErrMsg = lambda Encoding, LineNumber: \\r
329 '%s contains invalid %s characters on line %d.' % \\r
330 (FileName, Encoding, LineNumber)\r
331 while True:\r
332 LineNumber = LineNumber + 1\r
333 try:\r
334 Line = File.readline()\r
335 if Line == '':\r
336 EdkLogger.error('Unicode File Parser', PARSER_ERROR,\r
337 ErrMsg(Encoding, LineNumber))\r
338 Ucs2Info.encode(Line)\r
339 except:\r
340 EdkLogger.error('Unicode File Parser', PARSER_ERROR,\r
341 ErrMsg('UCS-2', LineNumber))\r
342\r
30fdf114
LG
343 #\r
344 # Get String name and value\r
345 #\r
346 def GetStringObject(self, Item):\r
30fdf114
LG
347 Language = ''\r
348 Value = ''\r
349\r
350 Name = Item.split()[1]\r
c3915fa5 351 # Check the string name\r
2bcc713e 352 if Name != '':\r
c3915fa5 353 MatchString = re.match('^[a-zA-Z][a-zA-Z0-9_]*$', Name, re.UNICODE)\r
b303ea72 354 if MatchString == None or MatchString.end(0) != len(Name):\r
c3915fa5 355 EdkLogger.error('Unicode File Parser', FORMAT_INVALID, 'The string token name %s defined in UNI file %s contains the invalid character.' % (Name, self.File))\r
30fdf114
LG
356 LanguageList = Item.split(u'#language ')\r
357 for IndexI in range(len(LanguageList)):\r
358 if IndexI == 0:\r
359 continue\r
360 else:\r
361 Language = LanguageList[IndexI].split()[0]\r
362 Value = LanguageList[IndexI][LanguageList[IndexI].find(u'\"') + len(u'\"') : LanguageList[IndexI].rfind(u'\"')] #.replace(u'\r\n', u'')\r
363 Language = GetLanguageCode(Language, self.IsCompatibleMode, self.File)\r
364 self.AddStringToList(Name, Language, Value)\r
365\r
366 #\r
367 # Get include file list and load them\r
368 #\r
369 def GetIncludeFile(self, Item, Dir):\r
370 FileName = Item[Item.find(u'#include ') + len(u'#include ') :Item.find(u' ', len(u'#include '))][1:-1]\r
371 self.LoadUniFile(FileName)\r
372\r
71f02911
CS
373 def StripComments(self, Line):\r
374 Comment = u'//'\r
375 CommentPos = Line.find(Comment)\r
376 while CommentPos >= 0:\r
377 # if there are non matched quotes before the comment header\r
378 # then we are in the middle of a string\r
379 # but we need to ignore the escaped quotes and backslashes.\r
380 if ((Line.count(u'"', 0, CommentPos) - Line.count(u'\\"', 0, CommentPos)) & 1) == 1:\r
381 CommentPos = Line.find (Comment, CommentPos + 1)\r
382 else:\r
45265a86
YL
383 return Line[:CommentPos].strip()\r
384 return Line.strip()\r
71f02911
CS
385 \r
386\r
30fdf114
LG
387 #\r
388 # Pre-process before parse .uni file\r
389 #\r
390 def PreProcess(self, File):\r
391 if not os.path.exists(File.Path) or not os.path.isfile(File.Path):\r
392 EdkLogger.error("Unicode File Parser", FILE_NOT_FOUND, ExtraData=File.Path)\r
393\r
30fdf114 394 try:\r
d80e451b 395 FileIn = self.OpenUniFile(LongFilePath(File.Path))\r
30fdf114
LG
396 except UnicodeError, X:\r
397 EdkLogger.error("build", FILE_READ_FAILURE, "File read failure: %s" % str(X), ExtraData=File.Path);\r
398 except:\r
399 EdkLogger.error("build", FILE_OPEN_FAILURE, ExtraData=File.Path);\r
400\r
401 Lines = []\r
402 #\r
403 # Use unique identifier\r
404 #\r
405 for Line in FileIn:\r
406 Line = Line.strip()\r
71f02911
CS
407 Line = Line.replace(u'\\\\', BACK_SLASH_PLACEHOLDER)\r
408 Line = self.StripComments(Line)\r
409\r
30fdf114 410 #\r
71f02911 411 # Ignore empty line\r
30fdf114 412 #\r
71f02911
CS
413 if len(Line) == 0: \r
414 continue \r
da92f276 415 \r
71f02911 416 \r
30fdf114
LG
417 Line = Line.replace(u'/langdef', u'#langdef')\r
418 Line = Line.replace(u'/string', u'#string')\r
419 Line = Line.replace(u'/language', u'#language')\r
420 Line = Line.replace(u'/include', u'#include')\r
421\r
422 Line = Line.replace(UNICODE_WIDE_CHAR, WIDE_CHAR)\r
423 Line = Line.replace(UNICODE_NARROW_CHAR, NARROW_CHAR)\r
424 Line = Line.replace(UNICODE_NON_BREAKING_CHAR, NON_BREAKING_CHAR)\r
425\r
30fdf114
LG
426 Line = Line.replace(u'\\r\\n', CR + LF)\r
427 Line = Line.replace(u'\\n', CR + LF)\r
428 Line = Line.replace(u'\\r', CR)\r
71f02911 429 Line = Line.replace(u'\\t', u' ')\r
8546dfea 430 Line = Line.replace(u'\t', u' ')\r
71f02911
CS
431 Line = Line.replace(u'\\"', u'"') \r
432 Line = Line.replace(u"\\'", u"'") \r
433 Line = Line.replace(BACK_SLASH_PLACEHOLDER, u'\\')\r
4afd3d04 434\r
314e2fb1
YZ
435 StartPos = Line.find(u'\\x')\r
436 while (StartPos != -1):\r
437 EndPos = Line.find(u'\\', StartPos + 1, StartPos + 7)\r
438 if EndPos != -1 and EndPos - StartPos == 6 :\r
439 if re.match('[a-fA-F0-9]{4}', Line[StartPos + 2 : EndPos], re.UNICODE):\r
440 EndStr = Line[EndPos: ]\r
441 UniStr = ('\u' + (Line[StartPos + 2 : EndPos])).decode('unicode_escape')\r
442 if EndStr.startswith(u'\\x') and len(EndStr) >= 7:\r
443 if EndStr[6] == u'\\' and re.match('[a-fA-F0-9]{4}', EndStr[2 : 6], re.UNICODE):\r
444 Line = Line[0 : StartPos] + UniStr + EndStr\r
445 else:\r
446 Line = Line[0 : StartPos] + UniStr + EndStr[1:]\r
ad319b93 447 StartPos = Line.find(u'\\x', StartPos + 1)\r
30fdf114
LG
448\r
449 IncList = gIncludePattern.findall(Line)\r
450 if len(IncList) == 1:\r
756ad8f8
LG
451 for Dir in [File.Dir] + self.IncludePathList:\r
452 IncFile = PathClass(str(IncList[0]), Dir)\r
453 if os.path.isfile(IncFile.Path):\r
454 Lines.extend(self.PreProcess(IncFile))\r
455 break\r
456 else:\r
457 EdkLogger.error("Unicode File Parser", FILE_NOT_FOUND, Message="Cannot find include file", ExtraData=str(IncList[0]))\r
30fdf114
LG
458 continue\r
459\r
460 Lines.append(Line)\r
461\r
462 return Lines\r
463\r
464 #\r
465 # Load a .uni file\r
466 #\r
467 def LoadUniFile(self, File = None):\r
468 if File == None:\r
469 EdkLogger.error("Unicode File Parser", PARSER_ERROR, 'No unicode file is given')\r
470 self.File = File\r
471 #\r
472 # Process special char in file\r
473 #\r
474 Lines = self.PreProcess(File)\r
475\r
476 #\r
477 # Get Unicode Information\r
478 #\r
479 for IndexI in range(len(Lines)):\r
480 Line = Lines[IndexI]\r
481 if (IndexI + 1) < len(Lines):\r
482 SecondLine = Lines[IndexI + 1]\r
483 if (IndexI + 2) < len(Lines):\r
484 ThirdLine = Lines[IndexI + 2]\r
485\r
486 #\r
487 # Get Language def information\r
488 #\r
489 if Line.find(u'#langdef ') >= 0:\r
490 self.GetLangDef(File, Line)\r
491 continue\r
492\r
493 Name = ''\r
494 Language = ''\r
495 Value = ''\r
496 #\r
497 # Get string def information format 1 as below\r
498 #\r
499 # #string MY_STRING_1\r
500 # #language eng\r
501 # My first English string line 1\r
502 # My first English string line 2\r
503 # #string MY_STRING_1\r
504 # #language spa\r
505 # Mi segunda secuencia 1\r
506 # Mi segunda secuencia 2\r
507 #\r
508 if Line.find(u'#string ') >= 0 and Line.find(u'#language ') < 0 and \\r
509 SecondLine.find(u'#string ') < 0 and SecondLine.find(u'#language ') >= 0 and \\r
510 ThirdLine.find(u'#string ') < 0 and ThirdLine.find(u'#language ') < 0:\r
511 Name = Line[Line.find(u'#string ') + len(u'#string ') : ].strip(' ')\r
512 Language = SecondLine[SecondLine.find(u'#language ') + len(u'#language ') : ].strip(' ')\r
513 for IndexJ in range(IndexI + 2, len(Lines)):\r
514 if Lines[IndexJ].find(u'#string ') < 0 and Lines[IndexJ].find(u'#language ') < 0:\r
515 Value = Value + Lines[IndexJ]\r
516 else:\r
517 IndexI = IndexJ\r
518 break\r
519 # Value = Value.replace(u'\r\n', u'')\r
520 Language = GetLanguageCode(Language, self.IsCompatibleMode, self.File)\r
c3915fa5 521 # Check the string name\r
b303ea72 522 if not self.IsCompatibleMode and Name != '':\r
c3915fa5 523 MatchString = re.match('^[a-zA-Z][a-zA-Z0-9_]*$', Name, re.UNICODE)\r
b303ea72 524 if MatchString == None or MatchString.end(0) != len(Name):\r
c3915fa5 525 EdkLogger.error('Unicode File Parser', FORMAT_INVALID, 'The string token name %s defined in UNI file %s contains the invalid character.' % (Name, self.File))\r
30fdf114
LG
526 self.AddStringToList(Name, Language, Value)\r
527 continue\r
528\r
529 #\r
530 # Get string def information format 2 as below\r
531 #\r
532 # #string MY_STRING_1 #language eng "My first English string line 1"\r
533 # "My first English string line 2"\r
534 # #language spa "Mi segunda secuencia 1"\r
535 # "Mi segunda secuencia 2"\r
536 # #string MY_STRING_2 #language eng "My first English string line 1"\r
537 # "My first English string line 2"\r
538 # #string MY_STRING_2 #language spa "Mi segunda secuencia 1"\r
539 # "Mi segunda secuencia 2"\r
540 #\r
541 if Line.find(u'#string ') >= 0 and Line.find(u'#language ') >= 0:\r
542 StringItem = Line\r
543 for IndexJ in range(IndexI + 1, len(Lines)):\r
544 if Lines[IndexJ].find(u'#string ') >= 0 and Lines[IndexJ].find(u'#language ') >= 0:\r
545 IndexI = IndexJ\r
546 break\r
547 elif Lines[IndexJ].find(u'#string ') < 0 and Lines[IndexJ].find(u'#language ') >= 0:\r
548 StringItem = StringItem + Lines[IndexJ]\r
549 elif Lines[IndexJ].count(u'\"') >= 2:\r
550 StringItem = StringItem[ : StringItem.rfind(u'\"')] + Lines[IndexJ][Lines[IndexJ].find(u'\"') + len(u'\"') : ]\r
551 self.GetStringObject(StringItem)\r
552 continue\r
553\r
554 #\r
555 # Load multiple .uni files\r
556 #\r
52302d4d 557 def LoadUniFiles(self, FileList):\r
30fdf114 558 if len(FileList) > 0:\r
52302d4d
LG
559 for File in FileList:\r
560 self.LoadUniFile(File)\r
30fdf114
LG
561\r
562 #\r
563 # Add a string to list\r
564 #\r
565 def AddStringToList(self, Name, Language, Value, Token = None, Referenced = False, UseOtherLangDef = '', Index = -1):\r
08dd311f
LG
566 for LangNameItem in self.LanguageDef:\r
567 if Language == LangNameItem[0]:\r
568 break\r
569 else:\r
570 EdkLogger.error('Unicode File Parser', FORMAT_NOT_SUPPORTED, "The language '%s' for %s is not defined in Unicode file %s." \\r
571 % (Language, Name, self.File))\r
572 \r
30fdf114
LG
573 if Language not in self.OrderedStringList:\r
574 self.OrderedStringList[Language] = []\r
79b74a03 575 self.OrderedStringDict[Language] = {}\r
30fdf114 576\r
756ad8f8 577 IsAdded = True\r
79b74a03
LG
578 if Name in self.OrderedStringDict[Language]:\r
579 IsAdded = False\r
580 if Value != None:\r
581 ItemIndexInList = self.OrderedStringDict[Language][Name]\r
582 Item = self.OrderedStringList[Language][ItemIndexInList]\r
583 Item.UpdateValue(Value)\r
47fea6af 584 Item.UseOtherLangDef = ''\r
756ad8f8
LG
585\r
586 if IsAdded:\r
30fdf114
LG
587 Token = len(self.OrderedStringList[Language])\r
588 if Index == -1:\r
589 self.OrderedStringList[Language].append(StringDefClassObject(Name, Value, Referenced, Token, UseOtherLangDef))\r
79b74a03 590 self.OrderedStringDict[Language][Name] = Token\r
756ad8f8
LG
591 for LangName in self.LanguageDef:\r
592 #\r
593 # New STRING token will be added into all language string lists.\r
594 # so that the unique STRING identifier is reserved for all languages in the package list. \r
595 #\r
596 if LangName[0] != Language:\r
597 if UseOtherLangDef != '':\r
598 OtherLangDef = UseOtherLangDef\r
599 else:\r
600 OtherLangDef = Language\r
601 self.OrderedStringList[LangName[0]].append(StringDefClassObject(Name, '', Referenced, Token, OtherLangDef))\r
79b74a03 602 self.OrderedStringDict[LangName[0]][Name] = len(self.OrderedStringList[LangName[0]]) - 1\r
30fdf114
LG
603 else:\r
604 self.OrderedStringList[Language].insert(Index, StringDefClassObject(Name, Value, Referenced, Token, UseOtherLangDef))\r
79b74a03 605 self.OrderedStringDict[Language][Name] = Index\r
30fdf114
LG
606\r
607 #\r
608 # Set the string as referenced\r
609 #\r
610 def SetStringReferenced(self, Name):\r
756ad8f8
LG
611 #\r
612 # String stoken are added in the same order in all language string lists.\r
613 # So, only update the status of string stoken in first language string list.\r
614 #\r
615 Lang = self.LanguageDef[0][0]\r
79b74a03
LG
616 if Name in self.OrderedStringDict[Lang]:\r
617 ItemIndexInList = self.OrderedStringDict[Lang][Name]\r
618 Item = self.OrderedStringList[Lang][ItemIndexInList]\r
619 Item.Referenced = True\r
620\r
30fdf114
LG
621 #\r
622 # Search the string in language definition by Name\r
623 #\r
624 def FindStringValue(self, Name, Lang):\r
79b74a03
LG
625 if Name in self.OrderedStringDict[Lang]:\r
626 ItemIndexInList = self.OrderedStringDict[Lang][Name]\r
627 return self.OrderedStringList[Lang][ItemIndexInList]\r
30fdf114
LG
628\r
629 return None\r
630\r
631 #\r
632 # Search the string in language definition by Token\r
633 #\r
634 def FindByToken(self, Token, Lang):\r
635 for Item in self.OrderedStringList[Lang]:\r
636 if Item.Token == Token:\r
637 return Item\r
638\r
639 return None\r
640\r
641 #\r
642 # Re-order strings and re-generate tokens\r
643 #\r
644 def ReToken(self):\r
645 #\r
756ad8f8 646 # Retoken all language strings according to the status of string stoken in the first language string.\r
30fdf114 647 #\r
756ad8f8 648 FirstLangName = self.LanguageDef[0][0]\r
52302d4d 649\r
79b74a03
LG
650 # Convert the OrderedStringList to be OrderedStringListByToken in order to faciliate future search by token\r
651 for LangNameItem in self.LanguageDef:\r
652 self.OrderedStringListByToken[LangNameItem[0]] = {}\r
653\r
52302d4d 654 #\r
756ad8f8 655 # Use small token for all referred string stoken.\r
52302d4d 656 #\r
756ad8f8
LG
657 RefToken = 0\r
658 for Index in range (0, len (self.OrderedStringList[FirstLangName])):\r
659 FirstLangItem = self.OrderedStringList[FirstLangName][Index]\r
660 if FirstLangItem.Referenced == True:\r
661 for LangNameItem in self.LanguageDef:\r
662 LangName = LangNameItem[0]\r
663 OtherLangItem = self.OrderedStringList[LangName][Index]\r
664 OtherLangItem.Referenced = True\r
665 OtherLangItem.Token = RefToken\r
79b74a03 666 self.OrderedStringListByToken[LangName][OtherLangItem.Token] = OtherLangItem\r
756ad8f8 667 RefToken = RefToken + 1\r
30fdf114
LG
668\r
669 #\r
756ad8f8 670 # Use big token for all unreferred string stoken.\r
30fdf114 671 #\r
756ad8f8
LG
672 UnRefToken = 0\r
673 for Index in range (0, len (self.OrderedStringList[FirstLangName])):\r
674 FirstLangItem = self.OrderedStringList[FirstLangName][Index]\r
675 if FirstLangItem.Referenced == False:\r
676 for LangNameItem in self.LanguageDef:\r
677 LangName = LangNameItem[0]\r
678 OtherLangItem = self.OrderedStringList[LangName][Index]\r
679 OtherLangItem.Token = RefToken + UnRefToken\r
79b74a03 680 self.OrderedStringListByToken[LangName][OtherLangItem.Token] = OtherLangItem\r
756ad8f8 681 UnRefToken = UnRefToken + 1\r
30fdf114
LG
682\r
683 #\r
684 # Show the instance itself\r
685 #\r
686 def ShowMe(self):\r
687 print self.LanguageDef\r
688 #print self.OrderedStringList\r
689 for Item in self.OrderedStringList:\r
690 print Item\r
691 for Member in self.OrderedStringList[Item]:\r
692 print str(Member)\r
693\r
694# This acts like the main() function for the script, unless it is 'import'ed into another\r
695# script.\r
696if __name__ == '__main__':\r
697 EdkLogger.Initialize()\r
698 EdkLogger.SetLevel(EdkLogger.DEBUG_0)\r
da92f276 699 a = UniFileClassObject([PathClass("C:\\Edk\\Strings.uni"), PathClass("C:\\Edk\\Strings2.uni")])\r
30fdf114
LG
700 a.ReToken()\r
701 a.ShowMe()\r