Skip to content

Fix: Updated bg_gui.py with fixes for SHO visualizer #25

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ramav87 merged 1 commit into from
Sep 15, 2025
Merged

Fix: Updated bg_gui.py with fixes for SHO visualizer #25

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 23 additions & 21 deletions BGlib/misc/bg_gui.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -199,37 +199,37 @@ def create_sho_tab(self):
self.do_fit_button.setEnabled(False)

# ===== Right side: visualization + output =====
right_layout = QVBoxLayout()

# After the SHO fit is complete, create and add the visualizer widget
from sho_visualizer_widget import SHOVisualizerWidget

def show_visualizer(self, raw_data, fit_data, freq_vec):
"""
Embed the visualizer directly into the right side of the Fit and View SHO tab.
"""
if hasattr(self, "sho_visualizer_widget"):
# Remove old visualizer if it exists
self.right_layout.removeWidget(self.sho_visualizer_widget)
self.sho_visualizer_widget.deleteLater()

# Create new visualizer
self.sho_visualizer_widget = SHOVisualizerWidget(raw_data, fit_data, freq_vec)
self.right_layout.addWidget(self.sho_visualizer_widget)

self.right_layout = QVBoxLayout()

# Output box for logs
self.output_box = QTextEdit()
self.output_box.setReadOnly(True)
self.output_box.setPlaceholderText("Output logs will appear here...")

right_layout.addWidget(self.output_box, 1)
self.right_layout.addWidget(self.output_box, 1)

main_layout.addLayout(controls_layout, 1)
main_layout.addLayout(right_layout, 2)
main_layout.addLayout(self.right_layout, 2)

tab.setLayout(main_layout)
return tab

# After the SHO fit is complete, create and add the visualizer widget
def show_visualizer(self, raw_data, fit_data, freq_vec):
"""
Embed the visualizer directly into the right side of the Fit and View SHO tab.
"""
from sho_visualizer_widget import SHOVisualizerWidget
if hasattr(self, "sho_visualizer_widget"):
# Remove old visualizer if it exists
self.right_layout.removeWidget(self.sho_visualizer_widget)
self.sho_visualizer_widget.deleteLater()

# Create new visualizer
self.sho_visualizer_widget = SHOVisualizerWidget(raw_data, fit_data, freq_vec)
self.right_layout.addWidget(self.sho_visualizer_widget)


# ==============================
# Tab 3: Fit Loops
# ==============================
Expand Down Expand Up @@ -292,7 +292,9 @@ def on_do_fit(self):
self.lower_bounds, self.upper_bounds = self.get_fitting_bounds()
self.fitter_output = self.fitter.do_fit()
self.fitter_output[0].data_type = 'spectral_image'
self.show_visualizer(np.array(self.fitter_output[0].compute()), np.array(self.fitter_output[1].compute()), np.array(self.freq_vec))
raw = np.array(self.beps_raw.compute())
fit_params = np.array(self.fitter_output[0].compute())
self.show_visualizer(raw, fit_params, np.array(self.freq_vec))
return
# ==============================
# Main application entry point
Expand Down