Camera

Complexity

Total Complexity

4

Size/Duplication

Total Lines	46
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	3	1
B	takeSnapshot()	0	26	2
A	saveSnapshot()	0	13	1

from niprov.dependencies import Dependencies
import webbrowser


class Camera(object):

    def __init__(self, dependencies):
        self.film = dependencies.getPictureCache()
        self.libs = dependencies.getLibraries()

    def saveSnapshot(self, data):
        """Plot an overview of the image and store it.

        Uses :class:`.PictureCache` as service that provides a file-like 
                handle to save the plotted picture to.
        Calls takeSnapshot() to do the actual plotting.

        Args:
            data (numpy.ndarray): Array of 2, 3 or 4 dimensions with image data.
        """
        newPicture = self.film.new()
        self.takeSnapshot(data, on=newPicture)
        self.film.keep(newPicture)

    def takeSnapshot(data, on='shouldnotbehere.png'):
        """Plot an overview of the image using matplotlib.pyplot.

        Args:
            data (numpy.ndarray): Array of 2, 3 or 4 dimensions with image data.
            on (str or file-like object): Where to save figure to.
        """
        plt = self.libs.pyplot

        ## 3D
        ndims = len(data.shape)
        sliceOrder = [1, 0, 2]

        fig, axs = plt.subplots(nrows=1, ncols=ndims)

        for d in range(ndims):
            slicing = [slice(None)]*ndims
            slicing[sliceOrder[d]] = int(data.shape[d]/2)
            axs[d].matshow(data[slicing].T, origin='lower', 
                cmap = plt.get_cmap('gray'), vmin = 0, vmax = data.max())
            axs[d].locator_params(nbins=3)
            axs[d].tick_params(axis='both', which='major', labelsize=8)

        plt.tight_layout()

        plt.savefig(on)