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

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

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

    xmlData.write('<?xml version="1.0" encoding="UTF-8"?>' + "\n")
    xmlData.write('<!DOCTYPE properties SYSTEM "cpf.dtd">' + "\n")
    xmlData.write('<properties>' + "\n")
    xmlData.write('    <struct name="EventsToActOn" type="string[]">' + "\n")
    xmlData.write('             <description>The names of the RobonetDataElement groups that can trigger this FloatPorter to read data.</description>' + "\n")
    xmlData.write('     </struct>' + "\n")
    rowNum = 0
    elementNum = 0
    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

    # do read elements
    xmlData.write('     <struct name="ReadElementNames" type="string[]">' + "\n")
    xmlData.write('             <description>The names of the RobonetDataElements this Porter will read and publish.</description>' + "\n")
    rowNum = 0
    for row in data:
        if rowNum == 0:
            elementNum = 0
        else:
            if(row[iName].strip().replace(' ', '_')):
                if(row[iDirection].lower() == "status"):
                    xmlData.write('             <simple name="Element' + str(elementNum) +'"   type="string"><value>r2/bat_pack/bms/' + str(row[iName].strip().replace(' ', '_')) + '</value> </simple>\n')
                    elementNum += 1
        rowNum +=1
    xmlData.write('     </struct>' + "\n")

    # do write elements
    xmlData.write('    <struct name="WriteElementNames" type="string[]">' + "\n")
    xmlData.write('             <description>The names of the RobonetDataElements this Porter will write subscribed data to.</description>' + "\n")
    rowNum = 0
    for row in data:
        if rowNum == 0:
            elementNum = 0
        else:
            if(row[iName].strip().replace(' ', '_')):
                if(row[iDirection].lower() == "control"):
                    xmlData.write('             <simple name="Element' + str(elementNum) +'"   type="string"><value>r2/bat_pack/bms/' + str(row[iName].strip().replace(' ', '_')) + '</value> </simple>\n')
                    elementNum += 1
        rowNum +=1
    xmlData.write('     </struct>' + "\n")

    #finish the file
    xmlData.write('</properties>')
    xmlData.close()

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