PowCon_MLVDS_register_map_to_numbergui.py

Go to the documentation of this file.

# BMS_MLVDS_register_map_to_token.py
# First row of the csv file must be header!
import sys, getopt
import csv
import re

def parseRegisterDescription(desc):
    #print('parsing ' + desc)
    regex = re.compile("([0-9a-zA-Z_]+)\(([0-9]+)\:([0-9]+)\)")
    registers = desc.split(' ')
    #print('splint into ' + str(len(registers)) + ' pieces')
    output = []
    for reg in registers:
        #print('finding regex match for ' + reg)
        try:
            match = regex.match(reg)
            mydict = {"id": match.group(1),
                      "size": int(match.group(2)) - int(match.group(3)) +1,
                      "index": int(match.group(3))}
            output.append(mydict)
        except AttributeError:
            #print('no match found')
            pass
    return output    

def main(argv):
    csvFile = 'PowCon_MLVDS_register_map_oct28.csv'
    statusFile = 'NumberGuiConfigPowConStatus.xml'
    controlFile = 'NumberGuiConfigPowConControl.xml'
    try:
        opts, args = getopt.getopt(argv, "hi:c:s:",["input=", "control=", "status="])
    except getopt.GetoptError:
        print("PowCon_MLVDS_register_map_to_numbergui.py -i <inputfile> -c <controlfile> -s <statusfile>")
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            print 'PowCon_MLVDS_register_map_to_numbergui.py -i <inputfile> -c <controlfile> -s <statusfile>'
            sys.exit()
        elif opt in ("-i", "--input"):
            csvFile = arg
        elif opt in ("-c", "--control"):
            controlFile = arg
        elif opt in ("-s", "--status"):
            statusFile = arg

    print("Opening " + csvFile)
    data = list( csv.reader(open(csvFile)) )

    if(len(data)>0):
        tags = data[0]
        # replace spaces w/ underscores in tag names
        for i in range(len(tags)):
            tags[i] = tags[i].strip().replace(' ', '_')
        if("name" in tags):
            iName = tags.index("name")
            print("found name tag at index: " + str(iName))
        else:
            print("can not find name tag")
            return    
        if("ROBONET_type" in tags):
            iDirection = tags.index("ROBONET_type")
            print("found ROBONET_type tag at index: " + str(iDirection))
        else:
            print("can not find ROBONET_type tag")
            return
        if("registers" in tags):
            iRegisters = tags.index("registers")
            print("found registers tag at index: " + str(iRegisters))
        else:
            print("can not find registers tag")
            return
    else:
        print("can not find header!")
        return

    # do status elements
    xmlFile = statusFile
    print("Writing output to " + xmlFile)
    xmlData = open(xmlFile, 'w')

    xmlData.write('<?xml version="1.0" encoding="ISO-8859-1"?>' + "\n")
    xmlData.write('<NumberGuiConfiguration>' + "\n")
    xmlData.write('    <Group id="r2/bat_pack/powcon">' + "\n")
    rowNum = 0
    for row in data:
        if rowNum == 0:
            pass
        else:
            if(row[iName].strip().replace(' ', '_')):
                #registers = parseRegisterDescription(row[iRegisters])
                if(row[iDirection].lower() == "status"):
                    xmlData.write('        <Element id="' + str(row[iName].strip().replace(' ', '_')) + '" type="float" />\n')
        rowNum +=1
    xmlData.write('    </Group>' + "\n")
    xmlData.write('</NumberGuiConfiguration>')
    xmlData.close()

    # do write elements
    xmlFile = controlFile
    print("Writing output to " + xmlFile)
    xmlData = open(xmlFile, 'w')

    xmlData.write('<?xml version="1.0" encoding="ISO-8859-1"?>' + "\n")
    xmlData.write('<NumberGuiConfiguration>' + "\n")
    xmlData.write('    <Group id="r2/bat_pack/powcon">' + "\n")
    rowNum = 0
    for row in data:
        if rowNum == 0:
            pass
        else:
            if(row[iName].strip().replace(' ', '_')):
                #registers = parseRegisterDescription(row[iRegisters])
                if(row[iDirection].lower() == "control"):
                    xmlData.write('        <Element id="' + str(row[iName].strip().replace(' ', '_')) + '" type="float" />\n')
        rowNum +=1
    xmlData.write('    </Group>' + "\n")
    xmlData.write('</NumberGuiConfiguration>')
    xmlData.close()

if __name__ == "__main__":
   main(sys.argv[1:])