Fixed crash when trying to get the label of a non-existant variable.
[pspp-builds.git] / src / ui / gui / compute-dialog.c
1 /* PSPPIRE - a graphical user interface for PSPP.
2    Copyright (C) 2007  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 #include <gtk/gtk.h>
19 #include "compute-dialog.h"
20 #include "helper.h"
21 #include "psppire-dialog.h"
22 #include "psppire-keypad.h"
23 #include "data-editor.h"
24 #include <gtksheet/gtksheet.h>
25 #include "psppire-var-store.h"
26 #include "dialog-common.h"
27 #include "dict-display.h"
28
29 #include <language/expressions/public.h>
30 #include <language/syntax-string-source.h>
31 #include "syntax-editor.h"
32
33 static void function_list_populate (GtkTreeView *tv);
34
35 static void insert_function_into_syntax_area (GtkTreeIter iter,
36                                               GtkWidget *text_view,
37                                               GtkTreeModel *model,
38                                               gpointer data
39                                               );
40
41 static void insert_source_row_into_text_view (GtkTreeIter iter,
42                                               GtkWidget *dest,
43                                               GtkTreeModel *model,
44                                               gpointer data
45                                               );
46
47
48
49 struct compute_dialog
50 {
51   GladeXML *xml;  /* The xml that generated the widgets */
52   PsppireDict *dict;
53   gboolean use_type;
54 };
55
56
57 static void
58 on_target_change (GObject *obj, struct compute_dialog *cd)
59 {
60   GtkWidget *target = get_widget_assert (cd->xml, "compute-entry1");
61   GtkWidget *type_and_label = get_widget_assert (cd->xml, "compute-button1");
62
63   const gchar *var_name = gtk_entry_get_text (GTK_ENTRY (target));
64   gboolean valid = var_name && strcmp ("", var_name);
65
66   gtk_widget_set_sensitive (type_and_label, valid);
67 }
68
69 static void
70 refresh (GObject *obj, const struct compute_dialog *cd)
71 {
72   GtkTextIter start, end;
73   GtkWidget *target = get_widget_assert (cd->xml, "compute-entry1");
74   GtkWidget *syntax_area = get_widget_assert (cd->xml, "compute-textview1");
75   GtkWidget *varlist = get_widget_assert (cd->xml, "compute-treeview1");
76   GtkWidget *funclist = get_widget_assert (cd->xml, "compute-treeview2");
77
78   GtkTextBuffer *buffer =
79     gtk_text_view_get_buffer (GTK_TEXT_VIEW (syntax_area));
80
81   GtkTreeSelection *selection;
82
83   /* Clear the target variable entry box */
84   gtk_entry_set_text (GTK_ENTRY (target), "");
85   g_signal_emit_by_name (target, "changed");
86
87   /* Clear the syntax area textbuffer */
88   gtk_text_buffer_get_start_iter (buffer, &start);
89   gtk_text_buffer_get_end_iter (buffer, &end);
90   gtk_text_buffer_delete (buffer, &start, &end);
91
92   /* Unselect all items in the treeview */
93   selection = gtk_tree_view_get_selection (GTK_TREE_VIEW (varlist));
94   gtk_tree_selection_unselect_all (selection);
95
96   /* And the other one */
97   selection = gtk_tree_view_get_selection (GTK_TREE_VIEW (funclist));
98   gtk_tree_selection_unselect_all (selection);
99 }
100
101
102 static void
103 erase_selection (GtkTextBuffer *buffer)
104 {
105   GtkTextIter start, end;
106
107   gtk_text_buffer_get_selection_bounds (buffer, &start, &end);
108
109   gtk_text_buffer_delete (buffer, &start, &end);
110 }
111
112
113
114 static void
115 on_keypad_button (PsppireKeypad *kp, const gchar *syntax, gpointer data)
116 {
117   GladeXML *xml = data;
118
119   GtkWidget *rhs = get_widget_assert (xml, "compute-textview1");
120
121   GtkTextBuffer *buffer = gtk_text_view_get_buffer (GTK_TEXT_VIEW (rhs));
122
123   erase_selection (buffer);
124
125   gtk_text_buffer_insert_at_cursor (buffer, syntax, strlen (syntax));
126
127   if (0 == strcmp (syntax, "()"))
128     {
129       GtkTextIter iter;
130       GtkTextMark *cursor = gtk_text_buffer_get_insert (buffer);
131       gtk_text_buffer_get_iter_at_mark (buffer, &iter, cursor);
132       gtk_text_iter_backward_cursor_position (&iter);
133       gtk_text_buffer_move_mark (buffer, cursor, &iter);
134     }
135
136 }
137
138 static void
139 erase (PsppireKeypad *kp, gpointer data)
140 {
141   GladeXML *xml = data;
142
143   GtkWidget *rhs = get_widget_assert (xml, "compute-textview1");
144
145   GtkTextBuffer *buffer = gtk_text_view_get_buffer (GTK_TEXT_VIEW (rhs));
146
147   erase_selection (buffer);
148 }
149
150 static char *
151 generate_syntax (const struct compute_dialog *cd)
152 {
153   gchar *text;
154   GString *string ;
155   const gchar *target_name ;
156   gchar *expression;
157   const gchar *label;
158   GtkTextIter start, end;
159   GtkWidget *target = get_widget_assert   (cd->xml, "compute-entry1");
160   GtkWidget *syntax_area = get_widget_assert (cd->xml, "compute-textview1");
161   GtkWidget *string_toggle = get_widget_assert (cd->xml, "radio-button-string");
162   GtkWidget *user_label_toggle =
163     get_widget_assert (cd->xml, "radio-button-user-label");
164   GtkWidget *width_entry = get_widget_assert (cd->xml, "type-and-label-width");
165   GtkWidget *label_entry = get_widget_assert (cd->xml,
166                                               "type-and-label-label-entry");
167
168
169   GtkTextBuffer *buffer =
170     gtk_text_view_get_buffer (GTK_TEXT_VIEW (syntax_area));
171
172   gtk_text_buffer_get_start_iter (buffer, &start);
173   gtk_text_buffer_get_end_iter (buffer, &end);
174
175   target_name = gtk_entry_get_text (GTK_ENTRY (target));
176
177   expression = gtk_text_buffer_get_text (buffer, &start, &end, FALSE);
178
179   string = g_string_sized_new (64);
180
181   if ( cd-> use_type &&
182        NULL == psppire_dict_lookup_var (cd->dict, target_name ))
183     {
184       if ( gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (string_toggle)))
185         {
186           const char *w = gtk_entry_get_text (GTK_ENTRY(width_entry));
187           g_string_append_printf (string,
188                                   "STRING %s (a%s).\n", target_name, w);
189         }
190       else
191         g_string_append_printf (string, "NUMERIC %s.\n", target_name);
192     }
193
194   if ( gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (user_label_toggle)))
195     label = gtk_entry_get_text (GTK_ENTRY (label_entry));
196   else
197     label = expression;
198
199   if ( strlen (label) > 0 )
200     g_string_append_printf (string, "VARIABLE LABEL %s '%s'.\n",
201                             target_name,
202                             label);
203
204   g_string_append_printf (string, "COMPUTE %s = %s.\n",
205                           target_name,
206                           expression
207                           );
208
209   g_string_append (string, "EXECUTE.\n");
210
211
212   g_free (expression);
213
214   text = string->str;
215
216   g_string_free (string, FALSE);
217
218   return text;
219 }
220
221 static void
222 reset_type_label_dialog (struct compute_dialog *cd)
223 {
224   const gchar *target_name;
225   struct variable *target_var;
226
227   GtkWidget *width_entry =
228     get_widget_assert (cd->xml, "type-and-label-width");
229
230   GtkWidget *label_entry =
231     get_widget_assert (cd->xml, "type-and-label-label-entry");
232
233   GtkWidget *numeric_target =
234     get_widget_assert (cd->xml, "radio-button-numeric");
235
236   GtkWidget *string_target =
237     get_widget_assert (cd->xml, "radio-button-string");
238
239
240   target_name = gtk_entry_get_text
241     (GTK_ENTRY (get_widget_assert (cd->xml, "compute-entry1")));
242
243
244   if ( (target_var = psppire_dict_lookup_var (cd->dict, target_name)) )
245     {
246       /* Existing Variable */
247       const gchar *label ;
248       GtkWidget *user_label =
249         get_widget_assert (cd->xml, "radio-button-user-label");
250
251       gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (user_label), TRUE);
252
253       label = var_get_label (target_var);
254
255       if ( label )
256         {
257           gtk_entry_set_text (GTK_ENTRY (label_entry), label);
258         }
259
260       gtk_widget_set_sensitive (width_entry, FALSE);
261
262       if ( var_is_numeric (target_var))
263         gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (numeric_target),
264                                       TRUE);
265       else
266         gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (string_target),
267                                       TRUE);
268
269       gtk_widget_set_sensitive (numeric_target, FALSE);
270       gtk_widget_set_sensitive (string_target, FALSE);
271     }
272   else
273     {
274       GtkWidget *expression =
275         get_widget_assert (cd->xml, "radio-button-expression-label");
276
277       gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (expression), TRUE);
278
279       gtk_widget_set_sensitive (width_entry, TRUE);
280       gtk_widget_set_sensitive (numeric_target, TRUE);
281       gtk_widget_set_sensitive (string_target, TRUE);
282
283       gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (numeric_target),
284                                     TRUE);
285     }
286
287 }
288
289 static void
290 run_type_label_dialog (GtkButton *b, gpointer data)
291 {
292   struct compute_dialog *cd = data;
293   gint response;
294
295   GtkWidget *subdialog = get_widget_assert (cd->xml, "type-and-label-dialog");
296   GtkWidget *dialog = get_widget_assert (cd->xml, "compute-variable-dialog");
297
298   gtk_window_set_transient_for (GTK_WINDOW (subdialog), GTK_WINDOW (dialog));
299
300   reset_type_label_dialog (cd);
301   response = psppire_dialog_run (PSPPIRE_DIALOG (subdialog));
302   if ( response == PSPPIRE_RESPONSE_CONTINUE)
303     cd->use_type = TRUE;
304 }
305
306
307 static void
308 on_expression_toggle (GtkToggleButton *button, gpointer data)
309 {
310   struct compute_dialog *cd = data;
311
312   GtkWidget *entry =
313     get_widget_assert (cd->xml, "type-and-label-label-entry");
314
315   if ( gtk_toggle_button_get_active (button))
316     {
317       gtk_entry_set_text (GTK_ENTRY (entry), "");
318       gtk_widget_set_sensitive (entry, FALSE);
319     }
320   else
321     {
322       const char *label;
323       struct variable *target_var;
324       const gchar *target_name = gtk_entry_get_text
325         (GTK_ENTRY (get_widget_assert (cd->xml, "compute-entry1")));
326
327       target_var = psppire_dict_lookup_var (cd->dict, target_name);
328       if ( target_var )
329         {
330           label = var_get_label (target_var);
331
332           if ( label )
333             gtk_entry_set_text (GTK_ENTRY (entry), label);
334         }
335       else
336         gtk_entry_set_text (GTK_ENTRY (entry), "");
337
338       gtk_widget_set_sensitive (entry, TRUE);
339     }
340 }
341
342
343 /* Return TRUE if the dialog box's widgets' state are such that clicking OK
344    might not result in erroneous syntax being generated */
345 static gboolean
346 contents_plausible (gpointer data)
347 {
348   struct compute_dialog *cd = data;
349
350   GtkWidget *target      = get_widget_assert (cd->xml, "compute-entry1");
351   GtkWidget *syntax_area = get_widget_assert (cd->xml, "compute-textview1");
352   GtkTextBuffer *buffer  =
353     gtk_text_view_get_buffer (GTK_TEXT_VIEW (syntax_area));
354
355   if ( 0 == strcmp ("", gtk_entry_get_text (GTK_ENTRY (target))))
356     return FALSE;
357
358   if ( gtk_text_buffer_get_char_count (buffer) == 0 )
359     return FALSE;
360
361   return TRUE;
362 }
363
364 /* Pops up the Compute dialog box */
365 void
366 compute_dialog (GObject *o, gpointer data)
367 {
368   gint response;
369   struct data_editor *de = data;
370
371   PsppireVarStore *vs;
372   struct compute_dialog scd;
373
374   GladeXML *xml = XML_NEW ("psppire.glade");
375
376   GtkWidget *dialog = get_widget_assert   (xml, "compute-variable-dialog");
377
378   GtkWidget *dict_view = get_widget_assert (xml, "compute-treeview1");
379   GtkWidget *functions = get_widget_assert (xml, "compute-treeview2");
380   GtkWidget *keypad    = get_widget_assert (xml, "psppire-keypad1");
381   GtkWidget *target    = get_widget_assert (xml, "compute-entry1");
382   GtkWidget *syntax_area = get_widget_assert (xml, "compute-textview1");
383   GtkWidget *var_selector = get_widget_assert (xml, "compute-selector1");
384   GtkWidget *func_selector = get_widget_assert (xml, "compute-selector2");
385   GtkWidget *type_and_label = get_widget_assert (xml, "compute-button1");
386
387   GtkWidget *expression =
388         get_widget_assert (xml, "radio-button-expression-label");
389
390
391   GtkSheet *var_sheet =
392     GTK_SHEET (get_widget_assert (de->xml, "variable_sheet"));
393
394   vs = PSPPIRE_VAR_STORE (gtk_sheet_get_model (var_sheet));
395
396
397   scd.dict = vs->dict;
398   scd.use_type = FALSE;
399
400   g_signal_connect (expression, "toggled",
401                     G_CALLBACK(on_expression_toggle), &scd);
402
403   gtk_window_set_transient_for (GTK_WINDOW (dialog), de->parent.window);
404
405
406   attach_dictionary_to_treeview (GTK_TREE_VIEW (dict_view),
407                                  vs->dict,
408                                  GTK_SELECTION_SINGLE, NULL);
409
410
411   psppire_selector_set_subjects (PSPPIRE_SELECTOR (var_selector),
412                                  dict_view, syntax_area,
413                                  insert_source_row_into_text_view,
414                                  NULL,
415                                  NULL);
416
417
418   function_list_populate (GTK_TREE_VIEW (functions));
419
420   psppire_selector_set_subjects (PSPPIRE_SELECTOR (func_selector),
421                                  functions, syntax_area,
422                                  insert_function_into_syntax_area,
423                                  NULL,
424                                  NULL);
425
426
427   scd.xml = xml;
428
429   psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (dialog),
430                                       contents_plausible, &scd);
431
432   g_signal_connect (target, "changed", G_CALLBACK (on_target_change), &scd);
433
434   g_signal_connect (dialog, "refresh", G_CALLBACK (refresh),  &scd);
435
436   g_signal_connect (keypad, "insert-syntax",
437                     G_CALLBACK (on_keypad_button),  xml);
438
439   g_signal_connect (keypad, "erase",
440                     G_CALLBACK (erase),  xml);
441
442
443   g_signal_connect (type_and_label, "clicked",
444                     G_CALLBACK (run_type_label_dialog),  &scd);
445
446
447
448   response = psppire_dialog_run (PSPPIRE_DIALOG (dialog));
449
450
451   switch (response)
452     {
453     case GTK_RESPONSE_OK:
454       {
455         gchar *syntax = generate_syntax (&scd);
456         struct getl_interface *sss = create_syntax_string_source (syntax);
457         execute_syntax (sss);
458
459         g_free (syntax);
460       }
461       break;
462     case PSPPIRE_RESPONSE_PASTE:
463       {
464         gchar *syntax = generate_syntax (&scd);
465
466         struct syntax_editor *se =
467           (struct syntax_editor *) window_create (WINDOW_SYNTAX, NULL);
468
469         gtk_text_buffer_insert_at_cursor (se->buffer, syntax, -1);
470
471         g_free (syntax);
472       }
473       break;
474     default:
475       break;
476     }
477
478   g_object_unref (xml);
479 }
480
481
482 enum {
483   COL_NAME,
484   COL_USAGE,
485   COL_ARITY
486 };
487
488
489 static void
490 function_list_populate (GtkTreeView *tv)
491 {
492   GtkListStore *liststore;
493   GtkTreeIter iter;
494   gint i;
495
496   const gint n_funcs = expr_get_function_cnt ();
497
498   liststore = gtk_list_store_new (3, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_INT);
499
500   for (i = 0 ; i < n_funcs ; ++i)
501     {
502       const struct operation *op = expr_get_function (i);
503
504       gtk_list_store_append (liststore, &iter);
505
506       gtk_list_store_set (liststore, &iter,
507                           COL_NAME, expr_operation_get_name (op),
508                           COL_USAGE, expr_operation_get_prototype (op),
509                           COL_ARITY, expr_operation_get_arg_cnt (op),
510                           -1);
511     }
512
513
514
515   /* Set the cell rendering */
516
517   {
518     GtkTreeViewColumn *col;
519     GtkCellRenderer *renderer;
520
521
522     col = gtk_tree_view_column_new ();
523
524     gtk_tree_view_append_column (tv, col);
525
526     renderer = gtk_cell_renderer_text_new ();
527
528     gtk_tree_view_column_pack_start (col, renderer, TRUE);
529
530     gtk_tree_view_column_add_attribute (col, renderer, "text", COL_USAGE);
531   }
532
533   gtk_tree_view_set_model (tv, GTK_TREE_MODEL (liststore));
534 }
535
536
537
538
539 static void
540 insert_function_into_syntax_area (GtkTreeIter iter,
541                                   GtkWidget *text_view,
542                                   GtkTreeModel *model,
543                                   gpointer data
544                                   )
545 {
546   GString *string;
547   GValue name_value = {0};
548   GValue arity_value = {0};
549   gint arity;
550   gint i;
551
552   GtkTextBuffer *buffer ;
553
554   g_return_if_fail (GTK_IS_TEXT_VIEW (text_view));
555
556   buffer = gtk_text_view_get_buffer (GTK_TEXT_VIEW (text_view));
557
558   gtk_tree_model_get_value (model, &iter, COL_NAME, &name_value);
559   gtk_tree_model_get_value (model, &iter, COL_ARITY, &arity_value);
560
561   arity = g_value_get_int (&arity_value);
562
563   string = g_string_new (g_value_get_string (&name_value));
564
565   g_string_append (string, "(");
566   for ( i = 0 ; i < arity -1 ; ++i )
567     {
568       g_string_append (string, "?,");
569     }
570   g_string_append (string, "?)");
571
572
573   erase_selection (buffer);
574
575   gtk_text_buffer_insert_at_cursor (buffer, string->str, string->len);
576
577   g_value_unset (&name_value);
578   g_value_unset (&arity_value);
579   g_string_free (string, TRUE);
580
581   /* Now position the cursor over the first '?' */
582   {
583     GtkTextIter insert;
584     GtkTextIter selectbound;
585     GtkTextMark *cursor = gtk_text_buffer_get_insert (buffer);
586     gtk_text_buffer_get_iter_at_mark (buffer, &insert, cursor);
587     for ( i = 0 ; i < arity ; ++i )
588       {
589         gtk_text_iter_backward_cursor_position (&insert);
590         gtk_text_iter_backward_cursor_position (&insert);
591       }
592     selectbound = insert;
593     gtk_text_iter_forward_cursor_position (&selectbound);
594
595     gtk_text_buffer_select_range (buffer, &insert, &selectbound);
596   }
597
598 }
599
600 /* Inserts the name of the selected variable into the destination widget.
601    The destination widget must be a GtkTextView
602  */
603 static void
604 insert_source_row_into_text_view (GtkTreeIter iter,
605                                   GtkWidget *dest,
606                                   GtkTreeModel *model,
607                                   gpointer data
608                                   )
609 {
610   GtkTreePath *path;
611   PsppireDict *dict;
612   gint *idx;
613   struct variable *var;
614   GtkTreeIter dict_iter;
615   gchar *name;
616   GtkTextBuffer *buffer;
617
618   g_return_if_fail (GTK_IS_TEXT_VIEW (dest));
619
620   if ( GTK_IS_TREE_MODEL_FILTER (model))
621     {
622       dict = PSPPIRE_DICT (gtk_tree_model_filter_get_model
623                            (GTK_TREE_MODEL_FILTER(model)));
624
625       gtk_tree_model_filter_convert_iter_to_child_iter (GTK_TREE_MODEL_FILTER
626                                                         (model),
627                                                         &dict_iter, &iter);
628     }
629   else
630     {
631       dict = PSPPIRE_DICT (model);
632       dict_iter = iter;
633     }
634
635   path = gtk_tree_model_get_path (GTK_TREE_MODEL (dict), &dict_iter);
636
637   idx = gtk_tree_path_get_indices (path);
638
639   var =  psppire_dict_get_variable (dict, *idx);
640
641   gtk_tree_path_free (path);
642
643   name = pspp_locale_to_utf8 (var_get_name (var), -1, NULL);
644
645   buffer = gtk_text_view_get_buffer (GTK_TEXT_VIEW (dest));
646
647   erase_selection (buffer);
648
649   gtk_text_buffer_insert_at_cursor (buffer, name, -1);
650
651   g_free (name);
652 }