import DefineException
import BProject
+from LoadException import VersionException, ToolchainException
+
def isBertosDir(directory):
return os.path.exists(directory + "/VERSION")
def bertosVersion(directory):
return open(directory + "/VERSION").readline().strip()
-def loadBertosProject(project_file):
+def loadBertosProject(project_file, info_dict):
project_data = pickle.loads(open(project_file, "r").read())
project_info = BProject.BProject()
project_info.setInfo("PROJECT_PATH", os.path.dirname(project_file))
- project_info.setInfo("SOURCES_PATH", project_data["SOURCES_PATH"])
- project_info.setInfo("TOOLCHAIN", project_data["TOOLCHAIN"])
+ if "SOURCES_PATH" in info_dict:
+ project_data["SOURCES_PATH"] = info_dict["SOURCES_PATH"]
+ if os.path.exists(project_data["SOURCES_PATH"]):
+ project_info.setInfo("SOURCES_PATH", project_data["SOURCES_PATH"])
+ else:
+ raise VersionException(project_info)
loadSourceTree(project_info)
cpu_name = project_data["CPU_NAME"]
project_info.setInfo("CPU_NAME", cpu_name)
tag_dict[element] = False
infos = project_info.info("CPU_INFOS")
for tag in tag_dict:
- if tag in infos["CPU_TAGS"] + [infos["CPU_NAME"], infos["CORE_CPU"], infos["TOOLCHAIN"]]:
+ if tag in infos["CPU_TAGS"] + [infos["CPU_NAME"], infos["TOOLCHAIN"]]:
tag_dict[tag] = True
else:
tag_dict[tag] = False
project_info.setInfo("ALL_CPU_TAGS", tag_dict)
+ if "TOOLCHAIN" in info_dict:
+ project_data["TOOLCHAIN"] = info_dict["TOOLCHAIN"]
+ if os.path.exists(project_data["TOOLCHAIN"]["path"]):
+ project_info.setInfo("TOOLCHAIN", project_data["TOOLCHAIN"])
+ else:
+ raise ToolchainException(project_info)
+ project_info.setInfo("SELECTED_FREQ", project_data["SELECTED_FREQ"])
+ project_info.setInfo("OUTPUT", project_data["OUTPUT"])
loadModuleData(project_info, True)
+ setEnabledModules(project_info, project_data["ENABLED_MODULES"])
+ return project_info
+
+def setEnabledModules(project_info, enabled_modules):
modules = project_info.info("MODULES")
+ files = {}
for module, information in modules.items():
- information["enabled"] = module in project_data["ENABLED_MODULES"]
+ information["enabled"] = module in enabled_modules
+ for dependency in information["depends"]:
+ if not dependency in modules:
+ if dependency in files:
+ files[dependency] += 1
+ else:
+ files[dependency] = 1
project_info.setInfo("MODULES", modules)
- return project_info
+ project_info.setInfo("FILES", files)
+
+def enabledModules(project_info):
+ enabled_modules = []
+ for name, module in project_info.info("MODULES").items():
+ if module["enabled"]:
+ enabled_modules.append(name)
+ return enabled_modules
+
+def mergeSources(srcdir, new_sources, old_sources):
+ # The current mergeSources function provide only a raw copy of the sources in the
+ # created project.
+ #
+ # TODO: implement the three way merge algorithm
+ #
+ shutil.rmtree(srcdir, True)
+ copytree.copytree(os.path.join(new_sources, "bertos"), srcdir, ignore_list=const.IGNORE_LIST)
def projectFileGenerator(project_info):
directory = project_info.info("PROJECT_PATH")
project_data["TOOLCHAIN"] = project_info.info("TOOLCHAIN")
project_data["CPU_NAME"] = project_info.info("CPU_NAME")
project_data["SELECTED_FREQ"] = project_info.info("SELECTED_FREQ")
+ project_data["OUTPUT"] = project_info.info("OUTPUT")
return pickle.dumps(project_data)
def createBertosProject(project_info, edit=False):
directory = project_info.info("PROJECT_PATH")
sources_dir = project_info.info("SOURCES_PATH")
- if os.path.isdir(directory) and not edit:
- shutil.rmtree(directory, True)
- os.makedirs(directory)
+ old_sources_dir = project_info.info("OLD_SOURCES_PATH")
+ if not edit:
+ if os.path.isdir(directory):
+ shutil.rmtree(directory, True)
+ os.makedirs(directory)
+ # Write the project file
f = open(directory + "/project.bertos", "w")
f.write(projectFileGenerator(project_info))
f.close()
# Destination source dir
srcdir = directory + "/bertos"
- shutil.rmtree(srcdir, True)
- copytree.copytree(sources_dir + "/bertos", srcdir, ignore_list=const.IGNORE_LIST)
+ if not edit:
+ # If not in editing mode it copies all the bertos sources in the /bertos subdirectory of the project
+ shutil.rmtree(srcdir, True)
+ copytree.copytree(sources_dir + "/bertos", srcdir, ignore_list=const.IGNORE_LIST)
+ elif old_sources_dir:
+ # If in editing mode it merges the current bertos sources with the selected ones
+ # TODO: implement the three way merge algotihm
+ #
+ mergeSources(srcdir, sources_dir, old_sources_dir)
# Destination makefile
makefile = directory + "/Makefile"
- if os.path.exists(makefile):
- os.remove(makefile)
makefile = open("mktemplates/Makefile").read()
makefile = makefileGenerator(project_info, makefile)
open(directory + "/Makefile", "w").write(makefile)
# Destination project dir
prjdir = directory + "/" + os.path.basename(directory)
- shutil.rmtree(prjdir, True)
- os.mkdir(prjdir)
+ if not edit:
+ shutil.rmtree(prjdir, True)
+ os.mkdir(prjdir)
# Destination hw files
hwdir = prjdir + "/hw"
- shutil.rmtree(hwdir, True)
- os.mkdir(hwdir)
+ if not edit:
+ shutil.rmtree(hwdir, True)
+ os.mkdir(hwdir)
# Copy all the hw files
for module, information in project_info.info("MODULES").items():
for hwfile in information["hw"]:
string = open(sources_dir + "/" + hwfile, "r").read()
- open(hwdir + "/" + os.path.basename(hwfile), "w").write(string)
+ hwfile_path = hwdir + "/" + os.path.basename(hwfile)
+ if not edit or not os.path.exists(hwfile_path):
+ # If not in editing mode it copies all the hw files. If in
+ # editing mode it copies only the files that don't exist yet
+ open(hwdir + "/" + os.path.basename(hwfile), "w").write(string)
# Destination configurations files
cfgdir = prjdir + "/cfg"
- shutil.rmtree(cfgdir, True)
- os.mkdir(cfgdir)
+ if not edit:
+ shutil.rmtree(cfgdir, True)
+ os.mkdir(cfgdir)
# Set properly the autoenabled parameters
for module, information in project_info.info("MODULES").items():
if "configuration" in information and information["configuration"] != "":
f = open(cfgdir + "/" + os.path.basename(configuration), "w")
f.write(string)
f.close()
- # Destinatio mk file
- makefile = open("mktemplates/template.mk", "r").read()
+ if not edit:
+ # Destination user mk file (only on project creation)
+ makefile = open("mktemplates/template.mk", "r").read()
+ makefile = mkGenerator(project_info, makefile)
+ open(prjdir + "/" + os.path.basename(prjdir) + ".mk", "w").write(makefile)
+ # Destination wizard mk file
+ makefile = open("mktemplates/template_wiz.mk", "r").read()
makefile = mkGenerator(project_info, makefile)
- open(prjdir + "/" + os.path.basename(prjdir) + ".mk", "w").write(makefile)
+ open(prjdir + "/" + os.path.basename(prjdir) + "_wiz.mk", "w").write(makefile)
# Destination main.c file
- main = open("srctemplates/main.c", "r").read()
- open(prjdir + "/main.c", "w").write(main)
+ if not edit:
+ main = open("srctemplates/main.c", "r").read()
+ open(prjdir + "/main.c", "w").write(main)
# Files for selected plugins
relevants_files = {}
for plugin in project_info.info("OUTPUT"):
"""
mk_data = {}
mk_data["$pname"] = os.path.basename(project_info.info("PROJECT_PATH"))
- mk_data["$cpuflag"] = project_info.info("CPU_INFOS")["CPU_FLAG_NAME"]
- mk_data["$cpuname"] = project_info.info("CPU_INFOS")["CORE_CPU"]
mk_data["$cpuclockfreq"] = project_info.info("SELECTED_FREQ")
- mk_data["$cflags"] = " ".join(project_info.info("CPU_INFOS")["C_FLAGS"])
- mk_data["$ldflags"] = " ".join(project_info.info("CPU_INFOS")["LD_FLAGS"])
- mk_data["$cppflags"] = " ".join(project_info.info("CPU_INFOS")["CPP_FLAGS"])
- mk_data["$cppaflags"] = " ".join(project_info.info("CPU_INFOS")["CPPA_FLAGS"])
- mk_data["$cxxflags"] = " ".join(project_info.info("CPU_INFOS")["CXX_FLAGS"])
- mk_data["$asflags"] = " ".join(project_info.info("CPU_INFOS")["AS_FLAGS"])
- mk_data["$arflags"] = " ".join(project_info.info("CPU_INFOS")["AR_FLAGS"])
+ cpu_mk_parameters = []
+ for key, value in project_info.info("CPU_INFOS").items():
+ if key.startswith(const.MK_PARAM_ID):
+ cpu_mk_parameters.append("%s = %s" %(key.replace("MK", mk_data["$pname"]), value))
+ mk_data["$cpuparameters"] = "\n".join(cpu_mk_parameters)
mk_data["$csrc"], mk_data["$pcsrc"], mk_data["$cppasrc"], mk_data["$cxxsrc"], mk_data["$asrc"], mk_data["$constants"] = csrcGenerator(project_info)
mk_data["$prefix"] = replaceSeparators(project_info.info("TOOLCHAIN")["path"].split("gcc")[0])
mk_data["$suffix"] = replaceSeparators(project_info.info("TOOLCHAIN")["path"].split("gcc")[1])
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(project_info.info("PROJECT_PATH")))
+ while makefile.find("$pname") != -1:
+ makefile = makefile.replace("$pname", os.path.basename(project_info.info("PROJECT_PATH")))
return makefile
def csrcGenerator(project_info):
cxxsrc.append(file)
for file in project_info.info("CPU_INFOS")["ASRC"]:
asrc.append(file)
+ csrc = set(csrc)
csrc = " \\\n\t".join(csrc) + " \\"
+ pcsrc = set(pcsrc)
pcsrc = " \\\n\t".join(pcsrc) + " \\"
+ cppasrc = set(cppasrc)
cppasrc = " \\\n\t".join(cppasrc) + " \\"
+ cxxsrc = set(cxxsrc)
cxxsrc = " \\\n\t".join(cxxsrc) + " \\"
+ asrc = set(asrc)
asrc = " \\\n\t".join(asrc) + " \\"
constants = "\n".join([os.path.basename(project_info.info("PROJECT_PATH")) + "_" + key + " = " + unicode(value) for key, value in constants.items()])
return csrc, pcsrc, cppasrc, cxxsrc, asrc, constants
for cpu in cpu_info:
tag_set |= set([cpu["CPU_NAME"]])
tag_set |= set(cpu["CPU_TAGS"])
- tag_set |= set([cpu["CORE_CPU"]])
tag_set |= set([cpu["TOOLCHAIN"]])
return tag_set
def updateConfigurationValues(def_conf, user_conf):
for param in def_conf["paramlist"]:
- def_conf[param[1]]["value"] = user_conf[param[1]]["value"]
+ if param[1] in user_conf and "value" in user_conf[param[1]]:
+ def_conf[param[1]]["value"] = user_conf[param[1]]["value"]
return def_conf
def findParameterType(parameter):