Mercurial > hg > nsaunier > traffic-intelligence
comparison python/offset-trajectories.py @ 161:41a5853ec495
added utility to offset the time intervales of complete trajectory files
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Tue, 20 Sep 2011 13:49:01 -0400 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 160:b0719b3ad3db | 161:41a5853ec495 |
|---|---|
| 1 #! /usr/bin/env python | |
| 2 import sys | |
| 3 | |
| 4 import ubc_utils | |
| 5 import utils | |
| 6 | |
| 7 if len(sys.argv) < 3: | |
| 8 print("Usage: %s filename offsetframes") | |
| 9 sys.exit() | |
| 10 | |
| 11 nFrames = int(sys.argv[2]) | |
| 12 | |
| 13 def modifyLines(objectNum, lines): | |
| 14 result = lines | |
| 15 tmp = lines[0].split(" ") | |
| 16 firstInstant = int(tmp[1])+nFrames | |
| 17 lastInstant = int(tmp[2])+nFrames | |
| 18 tmp[1] = str(firstInstant) | |
| 19 tmp[2] = str(lastInstant) | |
| 20 result[0] = " ".join(tmp) | |
| 21 return result | |
| 22 | |
| 23 ubc_utils.modifyTrajectoryFile(modifyLines, sys.argv[1], sys.argv[1]+".new") |