NiftiFile

Complexity

Total Complexity

3

Size/Duplication

Total Lines	29
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	inspect()	0	5	1
A	__init__()	0	5	1
A	attach()	0	15	1

from datetime import datetime
from niprov.basefile import BaseFile


class NiftiFile(BaseFile):

    def __init__(self, location, **kwargs):
        super(NiftiFile, self).__init__(location, **kwargs)
        self.libs = self.dependencies.getLibraries()
        self.polaroid = self.dependencies.getPolaroid()
        self.pictureCache = self.dependencies.getPictureCache()

    def attach(self, form='json'):
        """
        Attach the current provenance to the file by injecting it as a 
        json-encoded extension to the nifti header.

        Args:
            form (str): Data format in which to serialize provenance. Defaults 
                to 'json'.
        """
        img = self.libs.nibabel.load(self.path)
        provstr = self.getProvenance(form)
        ext = self.libs.nibabel.nifti1.Nifti1Extension('comment', provstr)
        hdr = img.get_header()
        hdr.extensions.append(ext)
        img.to_filename(self.path)

    def inspect(self):
        provenance = super(NiftiFile, self).inspect()
        img = self.libs.nibabel.load(self.path)
        self.polaroid.saveSnapshot(img.get_data(), to=self.pictureCache)
        return provenance