def read_image_to_register(registername, is_imzml, to_flip, moving):
mz = None
if is_imzml:
imzml = imzmlio.open_imzml(registername)
current_s = imzmlio.get_full_spectra(imzml)
max_x = max(imzml.coordinates, key=lambda item: item[0])[0]
max_y = max(imzml.coordinates, key=lambda item: item[1])[1]
max_z = max(imzml.coordinates, key=lambda item: item[2])[2]
array = imzmlio.get_images_from_spectra(current_s,
(max_x, max_y, max_z)).T
register = sitk.GetImageFromArray(array)
mz, _ = imzml.getspectrum(0)
else:
register = sitk.ReadImage(registername, sitk.sitkFloat32)
if to_flip:
dim = register.GetDimension()
flip = (True, False) + ((False, ) if dim > 2 else ())
register = sitk.Flip(register, flip)
dim = register.GetDimension()
register.SetSpacing([1 for i in range(dim)])
identity = np.identity(dim).tolist()
flat_list = [item for sublist in identity for item in sublist]
direction = tuple(flat_list)
register.SetDirection(flat_list)
return register, mz
def read_imzml(inputname):
imzml = imzmlio.open_imzml(inputname)
spectra = imzmlio.get_full_spectra(imzml)
max_x = max(imzml.coordinates, key=lambda item:item[0])[0]
max_y = max(imzml.coordinates, key=lambda item:item[1])[1]
max_z = max(imzml.coordinates, key=lambda item:item[2])[2]
image = imzmlio.get_images_from_spectra(spectra, (max_x, max_y, max_z))
mzs, intensities = imzml.getspectrum(0)
return spectra, image
import esmraldi.imzmlio as imzmlio
import numpy as np
import os
parser = argparse.ArgumentParser()
parser.add_argument("-i", "--input", help="Input 3D imzml")
parser.add_argument("-o", "--output", help="Output directory for 2D imzml")
args = parser.parse_args()
inputname = args.input
outname = args.output
imzml = imzmlio.open_imzml(inputname)
mz, I = imzml.getspectrum(0)
spectra = imzmlio.get_full_spectra(imzml)
max_x = max(imzml.coordinates, key=lambda item: item[0])[0]
max_y = max(imzml.coordinates, key=lambda item: item[1])[1]
max_z = max(imzml.coordinates, key=lambda item: item[2])[2]
image = imzmlio.get_images_from_spectra(spectra, (max_x, max_y, max_z))
coordinates = imzml.coordinates
print("Parsing all spectra")
intensities, new_coords = imzmlio.get_spectra_from_images(image)
previous_index = 0
array_coords = np.array(new_coords)
array_intensities = np.array(intensities)
split_intensities, split_coords = [], []