From 55b22d5bf19047578862b3f6d443b086d11d6ee4 Mon Sep 17 00:00:00 2001 From: rasmusvt Date: Thu, 12 May 2022 19:17:43 +0200 Subject: [PATCH] Small improvements to integrate function --- nafuma/xrd/io.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/nafuma/xrd/io.py b/nafuma/xrd/io.py index 495f2d1..6b5a762 100644 --- a/nafuma/xrd/io.py +++ b/nafuma/xrd/io.py @@ -62,6 +62,7 @@ def integrate_1d(data, options={}, index=0): if 'image' not in data.keys() or not data['image']: data['image'] = get_image_array(data['path'][index]) + # Instanciate the azimuthal integrator from pyFAI from the calibrant (.poni-file) ai = pyFAI.load(data['calibrant']) @@ -72,6 +73,9 @@ def integrate_1d(data, options={}, index=0): if not os.path.isdir(options['extract_folder']): os.makedirs(options['extract_folder']) + if not os.path.isdir(options['save_folder']): + os.makedirs(options['save_folder']) + res = ai.integrate1d(data['image'], options['nbins'], unit=options['unit'], filename=filename) @@ -348,7 +352,6 @@ def strip_headers_from_xy(path: str, filename=None) -> None: ext = path.split('.')[-1] filename = path.split(f'.{ext}')[0] + f'_noheaders.{ext}' - print(filename) with open(filename, 'w') as f: for line in xy: