From a8fbcf8345150b24d06b72bf39f780580ecd370f Mon Sep 17 00:00:00 2001
From: Ivan Usov <ivan.usov@psi.ch>
Date: Wed, 13 May 2020 15:05:54 +0200
Subject: [PATCH] Use dict for roi selections

---
 pyzebra/app.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/pyzebra/app.py b/pyzebra/app.py
index f02ba4b..08b860a 100644
--- a/pyzebra/app.py
+++ b/pyzebra/app.py
@@ -53,6 +53,7 @@ doc = curdoc()
 doc.title = "pyzebra"
 
 global curent_h5_data, current_index, det_data
+roi_selection = {}
 
 
 def update_image():
@@ -477,6 +478,7 @@ selection_list = TextAreaInput(rows=7)
 
 
 def selection_button_callback():
+    global roi_selection
     selection = [
         int(np.floor(det_x_range.start)),
         int(np.ceil(det_x_range.end)),
@@ -486,10 +488,13 @@ def selection_button_callback():
         int(np.ceil(frame_range.end)),
     ]
 
-    if selection_list.value == "":
-        selection_list.value = f"{selection}"
+    filename_id = filelist.value[-8:-4]
+    if filename_id in roi_selection:
+        roi_selection[f"{filename_id}"].append(selection)
     else:
-        selection_list.value = f"{selection_list.value},\n{selection}"
+        roi_selection[f"{filename_id}"] = [selection]
+
+    selection_list.value = str(roi_selection)
 
 
 selection_button = Button(label="Add selection")