X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=wizard%2Fbertos_utils.py;h=700d72e7ef65087eb0b826cbed6e0be2c47137da;hb=9f41d37cc8136d4a26e004a93ceb523e008f8486;hp=df26613f66785a662666a44edb5a076719566775;hpb=0baa18b329a9bc006fc74f7ef65581b04b43cfb7;p=bertos.git diff --git a/wizard/bertos_utils.py b/wizard/bertos_utils.py index df26613f..700d72e7 100644 --- a/wizard/bertos_utils.py +++ b/wizard/bertos_utils.py @@ -16,6 +16,7 @@ import re import shutil import const +import DefineException def isBertosDir(directory): return os.path.exists(directory + "/VERSION") @@ -23,15 +24,72 @@ def isBertosDir(directory): def bertosVersion(directory): return open(directory + "/VERSION").readline().strip() -def createBertosProject(directory, sources_dir, projectInfos): +def createBertosProject(projectInfos): + directory = projectInfos.info("PROJECT_PATH") + sourcesDir = projectInfos.info("SOURCES_PATH") if not os.path.isdir(directory): os.mkdir(directory) f = open(directory + "/project.bertos", "w") f.write(repr(projectInfos)) f.close() - shutil.rmtree(directory + "/bertos", True) - shutil.copytree(sources_dir + "/bertos", directory + "/bertos") - shutil.copy(sources_dir + "/Makefile", directory + "/Makefile") + ## Destination source dir + srcdir = directory + "/bertos" + shutil.rmtree(srcdir, True) + shutil.copytree(sourcesDir + "/bertos", srcdir) + ## Destination makefile + makefile = directory + "/Makefile" + if os.path.exists(makefile): + os.remove(makefile) + makefile = open("mktemplates/Makefile").read() + makefile = makefileGenerator(projectInfos, makefile) + open(directory + "/Makefile", "w").write(makefile) + ## Destination project dir + prjdir = directory + "/" + os.path.basename(directory) + shutil.rmtree(prjdir, True) + os.mkdir(prjdir) + ## Destination configurations files + cfgdir = prjdir + "/cfg" + shutil.rmtree(cfgdir, True) + os.mkdir(cfgdir) + for key, value in projectInfos.info("CONFIGURATIONS").items(): + string = open(sourcesDir + "/" + key, "r").read() + for parameter, infos in value.items(): + value = infos["value"] + if "unsigned" in infos["informations"].keys() and infos["informations"]["unsigned"]: + value += "U" + if "long" in infos["informations"].keys() and infos["informations"]["long"]: + value += "L" + string = sub(string, parameter, value) + f = open(cfgdir + "/" + os.path.basename(key), "w") + f.write(string) + f.close() + ## Destinatio mk file + makefile = open("mktemplates/template.mk", "r").read() + makefile = mkGenerator(projectInfos, makefile) + open(prjdir + "/" + os.path.basename(prjdir) + ".mk", "w").write(makefile) + +def mkGenerator(projectInfos, makefile): + """ + Generates the mk file for the current project. + """ + mkData = {} + mkData["pname"] = os.path.basename(projectInfos.info("PROJECT_PATH")) + mkData["cpuname"] = projectInfos.info("CPU_INFOS")["CPU_NAME"] + mkData["cflags"] = " ".join(projectInfos.info("CPU_INFOS")["C_FLAGS"]) + mkData["ldflags"] = " ".join(projectInfos.info("CPU_INFOS")["LD_FLAGS"]) + for key in mkData: + while makefile.find(key) != -1: + makefile = makefile.replace(key, mkData[key]) + return makefile + +def makefileGenerator(projectInfos, makefile): + """ + Generate the Makefile for the current project. + """ + # TODO: write a general function that works for both the mk file and the Makefile + while makefile.find("project_name") != -1: + makefile = makefile.replace("project_name", os.path.basename(projectInfos.info("PROJECT_PATH"))) + return makefile def getSystemPath(): path = os.environ["PATH"] @@ -75,16 +133,22 @@ def getToolchainInfo(output): info["thread"] = thread[0].split("Thread model: ")[1] return info -def findDefinitions(ftype, path): - L = os.walk(path) +def loadSourceTree(project): + fileList = [f for f in os.walk(project.info("SOURCES_PATH"))] + project.setInfo("FILE_LIST", fileList) + +def findDefinitions(ftype, project): + L = project.info("FILE_LIST") + definitions = [] for element in L: for filename in element[2]: if fnmatch.fnmatch(filename, ftype): - yield (filename, element[0]) + definitions.append((filename, element[0])) + return definitions -def loadCpuInfos(path): +def loadCpuInfos(project): cpuInfos = [] - for definition in findDefinitions(const.CPU_DEFINITION, path): + for definition in findDefinitions(const.CPU_DEFINITION, project): cpuInfos.append(getInfos(definition)) return cpuInfos @@ -146,15 +210,40 @@ def loadConfigurationInfos(path): "long": boolean indicating if the num is a long "value_list": the name of the enum for enum parameters """ - configurationInfos = {} - for comment, define in getDefinitionBlocks(open(path, "r").read()): - name, value = formatParamNameValue(define) - description, informations = getDescriptionInformations(comment) - configurationInfos[name] = {} - configurationInfos[name]["value"] = value - configurationInfos[name]["informations"] = informations - configurationInfos[name]["description"] = description - return configurationInfos + try: + configurationInfos = {} + for comment, define in getDefinitionBlocks(open(path, "r").read()): + name, value = formatParamNameValue(define) + description, informations = getDescriptionInformations(comment) + configurationInfos[name] = {} + configurationInfos[name]["value"] = value + configurationInfos[name]["informations"] = informations + if ("type" in configurationInfos[name]["informations"].keys() and + configurationInfos[name]["informations"]["type"] == "int" and + configurationInfos[name]["value"].find("L") != -1): + configurationInfos[name]["informations"]["long"] = True + configurationInfos[name]["value"] = configurationInfos[name]["value"].replace("L", "") + if ("type" in configurationInfos[name]["informations"].keys() and + configurationInfos[name]["informations"]["type"] == "int" and + configurationInfos[name]["value"].find("U") != -1): + configurationInfos[name]["informations"]["unsigned"] = True + configurationInfos[name]["value"] = configurationInfos[name]["value"].replace("U", "") + configurationInfos[name]["description"] = description + return configurationInfos + except SyntaxError: + raise DefineException.ConfigurationDefineException(path, name) + +def loadConfigurationInfosDict(project): + """ + Store in the project the configuration infos as a dict. + """ + modules = project.info("MODULES") + configurations = {} + for module, informations in modules.items(): + if len(informations["configuration"]) > 0: + configurations[informations["configuration"]] = loadConfigurationInfos(project.info("SOURCES_PATH") + + "/" + informations["configuration"]) + project.setInfo("CONFIGURATIONS", configurations) def loadModuleInfos(path): """ @@ -167,50 +256,106 @@ def loadModuleInfos(path): "enabled": contains False but the wizard will change if the user select the module """ - moduleInfos = {} - string = open(path, "r").read() - commentList = re.findall(r"/\*{2}\s*([^*]*\*(?:[^/*][^*]*\*+)*)/", string) - commentList = [" ".join(re.findall(r"^\s*\*?\s*(.*?)\s*?(?:/{2}.*?)?$", comment, re.MULTILINE)).strip() for comment in commentList] - for comment in commentList: - index = comment.find("$WIZARD_MODULE") - if index != -1: - exec(comment[index + 1:]) - moduleInfos[WIZARD_MODULE["name"]] = {"depends": WIZARD_MODULE["depends"], - "configuration": WIZARD_MODULE["configuration"], - "description": "", - "enabled": False} - return moduleInfos - return {} - -def loadModuleInfosDict(path): - """ - Return the dict containig all the modules + try: + moduleInfos = {} + string = open(path, "r").read() + commentList = re.findall(r"/\*{2}\s*([^*]*\*(?:[^/*][^*]*\*+)*)/", string) + commentList = [" ".join(re.findall(r"^\s*\*?\s*(.*?)\s*?(?:/{2}.*?)?$", comment, re.MULTILINE)).strip() for comment in commentList] + for comment in commentList: + index = comment.find("$WIZARD_MODULE") + if index != -1: + exec(comment[index + 1:]) + moduleInfos[WIZARD_MODULE["name"]] = {"depends": WIZARD_MODULE["depends"], + "configuration": WIZARD_MODULE["configuration"], + "description": "", + "enabled": False} + index = comment.find("\\brief") + if index != -1: + description = comment[index + 7:] + description = description[:description.find(" * ")] + moduleInfos[WIZARD_MODULE["name"]]["description"] = description + return moduleInfos + return {} + except SyntaxError: + raise DefineException.ModuleDefineException(path) + +def loadModuleInfosDict(project): + """ + Store in the project the dict containig all the modules """ moduleInfosDict = {} - for filename, path in findDefinitions("*.h", path): + for filename, path in findDefinitions("*.h", project): moduleInfosDict.update(loadModuleInfos(path + "/" + filename)) - return moduleInfosDict + project.setInfo("MODULES", moduleInfosDict) def loadDefineLists(path): """ Return a dict with the name of the list as key and a list of string as value """ - string = open(path, "r").read() - commentList = re.findall(r"/\*{2}\s*([^*]*\*(?:[^/*][^*]*\*+)*)/", string) - commentList = [" ".join(re.findall(r"^\s*\*?\s*(.*?)\s*?(?:/{2}.*?)?$", comment, re.MULTILINE)).strip() for comment in commentList] - listDict = {} - for comment in commentList: - index = comment.find("$WIZARD_LIST") - if index != -1: - exec(comment[index + 1:]) - listDict.update(WIZARD_LIST) - return listDict + try: + string = open(path, "r").read() + commentList = re.findall(r"/\*{2}\s*([^*]*\*(?:[^/*][^*]*\*+)*)/", string) + commentList = [" ".join(re.findall(r"^\s*\*?\s*(.*?)\s*?(?:/{2}.*?)?$", comment, re.MULTILINE)).strip() for comment in commentList] + listDict = {} + for comment in commentList: + index = comment.find("$WIZARD_LIST") + if index != -1: + exec(comment[index + 1:]) + listDict.update(WIZARD_LIST) + return listDict + except SyntaxError: + raise DefineException.EnumDefineException(path) -def loadDefineListsDict(path): +def loadDefineListsDict(project): """ - Return the dict containing all the define lists + Store in the project the dict containing all the define lists """ defineListsDict = {} - for filename, path in findDefinitions("*.h", path): + for filename, path in findDefinitions("*.h", project): defineListsDict.update(loadDefineLists(path + "/" + filename)) - return defineListsDict \ No newline at end of file + lists = project.info("LISTS") + if lists is not None: + defineListsDict.update(lists) + project.setInfo("LISTS", defineListsDict) + +def sub(string, parameter, value): + """ + Substitute the given value at the given parameter define in the given string + """ + return re.sub(r"(?P#define\s+" + parameter + r"\s+)([^\s]+)", r"\g" + value, string) + +def isInt(informations): + """ + Return True if the value is a simple int. + """ + if ("long" not in informatios.keys() or not informations["long"]) and ("unsigned" not in informations.keys() or informations["unsigned"]): + return True + else: + return False + +def isLong(informations): + """ + Return True if the value is a long. + """ + if "long" in informations.keys() and informations["long"] and "unsigned" not in informations.keys(): + return True + else: + return False + +def isUnsigned(informations): + """ + Return True if the value is an unsigned. + """ + if "unsigned" in informations.keys() and informations["unsigned"] and "long" not in informations.keys(): + return True + else: + return False + +def isUnsignedLong(informations): + """ + Return True if the value is an unsigned long. + """ + if "unsigned" in informations.keys() and "long" in informations.keys() and informations["unsigned"] and informations["long"]: + return True + else: + return False \ No newline at end of file