Mercurial Hosting > traffic-intelligence
comparison scripts/play-synced-video.py @ 821:26daf35180ad
finished modification and demo script to replay synchronized video (with same frame rate)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 22 Jun 2016 16:36:12 -0400 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
820:e73e7b644428 | 821:26daf35180ad |
---|---|
1 #! /usr/bin/env python | |
2 | |
3 import sys, argparse, os.path | |
4 import cvutils, utils | |
5 from metadata import createDatabase, Site, VideoSequence | |
6 from datetime import datetime, timedelta | |
7 | |
8 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) | |
10 parser.add_argument('-n', dest = 'siteId', help = 'site id or site name', required = True) | |
11 parser.add_argument('-t', dest = 'startTime', help = 'time to start playing (format %Y-%m-%d %H:%M:%S, eg 2011-06-22 10:00:39)', required = True) | |
12 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) | |
14 parser.add_argument('-s', dest = 'step', help = 'display every s image', default = 1, type = int) | |
15 | |
16 args = parser.parse_args() | |
17 | |
18 session = createDatabase(args.metadataFilename) | |
19 | |
20 if str.isdigit(args.siteId): | |
21 site = session.query(Site).filter(Site.idx == int(args.siteId)).first() | |
22 else: | |
23 site = session.query(Site).filter(Site.description.like('%'+args.siteId+'%')).first() | |
24 | |
25 if site is None: | |
26 print('Site {} was not found in {}. Exiting'.format(args.siteId, args.metadataFilename)) | |
27 sys.exit() | |
28 | |
29 dirname = os.path.split(args.metadataFilename)[0] | |
30 | |
31 startTime = datetime.strptime(args.startTime, utils.datetimeFormat) | |
32 videoSequences = session.query(VideoSequence).filter(VideoSequence.site == site).filter(VideoSequence.startTime <= startTime).all() | |
33 videoSequences = [v for v in videoSequences if v.containsInstant(startTime)] | |
34 filenames = [dirname+os.path.sep+v.getVideoSequenceFilename() for v in videoSequences] | |
35 firstFrameNums = [v.getFrameNum(startTime) for v in videoSequences] | |
36 | |
37 cvutils.playVideo(filenames, [v.cameraView.description for v in videoSequences], firstFrameNums, args.frameRate, rescale = args.rescale, step = args.step) |