From 60592749b96d9185272b6d50cb70a95afebd821d Mon Sep 17 00:00:00 2001
From: Thomas Kluyver <thomas.kluyver@xfel.eu>
Date: Fri, 15 Sep 2023 09:15:27 +0200
Subject: [PATCH] Propagate train timestamps to REMI output files

---
 notebooks/REMI/REMI_Digitize_and_Transform.ipynb | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
index f1653cf4e..1eaaaeb28 100644
--- a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
+++ b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
@@ -87,7 +87,7 @@
     "\n",
     "import pasha as psh\n",
     "from euxfel_bunch_pattern import indices_at_sase, indices_at_laser\n",
-    "from extra_data import RunDirectory\n",
+    "from extra_data import RunDirectory, by_id\n",
     "from extra_remi import Analysis, trigger_dt\n",
     "from extra_remi.util import timing\n",
     "from extra_remi.rd_resort import signal_dt, hit_dt\n",
@@ -1152,7 +1152,10 @@
     "    with DataFile.from_details(out_folder, out_aggregator, run, seq_id) as outp:\n",
     "        outp.create_metadata(like=dc, proposal=proposal, run=run, sequence=seq_id,\n",
     "                             control_sources=control_sources, instrument_channels=instrument_channels)\n",
-    "        outp.create_index(seq_train_ids)\n",
+    "        outp.create_index(\n",
+    "            seq_train_ids, \n",
+    "            timestamps=dc.select_trains(by_id[seq_train_ids]).train_timestamps().astype(np.uint64)\n",
+    "        )\n",
     "        \n",
     "        for det_name in remi['detector']:\n",
     "            cur_device_id = det_device_id.format(karabo_id=karabo_id, det_name=det_name.upper())\n",
-- 
GitLab