From e87dd0a1432e9edf32ff83d239d8a6a450f7b984 Mon Sep 17 00:00:00 2001 From: Yuanpeng Zhang Date: Fri, 14 Oct 2022 17:28:58 -0400 Subject: [PATCH] debug gr tree --- addie/calculate_gr/event_handler.py | 1 + addie/calculate_gr/gofrtree.py | 4 +++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/addie/calculate_gr/event_handler.py b/addie/calculate_gr/event_handler.py index 188ed299..97d78d4f 100644 --- a/addie/calculate_gr/event_handler.py +++ b/addie/calculate_gr/event_handler.py @@ -169,6 +169,7 @@ 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 index 4dc040ae..a86b4994 100644 --- 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):