Remove debug print
[bertos.git] / wizard / bertos_utils.py
1 #!/usr/bin/env python
2 # encoding: utf-8
3 #
4 # This file is part of BeRTOS.
5 #
6 # Bertos is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
10 #
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 # GNU General Public License for more details.
15 #
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19 #
20 # As a special exception, you may use this file as part of a free software
21 # library without restriction.  Specifically, if other files instantiate
22 # templates or use macros or inline functions from this file, or you compile
23 # this file and link it with other files to produce an executable, this
24 # file does not by itself cause the resulting executable to be covered by
25 # the GNU General Public License.  This exception does not however
26 # invalidate any other reasons why the executable file might be covered by
27 # the GNU General Public License.
28 #
29 # Copyright 2008 Develer S.r.l. (http://www.develer.com/)
30 #
31 # $Id$
32 #
33 # Author: Lorenzo Berni <duplo@develer.com>
34 #
35
36 import os
37 import fnmatch
38 import glob
39 import re
40 import shutil
41 # Use custom copytree function
42 import copytree
43 import pickle
44
45 import const
46 import plugins
47 import DefineException
48 import BProject
49
50 def isBertosDir(directory):
51    return os.path.exists(directory + "/VERSION")
52
53 def bertosVersion(directory):
54    return open(directory + "/VERSION").readline().strip()
55
56 def loadBertosProject(project_file):
57     project_data = pickle.loads(open(project_file, "r").read())
58     project_info = BProject.BProject()
59     project_info.setInfo("PROJECT_PATH", os.path.dirname(project_file))
60     project_info.setInfo("SOURCES_PATH", project_data["SOURCES_PATH"])
61     project_info.setInfo("TOOLCHAIN", project_data["TOOLCHAIN"])
62     project_info.setInfo("SELECTED_FREQ", project_data["SELECTED_FREQ"])
63     project_info.setInfo("OUTPUT", project_data["OUTPUT"])
64     loadSourceTree(project_info)
65     cpu_name = project_data["CPU_NAME"]
66     project_info.setInfo("CPU_NAME", cpu_name)
67     cpu_info = loadCpuInfos(project_info)
68     for cpu in cpu_info:
69         if cpu["CPU_NAME"] == cpu_name:
70             project_info.setInfo("CPU_INFOS", cpu)
71             break
72     tag_list = getTagSet(cpu_info)
73     # Create, fill and store the dict with the tags
74     tag_dict = {}
75     for element in tag_list:
76         tag_dict[element] = False
77     infos = project_info.info("CPU_INFOS")
78     for tag in tag_dict:
79         if tag in infos["CPU_TAGS"] + [infos["CPU_NAME"], infos["TOOLCHAIN"]]:
80             tag_dict[tag] = True
81         else:
82             tag_dict[tag] = False
83     project_info.setInfo("ALL_CPU_TAGS", tag_dict)
84     loadModuleData(project_info, True)
85     modules = project_info.info("MODULES")
86     for module, information in modules.items():
87         information["enabled"] = module in project_data["ENABLED_MODULES"]
88     project_info.setInfo("MODULES", modules)
89     return project_info
90
91 def mergeSources(srcdir, new_sources, old_sources):
92     # TODO: implement the three way merge algorithm
93     pass
94
95 def projectFileGenerator(project_info):
96     directory = project_info.info("PROJECT_PATH")
97     project_data = {}
98     enabled_modules = []
99     for module, information in project_info.info("MODULES").items():
100         if information["enabled"]:
101             enabled_modules.append(module)
102     project_data["ENABLED_MODULES"] = enabled_modules
103     project_data["SOURCES_PATH"] = project_info.info("SOURCES_PATH")
104     project_data["TOOLCHAIN"] = project_info.info("TOOLCHAIN")
105     project_data["CPU_NAME"] = project_info.info("CPU_NAME")
106     project_data["SELECTED_FREQ"] = project_info.info("SELECTED_FREQ")
107     project_data["OUTPUT"] = project_info.info("OUTPUT")
108     return pickle.dumps(project_data)
109
110 def createBertosProject(project_info, edit=False):
111     directory = project_info.info("PROJECT_PATH")
112     sources_dir = project_info.info("SOURCES_PATH")
113     if not edit:
114         if os.path.isdir(directory):
115             shutil.rmtree(directory, True)        
116         os.makedirs(directory)
117     # Write the project file
118     f = open(directory + "/project.bertos", "w")
119     f.write(projectFileGenerator(project_info))
120     f.close()
121     # Destination source dir
122     srcdir = directory + "/bertos"
123     if not edit:
124         # If not in editing mode it copies all the bertos sources in the /bertos subdirectory of the project
125         shutil.rmtree(srcdir, True)
126         copytree.copytree(sources_dir + "/bertos", srcdir, ignore_list=const.IGNORE_LIST)
127     else:
128         # If in editing mode it merges the current bertos sources with the selected ones
129         # TODO: implement the three way merge algotihm
130         #
131         # mergeSources(srcdir, sources_dir, old_sources_dir)
132         #
133         pass
134     # Destination makefile
135     makefile = directory + "/Makefile"
136     makefile = open("mktemplates/Makefile").read()
137     makefile = makefileGenerator(project_info, makefile)
138     open(directory + "/Makefile", "w").write(makefile)
139     # Destination project dir
140     prjdir = directory + "/" + os.path.basename(directory)
141     if not edit:
142         shutil.rmtree(prjdir, True)
143         os.mkdir(prjdir)
144     # Destination hw files
145     hwdir = prjdir + "/hw"
146     if not edit:
147         shutil.rmtree(hwdir, True)
148         os.mkdir(hwdir)
149     # Copy all the hw files
150     for module, information in project_info.info("MODULES").items():
151         for hwfile in information["hw"]:
152             string = open(sources_dir + "/" + hwfile, "r").read()
153             hwfile_path = hwdir + "/" + os.path.basename(hwfile)
154             if not edit or not os.path.exists(hwfile_path):
155                 # If not in editing mode it copies all the hw files. If in
156                 # editing mode it copies only the files that don't exist yet
157                 open(hwdir + "/" + os.path.basename(hwfile), "w").write(string)
158     # Destination configurations files
159     cfgdir = prjdir + "/cfg"
160     if not edit:
161         shutil.rmtree(cfgdir, True)
162         os.mkdir(cfgdir)
163     # Set properly the autoenabled parameters
164     for module, information in project_info.info("MODULES").items():
165         if "configuration" in information and information["configuration"] != "":
166             configurations = project_info.info("CONFIGURATIONS")
167             configuration = configurations[information["configuration"]]
168             for start, parameter in configuration["paramlist"]:
169                 if "type" in configuration[parameter]["informations"] and configuration[parameter]["informations"]["type"] == "autoenabled":
170                     configuration[parameter]["value"] = "1" if information["enabled"] else "0"
171             project_info.setInfo("CONFIGURATIONS", configurations)
172     # Copy all the configuration files
173     for configuration, information in project_info.info("CONFIGURATIONS").items():
174         string = open(sources_dir + "/" + configuration, "r").read()
175         for start, parameter in information["paramlist"]:
176             infos = information[parameter]
177             value = infos["value"]
178             if "unsigned" in infos["informations"] and infos["informations"]["unsigned"]:
179                 value += "U"
180             if "long" in infos["informations"] and infos["informations"]["long"]:
181                 value += "L"
182             string = sub(string, parameter, value)
183         f = open(cfgdir + "/" + os.path.basename(configuration), "w")
184         f.write(string)
185         f.close()
186     if not edit:
187         # Destination user mk file (only on project creation)
188         makefile = open("mktemplates/template.mk", "r").read()
189         makefile = mkGenerator(project_info, makefile)
190         open(prjdir + "/" + os.path.basename(prjdir) + ".mk", "w").write(makefile)
191     # Destination wizard mk file
192     makefile = open("mktemplates/template_wiz.mk", "r").read()
193     makefile = mkGenerator(project_info, makefile)
194     open(prjdir + "/" + os.path.basename(prjdir) + "_wiz.mk", "w").write(makefile)
195     # Destination main.c file
196     if not edit:
197         main = open("srctemplates/main.c", "r").read()
198         open(prjdir + "/main.c", "w").write(main)
199     # Files for selected plugins
200     relevants_files = {}
201     for plugin in project_info.info("OUTPUT"):
202         module = loadPlugin(plugin)
203         relevants_files[plugin] = module.createProject(project_info)
204     project_info.setInfo("RELEVANT_FILES", relevants_files)
205
206 def loadPlugin(plugin):
207     """
208     Returns the given plugin module.
209     """
210     return getattr(__import__("plugins", {}, {}, [plugin]), plugin)
211     
212 def mkGenerator(project_info, makefile):
213     """
214     Generates the mk file for the current project.
215     """
216     mk_data = {}
217     mk_data["$pname"] = os.path.basename(project_info.info("PROJECT_PATH"))
218     mk_data["$cpuclockfreq"] = project_info.info("SELECTED_FREQ")
219     cpu_mk_parameters = []
220     for key, value in project_info.info("CPU_INFOS").items():
221         if key.startswith("MK_"):
222             cpu_mk_parameters.append("%s = %s" %(key.replace("MK", mk_data["$pname"]), value))
223     mk_data["$cpuparameters"] = "\n".join(cpu_mk_parameters)
224     mk_data["$csrc"], mk_data["$pcsrc"], mk_data["$cppasrc"], mk_data["$cxxsrc"], mk_data["$asrc"], mk_data["$constants"] = csrcGenerator(project_info)
225     mk_data["$prefix"] = replaceSeparators(project_info.info("TOOLCHAIN")["path"].split("gcc")[0])
226     mk_data["$suffix"] = replaceSeparators(project_info.info("TOOLCHAIN")["path"].split("gcc")[1])
227     mk_data["$main"] = os.path.basename(project_info.info("PROJECT_PATH")) + "/main.c"
228     for key in mk_data:
229         while makefile.find(key) != -1:
230             makefile = makefile.replace(key, mk_data[key])
231     return makefile
232
233 def makefileGenerator(project_info, makefile):
234     """
235     Generate the Makefile for the current project.
236     """
237     # TODO write a general function that works for both the mk file and the Makefile
238     while makefile.find("$pname") != -1:
239         makefile = makefile.replace("$pname", os.path.basename(project_info.info("PROJECT_PATH")))
240     return makefile
241
242 def csrcGenerator(project_info):
243     modules = project_info.info("MODULES")
244     files = project_info.info("FILES")
245     if "harvard" in project_info.info("CPU_INFOS")["CPU_TAGS"]:
246         harvard = True
247     else:
248         harvard = False
249     # file to be included in CSRC variable
250     csrc = []
251     # file to be included in PCSRC variable
252     pcsrc = []
253     # files to be included in CPPASRC variable
254     cppasrc = []
255     # files to be included in CXXSRC variable
256     cxxsrc = []
257     # files to be included in ASRC variable
258     asrc = []
259     # constants to be included at the beginning of the makefile
260     constants = {}
261     for module, information in modules.items():
262         module_files = set([])
263         dependency_files = set([])
264         # assembly sources
265         asm_files = set([])
266         hwdir = os.path.basename(project_info.info("PROJECT_PATH")) + "/hw" 
267         if information["enabled"]:
268             if "constants" in information:
269                 constants.update(information["constants"])
270             cfiles, sfiles = findModuleFiles(module, project_info)
271             module_files |= set(cfiles)
272             asm_files |= set(sfiles)
273             for file in information["hw"]:
274                 if file.endswith(".c"):
275                     module_files |= set([hwdir + "/" + os.path.basename(file)])
276             for file_dependency in information["depends"]:
277                 if file_dependency in files:
278                     dependencyCFiles, dependencySFiles = findModuleFiles(file_dependency, project_info)
279                     dependency_files |= set(dependencyCFiles)
280                     asm_files |= set(dependencySFiles)
281             for file in module_files:
282                 if not harvard or information.get("harvard", "both") == "both":
283                     csrc.append(file)
284                 if harvard and "harvard" in information:
285                     pcsrc.append(file)
286             for file in dependency_files:
287                 csrc.append(file)
288             for file in project_info.info("CPU_INFOS")["C_SRC"]:
289                 csrc.append(file)
290             for file in project_info.info("CPU_INFOS")["PC_SRC"]:
291                 pcsrc.append(file)
292             for file in asm_files:
293                 cppasrc.append(file)
294     for file in project_info.info("CPU_INFOS")["CPPA_SRC"]:
295         cppasrc.append(file)
296     for file in project_info.info("CPU_INFOS")["CXX_SRC"]:
297         cxxsrc.append(file)
298     for file in project_info.info("CPU_INFOS")["ASRC"]:
299         asrc.append(file)
300     csrc = " \\\n\t".join(csrc) + " \\"
301     pcsrc = " \\\n\t".join(pcsrc) + " \\"
302     cppasrc = " \\\n\t".join(cppasrc) + " \\"
303     cxxsrc = " \\\n\t".join(cxxsrc) + " \\"
304     asrc = " \\\n\t".join(asrc) + " \\"
305     constants = "\n".join([os.path.basename(project_info.info("PROJECT_PATH")) + "_" + key + " = " + unicode(value) for key, value in constants.items()])
306     return csrc, pcsrc, cppasrc, cxxsrc, asrc, constants
307
308 def findModuleFiles(module, project_info):
309     # Find the files related to the selected module
310     cfiles = []
311     sfiles = []
312     # .c files related to the module and the cpu architecture
313     for filename, path in findDefinitions(module + ".c", project_info) + \
314             findDefinitions(module + "_" + project_info.info("CPU_INFOS")["TOOLCHAIN"] + ".c", project_info):
315         path = path.replace(project_info.info("SOURCES_PATH") + os.sep, "")
316         path = replaceSeparators(path)
317         cfiles.append(path + "/" + filename)
318     # .s files related to the module and the cpu architecture
319     for filename, path in findDefinitions(module + ".s", project_info) + \
320             findDefinitions(module + "_" + project_info.info("CPU_INFOS")["TOOLCHAIN"] + ".s", project_info) + \
321             findDefinitions(module + ".S", project_info) + \
322             findDefinitions(module + "_" + project_info.info("CPU_INFOS")["TOOLCHAIN"] + ".S", project_info):
323         path = path.replace(project_info.info("SOURCES_PATH") + os.sep, "")
324         path = replaceSeparators(path)
325         sfiles.append(path + "/" + filename)
326     # .c and .s files related to the module and the cpu tags
327     for tag in project_info.info("CPU_INFOS")["CPU_TAGS"]:
328         for filename, path in findDefinitions(module + "_" + tag + ".c", project_info):
329             path = path.replace(project_info.info("SOURCES_PATH") + os.sep, "")
330             if os.sep != "/":
331                 path = replaceSeparators(path)
332             cfiles.append(path + "/" + filename)
333         for filename, path in findDefinitions(module + "_" + tag + ".s", project_info) + \
334                 findDefinitions(module + "_" + tag + ".S", project_info):
335             path = path.replace(project_info.info("SOURCES_PATH") + os.sep, "")
336             path = replaceSeparators(path)
337             sfiles.append(path + "/" + filename)
338     return cfiles, sfiles
339
340 def replaceSeparators(path):
341     """
342     Replace the separators in the given path with unix standard separator.
343     """
344     if os.sep != "/":
345         while path.find(os.sep) != -1:
346             path = path.replace(os.sep, "/")
347     return path
348
349 def getSystemPath():
350     path = os.environ["PATH"]
351     if os.name == "nt":
352         path = path.split(";")
353     else:
354         path = path.split(":")
355     return path
356
357 def findToolchains(path_list):
358     toolchains = []
359     for element in path_list:
360         for toolchain in glob.glob(element+ "/" + const.GCC_NAME):
361             toolchains.append(toolchain)
362     return list(set(toolchains))
363
364 def getToolchainInfo(output):
365     info = {}
366     expr = re.compile("Target: .*")
367     target = expr.findall(output)
368     if len(target) == 1:
369         info["target"] = target[0].split("Target: ")[1]
370     expr = re.compile("gcc version [0-9,.]*")
371     version = expr.findall(output)
372     if len(version) == 1:
373         info["version"] = version[0].split("gcc version ")[1]
374     expr = re.compile("gcc version [0-9,.]* \(.*\)")
375     build = expr.findall(output)
376     if len(build) == 1:
377         build = build[0].split("gcc version ")[1]
378         build = build[build.find("(") + 1 : build.find(")")]
379         info["build"] = build
380     expr = re.compile("Configured with: .*")
381     configured = expr.findall(output)
382     if len(configured) == 1:
383         info["configured"] = configured[0].split("Configured with: ")[1]
384     expr = re.compile("Thread model: .*")
385     thread = expr.findall(output)
386     if len(thread) == 1:
387         info["thread"] = thread[0].split("Thread model: ")[1]
388     return info
389
390 def getToolchainName(toolchain_info):
391     name = "GCC " + toolchain_info["version"] + " - " + toolchain_info["target"].strip()
392     return name
393
394 def loadSourceTree(project):
395     fileList = [f for f in os.walk(project.info("SOURCES_PATH"))]
396     project.setInfo("FILE_LIST", fileList)
397
398 def findDefinitions(ftype, project):
399     L = project.info("FILE_LIST")
400     definitions = []
401     for element in L:
402         for filename in element[2]:
403             if fnmatch.fnmatch(filename, ftype):
404                 definitions.append((filename, element[0]))
405     return definitions
406
407 def loadCpuInfos(project):
408     cpuInfos = []
409     for definition in findDefinitions(const.CPU_DEFINITION, project):
410         cpuInfos.append(getInfos(definition))
411     return cpuInfos
412
413 def getTagSet(cpu_info):
414     tag_set = set([])
415     for cpu in cpu_info:
416         tag_set |= set([cpu["CPU_NAME"]])
417         tag_set |= set(cpu["CPU_TAGS"])
418         tag_set |= set([cpu["TOOLCHAIN"]])
419     return tag_set
420         
421
422 def getInfos(definition):
423     D = {}
424     D.update(const.CPU_DEF)
425     def include(filename, dict = D, directory=definition[1]):
426         execfile(directory + "/" + filename, {}, D)
427     D["include"] = include
428     include(definition[0], D)
429     D["CPU_NAME"] = definition[0].split(".")[0]
430     D["DEFINITION_PATH"] = definition[1] + "/" + definition[0]
431     del D["include"]
432     return D
433
434 def getCommentList(string):
435     comment_list = re.findall(r"/\*{2}\s*([^*]*\*(?:[^/*][^*]*\*+)*)/", string)
436     comment_list = [re.findall(r"^\s*\* *(.*?)$", comment, re.MULTILINE) for comment in comment_list]
437     return comment_list
438
439 def loadModuleDefinition(first_comment):
440     to_be_parsed = False
441     module_definition = {}
442     for num, line in enumerate(first_comment):
443         index = line.find("$WIZ$")
444         if index != -1:
445             to_be_parsed = True
446             try:
447                 exec line[index + len("$WIZ$ "):] in {}, module_definition
448             except:
449                 raise ParseError(num, line[index:])
450         elif line.find("\\brief") != -1:
451             module_definition["module_description"] = line[line.find("\\brief") + len("\\brief "):]
452     module_dict = {}
453     if "module_name" in module_definition:
454         module_name = module_definition[const.MODULE_DEFINITION["module_name"]]
455         del module_definition[const.MODULE_DEFINITION["module_name"]]
456         module_dict[module_name] = {}
457         if const.MODULE_DEFINITION["module_depends"] in module_definition:
458             depends = module_definition[const.MODULE_DEFINITION["module_depends"]]
459             del module_definition[const.MODULE_DEFINITION["module_depends"]]
460             if type(depends) == str:
461                 depends = (depends,)
462             module_dict[module_name]["depends"] = depends
463         else:
464             module_dict[module_name]["depends"] = ()
465         if const.MODULE_DEFINITION["module_configuration"] in module_definition:
466             module_dict[module_name]["configuration"] = module_definition[const.MODULE_DEFINITION["module_configuration"]]
467             del module_definition[const.MODULE_DEFINITION["module_configuration"]]
468         else:
469             module_dict[module_name]["configuration"] = ""
470         if "module_description" in module_definition:
471             module_dict[module_name]["description"] = module_definition["module_description"]
472             del module_definition["module_description"]
473         if const.MODULE_DEFINITION["module_harvard"] in module_definition:
474             harvard = module_definition[const.MODULE_DEFINITION["module_harvard"]]
475             module_dict[module_name]["harvard"] = harvard
476             del module_definition[const.MODULE_DEFINITION["module_harvard"]]
477         if const.MODULE_DEFINITION["module_hw"] in module_definition:
478             hw = module_definition[const.MODULE_DEFINITION["module_hw"]]
479             del module_definition[const.MODULE_DEFINITION["module_hw"]]
480             if type(hw) == str:
481                 hw = (hw, )
482             module_dict[module_name]["hw"] = hw
483         else:
484             module_dict[module_name]["hw"] = ()
485         if const.MODULE_DEFINITION["module_supports"] in module_definition:
486             supports = module_definition[const.MODULE_DEFINITION["module_supports"]]
487             del module_definition[const.MODULE_DEFINITION["module_supports"]]
488             module_dict[module_name]["supports"] = supports
489         module_dict[module_name]["constants"] = module_definition
490         module_dict[module_name]["enabled"] = False
491     return to_be_parsed, module_dict
492
493 def isSupported(project, module=None, property_id=None):
494     if not module and property_id:
495         item = project.info("CONFIGURATIONS")[property_id[0]][property_id[1]]["informations"]
496     else:
497         item = project.info("MODULES")[module]
498     tag_dict = project.info("ALL_CPU_TAGS")
499     if "supports" in item:
500         support_string = item["supports"]
501         supported = {}
502         try:
503             exec "supported = " + support_string in tag_dict, supported
504         except:
505             raise SupportedException(support_string)
506         return supported["supported"]
507     else:
508         return True
509
510 def loadDefineLists(comment_list):
511     define_list = {}
512     for comment in comment_list:
513         for num, line in enumerate(comment):
514             index = line.find("$WIZ$")
515             if index != -1:
516                 try:
517                     exec line[index + len("$WIZ$ "):] in {}, define_list
518                 except:
519                     raise ParseError(num, line[index:])
520     for key, value in define_list.items():
521         if type(value) == str:
522             define_list[key] = (value,)
523     return define_list
524
525 def getDescriptionInformations(comment):
526     """
527     Take the doxygen comment and strip the wizard informations, returning the tuple
528     (comment, wizard_information)
529     """
530     brief = ""
531     description = ""
532     information = {}
533     for num, line in enumerate(comment):
534         index = line.find("$WIZ$")
535         if index != -1:
536             if len(brief) == 0:
537                 brief += line[:index].strip()
538             else:
539                 description += " " + line[:index]
540             try:
541                 exec line[index + len("$WIZ$ "):] in {}, information
542             except:
543                 raise ParseError(num, line[index:])
544         else:
545             if len(brief) == 0:
546                 brief += line.strip()
547             else:
548                 description += " " + line
549                 description = description.strip()
550     return brief.strip(), description.strip(), information
551
552 def getDefinitionBlocks(text):
553     """
554     Take a text and return a list of tuple (description, name-value).
555     """
556     block = []
557     block_tmp = re.finditer(r"/\*{2}\s*([^*]*\*(?:[^/*][^*]*\*+)*)/\s*#define\s+((?:[^/]*?/?)+)\s*?(?:/{2,3}[^<].*?)?$", text, re.MULTILINE)
558     for match in block_tmp:
559         # Only the first element is needed
560         comment = match.group(1)
561         define = match.group(2)
562         start = match.start()
563         block.append(([re.findall(r"^\s*\* *(.*?)$", line, re.MULTILINE)[0] for line in comment.splitlines()], define, start))
564     for match in re.finditer(r"/{3}\s*([^<].*?)\s*#define\s+((?:[^/]*?/?)+)\s*?(?:/{2,3}[^<].*?)?$", text, re.MULTILINE):
565         comment = match.group(1)
566         define = match.group(2)
567         start = match.start()
568         block.append(([comment], define, start))
569     for match in re.finditer(r"#define\s*(.*?)\s*/{3}<\s*(.+?)\s*?(?:/{2,3}[^<].*?)?$", text, re.MULTILINE):
570         comment = match.group(2)
571         define = match.group(1)
572         start = match.start()
573         block.append(([comment], define, start))
574     return block
575
576 def loadModuleData(project, edit=False):
577     module_info_dict = {}
578     list_info_dict = {}
579     configuration_info_dict = {}
580     file_dict = {}
581     for filename, path in findDefinitions("*.h", project) + findDefinitions("*.c", project) + findDefinitions("*.s", project) + findDefinitions("*.S", project):
582         comment_list = getCommentList(open(path + "/" + filename, "r").read())
583         if len(comment_list) > 0:
584             module_info = {}
585             configuration_info = {}
586             try:
587                 to_be_parsed, module_dict = loadModuleDefinition(comment_list[0])
588             except ParseError, err:
589                 raise DefineException.ModuleDefineException(path, err.line_number, err.line)
590             for module, information in module_dict.items():
591                 if "depends" not in information:
592                     information["depends"] = ()
593                 information["depends"] += (filename.split(".")[0],)
594                 information["category"] = os.path.basename(path)
595                 if "configuration" in information and len(information["configuration"]):
596                     configuration = module_dict[module]["configuration"]
597                     try:
598                         configuration_info[configuration] = loadConfigurationInfos(project.info("SOURCES_PATH") + "/" + configuration)
599                     except ParseError, err:
600                         raise DefineException.ConfigurationDefineException(project.info("SOURCES_PATH") + "/" + configuration, err.line_number, err.line)
601                     if edit:
602                         try:
603                             path = os.path.basename(project.info("PROJECT_PATH"))
604                             directory = project.info("PROJECT_PATH")
605                             user_configuration = loadConfigurationInfos(directory + "/" + configuration.replace("bertos", path))
606                             configuration_info[configuration] = updateConfigurationValues(configuration_info[configuration], user_configuration)
607                         except ParseError, err:
608                             raise DefineException.ConfigurationDefineException(directory + "/" + configuration.replace("bertos", path))
609             module_info_dict.update(module_dict)
610             configuration_info_dict.update(configuration_info)
611             if to_be_parsed:
612                 try:
613                     list_dict = loadDefineLists(comment_list[1:])
614                     list_info_dict.update(list_dict)
615                 except ParseError, err:
616                     raise DefineException.EnumDefineException(path, err.line_number, err.line)
617     for filename, path in findDefinitions("*_" + project.info("CPU_INFOS")["TOOLCHAIN"] + ".h", project):
618         comment_list = getCommentList(open(path + "/" + filename, "r").read())
619         list_info_dict.update(loadDefineLists(comment_list))
620     for tag in project.info("CPU_INFOS")["CPU_TAGS"]:
621         for filename, path in findDefinitions("*_" + tag + ".h", project):
622             comment_list = getCommentList(open(path + "/" + filename, "r").read())
623             list_info_dict.update(loadDefineLists(comment_list))
624     project.setInfo("MODULES", module_info_dict)
625     project.setInfo("LISTS", list_info_dict)
626     project.setInfo("CONFIGURATIONS", configuration_info_dict)
627     project.setInfo("FILES", file_dict)
628
629 def formatParamNameValue(text):
630     """
631     Take the given string and return a tuple with the name of the parameter in the first position
632     and the value in the second.
633     """
634     block = re.findall("\s*([^\s]+)\s*(.+?)\s*$", text, re.MULTILINE)
635     return block[0]
636
637 def loadConfigurationInfos(path):
638     """
639     Return the module configurations found in the given file as a dict with the
640     parameter name as key and a dict containig the fields above as value:
641         "value": the value of the parameter
642         "description": the description of the parameter
643         "informations": a dict containig optional informations:
644             "type": "int" | "boolean" | "enum"
645             "min": the minimum value for integer parameters
646             "max": the maximum value for integer parameters
647             "long": boolean indicating if the num is a long
648             "unsigned": boolean indicating if the num is an unsigned
649             "value_list": the name of the enum for enum parameters
650     """
651     configuration_infos = {}
652     configuration_infos["paramlist"] = []
653     for comment, define, start in getDefinitionBlocks(open(path, "r").read()):
654         name, value = formatParamNameValue(define)
655         brief, description, informations = getDescriptionInformations(comment)
656         configuration_infos["paramlist"].append((start, name))
657         configuration_infos[name] = {}
658         configuration_infos[name]["value"] = value
659         configuration_infos[name]["informations"] = informations
660         if not "type" in configuration_infos[name]["informations"]:
661             configuration_infos[name]["informations"]["type"] = findParameterType(configuration_infos[name])
662         if ("type" in configuration_infos[name]["informations"] and
663                 configuration_infos[name]["informations"]["type"] == "int" and
664                 configuration_infos[name]["value"].find("L") != -1):
665             configuration_infos[name]["informations"]["long"] = True
666             configuration_infos[name]["value"] = configuration_infos[name]["value"].replace("L", "")
667         if ("type" in configuration_infos[name]["informations"] and
668                 configuration_infos[name]["informations"]["type"] == "int" and
669                 configuration_infos[name]["value"].find("U") != -1):
670             configuration_infos[name]["informations"]["unsigned"] = True
671             configuration_infos[name]["value"] = configuration_infos[name]["value"].replace("U", "")
672         configuration_infos[name]["description"] = description
673         configuration_infos[name]["brief"] = brief
674     return configuration_infos
675
676 def updateConfigurationValues(def_conf, user_conf):
677     for param in def_conf["paramlist"]:
678         def_conf[param[1]]["value"] = user_conf[param[1]]["value"]
679     return def_conf
680
681 def findParameterType(parameter):
682     if "value_list" in parameter["informations"]:
683         return "enum"
684     if "min" in parameter["informations"] or "max" in parameter["informations"] or re.match(r"^\d+U?L?$", parameter["value"]) != None:
685         return "int"
686
687 def sub(string, parameter, value):
688     """
689     Substitute the given value at the given parameter define in the given string
690     """
691     return re.sub(r"(?P<define>#define\s+" + parameter + r"\s+)([^\s]+)", r"\g<define>" + value, string)
692
693 def isInt(informations):
694     """
695     Return True if the value is a simple int.
696     """
697     if ("long" not in informatios or not informations["long"]) and ("unsigned" not in informations or informations["unsigned"]):
698         return True
699     else:
700         return False
701
702 def isLong(informations):
703     """
704     Return True if the value is a long.
705     """
706     if "long" in informations and informations["long"] and "unsigned" not in informations:
707         return True
708     else:
709         return False
710
711 def isUnsigned(informations):
712     """
713     Return True if the value is an unsigned.
714     """
715     if "unsigned" in informations and informations["unsigned"] and "long" not in informations:
716         return True
717     else:
718         return False
719
720 def isUnsignedLong(informations):
721     """
722     Return True if the value is an unsigned long.
723     """
724     if "unsigned" in informations and "long" in informations and informations["unsigned"] and informations["long"]:
725         return True
726     else:
727         return False
728
729 class ParseError(Exception):
730     def __init__(self, line_number, line):
731         Exception.__init__(self)
732         self.line_number = line_number
733         self.line = line
734
735 class SupportedException(Exception):
736     def __init__(self, support_string):
737         Exception.__init__(self)
738         self.support_string = support_string