tomo: tomo.py annotate

annotate tomo.py @ 63:98a83f03d91b draft

"planemo upload for repository https://github.com/rolfverberg/galaxytools commit a9ebbfca439737ae17ddcb0df84408155f964877"

author	rv43
date	Thu, 18 Aug 2022 14:36:46 +0000
parents	75dd6e15f628
children

rev	line source
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1 #!/usr/bin/env python3
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	3 # -- coding: utf-8 --
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	4 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	5 Created on Fri Dec 10 09:54:37 2021
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	6
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	7 @author: rv43
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	8 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	9
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	10 import logging
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	11
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	12 import os
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	13 import sys
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	14 import getopt
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	15 import re
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	16 import io
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	17 import argparse
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	18 import numpy as np
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	19 try:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	20 import numexpr as ne
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	21 except:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	22 pass
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	23 import multiprocessing as mp
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	24 try:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	25 import scipy.ndimage as spi
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	26 except:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	27 pass
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	28 try:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	29 import tomopy
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	30 except:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	31 pass
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	32 from time import time
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	33 try:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	34 from skimage.transform import iradon
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	35 except:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	36 pass
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	37 try:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	38 from skimage.restoration import denoise_tv_chambolle
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	39 except:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	40 pass
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	41
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	42 from detector import TomoDetectorConfig
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	43 from fit import Fit
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	44 from general import illegal_value, is_int, is_num, is_index_range, get_trailing_int, \
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	45 input_int, input_num, input_yesno, input_menu, findImageFiles, loadImageStack, clearPlot, \
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	46 draw_mask_1d, quickPlot, clearImshow, quickImshow, combine_tiffs_in_h5, Config
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	47 from general import selectImageRange, selectImageBounds
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	48
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	49 # the following tomopy routines don't run with more than 24 cores on Galaxy-Dev
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	50 # - tomopy.find_center_vo
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	51 # - tomopy.prep.stripe.remove_stripe_fw
46 d73e715e975b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac" rv43 parents: 45 diff changeset	52 num_core_tomopy_limit = 24
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	53
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	54 class set_numexpr_threads:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	55
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	56 def __init__(self, num_core):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	57 cpu_count = mp.cpu_count()
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	58 logging.debug(f'start: num_core={num_core} cpu_count={cpu_count}')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	59 if num_core is None or num_core < 1 or num_core > cpu_count:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	60 self.num_core = cpu_count
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	61 else:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	62 self.num_core = num_core
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	63 logging.debug(f'self.num_core={self.num_core}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	64
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	65 def __enter__(self):
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	66 self.num_core_org = ne.set_num_threads(self.num_core)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	67 logging.debug(f'self.num_core={self.num_core}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	68
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	69 def __exit__(self, exc_type, exc_value, traceback):
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	70 ne.set_num_threads(self.num_core_org)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	71
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	72 class ConfigTomo(Config):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	73 """Class for processing a config file.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	74 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	75
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	76 def __init__(self, config_file=None, config_dict=None):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	77 super().__init__(config_file, config_dict)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	78
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	79 def _validate_txt(self):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	80 """Returns False if any required config parameter is illegal or missing.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	81 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	82 is_valid = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	83
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	84 # Check for required first-level keys
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	85 pars_required = ['tdf_data_path', 'tbf_data_path', 'detector_id']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	86 pars_missing = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	87 is_valid = super().validate(pars_required, pars_missing)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	88 if len(pars_missing) > 0:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	89 logging.error(f'Missing item(s) in config file: {", ".join(pars_missing)}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	90 self.detector_id = self.config.get('detector_id')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	91
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	92 # Find tomography dark field images file/folder
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	93 self.tdf_data_path = self.config.get('tdf_data_path')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	94
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	95 # Find tomography bright field images file/folder
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	96 self.tbf_data_path = self.config.get('tbf_data_path')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	97
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	98 # Check number of tomography image stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	99 self.num_tomo_stacks = self.config.get('num_tomo_stacks', 1)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	100 if not is_int(self.num_tomo_stacks, 1):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	101 self.num_tomo_stacks = None
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	102 illegal_value(self.num_tomo_stacks, 'num_tomo_stacks', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	103 return False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	104 logging.info(f'num_tomo_stacks = {self.num_tomo_stacks}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	105
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	106 # Find tomography images file/folders and stack parameters
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	107 tomo_data_paths_indices = sorted({key:value for key,value in self.config.items()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	108 if 'tomo_data_path' in key}.items())
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	109 if len(tomo_data_paths_indices) != self.num_tomo_stacks:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	110 logging.error(f'Incorrect number of tomography data path names in config file')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	111 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	112 self.tomo_data_paths = [tomo_data_paths_indices[i][1] for i in range(self.num_tomo_stacks)]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	113 self.tomo_data_indices = [get_trailing_int(tomo_data_paths_indices[i][0])
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	114 if get_trailing_int(tomo_data_paths_indices[i][0]) else None
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	115 for i in range(self.num_tomo_stacks)]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	116 tomo_ref_height_indices = sorted({key:value for key,value in self.config.items()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	117 if 'z_pos' in key}.items())
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	118 if self.num_tomo_stacks > 1 and len(tomo_ref_height_indices) != self.num_tomo_stacks:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	119 logging.error(f'Incorrect number of tomography reference heights in config file')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	120 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	121 if len(tomo_ref_height_indices):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	122 self.tomo_ref_heights = [
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	123 tomo_ref_height_indices[i][1] for i in range(self.num_tomo_stacks)]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	124 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	125 self.tomo_ref_heights = [0.0]*self.num_tomo_stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	126
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	127 # Check tomo angle (theta) range
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	128 self.start_theta = self.config.get('start_theta', 0.)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	129 if not is_num(self.start_theta, 0.):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	130 illegal_value(self.start_theta, 'start_theta', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	131 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	132 logging.debug(f'start_theta = {self.start_theta}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	133 self.end_theta = self.config.get('end_theta', 180.)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	134 if not is_num(self.end_theta, self.start_theta):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	135 illegal_value(self.end_theta, 'end_theta', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	136 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	137 logging.debug(f'end_theta = {self.end_theta}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	138 self.num_thetas = self.config.get('num_thetas')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	139 if not (self.num_thetas is None or is_int(self.num_thetas, 1)):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	140 illegal_value(self.num_thetas, 'num_thetas', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	141 self.num_thetas = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	142 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	143 logging.debug(f'num_thetas = {self.num_thetas}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	144
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	145 return is_valid
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	146
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	147 def _validate_yaml(self):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	148 """Returns False if any required config parameter is illegal or missing.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	149 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	150 is_valid = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	151
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	152 # Check for required first-level keys
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	153 pars_required = ['dark_field', 'bright_field', 'stack_info', 'detector']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	154 pars_missing = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	155 is_valid = super().validate(pars_required, pars_missing)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	156 if len(pars_missing) > 0:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	157 logging.error(f'Missing item(s) in config file: {", ".join(pars_missing)}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	158 self.detector_id = self.config['detector'].get('id')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	159
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	160 # Find tomography dark field images file/folder
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	161 self.tdf_data_path = self.config['dark_field'].get('data_path')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	162
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	163 # Find tomography bright field images file/folder
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	164 self.tbf_data_path = self.config['bright_field'].get('data_path')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	165
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	166 # Check number of tomography image stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	167 stack_info = self.config['stack_info']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	168 self.num_tomo_stacks = stack_info.get('num', 1)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	169 if not is_int(self.num_tomo_stacks, 1):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	170 self.num_tomo_stacks = None
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	171 illegal_value(self.num_tomo_stacks, 'num_tomo_stacks', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	172 return False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	173 logging.info(f'num_tomo_stacks = {self.num_tomo_stacks}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	174
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	175 # Find tomography images file/folders and stack parameters
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	176 stacks = stack_info.get('stacks')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	177 if stacks is None or len(stacks) is not self.num_tomo_stacks:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	178 illegal_value(stacks, 'stacks', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	179 return False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	180 self.tomo_data_paths = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	181 self.tomo_data_indices = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	182 self.tomo_ref_heights = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	183 for stack in stacks:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	184 self.tomo_data_paths.append(stack.get('data_path'))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	185 self.tomo_data_indices.append(stack.get('index'))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	186 self.tomo_ref_heights.append(stack.get('ref_height'))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	187
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	188 # Check tomo angle (theta) range
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	189 theta_range = self.config.get('theta_range')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	190 if theta_range is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	191 self.start_theta = 0.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	192 self.end_theta = 180.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	193 self.num_thetas = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	194 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	195 self.start_theta = theta_range.get('start', 0.)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	196 if not is_num(self.start_theta, 0.):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	197 illegal_value(self.start_theta, 'theta_range:start', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	198 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	199 logging.debug(f'start_theta = {self.start_theta}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	200 self.end_theta = theta_range.get('end', 180.)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	201 if not is_num(self.end_theta, self.start_theta):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	202 illegal_value(self.end_theta, 'theta_range:end', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	203 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	204 logging.debug(f'end_theta = {self.end_theta}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	205 self.num_thetas = theta_range.get('num')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	206 if self.num_thetas and not is_int(self.num_thetas, 1):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	207 illegal_value(self.num_thetas, 'theta_range:num', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	208 self.num_thetas = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	209 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	210 logging.debug(f'num_thetas = {self.num_thetas}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	211
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	212 return is_valid
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	213
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	214 def validate(self):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	215 """Returns False if any required config parameter is illegal or missing.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	216 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	217 is_valid = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	218
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	219 # Check work_folder (shared by both file formats)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	220 work_folder = os.path.abspath(self.config.get('work_folder', ''))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	221 if not os.path.isdir(work_folder):
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	222 illegal_value(work_folder, 'work_folder', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	223 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	224 logging.info(f'work_folder: {work_folder}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	225
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	226 # Check data filetype (shared by both file formats)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	227 self.data_filetype = self.config.get('data_filetype', 'tif')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	228 if not isinstance(self.data_filetype, str) or (self.data_filetype != 'tif' and
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	229 self.data_filetype != 'h5'):
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	230 illegal_value(self.data_filetype, 'data_filetype', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	231
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	232 if self.suffix == '.yml' or self.suffix == '.yaml':
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	233 is_valid = self._validate_yaml()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	234 elif self.suffix == '.txt':
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	235 is_valid = self._validate_txt()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	236 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	237 logging.error(f'Undefined or illegal config file extension: {self.suffix}')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	238 is_valid = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	239
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	240 # Find tomography bright field images file/folder
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	241 if self.tdf_data_path:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	242 if self.data_filetype == 'h5':
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	243 if isinstance(self.tdf_data_path, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	244 if not os.path.isabs(self.tdf_data_path):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	245 self.tdf_data_path = os.path.abspath(
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	246 f'{work_folder}/{self.tdf_data_path}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	247 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	248 illegal_value(tdf_data_fil, 'tdf_data_path', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	249 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	250 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	251 if isinstance(self.tdf_data_path, int):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	252 self.tdf_data_path = os.path.abspath(
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	253 f'{work_folder}/{self.tdf_data_path}/nf')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	254 elif isinstance(self.tdf_data_path, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	255 if not os.path.isabs(self.tdf_data_path):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	256 self.tdf_data_path = os.path.abspath(
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	257 f'{work_folder}/{self.tdf_data_path}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	258 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	259 illegal_value(self.tdf_data_path, 'tdf_data_path', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	260 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	261 logging.info(f'dark field images path = {self.tdf_data_path}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	262
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	263 # Find tomography bright field images file/folder
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	264 if self.tbf_data_path:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	265 if self.data_filetype == 'h5':
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	266 if isinstance(self.tbf_data_path, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	267 if not os.path.isabs(self.tbf_data_path):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	268 self.tbf_data_path = os.path.abspath(
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	269 f'{work_folder}/{self.tbf_data_path}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	270 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	271 illegal_value(tbf_data_fil, 'tbf_data_path', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	272 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	273 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	274 if isinstance(self.tbf_data_path, int):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	275 self.tbf_data_path = os.path.abspath(
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	276 f'{work_folder}/{self.tbf_data_path}/nf')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	277 elif isinstance(self.tbf_data_path, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	278 if not os.path.isabs(self.tbf_data_path):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	279 self.tbf_data_path = os.path.abspath(
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	280 f'{work_folder}/{self.tbf_data_path}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	281 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	282 illegal_value(self.tbf_data_path, 'tbf_data_path', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	283 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	284 logging.info(f'bright field images path = {self.tbf_data_path}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	285
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	286 # Find tomography images file/folders and stack parameters
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	287 tomo_data_paths = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	288 tomo_data_indices = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	289 tomo_ref_heights = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	290 for data_path, index, ref_height in zip(self.tomo_data_paths, self.tomo_data_indices,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	291 self.tomo_ref_heights):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	292 if self.data_filetype == 'h5':
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	293 if isinstance(data_path, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	294 if not os.path.isabs(data_path):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	295 data_path = os.path.abspath(f'{work_folder}/{data_path}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	296 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	297 illegal_value(data_path, 'stack_info:stacks:data_path', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	298 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	299 data_path = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	300 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	301 if isinstance(data_path, int):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	302 data_path = os.path.abspath(f'{work_folder}/{data_path}/nf')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	303 elif isinstance(data_path, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	304 if not os.path.isabs(data_path):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	305 data_path = os.path.abspath(f'{work_folder}/{data_path}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	306 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	307 illegal_value(data_path, 'stack_info:stacks:data_path', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	308 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	309 data_path = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	310 tomo_data_paths.append(data_path)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	311 if index is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	312 if self.num_tomo_stacks > 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	313 logging.error('Missing stack_info:stacks:index in config file')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	314 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	315 index = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	316 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	317 index = 1
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	318 elif not isinstance(index, int):
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	319 illegal_value(index, 'stack_info:stacks:index', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	320 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	321 index = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	322 tomo_data_indices.append(index)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	323 if ref_height is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	324 if self.num_tomo_stacks > 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	325 logging.error('Missing stack_info:stacks:ref_height in config file')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	326 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	327 ref_height = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	328 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	329 ref_height = 0.
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	330 elif not is_num(ref_height):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	331 illegal_value(ref_height, 'stack_info:stacks:ref_height', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	332 is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	333 ref_height = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	334 # Set reference heights relative to first stack
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	335 if (len(tomo_ref_heights) and is_num(ref_height) and
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	336 is_num(tomo_ref_heights[0])):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	337 ref_height = (round(ref_height-tomo_ref_heights[0], 3))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	338 tomo_ref_heights.append(ref_height)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	339 tomo_ref_heights[0] = 0.0
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	340 logging.info('tomography data paths:')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	341 for i in range(self.num_tomo_stacks):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	342 logging.info(f' {tomo_data_paths[i]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	343 logging.info(f'tomography data path indices: {tomo_data_indices}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	344 logging.info(f'tomography reference heights: {tomo_ref_heights}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	345
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	346 # Update config in memory
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	347 if self.suffix == '.txt':
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	348 self.config = {}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	349 dark_field = self.config.get('dark_field')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	350 if dark_field is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	351 self.config['dark_field'] = {'data_path' : self.tdf_data_path}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	352 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	353 self.config['dark_field']['data_path'] = self.tdf_data_path
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	354 bright_field = self.config.get('bright_field')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	355 if bright_field is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	356 self.config['bright_field'] = {'data_path' : self.tbf_data_path}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	357 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	358 self.config['bright_field']['data_path'] = self.tbf_data_path
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	359 detector = self.config.get('detector')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	360 if detector is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	361 self.config['detector'] = {'id' : self.detector_id}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	362 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	363 detector['id'] = self.detector_id
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	364 self.config['work_folder'] = work_folder
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	365 self.config['data_filetype'] = self.data_filetype
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	366 stack_info = self.config.get('stack_info')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	367 if stack_info is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	368 stacks = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	369 for i in range(self.num_tomo_stacks):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	370 stacks.append({'data_path' : tomo_data_paths[i], 'index' : tomo_data_indices[i],
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	371 'ref_height' : tomo_ref_heights[i]})
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	372 self.config['stack_info'] = {'num' : self.num_tomo_stacks, 'stacks' : stacks}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	373 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	374 stack_info['num'] = self.num_tomo_stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	375 stacks = stack_info.get('stacks')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	376 for i,stack in enumerate(stacks):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	377 stack['data_path'] = tomo_data_paths[i]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	378 stack['index'] = tomo_data_indices[i]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	379 stack['ref_height'] = tomo_ref_heights[i]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	380 if self.num_thetas:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	381 theta_range = {'start' : self.start_theta, 'end' : self.end_theta,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	382 'num' : self.num_thetas}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	383 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	384 theta_range = {'start' : self.start_theta, 'end' : self.end_theta}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	385 self.config['theta_range'] = theta_range
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	386
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	387 # Cleanup temporary validation variables
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	388 del self.tdf_data_path
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	389 del self.tbf_data_path
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	390 del self.detector_id
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	391 del self.data_filetype
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	392 del self.num_tomo_stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	393 del self.tomo_data_paths
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	394 del self.tomo_data_indices
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	395 del self.tomo_ref_heights
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	396 del self.start_theta
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	397 del self.end_theta
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	398 del self.num_thetas
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	399
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	400 return is_valid
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	401
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	402 class Tomo:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	403 """Processing tomography data with misalignment.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	404 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	405
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	406 def __init__(self, config_file=None, config_dict=None, config_out=None, output_folder='.',
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	407 log_level='INFO', log_stream='tomo.log', galaxy_flag=False, test_mode=False,
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	408 num_core=-1):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	409 """Initialize with optional config input file or dictionary
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	410 """
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	411 self.num_core = None
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	412 self.config_out = config_out
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	413 self.output_folder = output_folder
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	414 self.galaxy_flag = galaxy_flag
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	415 self.test_mode = test_mode
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	416 self.save_plots = True # Make input argument?
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	417 self.save_plots_only = True # Make input argument?
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	418 self.cf = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	419 self.config = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	420 self.is_valid = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	421 self.tdf = np.array([])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	422 self.tbf = np.array([])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	423 self.tomo_stacks = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	424 self.tomo_recon_stacks = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	425
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	426 # Validate input parameters
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	427 if config_file is not None and not os.path.isfile(config_file):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	428 raise OSError(f'Invalid config_file input {config_file} {type(config_file)}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	429 if config_dict is not None and not isinstance(config_dict, dict):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	430 raise ValueError(f'Invalid config_dict input {config_dict} {type(config_dict)}')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	431 if self.config_out is not None and not isinstance(self.config_out, str):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	432 raise OSError(f'Invalid config_out input {self.config_out} {type(self.config_out)}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	433 if not os.path.isdir(output_folder):
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	434 os.mkdir(os.path.abspath(output_folder))
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	435 if isinstance(log_stream, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	436 path = os.path.split(log_stream)[0]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	437 if path and not os.path.isdir(path):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	438 raise OSError(f'Invalid log_stream path')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	439 if not os.path.isabs(path):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	440 log_stream = f'{output_folder}/{log_stream}'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	441 if not isinstance(galaxy_flag, bool):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	442 raise ValueError(f'Invalid galaxy_flag input {galaxy_flag} {type(galaxy_flag)}')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	443 if not isinstance(self.test_mode, bool):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	444 raise ValueError(f'Invalid test_mode input {self.test_mode} {type(self.test_mode)}')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	445 if not isinstance(num_core, int) or num_core < -1 or num_core == 0:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	446 raise ValueError(f'Invalid num_core input {num_core} {type(num_core)}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	447 if num_core == -1:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	448 self.num_core = mp.cpu_count()
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	449 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	450 self.num_core = num_core
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	451
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	452 # Set log configuration
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	453 logging_format = '%(asctime)s : %(levelname)s - %(module)s : %(funcName)s - %(message)s'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	454 if self.test_mode:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	455 self.save_plots_only = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	456 if isinstance(log_stream, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	457 logging.basicConfig(filename=f'{log_stream}', filemode='w',
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	458 format=logging_format, level=logging.INFO, force=True)
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	459 #format=logging_format, level=logging.WARNING, force=True)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	460 elif isinstance(log_stream, io.TextIOWrapper):
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	461 #logging.basicConfig(filemode='w', format=logging_format, level=logging.WARNING,
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	462 logging.basicConfig(filemode='w', format=logging_format, level=logging.INFO,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	463 stream=log_stream, force=True)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	464 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	465 raise ValueError(f'Invalid log_stream: {log_stream}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	466 logging.warning('Ignoring log_level argument in test mode')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	467 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	468 level = getattr(logging, log_level.upper(), None)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	469 if not isinstance(level, int):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	470 raise ValueError(f'Invalid log_level: {log_level}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	471 if log_stream is sys.stdout:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	472 logging.basicConfig(format=logging_format, level=level, force=True,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	473 handlers=[logging.StreamHandler()])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	474 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	475 if isinstance(log_stream, str):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	476 logging.basicConfig(filename=f'{log_stream}', filemode='w',
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	477 format=logging_format, level=level, force=True)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	478 elif isinstance(log_stream, io.TextIOWrapper):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	479 logging.basicConfig(filemode='w', format=logging_format, level=level,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	480 stream=log_stream, force=True)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	481 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	482 raise ValueError(f'Invalid log_stream: {log_stream}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	483 stream_handler = logging.StreamHandler()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	484 logging.getLogger().addHandler(stream_handler)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	485 stream_handler.setLevel(logging.WARNING)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	486 stream_handler.setFormatter(logging.Formatter(logging_format))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	487
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	488 # Check/set output config file name
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	489 if self.config_out is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	490 self.config_out = f'{self.output_folder}/config.yaml'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	491 elif (self.config_out is os.path.basename(self.config_out) and
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	492 not os.path.isabs(self.config_out)):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	493 self.config_out = f'{self.output_folder}/{self.config_out}'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	494
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	495 # Create config object and load config file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	496 self.cf = ConfigTomo(config_file, config_dict)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	497 if not self.cf.load_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	498 self.is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	499 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	500
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	501 if self.galaxy_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	502 assert(self.output_folder == '.')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	503 assert(self.test_mode is False)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	504 self.save_plots = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	505 self.save_plots_only = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	506 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	507 # Input validation is already performed during link_data_to_galaxy
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	508
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	509 # Check config file parameters
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	510 self.is_valid = self.cf.validate()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	511
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	512 # Load detector info file
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	513 df = TomoDetectorConfig(self.cf.config['detector']['id'])
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	514
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	515 # Check detector info file parameters
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	516 if df.valid:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	517 pixel_size = df.pixel_size
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	518 num_rows, num_columns = df.dimensions
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	519 if not pixel_size or not num_rows or not num_columns:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	520 self.is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	521 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	522 pixel_size = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	523 num_rows = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	524 num_columns = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	525 self.is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	526
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	527 # Update config
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	528 self.cf.config['detector']['pixel_size'] = pixel_size
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	529 self.cf.config['detector']['rows'] = num_rows
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	530 self.cf.config['detector']['columns'] = num_columns
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	531 logging.debug(f'pixel_size = self.cf.config["detector"]["pixel_size"]')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	532 logging.debug(f'num_rows: {self.cf.config["detector"]["rows"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	533 logging.debug(f'num_columns: {self.cf.config["detector"]["columns"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	534
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	535 # Safe config to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	536 if self.is_valid:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	537 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	538
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	539 # Initialize shortcut to config
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	540 self.config = self.cf.config
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	541
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	542 # Initialize tomography stack
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	543 num_tomo_stacks = self.config['stack_info']['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	544 if num_tomo_stacks:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	545 self.tomo_stacks = [np.array([]) for _ in range(num_tomo_stacks)]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	546 self.tomo_recon_stacks = [np.array([]) for _ in range(num_tomo_stacks)]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	547
2 b8977c98800b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0c40fb274acc13ebf7054379bb49ed13acec918e" rv43 parents: 1 diff changeset	548 logging.debug(f'num_core = {self.num_core}')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	549 logging.debug(f'config_file = {config_file}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	550 logging.debug(f'config_dict = {config_dict}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	551 logging.debug(f'config_out = {self.config_out}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	552 logging.debug(f'output_folder = {self.output_folder}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	553 logging.debug(f'log_stream = {log_stream}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	554 logging.debug(f'log_level = {log_level}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	555 logging.debug(f'galaxy_flag = {self.galaxy_flag}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	556 logging.debug(f'test_mode = {self.test_mode}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	557 logging.debug(f'save_plots = {self.save_plots}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	558 logging.debug(f'save_plots_only = {self.save_plots_only}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	559
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	560 def _selectImageRanges(self, available_stacks=None):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	561 """Select image files to be included in analysis.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	562 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	563 self.is_valid = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	564 stack_info = self.config['stack_info']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	565 if available_stacks is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	566 available_stacks = [False]*stack_info['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	567 elif len(available_stacks) != stack_info['num']:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	568 logging.warning('Illegal dimension of available_stacks in getImageFiles '+
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	569 f'({len(available_stacks)}');
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	570 available_stacks = [False]*stack_info['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	571
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	572 # Check number of tomography angles/thetas
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	573 num_thetas = self.config['theta_range'].get('num')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	574 if num_thetas is None:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	575 num_thetas = input_int('\nEnter the number of thetas', 1)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	576 elif not is_int(num_thetas, 0):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	577 illegal_value(num_thetas, 'num_thetas', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	578 self.is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	579 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	580 self.config['theta_range']['num'] = num_thetas
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	581 logging.debug(f'num_thetas = {self.config["theta_range"]["num"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	582
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	583 # Find tomography dark field images
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	584 dark_field = self.config['dark_field']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	585 img_start = dark_field.get('img_start', -1)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	586 img_offset = dark_field.get('img_offset', -1)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	587 num_imgs = dark_field.get('num', 0)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	588 if not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	589 img_start, img_offset, num_imgs = selectImageRange(img_start, img_offset,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	590 num_imgs, 'dark field')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	591 if img_start < 0 or num_imgs < 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	592 logging.error('Unable to find suitable dark field images')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	593 if dark_field['data_path']:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	594 self.is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	595 dark_field['img_start'] = img_start
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	596 dark_field['img_offset'] = img_offset
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	597 dark_field['num'] = num_imgs
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	598 logging.debug(f'Dark field image start index: {dark_field["img_start"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	599 logging.debug(f'Dark field image offset: {dark_field["img_offset"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	600 logging.debug(f'Number of dark field images: {dark_field["num"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	601
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	602 # Find tomography bright field images
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	603 bright_field = self.config['bright_field']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	604 img_start = bright_field.get('img_start', -1)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	605 img_offset = bright_field.get('img_offset', -1)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	606 num_imgs = bright_field.get('num', 0)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	607 if not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	608 img_start, img_offset, num_imgs = selectImageRange(img_start, img_offset,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	609 num_imgs, 'bright field')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	610 if img_start < 0 or num_imgs < 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	611 logging.error('Unable to find suitable bright field images')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	612 self.is_valid = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	613 bright_field['img_start'] = img_start
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	614 bright_field['img_offset'] = img_offset
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	615 bright_field['num'] = num_imgs
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	616 logging.debug(f'Bright field image start index: {bright_field["img_start"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	617 logging.debug(f'Bright field image offset: {bright_field["img_offset"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	618 logging.debug(f'Number of bright field images: {bright_field["num"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	619
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	620 # Find tomography images
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	621 for i,stack in enumerate(stack_info['stacks']):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	622 # Check if stack is already loaded or available
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	623 if self.tomo_stacks[i].size or available_stacks[i]:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	624 continue
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	625 index = stack['index']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	626 img_start = stack.get('img_start', -1)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	627 img_offset = stack.get('img_offset', -1)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	628 num_imgs = stack.get('num', 0)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	629 if not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	630 img_start, img_offset, num_imgs = selectImageRange(img_start, img_offset,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	631 num_imgs, f'tomography stack {index}', num_thetas)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	632 if img_start < 0 or num_imgs != num_thetas:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	633 logging.error('Unable to find suitable tomography images')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	634 self.is_valid = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	635 stack['img_start'] = img_start
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	636 stack['img_offset'] = img_offset
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	637 stack['num'] = num_imgs
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	638 logging.debug(f'Tomography stack {index} image start index: {stack["img_start"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	639 logging.debug(f'Tomography stack {index} image offset: {stack["img_offset"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	640 logging.debug(f'Number of tomography images for stack {index}: {stack["num"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	641
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	642 # Safe updated config to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	643 if self.is_valid:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	644 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	645
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	646 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	647
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	648 def _genDark(self, tdf_files):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	649 """Generate dark field.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	650 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	651 # Load the dark field images
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	652 logging.debug('Loading dark field...')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	653 dark_field = self.config['dark_field']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	654 tdf_stack = loadImageStack(tdf_files, self.config['data_filetype'],
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	655 dark_field['img_offset'], dark_field['num'])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	656
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	657 # Take median
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	658 self.tdf = np.median(tdf_stack, axis=0)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	659 del tdf_stack
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	660
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	661 # Remove dark field intensities above the cutoff
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	662 tdf_cutoff = dark_field.get('cutoff')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	663 if tdf_cutoff is not None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	664 if not is_num(tdf_cutoff, 0):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	665 logging.warning(f'Ignoring illegal value of tdf_cutoff {tdf_cutoff}')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	666 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	667 self.tdf[self.tdf > tdf_cutoff] = np.nan
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	668 logging.debug(f'tdf_cutoff = {tdf_cutoff}')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	669
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	670 tdf_mean = np.nanmean(self.tdf)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	671 logging.debug(f'tdf_mean = {tdf_mean}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	672 np.nan_to_num(self.tdf, copy=False, nan=tdf_mean, posinf=tdf_mean, neginf=0.)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	673 if self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	674 quickImshow(self.tdf, title='dark field', path='setup_pngs',
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	675 save_fig=True, save_only=True)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	676 elif not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	677 quickImshow(self.tdf, title='dark field', path=self.output_folder,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	678 save_fig=self.save_plots, save_only=self.save_plots_only)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	679
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	680 def _genBright(self, tbf_files):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	681 """Generate bright field.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	682 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	683 # Load the bright field images
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	684 logging.debug('Loading bright field...')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	685 bright_field = self.config['bright_field']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	686 tbf_stack = loadImageStack(tbf_files, self.config['data_filetype'],
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	687 bright_field['img_offset'], bright_field['num'])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	688
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	689 # Take median
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	690 """Median or mean: It may be best to try the median because of some image
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	691 artifacts that arise due to crinkles in the upstream kapton tape windows
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	692 causing some phase contrast images to appear on the detector.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	693 One thing that also may be useful in a future implementation is to do a
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	694 brightfield adjustment on EACH frame of the tomo based on a ROI in the
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	695 corner of the frame where there is no sample but there is the direct X-ray
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	696 beam because there is frame to frame fluctuations from the incoming beam.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	697 We don’t typically account for them but potentially could.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	698 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	699 self.tbf = np.median(tbf_stack, axis=0)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	700 del tbf_stack
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	701
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	702 # Subtract dark field
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	703 if self.tdf.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	704 self.tbf -= self.tdf
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	705 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	706 logging.warning('Dark field unavailable')
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	707 if self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	708 quickImshow(self.tbf, title='bright field', path='setup_pngs',
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	709 save_fig=True, save_only=True)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	710 elif not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	711 quickImshow(self.tbf, title='bright field', path=self.output_folder,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	712 save_fig=self.save_plots, save_only=self.save_plots_only)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	713
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	714 def _setDetectorBounds(self, tomo_stack_files):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	715 """Set vertical detector bounds for image stack.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	716 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	717 preprocess = self.config.get('preprocess')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	718 if preprocess is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	719 img_x_bounds = [None, None]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	720 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	721 img_x_bounds = preprocess.get('img_x_bounds', [0, self.tbf.shape[0]])
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	722 if img_x_bounds[0] is not None and img_x_bounds[1] is not None:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	723 if img_x_bounds[0] < 0:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	724 illegal_value(img_x_bounds[0], 'preprocess:img_x_bounds[0]', 'config file')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	725 img_x_bounds[0] = 0
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	726 if not is_index_range(img_x_bounds, 0, self.tbf.shape[0]):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	727 illegal_value(img_x_bounds[1], 'preprocess:img_x_bounds[1]', 'config file')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	728 img_x_bounds[1] = self.tbf.shape[0]
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	729 if self.test_mode:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	730 # Update config and save to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	731 if preprocess is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	732 self.cf.config['preprocess'] = {'img_x_bounds' : [0, self.tbf.shape[0]]}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	733 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	734 preprocess['img_x_bounds'] = img_x_bounds
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	735 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	736 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	737
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	738 # Check reference heights
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	739 pixel_size = self.config['detector']['pixel_size']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	740 if pixel_size is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	741 raise ValueError('Detector pixel size unavailable')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	742 if not self.tbf.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	743 raise ValueError('Bright field unavailable')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	744 num_x_min = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	745 num_tomo_stacks = self.config['stack_info']['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	746 stacks = self.config['stack_info']['stacks']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	747 if num_tomo_stacks > 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	748 delta_z = stacks[1]['ref_height']-stacks[0]['ref_height']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	749 for i in range(2, num_tomo_stacks):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	750 delta_z = min(delta_z, stacks[i]['ref_height']-stacks[i-1]['ref_height'])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	751 logging.debug(f'delta_z = {delta_z}')
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	752 num_x_min = int((delta_z-0.5*pixel_size)/pixel_size)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	753 logging.debug(f'num_x_min = {num_x_min}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	754 if num_x_min > self.tbf.shape[0]:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	755 logging.warning('Image bounds and pixel size prevent seamless stacking')
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	756 num_x_min = None
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	757
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	758 # Select image bounds
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	759 if self.galaxy_flag:
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	760 x_sum = np.sum(self.tbf, 1)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	761 x_sum_min = x_sum.min()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	762 x_sum_max = x_sum.max()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	763 x_low = 0
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	764 x_upp = x_sum.size
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	765 if num_x_min is not None:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	766 fit = Fit.fit_data(np.array(range(len(x_sum))), x_sum, models='rectangle',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	767 form='atan', guess=True)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	768 parameters = fit.best_values
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	769 x_low = parameters.get('center1', None)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	770 x_upp = parameters.get('center2', None)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	771 sig_low = parameters.get('sigma1', None)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	772 sig_upp = parameters.get('sigma2', None)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	773 if (x_low is not None and x_upp is not None and sig_low is not None and
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	774 sig_upp is not None and 0 <= x_low < x_upp <= x_sum.size and
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	775 (sig_low+sig_upp)/(x_upp-x_low) < 0.1):
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	776 if num_tomo_stacks == 1 or num_x_min is None:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	777 x_low = int(x_low-(x_upp-x_low)/10)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	778 x_upp = int(x_upp+(x_upp-x_low)/10)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	779 else:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	780 x_low = int((x_low+x_upp)/2-num_x_min/2)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	781 x_upp = x_low+num_x_min
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	782 if x_low < 0:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	783 x_low = 0
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	784 if x_upp > x_sum.size:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	785 x_upp = x_sum.size
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	786 else:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	787 x_low = 0
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	788 x_upp = x_sum.size
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	789 quickPlot((range(x_sum.size), x_sum),
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	790 ([x_low, x_low], [x_sum_min, x_sum_max], 'r-'),
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	791 ([x_upp-1, x_upp-1], [x_sum_min, x_sum_max], 'r-'),
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	792 title=f'sum bright field over theta/y (row bounds: [{x_low}, {x_upp}])',
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	793 path='setup_pngs', name='detectorbounds.png', save_fig=True, save_only=True,
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	794 show_grid=True)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	795 for i,stack in enumerate(stacks):
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	796 tomo_stack = loadImageStack(tomo_stack_files[i], self.config['data_filetype'],
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	797 stack['img_offset'], 1)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	798 tomo_stack = tomo_stack[0,:,:]
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	799 if num_x_min is not None:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	800 tomo_stack_max = tomo_stack.max()
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	801 tomo_stack[x_low,:] = tomo_stack_max
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	802 tomo_stack[x_upp-1,:] = tomo_stack_max
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	803 title = f'tomography image at theta={self.config["theta_range"]["start"]}'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	804 quickImshow(tomo_stack, title=title, path='setup_pngs',
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	805 name=f'tomo_{stack["index"]}.png', save_fig=True, save_only=True,
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	806 show_grid=True)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	807 del tomo_stack
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	808
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	809 # Update config and save to file
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	810 img_x_bounds = [x_low, x_upp]
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	811 logging.debug(f'img_x_bounds: {img_x_bounds}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	812 if preprocess is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	813 self.cf.config['preprocess'] = {'img_x_bounds' : img_x_bounds}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	814 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	815 preprocess['img_x_bounds'] = img_x_bounds
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	816 self.cf.saveFile(self.config_out)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	817 del x_sum
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	818 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	819
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	820 # For one tomography stack only: load the first image
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	821 title = None
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	822 quickImshow(self.tbf, title='bright field')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	823 if num_tomo_stacks == 1:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	824 tomo_stack = loadImageStack(tomo_stack_files[0], self.config['data_filetype'],
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	825 stacks[0]['img_offset'], 1)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	826 title = f'tomography image at theta={self.config["theta_range"]["start"]}'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	827 quickImshow(tomo_stack[0,:,:], title=title)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	828 tomo_or_bright = input_menu(['bright field', 'first tomography image'],
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	829 header='\nSelect image bounds from')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	830 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	831 print('\nSelect image bounds from bright field')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	832 tomo_or_bright = 0
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	833 if tomo_or_bright:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	834 x_sum = np.sum(tomo_stack[0,:,:], 1)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	835 use_bounds = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	836 if img_x_bounds[0] is not None and img_x_bounds[1] is not None:
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	837 tmp = np.copy(tomo_stack[0,:,:])
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	838 tmp_max = tmp.max()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	839 tmp[img_x_bounds[0],:] = tmp_max
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	840 tmp[img_x_bounds[1]-1,:] = tmp_max
43 0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	841 title = f'tomography image at theta={self.config["theta_range"]["start"]}'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	842 quickImshow(tmp, title=title)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	843 del tmp
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	844 x_sum_min = x_sum.min()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	845 x_sum_max = x_sum.max()
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	846 quickPlot((range(x_sum.size), x_sum),
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	847 ([img_x_bounds[0], img_x_bounds[0]], [x_sum_min, x_sum_max], 'r-'),
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	848 ([img_x_bounds[1]-1, img_x_bounds[1]-1], [x_sum_min, x_sum_max], 'r-'),
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	849 title='sum over theta and y')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	850 print(f'lower bound = {img_x_bounds[0]} (inclusive)\n'+
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	851 f'upper bound = {img_x_bounds[1]} (exclusive)]')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	852 use_bounds = input_yesno('Accept these bounds (y/n)?', 'y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	853 if not use_bounds:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	854 img_x_bounds = list(selectImageBounds(tomo_stack[0,:,:], 0,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	855 img_x_bounds[0], img_x_bounds[1], num_x_min,
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	856 f'tomography image at theta={self.config["theta_range"]["start"]}'))
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	857 if num_x_min is not None and img_x_bounds[1]-img_x_bounds[0]+1 < num_x_min:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	858 logging.warning('Image bounds and pixel size prevent seamless stacking')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	859 title = f'tomography image at theta={self.config["theta_range"]["start"]}'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	860 quickImshow(tomo_stack[0,:,:], title=title, path=self.output_folder,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	861 save_fig=self.save_plots, save_only=True)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	862 quickPlot(range(img_x_bounds[0], img_x_bounds[1]),
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	863 x_sum[img_x_bounds[0]:img_x_bounds[1]],
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	864 title='sum over theta and y', path=self.output_folder,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	865 save_fig=self.save_plots, save_only=True)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	866 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	867 x_sum = np.sum(self.tbf, 1)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	868 x_sum_min = x_sum.min()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	869 x_sum_max = x_sum.max()
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	870 use_bounds = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	871 if img_x_bounds[0] is not None and img_x_bounds[1] is not None:
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	872 tmp = np.copy(self.tbf)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	873 tmp_max = tmp.max()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	874 tmp[img_x_bounds[0],:] = tmp_max
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	875 tmp[img_x_bounds[1]-1,:] = tmp_max
43 0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	876 title = 'bright field'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	877 quickImshow(tmp, title=title)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	878 del tmp
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	879 quickPlot((range(x_sum.size), x_sum),
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	880 ([img_x_bounds[0], img_x_bounds[0]], [x_sum_min, x_sum_max], 'r-'),
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	881 ([img_x_bounds[1]-1, img_x_bounds[1]-1], [x_sum_min, x_sum_max], 'r-'),
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	882 title='sum over theta and y')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	883 print(f'lower bound = {img_x_bounds[0]} (inclusive)\n'+
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	884 f'upper bound = {img_x_bounds[1]} (exclusive)]')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	885 use_bounds = input_yesno('Accept these bounds (y/n)?', 'y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	886 if not use_bounds:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	887 use_fit = False
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	888 fit = Fit.fit_data(np.array(range(len(x_sum))), x_sum, models='rectangle',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	889 form='atan', guess=True)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	890 parameters = fit.best_values
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	891 x_low = parameters.get('center1', None)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	892 x_upp = parameters.get('center2', None)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	893 sig_low = parameters.get('sigma1', None)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	894 sig_upp = parameters.get('sigma2', None)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	895 if (x_low is not None and x_upp is not None and sig_low is not None and
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	896 sig_upp is not None and 0 <= x_low < x_upp <= x_sum.size and
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	897 (sig_low+sig_upp)/(x_upp-x_low) < 0.1):
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	898 if num_tomo_stacks == 1 or num_x_min is None:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	899 x_low = int(x_low-(x_upp-x_low)/10)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	900 x_upp = int(x_upp+(x_upp-x_low)/10)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	901 else:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	902 x_low = int((x_low+x_upp)/2-num_x_min/2)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	903 x_upp = x_low+num_x_min
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	904 if x_low < 0:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	905 x_low = 0
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	906 if x_upp > x_sum.size:
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	907 x_upp = x_sum.size
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	908 tmp = np.copy(self.tbf)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	909 tmp_max = tmp.max()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	910 tmp[x_low,:] = tmp_max
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	911 tmp[x_upp-1,:] = tmp_max
43 0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	912 title = 'bright field'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	913 quickImshow(tmp, title=title)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	914 del tmp
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	915 quickPlot((range(x_sum.size), x_sum),
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	916 ([x_low, x_low], [x_sum_min, x_sum_max], 'r-'),
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	917 ([x_upp, x_upp], [x_sum_min, x_sum_max], 'r-'),
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	918 title='sum over theta and y')
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	919 print(f'lower bound = {x_low} (inclusive)')
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	920 print(f'upper bound = {x_upp} (exclusive)]')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	921 use_fit = input_yesno('Accept these bounds (y/n)?', 'y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	922 if use_fit:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	923 img_x_bounds = [x_low, x_upp]
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	924 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	925 accept = False
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	926 while not accept:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	927 mask, img_x_bounds = draw_mask_1d(x_sum, title='select x data range',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	928 legend='sum over theta and y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	929 print(f'img_x_bounds = {img_x_bounds}')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	930 while (len(img_x_bounds) != 1 or (len(x_sum) >= num_x_min and
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	931 img_x_bounds[0][1]-img_x_bounds[0][0]+1 < num_x_min)):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	932 exit('Should not be here')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	933 print('Please select exactly one continuous range')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	934 mask, img_x_bounds = draw_mask_1d(x_sum, title='select x data range',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	935 legend='sum over theta and y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	936 img_x_bounds = list(img_x_bounds[0])
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	937 quickPlot(x_sum, vlines=img_x_bounds, title='sum over theta and y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	938 print(f'img_x_bounds = {img_x_bounds} (lower bound inclusive, upper bound '+
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	939 'exclusive)')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	940 accept = input_yesno('Accept these bounds (y/n)?', 'y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	941 if not accept:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	942 img_x_bounds = None
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	943 if num_x_min is not None and img_x_bounds[1]-img_x_bounds[0]+1 < num_x_min:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	944 logging.warning('Image bounds and pixel size prevent seamless stacking')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	945 #quickPlot(range(img_x_bounds[0], img_x_bounds[1]),
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	946 # x_sum[img_x_bounds[0]:img_x_bounds[1]],
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	947 # title='sum over theta and y', path=self.output_folder,
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	948 # save_fig=self.save_plots, save_only=True)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	949 quickPlot((range(x_sum.size), x_sum),
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	950 ([img_x_bounds[0], img_x_bounds[0]], [x_sum_min, x_sum_max], 'r-'),
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	951 ([img_x_bounds[1], img_x_bounds[1]], [x_sum_min, x_sum_max], 'r-'),
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	952 title='sum over theta and y', path=self.output_folder,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	953 save_fig=self.save_plots, save_only=True)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	954 del x_sum
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	955 for i,stack in enumerate(stacks):
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	956 tomo_stack = loadImageStack(tomo_stack_files[i], self.config['data_filetype'],
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	957 stack['img_offset'], 1)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	958 tomo_stack = tomo_stack[0,:,:]
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	959 if num_x_min is not None:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	960 tomo_stack_max = tomo_stack.max()
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	961 tomo_stack[img_x_bounds[0],:] = tomo_stack_max
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	962 tomo_stack[img_x_bounds[1]-1,:] = tomo_stack_max
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	963 title = f'tomography image at theta={self.config["theta_range"]["start"]}'
43 0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	964 if self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	965 quickImshow(tomo_stack, title=title, path='setup_pngs',
43 0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	966 name=f'tomo_{stack["index"]}.png', save_fig=True, save_only=True,
0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	967 show_grid=True)
0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	968 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	969 quickImshow(tomo_stack, title=title, path=self.output_folder,
43 0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	970 name=f'tomo_{stack["index"]}.png', save_fig=self.save_plots,
0b9ce9489ecc "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 006ce85b82c2f8eb386cb1648803d7768016c067" rv43 parents: 41 diff changeset	971 save_only=True, show_grid=True)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	972 del tomo_stack
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	973 logging.debug(f'img_x_bounds: {img_x_bounds}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	974
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	975 if self.save_plots_only:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	976 clearImshow('bright field')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	977 clearPlot('sum over theta and y')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	978 if title:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	979 clearPlot(title)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	980
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	981 # Update config and save to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	982 if preprocess is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	983 self.cf.config['preprocess'] = {'img_x_bounds' : img_x_bounds}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	984 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	985 preprocess['img_x_bounds'] = img_x_bounds
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	986 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	987
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	988 def _setZoomOrSkip(self):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	989 """Set zoom and/or theta skip to reduce memory the requirement for the analysis.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	990 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	991 preprocess = self.config.get('preprocess')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	992 zoom_perc = 100
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	993 if not self.galaxy_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	994 if preprocess is None or 'zoom_perc' not in preprocess:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	995 if input_yesno('\nDo you want to zoom in to reduce memory requirement (y/n)?', 'n'):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	996 zoom_perc = input_int(' Enter zoom percentage', 1, 100)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	997 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	998 zoom_perc = preprocess['zoom_perc']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	999 if is_num(zoom_perc, 1., 100.):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1000 zoom_perc = int(zoom_perc)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1001 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1002 illegal_value(zoom_perc, 'preprocess:zoom_perc', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1003 zoom_perc = 100
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1004 num_theta_skip = 0
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1005 if not self.galaxy_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1006 if preprocess is None or 'num_theta_skip' not in preprocess:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1007 if input_yesno('Do you want to skip thetas to reduce memory requirement (y/n)?',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1008 'n'):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1009 num_theta_skip = input_int(' Enter the number skip theta interval', 0,
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1010 self.num_thetas-1)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1011 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1012 num_theta_skip = preprocess['num_theta_skip']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1013 if not is_int(num_theta_skip, 0):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1014 illegal_value(num_theta_skip, 'preprocess:num_theta_skip', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1015 num_theta_skip = 0
40 fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1016 logging.debug(f'zoom_perc = {zoom_perc}')
fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1017 logging.debug(f'num_theta_skip = {num_theta_skip}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1018
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1019 # Update config and save to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1020 if preprocess is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1021 self.cf.config['preprocess'] = {'zoom_perc' : zoom_perc,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1022 'num_theta_skip' : num_theta_skip}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1023 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1024 preprocess['zoom_perc'] = zoom_perc
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1025 preprocess['num_theta_skip'] = num_theta_skip
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1026 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1027
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1028 def _loadTomo(self, base_name, index, required=False):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1029 """Load a tomography stack.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1030 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1031 # stack order: row,theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1032 zoom_perc = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1033 preprocess = self.config.get('preprocess')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1034 if preprocess:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1035 zoom_perc = preprocess.get('zoom_perc')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1036 if zoom_perc is None or zoom_perc == 100:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1037 title = f'{base_name} fullres'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1038 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1039 title = f'{base_name} {zoom_perc}p'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1040 title += f'_{index}'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1041 tomo_file = re.sub(r"\s+", '_', f'{self.output_folder}/{title}.npy')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1042 load_flag = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1043 available = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1044 if os.path.isfile(tomo_file):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1045 available = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1046 if required:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1047 load_flag = True
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1048 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1049 load_flag = input_yesno(f'\nDo you want to load {tomo_file} (y/n)?')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1050 stack = np.array([])
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1051 if load_flag:
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1052 t0 = time()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1053 logging.info(f'Loading {tomo_file} ...')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1054 try:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1055 stack = np.load(tomo_file)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1056 except IOError or ValueError:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1057 stack = np.array([])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1058 logging.error(f'Error loading {tomo_file}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1059 logging.info(f'... done in {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1060 if stack.size:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1061 quickImshow(stack[:,0,:], title=title, path=self.output_folder,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1062 save_fig=self.save_plots, save_only=self.save_plots_only)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1063 return stack, available
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1064
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1065 def _saveTomo(self, base_name, stack, index=None):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1066 """Save a tomography stack.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1067 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1068 zoom_perc = None
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1069 preprocess = self.config.get('preprocess')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1070 if preprocess:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1071 zoom_perc = preprocess.get('zoom_perc')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1072 if zoom_perc is None or zoom_perc == 100:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1073 title = f'{base_name} fullres'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1074 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1075 title = f'{base_name} {zoom_perc}p'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1076 if index:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1077 title += f'_{index}'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1078 tomo_file = re.sub(r"\s+", '_', f'{self.output_folder}/{title}.npy')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1079 t0 = time()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1080 logging.info(f'Saving {tomo_file} ...')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1081 np.save(tomo_file, stack)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1082 logging.info(f'... done in {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1083
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1084 def _genTomo(self, tomo_stack_files, available_stacks, num_core=None):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1085 """Generate tomography fields.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1086 """
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1087 if num_core is None:
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1088 num_core = self.num_core
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1089 stacks = self.config['stack_info']['stacks']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1090 assert(len(self.tomo_stacks) == self.config['stack_info']['num'])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1091 assert(len(self.tomo_stacks) == len(stacks))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1092 if len(available_stacks) != len(stacks):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1093 logging.warning('Illegal dimension of available_stacks in _genTomo'+
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1094 f'({len(available_stacks)}');
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1095 available_stacks = [False]*self.num_tomo_stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1096
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1097 preprocess = self.config.get('preprocess')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1098 if preprocess is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1099 img_x_bounds = [0, self.tbf.shape[0]]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1100 img_y_bounds = [0, self.tbf.shape[1]]
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1101 zoom_perc = 100
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1102 num_theta_skip = 0
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1103 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1104 img_x_bounds = preprocess.get('img_x_bounds', [0, self.tbf.shape[0]])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1105 img_y_bounds = preprocess.get('img_y_bounds', [0, self.tbf.shape[1]])
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1106 zoom_perc = preprocess.get('zoom_perc', 100)
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1107 num_theta_skip = preprocess.get('num_theta_skip', 0)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1108
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1109 if self.tdf.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1110 tdf = self.tdf[img_x_bounds[0]:img_x_bounds[1],img_y_bounds[0]:img_y_bounds[1]]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1111 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1112 logging.warning('Dark field unavailable')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1113 if not self.tbf.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1114 raise ValueError('Bright field unavailable')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1115 tbf = self.tbf[img_x_bounds[0]:img_x_bounds[1],img_y_bounds[0]:img_y_bounds[1]]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1116
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1117 for i,stack in enumerate(stacks):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1118 # Check if stack is already loaded or available
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1119 if self.tomo_stacks[i].size or available_stacks[i]:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1120 continue
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1121
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1122 # Load a stack of tomography images
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1123 index = stack['index']
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1124 t0 = time()
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1125 tomo_stack = loadImageStack(tomo_stack_files[i], self.config['data_filetype'],
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1126 stack['img_offset'], self.config['theta_range']['num'], num_theta_skip,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1127 img_x_bounds, img_y_bounds)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1128 tomo_stack = tomo_stack.astype('float64')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1129 logging.debug(f'loading stack {index} took {time()-t0:.2f} seconds!')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1130
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1131 # Subtract dark field
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1132 if self.tdf.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1133 t0 = time()
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1134 with set_numexpr_threads(self.num_core):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1135 ne.evaluate('tomo_stack-tdf', out=tomo_stack)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1136 logging.debug(f'subtracting dark field took {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1137
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1138 # Normalize
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1139 t0 = time()
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1140 with set_numexpr_threads(self.num_core):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1141 ne.evaluate('tomo_stack/tbf', out=tomo_stack, truediv=True)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1142 logging.debug(f'normalizing took {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1143
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1144 # Remove non-positive values and linearize data
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1145 t0 = time()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1146 cutoff = 1.e-6
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1147 with set_numexpr_threads(self.num_core):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1148 ne.evaluate('where(tomo_stack<cutoff, cutoff, tomo_stack)', out=tomo_stack)
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1149 with set_numexpr_threads(self.num_core):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1150 ne.evaluate('-log(tomo_stack)', out=tomo_stack)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1151 logging.debug('removing non-positive values and linearizing data took '+
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1152 f'{time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1153
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1154 # Get rid of nans/infs that may be introduced by normalization
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1155 t0 = time()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1156 np.where(np.isfinite(tomo_stack), tomo_stack, 0.)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1157 logging.debug(f'remove nans/infs took {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1158
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1159 # Downsize tomography stack to smaller size
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1160 tomo_stack = tomo_stack.astype('float32')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1161 if not self.galaxy_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1162 title = f'red stack fullres {index}'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1163 if not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1164 quickImshow(tomo_stack[0,:,:], title=title, path=self.output_folder,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1165 save_fig=self.save_plots, save_only=self.save_plots_only)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1166 if zoom_perc != 100:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1167 t0 = time()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1168 logging.info(f'Zooming in ...')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1169 tomo_zoom_list = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1170 for j in range(tomo_stack.shape[0]):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1171 tomo_zoom = spi.zoom(tomo_stack[j,:,:], 0.01*zoom_perc)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1172 tomo_zoom_list.append(tomo_zoom)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1173 tomo_stack = np.stack([tomo_zoom for tomo_zoom in tomo_zoom_list])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1174 logging.info(f'... done in {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1175 del tomo_zoom_list
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1176 if not self.galaxy_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1177 title = f'red stack {zoom_perc}p {index}'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1178 if not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1179 quickImshow(tomo_stack[0,:,:], title=title, path=self.output_folder,
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1180 save_fig=self.save_plots, save_only=self.save_plots_only)
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1181
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1182 # Convert tomography stack from theta,row,column to row,theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1183 tomo_stack = np.swapaxes(tomo_stack, 0, 1)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1184
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1185 # Save tomography stack to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1186 if not self.galaxy_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1187 if not self.test_mode:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1188 self._saveTomo('red stack', tomo_stack, index)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1189 else:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1190 np.savetxt(f'{self.output_folder}/red_stack_{index}.txt',
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1191 tomo_stack[0,:,:], fmt='%.6e')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1192
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1193 # Combine stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1194 t0 = time()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1195 self.tomo_stacks[i] = tomo_stack
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1196 logging.debug(f'combining nstack took {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1197
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1198 # Update config and save to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1199 stack['preprocessed'] = True
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1200 stack.pop('reconstructed', 'reconstructed not found')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1201 find_center = self.config.get('find_center')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1202 if find_center:
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1203 find_center.pop('completed', 'completed not found')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1204 if self.test_mode:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1205 find_center.pop('lower_center_offset', 'lower_center_offset not found')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1206 find_center.pop('upper_center_offset', 'upper_center_offset not found')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1207 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1208
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1209 if self.tdf.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1210 del tdf
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1211 del tbf
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1212
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1213 def _reconstructOnePlane(self, tomo_plane_T, center, thetas_deg, eff_pixel_size,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1214 cross_sectional_dim, plot_sinogram=True, num_core=1):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1215 """Invert the sinogram for a single tomography plane.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1216 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1217 # tomo_plane_T index order: column,theta
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1218 assert(0 <= center < tomo_plane_T.shape[0])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1219 center_offset = center-tomo_plane_T.shape[0]/2
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1220 two_offset = 2*int(np.round(center_offset))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1221 two_offset_abs = np.abs(two_offset)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1222 max_rad = int(0.5(cross_sectional_dim/eff_pixel_size)1.1) # 10% slack to avoid edge effects
47 c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	1223 if max_rad > 0.5*tomo_plane_T.shape[0]:
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	1224 max_rad = 0.5*tomo_plane_T.shape[0]
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1225 dist_from_edge = max(1, int(np.floor((tomo_plane_T.shape[0]-two_offset_abs)/2.)-max_rad))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1226 if two_offset >= 0:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1227 logging.debug(f'sinogram range = [{two_offset+dist_from_edge}, {-dist_from_edge}]')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1228 sinogram = tomo_plane_T[two_offset+dist_from_edge:-dist_from_edge,:]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1229 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1230 logging.debug(f'sinogram range = [{dist_from_edge}, {two_offset-dist_from_edge}]')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1231 sinogram = tomo_plane_T[dist_from_edge:two_offset-dist_from_edge,:]
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1232 if plot_sinogram and not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1233 quickImshow(sinogram.T, f'sinogram center offset{center_offset:.2f}',
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1234 path=self.output_folder, save_fig=self.save_plots,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1235 save_only=self.save_plots_only, aspect='auto')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1236
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1237 # Inverting sinogram
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1238 t0 = time()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1239 recon_sinogram = iradon(sinogram, theta=thetas_deg, circle=True)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1240 logging.debug(f'inverting sinogram took {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1241 del sinogram
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1242
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1243 # Performing Gaussian filtering and removing ring artifacts
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1244 recon_parameters = self.config.get('recon_parameters')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1245 if recon_parameters is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1246 sigma = 1.0
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1247 ring_width = 15
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1248 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1249 sigma = recon_parameters.get('gaussian_sigma', 1.0)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1250 if not is_num(sigma, 0.0):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1251 logging.warning(f'Illegal gaussian_sigma ({sigma}) in _reconstructOnePlane, '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1252 'set to a default value of 1.0')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1253 sigma = 1.0
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1254 ring_width = recon_parameters.get('ring_width', 15)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1255 if not is_int(ring_width, 0):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1256 logging.warning(f'Illegal ring_width ({ring_width}) in _reconstructOnePlane, '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1257 'set to a default value of 15')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1258 ring_width = 15
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1259 t0 = time()
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1260 recon_sinogram = spi.gaussian_filter(recon_sinogram, sigma, mode='nearest')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1261 recon_clean = np.expand_dims(recon_sinogram, axis=0)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1262 del recon_sinogram
38 c09b8ee8f68f "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 62ea2c5a3629c5e4b5b59fadc282214326f093aa" rv43 parents: 37 diff changeset	1263 t1 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1264 logging.debug(f'running remove_ring on {num_core} cores ...')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1265 recon_clean = tomopy.misc.corr.remove_ring(recon_clean, rwidth=ring_width, ncore=num_core)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1266 logging.debug(f'... remove_ring took {time()-t1:.2f} seconds!')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1267 logging.debug(f'filtering and removing ring artifact took {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1268 return recon_clean
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1269
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1270 def _plotEdgesOnePlane(self, recon_plane, title, path=None):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1271 vis_parameters = self.config.get('vis_parameters')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1272 if vis_parameters is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1273 weight = 0.1
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1274 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1275 weight = vis_parameters.get('denoise_weight', 0.1)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1276 if not is_num(weight, 0.0):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1277 logging.warning(f'Illegal weight ({weight}) in _plotEdgesOnePlane, '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1278 'set to a default value of 0.1')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1279 weight = 0.1
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1280 edges = denoise_tv_chambolle(recon_plane, weight=weight)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1281 vmax = np.max(edges[0,:,:])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1282 vmin = -vmax
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1283 if self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1284 quickImshow(edges[0,:,:], title, path=path, save_fig=True, save_only=True,
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1285 cmap='gray', vmin=vmin, vmax=vmax)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1286 else:
11 ec7c3e84d611 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 6d01e142af3a0daad7ab37d4ff02d0f1797057a7" rv43 parents: 7 diff changeset	1287 if path is None:
ec7c3e84d611 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 6d01e142af3a0daad7ab37d4ff02d0f1797057a7" rv43 parents: 7 diff changeset	1288 path=self.output_folder
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1289 quickImshow(edges[0,:,:], f'{title} coolwarm', path=path,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1290 save_fig=self.save_plots, save_only=self.save_plots_only, cmap='coolwarm')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1291 quickImshow(edges[0,:,:], f'{title} gray', path=path,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1292 save_fig=self.save_plots, save_only=self.save_plots_only, cmap='gray',
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1293 vmin=vmin, vmax=vmax)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1294 del edges
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1295
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1296 def _findCenterOnePlane(self, sinogram, row, thetas_deg, eff_pixel_size, cross_sectional_dim,
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1297 tol=0.1, num_core=1, galaxy_param=None):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1298 """Find center for a single tomography plane.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1299 """
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1300 if self.galaxy_flag:
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1301 assert(isinstance(galaxy_param, dict))
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1302 if not os.path.exists('find_center_pngs'):
40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1303 os.mkdir('find_center_pngs')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1304 # sinogram index order: theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1305 # need index order column,theta for iradon, so take transpose
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1306 sinogram_T = sinogram.T
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1307 center = sinogram.shape[1]/2
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1308
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1309 # try automatic center finding routines for initial value
38 c09b8ee8f68f "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 62ea2c5a3629c5e4b5b59fadc282214326f093aa" rv43 parents: 37 diff changeset	1310 t0 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1311 if num_core > num_core_tomopy_limit:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1312 logging.debug(f'running find_center_vo on {num_core_tomopy_limit} cores ...')
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1313 tomo_center = tomopy.find_center_vo(sinogram, ncore=num_core_tomopy_limit)
37 51dc93d0806a "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 87acf8a257d5d51cf249de085e555bb11d1dbd88" rv43 parents: 36 diff changeset	1314 else:
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1315 logging.debug(f'running find_center_vo on {num_core} cores ...')
37 51dc93d0806a "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 87acf8a257d5d51cf249de085e555bb11d1dbd88" rv43 parents: 36 diff changeset	1316 tomo_center = tomopy.find_center_vo(sinogram, ncore=num_core)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1317 logging.debug(f'... find_center_vo took {time()-t0:.2f} seconds!')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1318 center_offset_vo = tomo_center-center
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1319 if self.test_mode:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1320 logging.info(f'Center at row {row} using Nghia Vo’s method = {center_offset_vo:.2f}')
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1321 del sinogram_T
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1322 return float(center_offset_vo)
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1323 elif self.galaxy_flag:
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1324 logging.info(f'Center at row {row} using Nghia Vo’s method = {center_offset_vo:.2f}')
38 c09b8ee8f68f "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 62ea2c5a3629c5e4b5b59fadc282214326f093aa" rv43 parents: 37 diff changeset	1325 t0 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1326 logging.debug(f'running _reconstructOnePlane on {num_core} cores ...')
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1327 recon_plane = self._reconstructOnePlane(sinogram_T, tomo_center, thetas_deg,
40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1328 eff_pixel_size, cross_sectional_dim, False, num_core)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1329 logging.debug(f'... _reconstructOnePlane took {time()-t0:.2f} seconds!')
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1330 title = f'edges row{row} center offset{center_offset_vo:.2f} Vo'
40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1331 self._plotEdgesOnePlane(recon_plane, title, path='find_center_pngs')
40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1332 del recon_plane
40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1333 if not galaxy_param['center_type_selector']:
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1334 del sinogram_T
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1335 return float(center_offset_vo)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1336 else:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1337 print(f'Center at row {row} using Nghia Vo’s method = {center_offset_vo:.2f}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1338 recon_plane = self._reconstructOnePlane(sinogram_T, tomo_center, thetas_deg,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1339 eff_pixel_size, cross_sectional_dim, False, num_core)
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1340 title = f'edges row{row} center offset{center_offset_vo:.2f} Vo'
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1341 self._plotEdgesOnePlane(recon_plane, title)
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1342 if not self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1343 if input_yesno('Try finding center using phase correlation (y/n)?', 'n'):
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1344 tomo_center = tomopy.find_center_pc(sinogram, sinogram, tol=0.1,
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1345 rotc_guess=tomo_center)
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1346 error = 1.
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1347 while error > tol:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1348 prev = tomo_center
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1349 tomo_center = tomopy.find_center_pc(sinogram, sinogram, tol=tol,
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1350 rotc_guess=tomo_center)
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1351 error = np.abs(tomo_center-prev)
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1352 center_offset = tomo_center-center
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1353 print(f'Center at row {row} using phase correlation = {center_offset:.2f}')
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1354 recon_plane = self._reconstructOnePlane(sinogram_T, tomo_center, thetas_deg,
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1355 eff_pixel_size, cross_sectional_dim, False, num_core)
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1356 title = f'edges row{row} center_offset{center_offset:.2f} PC'
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1357 self._plotEdgesOnePlane(recon_plane, title)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1358 if input_yesno('Accept a center location (y) or continue search (n)?', 'y'):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1359 center_offset = input_num(' Enter chosen center offset', -center, center,
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1360 center_offset_vo)
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1361 del sinogram_T
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1362 del recon_plane
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1363 return float(center_offset)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1364
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1365 # perform center finding search
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1366 while True:
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1367 if self.galaxy_flag and galaxy_param and galaxy_param['center_type_selector']:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1368 set_center = center_offset_vo
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1369 if galaxy_param['center_type_selector'] == 'user':
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1370 set_center = galaxy_param['set_center']
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1371 set_range = galaxy_param['set_range']
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1372 center_offset_step = galaxy_param['set_step']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1373 if (not is_num(set_range, 0) or not is_num(center_offset_step) or
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1374 center_offset_step <= 0):
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1375 logging.warning('Illegal center finding search parameter, skip search')
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1376 del sinogram_T
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1377 return float(center_offset_vo)
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1378 set_range = center_offset_step*max(1, int(set_range/center_offset_step))
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1379 center_offset_low = set_center-set_range
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1380 center_offset_upp = set_center+set_range
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1381 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1382 center_offset_low = input_int('\nEnter lower bound for center offset',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1383 -center, center)
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1384 center_offset_upp = input_int('Enter upper bound for center offset',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1385 center_offset_low, center)
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1386 if center_offset_upp == center_offset_low:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1387 center_offset_step = 1
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1388 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1389 center_offset_step = input_int('Enter step size for center offset search',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1390 1, center_offset_upp-center_offset_low)
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1391 num_center_offset = 1+int((center_offset_upp-center_offset_low)/center_offset_step)
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1392 center_offsets = np.linspace(center_offset_low, center_offset_upp, num_center_offset)
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1393 for center_offset in center_offsets:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1394 if center_offset == center_offset_vo:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1395 continue
38 c09b8ee8f68f "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 62ea2c5a3629c5e4b5b59fadc282214326f093aa" rv43 parents: 37 diff changeset	1396 t0 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1397 logging.debug(f'running _reconstructOnePlane on {num_core} cores ...')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1398 recon_plane = self._reconstructOnePlane(sinogram_T, center_offset+center,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1399 thetas_deg, eff_pixel_size, cross_sectional_dim, False, num_core)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1400 logging.debug(f'... _reconstructOnePlane took {time()-t0:.2f} seconds!')
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1401 title = f'edges row{row} center_offset{center_offset:.2f}'
11 ec7c3e84d611 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 6d01e142af3a0daad7ab37d4ff02d0f1797057a7" rv43 parents: 7 diff changeset	1402 if self.galaxy_flag:
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1403 self._plotEdgesOnePlane(recon_plane, title, path='find_center_pngs')
11 ec7c3e84d611 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 6d01e142af3a0daad7ab37d4ff02d0f1797057a7" rv43 parents: 7 diff changeset	1404 else:
ec7c3e84d611 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 6d01e142af3a0daad7ab37d4ff02d0f1797057a7" rv43 parents: 7 diff changeset	1405 self._plotEdgesOnePlane(recon_plane, title)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1406 if self.galaxy_flag or input_int('\nContinue (0) or end the search (1)', 0, 1):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1407 break
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1408
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1409 del sinogram_T
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1410 del recon_plane
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1411 if self.galaxy_flag:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1412 center_offset = center_offset_vo
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1413 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1414 center_offset = input_num(' Enter chosen center offset', -center, center)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1415 return float(center_offset)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1416
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1417 def _reconstructOneTomoStack(self, tomo_stack, thetas, row_bounds=None, center_offsets=[],
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1418 num_core=1, algorithm='gridrec'):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1419 """reconstruct a single tomography stack.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1420 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1421 # stack order: row,theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1422 # thetas must be in radians
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1423 # centers_offset: tomography axis shift in pixels relative to column center
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1424 # RV should we remove stripes?
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1425 # https://tomopy.readthedocs.io/en/latest/api/tomopy.prep.stripe.html
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1426 # RV should we remove rings?
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1427 # https://tomopy.readthedocs.io/en/latest/api/tomopy.misc.corr.html
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1428 # RV: Add an option to do (extra) secondary iterations later or to do some sort of convergence test?
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1429 if row_bounds is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1430 row_bounds = [0, tomo_stack.shape[0]]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1431 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1432 if not is_index_range(row_bounds, 0, tomo_stack.shape[0]):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1433 raise ValueError('Illegal row bounds in reconstructOneTomoStack')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1434 if thetas.size != tomo_stack.shape[1]:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1435 raise ValueError('theta dimension mismatch in reconstructOneTomoStack')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1436 if not len(center_offsets):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1437 centers = np.zeros((row_bounds[1]-row_bounds[0]))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1438 elif len(center_offsets) == 2:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1439 centers = np.linspace(center_offsets[0], center_offsets[1],
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1440 row_bounds[1]-row_bounds[0])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1441 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1442 if center_offsets.size != row_bounds[1]-row_bounds[0]:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1443 raise ValueError('center_offsets dimension mismatch in reconstructOneTomoStack')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1444 centers = center_offsets
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1445 centers += tomo_stack.shape[2]/2
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1446
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1447 # Removing horizontal stripe and ring artifacts and perform secondary iterations
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1448 recon_parameters = self.config.get('recon_parameters')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1449 if recon_parameters is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1450 sigma = 2.0
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1451 secondary_iters = 0
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1452 ring_width = 15
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1453 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1454 sigma = recon_parameters.get('stripe_fw_sigma', 2.0)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1455 if not is_num(sigma, 0):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1456 logging.warning(f'Illegal stripe_fw_sigma ({sigma}) in '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1457 '_reconstructOneTomoStack, set to a default value of 2.0')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1458 ring_width = 15
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1459 secondary_iters = recon_parameters.get('secondary_iters', 0)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1460 if not is_int(secondary_iters, 0):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1461 logging.warning(f'Illegal secondary_iters ({secondary_iters}) in '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1462 '_reconstructOneTomoStack, set to a default value of 0 (skip them)')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1463 ring_width = 0
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1464 ring_width = recon_parameters.get('ring_width', 15)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1465 if not is_int(ring_width, 0):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1466 logging.warning(f'Illegal ring_width ({ring_width}) in _reconstructOnePlane, '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1467 'set to a default value of 15')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1468 ring_width = 15
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1469 if True:
39 8a3036b0a34c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit ebd998fc223b35670269b851ff6d12178ced3ee8" rv43 parents: 38 diff changeset	1470 t0 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1471 if num_core > num_core_tomopy_limit:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1472 logging.debug('running remove_stripe_fw on {num_core_tomopy_limit} cores ...')
40 fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1473 tomo_stack = tomopy.prep.stripe.remove_stripe_fw(
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1474 tomo_stack[row_bounds[0]:row_bounds[1]], sigma=sigma,
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1475 ncore=num_core_tomopy_limit)
40 fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1476 else:
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1477 logging.debug(f'running remove_stripe_fw on {num_core} cores ...')
40 fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1478 tomo_stack = tomopy.prep.stripe.remove_stripe_fw(
fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1479 tomo_stack[row_bounds[0]:row_bounds[1]], sigma=sigma, ncore=num_core)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1480 logging.debug(f'... tomopy.prep.stripe.remove_stripe_fw took {time()-t0:.2f} seconds!')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1481 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1482 tomo_stack = tomo_stack[row_bounds[0]:row_bounds[1]]
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1483 logging.debug('performing initial reconstruction')
39 8a3036b0a34c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit ebd998fc223b35670269b851ff6d12178ced3ee8" rv43 parents: 38 diff changeset	1484 t0 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1485 logging.debug(f'running recon on {num_core} cores ...')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1486 tomo_recon_stack = tomopy.recon(tomo_stack, thetas, centers, sinogram_order=True,
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1487 algorithm=algorithm, ncore=num_core)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1488 logging.debug(f'... recon took {time()-t0:.2f} seconds!')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1489 if secondary_iters > 0:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1490 logging.debug(f'running {secondary_iters} secondary iterations')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1491 #options = {'method':'SIRT_CUDA', 'proj_type':'cuda', 'num_iter':secondary_iters}
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1492 #RV: doesn't work for me: "Error: CUDA error 803: system has unsupported display driver /
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1493 # cuda driver combination."
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1494 #options = {'method':'SIRT', 'proj_type':'linear', 'MinConstraint': 0, 'num_iter':secondary_iters}
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1495 #SIRT did not finish while running overnight
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1496 #options = {'method':'SART', 'proj_type':'linear', 'num_iter':secondary_iters}
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1497 options = {'method':'SART', 'proj_type':'linear', 'MinConstraint': 0,
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1498 'num_iter':secondary_iters}
39 8a3036b0a34c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit ebd998fc223b35670269b851ff6d12178ced3ee8" rv43 parents: 38 diff changeset	1499 t0 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1500 logging.debug(f'running recon on {num_core} cores ...')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1501 tomo_recon_stack = tomopy.recon(tomo_stack, thetas, centers,
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1502 init_recon=tomo_recon_stack, options=options, sinogram_order=True,
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1503 algorithm=tomopy.astra, ncore=num_core)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1504 logging.debug(f'... recon took {time()-t0:.2f} seconds!')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1505 if True:
39 8a3036b0a34c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit ebd998fc223b35670269b851ff6d12178ced3ee8" rv43 parents: 38 diff changeset	1506 t0 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1507 logging.debug(f'running remove_ring on {num_core} cores ...')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1508 tomopy.misc.corr.remove_ring(tomo_recon_stack, rwidth=ring_width, out=tomo_recon_stack,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1509 ncore=num_core)
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1510 logging.debug(f'... remove_ring took {time()-t0:.2f} seconds!')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1511 return tomo_recon_stack
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1512
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1513 def findImageFiles(self, tiff_to_h5_flag = False):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1514 """Find all available image files.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1515 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1516 self.is_valid = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1517
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1518 # Find dark field images
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1519 dark_field = self.config['dark_field']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1520 img_start, num_imgs, dark_files = findImageFiles(
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1521 dark_field['data_path'], self.config['data_filetype'], 'dark field')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1522 if img_start < 0 or num_imgs < 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1523 logging.error('Unable to find suitable dark field images')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1524 if dark_field['data_path']:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1525 self.is_valid = False
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1526 img_start_old = dark_field.get('img_start')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1527 num_imgs_old = dark_field.get('num')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1528 if num_imgs_old is None:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1529 dark_field['num'] = num_imgs
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1530 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1531 if num_imgs_old > num_imgs:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1532 logging.error('Inconsistent number of availaible dark field images')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1533 if dark_field['data_path']:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1534 self.is_valid = False
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1535 if img_start_old is None:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1536 dark_field['img_start'] = img_start
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1537 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1538 if img_start_old < img_start:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1539 logging.error('Inconsistent image start index for dark field images')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1540 if dark_field['data_path']:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1541 self.is_valid = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1542 logging.info(f'Number of dark field images = {dark_field["num"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1543 logging.info(f'Dark field image start index = {dark_field["img_start"]}')
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1544 if num_imgs and tiff_to_h5_flag and self.config['data_filetype'] == 'tif':
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1545 dark_files = combine_tiffs_in_h5(dark_files, num_imgs,
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1546 f'{self.config["work_folder"]}/dark_field.h5')
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1547 dark_field['data_path'] = dark_files[0]
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1548
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1549 # Find bright field images
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1550 bright_field = self.config['bright_field']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1551 img_start, num_imgs, bright_files = findImageFiles(
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1552 bright_field['data_path'], self.config['data_filetype'], 'bright field')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1553 if img_start < 0 or num_imgs < 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1554 logging.error('Unable to find suitable bright field images')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1555 self.is_valid = False
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1556 img_start_old = bright_field.get('img_start')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1557 num_imgs_old = bright_field.get('num')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1558 if num_imgs_old is None:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1559 bright_field['num'] = num_imgs
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1560 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1561 if num_imgs_old > num_imgs:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1562 logging.error('Inconsistent number of availaible bright field images')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1563 self.is_valid = False
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1564 if img_start_old is None:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1565 bright_field['img_start'] = img_start
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1566 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1567 if img_start_old < img_start:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1568 logging.warning('Inconsistent image start index for bright field images')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1569 self.is_valid = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1570 logging.info(f'Number of bright field images = {bright_field["num"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1571 logging.info(f'Bright field image start index = {bright_field["img_start"]}')
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1572 if num_imgs and tiff_to_h5_flag and self.config['data_filetype'] == 'tif':
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1573 bright_files = combine_tiffs_in_h5(bright_files, num_imgs,
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1574 f'{self.config["work_folder"]}/bright_field.h5')
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1575 bright_field['data_path'] = bright_files[0]
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1576
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1577 # Find tomography images
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1578 tomo_stack_files = []
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1579 for stack in self.config['stack_info']['stacks']:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1580 index = stack['index']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1581 img_start, num_imgs, tomo_files = findImageFiles(
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1582 stack['data_path'], self.config['data_filetype'], f'tomography set {index}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1583 if img_start < 0 or num_imgs < 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1584 logging.error('Unable to find suitable tomography images')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1585 self.is_valid = False
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1586 img_start_old = stack.get('img_start')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1587 num_imgs_old = stack.get('num')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1588 if num_imgs_old is None:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1589 stack['num'] = num_imgs
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1590 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1591 if num_imgs_old > num_imgs:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1592 logging.error('Inconsistent number of availaible tomography images')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1593 self.is_valid = False
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1594 if img_start_old is None:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1595 stack['img_start'] = img_start
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1596 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1597 if img_start_old < img_start:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1598 logging.warning('Inconsistent image start index for tomography images')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1599 self.is_valid = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1600 logging.info(f'Number of tomography images for set {index} = {stack["num"]}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1601 logging.info(f'Tomography set {index} image start index = {stack["img_start"]}')
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1602 if num_imgs and tiff_to_h5_flag and self.config['data_filetype'] == 'tif':
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1603 tomo_files = combine_tiffs_in_h5(tomo_files, num_imgs,
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1604 f'{self.config["work_folder"]}/tomo_field_{index}.h5')
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1605 stack['data_path'] = tomo_files[0]
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1606 tomo_stack_files.append(tomo_files)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1607 del tomo_files
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1608
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1609 # Safe updated config
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1610 if tiff_to_h5_flag:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1611 self.config['data_filetype'] == 'h5'
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1612 if self.is_valid:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1613 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1614
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1615 return dark_files, bright_files, tomo_stack_files
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1616
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1617 def loadTomoStacks(self, input_name, recon_flag=False):
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1618 """Load tomography stacks (only for Galaxy).
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1619 """
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1620 assert(self.galaxy_flag)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1621 t0 = time()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1622 logging.info(f'Loading preprocessed tomography stack from {input_name} ...')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1623 stack_info = self.config['stack_info']
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1624 stacks = stack_info.get('stacks')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1625 assert(len(self.tomo_stacks) == stack_info['num'])
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1626 with np.load(input_name) as f:
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1627 if recon_flag:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1628 for i,stack in enumerate(stacks):
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1629 self.tomo_recon_stacks[i] = f[f'set_{stack["index"]}']
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1630 logging.info(f'loaded stack {i}: index = {stack["index"]}, shape = '+
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1631 f'{self.tomo_recon_stacks[i].shape}')
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1632 else:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1633 for i,stack in enumerate(stacks):
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1634 self.tomo_stacks[i] = f[f'set_{stack["index"]}']
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1635 logging.info(f'loaded stack {i}: index = {stack["index"]}, shape = '+
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	1636 f'{self.tomo_stacks[i].shape}')
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1637 logging.info(f'... done in {time()-t0:.2f} seconds!')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1638
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1639 def genTomoStacks(self, galaxy_param=None, num_core=None):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1640 """Preprocess tomography images.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1641 """
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1642 if num_core is None:
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1643 num_core = self.num_core
33 afb612b64f26 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit c566c32f7c1506ab271ae08eb14a2982ee511938" rv43 parents: 32 diff changeset	1644 logging.info(f'num_core = {num_core}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1645 # Try loading any already preprocessed stacks (skip in Galaxy)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1646 # preprocessed stack order for each one in stack: row,theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1647 stack_info = self.config['stack_info']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1648 stacks = stack_info['stacks']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1649 num_tomo_stacks = stack_info['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1650 assert(num_tomo_stacks == len(self.tomo_stacks))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1651 available_stacks = [False]*num_tomo_stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1652 if self.galaxy_flag:
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1653 assert(isinstance(galaxy_param, dict))
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1654 tdf_files = galaxy_param['tdf_files']
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1655 tbf_files = galaxy_param['tbf_files']
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1656 tomo_stack_files = galaxy_param['tomo_stack_files']
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1657 assert(num_tomo_stacks == len(tomo_stack_files))
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1658 if not os.path.exists('setup_pngs'):
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1659 os.mkdir('setup_pngs')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1660 else:
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1661 if galaxy_param:
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1662 logging.warning('Ignoring galaxy_param in genTomoStacks (only for Galaxy)')
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1663 galaxy_param = None
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1664 tdf_files, tbf_files, tomo_stack_files = self.findImageFiles()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1665 if not self.is_valid:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1666 return
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1667 if self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1668 required = True
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1669 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1670 required = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1671 for i,stack in enumerate(stacks):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1672 if not self.tomo_stacks[i].size and stack.get('preprocessed', False):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1673 self.tomo_stacks[i], available_stacks[i] = \
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1674 self._loadTomo('red stack', stack['index'], required=required)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1675
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1676 # Preprocess any unloaded stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1677 if False in available_stacks:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1678 logging.debug('Preprocessing tomography images')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1679
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1680 # Check required image files (skip in Galaxy)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1681 if not self.galaxy_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1682 self._selectImageRanges(available_stacks)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1683 if not self.is_valid:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1684 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1685
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1686 # Generate dark field
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1687 if tdf_files:
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1688 self._genDark(tdf_files)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1689
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1690 # Generate bright field
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1691 self._genBright(tbf_files)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1692
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1693 # Set vertical detector bounds for image stack
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1694 self._setDetectorBounds(tomo_stack_files)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1695
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1696 # Set zoom and/or theta skip to reduce memory the requirement
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1697 self._setZoomOrSkip()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1698
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1699 # Generate tomography fields
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1700 self._genTomo(tomo_stack_files, available_stacks, num_core)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1701
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1702 # Save tomography stack to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1703 if self.galaxy_flag:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1704 t0 = time()
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	1705 output_name = galaxy_param['output_name']
2 b8977c98800b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0c40fb274acc13ebf7054379bb49ed13acec918e" rv43 parents: 1 diff changeset	1706 logging.info(f'Saving preprocessed tomography stack to {output_name} ...')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1707 save_stacks = {f'set_{stack["index"]}':tomo_stack
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1708 for stack,tomo_stack in zip(stacks,self.tomo_stacks)}
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1709 np.savez(output_name, **save_stacks)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1710 logging.info(f'... done in {time()-t0:.2f} seconds!')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1711
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1712 del available_stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1713
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1714 # Adjust sample reference height, update config and save to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1715 preprocess = self.config.get('preprocess')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1716 if preprocess is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1717 img_x_bounds = [0, self.tbf.shape[0]]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1718 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1719 img_x_bounds = preprocess.get('img_x_bounds', [0, self.tbf.shape[0]])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1720 pixel_size = self.config['detector']['pixel_size']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1721 if pixel_size is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1722 raise ValueError('Detector pixel size unavailable')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1723 pixel_size *= img_x_bounds[0]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1724 for stack in stacks:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1725 stack['ref_height'] = stack['ref_height']+pixel_size
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1726 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1727
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1728 def findCenters(self, galaxy_param=None, num_core=None):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1729 """Find rotation axis centers for the tomography stacks.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1730 """
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1731 if num_core is None:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1732 num_core = self.num_core
40 fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1733 logging.info(f'num_core = {num_core}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1734 logging.debug('Find centers for tomography stacks')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1735 stacks = self.config['stack_info']['stacks']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1736 available_stacks = [stack['index'] for stack in stacks if stack.get('preprocessed', False)]
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1737 logging.debug('Available stacks: {available_stacks}')
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1738 if self.galaxy_flag:
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1739 row_bounds = galaxy_param['row_bounds']
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1740 center_rows = galaxy_param['center_rows']
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1741 center_type_selector = galaxy_param['center_type_selector']
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1742 if center_type_selector:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1743 if center_type_selector == 'vo':
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1744 set_center = None
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1745 elif center_type_selector == 'user':
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1746 set_center = galaxy_param['set_center']
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1747 else:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1748 logging.error('Illegal center_type_selector entry in galaxy_param '+
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1749 f'({center_type_selector})')
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1750 galaxy_param['center_type_selector'] = None
40 fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1751 logging.debug(f'row_bounds = {row_bounds}')
fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1752 logging.debug(f'center_rows = {center_rows}')
fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1753 logging.debug(f'center_type_selector = {center_type_selector}')
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1754 else:
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1755 if galaxy_param:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1756 logging.warning('Ignoring galaxy_param in findCenters (only for Galaxy)')
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1757 galaxy_param = None
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1758
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1759 # Check for valid available center stack index
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1760 find_center = self.config.get('find_center')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1761 center_stack_index = None
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1762 if find_center and 'center_stack_index' in find_center:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1763 center_stack_index = find_center['center_stack_index']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1764 if (not isinstance(center_stack_index, int) or
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1765 center_stack_index not in available_stacks):
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1766 illegal_value(center_stack_index, 'find_center:center_stack_index', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1767 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1768 if self.test_mode:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1769 assert(find_center.get('completed', False) == False)
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1770 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1771 print('\nFound calibration center offset info for stack '+
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1772 f'{center_stack_index}')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1773 if (input_yesno('Do you want to use this again (y/n)?', 'y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1774 and find_center.get('completed', False) == True):
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1775 return
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1776
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1777 # Load the required preprocessed stack
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1778 # preprocessed stack order: row,theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1779 num_tomo_stacks = self.config['stack_info']['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1780 assert(len(stacks) == num_tomo_stacks)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1781 assert(len(self.tomo_stacks) == num_tomo_stacks)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1782 if num_tomo_stacks == 1:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1783 if not center_stack_index:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1784 center_stack_index = stacks[0]['index']
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1785 elif center_stack_index != stacks[0]['index']:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1786 raise ValueError(f'Inconsistent center_stack_index {center_stack_index}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1787 if not self.tomo_stacks[0].size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1788 self.tomo_stacks[0], available = self._loadTomo('red stack', center_stack_index,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1789 required=True)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1790 center_stack = self.tomo_stacks[0]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1791 if not center_stack.size:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1792 stacks[0]['preprocessed'] = False
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1793 raise OSError('Unable to load the required preprocessed tomography stack')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1794 assert(stacks[0].get('preprocessed', False) == True)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1795 elif self.galaxy_flag:
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1796 center_stack_index = stacks[int(num_tomo_stacks/2)]['index']
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1797 tomo_stack_index = available_stacks.index(center_stack_index)
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1798 center_stack = self.tomo_stacks[tomo_stack_index]
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1799 if not center_stack.size:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1800 stacks[tomo_stack_index]['preprocessed'] = False
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1801 raise OSError('Unable to load the required preprocessed tomography stack')
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1802 assert(stacks[tomo_stack_index].get('preprocessed', False) == True)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1803 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1804 while True:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1805 if not center_stack_index:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1806 center_stack_index = input_int('\nEnter tomography stack index to get '
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1807 f'rotation axis centers ({available_stacks})', inset=available_stacks)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1808 while center_stack_index not in available_stacks:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1809 center_stack_index = input_int('\nEnter tomography stack index to get '
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1810 f'rotation axis centers ({available_stacks})', inset=available_stacks)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1811 tomo_stack_index = available_stacks.index(center_stack_index)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1812 if not self.tomo_stacks[tomo_stack_index].size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1813 self.tomo_stacks[tomo_stack_index], available = self._loadTomo(
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1814 'red stack', center_stack_index, required=True)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1815 center_stack = self.tomo_stacks[tomo_stack_index]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1816 if not center_stack.size:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1817 stacks[tomo_stack_index]['preprocessed'] = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1818 logging.error(f'Unable to load the {center_stack_index}th '+
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1819 'preprocessed tomography stack, pick another one')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1820 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1821 break
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1822 assert(stacks[tomo_stack_index].get('preprocessed', False) == True)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1823 if find_center is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1824 self.config['find_center'] = {'center_stack_index' : center_stack_index}
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1825 find_center = self.config['find_center']
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1826 else:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1827 find_center['center_stack_index'] = center_stack_index
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1828 if not self.galaxy_flag:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1829 row_bounds = find_center.get('row_bounds', None)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1830 center_rows = [find_center.get('lower_row', None),
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1831 find_center.get('upper_row', None)]
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1832 if row_bounds is None:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1833 row_bounds = [0, center_stack.shape[0]]
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1834 if row_bounds[0] == -1:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1835 row_bounds[0] = 0
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1836 if row_bounds[1] == -1:
845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1837 row_bounds[1] = center_stack.shape[0]
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1838 if center_rows[0] == -1:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1839 center_rows[0] = 0
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1840 if center_rows[1] == -1:
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	1841 center_rows[1] = center_stack.shape[0]-1
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1842 if not is_index_range(row_bounds, 0, center_stack.shape[0]):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1843 illegal_value(row_bounds, 'row_bounds', 'Tomo:findCenters')
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1844 return
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1845
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1846 # Set thetas (in degrees)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1847 theta_range = self.config['theta_range']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1848 theta_start = theta_range['start']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1849 theta_end = theta_range['end']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1850 num_theta = theta_range['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1851 num_theta_skip = self.config['preprocess'].get('num_theta_skip', 0)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1852 thetas_deg = np.linspace(theta_start, theta_end, int(num_theta/(num_theta_skip+1)),
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1853 endpoint=False)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1854
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1855 # Get non-overlapping sample row boundaries
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1856 zoom_perc = self.config['preprocess'].get('zoom_perc', 100)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1857 pixel_size = self.config['detector']['pixel_size']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1858 if pixel_size is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1859 raise ValueError('Detector pixel size unavailable')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1860 eff_pixel_size = 100.*pixel_size/zoom_perc
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1861 logging.debug(f'eff_pixel_size = {eff_pixel_size}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1862 if num_tomo_stacks == 1:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1863 accept = True
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1864 if not self.test_mode and not self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1865 accept = False
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1866 print('\nSelect bounds for image reconstruction')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1867 if is_index_range(row_bounds, 0, center_stack.shape[0]):
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1868 a_tmp = np.copy(center_stack[:,0,:])
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1869 a_tmp_max = a_tmp.max()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1870 a_tmp[row_bounds[0],:] = a_tmp_max
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1871 a_tmp[row_bounds[1]-1,:] = a_tmp_max
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1872 print(f'lower bound = {row_bounds[0]} (inclusive)')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1873 print(f'upper bound = {row_bounds[1]} (exclusive)')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1874 quickImshow(a_tmp, title=f'center stack theta={theta_start}',
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1875 aspect='auto')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1876 del a_tmp
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1877 accept = input_yesno('Accept these bounds (y/n)?', 'y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1878 if accept:
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1879 n1 = row_bounds[0]
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1880 n2 = row_bounds[1]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1881 else:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1882 n1, n2 = selectImageBounds(center_stack[:,0,:], 0,
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1883 title=f'center stack theta={theta_start}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1884 else:
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1885 tomo_ref_heights = [stack['ref_height'] for stack in stacks]
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1886 n1 = int((1.+(tomo_ref_heights[0]+center_stack.shape[0]*eff_pixel_size-
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1887 tomo_ref_heights[1])/eff_pixel_size)/2)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1888 n2 = center_stack.shape[0]-n1
40 fa94fe25ca46 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 9978af8877d8e9722d252ab9bee4cbd4654aa100" rv43 parents: 39 diff changeset	1889 logging.debug(f'n1 = {n1}, n2 = {n2} (n2-n1) = {(n2-n1)*eff_pixel_size:.3f} mm')
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1890 if not self.test_mode and not self.galaxy_flag:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1891 tmp = center_stack[:,0,:]
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1892 tmp_max = tmp.max()
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1893 tmp[n1,:] = tmp_max
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1894 tmp[n2-1,:] = tmp_max
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1895 if is_index_range(center_rows, 0, tmp.shape[0]):
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1896 tmp[center_rows[0],:] = tmp_max
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1897 tmp[center_rows[1]-1,:] = tmp_max
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1898 extent = [0, tmp.shape[1], tmp.shape[0], 0]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1899 quickImshow(tmp, title=f'center stack theta={theta_start}',
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1900 path=self.output_folder, extent=extent, save_fig=self.save_plots,
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1901 save_only=self.save_plots_only, aspect='auto')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1902 del tmp
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1903 #extent = [0, center_stack.shape[2], n2, n1]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1904 #quickImshow(center_stack[n1:n2,0,:], title=f'center stack theta={theta_start}',
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1905 # path=self.output_folder, extent=extent, save_fig=self.save_plots,
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1906 # save_only=self.save_plots_only, show_grid=True, aspect='auto')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1907
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1908 # Get cross sectional diameter in mm
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1909 cross_sectional_dim = center_stack.shape[2]*eff_pixel_size
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1910 logging.debug(f'cross_sectional_dim = {cross_sectional_dim}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1911
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1912 # Determine center offset at sample row boundaries
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1913 logging.info('Determine center offset at sample row boundaries')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1914
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1915 # Lower row center
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1916 use_row = False
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1917 use_center = False
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1918 row = center_rows[0]
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1919 if self.test_mode or self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1920 assert(is_int(row, n1, n2-2))
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1921 if is_int(row, n1, n2-2):
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1922 if self.test_mode or self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1923 use_row = True
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1924 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1925 quickImshow(center_stack[:,0,:], title=f'theta={theta_start}', aspect='auto')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1926 use_row = input_yesno('\nCurrent row index for lower center = '
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1927 f'{row}, use this value (y/n)?', 'y')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1928 if self.save_plots_only:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1929 clearImshow(f'theta={theta_start}')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1930 if use_row:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1931 center_offset = find_center.get('lower_center_offset')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1932 if is_num(center_offset):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1933 use_center = input_yesno('Current lower center offset = '+
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1934 f'{center_offset}, use this value (y/n)?', 'y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1935 if not use_center:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1936 if not use_row:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1937 if not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1938 quickImshow(center_stack[:,0,:], title=f'theta={theta_start}',
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1939 aspect='auto')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1940 row = input_int('\nEnter row index to find lower center', n1, n2-2, n1)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1941 if row == '':
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1942 row = n1
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1943 if self.save_plots_only:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1944 clearImshow(f'theta={theta_start}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1945 # center_stack order: row,theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1946 center_offset = self._findCenterOnePlane(center_stack[row,:,:], row, thetas_deg,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1947 eff_pixel_size, cross_sectional_dim, num_core=num_core,
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1948 galaxy_param=galaxy_param)
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1949 logging.info(f'lower_center_offset = {center_offset:.2f} {type(center_offset)}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1950
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1951 # Update config and save to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1952 find_center['row_bounds'] = [n1, n2]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1953 find_center['lower_row'] = row
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1954 find_center['lower_center_offset'] = center_offset
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1955 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1956 lower_row = row
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1957
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1958 # Upper row center
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1959 use_row = False
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1960 use_center = False
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1961 row = center_rows[1]
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1962 if self.test_mode or self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1963 assert(is_int(row, lower_row+1, n2-1))
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1964 if is_int(row, lower_row+1, n2-1):
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1965 if self.test_mode or self.galaxy_flag:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1966 use_row = True
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1967 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1968 quickImshow(center_stack[:,0,:], title=f'theta={theta_start}', aspect='auto')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1969 use_row = input_yesno('\nCurrent row index for upper center = '
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1970 f'{row}, use this value (y/n)?', 'y')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1971 if self.save_plots_only:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1972 clearImshow(f'theta={theta_start}')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1973 if use_row:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1974 center_offset = find_center.get('upper_center_offset')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1975 if is_num(center_offset):
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	1976 use_center = input_yesno('Current upper center offset = '+
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1977 f'{center_offset}, use this value (y/n)?', 'y')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1978 if not use_center:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1979 if not use_row:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1980 if not self.test_mode:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1981 quickImshow(center_stack[:,0,:], title=f'theta={theta_start}',
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	1982 aspect='auto')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1983 row = input_int('\nEnter row index to find upper center', lower_row+1, n2-1, n2-1)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1984 if row == '':
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1985 row = n2-1
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1986 if self.save_plots_only:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	1987 clearImshow(f'theta={theta_start}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1988 # center_stack order: row,theta,column
24 817a8101d2e4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 3d7910473008ec94bc0b4580d62e61310e69917d" rv43 parents: 23 diff changeset	1989 center_offset = self._findCenterOnePlane(center_stack[row,:,:], row, thetas_deg,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	1990 eff_pixel_size, cross_sectional_dim, num_core=num_core,
13 40395e60d2be "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e61a62f5b5950b3142fedad2c0e49a00f66f9980" rv43 parents: 11 diff changeset	1991 galaxy_param=galaxy_param)
7 845270a96464 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit a08ed1264ea7b7c878ba96801e4b27ef621fb6be" rv43 parents: 3 diff changeset	1992 logging.info(f'upper_center_offset = {center_offset:.2f}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1993 del center_stack
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1994
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1995 # Update config and save to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1996 find_center['upper_row'] = row
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1997 find_center['upper_center_offset'] = center_offset
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1998 find_center['completed'] = True
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	1999 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2000
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2001 def checkCenters(self):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2002 """Check centers for the tomography stacks.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2003 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2004 #RV TODO load all stacks and check at all stack boundaries
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2005 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2006 logging.debug('Check centers for tomography stacks')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2007 center_stack_index = self.config.get('center_stack_index')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2008 if center_stack_index is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2009 raise ValueError('Unable to read center_stack_index from config')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2010 center_stack_index = self.tomo_stacks[self.tomo_data_indices.index(center_stack_index)]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2011 lower_row = self.config.get('lower_row')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2012 if lower_row is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2013 raise ValueError('Unable to read lower_row from config')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2014 lower_center_offset = self.config.get('lower_center_offset')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2015 if lower_center_offset is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2016 raise ValueError('Unable to read lower_center_offset from config')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2017 upper_row = self.config.get('upper_row')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2018 if upper_row is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2019 raise ValueError('Unable to read upper_row from config')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2020 upper_center_offset = self.config.get('upper_center_offset')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2021 if upper_center_offset is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2022 raise ValueError('Unable to read upper_center_offset from config')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2023 center_slope = (upper_center_offset-lower_center_offset)/(upper_row-lower_row)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2024 shift = upper_center_offset-lower_center_offset
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2025 if lower_row == 0:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2026 logging.warning(f'lower_row == 0: one row offset between both planes')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2027 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2028 lower_row -= 1
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2029 lower_center_offset -= center_slope
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2030
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2031 # stack order: stack,row,theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2032 if center_stack_index:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2033 stack1 = self.tomo_stacks[center_stack_index-1]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2034 stack2 = self.tomo_stacks[center_stack_index]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2035 if not stack1.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2036 logging.error(f'Unable to load required tomography stack {stack1}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2037 elif not stack2.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2038 logging.error(f'Unable to load required tomography stack {stack1}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2039 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2040 assert(0 <= lower_row < stack2.shape[0])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2041 assert(0 <= upper_row < stack1.shape[0])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2042 plane1 = stack1[upper_row,:]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2043 plane2 = stack2[lower_row,:]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2044 for i in range(-2, 3):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2045 shift_i = shift+2*i
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2046 plane1_shifted = spi.shift(plane2, [0, shift_i])
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2047 quickPlot((plane1[0,:],), (plane1_shifted[0,:],),
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2048 title=f'stacks {stack1} {stack2} shifted {2*i} theta={self.start_theta}',
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2049 path=self.output_folder, save_fig=self.save_plots,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2050 save_only=self.save_plots_only)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2051 if center_stack_index < self.num_tomo_stacks-1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2052 stack1 = self.tomo_stacks[center_stack_index]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2053 stack2 = self.tomo_stacks[center_stack_index+1]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2054 if not stack1.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2055 logging.error(f'Unable to load required tomography stack {stack1}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2056 elif not stack2.size:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2057 logging.error(f'Unable to load required tomography stack {stack1}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2058 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2059 assert(0 <= lower_row < stack2.shape[0])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2060 assert(0 <= upper_row < stack1.shape[0])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2061 plane1 = stack1[upper_row,:]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2062 plane2 = stack2[lower_row,:]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2063 for i in range(-2, 3):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2064 shift_i = -shift+2*i
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2065 plane1_shifted = spi.shift(plane2, [0, shift_i])
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2066 quickPlot((plane1[0,:],), (plane1_shifted[0,:],),
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2067 title=f'stacks {stack1} {stack2} shifted {2*i} theta={start_theta}',
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2068 path=self.output_folder, save_fig=self.save_plots,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2069 save_only=self.save_plots_only)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2070 del plane1, plane2, plane1_shifted
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2071
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2072 # Update config file
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2073 self.config = update('config.txt', 'check_centers', True, 'find_centers')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2074
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2075 def reconstructTomoStacks(self, galaxy_param=None, num_core=None):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2076 """Reconstruct tomography stacks.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2077 """
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2078 if num_core is None:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2079 num_core = self.num_core
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	2080 logging.info(f'num_core = {num_core}')
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2081 if self.galaxy_flag:
3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2082 assert(galaxy_param)
3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2083 if not os.path.exists('center_slice_pngs'):
3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2084 os.mkdir('center_slice_pngs')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2085 logging.debug('Reconstruct tomography stacks')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2086 stacks = self.config['stack_info']['stacks']
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2087 assert(len(self.tomo_stacks) == self.config['stack_info']['num'])
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2088 assert(len(self.tomo_stacks) == len(stacks))
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2089 assert(len(self.tomo_recon_stacks) == len(stacks))
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2090 if self.galaxy_flag:
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2091 assert(isinstance(galaxy_param, dict))
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2092 # Get rotation axis centers
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2093 center_offsets = galaxy_param['center_offsets']
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2094 assert(isinstance(center_offsets, list) and len(center_offsets) == 2)
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2095 lower_center_offset = center_offsets[0]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2096 assert(is_num(lower_center_offset))
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2097 upper_center_offset = center_offsets[1]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2098 assert(is_num(upper_center_offset))
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2099 else:
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2100 if galaxy_param:
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2101 logging.warning('Ignoring galaxy_param in reconstructTomoStacks (only for Galaxy)')
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2102 galaxy_param = None
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2103 lower_center_offset = None
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2104 upper_center_offset = None
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2105
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2106 # Get rotation axis rows and centers
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2107 find_center = self.config['find_center']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2108 lower_row = find_center.get('lower_row')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2109 if lower_row is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2110 logging.error('Unable to read lower_row from config')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2111 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2112 upper_row = find_center.get('upper_row')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2113 if upper_row is None:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2114 logging.error('Unable to read upper_row from config')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2115 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2116 logging.debug(f'lower_row = {lower_row} upper_row = {upper_row}')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2117 assert(lower_row < upper_row)
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2118 if lower_center_offset is None:
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2119 lower_center_offset = find_center.get('lower_center_offset')
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2120 if lower_center_offset is None:
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2121 logging.error('Unable to read lower_center_offset from config')
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2122 return
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2123 if upper_center_offset is None:
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2124 upper_center_offset = find_center.get('upper_center_offset')
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2125 if upper_center_offset is None:
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2126 logging.error('Unable to read upper_center_offset from config')
1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2127 return
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2128 center_slope = (upper_center_offset-lower_center_offset)/(upper_row-lower_row)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2129
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2130 # Set thetas (in radians)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2131 theta_range = self.config['theta_range']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2132 theta_start = theta_range['start']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2133 theta_end = theta_range['end']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2134 num_theta = theta_range['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2135 num_theta_skip = self.config['preprocess'].get('num_theta_skip', 0)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2136 thetas = np.radians(np.linspace(theta_start, theta_end,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2137 int(num_theta/(num_theta_skip+1)), endpoint=False))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2138
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2139 # Reconstruct tomo stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2140 zoom_perc = self.config['preprocess'].get('zoom_perc', 100)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2141 if zoom_perc == 100:
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2142 basetitle = 'recon stack fullres'
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2143 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2144 basetitle = f'recon stack {zoom_perc}p'
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2145 load_error = False
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2146 for i,stack in enumerate(stacks):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2147 # Check if stack can be loaded
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2148 # reconstructed stack order for each one in stack : row/z,x,y
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2149 # preprocessed stack order for each one in stack: row,theta,column
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2150 index = stack['index']
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2151 if not self.galaxy_flag:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2152 available = False
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2153 if stack.get('reconstructed', False):
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2154 self.tomo_recon_stacks[i], available = self._loadTomo('recon stack', index)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2155 if self.tomo_recon_stacks[i].size or available:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2156 if self.tomo_stacks[i].size:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2157 self.tomo_stacks[i] = np.array([])
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2158 assert(stack.get('preprocessed', False) == True)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2159 assert(stack.get('reconstructed', False) == True)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2160 continue
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2161 stack['reconstructed'] = False
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2162 if not self.tomo_stacks[i].size:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2163 self.tomo_stacks[i], available = self._loadTomo('red stack', index,
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2164 required=True)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2165 if not self.tomo_stacks[i].size:
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2166 logging.error(f'Unable to load tomography stack {index} for reconstruction')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2167 stack[i]['preprocessed'] = False
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2168 load_error = True
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2169 continue
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2170 assert(0 <= lower_row < upper_row < self.tomo_stacks[i].shape[0])
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2171 center_offsets = [lower_center_offset-lower_row*center_slope,
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2172 upper_center_offset+(self.tomo_stacks[i].shape[0]-1-upper_row)*center_slope]
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2173 t0 = time()
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	2174 logging.debug(f'running _reconstructOneTomoStack on {num_core} cores ...')
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2175 self.tomo_recon_stacks[i]= self._reconstructOneTomoStack(self.tomo_stacks[i],
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2176 thetas, center_offsets=center_offsets, num_core=num_core,
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2177 algorithm='gridrec')
41 ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	2178 logging.debug(f'... _reconstructOneTomoStack took {time()-t0:.2f} seconds!')
ef5c2f7b49ec "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 0d207c80e38a6019595ebe178f5678372b75f3e7" rv43 parents: 40 diff changeset	2179 logging.info(f'Reconstruction of stack {index} took {time()-t0:.2f} seconds!')
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2180 if self.galaxy_flag:
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2181 x_slice = int(self.tomo_recon_stacks[i].shape[0]/2)
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2182 title = f'{basetitle} {index} xslice{x_slice}'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2183 quickImshow(self.tomo_recon_stacks[i][x_slice,:,:], title=title,
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2184 path='center_slice_pngs', save_fig=True, save_only=True)
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2185 y_slice = int(self.tomo_recon_stacks[i].shape[1]/2)
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2186 title = f'{basetitle} {index} yslice{y_slice}'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2187 quickImshow(self.tomo_recon_stacks[i][:,y_slice,:], title=title,
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2188 path='center_slice_pngs', save_fig=True, save_only=True)
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2189 z_slice = int(self.tomo_recon_stacks[i].shape[2]/2)
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2190 title = f'{basetitle} {index} zslice{z_slice}'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2191 quickImshow(self.tomo_recon_stacks[i][:,:,z_slice], title=title,
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2192 path='center_slice_pngs', save_fig=True, save_only=True)
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2193 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2194 x_slice = int(self.tomo_recon_stacks[i].shape[0]/2)
21 3caba2116858 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit e32e37ee46f47b185f924884685eb1d464761e86" rv43 parents: 20 diff changeset	2195 title = f'{basetitle} {index} xslice{x_slice}'
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2196 quickImshow(self.tomo_recon_stacks[i][x_slice,:,:], title=title,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2197 path=self.output_folder, save_fig=self.save_plots,
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2198 save_only=self.save_plots_only)
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2199 y_slice = int(self.tomo_recon_stacks[i].shape[1]/2)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2200 title = f'{basetitle} {index} yslice{y_slice}'
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2201 quickImshow(self.tomo_recon_stacks[i][:,y_slice,:], title=title,
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2202 path=self.output_folder, save_fig=self.save_plots,
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2203 save_only=self.save_plots_only)
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2204 z_slice = int(self.tomo_recon_stacks[i].shape[2]/2)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2205 title = f'{basetitle} {index} zslice{z_slice}'
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2206 quickImshow(self.tomo_recon_stacks[i][:,:,z_slice], title=title,
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2207 path=self.output_folder, save_fig=self.save_plots,
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2208 save_only=self.save_plots_only)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2209 # quickPlot(self.tomo_recon_stacks[i]
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2210 # [x_slice,int(self.tomo_recon_stacks[i].shape[1]/2),:],
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2211 # title=f'{title} cut{int(self.tomo_recon_stacks[i].shape[1]/2)}',
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2212 # path=self.output_folder, save_fig=self.save_plots,
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2213 # save_only=self.save_plots_only)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2214 if not self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2215 self._saveTomo('recon stack', self.tomo_recon_stacks[i], index)
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2216 self.tomo_stacks[i] = np.array([])
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2217
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2218 # Update config and save to file
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2219 stack['reconstructed'] = True
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2220 combine_stacks = self.config.get('combine_stacks')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2221 if combine_stacks and index in combine_stacks.get('stacks', []):
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2222 combine_stacks['stacks'].remove(index)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2223 self.cf.saveFile(self.config_out)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2224
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2225 if self.galaxy_flag:
47 c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2226 # Save reconstructed tomography stack to file
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2227 t0 = time()
15 1bcca1f2adb4 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 38c19bf5addbc46f45d598f981bb1a48f7bca691" rv43 parents: 13 diff changeset	2228 output_name = galaxy_param['output_name']
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2229 logging.info(f'Saving reconstructed tomography stack to {output_name} ...')
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2230 save_stacks = {f'set_{stack["index"]}':tomo_stack
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2231 for stack,tomo_stack in zip(stacks,self.tomo_recon_stacks)}
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2232 np.savez(output_name, **save_stacks)
f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2233 logging.info(f'... done in {time()-t0:.2f} seconds!')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2234
47 c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2235 # Create cross section profile in yz-plane
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2236 tomosum = 0
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2237 [tomosum := tomosum+np.sum(tomo_recon_stack, axis=(0,2)) for tomo_recon_stack in
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2238 self.tomo_recon_stacks]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2239 quickPlot(tomosum, title='recon stack sum yz', path='center_slice_pngs',
47 c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2240 save_fig=True, save_only=True)
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2241
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2242 # Create cross section profile in xz-plane
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2243 tomosum = 0
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2244 [tomosum := tomosum+np.sum(tomo_recon_stack, axis=(0,1)) for tomo_recon_stack in
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2245 self.tomo_recon_stacks]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2246 quickPlot(tomosum, title='recon stack sum xz', path='center_slice_pngs',
47 c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2247 save_fig=True, save_only=True)
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2248
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2249 # Create cross section profile in xy-plane
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2250 num_tomo_stacks = len(stacks)
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2251 row_bounds = self.config['find_center']['row_bounds']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2252 if not is_index_range(row_bounds, 0, self.tomo_recon_stacks[0].shape[0]):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2253 illegal_value(row_bounds, 'find_center:row_bounds', 'config file')
47 c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2254 return
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2255 if num_tomo_stacks == 1:
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2256 low_bound = row_bounds[0]
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2257 else:
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2258 low_bound = 0
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2259 tomosum = np.sum(self.tomo_recon_stacks[0][low_bound:row_bounds[1],:,:], axis=(1,2))
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2260 if num_tomo_stacks > 2:
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2261 tomosum = np.concatenate([tomosum]+
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2262 [np.sum(self.tomo_recon_stacks[i][row_bounds[0]:row_bounds[1],:,:],
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2263 axis=(1,2)) for i in range(1, num_tomo_stacks-1)])
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2264 if num_tomo_stacks > 1:
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2265 tomosum = np.concatenate([tomosum,
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2266 np.sum(self.tomo_recon_stacks[num_tomo_stacks-1][row_bounds[0]:,:,:],
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2267 axis=(1,2))])
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2268 quickPlot(tomosum, title='recon stack sum xy', path='center_slice_pngs',
47 c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2269 save_fig=True, save_only=True)
c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2270
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2271 def combineTomoStacks(self, galaxy_param=None):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2272 """Combine the reconstructed tomography stacks.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2273 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2274 # stack order: stack,row(z),x,y
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2275 if self.galaxy_flag:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2276 assert(galaxy_param)
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2277 if not os.path.exists('combine_pngs'):
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2278 os.mkdir('combine_pngs')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2279 logging.debug('Combine reconstructed tomography stacks')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2280 stack_info = self.config['stack_info']
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2281 stacks = stack_info['stacks']
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2282 assert(len(self.tomo_recon_stacks) == stack_info['num'])
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2283 assert(len(self.tomo_recon_stacks) == len(stacks))
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2284 if self.galaxy_flag:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2285 assert(isinstance(galaxy_param, dict))
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2286 # Get image bounds
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2287 x_bounds = galaxy_param['x_bounds']
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2288 assert(isinstance(x_bounds, list) and len(x_bounds) == 2)
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2289 y_bounds = galaxy_param['y_bounds']
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2290 assert(isinstance(y_bounds, list) and len(y_bounds) == 2)
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2291 z_bounds = galaxy_param['z_bounds']
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2292 assert(isinstance(z_bounds, list) and len(z_bounds) == 2)
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2293 else:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2294 if galaxy_param:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2295 logging.warning('Ignoring galaxy_param in reconstructTomoStacks (only for Galaxy)')
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2296 galaxy_param = None
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2297
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2298 # Load any unloaded reconstructed stacks
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2299 for i,stack in enumerate(stacks):
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2300 available = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2301 if not self.tomo_recon_stacks[i].size and stack.get('reconstructed', False):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2302 self.tomo_recon_stacks[i], available = self._loadTomo('recon stack',
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2303 stack['index'], required=True)
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2304 if not self.tomo_recon_stacks[i].size:
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2305 logging.error(f'Unable to load reconstructed stack {stack["index"]}')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2306 stack['reconstructed'] = False
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2307 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2308 if i:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2309 if (self.tomo_recon_stacks[i].shape[1] != self.tomo_recon_stacks[0].shape[1] or
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2310 self.tomo_recon_stacks[i].shape[1] != self.tomo_recon_stacks[0].shape[1]):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2311 logging.error('Incompatible reconstructed tomography stack dimensions')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2312 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2313
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2314 # Get center stack boundaries
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2315 row_bounds = self.config['find_center']['row_bounds']
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2316 if not is_index_range(row_bounds, 0, self.tomo_recon_stacks[0].shape[0]):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2317 illegal_value(row_bounds, 'find_center:row_bounds', 'config file')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2318 return
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2319
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2320 # Selecting x bounds (in yz-plane)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2321 tomosum = 0
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2322 #RV FIX :=
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2323 [tomosum := tomosum+np.sum(tomo_recon_stack, axis=(0,2)) for tomo_recon_stack in
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2324 self.tomo_recon_stacks]
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2325 combine_stacks = self.config.get('combine_stacks')
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2326 if self.galaxy_flag:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2327 if x_bounds[0] == -1:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2328 x_bounds[0] = 0
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2329 if x_bounds[1] == -1:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2330 x_bounds[1] = tomosum.size
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2331 if not is_index_range(x_bounds, 0, tomosum.size):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2332 illegal_value(x_bounds, 'x_bounds', 'galaxy input')
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2333 tomosum_min = tomosum.min()
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2334 tomosum_max = tomosum.max()
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2335 quickPlot((range(tomosum.size), tomosum),
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2336 ([x_bounds[0], x_bounds[0]], [tomosum_min, tomosum_max], 'r-'),
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2337 ([x_bounds[1]-1, x_bounds[1]-1], [tomosum_min, tomosum_max], 'r-'),
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2338 title=f'recon stack sum yz', path='combine_pngs', save_fig=True, save_only=True)
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2339 else:
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2340 x_bounds = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2341 change_x_bounds = 'y'
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2342 if combine_stacks and 'x_bounds' in combine_stacks:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2343 x_bounds = combine_stacks['x_bounds']
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2344 if is_index_range(x_bounds, 0, tomosum.size):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2345 if not self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2346 quickPlot(tomosum, vlines=x_bounds, title='recon stack sum yz')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2347 print(f'x_bounds = {x_bounds} (lower bound inclusive, upper bound '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2348 'exclusive)')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2349 change_x_bounds = 'n'
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2350 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2351 illegal_value(x_bounds, 'combine_stacks:x_bounds', 'config file')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2352 x_bounds = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2353 if self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2354 if x_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2355 x_bounds = [0, tomosum.size]
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2356 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2357 if not input_yesno('\nDo you want to change the image x-bounds (y/n)?',
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2358 change_x_bounds):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2359 if x_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2360 x_bounds = [0, tomosum.size]
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2361 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2362 accept = False
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2363 if x_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2364 index_ranges = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2365 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2366 index_ranges = [x_bounds]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2367 while not accept:
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2368 mask, x_bounds = draw_mask_1d(tomosum, current_index_ranges=index_ranges,
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2369 title='select x data range', legend='recon stack sum yz')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2370 while len(x_bounds) != 1:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2371 print('Please select exactly one continuous range')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2372 mask, x_bounds = draw_mask_1d(tomosum, title='select x data range',
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2373 legend='recon stack sum yz')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2374 x_bounds = list(x_bounds[0])
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2375 quickPlot(tomosum, vlines=x_bounds, title='recon stack sum yz')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2376 print(f'x_bounds = {x_bounds} (lower bound inclusive, upper bound '+
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2377 'exclusive)')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2378 accept = input_yesno('Accept these bounds (y/n)?', 'y')
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2379 if self.save_plots_only:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2380 clearPlot('recon stack sum yz')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2381 logging.info(f'x_bounds = {x_bounds}')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2382
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2383 # Selecting y bounds (in xz-plane)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2384 tomosum = 0
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2385 #RV FIX :=
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2386 [tomosum := tomosum+np.sum(tomo_recon_stack, axis=(0,1)) for tomo_recon_stack in
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2387 self.tomo_recon_stacks]
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2388 if self.galaxy_flag:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2389 if y_bounds[0] == -1:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2390 y_bounds[0] = 0
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2391 if y_bounds[1] == -1:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2392 y_bounds[1] = tomosum.size
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2393 if not is_index_range(y_bounds, 0, tomosum.size):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2394 illegal_value(y_bounds, 'y_bounds', 'galaxy input')
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2395 tomosum_min = tomosum.min()
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2396 tomosum_max = tomosum.max()
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2397 quickPlot((range(tomosum.size), tomosum),
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2398 ([y_bounds[0], y_bounds[0]], [tomosum_min, tomosum_max], 'r-'),
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2399 ([y_bounds[1]-1, y_bounds[1]-1], [tomosum_min, tomosum_max], 'r-'),
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2400 title=f'recon stack sum xz', path='combine_pngs', save_fig=True, save_only=True)
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2401 else:
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2402 y_bounds = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2403 change_y_bounds = 'y'
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2404 if combine_stacks and 'y_bounds' in combine_stacks:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2405 y_bounds = combine_stacks['y_bounds']
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2406 if is_index_range(y_bounds, 0, tomosum.size):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2407 if not self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2408 quickPlot(tomosum, vlines=y_bounds, title='recon stack sum xz')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2409 print(f'y_bounds = {y_bounds} (lower bound inclusive, upper bound '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2410 'exclusive)')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2411 change_y_bounds = 'n'
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2412 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2413 illegal_value(y_bounds, 'combine_stacks:y_bounds', 'config file')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2414 y_bounds = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2415 if self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2416 if y_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2417 y_bounds = [0, tomosum.size]
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2418 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2419 if not input_yesno('\nDo you want to change the image y-bounds (y/n)?',
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2420 change_y_bounds):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2421 if y_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2422 y_bounds = [0, tomosum.size]
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2423 else:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2424 accept = False
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2425 if y_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2426 index_ranges = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2427 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2428 index_ranges = [y_bounds]
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2429 while not accept:
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2430 mask, y_bounds = draw_mask_1d(tomosum, current_index_ranges=index_ranges,
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2431 title='select x data range', legend='recon stack sum xz')
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2432 while len(y_bounds) != 1:
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2433 print('Please select exactly one continuous range')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2434 mask, y_bounds = draw_mask_1d(tomosum, title='select x data range',
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2435 legend='recon stack sum xz')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2436 y_bounds = list(y_bounds[0])
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2437 quickPlot(tomosum, vlines=y_bounds, title='recon stack sum xz')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2438 print(f'y_bounds = {y_bounds} (lower bound inclusive, upper bound '+
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2439 'exclusive)')
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2440 accept = input_yesno('Accept these bounds (y/n)?', 'y')
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2441 if self.save_plots_only:
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2442 clearPlot('recon stack sum xz')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2443 logging.info(f'y_bounds = {y_bounds}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2444
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2445 # Combine reconstructed tomography stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2446 logging.info(f'Combining reconstructed stacks ...')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2447 t0 = time()
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2448 num_tomo_stacks = len(stacks)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2449 if num_tomo_stacks == 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2450 low_bound = row_bounds[0]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2451 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2452 low_bound = 0
47 c693117d01e9 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit baa3866a541a552710bec26ae5d37a8c624563ac-dirty" rv43 parents: 46 diff changeset	2453 tomo_recon_combined = self.tomo_recon_stacks[0][low_bound:row_bounds[1],
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2454 x_bounds[0]:x_bounds[1],y_bounds[0]:y_bounds[1]]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2455 if num_tomo_stacks > 2:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2456 tomo_recon_combined = np.concatenate([tomo_recon_combined]+
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2457 [self.tomo_recon_stacks[i][row_bounds[0]:row_bounds[1],
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2458 x_bounds[0]:x_bounds[1],y_bounds[0]:y_bounds[1]]
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2459 for i in range(1, num_tomo_stacks-1)])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2460 if num_tomo_stacks > 1:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2461 tomo_recon_combined = np.concatenate([tomo_recon_combined]+
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2462 [self.tomo_recon_stacks[num_tomo_stacks-1][row_bounds[0]:,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2463 x_bounds[0]:x_bounds[1],y_bounds[0]:y_bounds[1]]])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2464 logging.info(f'... done in {time()-t0:.2f} seconds!')
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2465 combined_stacks = [stack['index'] for stack in stacks]
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2466
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2467 # Selecting z bounds (in xy-plane)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2468 tomosum = np.sum(tomo_recon_combined, axis=(1,2))
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2469 if self.galaxy_flag:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2470 if z_bounds[0] == -1:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2471 z_bounds[0] = 0
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2472 if z_bounds[1] == -1:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2473 z_bounds[1] = tomosum.size
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2474 if not is_index_range(z_bounds, 0, tomosum.size):
26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2475 illegal_value(z_bounds, 'z_bounds', 'galaxy input')
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2476 tomosum_min = tomosum.min()
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2477 tomosum_max = tomosum.max()
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2478 quickPlot((range(tomosum.size), tomosum),
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2479 ([z_bounds[0], z_bounds[0]], [tomosum_min, tomosum_max], 'r-'),
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2480 ([z_bounds[1]-1, z_bounds[1]-1], [tomosum_min, tomosum_max], 'r-'),
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2481 title=f'recon stack sum xy', path='combine_pngs', save_fig=True, save_only=True)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2482 else:
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2483 z_bounds = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2484 if combine_stacks and 'z_bounds' in combine_stacks:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2485 z_bounds = combine_stacks['z_bounds']
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2486 if is_index_range(z_bounds, 0, tomosum.size):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2487 if not self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2488 quickPlot(tomosum, vlines=z_bounds, title='recon stack sum xy')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2489 print(f'z_bounds = {z_bounds} (lower bound inclusive, upper bound '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2490 'exclusive)')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2491 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2492 illegal_value(z_bounds, 'combine_stacks:z_bounds', 'config file')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2493 z_bounds = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2494 if self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2495 if z_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2496 z_bounds = [0, tomosum.size]
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2497 else:
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2498 if not input_yesno('\nDo you want to change the image z-bounds (y/n)?', 'n'):
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2499 if z_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2500 z_bounds = [0, tomosum.size]
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2501 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2502 accept = False
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2503 if z_bounds is None:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2504 index_ranges = None
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2505 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2506 index_ranges = [z_bounds]
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2507 while not accept:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2508 mask, z_bounds = draw_mask_1d(tomosum, current_index_ranges=index_ranges,
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2509 title='select x data range', legend='recon stack sum xy')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2510 while len(z_bounds) != 1:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2511 print('Please select exactly one continuous range')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2512 mask, z_bounds = draw_mask_1d(tomosum, title='select x data range',
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2513 legend='recon stack sum xy')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2514 z_bounds = list(z_bounds[0])
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2515 quickPlot(tomosum, vlines=z_bounds, title='recon stack sum xy')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2516 print(f'z_bounds = {z_bounds} (lower bound inclusive, upper bound '+
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2517 'exclusive)')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2518 accept = input_yesno('Accept these bounds (y/n)?', 'y')
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2519 if self.save_plots_only:
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2520 clearPlot('recon stack sum xy')
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2521 logging.info(f'z_bounds = {z_bounds}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2522
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2523 # Plot center slices
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2524 if self.galaxy_flag:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2525 path = 'combine_pngs'
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2526 save_fig = True
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2527 save_only = True
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2528 else:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2529 path = self.output_folder
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2530 save_fig = self.save_plots
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2531 save_only = self.save_plots_only
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2532 quickImshow(tomo_recon_combined[int(tomo_recon_combined.shape[0]/2),:,:],
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2533 title=f'recon combined xslice{int(tomo_recon_combined.shape[0]/2)}',
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2534 path=path, save_fig=save_fig, save_only=save_only)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2535 quickImshow(tomo_recon_combined[:,int(tomo_recon_combined.shape[1]/2),:],
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2536 title=f'recon combined yslice{int(tomo_recon_combined.shape[1]/2)}',
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2537 path=path, save_fig=save_fig, save_only=save_only)
49 26f99fdd8d61 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 4f7738d02f4a3fd91373f43937ed311b6fe11a12" rv43 parents: 48 diff changeset	2538 quickImshow(tomo_recon_combined[:,:,int(tomo_recon_combined.shape[2]/2)],
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2539 title=f'recon combined zslice{int(tomo_recon_combined.shape[2]/2)}',
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2540 path=path, save_fig=save_fig, save_only=save_only)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2541
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2542 # Save combined reconstructed tomography stack or test mode data to file
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2543 if self.galaxy_flag:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2544 t0 = time()
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2545 output_name = galaxy_param['output_name']
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2546 logging.info(f'Saving combined reconstructed tomography stack to {output_name} ...')
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2547 np.save(output_name, tomo_recon_combined)
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2548 logging.info(f'... done in {time()-t0:.2f} seconds!')
61 75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2549 elif self.test_mode:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2550 zoom_perc = self.config['preprocess'].get('zoom_perc', 100)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2551 filename = 'recon combined sum xy'
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2552 if zoom_perc is None or zoom_perc == 100:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2553 filename += ' fullres.dat'
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2554 else:
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2555 filename += f' {zoom_perc}p.dat'
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2556 quickPlot(tomosum, title='recon combined sum xy',
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2557 path=self.output_folder, save_fig=self.save_plots,
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2558 save_only=self.save_plots_only)
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2559 np.savetxt(f'{self.output_folder}/recon_combined.txt',
75dd6e15f628 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 47b7bc203de6eac4d3e081b1fae8acb352f54b55" rv43 parents: 49 diff changeset	2560 tomo_recon_combined[int(tomosum.size/2),:,:], fmt='%.6e')
48 059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2561 else:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2562 base_name = 'recon combined'
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2563 for stack in stacks:
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2564 base_name += f' {stack["index"]}'
059819ea1f0e "planemo upload for repository https://github.com/rolfverberg/galaxytools commit b97c6a8f181dea6d2f9cfa2069b86d30dcc47b4d" rv43 parents: 47 diff changeset	2565 self._saveTomo(base_name, tomo_recon_combined)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2566
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2567 # Update config and save to file
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2568 if combine_stacks:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2569 combine_stacks['x_bounds'] = x_bounds
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2570 combine_stacks['y_bounds'] = y_bounds
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2571 combine_stacks['z_bounds'] = z_bounds
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2572 combine_stacks['stacks'] = combined_stacks
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2573 else:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2574 self.config['combine_stacks'] = {'x_bounds' : x_bounds, 'y_bounds' : y_bounds,
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2575 'z_bounds' : z_bounds, 'stacks' : combined_stacks}
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2576 self.cf.saveFile(self.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2577
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2578 def runTomo(config_file=None, config_dict=None, output_folder='.', log_level='INFO',
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2579 test_mode=False, num_core=-1):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2580 """Run a tomography analysis.
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2581 """
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2582 # Instantiate Tomo object
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2583 tomo = Tomo(config_file=config_file, output_folder=output_folder, log_level=log_level,
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2584 test_mode=test_mode, num_core=num_core)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2585 if not tomo.is_valid:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2586 raise ValueError('Invalid config and/or detector file provided.')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2587
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2588 # Preprocess the image files
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2589 assert(tomo.config['stack_info'])
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2590 num_tomo_stacks = tomo.config['stack_info']['num']
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2591 assert(num_tomo_stacks == len(tomo.tomo_stacks))
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2592 preprocessed_stacks = []
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2593 if not tomo.test_mode:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2594 preprocess = tomo.config.get('preprocess', None)
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2595 if preprocess:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2596 preprocessed_stacks = [stack['index'] for stack in tomo.config['stack_info']['stacks']
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2597 if stack.get('preprocessed', False)]
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2598 if len(preprocessed_stacks) != num_tomo_stacks:
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2599 tomo.genTomoStacks()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2600 if not tomo.is_valid:
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2601 IOError('Unable to load all required image files.')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2602 tomo.cf.saveFile(tomo.config_out)
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2603
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2604 # Find centers
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2605 find_center = tomo.config.get('find_center')
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2606 if find_center is None or not find_center.get('completed', False):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2607 tomo.findCenters()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2608
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2609 # Check centers
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2610 #if num_tomo_stacks > 1 and not tomo.config.get('check_centers', False):
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2611 # tomo.checkCenters()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2612
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2613 # Reconstruct tomography stacks
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2614 assert(tomo.config['stack_info']['stacks'])
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2615 reconstructed_stacks = [stack['index'] for stack in tomo.config['stack_info']['stacks']
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2616 if stack.get('reconstructed', False)]
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2617 if len(reconstructed_stacks) != num_tomo_stacks:
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2618 tomo.reconstructTomoStacks()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2619
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2620 # Combine reconstructed tomography stacks
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2621 reconstructed_stacks = [stack['index'] for stack in tomo.config['stack_info']['stacks']
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2622 if stack.get('reconstructed', False)]
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2623 combine_stacks = tomo.config.get('combine_stacks')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2624 if len(reconstructed_stacks) and (combine_stacks is None or
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2625 combine_stacks.get('stacks') != reconstructed_stacks):
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2626 tomo.combineTomoStacks()
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2627
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2628 #%%============================================================================
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2629 if __name__ == '__main__':
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2630
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2631 # Parse command line arguments
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2632 parser = argparse.ArgumentParser(
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2633 description='Tomography reconstruction')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2634 parser.add_argument('-c', '--config',
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2635 default=None,
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2636 help='Input config')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2637 parser.add_argument('-o', '--output_folder',
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2638 default='.',
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2639 help='Output folder')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2640 parser.add_argument('-l', '--log_level',
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2641 default='INFO',
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2642 help='Log level')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2643 parser.add_argument('-t', '--test_mode',
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2644 action='store_true',
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2645 default=False,
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2646 help='Test mode flag')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2647 parser.add_argument('--num_core',
3 f9c52762c32c "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6" rv43 parents: 2 diff changeset	2648 type=int,
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2649 default=-1,
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2650 help='Number of cores')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2651 args = parser.parse_args()
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2652
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2653 if args.config is None:
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2654 if os.path.isfile('config.yaml'):
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2655 args.config = 'config.yaml'
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2656 else:
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2657 args.config = 'config.txt'
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2658
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2659 # Set basic log configuration
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2660 logging_format = '%(asctime)s : %(levelname)s - %(module)s : %(funcName)s - %(message)s'
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2661 if not args.test_mode:
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2662 level = getattr(logging, args.log_level.upper(), None)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2663 if not isinstance(level, int):
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2664 raise ValueError(f'Invalid log_level: {args.log_level}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2665 logging.basicConfig(format=logging_format, level=level, force=True,
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2666 handlers=[logging.StreamHandler()])
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2667
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2668 logging.debug(f'config = {args.config}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2669 logging.debug(f'output_folder = {args.output_folder}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2670 logging.debug(f'log_level = {args.log_level}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2671 logging.debug(f'test_mode = {args.test_mode}')
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2672 logging.debug(f'num_core = {args.num_core}')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2673
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2674 # Run tomography analysis
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2675 runTomo(config_file=args.config, output_folder=args.output_folder, log_level=args.log_level,
e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2676 test_mode=args.test_mode, num_core=args.num_core)
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2677
cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2678 #%%============================================================================
1 e4778148df6b "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7bc44bd6a5a7d503fc4d6f6fb67b3a04a631430b" rv43 parents: 0 diff changeset	2679 # input('Press any key to continue')
0 cb1b0d757704 "planemo upload for repository https://github.com/rolfverberg/galaxytools commit 2da52c7db6def807073a1d437a00e0e2a8e7e72e" rv43 parents: diff changeset	2680 #%%============================================================================

Mercurial > repos > rv43 > tomo

annotate tomo.py @ 63:98a83f03d91b draft