Merge pull request #387 from neutrons/debug_gr_tree

debug gr tree
neutrons · Oct 22, 2022 · f2ae93d · f2ae93d · github-actions · Oct 22, 2022
2 parents 5af9399 + 4032f7f
commit f2ae93d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/addie/calculate_gr/event_handler.py b/addie/calculate_gr/event_handler.py
@@ -169,6 +169,8 @@ def generate_gr_step1(main_window):
     generate_gr_step2(main_window,
                       sq_ws_name_list=sq_ws_name_list)
 
+    gr_widgets_status(main_window, True)
+
 
 def generate_gr_step2(main_window, sq_ws_name_list):
     """Generate G(r) from specified S(Q) workspaces

diff --git a/addie/calculate_gr/gofrtree.py b/addie/calculate_gr/gofrtree.py
@@ -80,6 +80,7 @@ def pop_up_menu(self):
                 print('[Error] Nodes of different levels are selected.')
 
         if leaf_level == 1:
+            self.removeAction(self._action_remove_plot)
             self.addAction(self._action_plot)
             self.addAction(self._action_ipython)
             self.addAction(self._action_delete)
@@ -205,7 +206,8 @@ def is_gr_empty(self):
         gr_exists = False
         for key in self._leafDict.keys():
             if key.startswith('G(r)'):
-                gr_exists = True
+                if len(self._leafDict[key]) > 0:
+                    gr_exists = True
         return not gr_exists
 
     def is_sofq_empty(self):