Merge remote branch 'origin/master' into import-gui
[pspp] / src / ui / gui / oneway-anova-dialog.c
1 /* PSPPIRE - a graphical user interface for PSPP.
2    Copyright (C) 2007, 2010, 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
18 #include <config.h>
19 #include <gtk/gtk.h>
20 #include "oneway-anova-dialog.h"
21 #include "psppire-dict.h"
22 #include "psppire-var-store.h"
23 #include "psppire-var-view.h"
24 #include "builder-wrapper.h"
25 #include "psppire-data-window.h"
26 #include "psppire-dialog.h"
27 #include "dialog-common.h"
28 #include "psppire-acr.h"
29 #include "psppire-selector.h"
30 #include "dict-display.h"
31
32 #include "executor.h"
33 #include "helper.h"
34
35 #include "gettext.h"
36 #define _(msgid) gettext (msgid)
37 #define N_(msgid) msgid
38
39
40 struct contrasts_subdialog;
41 struct oneway_anova_dialog;
42
43
44 static void run_contrasts_dialog (struct oneway_anova_dialog *);
45 static void push_new_store (GArray *, struct contrasts_subdialog *);
46
47 static void next (GtkWidget *, gpointer);
48 static void prev (GtkWidget *, gpointer);
49
50
51 struct contrasts_subdialog
52 {
53   /* Contrasts Dialog widgets */
54   GtkWidget *contrasts_dialog;
55   GtkWidget *stack_label;
56   PsppireAcr *acr;
57
58
59   /* Gets copied into contrasts when "Continue"
60      is clicked */
61   GArray *temp_contrasts;
62
63   /* Index into the temp_contrasts */
64   guint c;
65
66   GtkWidget *prev;
67   GtkWidget *next;
68
69   GtkWidget *ctotal;
70 };
71
72
73 struct oneway_anova_dialog
74 {
75   PsppireDict *dict;
76
77   GtkWidget *factor_entry;
78   GtkWidget *vars_treeview;
79   GtkWindow *dialog;
80   GArray *contrasts_array;
81
82   GtkToggleButton *descriptives;
83   GtkToggleButton *homogeneity;
84
85   struct contrasts_subdialog contrasts;
86 };
87
88 static gboolean
89 dialog_state_valid (gpointer data)
90 {
91   struct oneway_anova_dialog *ow = data;
92
93   GtkTreeModel *vars =
94     gtk_tree_view_get_model (GTK_TREE_VIEW (ow->vars_treeview));
95
96   GtkTreeIter notused;
97
98   if ( !gtk_tree_model_get_iter_first (vars, &notused) )
99     return FALSE;
100
101   if ( 0 == strcmp ("", gtk_entry_get_text (GTK_ENTRY (ow->factor_entry))))
102     return FALSE;
103
104   return TRUE;
105 }
106
107 static gchar * generate_syntax (const struct oneway_anova_dialog *);
108
109
110 static void
111 refresh (struct oneway_anova_dialog *ow)
112 {
113   GtkTreeModel *model =
114     gtk_tree_view_get_model (GTK_TREE_VIEW (ow->vars_treeview));
115
116   gtk_entry_set_text (GTK_ENTRY (ow->factor_entry), "");
117
118   gtk_list_store_clear (GTK_LIST_STORE (model));
119 }
120
121
122
123 /* Pops up the dialog box */
124 void
125 oneway_anova_dialog (PsppireDataWindow *de)
126 {
127   gint response;
128
129   PsppireVarStore *vs = NULL;
130
131   struct oneway_anova_dialog ow;
132
133   GtkBuilder *builder = builder_new ("oneway.ui");
134
135   GtkWidget *dict_view =
136     get_widget_assert (builder, "oneway-anova-treeview1");
137
138   GtkWidget *selector2 =
139     get_widget_assert (builder, "oneway-anova-selector2");
140
141   GtkWidget *contrasts_button =
142     get_widget_assert (builder, "contrasts-button");
143
144
145   g_signal_connect_swapped (contrasts_button, "clicked",
146                     G_CALLBACK (run_contrasts_dialog), &ow);
147
148
149   ow.factor_entry = get_widget_assert (builder, "oneway-anova-entry");
150   ow.vars_treeview =
151     get_widget_assert (builder, "oneway-anova-treeview2");
152
153   ow.descriptives =
154     GTK_TOGGLE_BUTTON (get_widget_assert (builder, "checkbutton1"));
155
156   ow.homogeneity =
157     GTK_TOGGLE_BUTTON (get_widget_assert (builder, "checkbutton2"));
158
159   g_object_get (de->data_editor, "var-store", &vs, NULL);
160
161   g_object_get (vs, "dictionary", &ow.dict, NULL);
162
163   ow.dialog =
164     GTK_WINDOW (get_widget_assert (builder, "oneway-anova-dialog"));
165
166   gtk_window_set_transient_for (ow.dialog, GTK_WINDOW (de));
167
168   g_object_set (dict_view, "model", ow.dict, NULL);
169
170
171   psppire_selector_set_filter_func (PSPPIRE_SELECTOR (selector2),
172                                     is_currently_in_entry);
173
174
175   g_signal_connect_swapped (ow.dialog, "refresh", G_CALLBACK (refresh),  &ow);
176
177
178
179   psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (ow.dialog),
180                                       dialog_state_valid, &ow);
181
182
183   {
184     struct contrasts_subdialog *cd = &ow.contrasts;
185     GtkEntry *entry = GTK_ENTRY (get_widget_assert (builder, "entry1"));
186
187     cd->acr = PSPPIRE_ACR (get_widget_assert (builder, "psppire-acr1"));
188     cd->contrasts_dialog = get_widget_assert (builder, "contrasts-dialog");
189
190     cd->next = get_widget_assert (builder, "next-button");
191     cd->prev = get_widget_assert (builder, "prev-button");
192     cd->ctotal = get_widget_assert (builder, "entry2");
193
194     cd->stack_label = get_widget_assert (builder, "contrast-stack-label");
195
196     /* Contrasts */
197     ow.contrasts_array = g_array_new (FALSE, FALSE, sizeof (GtkListStore *));
198
199     g_signal_connect (cd->next, "clicked", G_CALLBACK (next), cd);
200     g_signal_connect (cd->prev, "clicked", G_CALLBACK (prev), cd);
201
202     psppire_acr_set_entry (cd->acr, entry);
203
204     gtk_window_set_transient_for (GTK_WINDOW (cd->contrasts_dialog),
205                                   GTK_WINDOW (de));
206   }
207
208   response = psppire_dialog_run (PSPPIRE_DIALOG (ow.dialog));
209
210   switch (response)
211     {
212     case GTK_RESPONSE_OK:
213       g_free (execute_syntax_string (de, generate_syntax (&ow)));
214       break;
215     case PSPPIRE_RESPONSE_PASTE:
216       g_free (paste_syntax_to_window (generate_syntax (&ow)));
217       break;
218     default:
219       break;
220     }
221
222   g_array_free (ow.contrasts_array, FALSE);
223
224   g_object_unref (builder);
225 }
226
227
228 static gchar * generate_syntax (const struct oneway_anova_dialog *ow)
229 {
230   gchar *text;
231   gint i;
232   gboolean descriptives = gtk_toggle_button_get_active (ow->descriptives);
233   gboolean homogeneity = gtk_toggle_button_get_active (ow->homogeneity);
234   struct string dss;
235
236   ds_init_cstr (&dss, "ONEWAY /VARIABLES=");
237
238   psppire_var_view_append_names_str (PSPPIRE_VAR_VIEW (ow->vars_treeview), 0, &dss);
239
240   ds_put_cstr (&dss, " BY ");
241
242   ds_put_cstr (&dss, gtk_entry_get_text (GTK_ENTRY (ow->factor_entry)));
243
244   if (descriptives || homogeneity )
245     {
246       ds_put_cstr (&dss, "\n\t/STATISTICS=");
247       if (descriptives)
248         ds_put_cstr (&dss, "DESCRIPTIVES ");
249       if (homogeneity)
250         ds_put_cstr (&dss, "HOMOGENEITY ");
251     }
252
253   for (i = 0 ; i < ow->contrasts_array->len ; ++i )
254     {
255       GtkListStore *ls = g_array_index (ow->contrasts_array, GtkListStore*, i);
256       GtkTreeIter iter;
257       gboolean ok;
258
259       ds_put_cstr (&dss, "\n\t/CONTRAST=");
260
261       for (ok = gtk_tree_model_get_iter_first (GTK_TREE_MODEL(ls),
262                                                &iter);
263            ok;
264            ok = gtk_tree_model_iter_next (GTK_TREE_MODEL (ls), &iter))
265         {
266           gdouble v;
267
268           gtk_tree_model_get (GTK_TREE_MODEL (ls), &iter, 0, &v, -1);
269
270           ds_put_c_format (&dss, " %g", v);
271         }
272     }
273
274   ds_put_cstr (&dss, ".\n");
275
276   text = ds_steal_cstr (&dss);
277   ds_destroy (&dss);
278
279   return text;
280 }
281
282
283 \f
284 /* Contrasts stuff */
285
286
287
288
289 /* Callback for when the list store currently associated with the
290    treeview has changed.  It sets the widgets of the subdialog
291    to reflect the store's new state.
292 */
293 static void
294 list_store_changed (struct contrasts_subdialog *csd)
295 {
296   gboolean ok;
297   gdouble total = 0.0;
298   GtkTreeIter iter;
299   GtkTreeModel *ls = NULL;
300   gchar *text =
301     g_strdup_printf (_("Contrast %d of %d"),
302                      csd->c, csd->temp_contrasts->len);
303
304   gtk_label_set_label (GTK_LABEL (csd->stack_label), text);
305
306   g_free (text);
307
308   gtk_widget_set_sensitive (csd->prev, csd->c > 1);
309
310   if ( csd->c > 0 )
311     ls = g_array_index (csd->temp_contrasts, GtkTreeModel*, csd->c - 1);
312
313   psppire_acr_set_model (csd->acr, GTK_LIST_STORE (ls));
314
315   /* Sensitive iff the liststore has two items or more */
316   gtk_widget_set_sensitive (csd->next,
317                             gtk_tree_model_iter_nth_child
318                             (ls, &iter,  NULL, 1));
319
320   for (ok = gtk_tree_model_get_iter_first (ls, &iter);
321        ok;
322        ok = gtk_tree_model_iter_next (ls, &iter)
323        )
324     {
325       gdouble v;
326       gtk_tree_model_get (ls, &iter, 0, &v, -1);
327       total += v;
328     }
329
330   text = g_strdup_printf ("%g", total);
331
332   gtk_entry_set_text (GTK_ENTRY (csd->ctotal), text);
333
334   g_free (text);
335 }
336
337
338
339 /* Copy the contrasts array into the local array */
340 static GArray *
341 clone_contrasts_array (GArray *src_array)
342 {
343   gint i;
344
345   GArray *dest_array =
346     g_array_sized_new (FALSE, FALSE, sizeof (GtkListStore *),
347                        src_array->len);
348
349   for (i = 0 ; i < src_array->len ; ++i )
350     {
351
352       GtkTreeIter src_iter;
353       GtkListStore *src = g_array_index (src_array, GtkListStore*, i);
354       GtkListStore *dest;
355
356       /* Refuse to copy empty stores */
357       if (! gtk_tree_model_get_iter_first (GTK_TREE_MODEL (src),
358                                            &src_iter))
359         continue;
360
361       dest = clone_list_store (src);
362
363       g_array_append_val (dest_array, dest);
364     }
365
366   return dest_array;
367 }
368
369
370 static void
371 run_contrasts_dialog (struct oneway_anova_dialog *ow)
372 {
373   struct contrasts_subdialog *csd = &ow->contrasts;
374   gint response;
375
376   csd->temp_contrasts = clone_contrasts_array (ow->contrasts_array);
377
378   csd->c = 1;
379
380   push_new_store (csd->temp_contrasts, csd);
381
382   response = psppire_dialog_run (PSPPIRE_DIALOG (csd->contrasts_dialog));
383
384   if ( response == PSPPIRE_RESPONSE_CONTINUE )
385     {
386       ow->contrasts_array = clone_contrasts_array (csd->temp_contrasts);
387     }
388
389   /* Destroy the temp contrasts here */
390
391 }
392
393
394 static void
395 push_new_store (GArray *contrast_stack, struct contrasts_subdialog *csd)
396 {
397   GtkListStore *ls = gtk_list_store_new (1, G_TYPE_DOUBLE);
398
399   g_array_append_val (contrast_stack, ls);
400
401   g_signal_connect_swapped (ls, "row-deleted",
402                             G_CALLBACK (list_store_changed), csd);
403
404   g_signal_connect_swapped (ls, "row-changed",
405                             G_CALLBACK (list_store_changed), csd);
406
407   list_store_changed (csd);
408 }
409
410 static void
411 next (GtkWidget *widget, gpointer data)
412 {
413   struct contrasts_subdialog *csd = data;
414
415   if (csd->c >= csd->temp_contrasts->len)
416     push_new_store (csd->temp_contrasts, csd);
417
418   csd->c++;
419
420   list_store_changed (csd);
421 }
422
423
424 static void
425 prev (GtkWidget *widget, gpointer data)
426 {
427   struct contrasts_subdialog *csd = data;
428
429   if ( csd->c > 0 )
430     --csd->c;
431
432   list_store_changed (csd);
433 }
434
435