Mercurial Hosting > traffic-intelligence
view python/ubc_utils.py @ 56:61fe73df2d36
created new package, moved Interaction class and created Crossing class
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 27 Oct 2010 18:32:27 -0400 |
parents | 0a5bdbf0d1b4 |
children | b8b3768f8d54 |
line wrap: on
line source
#! /usr/bin/env python '''Various utilities to load data saved by the UBC tool(s)''' import utils from moving import MovingObject, TimeInterval, Trajectory __metaclass__ = type userTypeNames = ['car', 'pedestrian', 'twowheels', 'bus' 'truck'] fileTypeNames = ['feature', 'object', 'prototype', 'contoursequence'] def getFileType(s): 'Finds the type in fileTypeNames' for fileType in fileTypeNames: if s.find(fileType)>0: return fileType return '' def isFileType(s, fileType): return (s.find(fileType)>0) def saveTrajectoryUserTypes(inFilename, outFilename, objects): '''The program saves the objects, by just copying the corresponding trajectory and velocity data from the inFilename, and saving the characteristics in objects (first line) into outFilename''' infile = utils.openCheck(inFilename) outfile = utils.openCheck(outFilename,'w') if (inFilename.find('features') >= 0) or (not infile) or (not outfile): return lines = utils.getLines(infile) objNum = 0 # in inFilename while lines != []: # find object in objects (index i) i = 0 while (i<len(objects)) and (objects[i].num != objNum): i+=1 if i<len(objects): l = lines[0].split(' ') l[3] = str(objects[i].userType) outfile.write(' '.join(l)+'\n') for l in lines[1:]: outfile.write(l+'\n') outfile.write(utils.delimiterChar+'\n') # next object objNum += 1 lines = utils.getLines(infile) print('read {0} objects'.format(objNum)) def loadTrajectories(filename, nObjects = -1): '''Loads trajectories''' file = utils.openCheck(filename) if (not file): return [] objects = [] objNum = 0 objectType = getFileType(filename) lines = utils.getLines(file) while (lines != []) and ((nObjects<0) or (objNum<nObjects)): l = lines[0].split(' ') parsedLine = [int(n) for n in l[:4]] obj = MovingObject(num = objNum, timeInterval = TimeInterval(parsedLine[1],parsedLine[2])) #add = True if len(lines) >= 3: obj.positions = Trajectory.load(lines[1], lines[2]) if len(lines) >= 5: obj.velocities = Trajectory.load(lines[3], lines[4]) if objectType == 'object': obj.userType = parsedLine[3] obj.nObjects = float(l[4]) obj.featureNumbers = [int(n) for n in l[5:]] # load contour data if available if len(lines) >= 6: obj.contourType = utils.line2Floats(lines[6]) obj.contourOrigins = Trajectory.load(lines[7], lines[8]) obj.contourSizes = Trajectory.load(lines[9], lines[10]) elif objectType == 'prototype': obj.userType = parsedLine[3] obj.nMatchings = int(l[4]) if len(lines) != 2: objects.append(obj) objNum+=1 else: print("Error two lines of data for feature %d"%(f.num)) lines = utils.getLines(file) file.close() return objects def loadCollisionPoints(filename, nPoints = -1): '''Loads collision points and returns a dict with keys as a pair of the numbers of the two interacting objects''' file = utils.openCheck(filename) if (not file): return [] points = {} num = 0 lines = utils.getLines(file) while (lines != []) and ((nPoints<0) or (num<nPoints)): parsedLine = [int(n) for n in lines[0].split(' ')] protagonistNums = (parsedLine[0], parsedLine[1]) points[protagonistNums] = [[float(n) for n in lines[1].split(' ')], [float(n) for n in lines[2].split(' ')]] num+=1 lines = utils.getLines(file) file.close() return points