Adding support for QTM files with header information included.

Author: alexarje

mocaptoolbox.py

@@ -12,51 +12,70 @@
 
 #---- The mocap data class ----
 class data:
-    
+
     timederOrder = 0;
     filename = ''
-            
+
     #--- initializer ---
-    def __init__(self, freq, nMarkers, nFrames, markerNames, data, filename):
+    def __init__(self, freq, nMarkers, nCameras, nFrames, markerNames, data, filename, timestamp, trajectoryTypes, header):
         self.freq = freq
         self.nMarkers = nMarkers
+        self.nCameras = nCameras
         self.nFrames = nFrames
         self.markerNames = markerNames
         self.data = data
         self.filename = filename
+        self.timestamp = timestamp
+        self.trajectoryTypes = trajectoryTypes
+        self.header = header
 
     #--- class method to read tsv files ---
     @classmethod
     def readTsv(cls, filename):
-        
+
         file = open(filename)
-        
+                
         #reading tsv header until marker names line
         line = 'asdfasdfa'
         while line[0:8] != 'MARKER_N':
             line = file.readline()
-            
+ 
             if line[0:8] == 'NO_OF_FR':
-                nFrames     = int(line.split()[1])
+                nFrames      = int(line.split()[1])
+            if line[0:8] == 'NO_OF_CA':
+                nCameras     = int(line.split()[1])
             if line[0:8] == 'NO_OF_MA':
                 nMarkers     = int(line.split()[1])
             if line[0:8] == 'FREQUENC':
-                freq     = float(line.split()[1])
-        
-        #line is now marker names line
+                freq         = float(line.split()[1])
+            if line[0:8] == 'TIME_STA':
+                timestamp    = str(line.split()[1:4])
+ 
+        #reading marker names
         markerNames = line.split()[1:]
-        
+
+        #reading additional lines (not in all QTM-TSV files)
+        line = file.readline()
+
+        if line[0:8] == 'TRAJECTO':
+            trajectoryTypes = str(line.split()[1:])
+
+        line = file.readline()
+
+        if line[0:5] == 'Frame':
+            header = line.split("\t")
+                        
         #the rest is marker data
-        data = np.genfromtxt(file)
-        
+        data = np.genfromtxt(file, delimiter='\t', invalid_raise = False)
+
         file.close
-            
-        return cls(freq, nMarkers, nFrames, markerNames, data, filename)
+
+        return cls(freq, nMarkers, nCameras, nFrames, markerNames, data, filename, timestamp, trajectoryTypes, header)
 
 
 #---- A norm data class ---- ### this class should probably be removed see todo comment at the top.. 
 class normData:
-        
+
     #--- initializer ---
     def __init__(self, data):
         self.freq = data.freq