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 (GObject *o, gpointer data)
 226 {
 227   gint response;
 228   struct regression_dialog rd;
 229
 230   GtkBuilder *xml = builder_new ("regression.ui");
 231   PsppireDataWindow *de = PSPPIRE_DATA_WINDOW (data);
 232   PsppireVarStore *vs;
 233
 234   GtkWidget *dialog = get_widget_assert   (xml, "regression-dialog");
 235   GtkWidget *source = get_widget_assert   (xml, "dict-view");
 236   GtkWidget *dest_dep =   get_widget_assert   (xml, "dep-view");
 237   GtkWidget *dest_indep =   get_widget_assert   (xml, "indep-view");
 238   GtkWidget *stat_button = get_widget_assert (xml, "stat-button");
 239   GtkWidget *save_button = get_widget_assert (xml, "save-button");
 240
 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
 262   rd.save_dialog = get_widget_assert (xml, "save-dialog");
 263   rd.pred_button = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "pred-button"));
 264   rd.resid_button = GTK_TOGGLE_BUTTON (get_widget_assert (xml, "resid-button"));
 265   rd.stat_dialog = get_widget_assert (xml, "statistics-dialog");
 266
 267   rd.stat = gtk_tree_view_get_model (GTK_TREE_VIEW (rd.stat_view));
 268   rd.current_opts.pred = FALSE;
 269   rd.current_opts.resid = FALSE;
 270
 271   gtk_window_set_transient_for (GTK_WINDOW (rd.save_dialog), GTK_WINDOW (de));
 272   gtk_window_set_transient_for (GTK_WINDOW (rd.stat_dialog), GTK_WINDOW (de));
 273
 274   g_signal_connect (dialog, "refresh", G_CALLBACK (refresh),  &rd);
 275
 276   psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (dialog),
 277                                       dialog_state_valid, &rd);
 278
 279   g_signal_connect_swapped (stat_button, "clicked",
 280                             G_CALLBACK (on_statistics_clicked),  &rd);
 281   g_signal_connect_swapped (save_button, "clicked",
 282                             G_CALLBACK (on_save_clicked),  &rd);
 283
 284   response = psppire_dialog_run (PSPPIRE_DIALOG (dialog));
 285
 286
 287   switch (response)
 288     {
 289     case GTK_RESPONSE_OK:
 290       {
 291         gchar *syntax = generate_syntax (&rd);
 292
 293         struct getl_interface *sss = create_syntax_string_source (syntax);
 294         execute_syntax (sss);
 295
 296         g_free (syntax);
 297       }
 298       break;
 299     case PSPPIRE_RESPONSE_PASTE:
 300       {
 301         gchar *syntax = generate_syntax (&rd);
 302         paste_syntax_in_new_window (syntax);
 303
 304         g_free (syntax);
 305       }
 306       break;
 307     default:
 308       break;
 309     }
 310
 311   g_object_unref (xml);
 312 }