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