From 1724569f9b32c8d118c66574183a5e7f6565b0c8 Mon Sep 17 00:00:00 2001
From: Remi Lehe <remi.lehe@normalesup.org>
Date: Tue, 11 Jul 2023 11:10:33 -0700
Subject: [PATCH] Store additional beam metadata (#396)

---
 src/particles/elements/diagnostics/openPMD.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/particles/elements/diagnostics/openPMD.cpp b/src/particles/elements/diagnostics/openPMD.cpp
index 3fc28a794..a7b8e4e3c 100644
--- a/src/particles/elements/diagnostics/openPMD.cpp
+++ b/src/particles/elements/diagnostics/openPMD.cpp
@@ -257,6 +257,11 @@ namespace detail
             }
         }
 
+        // beam mass
+        beam.setAttribute( "mass", ref_part.mass );
+        beam.setAttribute( "beta_ref", ref_part.beta() );
+        beam.setAttribute( "gamma_ref", ref_part.gamma() );
+
         // openPMD coarse position
         {
             beam["positionOffset"]["x"].resetDataset(d_fl);