import DefineException
import BProject
+from _wizard_version import WIZARD_VERSION
+
+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_dir = os.path.dirname(project_file)
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"])
- project_info.setInfo("SELECTED_FREQ", project_data["SELECTED_FREQ"])
- project_info.setInfo("OUTPUT", project_data["OUTPUT"])
+ # Check for the Wizard version
+ wizard_version = project_data.get("WIZARD_VERSION", 0)
+ if not wizard_version:
+ project_data["SOURCES_PATH"] = project_dir
+ 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)
+ if not isBertosDir(os.path.dirname(project_file)):
+ version_file = open(os.path.join(const.DATA_DIR, "vtemplates/VERSION"), "r").read()
+ open(os.path.join(project_dir, "VERSION"), "w").write(version_file.replace("$version", "").strip())
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
- pass
+ #
+ 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")
if information["enabled"]:
enabled_modules.append(module)
project_data["ENABLED_MODULES"] = enabled_modules
- project_data["SOURCES_PATH"] = project_info.info("SOURCES_PATH")
+ # Use the local BeRTOS version instead of the original one
+ # project_data["SOURCES_PATH"] = project_info.info("SOURCES_PATH")
+ project_data["SOURCES_PATH"] = directory
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")
+ project_data["WIZARD_VERSION"] = WIZARD_VERSION
return pickle.dumps(project_data)
def createBertosProject(project_info, edit=False):
directory = project_info.info("PROJECT_PATH")
sources_dir = project_info.info("SOURCES_PATH")
+ old_sources_dir = project_info.info("OLD_SOURCES_PATH")
if not edit:
if os.path.isdir(directory):
shutil.rmtree(directory, True)
f = open(directory + "/project.bertos", "w")
f.write(projectFileGenerator(project_info))
f.close()
+ # VERSION file
+ version_file = open(os.path.join(const.DATA_DIR, "vtemplates/VERSION"), "r").read()
+ open(directory + "/VERSION", "w").write(versionFileGenerator(project_info, version_file))
# Destination source dir
srcdir = directory + "/bertos"
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)
- else:
+ 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)
- #
- pass
+ mergeSources(srcdir, sources_dir, old_sources_dir)
# Destination makefile
makefile = directory + "/Makefile"
- makefile = open("mktemplates/Makefile").read()
+ makefile = open(os.path.join(const.DATA_DIR, "mktemplates/Makefile"), 'r').read()
makefile = makefileGenerator(project_info, makefile)
open(directory + "/Makefile", "w").write(makefile)
# Destination project dir
f.close()
if not edit:
# Destination user mk file (only on project creation)
- makefile = open("mktemplates/template.mk", "r").read()
- makefile = mkGenerator(project_info, makefile)
+ makefile = open(os.path.join(const.DATA_DIR, "mktemplates/template.mk"), "r").read()
+ # Deadly performances loss was here :(
+ makefile = userMkGenerator(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 = open(os.path.join(const.DATA_DIR, "mktemplates/template_wiz.mk"), "r").read()
makefile = mkGenerator(project_info, makefile)
open(prjdir + "/" + os.path.basename(prjdir) + "_wiz.mk", "w").write(makefile)
# Destination main.c file
if not edit:
- main = open("srctemplates/main.c", "r").read()
+ main = open(os.path.join(const.DATA_DIR, "srctemplates/main.c"), "r").read()
open(prjdir + "/main.c", "w").write(main)
# Files for selected plugins
relevants_files = {}
Returns the given plugin module.
"""
return getattr(__import__("plugins", {}, {}, [plugin]), plugin)
-
+
+def versionFileGenerator(project_info, version_file):
+ version = bertosVersion(project_info.info("SOURCES_PATH"))
+ return version_file.replace('$version', version)
+
+def userMkGenerator(project_info, makefile):
+ mk_data = {}
+ mk_data["$pname"] = os.path.basename(project_info.info("PROJECT_PATH"))
+ for key in mk_data:
+ while makefile.find(key) != -1:
+ makefile = makefile.replace(key, mk_data[key])
+ return makefile
+
def mkGenerator(project_info, makefile):
"""
Generates the mk file for the current project.
"""
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])
mk_data["$main"] = os.path.basename(project_info.info("PROJECT_PATH")) + "/main.c"
- mk_data["$programmercpu"] = project_info.info("CPU_INFOS")["PROGRAMMER_CPU"]
- mk_data["$flashscript"] = project_info.info("CPU_INFOS")["FLASH_SCRIPT"]
- mk_data["$debugscript"] = project_info.info("CPU_INFOS")["DEBUG_SCRIPT"]
for key in mk_data:
while makefile.find(key) != -1:
makefile = makefile.replace(key, mk_data[key])
for file in information["hw"]:
if file.endswith(".c"):
module_files |= set([hwdir + "/" + os.path.basename(file)])
- for file_dependency in information["depends"]:
- if file_dependency in files:
+ for file_dependency in information["depends"] + tuple(files.keys()):
dependencyCFiles, dependencySFiles = findModuleFiles(file_dependency, project_info)
dependency_files |= set(dependencyCFiles)
asm_files |= set(dependencySFiles)
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
return name
def loadSourceTree(project):
- fileList = [f for f in os.walk(project.info("SOURCES_PATH"))]
+ # Index only the SOURCES_PATH/bertos content
+ bertos_sources_dir = os.path.join(project.info("SOURCES_PATH"), 'bertos')
+ if os.path.exists(bertos_sources_dir):
+ fileList = [f for f in os.walk(bertos_sources_dir)]
+ else:
+ fileList = []
project.setInfo("FILE_LIST", fileList)
+_cached_queries = {}
+
def findDefinitions(ftype, project):
+ definitions = _cached_queries.get(ftype, None)
+ if definitions is not None:
+ return definitions
L = project.info("FILE_LIST")
definitions = []
for element in L:
for filename in element[2]:
if fnmatch.fnmatch(filename, ftype):
definitions.append((filename, element[0]))
+ _cached_queries[ftype] = definitions
return definitions
def loadCpuInfos(project):
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
"long": boolean indicating if the num is a long
"unsigned": boolean indicating if the num is an unsigned
"value_list": the name of the enum for enum parameters
+ "conditional_deps": the list of conditional dependencies for boolean parameters
"""
configuration_infos = {}
configuration_infos["paramlist"] = []
configuration_infos[name]["value"].find("U") != -1):
configuration_infos[name]["informations"]["unsigned"] = True
configuration_infos[name]["value"] = configuration_infos[name]["value"].replace("U", "")
+ if "conditional_deps" in configuration_infos[name]["informations"]:
+ if (type(configuration_infos[name]["informations"]["conditional_deps"]) == str or
+ type(configuration_infos[name]["informations"]["conditional_deps"]) == unicode):
+ configuration_infos[name]["informations"]["conditional_deps"] = (configuration_infos[name]["informations"]["conditional_deps"], )
+ elif type(configuration_infos[name]["informations"]["conditional_deps"]) == tuple:
+ pass
+ else:
+ configuration_infos[name]["informations"]["conditional_deps"] = ()
configuration_infos[name]["description"] = description
configuration_infos[name]["brief"] = brief
return configuration_infos
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):