New module psppire-var-view
[pspp-builds.git] / 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 (GObject *o, gpointer data)
127 {
128   gint response;
129   PsppireDataWindow *de = PSPPIRE_DATA_WINDOW (data);
130
131   PsppireVarStore *vs = NULL;
132
133   struct oneway_anova_dialog ow;
134
135   GtkBuilder *builder = builder_new ("oneway.ui");
136
137   GtkWidget *dict_view =
138     get_widget_assert (builder, "oneway-anova-treeview1");
139
140   GtkWidget *selector2 =
141     get_widget_assert (builder, "oneway-anova-selector2");
142
143   GtkWidget *contrasts_button =
144     get_widget_assert (builder, "contrasts-button");
145
146
147   g_signal_connect_swapped (contrasts_button, "clicked",
148                     G_CALLBACK (run_contrasts_dialog), &ow);
149
150
151   ow.factor_entry = get_widget_assert (builder, "oneway-anova-entry");
152   ow.vars_treeview =
153     get_widget_assert (builder, "oneway-anova-treeview2");
154
155   ow.descriptives =
156     GTK_TOGGLE_BUTTON (get_widget_assert (builder, "checkbutton1"));
157
158   ow.homogeneity =
159     GTK_TOGGLE_BUTTON (get_widget_assert (builder, "checkbutton2"));
160
161   g_object_get (de->data_editor, "var-store", &vs, NULL);
162
163   g_object_get (vs, "dictionary", &ow.dict, NULL);
164
165   ow.dialog =
166     GTK_WINDOW (get_widget_assert (builder, "oneway-anova-dialog"));
167
168   gtk_window_set_transient_for (ow.dialog, GTK_WINDOW (de));
169
170   g_object_set (dict_view, "model", ow.dict, NULL);
171
172
173   psppire_selector_set_filter_func (PSPPIRE_SELECTOR (selector2),
174                                     is_currently_in_entry);
175
176
177   g_signal_connect_swapped (ow.dialog, "refresh", G_CALLBACK (refresh),  &ow);
178
179
180
181   psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (ow.dialog),
182                                       dialog_state_valid, &ow);
183
184
185   {
186     struct contrasts_subdialog *cd = &ow.contrasts;
187     GtkEntry *entry = GTK_ENTRY (get_widget_assert (builder, "entry1"));
188
189     cd->acr = PSPPIRE_ACR (get_widget_assert (builder, "psppire-acr1"));
190     cd->contrasts_dialog = get_widget_assert (builder, "contrasts-dialog");
191
192     cd->next = get_widget_assert (builder, "next-button");
193     cd->prev = get_widget_assert (builder, "prev-button");
194     cd->ctotal = get_widget_assert (builder, "entry2");
195
196     cd->stack_label = get_widget_assert (builder, "contrast-stack-label");
197
198     /* Contrasts */
199     ow.contrasts_array = g_array_new (FALSE, FALSE, sizeof (GtkListStore *));
200
201     g_signal_connect (cd->next, "clicked", G_CALLBACK (next), cd);
202     g_signal_connect (cd->prev, "clicked", G_CALLBACK (prev), cd);
203
204     psppire_acr_set_entry (cd->acr, entry);
205
206     gtk_window_set_transient_for (GTK_WINDOW (cd->contrasts_dialog),
207                                   GTK_WINDOW (de));
208   }
209
210   response = psppire_dialog_run (PSPPIRE_DIALOG (ow.dialog));
211
212   switch (response)
213     {
214     case GTK_RESPONSE_OK:
215       {
216         gchar *syntax = generate_syntax (&ow);
217
218         struct getl_interface *sss = create_syntax_string_source (syntax);
219         execute_syntax (sss);
220
221         g_free (syntax);
222       }
223       break;
224     case PSPPIRE_RESPONSE_PASTE:
225       {
226         gchar *syntax = generate_syntax (&ow);
227         paste_syntax_in_new_window (syntax);
228
229         g_free (syntax);
230       }
231       break;
232     default:
233       break;
234     }
235
236   g_array_free (ow.contrasts_array, FALSE);
237
238   g_object_unref (builder);
239 }
240
241
242 static gchar * generate_syntax (const struct oneway_anova_dialog *ow)
243 {
244   gchar *text;
245   gint i;
246   gboolean descriptives = gtk_toggle_button_get_active (ow->descriptives);
247   gboolean homogeneity = gtk_toggle_button_get_active (ow->homogeneity);
248
249   GString *str = g_string_new ("ONEWAY /VARIABLES=");
250
251   psppire_var_view_append_names (PSPPIRE_VAR_VIEW (ow->vars_treeview), 0, str);
252
253   g_string_append (str, " BY ");
254
255   g_string_append (str, gtk_entry_get_text (GTK_ENTRY (ow->factor_entry)));
256
257   if (descriptives || homogeneity )
258     {
259       g_string_append (str, "\n\t/STATISTICS=");
260       if (descriptives)
261         g_string_append (str, "DESCRIPTIVES ");
262       if (homogeneity)
263         g_string_append (str, "HOMOGENEITY ");
264     }
265
266   for (i = 0 ; i < ow->contrasts_array->len ; ++i )
267     {
268       GtkListStore *ls = g_array_index (ow->contrasts_array, GtkListStore*, i);
269       GtkTreeIter iter;
270       gboolean ok;
271
272       g_string_append (str, "\n\t/CONTRAST=");
273
274       for (ok = gtk_tree_model_get_iter_first (GTK_TREE_MODEL(ls),
275                                                &iter);
276            ok;
277            ok = gtk_tree_model_iter_next (GTK_TREE_MODEL (ls), &iter))
278         {
279           gdouble v;
280
281           gtk_tree_model_get (GTK_TREE_MODEL (ls), &iter, 0, &v, -1);
282
283           g_string_append_printf (str, " %g", v);
284         }
285     }
286
287   g_string_append (str, ".\n");
288
289   text = str->str;
290   g_string_free (str, FALSE);
291
292   return text;
293 }
294
295
296 \f
297 /* Contrasts stuff */
298
299
300
301
302 /* Callback for when the list store currently associated with the
303    treeview has changed.  It sets the widgets of the subdialog
304    to reflect the store's new state.
305 */
306 static void
307 list_store_changed (struct contrasts_subdialog *csd)
308 {
309   gboolean ok;
310   gdouble total = 0.0;
311   GtkTreeIter iter;
312   GtkTreeModel *ls = NULL;
313   gchar *text =
314     g_strdup_printf (_("Contrast %d of %d"),
315                      csd->c, csd->temp_contrasts->len);
316
317   gtk_label_set_label (GTK_LABEL (csd->stack_label), text);
318
319   g_free (text);
320
321   gtk_widget_set_sensitive (csd->prev, csd->c > 1);
322
323   if ( csd->c > 0 )
324     ls = g_array_index (csd->temp_contrasts, GtkTreeModel*, csd->c - 1);
325
326   psppire_acr_set_model (csd->acr, GTK_LIST_STORE (ls));
327
328   /* Sensitive iff the liststore has two items or more */
329   gtk_widget_set_sensitive (csd->next,
330                             gtk_tree_model_iter_nth_child
331                             (ls, &iter,  NULL, 1));
332
333   for (ok = gtk_tree_model_get_iter_first (ls, &iter);
334        ok;
335        ok = gtk_tree_model_iter_next (ls, &iter)
336        )
337     {
338       gdouble v;
339       gtk_tree_model_get (ls, &iter, 0, &v, -1);
340       total += v;
341     }
342
343   text = g_strdup_printf ("%g", total);
344
345   gtk_entry_set_text (GTK_ENTRY (csd->ctotal), text);
346
347   g_free (text);
348 }
349
350
351
352 /* Copy the contrasts array into the local array */
353 static GArray *
354 clone_contrasts_array (GArray *src_array)
355 {
356   gint i;
357
358   GArray *dest_array =
359     g_array_sized_new (FALSE, FALSE, sizeof (GtkListStore *),
360                        src_array->len);
361
362   for (i = 0 ; i < src_array->len ; ++i )
363     {
364
365       GtkTreeIter src_iter;
366       GtkListStore *src = g_array_index (src_array, GtkListStore*, i);
367       GtkListStore *dest;
368
369       /* Refuse to copy empty stores */
370       if (! gtk_tree_model_get_iter_first (GTK_TREE_MODEL (src),
371                                            &src_iter))
372         continue;
373
374       dest = clone_list_store (src);
375
376       g_array_append_val (dest_array, dest);
377     }
378
379   return dest_array;
380 }
381
382
383 static void
384 run_contrasts_dialog (struct oneway_anova_dialog *ow)
385 {
386   struct contrasts_subdialog *csd = &ow->contrasts;
387   gint response;
388
389   csd->temp_contrasts = clone_contrasts_array (ow->contrasts_array);
390
391   csd->c = 1;
392
393   push_new_store (csd->temp_contrasts, csd);
394
395   response = psppire_dialog_run (PSPPIRE_DIALOG (csd->contrasts_dialog));
396
397   if ( response == PSPPIRE_RESPONSE_CONTINUE )
398     {
399       ow->contrasts_array = clone_contrasts_array (csd->temp_contrasts);
400     }
401
402   /* Destroy the temp contrasts here */
403
404 }
405
406
407 static void
408 push_new_store (GArray *contrast_stack, struct contrasts_subdialog *csd)
409 {
410   GtkListStore *ls = gtk_list_store_new (1, G_TYPE_DOUBLE);
411
412   g_array_append_val (contrast_stack, ls);
413
414   g_signal_connect_swapped (ls, "row-deleted",
415                             G_CALLBACK (list_store_changed), csd);
416
417   g_signal_connect_swapped (ls, "row-changed",
418                             G_CALLBACK (list_store_changed), csd);
419
420   list_store_changed (csd);
421 }
422
423 static void
424 next (GtkWidget *widget, gpointer data)
425 {
426   struct contrasts_subdialog *csd = data;
427
428   if (csd->c >= csd->temp_contrasts->len)
429     push_new_store (csd->temp_contrasts, csd);
430
431   csd->c++;
432
433   list_store_changed (csd);
434 }
435
436
437 static void
438 prev (GtkWidget *widget, gpointer data)
439 {
440   struct contrasts_subdialog *csd = data;
441
442   if ( csd->c > 0 )
443     --csd->c;
444
445   list_store_changed (csd);
446 }
447
448