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