--- a/tomo_reconstruct.py	Wed Apr 13 15:53:06 2022 +0000
+++ b/tomo_reconstruct.py	Wed Apr 13 16:02:44 2022 +0000
@@ -4,6 +4,7 @@
 
 import sys
 import argparse
+import tracemalloc
 
 from tomo import Tomo
 
@@ -16,6 +17,8 @@
             help='Preprocessed image file stacks')
     parser.add_argument('-c', '--config',
             help='Input config')
+    parser.add_argument('--center_offsets',
+            required=True, nargs=2, type=float, help='Reconstruction center axis offsets')
     parser.add_argument('--output_config',
             help='Output config')
     parser.add_argument('--output_data',
@@ -26,6 +29,9 @@
             help='Log file')
     args = parser.parse_args()
 
+    # Starting memory monitoring
+    tracemalloc.start()
+
     # Set basic log configuration
     logging_format = '%(asctime)s : %(levelname)s - %(module)s : %(funcName)s - %(message)s'
     log_level = 'INFO'
@@ -37,6 +43,7 @@
 
     logging.debug(f'input_stacks = {args.input_stacks}')
     logging.debug(f'config = {args.config}')
+    logging.debug(f'center_offsets = {args.center_offsets} {type(args.center_offsets)}')
     logging.debug(f'output_config = {args.output_config}')
     logging.debug(f'output_data = {args.output_data}')
     logging.debug(f'log = {args.log}')
@@ -53,7 +60,14 @@
     tomo.loadTomoStacks(args.input_stacks)
 
     # Reconstruct tomography stacks
-    tomo.reconstructTomoStacks(args.output_data)
+    galaxy_param = {'center_offsets' : args.center_offsets, 'output_name' : args.output_data}
+    tomo.reconstructTomoStacks(galaxy_param)
+
+    # Displaying memory usage
+    logging.info(f'Memory usage: {tracemalloc.get_traced_memory()}')
+
+    # stopping memory monitoring
+    tracemalloc.stop()
 
 if __name__ == "__main__":
     __main__()