surfaces.txt

[NaN]

The script looks for list.txt files in root, creates surfaces.txt and updates list.txt.

Code:

import os

def locate(pattern, root=os.getcwd()):
    for path, dirs, files in os.walk(root):
        for filename in [os.path.abspath(os.path.join(path, filename)) for filename in files if filename.endswith(pattern)]:
            yield filename

root = 'E:/vdrift/vdrift-data/tracks'
pattern = 'list.txt'

for filename in locate(pattern, root):
    print filename
    objects = []
    object = []
    file = open(filename)
    line = file.readline().rstrip()
    while line != '':
        if '.joe' in line:
            if len(object): objects.append(object)
            object = []
            object.append(line)
        elif len(object) > 0 and line != '\n' and line != '' and not line.startswith('#'):
            object.append(line)
        line = file.readline()
    objects.append(object)
    file.close()
    #print 'objects: ' + str(len(objects))
    surfaces = []
    for object in objects:
        surface = [object[6], object[7], object[10], object[11], object[12], object[13]]
        i = 0
        while i < len(surfaces):
            if surface == surfaces[i]: break
            i = i + 1
        if i == len(surfaces):
            surfaces.append(surface)
        if len(object) == 14:
            object.extend(['0\n', '0\n', str(i)+'\n'])
        elif len(object) == 16:
            object.append(str(i)+'\n')
        else:
            object[16] = str(i)+'\n'
    #print 'surfaces: ' + str(len(surfaces))
    dir = os.path.dirname(filename)
    sfile = open(dir + '/surfaces.txt', 'w')
    id = 0
    for surface in surfaces:
        #print surface
        sfile.write('[surface-'+ str(id).zfill(2) + ']\n')
        #sfile.write('ID = ' + str(id) + '\n')
        sfile.write('Type = asphalt\n')
        sfile.write('BumpWaveLength = ' + str(surface[0]))
        sfile.write('BumpAmplitude = ' + str(surface[1]))
        sfile.write('FrictionNonTread = ' + str(surface[2]))
        sfile.write('FrictionTread = ' + str(surface[3]))
        sfile.write('RollResistanceCoefficient = ' + str(surface[4]))
        sfile.write('RollingDrag = ' + str(surface[5]))
        sfile.write('\n')
        id = id + 1
    sfile.close()
    #print 'write objects'
    file = open(filename, 'w')
    file.write('17\n')
    i = 0
    for object in objects:
        file.write('\n#entry ' + str(i) + '\n')
        i = i + 1
        for entry in object:
            file.write(str(entry))
    file.close()

By the way, I am currently updating the import_jpk/export_jpk scripts to support object parameters.