diff options
Diffstat (limited to 'src/gui_align.c')
-rw-r--r-- | src/gui_align.c | 235 |
1 files changed, 235 insertions, 0 deletions
diff --git a/src/gui_align.c b/src/gui_align.c new file mode 100644 index 00000000..b712681b --- /dev/null +++ b/src/gui_align.c @@ -0,0 +1,235 @@ +/* + * gui_align.c + * + * Align detector via CrystFEL GUI + * + * Copyright © 2024 Deutsches Elektronen-Synchrotron DESY, + * a research centre of the Helmholtz Association. + * + * Authors: + * 2024 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 <string.h> +#include <gtk/gtk.h> +#include <assert.h> + +#include "version.h" +#include "gui_project.h" +#include "crystfel_gui.h" +#include "gtk-util-routines.h" + + +struct align_window +{ + struct crystfelproject *proj; + GtkWidget *window; + GtkWidget *input_combo; + GtkWidget *out_of_plane; + GtkWidget *level; +}; + + + +static int run_align(const char *input_name, int level, int out_of_plane, + const char *out_geom, struct crystfelproject *proj) +{ + GSubprocess *sp; + struct gui_indexing_result *res; + GError *error; + const char *cmdline[16]; + char level_str[64]; + GFile *gstream; + GFile *gworkdir; + GFile *ggeom; + GFile *gmilledir; + char *input_geom; + int i; + char *mille_dir; + + res = find_indexing_result_by_name(proj, input_name); + if ( res == NULL ) { + ERROR("Results for '%s' not found!\n", input_name); + return 1; + } + + /* Figure out working directory for indexing job */ + if ( res->n_streams < 1 ) { + ERROR("No streams?\n"); + return 1; + } + gstream = g_file_new_for_path(res->streams[0]); + gworkdir = g_file_get_parent(gstream); + g_object_unref(gstream); + + /* Look for Millepede files */ + gmilledir = g_file_get_child(gworkdir, "mille-data"); + if ( !g_file_query_exists(gmilledir, NULL) ) { + ERROR("No detector alignment data found for indexing run '%s'\n", input_name); + return 1; + } + mille_dir = g_file_get_path(gmilledir); + g_object_unref(gmilledir); + + /* Input geometry file */ + ggeom = g_file_get_child(gworkdir, "detector.geom"); + input_geom = g_file_get_path(ggeom); + g_object_unref(ggeom); + + /* Build command line */ + snprintf(level_str, 63, "%i", level); + cmdline[0] = "align_detector"; + cmdline[1] = "-g"; + cmdline[2] = input_geom; + cmdline[3] = "--level"; + cmdline[4] = level_str; + cmdline[5] = "-o"; + cmdline[6] = out_geom; + if ( out_of_plane ) { + cmdline[7] = "--out-of-plane"; + cmdline[8] = mille_dir; + cmdline[9] = NULL; + } else { + cmdline[7] = mille_dir; + cmdline[8] = NULL; + } + + STATUS("Running program: "); + i = 0; + while ( cmdline[i] != NULL ) { + STATUS("%s ", cmdline[i++]); + } + STATUS("\n"); + + error = NULL; + sp = g_subprocess_newv(cmdline, G_SUBPROCESS_FLAGS_NONE, &error); + if ( sp == NULL ) { + ERROR("Failed to run align_detector: %s\n", error->message); + g_error_free(error); + return 1; + } + + g_object_unref(gworkdir); + g_free(mille_dir); + g_free(input_geom); + + return 0; +} + + +static void align_response_sig(GtkWidget *dialog, gint resp, + struct align_window *win) +{ + int r = 0; + + if ( resp == GTK_RESPONSE_ACCEPT ) { + + int level; + const char *input_name; + int out_of_plane; + gchar *filename; + + level = gtk_spin_button_get_value_as_int(GTK_SPIN_BUTTON(win->level)); + + input_name = get_combo_id(win->input_combo); + if ( input_name == NULL ) { + ERROR("Please select the input\n"); + r = 1; + } + + out_of_plane = get_bool(win->out_of_plane); + filename = gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(dialog)); + + r = run_align(input_name, level, out_of_plane, filename, win->proj); + + g_free(filename); + } + + if ( !r ) gtk_widget_destroy(dialog); +} + + +gint align_sig(GtkWidget *widget, struct crystfelproject *proj) +{ + GtkWidget *hbox; + GtkWidget *label; + struct align_window *win; + int i; + + win = malloc(sizeof(struct align_window)); + if ( win == NULL ) return 0; + + win->proj = proj; + + win->window = gtk_file_chooser_dialog_new("Align detector", + GTK_WINDOW(proj->window), + GTK_FILE_CHOOSER_ACTION_SAVE, + GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL, + GTK_STOCK_SAVE, GTK_RESPONSE_ACCEPT, + NULL); + gtk_file_chooser_set_do_overwrite_confirmation(GTK_FILE_CHOOSER(win->window), + TRUE); + + hbox = gtk_hbox_new(FALSE, 0.0); + gtk_file_chooser_set_extra_widget(GTK_FILE_CHOOSER(win->window), + GTK_WIDGET(hbox)); + gtk_container_set_border_width(GTK_CONTAINER(hbox), 4); + + label = gtk_label_new("Refine using indexing result:"); + gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(label), + FALSE, FALSE, 4.0); + win->input_combo = gtk_combo_box_text_new(); + gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(win->input_combo), + FALSE, FALSE, 4.0); + for ( i=0; i<proj->n_results; i++ ) { + gtk_combo_box_text_append(GTK_COMBO_BOX_TEXT(win->input_combo), + proj->results[i].name, + proj->results[i].name); + } + gtk_combo_box_set_active_id(GTK_COMBO_BOX(win->input_combo), + selected_result(proj)); + + label = gtk_label_new("Hierarchy level:"); + gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(label), + FALSE, FALSE, 16.0); + win->level = gtk_spin_button_new_with_range(0.0, 9.0, 1.0); + gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(win->level), + FALSE, FALSE, 4.0); + gtk_widget_set_tooltip_text(win->level, "--level"); + + win->out_of_plane = gtk_check_button_new_with_label("Include out-of-plane positions and tilts"); + gtk_box_pack_start(GTK_BOX(hbox), GTK_WIDGET(win->out_of_plane), + FALSE, FALSE, 4.0); + gtk_widget_set_tooltip_text(win->out_of_plane, "--out-of-plane"); + + g_signal_connect(G_OBJECT(win->window), "response", + G_CALLBACK(align_response_sig), win); + + gtk_dialog_set_default_response(GTK_DIALOG(win->window), + GTK_RESPONSE_CLOSE); + gtk_widget_show_all(win->window); + + return FALSE; +} |