Subversion Repositories HomeAutomation

#!/bin/env python

# Written by Mattias Runge 2008-05-13

import os

import sys

import getopt

moddir = '../../module'

localmoddir = 'modules'

def getModules():

    modules = []

    for fileName in os.listdir(moddir):

        if fileName[0] != '.':

            modules.append(fileName)

    return modules

def getLocalModules():

    modules = []

    for fileName in os.listdir(localmoddir):

        if fileName[0] != '.':

            modules.append(fileName)

    return modules

def parseModuleIdFromFile(moduleName, fileName):

    config_inc = open(fileName, 'r')

    config_inc_lines = config_inc.readlines()

    config_inc.close()

    for line in config_inc_lines:

        line = line.strip("\n").strip(" ")

        if line.find(moduleName + "_ID") != -1:

            parts = line.split("=")

            return parts[1].strip(" ")

    return -1;

def getFreeModuleId(moduleName):

    takenIds = []

    for applicationName in os.listdir("../"):

        if applicationName[0] != '.' and os.path.exists("../" + applicationName + "/config.inc"):

            takenId = parseModuleIdFromFile(moduleName, "../" + applicationName + "/config.inc")

            if takenId != -1 and takenId != "<ID>":

                takenIds.append(int(takenId, 16))

    if len(takenIds) == 0:

        return "0x01"

    takenIds.sort()

    lastId = 0

    for takenId in takenIds:

        if takenId != lastId+1:

            return hex(lastId+1)

    return hex(lastId+1)

def compileMainFile():

    print "Compiling new main.c..."

    main_c_template = open('src/main.c.template', 'r')

    main_c_template_lines = main_c_template.readlines()

    main_c_template.close()

    modules = getLocalModules()

    main_c = open('src/main.c', 'w')

    for main_c_template_line in main_c_template_lines:

        pos_include = main_c_template_line.find('%INCLUDE')

        pos_init = main_c_template_line.find('%INIT')

        pos_process = main_c_template_line.find('%PROCESS')

        pos_list = main_c_template_line.find('%LIST')

        pos_handlemsg = main_c_template_line.find('%HANDLEMSG')

        if pos_include != -1:

            for moduleName in modules:

                main_c.write(main_c_template_line[:pos_include] + "#include \"../modules/" + moduleName + "/" + moduleName + ".h\"\n")

        elif pos_init != -1:

            for moduleName in modules:

                main_c.write(main_c_template_line[:pos_init] + moduleName + "_Init();\n")

        elif pos_process != -1:

            for moduleName in modules:

                main_c.write(main_c_template_line[:pos_process] + moduleName + "_Process();\n")

        elif pos_list != -1:

            count = 1

            for moduleName in modules:

                main_c.write(main_c_template_line[:pos_list] + moduleName + "_List(" + str(count) + ");\n")

                count += 1

        elif pos_handlemsg != -1:

            for moduleName in modules:

                main_c.write(main_c_template_line[:pos_handlemsg] + moduleName + "_HandleMessage(&rxMsg);\n")

        else:

            main_c.write(main_c_template_line)

    main_c.close()

    print "Creation of main.c complete"

def readConfigSection(fileName, sectionName):

    fileInstance = open(fileName, 'r')

    lines = fileInstance.readlines()

    fileInstance.close()

    sectionLines = []

    inSection = False

    for line in lines:

        if not inSection:

            if line.find("## Section " + sectionName) != -1:

                inSection = True

                sectionLines.append(line.strip("\n"))

        else:

            if line.find("## End section " + sectionName) != -1:

                inSection = False

            sectionLines.append(line.strip("\n"))

    return sectionLines

def updateConfigFile():

    print "Updating config.inc..."

    modules = getLocalModules()

    moduleSections = {}

    applicationSection = []

    if not os.path.exists("config.inc"):

        applicationSection = readConfigSection("src/config.inc.template", "application")

        for moduleName in modules:

            moduleSections[moduleName] = readConfigSection(localmoddir + "/" + moduleName + "/config.inc.template", moduleName)

    else:

        applicationSection = readConfigSection("config.inc", "application")

        for moduleName in modules:

            moduleSections[moduleName] = readConfigSection("config.inc", moduleName)

            if len(moduleSections[moduleName]) <= 1:

                moduleSections[moduleName] = readConfigSection(localmoddir + "/" + moduleName + "/config.inc.template", moduleName)

    timers = 0

    for moduleName, moduleSection in moduleSections.iteritems():

        c = 0

        for line in moduleSection:

            if line.find("<ID>") != -1:

                line = line.replace("<ID>", getFreeModuleId(moduleName))

            elif line.find("<TIMER>") != -1:

                line = line[:line.find("=")+1] + hex(timers) + " /* <TIMER> -- DO NOT REMOVE THIS COMMENT -- */"

                timers += 1

            moduleSection[c] = line;

            c += 1

    c = 0

    for line in applicationSection:

        if line.find("<TIMER>") != -1:

            line = line[:line.find("=")+1] + hex(timers) + " /* <TIMER> -- DO NOT REMOVE THIS COMMENT -- */"

            timers += 1

        elif line.find("<NUMBER_OF_TIMERS>") != -1:

            line = line.replace("<NUMBER_OF_TIMERS>", hex(timers))

        elif line.find("<NUMBER_OF_MODULES>") != -1:

            line = line.replace("<NUMBER_OF_MODULES>", hex(len(modules)))

        applicationSection[c] = line;

        c += 1

    config_inc = open('config.inc', 'w')

    for line in applicationSection:

        config_inc.write(line + "\n")

    for moduleName, moduleSection in moduleSections.iteritems():

        for line in moduleSection:

            config_inc.write(line + "\n")

    config_inc.close()

    print "Update of config.inc complete"

def compileSourcesFile():

    print "Compiling sources.inc..."

    uniqueSet = []

    sources_list_template = open('src/sources.list.template', 'r')

    sources_list_template_lines = sources_list_template.readlines()

    sources_list_template.close()

    for sources_list_template_line in sources_list_template_lines:

        if len(sources_list_template_line) > 0:

            uniqueSet.append(sources_list_template_line.strip("\n"))

    modules = getLocalModules()

    for moduleName in modules:

        sources_list = open(localmoddir + "/" + moduleName + "/sources.list", 'r')

        sources_list_lines = sources_list.readlines()

        sources_list.close()

        for sources_list_line in sources_list_lines:

            if len(sources_list_line) > 0:

                uniqueSet.append(sources_list_line.strip("\n"))

    sourcesString = "SOURCES = "

    #FIXME: Make uniqueSet actually unique

    for line in uniqueSet:

        sourcesString += line + " "

    sources_inc = open('sources.inc', 'w')

    sources_inc.write(sourcesString.strip(" ") + "\n")

    sources_inc.close()

    print "Creation of sources.inc complete"

def regenerateModules():

    print "Regenerating modules..."

    compileSourcesFile()

    compileMainFile()

    updateConfigFile()

    print "Regenerating modules complete"

def addModule(moduleName):

    print "Trying to add module " + moduleName

    try:

        os.symlink("../" + moddir + "/" + moduleName, localmoddir + "/" + moduleName)

        print "Added module successfully"

    except OSError, (errno, strerror):

        if errno == 17:

            print "A link, file or directory named " + moduleName + " is already present in the modules directory, not linking"

        else:

            raise

    print ""

    regenerateModules()

def delModule(moduleName):

    print "Trying to remove module " + moduleName

    try:

        os.unlink(localmoddir + "/" + moduleName)

        print "Removed module successfully"

    except OSError, (errno, strerror):

        if errno == 2:

            print "No such " + moduleName + " present in the modules directory, not unlinking"

        else:

            raise

    print ""

    regenerateModules()

def usage():

    print "Options:"

    print "\t--regenare\t\tRegenarate modules and files"

    print "\t--list\t\t\tList avalible modules"

    print "\t--help\t\t\tPrint this help"

    print "\t--add=<module name>\tAdds a module to the application"

    print ""

def main():

    try:

        opts, args = getopt.getopt(sys.argv[1:], "hlrd", ["help", "list", "regenerate", "add=", "del="])

    except getopt.GetoptError, err:

        # print help information and exit:

        print str(err) # will print something like "option -a not recognized"

        usage()

        sys.exit(2)

    if len(sys.argv) == 1:

        usage()

        sys.exit()

    for o, a in opts:

        if o in ("-l", "--list"):

            print getModules()

        elif o in ("-h", "--help"):

            usage()

            sys.exit()

        elif o in ("-r", "--regenerate"):

            regenerateModules()

        elif o in ("--add"):

            if len(a) == 0:

                print "No modulename specified"

                usage()

                sys.exit()

            if a not in getModules():

                print a + " is not a recognized module name"

                sys.exit()

            addModule(a)

        elif o in ("--del"):

            if len(a) == 0:

                print "No modulename specified"

                usage()

                sys.exit()

            if a not in getModules():

                print a + " is not a recognized module name"

                sys.exit()

            delModule(a)

        else:

            assert False, "unhandled option"

if __name__ == "__main__":

    main()