wizard/bertos_utils.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3 #
   4 # Copyright 2008 Develer S.r.l. (http://www.develer.com/)
   5 # All rights reserved.
   6 #
   7 # $Id:$
   8 #
   9 # Author: Lorenzo Berni <duplo@develer.com>
  10 #
  11
  12 import os
  13 import fnmatch
  14 import glob
  15 import re
  16
  17 import const
  18
  19 def isBertosDir(directory):
  20    return os.path.exists(directory + "/VERSION")
  21
  22 def bertosVersion(directory):
  23    return open(directory + "/VERSION").readline().strip()
  24
  25 def createBertosProject(directory):
  26     if not os.path.isdir(directory):
  27         os.mkdir(directory)
  28     open(directory + "/project.bertos", "w")
  29
  30 def getSystemPath():
  31     path = os.environ["PATH"]
  32     if os.name == "nt":
  33         path = path.split(";")
  34     else:
  35         path = path.split(":")
  36     return path
  37
  38 def findToolchains(pathList):
  39     toolchains = []
  40     for element in pathList:
  41         for toolchain in glob.glob(element+ "/" + const.GCC_NAME):
  42             if not os.path.islink(toolchain):
  43                 toolchains.append(toolchain)
  44     return list(set(toolchains))
  45
  46 def getToolchainInfo(output):
  47     info = {}
  48     expr = re.compile("Target: .*")
  49     target = expr.findall(output)
  50     if len(target) == 1:
  51         info["target"] = target[0].split("Target: ")[1]
  52     expr = re.compile("gcc version [0-9,.]*")
  53     version = expr.findall(output)
  54     if len(version) == 1:
  55         info["version"] = version[0].split("gcc version ")[1]
  56     expr = re.compile("gcc version [0-9,.]* \(.*\)")
  57     build = expr.findall(output)
  58     if len(build) == 1:
  59         build = build[0].split("gcc version ")[1]
  60         build = build[build.find("(") + 1 : build.find(")")]
  61         info["build"] = build
  62     expr = re.compile("Configured with: .*")
  63     configured = expr.findall(output)
  64     if len(configured) == 1:
  65         info["configured"] = configured[0].split("Configured with: ")[1]
  66     expr = re.compile("Thread model: .*")
  67     thread = expr.findall(output)
  68     if len(thread) == 1:
  69         info["thread"] = thread[0].split("Thread model: ")[1]
  70     return info
  71
  72 def findDefinitions(ftype, path):
  73     L = os.walk(path)
  74     for element in L:
  75         for filename in element[2]:
  76             if fnmatch.fnmatch(filename, ftype):
  77                 yield (filename, element[0])
  78
  79 def loadCpuInfos(path):
  80     cpuInfos = []
  81     for definition in findDefinitions(const.CPU_DEFINITION, path):
  82         cpuInfos.append(getInfos(definition))
  83     return cpuInfos
  84
  85 def getInfos(definition):
  86     D = {}
  87     D.update(const.CPU_DEF)
  88     def include(filename, dict = D, directory=definition[1]):
  89         execfile(directory + "/" + filename, {}, D)
  90     D["include"] = include
  91     include(definition[0], D)
  92     D["CPU_NAME"] = definition[0].split(".")[0]
  93     D["DEFINITION_PATH"] = definition[1] + "/" + definition[0]
  94     del D["include"]
  95     return D
  96
  97 def getDefinitionBlocks(text):
  98     """
  99     Take a text and return a list of tuple (description, name-value).
 100     """
 101     block = []
 102     block_tmp = re.findall(r"/\*{2}\s*([^*]*\*(?:[^/*][^*]*\*+)*)/\s*#define\s+((?:[^/]*?/?)+)\s*?(?:/{2,3}[^<].*?)?$", text, re.MULTILINE)
 103     for comment, define in block_tmp:
 104         block.append((" ".join(re.findall(r"^\s*\*?\s*(.*?)\s*?(?:/{2}.*?)?$", comment, re.MULTILINE)).strip(), define))
 105     block += re.findall(r"/{3}\s*([^<].*?)\s*#define\s+((?:[^/]*?/?)+)\s*?(?:/{2,3}[^<].*?)?$", text, re.MULTILINE)
 106     block += [(comment, define) for define, comment in re.findall(r"#define\s*(.*?)\s*/{3}<\s*(.+?)\s*?(?:/{2,3}[^<].*?)?$", text, re.MULTILINE)]
 107     return block
 108
 109 def formatModuleNameValue(text):
 110     """
 111     Take the given string and return a tuple with the name of the parameter in the first position
 112     and the value in the second.
 113     """
 114     block = re.findall("\s*([^\s]+)\s*(.+?)\s*$", text, re.MULTILINE)
 115     return block[0]
 116
 117 def loadModuleInfos(path):
 118     """
 119     Return the module configurations found in the given path as a dict with the name as key
 120     and a dict as value. The value dict has the parameter name as key and has "value" and "description"
 121     fields.
 122     """
 123     moduleInfos = {}
 124     for definition in findDefinitions(const.MODULE_CONFIGURATION, path):
 125         moduleName = definition[0].replace("cfg_", "").replace(".h", "")
 126         moduleInfos[moduleName] = {}
 127         for description, define in getDefinitionBlocks(open(definition[1] + "/" + definition[0], "r").read()):
 128             name, value = formatModuleNameValue(define)
 129             moduleInfos[moduleName][name] = {}
 130             moduleInfos[moduleName][name]["value"] = value
 131             moduleInfos[moduleName][name]["description"] = description
 132     return moduleInfos