7b027d9badae5380e8505a007e188ede868318bd
[pspp] / src / ui / gui / autorecode-dialog.c
1 /* PSPPIRE - a graphical user interface for PSPP.
2    Copyright (C) 2011, 2012  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 "builder-wrapper.h"
31 #include "helper.h"
32
33 #include <gtk/gtk.h>
34
35 #include "gettext.h"
36 #define _(msgid) gettext (msgid)
37 #define N_(msgid) msgid
38
39 /* Name-Label pair */
40 struct nlp
41 {
42   char *name;
43   char *label;
44 };
45
46 struct autorecode
47 {
48   PsppireDict *dict;
49   GtkWidget *change_button;
50   GtkWidget *new_name_entry;
51   GtkWidget *var_view;
52
53   /* A hash table of struct nlp's indexed by variable */
54   GHashTable *varmap;
55
56   GtkWidget *ascending;
57   GtkWidget *group;
58   GtkWidget *blank;
59 };
60
61 static struct nlp *
62 nlp_create (const char *name, const char *label)
63 {
64   struct nlp *nlp = xmalloc (sizeof *nlp);
65
66   nlp->name = g_strdup (name);
67
68   nlp->label = NULL;
69
70   if ( label != NULL && 0 != strcmp ("", label))
71     nlp->label = g_strdup (label);
72
73   return nlp;
74 }
75
76 static void
77 nlp_destroy (gpointer data)
78 {
79   struct nlp *nlp = data ;
80   if ( ! nlp )
81     return;
82
83   g_free (nlp->name);
84   g_free (nlp->label);
85   g_free (nlp);
86 }
87
88 static void
89 on_change_clicked (GObject *obj, gpointer data)
90 {
91   struct autorecode *rd = data;
92   struct variable *var = NULL;
93   struct nlp *nlp;
94   GtkTreeModel *model = psppire_var_view_get_current_model (PSPPIRE_VAR_VIEW (rd->var_view));
95   GtkTreeIter iter;
96   GtkTreeSelection *selection =
97     gtk_tree_view_get_selection (GTK_TREE_VIEW (rd->var_view));
98
99   GList *rows = gtk_tree_selection_get_selected_rows (selection, &model);
100
101   const gchar *dest_var_name =
102     gtk_entry_get_text (GTK_ENTRY (rd->new_name_entry));
103
104   if ( NULL == rows || rows->next != NULL)
105     goto finish;
106
107   gtk_tree_model_get_iter (model, &iter, rows->data);
108
109   gtk_tree_model_get (model, &iter, 0, &var, -1);
110
111   g_hash_table_remove (rd->varmap, var);
112
113   nlp = nlp_create (dest_var_name, NULL);
114
115   g_hash_table_insert (rd->varmap, var, nlp);
116
117   gtk_tree_model_row_changed (model, rows->data, &iter);
118
119  finish:
120   g_list_foreach (rows, (GFunc) gtk_tree_path_free, NULL);
121   g_list_free (rows);
122 }
123
124
125 static char * generate_syntax (const struct autorecode *rd);
126
127
128 static void
129 refresh (PsppireDialog *dialog, struct autorecode *rd)
130 {
131   GtkTreeModel *target_list = gtk_tree_view_get_model (GTK_TREE_VIEW (rd->var_view));
132
133   gtk_entry_set_text (GTK_ENTRY (rd->new_name_entry), "");
134   gtk_widget_set_sensitive  (rd->new_name_entry, FALSE);
135   gtk_widget_set_sensitive  (rd->change_button, FALSE);
136
137   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (rd->ascending), TRUE);
138   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (rd->group), FALSE);
139   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (rd->blank), FALSE);
140
141   if (rd->varmap )
142     g_hash_table_remove_all (rd->varmap);
143
144   gtk_list_store_clear (GTK_LIST_STORE (target_list));
145 }
146
147
148 static gboolean
149 dialog_state_valid (gpointer data)
150 {
151   struct autorecode *rd = data;
152
153   GtkTreeModel *model = gtk_tree_view_get_model (GTK_TREE_VIEW (rd->var_view));
154
155   if (g_hash_table_size (rd->varmap) != gtk_tree_model_iter_n_children (model, NULL) )
156     return FALSE;
157
158  return TRUE;
159 }
160
161
162 static void
163 on_entry_change (struct autorecode *rd)
164 {
165   gboolean valid = TRUE;
166   const char *text = gtk_entry_get_text (GTK_ENTRY (rd->new_name_entry));
167
168   if ( 0 == strcmp ("", text))
169     valid = FALSE;
170   else if (psppire_dict_lookup_var (rd->dict, text))
171     valid = FALSE;
172   else
173     {
174       GHashTableIter iter;
175       gpointer key, value;
176
177       g_hash_table_iter_init (&iter, rd->varmap);
178       while (g_hash_table_iter_next (&iter, &key, &value)) 
179         {
180           struct nlp *nlp = value;
181           
182           if ( 0 == strcmp (nlp->name, text))
183             {
184               valid = FALSE;
185               break;
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 = psppire_var_view_get_current_model (PSPPIRE_VAR_VIEW (rd->var_view));
203
204   GList *rows = gtk_tree_selection_get_selected_rows (selection, &model);
205
206   if ( rows && !rows->next)
207     {
208       /* Exactly one row is selected */
209       struct nlp *nlp;
210       struct variable *var;
211       gboolean ok;
212       GtkTreeIter iter;
213
214       gtk_widget_set_sensitive  (rd->new_name_entry, TRUE);
215       gtk_widget_set_sensitive  (rd->change_button, TRUE);      
216
217
218       ok = gtk_tree_model_get_iter (model, &iter, (GtkTreePath*) rows->data);
219       g_return_if_fail (ok);
220
221       gtk_tree_model_get (model, &iter, 0, &var, -1);
222
223       nlp = g_hash_table_lookup (rd->varmap, var);
224
225       if (nlp)
226         gtk_entry_set_text (GTK_ENTRY (rd->new_name_entry), nlp->name ? nlp->name : "");
227       else
228         gtk_entry_set_text (GTK_ENTRY (rd->new_name_entry), "");
229     }
230   else
231     {
232       gtk_entry_set_text (GTK_ENTRY (rd->new_name_entry), "");
233       gtk_widget_set_sensitive  (rd->new_name_entry, FALSE);
234       gtk_widget_set_sensitive  (rd->change_button, FALSE);
235     }
236
237   g_list_foreach (rows, (GFunc) gtk_tree_path_free, NULL);
238   g_list_free (rows);
239 }
240
241 static void
242 render_new_var_name (GtkTreeViewColumn *tree_column,
243                      GtkCellRenderer *cell,
244                      GtkTreeModel *tree_model,
245                      GtkTreeIter *iter,
246                      gpointer data)
247 {
248   struct nlp *nlp = NULL;
249   struct autorecode *rd = data;
250
251   struct variable *var = NULL;
252
253   gtk_tree_model_get (tree_model, iter, 
254                       0, &var,
255                       -1);
256
257   nlp = g_hash_table_lookup (rd->varmap, var);
258
259   if ( nlp )
260     g_object_set (cell, "text", nlp->name, NULL);
261   else
262     g_object_set (cell, "text", "", NULL);
263 }
264
265
266
267 /* Pops up the Autorecode dialog box */
268 void
269 autorecode_dialog (PsppireDataWindow *de)
270 {
271   struct autorecode rd;
272   gint response;
273
274   GtkBuilder *xml = builder_new ("autorecode.ui");
275
276   GtkWidget *dialog = get_widget_assert   (xml, "autorecode-dialog");
277   GtkWidget *source = get_widget_assert   (xml, "dict-view");
278
279   rd.var_view = get_widget_assert   (xml, "var-view");
280
281   rd.new_name_entry = get_widget_assert (xml, "entry1");
282   rd.change_button = get_widget_assert (xml, "button1");
283   rd.ascending = get_widget_assert (xml, "radiobutton1");
284   rd.group = get_widget_assert (xml, "checkbutton1");
285   rd.blank = get_widget_assert (xml, "checkbutton2");
286
287     {
288       GtkTreeSelection *sel;
289
290       GtkCellRenderer *renderer = gtk_cell_renderer_text_new ();
291
292       GtkTreeViewColumn *col = gtk_tree_view_column_new_with_attributes (_("New"),
293                                                                          renderer,
294                                                                          "text", NULL,
295                                                                          NULL);
296
297       gtk_tree_view_column_set_cell_data_func (col, renderer,
298                                                render_new_var_name,
299                                                &rd, NULL);
300
301       gtk_tree_view_append_column (GTK_TREE_VIEW (rd.var_view), col);
302
303
304       col = gtk_tree_view_get_column (GTK_TREE_VIEW (rd.var_view), 0);
305
306       g_object_set (col, "title", _("Old"), NULL);
307
308       g_object_set (rd.var_view, "headers-visible", TRUE, NULL);
309
310       rd.varmap = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, nlp_destroy);
311
312
313       sel = gtk_tree_view_get_selection (GTK_TREE_VIEW (rd.var_view));
314
315
316       g_signal_connect (sel, "changed",
317                         G_CALLBACK (on_selection_change), &rd);
318
319       g_signal_connect (rd.change_button, "clicked",
320                         G_CALLBACK (on_change_clicked),  &rd);
321
322       g_signal_connect_swapped (rd.new_name_entry, "changed",
323                         G_CALLBACK (on_entry_change),  &rd);
324
325     }
326
327
328   gtk_window_set_transient_for (GTK_WINDOW (dialog), GTK_WINDOW (de));
329
330   g_object_get (de->data_editor, "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 }