Merge pull request #3 from LolloCappo/master

Add SFMOV package for reading infrared data from FLIR Cameras

Author: jankoslavic

docs/source/index.rst

@@ -23,6 +23,7 @@ The ``io`` module is imported as follows:
    uff
    lvm
    mraw
+   sfmov
 
 
 .. _GitHub: https://github.com/sdypy/sdypy-io

docs/source/sfmov.rst

@@ -0,0 +1,24 @@
+SFMOV file format
+=================
+
+FLIR SFMOV File Reader.
+
+The ``sfmov`` module provides a direct link to the `pysfmov <https://github.com/LolloCappo/pysfmov>`_ package.
+
+A simple example:
+
+.. code-block:: python
+
+    # Import the module
+    import pysfmov 
+
+    # Set the path and the filename
+    filename = './data/rec.sfmov' 
+
+    # Get data from sfmov file
+    data = pysfmov.get_data(filename) 
+
+    # Get meta data from sfmov file
+    meta_data = pysfmov.get_meta_data(filename)
+
+For reference, see the `pyMRAW <https://github.com/LolloCappo/pysfmov>`_.

sdypy/io/__init__.py

@@ -7,3 +7,4 @@
 import pyuff as uff
 import lvm_read as lvm
 import pyMRAW as mraw
+from sfmov import sfmovcd 

sdypy/io/sfmov/__init__.py

@@ -0,0 +1,49 @@
+import numpy as np
+
+def get_meta_data(filename):
+    """
+    Function to open sfmov image sequences (exported from FLIR ResearchIR) and obtain metadata
+    
+    Arguments:
+        filename {str} -- path to the file
+    
+    Returns:
+        dict -- a dictionary of metadata
+    """
+    with open(filename, 'rt', errors='ignore') as f:
+        meta = {}
+        for line in f:
+            if line[:11] == 'saf_padding':
+                break
+            a = line[:-1].split(' ')
+            meta[a[0]] = a[1]
+
+    int_values = ['xPixls', 'yPixls', 'NumDPs']
+
+    for i in int_values:
+        meta[i] = int(meta[i])
+
+    return meta
+
+    
+def get_data(filename):
+    """
+    Function to open sfmov image sequences (exported from FLIR ResearchIR) and obtain raw data
+    
+    Arguments:
+        filename {str} -- path to the file
+    
+    Returns:
+        data -- raw data
+    """
+    meta = get_meta_data(filename=filename)
+    f = open(filename,'rb') 
+    f.seek(f.read().find(b'DATA')+6)
+    
+    if meta['DaType'] == 'Flt32':
+        ty = np.float32
+    else:
+        ty = np.uint16
+        
+    data = np.fromfile(f, dtype=ty).reshape(-1, meta['yPixls'], meta['xPixls'])
+    return data