diff python/storage.py @ 895:739acd338cc0

added script to extract camera info from tacal file by Lund University (T analyst)
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 01 Jun 2017 14:54:11 -0400
parents ff92801e5c54
children 1466a63dd1cf
line wrap: on
line diff
--- a/python/storage.py	Thu Jun 01 10:33:07 2017 -0400
+++ b/python/storage.py	Thu Jun 01 14:54:11 2017 -0400
@@ -1209,6 +1209,22 @@
         
     out.close()
 
+def loadPinholeCameraModel(filename, tanalystFormat = True):
+    '''Loads the data from a file containing the camera parameters
+    (pinhole camera model, http://docs.opencv.org/2.4/modules/calib3d/doc/camera_calibration_and_3d_reconstruction.html)
+    and returns a dictionary'''
+    if tanalystFormat:
+        f = openCheck(filename, quitting = True)
+        content = getLines(f)
+        cameraData = {}
+        for l in content:
+            tmp = l.split(':')
+            cameraData[tmp[0]] = float(tmp[1].strip().replace(',','.'))
+        return cameraData
+    else:
+        print('Unknown camera model (not tanalyst format')
+        return None
+
 def savePositionsToCsv(f, obj):
     timeInterval = obj.getTimeInterval()
     positions = obj.getPositions()