Add the Makefile template in the mktemplates directory

[bertos.git] / wizard / bertos_utils.py

diff --git a/wizard/bertos_utils.py b/wizard/bertos_utils.py
index df26613f66785a662666a44edb5a076719566775..05fd35465c5901119c9fea424dfcc409e9e204e2 100644 (file)
--- a/wizard/bertos_utils.py
+++ b/wizard/bertos_utils.py
@@ -23,15 +23,70 @@ 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(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 "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 + "/" + "project.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"]
@@ -213,4 +268,10 @@ def loadDefineListsDict(path):
     defineListsDict = {}
     for filename, path in findDefinitions("*.h", path):
         defineListsDict.update(loadDefineLists(path + "/" + filename))
-    return defineListsDict
\ No newline at end of file
+    return 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>#define\s+" + parameter + r"\s+)([^\s]+)", r"\g<define>" + value, string)