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