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