/*
 * 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:
			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 peaksearch_half_pixel_sig(GtkWidget *checkbox,
                                      struct crystfelproject *proj)
{
	int val = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(checkbox));
	proj->peak_search_params.half_pixel_shift = val;
	update_peaks(proj);
}


static void add_check_param(GtkWidget *params_box, const char *labeltext,
                            int initial_val, GCallback act_cb,
                            struct crystfelproject *proj)
{
	GtkWidget *checkbox;

	checkbox = gtk_check_button_new_with_label(labeltext);
	gtk_box_pack_start(GTK_BOX(params_box),
	                   GTK_WIDGET(checkbox), FALSE, FALSE, 8.0);
	gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(checkbox),
	                             initial_val);
	g_signal_connect(G_OBJECT(checkbox), "toggled",
	                 G_CALLBACK(act_cb), 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 if ( strcmp(algo_id, "hdf5") == 0 ) {

		proj->peak_search_params.method = PEAK_HDF5;

		add_check_param(proj->peak_params, "Half pixel shift",
		          proj->peak_search_params.half_pixel_shift,
		          G_CALLBACK(peaksearch_half_pixel_sig), proj);

	} else if ( strcmp(algo_id, "cxi") == 0 ) {

		proj->peak_search_params.method = PEAK_CXI;

		add_check_param(proj->peak_params, "Half pixel shift",
		          proj->peak_search_params.half_pixel_shift,
		          G_CALLBACK(peaksearch_half_pixel_sig), proj);

	} else {
		ERROR("Unrecognised peak search '%s'\n", algo_id);
	}

	/* 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_id(GTK_COMBO_BOX(combo),
	                            str_peaksearch(proj->peak_search_params.method));
	proj->type_combo = combo;

	gtk_widget_show_all(dialog);

	return FALSE;
}