1 /* PSPPIRE - a graphical user interface for PSPP.
2 Copyright (C) 2008, 2010, 2011, 2012 Free Software Foundation
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.
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.
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/>. */
19 #include "checkbox-treeview.h"
20 #include "regression-dialog.h"
26 #include <ui/gui/psppire-data-window.h>
27 #include <ui/gui/dialog-common.h>
28 #include <ui/gui/dict-display.h>
29 #include <ui/gui/builder-wrapper.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>
37 #define _(msgid) gettext (msgid)
38 #define N_(msgid) msgid
41 #define REGRESSION_STATS \
42 RG (COEFF, N_("Coeff")) \
44 RG (ANOVA, N_("Anova")) \
48 #define RG(NAME, LABEL) RG_##NAME,
56 #define RG(NAME, LABEL) B_RG_##NAME = 1u << RG_##NAME,
59 B_RG_STATS_ALL = (1u << N_REGRESSION_STATS) - 1,
60 B_RG_STATS_DEFAULT = B_RG_ANOVA | B_RG_COEFF | B_RG_R
63 static const struct checkbox_entry_item stats[] =
65 #define RG(NAME, LABEL) {#NAME, LABEL},
75 struct regression_dialog
77 GtkTreeView *dep_vars;
78 GtkTreeView *indep_vars;
81 GtkToggleButton *resid_button;
82 GtkToggleButton *pred_button;
84 GtkWidget *stat_dialog;
85 GtkWidget *save_dialog;
89 struct save_options current_opts;
93 refresh (PsppireDialog *dialog, struct regression_dialog *rd)
95 GtkTreeModel *liststore = gtk_tree_view_get_model (rd->dep_vars);
96 gtk_list_store_clear (GTK_LIST_STORE (liststore));
98 liststore = gtk_tree_view_get_model (rd->indep_vars);
99 gtk_list_store_clear (GTK_LIST_STORE (liststore));
103 on_statistics_clicked (struct regression_dialog *rd)
105 GtkListStore *liststore;
108 liststore = clone_list_store (GTK_LIST_STORE (rd->stat));
110 ret = psppire_dialog_run (PSPPIRE_DIALOG (rd->stat_dialog));
112 if ( ret != PSPPIRE_RESPONSE_CONTINUE )
114 gtk_tree_view_set_model (GTK_TREE_VIEW (rd->stat_view) , GTK_TREE_MODEL (liststore));
115 rd->stat = GTK_TREE_MODEL (liststore);
117 g_object_unref (liststore);
121 on_save_clicked (struct regression_dialog *rd)
124 if (rd->current_opts.pred)
126 gtk_toggle_button_set_active (rd->pred_button, TRUE);
128 if (rd->current_opts.resid)
130 gtk_toggle_button_set_active (rd->resid_button, TRUE);
133 ret = psppire_dialog_run (PSPPIRE_DIALOG (rd->save_dialog));
135 if ( ret == PSPPIRE_RESPONSE_CONTINUE )
137 rd->current_opts.pred = (gtk_toggle_button_get_active (rd->pred_button) == TRUE)
139 rd->current_opts.resid = (gtk_toggle_button_get_active (rd->resid_button) == TRUE)
145 generate_syntax (const struct regression_dialog *rd)
154 GString *string = g_string_new ("REGRESSION");
156 g_string_append (string, "\n\t/VARIABLES=");
157 psppire_var_view_append_names (PSPPIRE_VAR_VIEW (rd->indep_vars), 0, string);
158 g_string_append (string, "\n\t/DEPENDENT=\t");
159 psppire_var_view_append_names (PSPPIRE_VAR_VIEW (rd->dep_vars), 0, string);
162 for (i = 0, ok = gtk_tree_model_get_iter_first (rd->stat, &iter); ok;
163 i++, ok = gtk_tree_model_iter_next (rd->stat, &iter))
166 gtk_tree_model_get (rd->stat, &iter,
167 CHECKBOX_COLUMN_SELECTED, &toggled, -1);
171 selected &= ~(1u << i);
176 g_string_append (string, "\n\t/STATISTICS=");
178 for (i = 0; i < N_REGRESSION_STATS; i++)
179 if (selected & (1u << i))
182 g_string_append (string, " ");
183 g_string_append (string, stats[i].name);
186 if (rd->current_opts.pred || rd->current_opts.resid)
188 g_string_append (string, "\n\t/SAVE=");
189 if (rd->current_opts.pred)
190 g_string_append (string, " PRED");
191 if (rd->current_opts.resid)
192 g_string_append (string, " RESID");
194 g_string_append (string, ".\n");
198 g_string_free (string, FALSE);
203 /* Dialog is valid iff at least one dependent and one independent variable have
206 dialog_state_valid (gpointer data)
208 struct regression_dialog *rd = data;
210 GtkTreeModel *dep_vars = gtk_tree_view_get_model (rd->dep_vars);
211 GtkTreeModel *indep_vars = gtk_tree_view_get_model (rd->indep_vars);
215 return (gtk_tree_model_get_iter_first (dep_vars, ¬used)
216 && gtk_tree_model_get_iter_first (indep_vars, ¬used));
219 /* Pops up the Regression dialog box */
221 regression_dialog (PsppireDataWindow *de)
224 struct regression_dialog rd;
226 GtkBuilder *xml = builder_new ("regression.ui");
229 GtkWidget *dialog = get_widget_assert (xml, "regression-dialog");
230 GtkWidget *source = get_widget_assert (xml, "dict-view");
231 GtkWidget *dest_dep = get_widget_assert (xml, "dep-view");
232 GtkWidget *dest_indep = get_widget_assert (xml, "indep-view");
233 GtkWidget *stat_button = get_widget_assert (xml, "stat-button");
234 GtkWidget *save_button = get_widget_assert (xml, "save-button");
236 GtkWidget *dep_selector = get_widget_assert (xml, "dep-selector");
238 rd.stat_view = get_widget_assert (xml, "stat-view");
240 g_object_get (de->data_editor, "var-store", &vs, NULL);
243 put_checkbox_items_in_treeview (GTK_TREE_VIEW(rd.stat_view),
249 gtk_window_set_transient_for (GTK_WINDOW (dialog), GTK_WINDOW (de));
251 g_object_get (vs, "dictionary", &rd.dict, NULL);
252 g_object_set (source, "model", rd.dict, NULL);
254 rd.dep_vars = GTK_TREE_VIEW (dest_dep);
255 rd.indep_vars = GTK_TREE_VIEW (dest_indep);
257 psppire_selector_set_allow (PSPPIRE_SELECTOR (dep_selector), numeric_only);
259 rd.save_dialog = get_widget_assert (xml, "save-dialog");
260 rd.pred_button = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "pred-button"));
261 rd.resid_button = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "resid-button"));
262 rd.stat_dialog = get_widget_assert (xml, "statistics-dialog");
264 rd.stat = gtk_tree_view_get_model (GTK_TREE_VIEW (rd.stat_view));
265 rd.current_opts.pred = FALSE;
266 rd.current_opts.resid = FALSE;
268 gtk_window_set_transient_for (GTK_WINDOW (rd.save_dialog), GTK_WINDOW (de));
269 gtk_window_set_transient_for (GTK_WINDOW (rd.stat_dialog), GTK_WINDOW (de));
271 g_signal_connect (dialog, "refresh", G_CALLBACK (refresh), &rd);
273 psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (dialog),
274 dialog_state_valid, &rd);
276 g_signal_connect_swapped (stat_button, "clicked",
277 G_CALLBACK (on_statistics_clicked), &rd);
278 g_signal_connect_swapped (save_button, "clicked",
279 G_CALLBACK (on_save_clicked), &rd);
281 response = psppire_dialog_run (PSPPIRE_DIALOG (dialog));
286 case GTK_RESPONSE_OK:
287 g_free (execute_syntax_string (de, generate_syntax (&rd)));
289 case PSPPIRE_RESPONSE_PASTE:
290 g_free (paste_syntax_to_window (generate_syntax (&rd)));
296 g_object_unref (xml);