Categoricals.c Create entries for all groups.
[pspp] / src / ui / gui / regression-dialog.c
1 /* PSPPIRE - a graphical user interface for PSPP.
2    Copyright (C) 2008  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 "regression-dialog.h"
21 #include "executor.h"
22
23 #include <gtk/gtk.h>
24 #include <stdlib.h>
25
26 #include <language/syntax-string-source.h>
27 #include <ui/gui/psppire-data-window.h>
28 #include <ui/gui/dialog-common.h>
29 #include <ui/gui/dict-display.h>
30 #include <ui/gui/helper.h>
31 #include <ui/gui/psppire-dialog.h>
32 #include <ui/gui/psppire-var-store.h>
33 #include <ui/gui/psppire-var-view.h>
34
35
36 #include "gettext.h"
37 #define _(msgid) gettext (msgid)
38 #define N_(msgid) msgid
39
40
41 #define REGRESSION_STATS                       \
42   RG (COEFF, N_("Coeff"))                         \
43   RG (R, N_("R"))         \
44   RG (ANOVA, N_("Anova"))                   \
45   RG (BCOV, N_("Bcov"))
46 enum
47   {
48 #define RG(NAME, LABEL) RG_##NAME,
49     REGRESSION_STATS
50 #undef RG
51     N_REGRESSION_STATS
52   };
53
54 enum
55   {
56 #define RG(NAME, LABEL) B_RG_##NAME = 1u << RG_##NAME,
57     REGRESSION_STATS
58 #undef RG
59     B_RG_STATS_ALL = (1u << N_REGRESSION_STATS) - 1,
60     B_RG_STATS_DEFAULT = B_RG_ANOVA | B_RG_COEFF | B_RG_R
61   };
62
63 static const struct checkbox_entry_item stats[] =
64   {
65 #define RG(NAME, LABEL) {#NAME, LABEL},
66     REGRESSION_STATS
67 #undef RG
68   };
69
70 struct save_options
71 {
72   gboolean pred;
73   gboolean resid;
74 };
75 struct regression_dialog
76 {
77   GtkTreeView *dep_vars;
78   GtkTreeView *indep_vars;
79   PsppireDict *dict;
80
81   GtkToggleButton *resid_button;
82   GtkToggleButton *pred_button;
83
84   GtkWidget *stat_dialog;
85   GtkWidget *save_dialog;
86
87   GtkWidget *stat_view;
88   GtkTreeModel *stat;
89   struct save_options current_opts;
90 };
91
92 static void
93 refresh (PsppireDialog *dialog, struct regression_dialog *rd)
94 {
95   GtkTreeModel *liststore = gtk_tree_view_get_model (rd->dep_vars);
96   gtk_list_store_clear (GTK_LIST_STORE (liststore));
97
98   liststore = gtk_tree_view_get_model (rd->indep_vars);
99   gtk_list_store_clear (GTK_LIST_STORE (liststore));
100 }
101
102 static void
103 on_statistics_clicked (struct regression_dialog *rd)
104 {
105   GtkListStore *liststore;
106   int ret;
107
108   liststore = clone_list_store (GTK_LIST_STORE (rd->stat));
109
110   ret = psppire_dialog_run (PSPPIRE_DIALOG (rd->stat_dialog));
111
112   if ( ret == PSPPIRE_RESPONSE_CONTINUE )
113     {
114       g_object_unref (liststore);
115     }
116   else
117     {
118       g_object_unref (rd->stat);
119       gtk_tree_view_set_model (GTK_TREE_VIEW (rd->stat_view) , GTK_TREE_MODEL (liststore));
120       rd->stat = GTK_TREE_MODEL (liststore);
121     }
122 }
123
124 static void
125 on_save_clicked (struct regression_dialog *rd)
126 {
127   int ret;
128   if (rd->current_opts.pred)
129     {
130       gtk_toggle_button_set_active (rd->pred_button, TRUE);
131     }
132   if (rd->current_opts.resid)
133     {
134       gtk_toggle_button_set_active (rd->resid_button, TRUE);
135     }
136
137   ret = psppire_dialog_run (PSPPIRE_DIALOG (rd->save_dialog));
138
139   if ( ret == PSPPIRE_RESPONSE_CONTINUE )
140     {
141       rd->current_opts.pred = (gtk_toggle_button_get_active (rd->pred_button) == TRUE)
142         ? TRUE : FALSE;
143       rd->current_opts.resid = (gtk_toggle_button_get_active (rd->resid_button) == TRUE)
144         ? TRUE : FALSE;
145     }
146 }
147
148 static char *
149 generate_syntax (const struct regression_dialog *rd)
150 {
151   gint i;
152   int n;
153   guint selected;
154   GtkTreeIter iter;
155   gboolean ok;
156
157   gchar *text;
158   GString *string = g_string_new ("REGRESSION");
159
160   g_string_append (string, "\n\t/VARIABLES=");
161   psppire_var_view_append_names (PSPPIRE_VAR_VIEW (rd->indep_vars), 0, string);
162   g_string_append (string, "\n\t/DEPENDENT=\t");
163   psppire_var_view_append_names (PSPPIRE_VAR_VIEW (rd->dep_vars), 0, string);
164
165   selected = 0;
166   for (i = 0, ok = gtk_tree_model_get_iter_first (rd->stat, &iter); ok; 
167        i++, ok = gtk_tree_model_iter_next (rd->stat, &iter))
168     {
169       gboolean toggled;
170       gtk_tree_model_get (rd->stat, &iter,
171                           CHECKBOX_COLUMN_SELECTED, &toggled, -1); 
172       if (toggled) 
173         selected |= 1u << i; 
174       else 
175         selected &= ~(1u << i);
176     }
177
178   if (selected)
179     {
180       g_string_append (string, "\n\t/STATISTICS=");
181       n = 0;
182       for (i = 0; i < N_REGRESSION_STATS; i++)
183         if (selected & (1u << i))
184           {
185             if (n++)
186               g_string_append (string, " ");
187             g_string_append (string, stats[i].name);
188           }
189     }
190   if (rd->current_opts.pred || rd->current_opts.resid)
191     {
192       g_string_append (string, "\n\t/SAVE=");
193       if (rd->current_opts.pred)
194         g_string_append (string, " PRED");
195       if (rd->current_opts.resid)
196         g_string_append (string, " RESID");
197     }
198   g_string_append (string, ".\n");
199
200   text = string->str;
201
202   g_string_free (string, FALSE);
203
204   return text;
205 }
206
207 /* Dialog is valid iff at least one dependent and one independent variable have
208    been selected. */
209 static gboolean
210 dialog_state_valid (gpointer data)
211 {
212   struct regression_dialog *rd = data;
213
214   GtkTreeModel *dep_vars = gtk_tree_view_get_model (rd->dep_vars);
215   GtkTreeModel *indep_vars = gtk_tree_view_get_model (rd->indep_vars);
216
217   GtkTreeIter notused;
218
219   return (gtk_tree_model_get_iter_first (dep_vars, &notused)
220     && gtk_tree_model_get_iter_first (indep_vars, &notused));
221 }
222
223 /* Pops up the Regression dialog box */
224 void
225 regression_dialog (PsppireDataWindow *de)
226 {
227   gint response;
228   struct regression_dialog rd;
229
230   GtkBuilder *xml = builder_new ("regression.ui");
231   PsppireVarStore *vs;
232
233   GtkWidget *dialog = get_widget_assert   (xml, "regression-dialog");
234   GtkWidget *source = get_widget_assert   (xml, "dict-view");
235   GtkWidget *dest_dep =   get_widget_assert   (xml, "dep-view");
236   GtkWidget *dest_indep =   get_widget_assert   (xml, "indep-view");
237   GtkWidget *stat_button = get_widget_assert (xml, "stat-button");
238   GtkWidget *save_button = get_widget_assert (xml, "save-button");
239
240   GtkWidget *dep_selector = get_widget_assert (xml, "dep-selector");
241
242   rd.stat_view = get_widget_assert (xml, "stat-view");
243
244   g_object_get (de->data_editor, "var-store", &vs, NULL);
245
246
247   put_checkbox_items_in_treeview (GTK_TREE_VIEW(rd.stat_view),
248                                   B_RG_STATS_DEFAULT,
249                                   N_REGRESSION_STATS,
250                                   stats
251                                   );
252
253   gtk_window_set_transient_for (GTK_WINDOW (dialog), GTK_WINDOW (de));
254
255   g_object_get (vs, "dictionary", &rd.dict, NULL);
256   g_object_set (source, "model", rd.dict, NULL);
257
258   rd.dep_vars = GTK_TREE_VIEW (dest_dep);
259   rd.indep_vars = GTK_TREE_VIEW (dest_indep);
260
261   psppire_selector_set_allow (PSPPIRE_SELECTOR (dep_selector), numeric_only);
262
263   rd.save_dialog = get_widget_assert (xml, "save-dialog");
264   rd.pred_button = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "pred-button"));
265   rd.resid_button = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "resid-button"));
266   rd.stat_dialog = get_widget_assert (xml, "statistics-dialog");
267
268   rd.stat = gtk_tree_view_get_model (GTK_TREE_VIEW (rd.stat_view));
269   rd.current_opts.pred = FALSE;
270   rd.current_opts.resid = FALSE;
271
272   gtk_window_set_transient_for (GTK_WINDOW (rd.save_dialog), GTK_WINDOW (de));
273   gtk_window_set_transient_for (GTK_WINDOW (rd.stat_dialog), GTK_WINDOW (de));
274
275   g_signal_connect (dialog, "refresh", G_CALLBACK (refresh),  &rd);
276
277   psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (dialog),
278                                       dialog_state_valid, &rd);
279
280   g_signal_connect_swapped (stat_button, "clicked",
281                             G_CALLBACK (on_statistics_clicked),  &rd);
282   g_signal_connect_swapped (save_button, "clicked",
283                             G_CALLBACK (on_save_clicked),  &rd);
284
285   response = psppire_dialog_run (PSPPIRE_DIALOG (dialog));
286
287
288   switch (response)
289     {
290     case GTK_RESPONSE_OK:
291       {
292         gchar *syntax = generate_syntax (&rd);
293
294         struct getl_interface *sss = create_syntax_string_source (syntax);
295         execute_syntax (sss);
296
297         g_free (syntax);
298       }
299       break;
300     case PSPPIRE_RESPONSE_PASTE:
301       {
302         gchar *syntax = generate_syntax (&rd);
303         paste_syntax_to_window (syntax);
304
305         g_free (syntax);
306       }
307       break;
308     default:
309       break;
310     }
311
312   g_object_unref (xml);
313 }