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