comparison tomo_reconstruct.py @ 3:f9c52762c32c draft

"planemo upload for repository https://github.com/rolfverberg/galaxytools commit 7dce44d576e4149f31bdc2ee4dce0bb6962badb6"
author rv43
date Tue, 05 Apr 2022 18:23:54 +0000
parents
children 1bcca1f2adb4
comparison
equal deleted inserted replaced
2:b8977c98800b 3:f9c52762c32c
1 #!/usr/bin/env python3
2
3 import logging
4
5 import sys
6 import argparse
7
8 from tomo import Tomo
9
10 def __main__():
11
12 # Parse command line arguments
13 parser = argparse.ArgumentParser(
14 description='Perfrom a tomography reconstruction')
15 parser.add_argument('-i', '--input_stacks',
16 help='Preprocessed image file stacks')
17 parser.add_argument('-c', '--config',
18 help='Input config')
19 parser.add_argument('--output_config',
20 help='Output config')
21 parser.add_argument('--output_data',
22 help='Reconstructed tomography data')
23 parser.add_argument('-l', '--log',
24 type=argparse.FileType('w'),
25 default=sys.stdout,
26 help='Log file')
27 args = parser.parse_args()
28
29 # Set basic log configuration
30 logging_format = '%(asctime)s : %(levelname)s - %(module)s : %(funcName)s - %(message)s'
31 log_level = 'INFO'
32 level = getattr(logging, log_level.upper(), None)
33 if not isinstance(level, int):
34 raise ValueError(f'Invalid log_level: {log_level}')
35 logging.basicConfig(format=logging_format, level=level, force=True,
36 handlers=[logging.StreamHandler()])
37
38 logging.debug(f'input_stacks = {args.input_stacks}')
39 logging.debug(f'config = {args.config}')
40 logging.debug(f'output_config = {args.output_config}')
41 logging.debug(f'output_data = {args.output_data}')
42 logging.debug(f'log = {args.log}')
43 logging.debug(f'is log stdout? {args.log is sys.stdout}')
44
45 # Instantiate Tomo object
46 tomo = Tomo(config_file=args.config, config_out=args.output_config, log_level=log_level,
47 log_stream=args.log, galaxy_flag=True)
48 if not tomo.is_valid:
49 raise ValueError('Invalid config file provided.')
50 logging.debug(f'config:\n{tomo.config}')
51
52 # Load preprocessed image files
53 tomo.loadTomoStacks(args.input_stacks)
54
55 # Reconstruct tomography stacks
56 tomo.reconstructTomoStacks(args.output_data)
57
58 if __name__ == "__main__":
59 __main__()
60