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)
        lastId = takenId

    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("TIMER_NUM_TIMERS=") != -1:
            line = "TIMER_NUM_TIMERS=" + hex(timers)
        elif line.find("NUMBER_OF_MODULES=") != -1:
            line = "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()