Mercurial Hosting > traffic-intelligence
diff python/metadata.py @ 969:5d788d2e8ffc
work in progress
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 07 Dec 2017 17:03:09 -0500 |
parents | 32a34a143c27 |
children | bf401567a933 |
line wrap: on
line diff
--- a/python/metadata.py Thu Dec 07 15:13:06 2017 -0500 +++ b/python/metadata.py Thu Dec 07 17:03:09 2017 -0500 @@ -1,5 +1,5 @@ from datetime import datetime, timedelta -from os import path +from os import path, listdir, sep from math import floor from numpy import zeros, loadtxt, array @@ -8,8 +8,8 @@ from sqlalchemy.orm import relationship, backref, sessionmaker from sqlalchemy.ext.declarative import declarative_base -from utils import datetimeFormat, removeExtension -from cvutils import computeUndistortMaps +from utils import datetimeFormat, removeExtension, getExtension +from cvutils import computeUndistortMaps, videoFilenameExtensions from moving import TimeInterval """ @@ -363,7 +363,6 @@ eg somedirectory/montreal/ contains intersection1, intersection2, etc. The site names would be somedirectory/montreal/intersection1, somedirectory/montreal/intersection2, etc.''' - from os import listdir, path, sep sites = [] cameraViews = [] names = listdir(directoryName) @@ -377,3 +376,15 @@ session.add_all(cameraViews) session.commit() # TODO crawler for video files? + +def initializeVideos(session, site, cameraView, directoryName, startTime = None, datetimeFormat = None): + '''Initializes videos with time or tries to guess it from filename + directoryName should contain the videos to find and be the relative path from the site location''' + names = listdir(directoryName) + videoSequences = [] + for name in names: + extension = getExtension(name) + print(name, extension) + if extension in videoFilenameExtensions: + videoSequences.append(VideoSequence(directoryName+sep+name, startTime, None, cameraView, directoryName+sep+'.sqlite')) + return videoSequences