aboutsummaryrefslogtreecommitdiff
path: root/src/gui_peaksearch.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/gui_peaksearch.c')
-rw-r--r--src/gui_peaksearch.c369
1 files changed, 369 insertions, 0 deletions
diff --git a/src/gui_peaksearch.c b/src/gui_peaksearch.c
new file mode 100644
index 00000000..149aacc8
--- /dev/null
+++ b/src/gui_peaksearch.c
@@ -0,0 +1,369 @@
+/*
+ * gui_peaksearch.c
+ *
+ * Peak search parts of GUI
+ *
+ * Copyright © 2020 Deutsches Elektronen-Synchrotron DESY,
+ * a research centre of the Helmholtz Association.
+ *
+ * Authors:
+ * 2020 Thomas White <taw@physics.org>
+ *
+ * This file is part of CrystFEL.
+ *
+ * CrystFEL is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * CrystFEL is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with CrystFEL. If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdarg.h>
+#include <stdio.h>
+#include <getopt.h>
+#include <string.h>
+#include <gtk/gtk.h>
+#include <gdk/gdkkeysyms-compat.h>
+#include <assert.h>
+
+#include <datatemplate.h>
+#include <peaks.h>
+
+#include "crystfel_gui.h"
+#include "crystfelimageview.h"
+
+
+void update_peaks(struct crystfelproject *proj)
+{
+ struct image *image;
+
+ if ( proj->n_frames == 0 ) return;
+
+ if ( proj->show_peaks ) {
+
+ image = crystfel_image_view_get_image_struct(CRYSTFEL_IMAGE_VIEW(proj->imageview));
+ if ( image == NULL ) return;
+
+ switch ( proj->peak_search_params.method ) {
+
+ case PEAK_ZAEF:
+ search_peaks(image,
+ proj->peak_search_params.threshold,
+ proj->peak_search_params.min_sq_gradient,
+ proj->peak_search_params.min_snr,
+ proj->peak_search_params.pk_inn,
+ proj->peak_search_params.pk_mid,
+ proj->peak_search_params.pk_out,
+ 1);
+ break;
+
+ case PEAK_PEAKFINDER8:
+ STATUS("NB peakfinder8 doesn't yet use new API\n");
+ search_peaks_peakfinder8(image, 2048,
+ proj->peak_search_params.threshold,
+ proj->peak_search_params.min_snr,
+ proj->peak_search_params.min_pix_count,
+ proj->peak_search_params.max_pix_count,
+ proj->peak_search_params.local_bg_radius,
+ proj->peak_search_params.min_res,
+ proj->peak_search_params.max_res,
+ 1);
+ break;
+
+ default:
+ ERROR("This peak detection method not implemented!\n");
+ break;
+
+ }
+
+ crystfel_image_view_set_peaks(CRYSTFEL_IMAGE_VIEW(proj->imageview),
+ image->features, 0);
+
+ } else {
+ crystfel_image_view_set_peaks(CRYSTFEL_IMAGE_VIEW(proj->imageview),
+ NULL, 0);
+ }
+}
+
+
+static int get_val(GtkWidget *entry, float *v)
+{
+ const char *text;
+ float val;
+ text = gtk_entry_get_text(GTK_ENTRY(entry));
+ if (sscanf(text, "%f", &val) != 1) {
+ ERROR("Invalid value\n");
+ return 1;
+ }
+ *v = val;
+ return 0;
+}
+
+
+static void peaksearch_threshold_sig(GtkWidget *entry,
+ struct crystfelproject *proj)
+{
+ float val;
+ if ( get_val(entry, &val) ) return;
+ proj->peak_search_params.threshold = val;
+ update_peaks(proj);
+}
+
+
+static void peaksearch_sqgradient_sig(GtkWidget *entry,
+ struct crystfelproject *proj)
+{
+ float val;
+ if ( get_val(entry, &val) ) return;
+ proj->peak_search_params.min_sq_gradient = val;
+ update_peaks(proj);
+}
+
+
+static void peaksearch_snr_sig(GtkWidget *entry,
+ struct crystfelproject *proj)
+{
+ float val;
+ if ( get_val(entry, &val) ) return;
+ proj->peak_search_params.min_snr = val;
+ update_peaks(proj);
+}
+
+
+static void peaksearch_min_pix_count_sig(GtkWidget *entry,
+ struct crystfelproject *proj)
+{
+ float val;
+ if ( get_val(entry, &val) ) return;
+ proj->peak_search_params.min_pix_count = val;
+ update_peaks(proj);
+}
+
+
+static void peaksearch_max_pix_count_sig(GtkWidget *entry,
+ struct crystfelproject *proj)
+{
+ float val;
+ if ( get_val(entry, &val) ) return;
+ proj->peak_search_params.max_pix_count = val;
+ update_peaks(proj);
+}
+
+
+static void peaksearch_local_bg_radius_sig(GtkWidget *entry,
+ struct crystfelproject *proj)
+{
+ float val;
+ if ( get_val(entry, &val) ) return;
+ proj->peak_search_params.local_bg_radius = val;
+ update_peaks(proj);
+}
+
+
+static void peaksearch_min_res_sig(GtkWidget *entry,
+ struct crystfelproject *proj)
+{
+ float val;
+ if ( get_val(entry, &val) ) return;
+ proj->peak_search_params.min_res = val;
+ update_peaks(proj);
+}
+
+
+static void peaksearch_max_res_sig(GtkWidget *entry,
+ struct crystfelproject *proj)
+{
+ float val;
+ if ( get_val(entry, &val) ) return;
+ proj->peak_search_params.max_res = val;
+ update_peaks(proj);
+}
+
+
+static void add_param(GtkWidget *params_box, const char *labeltext,
+ float initial_val, GCallback act_cb,
+ struct crystfelproject *proj)
+{
+ GtkWidget *hbox;
+ GtkWidget *label;
+ GtkWidget *entry;
+ char tmp[64];
+
+ hbox = gtk_hbox_new(FALSE, 0.0);
+ gtk_box_pack_start(GTK_BOX(params_box),
+ GTK_WIDGET(hbox), FALSE, FALSE, 8.0);
+ label = gtk_label_new(labeltext);
+ gtk_misc_set_alignment(GTK_MISC(label), 1.0, 0.5);
+ gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(label),
+ FALSE, FALSE, 2.0);
+ entry = gtk_entry_new();
+ gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(entry),
+ TRUE, TRUE, 2.0);
+ snprintf(tmp, 63, "%.2f", initial_val);
+ gtk_entry_set_text(GTK_ENTRY(entry), tmp);
+ g_signal_connect(G_OBJECT(entry), "activate",
+ G_CALLBACK(act_cb), proj);
+}
+
+
+static void peaksearch_algo_changed(GtkWidget *combo,
+ struct crystfelproject *proj)
+{
+ const char *algo_id;
+
+ if ( proj->peak_params != NULL ) {
+ gtk_container_remove(GTK_CONTAINER(proj->peak_vbox),
+ proj->peak_params);
+ proj->peak_params = NULL;
+ }
+
+ proj->peak_params = gtk_vbox_new(FALSE, 0.0);
+ gtk_box_pack_start(GTK_BOX(proj->peak_vbox),
+ GTK_WIDGET(proj->peak_params),
+ FALSE, FALSE, 8.0);
+
+ algo_id = gtk_combo_box_get_active_id(GTK_COMBO_BOX(combo));
+
+ if ( strcmp(algo_id, "zaef") == 0 ) {
+
+ proj->peak_search_params.method = PEAK_ZAEF;
+
+ add_param(proj->peak_params, "Threshold:",
+ proj->peak_search_params.threshold,
+ G_CALLBACK(peaksearch_threshold_sig), proj);
+ add_param(proj->peak_params, "Minimum squared gradient:",
+ proj->peak_search_params.min_sq_gradient,
+ G_CALLBACK(peaksearch_sqgradient_sig), proj);
+ add_param(proj->peak_params, "Minimum signal/noise ratio:",
+ proj->peak_search_params.min_snr,
+ G_CALLBACK(peaksearch_snr_sig), proj);
+
+ } else if ( strcmp(algo_id, "peakfinder8") == 0 ) {
+
+ proj->peak_search_params.method = PEAK_PEAKFINDER8;
+
+ add_param(proj->peak_params, "Threshold:",
+ proj->peak_search_params.threshold,
+ G_CALLBACK(peaksearch_threshold_sig), proj);
+ add_param(proj->peak_params, "Minimum signal/noise ratio:",
+ proj->peak_search_params.min_snr,
+ G_CALLBACK(peaksearch_snr_sig), proj);
+ add_param(proj->peak_params, "Minimum number of pixels:",
+ proj->peak_search_params.min_pix_count,
+ G_CALLBACK(peaksearch_min_pix_count_sig), proj);
+ add_param(proj->peak_params, "Maximum number of pixels:",
+ proj->peak_search_params.max_pix_count,
+ G_CALLBACK(peaksearch_max_pix_count_sig), proj);
+ add_param(proj->peak_params, "Local background radius:",
+ proj->peak_search_params.local_bg_radius,
+ G_CALLBACK(peaksearch_local_bg_radius_sig), proj);
+ add_param(proj->peak_params, "Minimum resolution (pixels):",
+ proj->peak_search_params.min_res,
+ G_CALLBACK(peaksearch_min_res_sig), proj);
+ add_param(proj->peak_params, "Maximum resolution (pixels):",
+ proj->peak_search_params.max_res,
+ G_CALLBACK(peaksearch_max_res_sig), proj);
+
+ } /* else no parameters! */
+
+ /* FIXME: Radii */
+
+ gtk_widget_show_all(proj->peak_vbox);
+ update_peaks(proj);
+}
+
+
+static void peaksearch_response_sig(GtkWidget *dialog, gint resp,
+ struct crystfelproject *proj)
+{
+
+ if ( (resp == GTK_RESPONSE_DELETE_EVENT)
+ || (resp == GTK_RESPONSE_CANCEL) )
+ {
+ proj->peak_search_params = proj->original_params;
+ update_peaks(proj);
+ }
+
+ gtk_widget_destroy(dialog);
+ proj->peak_vbox = NULL;
+ proj->peak_params = NULL;
+}
+
+
+gint peaksearch_sig(GtkWidget *widget, struct crystfelproject *proj)
+{
+ GtkWidget *dialog;
+ GtkWidget *content_area;
+ GtkWidget *vbox;
+ GtkWidget *hbox;
+ GtkWidget *label;
+ GtkWidget *combo;
+ GtkWidget *w;
+
+ if ( proj->peak_params != NULL ) return FALSE;
+
+ proj->show_peaks = 1;
+
+ w = gtk_ui_manager_get_widget(proj->ui, "/ui/mainwindow/view/peaks");
+ gtk_check_menu_item_set_active(GTK_CHECK_MENU_ITEM(w), 1);
+
+ /* Take a copy of the original parameters, for reverting */
+ proj->original_params = proj->peak_search_params;
+
+ dialog = gtk_dialog_new_with_buttons("Peak search",
+ GTK_WINDOW(proj->window),
+ GTK_DIALOG_DESTROY_WITH_PARENT,
+ "Discard changes", GTK_RESPONSE_CANCEL,
+ "Confirm", GTK_RESPONSE_ACCEPT,
+ NULL);
+
+ g_signal_connect(G_OBJECT(dialog), "response",
+ G_CALLBACK(peaksearch_response_sig), proj);
+
+ vbox = gtk_vbox_new(FALSE, 0.0);
+ content_area = gtk_dialog_get_content_area(GTK_DIALOG(dialog));
+ gtk_container_add(GTK_CONTAINER(content_area), vbox);
+ gtk_container_set_border_width(GTK_CONTAINER(content_area), 8);
+ proj->peak_vbox = vbox;
+
+ hbox = gtk_hbox_new(FALSE, 0.0);
+ gtk_box_pack_start(GTK_BOX(vbox), GTK_WIDGET(hbox), FALSE, FALSE, 8.0);
+ label = gtk_label_new("Peak search algorithm");
+ gtk_misc_set_alignment(GTK_MISC(label), 1.0, 0.5);
+ gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(label), FALSE, FALSE, 2.0);
+ combo = gtk_combo_box_text_new();
+ gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(combo), TRUE, TRUE, 2.0);
+ gtk_combo_box_text_append(GTK_COMBO_BOX_TEXT(combo), "zaef",
+ "Zaefferer gradient search (zaef)");
+ gtk_combo_box_text_append(GTK_COMBO_BOX_TEXT(combo), "cxi",
+ "Get list from CXI file");
+ gtk_combo_box_text_append(GTK_COMBO_BOX_TEXT(combo), "hdf5",
+ "Get list from HDF5 file");
+ gtk_combo_box_text_append(GTK_COMBO_BOX_TEXT(combo), "peakfinder8",
+ "Radial background estimation (peakfinder8)");
+ #ifdef HAVE_FDIP
+ gtk_combo_box_text_append(GTK_COMBO_BOX_TEXT(combo), "peakfinder9",
+ "Local background estimation (peakfinder9)");
+ #endif
+ g_signal_connect(G_OBJECT(combo), "changed",
+ G_CALLBACK(peaksearch_algo_changed), proj);
+ gtk_combo_box_set_active(GTK_COMBO_BOX(combo), 0);
+ proj->type_combo = combo;
+
+ gtk_widget_show_all(dialog);
+
+ return FALSE;
+}
generated by cgit v1.2.3 (git 2.25.1) at 2025-02-02 00:02:09 +0000