Mercurial Hosting > traffic-intelligence
annotate python/offset-trajectories.py @ 231:249d65ff6c35
merged modifications for windows
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Mon, 02 Jul 2012 23:49:39 -0400 |
parents | 41a5853ec495 |
children |
rev | line source |
---|---|
161
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
1 #! /usr/bin/env python |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
2 import sys |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
3 |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
4 import ubc_utils |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
5 import utils |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
6 |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
7 if len(sys.argv) < 3: |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
8 print("Usage: %s filename offsetframes") |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
9 sys.exit() |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
10 |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
11 nFrames = int(sys.argv[2]) |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
12 |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
13 def modifyLines(objectNum, lines): |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
14 result = lines |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
15 tmp = lines[0].split(" ") |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
16 firstInstant = int(tmp[1])+nFrames |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
17 lastInstant = int(tmp[2])+nFrames |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
18 tmp[1] = str(firstInstant) |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
19 tmp[2] = str(lastInstant) |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
20 result[0] = " ".join(tmp) |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
21 return result |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
22 |
41a5853ec495
added utility to offset the time intervales of complete trajectory files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
23 ubc_utils.modifyTrajectoryFile(modifyLines, sys.argv[1], sys.argv[1]+".new") |