eab7ab2deb01121323f9de86c15a0792bb926d67
[pspp-builds.git] / src / ui / gui / crosstabs-dialog.c
1 /* PSPPIRE - a graphical user interface for PSPP.
2    Copyright (C) 2008, 2010, 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 "checkbox-treeview.h"
20 #include "crosstabs-dialog.h"
21 #include "psppire-var-view.h"
22
23 #include <gtk/gtk.h>
24 #include <stdlib.h>
25
26 #include <ui/gui/psppire-data-window.h>
27 #include <ui/gui/dialog-common.h>
28 #include <ui/gui/dict-display.h>
29 #include "executor.h"
30 #include <ui/gui/psppire-dialog.h>
31 #include <ui/gui/psppire-var-store.h>
32 #include <ui/gui/helper.h>
33
34 #include "gettext.h"
35 #define _(msgid) gettext (msgid)
36 #define N_(msgid) msgid
37
38
39 #define CROSSTABS_STATS                       \
40   CS (CHISQ, N_("Chisq"))                         \
41   CS (PHI, N_("Phi"))         \
42   CS (CC, N_("CC"))                   \
43   CS (LAMBDA, N_("Lambda"))                   \
44   CS (UC, N_("UC")) \
45   CS (BTAU, N_("BTau"))                 \
46   CS (CTAU, N_("CTau"))                 \
47   CS (RISK, N_("Risk"))  \
48   CS (GAMMA, N_("Gamma"))                       \
49   CS (D, N_("D"))                         \
50   CS (KAPPA, N_("Kappa"))                 \
51   CS (ETA, N_("Eta"))  \
52   CS (CORR, N_("Corr")) \
53   CS (STATS_NONE, N_("None"))
54
55 #define CROSSTABS_CELLS \
56   CS (COUNT, N_("Count"))   \
57   CS (ROW, N_("Row"))   \
58   CS (COLUMN, N_("Column"))   \
59   CS (TOTAL, N_("Total"))   \
60   CS (EXPECTED, N_("Expected"))   \
61   CS (RESIDUAL, N_("Residual"))   \
62   CS (SRESIDUAL, N_("Std. Residual"))   \
63   CS (ASRESIDUAL, N_("Adjusted Std. Residual"))   \
64   CS (CELLS_NONE, N_("None"))
65
66 enum
67   {
68 #define CS(NAME, LABEL) CS_##NAME,
69     CROSSTABS_STATS
70 #undef CS
71     N_CROSSTABS_STATS
72   };
73
74 enum
75   {
76 #define CS(NAME, LABEL) CS_##NAME,
77     CROSSTABS_CELLS
78 #undef CS
79     N_CROSSTABS_CELLS
80   };
81
82 enum
83   {
84 #define CS(NAME, LABEL) B_CS_##NAME = 1u << CS_##NAME,
85     CROSSTABS_STATS
86     CROSSTABS_CELLS
87 #undef CS
88     B_CS_STATS_ALL = (1u << N_CROSSTABS_STATS) - 1,
89     B_CS_CELLS_ALL = (1u << N_CROSSTABS_CELLS) - 1,
90     B_CS_STATS_DEFAULT = B_CS_CHISQ,
91     B_CS_CELL_DEFAULT = B_CS_COUNT | B_CS_ROW | B_CS_COLUMN | B_CS_TOTAL,
92     B_CS_NONE
93   };
94
95 static const struct checkbox_entry_item stats[] =
96   {
97 #define CS(NAME, LABEL) {#NAME, LABEL},
98     CROSSTABS_STATS \
99     CS(NONE, N_("None"))
100 #undef CS
101   };
102
103 static const struct checkbox_entry_item cells[] =
104   {
105 #define CS(NAME, LABEL) {#NAME, LABEL},
106     CROSSTABS_CELLS \
107     CS(NONE, N_("None"))
108 #undef CS
109   };
110
111 struct format_options
112 {
113   gboolean avalue;
114   gboolean pivot;
115   gboolean table;
116 };
117
118 struct crosstabs_dialog
119 {
120   GtkTreeView *row_vars;
121   GtkTreeView *col_vars;
122   PsppireDict *dict;
123
124   GtkToggleButton *table_button;
125   GtkToggleButton *pivot_button;
126
127   GtkWidget *format_dialog;
128   GtkWidget *cell_dialog;
129   GtkWidget *stat_dialog;
130
131   GtkToggleButton  *avalue;
132   GtkTreeModel *stat;
133   GtkTreeModel *cell;
134
135   GtkWidget *stat_view;
136   GtkWidget *cell_view;
137   struct format_options current_opts;
138 };
139
140 static void
141 refresh (PsppireDialog *dialog, struct crosstabs_dialog *cd)
142 {
143   GtkTreeModel *liststore = gtk_tree_view_get_model (cd->row_vars);
144   gtk_list_store_clear (GTK_LIST_STORE (liststore));
145   
146   liststore = gtk_tree_view_get_model (cd->col_vars);
147   gtk_list_store_clear (GTK_LIST_STORE (liststore));
148 }
149 static void
150 on_format_clicked (struct crosstabs_dialog *cd)
151 {
152   int ret;
153   gboolean lab;
154   gboolean no_lab;
155   gboolean no_val_lab;
156
157   if (cd->current_opts.avalue)
158     {
159       gtk_toggle_button_set_active (cd->avalue, TRUE);
160     }
161   if (cd->current_opts.table)
162     {
163       gtk_toggle_button_set_active (cd->table_button, TRUE);
164     }
165   if (cd->current_opts.pivot)
166     {
167       gtk_toggle_button_set_active (cd->pivot_button, TRUE);
168     }
169
170   ret = psppire_dialog_run (PSPPIRE_DIALOG (cd->format_dialog));
171
172   if ( ret == PSPPIRE_RESPONSE_CONTINUE )
173     {
174       cd->current_opts.avalue = (gtk_toggle_button_get_active (cd->avalue) == TRUE ) 
175         ? TRUE : FALSE;
176       cd->current_opts.table = (gtk_toggle_button_get_active (cd->table_button) == TRUE)
177         ? TRUE : FALSE;
178       cd->current_opts.pivot = (gtk_toggle_button_get_active (cd->pivot_button) == TRUE)
179         ? TRUE : FALSE;
180     }
181 }
182
183 static void
184 on_statistics_clicked (struct crosstabs_dialog *cd)
185 {
186   GtkListStore *liststore;
187   int ret;
188
189   liststore = clone_list_store (GTK_LIST_STORE (cd->stat));
190
191   ret = psppire_dialog_run (PSPPIRE_DIALOG (cd->stat_dialog));
192
193   if ( ret == PSPPIRE_RESPONSE_CONTINUE )
194     {
195       g_object_unref (liststore);
196     }
197   else
198     {
199       g_object_unref (cd->stat);
200       gtk_tree_view_set_model (GTK_TREE_VIEW (cd->stat_view) , GTK_TREE_MODEL (liststore));
201       cd->stat = GTK_TREE_MODEL (liststore);
202     }
203 }
204 static void
205 on_cell_clicked (struct crosstabs_dialog *cd)
206 {
207   GtkListStore *liststore;
208   int ret;
209
210   liststore = clone_list_store (GTK_LIST_STORE (cd->cell));
211
212   ret = psppire_dialog_run (PSPPIRE_DIALOG (cd->cell_dialog));
213
214   if ( ret == PSPPIRE_RESPONSE_CONTINUE )
215     {
216       g_object_unref (liststore);
217     }
218   else
219     {
220       g_object_unref (cd->cell);
221       gtk_tree_view_set_model (GTK_TREE_VIEW (cd->cell_view) , GTK_TREE_MODEL (liststore));
222       cd->cell = GTK_TREE_MODEL (liststore);
223     }
224 }
225
226 static char *
227 generate_syntax (const struct crosstabs_dialog *cd)
228 {
229   gint i;
230   int n;
231   guint selected;
232   GtkTreeIter iter;
233   gboolean ok;
234
235   gchar *text;
236   GString *string = g_string_new ("CROSSTABS");
237
238   g_string_append (string, "\n\t/TABLES=");
239   psppire_var_view_append_names (PSPPIRE_VAR_VIEW (cd->row_vars), 0, string);
240   g_string_append (string, "\tBY\t");
241   psppire_var_view_append_names (PSPPIRE_VAR_VIEW (cd->col_vars), 0, string);
242
243   g_string_append (string, "\n\t/FORMAT=");
244
245   if (cd->current_opts.avalue)
246     {
247       g_string_append (string, "AVALUE");
248     }
249   else 
250     {
251       g_string_append (string, "DVALUE");
252     }
253   g_string_append (string, " ");
254   if (cd->current_opts.table)
255     g_string_append (string, "TABLES");
256   else
257     g_string_append (string, "NOTABLES");
258   g_string_append (string, " ");
259
260   if (cd->current_opts.pivot)
261     g_string_append (string, "PIVOT");
262   else 
263     g_string_append (string, "NOPIVOT");
264
265   selected = 0;
266   for (i = 0, ok = gtk_tree_model_get_iter_first (cd->stat, &iter); ok; 
267        i++, ok = gtk_tree_model_iter_next (cd->stat, &iter))
268     {
269       gboolean toggled;
270       gtk_tree_model_get (cd->stat, &iter,
271                           CHECKBOX_COLUMN_SELECTED, &toggled, -1); 
272       if (toggled) 
273         selected |= 1u << i; 
274       else 
275         selected &= ~(1u << i);
276     }
277
278   if (!(selected & (1u << CS_STATS_NONE)))
279     {
280       if (selected)
281         {
282           g_string_append (string, "\n\t/STATISTICS=");
283           n = 0;
284           for (i = 0; i < N_CROSSTABS_STATS; i++)
285             if (selected & (1u << i))
286               {
287                 if (n++)
288                   g_string_append (string, " ");
289                 g_string_append (string, stats[i].name);
290               }
291         }
292     }
293
294   selected = 0;
295   for (i = 0, ok = gtk_tree_model_get_iter_first (cd->cell, &iter); ok; 
296        i++, ok = gtk_tree_model_iter_next (cd->cell, &iter))
297     {
298       gboolean toggled;
299       gtk_tree_model_get (cd->cell, &iter,
300                           CHECKBOX_COLUMN_SELECTED, &toggled, -1); 
301       if (toggled) 
302         selected |= 1u << i; 
303       else 
304         selected &= ~(1u << i);
305     }
306
307   g_string_append (string, "\n\t/CELLS=");
308   if (selected & (1u << CS_CELLS_NONE))
309     g_string_append (string, "NONE");
310   else
311     {
312       n = 0;
313       for (i = 0; i < N_CROSSTABS_CELLS; i++)
314         if (selected & (1u << i))
315           {
316             if (n++)
317               g_string_append (string, " ");
318             g_string_append (string, cells[i].name);
319           }
320     }
321   
322   g_string_append (string, ".\n");
323
324   text = string->str;
325
326   g_string_free (string, FALSE);
327
328   return text;
329 }
330
331 /* Dialog is valid iff at least one row and one column variable has
332    been selected. */
333 static gboolean
334 dialog_state_valid (gpointer data)
335 {
336   struct crosstabs_dialog *cd = data;
337
338   GtkTreeModel *row_vars = gtk_tree_view_get_model (cd->row_vars);
339   GtkTreeModel *col_vars = gtk_tree_view_get_model (cd->col_vars);
340
341   GtkTreeIter notused;
342
343   return (gtk_tree_model_get_iter_first (row_vars, &notused) 
344     && gtk_tree_model_get_iter_first (col_vars, &notused));
345 }
346
347 /* Pops up the Crosstabs dialog box */
348 void
349 crosstabs_dialog (PsppireDataWindow *de)
350 {
351   gint response;
352   struct crosstabs_dialog cd;
353
354   GtkBuilder *xml = builder_new ("crosstabs.ui");
355   PsppireVarStore *vs = NULL;
356   PsppireDict *dict = NULL;
357
358
359   GtkWidget *dialog = get_widget_assert   (xml, "crosstabs-dialog");
360   GtkWidget *source = get_widget_assert   (xml, "dict-treeview");
361   GtkWidget *dest_rows =   get_widget_assert   (xml, "rows");
362   GtkWidget *dest_cols =   get_widget_assert   (xml, "cols");
363   GtkWidget *format_button = get_widget_assert (xml, "format-button");
364   GtkWidget *stat_button = get_widget_assert (xml, "stats-button");
365   GtkWidget *cell_button = get_widget_assert (xml, "cell-button");
366
367
368   cd.stat_view = get_widget_assert (xml, "stats-view");
369   cd.cell_view = get_widget_assert (xml, "cell-view");
370
371   g_object_get (de->data_editor, "var-store", &vs, NULL);
372
373   put_checkbox_items_in_treeview (GTK_TREE_VIEW(cd.stat_view),
374                                   B_CS_STATS_DEFAULT,
375                                   N_CROSSTABS_STATS,
376                                   stats
377                                   );
378   put_checkbox_items_in_treeview (GTK_TREE_VIEW(cd.cell_view),
379                                   B_CS_CELL_DEFAULT,
380                                   N_CROSSTABS_CELLS,
381                                   cells
382                                   );
383
384   gtk_window_set_transient_for (GTK_WINDOW (dialog), GTK_WINDOW (de));
385
386   g_object_get (vs, "dictionary", &dict, NULL);
387   g_object_set (source, "model", dict, NULL);
388
389   cd.row_vars = GTK_TREE_VIEW (dest_rows);
390   cd.col_vars = GTK_TREE_VIEW (dest_cols);
391   g_object_get (vs, "dictionary", &cd.dict, NULL);
392   cd.format_dialog = get_widget_assert (xml, "format-dialog");
393   cd.table_button = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "print-tables"));
394   cd.pivot_button = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "pivot"));
395   cd.stat_dialog = get_widget_assert (xml, "stat-dialog");
396   cd.cell_dialog = get_widget_assert (xml, "cell-dialog");
397
398   cd.stat = gtk_tree_view_get_model (GTK_TREE_VIEW (cd.stat_view));
399   cd.cell = gtk_tree_view_get_model (GTK_TREE_VIEW (cd.cell_view));
400   cd.avalue = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "ascending"));
401   cd.current_opts.avalue = TRUE;
402   cd.current_opts.table = TRUE;
403   cd.current_opts.pivot = TRUE;
404
405   gtk_window_set_transient_for (GTK_WINDOW (cd.format_dialog), GTK_WINDOW (de));
406   gtk_window_set_transient_for (GTK_WINDOW (cd.cell_dialog), GTK_WINDOW (de));
407   gtk_window_set_transient_for (GTK_WINDOW (cd.stat_dialog), GTK_WINDOW (de));
408
409   g_signal_connect (dialog, "refresh", G_CALLBACK (refresh),  &cd);
410
411   psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (dialog),
412                                       dialog_state_valid, &cd);
413
414   g_signal_connect_swapped (format_button, "clicked",
415                             G_CALLBACK (on_format_clicked),  &cd);
416   g_signal_connect_swapped (stat_button, "clicked",
417                             G_CALLBACK (on_statistics_clicked),  &cd);
418   g_signal_connect_swapped (cell_button, "clicked",
419                             G_CALLBACK (on_cell_clicked),  &cd);
420
421   response = psppire_dialog_run (PSPPIRE_DIALOG (dialog));
422
423
424   switch (response)
425     {
426     case GTK_RESPONSE_OK:
427       g_free (execute_syntax_string (de, generate_syntax (&cd)));
428       break;
429     case PSPPIRE_RESPONSE_PASTE:
430       g_free (paste_syntax_to_window (generate_syntax (&cd)));
431       break;
432     default:
433       break;
434     }
435
436   g_object_unref (xml);
437 }