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