From d51f03bc6bc45a04f5514d94d43c44cd12bb540c Mon Sep 17 00:00:00 2001
From: Gamelin Alexis <gamelin@synchrotron-soleil.fr>
Date: Wed, 25 Mar 2020 14:47:30 +0100
Subject: [PATCH] Store alive attribute for PhaseSpaceMonitor class

---
 tracking/monitors.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tracking/monitors.py b/tracking/monitors.py
index 078c58a..6dcc526 100644
--- a/tracking/monitors.py
+++ b/tracking/monitors.py
@@ -311,8 +311,10 @@ class PhaseSpaceMonitor(Monitor):
         self.bunch_number = bunch_number
         self.mp_number = int(mp_number)
         group_name = "PhaseSpaceData_" + str(self.bunch_number)
-        dict_buffer = {"particles":(self.mp_number, 6, buffer_size)}
-        dict_file = {"particles":(self.mp_number, 6, total_size)}
+        dict_buffer = {"particles":(self.mp_number, 6, buffer_size), 
+                       "alive":(self.mp_number, buffer_size)}
+        dict_file = {"particles":(self.mp_number, 6, total_size),
+                     "alive":(self.mp_number, total_size)}
         self.monitor_init(group_name, save_every, buffer_size, total_size,
                           dict_buffer, dict_file, file_name, mpi_mode)
         
-- 
GitLab