src/ui/gui/autorecode-dialog.c

   1 /* PSPPIRE - a graphical user interface for PSPP.
   2    Copyright (C) 2011  Free Software Foundation
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation, either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
  16
  17 #include <config.h>
  18
  19 #include "dialog-common.h"
  20 #include "autorecode-dialog.h"
  21 #include "psppire-selector.h"
  22 #include "psppire-dictview.h"
  23 #include "psppire-var-ptr.h"
  24 #include "psppire-dialog.h"
  25
  26 #include "psppire-data-window.h"
  27 #include "psppire-var-view.h"
  28
  29 #include "executor.h"
  30 #include "helper.h"
  31
  32 #include <gtk/gtk.h>
  33
  34 #include "gettext.h"
  35 #define _(msgid) gettext (msgid)
  36 #define N_(msgid) msgid
  37
  38 /* Name-Label pair */
  39 struct nlp
  40 {
  41   char *name;
  42   char *label;
  43 };
  44
  45 struct autorecode
  46 {
  47   PsppireDict *dict;
  48   GtkWidget *change_button;
  49   GtkWidget *new_name_entry;
  50   GtkWidget *var_view;
  51
  52   /* A hash table of struct nlp's indexed by variable */
  53   GHashTable *varmap;
  54
  55   GtkWidget *ascending;
  56   GtkWidget *group;
  57   GtkWidget *blank;
  58 };
  59
  60 static struct nlp *
  61 nlp_create (const char *name, const char *label)
  62 {
  63   struct nlp *nlp = xmalloc (sizeof *nlp);
  64
  65   nlp->name = g_strdup (name);
  66
  67   nlp->label = NULL;
  68
  69   if ( label != NULL && 0 != strcmp ("", label))
  70     nlp->label = g_strdup (label);
  71
  72   return nlp;
  73 }
  74
  75 static void
  76 nlp_destroy (gpointer data)
  77 {
  78   struct nlp *nlp = data ;
  79   if ( ! nlp )
  80     return;
  81
  82   g_free (nlp->name);
  83   g_free (nlp->label);
  84   g_free (nlp);
  85 }
  86
  87 static void
  88 on_change_clicked (GObject *obj, gpointer data)
  89 {
  90   struct autorecode *rd = data;
  91   struct variable *var = NULL;
  92   struct nlp *nlp;
  93   GtkTreeModel *model = GTK_TREE_MODEL (PSPPIRE_VAR_VIEW (rd->var_view)->list);
  94   GtkTreeIter iter;
  95   GtkTreeSelection *selection =
  96     gtk_tree_view_get_selection (GTK_TREE_VIEW (rd->var_view));
  97
  98   GList *rows = gtk_tree_selection_get_selected_rows (selection, &model);
  99
 100   const gchar *dest_var_name =
 101     gtk_entry_get_text (GTK_ENTRY (rd->new_name_entry));
 102
 103   if ( NULL == rows || rows->next != NULL)
 104     goto finish;
 105
 106   gtk_tree_model_get_iter (model, &iter, rows->data);
 107
 108   gtk_tree_model_get (model, &iter, 0, &var, -1);
 109
 110   g_hash_table_remove (rd->varmap, var);
 111
 112   nlp = nlp_create (dest_var_name, NULL);
 113
 114   g_hash_table_insert (rd->varmap, var, nlp);
 115
 116   gtk_tree_model_row_changed (model, rows->data, &iter);
 117
 118  finish:
 119   g_list_foreach (rows, (GFunc) gtk_tree_path_free, NULL);
 120   g_list_free (rows);
 121 }
 122
 123
 124 static char * generate_syntax (const struct autorecode *rd);
 125
 126
 127 static void
 128 refresh (PsppireDialog *dialog, struct autorecode *rd)
 129 {
 130   GtkTreeModel *target_list = gtk_tree_view_get_model (GTK_TREE_VIEW (rd->var_view));
 131
 132   gtk_entry_set_text (GTK_ENTRY (rd->new_name_entry), "");
 133   gtk_widget_set_sensitive  (rd->new_name_entry, FALSE);
 134   gtk_widget_set_sensitive  (rd->change_button, FALSE);
 135
 136   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (rd->ascending), TRUE);
 137   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (rd->group), FALSE);
 138   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (rd->blank), FALSE);
 139
 140   if (rd->varmap )
 141     g_hash_table_remove_all (rd->varmap);
 142
 143   gtk_list_store_clear (GTK_LIST_STORE (target_list));
 144 }
 145
 146
 147 static gboolean
 148 dialog_state_valid (gpointer data)
 149 {
 150   struct autorecode *rd = data;
 151
 152   GtkTreeModel *model = gtk_tree_view_get_model (GTK_TREE_VIEW (rd->var_view));
 153
 154   if (g_hash_table_size (rd->varmap) != gtk_tree_model_iter_n_children (model, NULL) )
 155     return FALSE;
 156
 157  return TRUE;
 158 }
 159
 160
 161 static void
 162 on_entry_change (struct autorecode *rd)
 163 {
 164   gboolean valid = TRUE;
 165   const char *text = gtk_entry_get_text (GTK_ENTRY (rd->new_name_entry));
 166
 167   if ( 0 == strcmp ("", text))
 168     valid = FALSE;
 169   else if (psppire_dict_lookup_var (rd->dict, text))
 170     valid = FALSE;
 171   else
 172     {
 173       GHashTableIter iter;
 174       gpointer key, value;
 175
 176       g_hash_table_iter_init (&iter, rd->varmap);
 177       while (g_hash_table_iter_next (&iter, &key, &value))
 178         {
 179           struct nlp *nlp = value;
 180
 181           if ( 0 == strcmp (nlp->name, text))
 182             {
 183               valid = FALSE;
 184               break;
 185             }
 186
 187         }
 188     }
 189
 190   gtk_widget_set_sensitive  (rd->change_button, valid);
 191 }
 192
 193 /* Callback which gets called when a new row is selected
 194    in the variable treeview.
 195    It sets the name and label entry widgets to reflect the
 196    currently selected row.
 197  */
 198 static void
 199 on_selection_change (GtkTreeSelection *selection, gpointer data)
 200 {
 201   struct autorecode *rd = data;
 202   GtkTreeModel *model = GTK_TREE_MODEL (PSPPIRE_VAR_VIEW (rd->var_view)->list);
 203   GList *rows = gtk_tree_selection_get_selected_rows (selection, &model);
 204
 205   if ( rows && !rows->next)
 206     {
 207       /* Exactly one row is selected */
 208       struct nlp *nlp;
 209       struct variable *var;
 210       gboolean ok;
 211       GtkTreeIter iter;
 212
 213       gtk_widget_set_sensitive  (rd->new_name_entry, TRUE);
 214       gtk_widget_set_sensitive  (rd->change_button, TRUE);
 215
 216
 217       ok = gtk_tree_model_get_iter (model, &iter, (GtkTreePath*) rows->data);
 218
 219       gtk_tree_model_get (model, &iter, 0, &var, -1);
 220
 221       nlp = g_hash_table_lookup (rd->varmap, var);
 222
 223       if (nlp)
 224         gtk_entry_set_text (GTK_ENTRY (rd->new_name_entry), nlp->name ? nlp->name : "");
 225       else
 226         gtk_entry_set_text (GTK_ENTRY (rd->new_name_entry), "");
 227     }
 228   else
 229     {
 230       gtk_entry_set_text (GTK_ENTRY (rd->new_name_entry), "");
 231       gtk_widget_set_sensitive  (rd->new_name_entry, FALSE);
 232       gtk_widget_set_sensitive  (rd->change_button, FALSE);
 233     }
 234
 235   g_list_foreach (rows, (GFunc) gtk_tree_path_free, NULL);
 236   g_list_free (rows);
 237 }
 238
 239 static void
 240 render_new_var_name (GtkTreeViewColumn *tree_column,
 241                      GtkCellRenderer *cell,
 242                      GtkTreeModel *tree_model,
 243                      GtkTreeIter *iter,
 244                      gpointer data)
 245 {
 246   struct nlp *nlp = NULL;
 247   struct autorecode *rd = data;
 248
 249   struct variable *var = NULL;
 250
 251   gtk_tree_model_get (tree_model, iter,
 252                       0, &var,
 253                       -1);
 254
 255   nlp = g_hash_table_lookup (rd->varmap, var);
 256
 257   if ( nlp )
 258     g_object_set (cell, "text", nlp->name, NULL);
 259   else
 260     g_object_set (cell, "text", "", NULL);
 261 }
 262
 263
 264
 265 /* Pops up the Autorecode dialog box */
 266 void
 267 autorecode_dialog (PsppireDataWindow *de)
 268 {
 269   struct autorecode rd;
 270   gint response;
 271
 272   GtkBuilder *xml = builder_new ("autorecode.ui");
 273   PsppireVarStore *vs;
 274
 275   GtkWidget *dialog = get_widget_assert   (xml, "autorecode-dialog");
 276   GtkWidget *source = get_widget_assert   (xml, "dict-view");
 277
 278   rd.var_view = get_widget_assert   (xml, "var-view");
 279
 280   rd.new_name_entry = get_widget_assert (xml, "entry1");
 281   rd.change_button = get_widget_assert (xml, "button1");
 282   rd.ascending = get_widget_assert (xml, "radiobutton1");
 283   rd.group = get_widget_assert (xml, "checkbutton1");
 284   rd.blank = get_widget_assert (xml, "checkbutton2");
 285
 286     {
 287       GtkTreeSelection *sel;
 288
 289       GtkCellRenderer *renderer = gtk_cell_renderer_text_new ();
 290
 291       GtkTreeViewColumn *col = gtk_tree_view_column_new_with_attributes (_("New"),
 292                                                                          renderer,
 293                                                                          "text", NULL,
 294                                                                          NULL);
 295
 296       gtk_tree_view_column_set_cell_data_func (col, renderer,
 297                                                render_new_var_name,
 298                                                &rd, NULL);
 299
 300       gtk_tree_view_append_column (GTK_TREE_VIEW (rd.var_view), col);
 301
 302
 303       col = gtk_tree_view_get_column (GTK_TREE_VIEW (rd.var_view), 0);
 304
 305       g_object_set (col, "title", _("Old"), NULL);
 306
 307       g_object_set (rd.var_view, "headers-visible", TRUE, NULL);
 308
 309       rd.varmap = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, nlp_destroy);
 310
 311
 312       sel = gtk_tree_view_get_selection (GTK_TREE_VIEW (rd.var_view));
 313
 314
 315       g_signal_connect (sel, "changed",
 316                         G_CALLBACK (on_selection_change), &rd);
 317
 318       g_signal_connect (rd.change_button, "clicked",
 319                         G_CALLBACK (on_change_clicked),  &rd);
 320
 321       g_signal_connect_swapped (rd.new_name_entry, "changed",
 322                         G_CALLBACK (on_entry_change),  &rd);
 323
 324     }
 325
 326   g_object_get (de->data_editor, "var-store", &vs, NULL);
 327
 328   gtk_window_set_transient_for (GTK_WINDOW (dialog), GTK_WINDOW (de));
 329
 330   g_object_get (vs, "dictionary", &rd.dict, NULL);
 331   g_object_set (source, "model", rd.dict, NULL);
 332
 333
 334   g_signal_connect (dialog, "refresh", G_CALLBACK (refresh),  &rd);
 335
 336   psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (dialog),
 337                                       dialog_state_valid, &rd);
 338
 339   response = psppire_dialog_run (PSPPIRE_DIALOG (dialog));
 340
 341
 342   switch (response)
 343     {
 344     case GTK_RESPONSE_OK:
 345       g_free (execute_syntax_string (de, generate_syntax (&rd)));
 346       break;
 347     case PSPPIRE_RESPONSE_PASTE:
 348       g_free (paste_syntax_to_window (generate_syntax (&rd)));
 349       break;
 350     default:
 351       break;
 352     }
 353
 354   g_object_unref (xml);
 355 }
 356
 357
 358 \f
 359
 360 static char *
 361 generate_syntax (const struct autorecode *rd)
 362 {
 363   GHashTableIter iter;
 364   gpointer key, value;
 365   gchar *text;
 366   GString *string = g_string_new ("AUTORECODE");
 367
 368   g_string_append (string, "\n\tVARIABLES =");
 369
 370   g_hash_table_iter_init (&iter, rd->varmap);
 371   while (g_hash_table_iter_next (&iter, &key, &value))
 372   {
 373     struct variable *var = key;
 374     g_string_append (string, " ");
 375     g_string_append (string, var_get_name (var));
 376   }
 377
 378   g_string_append (string, " INTO");
 379
 380   g_hash_table_iter_init (&iter, rd->varmap);
 381   while (g_hash_table_iter_next (&iter, &key, &value))
 382   {
 383     struct nlp *nlp  = value;
 384     g_string_append (string, " ");
 385     g_string_append (string, nlp->name);
 386   }
 387
 388   if ( ! gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (rd->ascending)))
 389     g_string_append (string, "\n\t/DESCENDING");
 390
 391   if ( gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (rd->group)))
 392     g_string_append (string, "\n\t/GROUP");
 393
 394   if ( gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (rd->blank)))
 395     g_string_append (string, "\n\t/BLANK");
 396
 397   g_string_append (string, ".\n");
 398
 399   text = string->str;
 400
 401   g_string_free (string, FALSE);
 402
 403   return text;
 404 }