FifFile

Complexity

Total Complexity

4

Size/Duplication

Total Lines	44
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	3	1
B	inspect()	0	24	2
A	attach()	0	13	1

from __future__ import division
from datetime import datetime, timedelta
from niprov.basefile import BaseFile
from niprov.libraries import Libraries


class FifFile(BaseFile):

    def __init__(self, location, **kwargs):
        super(FifFile, self).__init__(location, **kwargs)
        self.libs = self.dependencies.getLibraries()

    def inspect(self):
        provenance = super(FifFile, self).inspect()
        #TODO try statement loop to check & inspect different fif filetypes
        """ try:
                img = self.libs.mne.io.Raw(self.path, allow_maxshield=True)
                except ValueError:
                    pass
                else:
                    inspect file
                    Return
        """
        img = self.libs.mne.io.Raw(self.path, allow_maxshield=True)
        sub = img.info['subject_info']
        if sub is not None:
            provenance['subject'] = sub['first_name']+' '+sub['last_name']
        provenance['project'] = img.info['proj_name']
        acqTS = img.info['meas_date'][0]
        provenance['acquired'] = datetime.fromtimestamp(acqTS)
        T = img.last_samp - img.first_samp + 1
        provenance['dimensions'] = [img.info['nchan'], T]
        provenance['sampling-frequency'] = img.info['sfreq']
        provenance['duration'] = timedelta(seconds=T/img.info['sfreq'])
        provenance['modality'] = 'MEG'
        return provenance

    def attach(self, form='json'):
        """
        Attach the current provenance to the file by appending it as a
        json-encoded string to the 'description' header field.

        Args:
            form (str): Data format in which to serialize provenance. Defaults 
                to 'json'.
        """
        info = self.libs.mne.io.read_info(self.path)
        provstr = self.getProvenance(form)
        info['description'] = info['description']+' NIPROV:'+provstr
        self.libs.mne.io.write_info(self.path, info)