38 lines
916 B
Python
38 lines
916 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
|
|
|
|
## 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, list) else [tiffFiles]
|
|
|
|
def _process(self, data, metadata):
|
|
layers = list()
|
|
metadata = list()
|
|
|
|
for i, file in enumerate(self.files):
|
|
print('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]
|
|
|