BMS_MLVDS_register_map_to_registers.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 getSize(arg):
    if("uint16_t" in arg or "int16_t" in arg):
        return 16
    elif("uint32_t" in arg or "int32_t" in arg):
        return 32
    elif("float" in arg):
        return 64

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 = 'BMS_MLVDS_register_map_oct28.csv'
    xmlFile = 'BmsRegisters.xml'
    try:
        opts, args = getopt.getopt(argv, "hi:o:",["input=", "output="])
    except getopt.GetoptError:
        print("BMS_MLVDS_register_map_to_registers.py -i <inputfile>")
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            print 'BMS_MLVDS_register_map_to_registers.py -i <inputfile>'
            sys.exit()
        elif opt in ("-i", "--input"):
            csvFile = arg
        elif opt in ("-o", "--output"):
            xmlFile = arg

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

    print("Writing to " + xmlFile)
    xmlData = open(xmlFile, 'w')

    xmlData.write('<?xml version="1.0" encoding="ISO-8859-1"?>' + "\n")
    xmlData.write('<RegisterData> ' + "\n")
    xmlData.write('    <Properties>' + "\n")
    xmlData.write('        <NodeType>BMS</NodeType>' + "\n")
    xmlData.write('    </Properties>' + "\n")
    xmlData.write('    <Registers>' + "\n")
    rowNum = 0
    for row in data:
        if rowNum == 0:
            tags = row
            # 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")
                break    
            if("data_type" in tags):
                iDataType = tags.index("data_type")
                print("found datatype tag at index: " + str(iDataType))
            else:
                print("can not find datatype tag")
                break
            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")
                break
            if("registers" in tags):
                iRegisters = tags.index("registers")
                print("found registers tag at index: " + str(iRegisters))
            else:
                print("can not find registers tag")
                break
            if("description" in tags):
                iDescription = tags.index("description")
                print("found description tag at index: " + str(iDescription))
            else:
                print("can not find description tag")
                iDescription = -1
        else:
            if(row[iName].strip().replace(' ', '_')):
                registers = parseRegisterDescription(row[iRegisters])
                size = getSize(row[iDataType])
                #print('size: ' + str(size))
                if(len(registers)>0):
                    xmlData.write('        <Register size="' + str(getSize(row[iDataType])) + '" type="' + str(row[iDirection]) + '" id="' + str(row[iName].strip().replace(' ', '_')) + '">\n')
                    xmlData.write('            <Certificate mask="0x0" value="0x0"/> \n')
                    xmlData.write('            <Entries> \n')
                    for reg in registers:
                        #print('reg: ' + str(reg))
                        xmlData.write('                <Entry size="' + str(reg['size']) + '" id="' + str(reg['id']) + '" index="' + str(reg['index']) + '">')
                        if(iDescription == -1):
                           xmlData.write('<Description>' + row[iDescription] + '</Description></Entry> \n')
                        else:
                           xmlData.write('<Description>' + row[iDescription] + '</Description></Entry> \n')
                    xmlData.write('            </Entries> \n')
                    xmlData.write('        </Register> \n')
        rowNum +=1
    xmlData.write('    </Registers>' + "\n")
    xmlData.write('</RegisterData>')

    xmlData.close()

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