Mercurial > hg > nsaunier > traffic-intelligence

comparison scripts/play-synced-videos.py @ 971:9897a13772fb

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added utils to load video sequence in metadata
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 13 Dec 2017 14:06:20 -0500
parents 75530d8c0090
children 7463c9bc846b
comparison
equal deleted inserted replaced
970:bf401567a933 971:9897a13772fb
1 #! /usr/bin/env python 1 #! /usr/bin/env python
2 2
3 import sys, argparse, os.path 3 import sys, argparse, os.path
4 import cvutils, utils 4 import cvutils, utils
5 from metadata import createDatabase, Site, CameraView, VideoSequence 5 from metadata import createDatabase, Site, CameraView, VideoSequence, getSite
6 from datetime import datetime, timedelta 6 from datetime import datetime, timedelta
7
8 timeConverter = utils.TimeConverter()
7 9
8 parser = argparse.ArgumentParser(description='The program displays several views of the same site synchronously.') 10 parser = argparse.ArgumentParser(description='The program displays several views of the same site synchronously.')
9 parser.add_argument('-i', dest = 'metadataFilename', help = 'name of the metadata file', required = True) 11 parser.add_argument('-i', dest = 'metadataFilename', help = 'name of the metadata file', required = True)
10 parser.add_argument('-n', dest = 'siteId', help = 'site id or site name', required = True) 12 parser.add_argument('-n', dest = 'siteId', help = 'site id or site name', required = True)
11 parser.add_argument('-f', dest = 'startTime', help = 'time to start playing (format %%Y-%%m-%%d %%H:%%M:%%S, eg 2011-06-22 10:00:39)', required = True) 13 parser.add_argument('-f', dest = 'startTime', help = 'time to start playing (format %%Y-%%m-%%d %%H:%%M:%%S, eg 2011-06-22 10:00:39)', required = True, type = timeConverter.convert)
12 parser.add_argument('--fps', dest = 'frameRate', help = 'approximate frame rate to replay', default = -1, type = float) 14 parser.add_argument('--fps', dest = 'frameRate', help = 'approximate frame rate to replay', default = -1, type = float)
13 parser.add_argument('-r', dest = 'rescale', help = 'rescaling factor for the displayed image', default = 1., type = float) 15 parser.add_argument('-r', dest = 'rescale', help = 'rescaling factor for the displayed image', default = 1., type = float)
14 parser.add_argument('-s', dest = 'step', help = 'display every s image', default = 1, type = int) 16 parser.add_argument('-s', dest = 'step', help = 'display every s image', default = 1, type = int)
15 17
16 args = parser.parse_args() 18 args = parser.parse_args()
17 19
18 session = createDatabase(args.metadataFilename) 20 session = createDatabase(args.metadataFilename)
19 21
20 site = Site.getSite(session, args.siteId) 22 site = getSite(session, args.siteId)
21 if site is None: 23 if site is None:
22 print('Site {} was not found in {}. Exiting'.format(args.siteId, args.metadataFilename)) 24 print('Site {} was not found in {}. Exiting'.format(args.siteId, args.metadataFilename))
23 sys.exit() 25 sys.exit()
24 else: 26 else:
25 site = site[0] 27 site = site[0]