Forgot to also add to batch cells

FABLE-3DXRD · Mar 6, 2024 · ce7fcdb · ce7fcdb
1 parent bb55dac
commit ce7fcdb
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/ImageD11/nbGui/S3DXRD/2_S3DXRD_sinograms_map.ipynb b/ImageD11/nbGui/S3DXRD/2_S3DXRD_sinograms_map.ipynb
@@ -1293,6 +1293,16 @@
     "            if \"slice_recon\" in hin.keys():\n",
     "                print(f\"Already reconstructed {dataset} in {sample}, skipping\")\n",
     "                continue\n",
+    "        \n",
+    "        if not hasattr(ds, \"ring_currents\"):\n",
+    "            ring_currents = []\n",
+    "            with h5py.File(ds.masterfile, \"r\") as h5in:\n",
+    "                for scan in ds.scans:\n",
+    "                    ring_current = float(h5in[scan][\"instrument/machine/current\"][()])\n",
+    "                    ring_currents.append(ring_current)\n",
+    "\n",
+    "            ds.ring_currents = np.array(ring_currents)\n",
+    "            ds.ring_currents_scaled = np.array(ring_currents/np.max(ring_currents))\n",
     "            \n",
     "        cf_4d = ImageD11.columnfile.columnfile(ds.col4dfile)\n",
     "        cf_4d.parameters.loadparameters(par_path)\n",

diff --git a/ImageD11/nbGui/S3DXRD/2_S3DXRD_sinograms_map_minor_phase.ipynb b/ImageD11/nbGui/S3DXRD/2_S3DXRD_sinograms_map_minor_phase.ipynb
@@ -1195,6 +1195,17 @@
     "            if \"slice_recon\" in hin.keys():\n",
     "                print(f\"Already reconstructed {dataset} in {sample}, skipping\")\n",
     "                continue\n",
+    "        \n",
+    "        \n",
+    "        if not hasattr(ds, \"ring_currents\"):\n",
+    "            ring_currents = []\n",
+    "            with h5py.File(ds.masterfile, \"r\") as h5in:\n",
+    "                for scan in ds.scans:\n",
+    "                    ring_current = float(h5in[scan][\"instrument/machine/current\"][()])\n",
+    "                    ring_currents.append(ring_current)\n",
+    "\n",
+    "            ds.ring_currents = np.array(ring_currents)\n",
+    "            ds.ring_currents_scaled = np.array(ring_currents/np.max(ring_currents))\n",
     "            \n",
     "        cf_4d = ImageD11.columnfile.columnfile(ds.col4dfile)\n",
     "        cf_4d.parameters.loadparameters(par_path)\n",