From b5effcd32fe85e876bbd69938478dbef53a5c7c4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Le=20Guyader?= <loic.le.guyader@xfel.eu>
Date: Tue, 11 Apr 2023 11:31:04 +0200
Subject: [PATCH] Load data only once in BOZ small data
---
doc/BOZ analysis part II.1 Small data.ipynb | 12 +++---------
1 file changed, 3 insertions(+), 9 deletions(-)
diff --git a/doc/BOZ analysis part II.1 Small data.ipynb b/doc/BOZ analysis part II.1 Small data.ipynb
index b2c1a8f..eac18f0 100644
--- a/doc/BOZ analysis part II.1 Small data.ipynb
+++ b/doc/BOZ analysis part II.1 Small data.ipynb
@@ -125,7 +125,8 @@
"metadata": {},
"outputs": [],
"source": [
- "arr_dark, tid_dark = boz.load_dssc_module(proposal, darkrunNB, moduleNB)"
+ "arr_dark, tid_dark = boz.load_dssc_module(proposal, darkrunNB, moduleNB)",
+ "arr_dark = arr_dark.rechunk(('auto',-1,-1,-1))\n"
]
},
{
@@ -175,6 +176,7 @@
"source": [
"dark = boz.average_module(arr_dark).compute()\n",
"arr, tid = boz.load_dssc_module(proposal, runNB, moduleNB)\n",
+ "arr = arr.rechunk(('auto',-1,-1,-1))\n",
"data = boz.average_module(arr, dark=dark).compute()\n",
"pp = data.mean(axis=(1,2)) # pulseId resolved mean\n",
"dataM = data.mean(axis=0) # mean over pulseId"
@@ -370,14 +372,6 @@
"metadata": {},
"outputs": [],
"source": [
- "# load data\n",
- "arr, tid = boz.load_dssc_module(proposal, runNB, moduleNB)\n",
- "arr_dark, tid_dark = boz.load_dssc_module(proposal, darkrunNB, moduleNB)\n",
- "\n",
- "# make sure to rechunk the arrays\n",
- "arr = arr.rechunk(('auto',-1,-1,-1))\n",
- "arr_dark = arr_dark.rechunk(('auto',-1,-1,-1))\n",
- "\n",
"#no correction\n",
"#data = boz.process(np.arange(2**9), arr_dark, arr, tid, rois, params.get_mask(),\n",
"# np.ones_like(ff), params.sat_level)\n",
--
GitLab