35 lines
874 B
Python
35 lines
874 B
Python
#!/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
|
|
|
|
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]
|
|
|