#!/usr/bin/python
# -*- coding: utf-8 -*-
# \file LoadTIFF.py
# \brief TODO
# \author Florent Guiotte <florent.guiotte@gmail.com>
# \version 0.1
# \date 04 avril 2018
#
# TODO details

from ld2dap.core import Input, Stack
import numpy as np

## TODO: dep
import sys
sys.path.append('../triskele/python')
import triskele

class LoadTIFF(Input):
    def __init__(self, tiffFiles):
        super().__init__(self.__class__.__name__)
        self.files = tiffFiles if isinstance(tiffFiles, (tuple, list)) else [tiffFiles]

    def _process(self, data, metadata):
        layers = list()
        metadata = list()

        for i, file in enumerate(self.files):
            self.logger.info('Loading {}'.format(file))
            layers.append(triskele.read(file))
            metadata.append(Stack(i, desc=file, symb='I_{{{}}}'.format(i)))

        return np.stack(layers, axis=2), metadata

    def I(self, i):
        return self.files[i]