CORRELATIONS: Fixed bug displaying non-sqaure correlation matrices
[pspp] / src / ui / gui / chi-square-dialog.c
1 /* PSPPIRE - a graphical user interface for PSPP.
2    Copyright (C) 2010, 2011, 2012  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 "chi-square-dialog.h"
20
21 #include "psppire-dialog.h"
22 #include "psppire-var-view.h"
23 #include "psppire-acr.h"
24 #include "dialog-common.h"
25
26 #include "builder-wrapper.h"
27 #include "executor.h"
28 #include "helper.h"
29
30 #include <gtk/gtk.h>
31
32 struct chisquare_dialog
33 {
34   PsppireDict *dict;
35   GtkWidget *var_view;
36
37   GtkWidget *button1;
38   GtkWidget *button2;
39
40   GtkWidget *range_button;
41   GtkWidget *value_lower;
42   GtkWidget *value_upper;
43
44   GtkWidget *values_button;
45
46   GtkListStore *expected_list;
47 };
48
49 static void
50 set_sensitivity (GtkToggleButton *button, GtkWidget *w)
51 {
52   gboolean state = gtk_toggle_button_get_active (button);
53   gtk_widget_set_sensitive (w, state);
54 }
55
56
57 static gboolean
58 dialog_state_valid (gpointer data)
59 {
60   struct chisquare_dialog *csd = data;
61
62   GtkTreeModel *vars =
63     gtk_tree_view_get_model (GTK_TREE_VIEW (csd->var_view));
64
65   GtkTreeIter notused;
66
67   if ( !gtk_tree_model_get_iter_first (vars, &notused) )
68     return FALSE;
69
70   return TRUE;
71 }
72
73
74 static void
75 refresh (struct chisquare_dialog *csd)
76 {
77   GtkTreeModel *liststore =
78     gtk_tree_view_get_model (GTK_TREE_VIEW (csd->var_view));
79
80   gtk_list_store_clear (GTK_LIST_STORE (liststore));
81
82   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (csd->button1), TRUE);
83
84   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (csd->button2), TRUE);
85 }
86
87
88
89 static char *
90 generate_syntax (const struct chisquare_dialog *scd)
91 {
92   gchar *text;
93   GString *string;
94
95
96   string = g_string_new ("NPAR TEST\n\t/CHISQUARE=");
97
98   psppire_var_view_append_names (PSPPIRE_VAR_VIEW (scd->var_view), 0, string);
99
100
101   if ( gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (scd->range_button)))
102     {
103       g_string_append (string, "(");
104       
105       g_string_append (string, 
106                        gtk_entry_get_text (GTK_ENTRY (scd->value_lower)));
107
108       g_string_append (string, ", ");
109
110       g_string_append (string,
111                        gtk_entry_get_text (GTK_ENTRY (scd->value_upper)));
112
113       g_string_append (string, ")");
114     }
115
116
117
118
119   if ( gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (scd->values_button)))
120     {
121       GtkListStore *ls = scd->expected_list;
122       GtkTreeIter iter;
123       gboolean ok;
124
125       g_string_append (string, "\n\t");
126       g_string_append (string, "/EXPECTED = ");
127
128       
129       for (ok = gtk_tree_model_get_iter_first (GTK_TREE_MODEL(ls),
130                                                &iter);
131            ok;
132            ok = gtk_tree_model_iter_next (GTK_TREE_MODEL (ls), &iter))
133         {
134           gdouble v;
135
136           gtk_tree_model_get (GTK_TREE_MODEL (ls), &iter, 0, &v, -1);
137
138           g_string_append_printf (string, " %g", v);
139         }
140
141
142
143     }
144
145   g_string_append (string, ".\n");
146
147   text = string->str;
148
149   g_string_free (string, FALSE);
150
151   return text;
152 }
153
154
155
156 /* Pops up the Chi-Square dialog box */
157 void
158 chisquare_dialog (PsppireDataWindow *dw)
159 {
160   gint response;
161
162   struct chisquare_dialog csd;
163
164   GtkBuilder *xml = builder_new ("chi-square.ui");
165   PsppireVarStore *vs;
166
167   GtkWidget *dialog = get_widget_assert   (xml, "chisquare-dialog");
168
169   GtkWidget *range_table = get_widget_assert   (xml, "range-table");
170
171
172
173   GtkWidget *values_acr = get_widget_assert   (xml, "psppire-acr1");
174   GtkWidget *expected_value_entry =
175     get_widget_assert   (xml, "expected-value-entry");
176
177
178   GtkWidget *dict_view = get_widget_assert   (xml, "dict-view");
179
180   csd.expected_list = gtk_list_store_new (1, G_TYPE_DOUBLE);
181
182   csd.button1 = get_widget_assert   (xml, "radiobutton1");
183   csd.button2 = get_widget_assert   (xml, "radiobutton3");
184   csd.var_view = get_widget_assert   (xml, "variables-treeview");
185
186   csd.range_button = get_widget_assert   (xml, "radiobutton4");
187   csd.value_lower = get_widget_assert   (xml, "entry1");
188   csd.value_upper = get_widget_assert   (xml, "entry2");
189
190   csd.values_button = get_widget_assert   (xml, "radiobutton2");
191
192   g_object_get (dw->data_editor, "var-store", &vs, NULL);
193
194   gtk_window_set_transient_for (GTK_WINDOW (dialog), GTK_WINDOW (dw));
195  
196
197   g_object_get (vs, "dictionary", &csd.dict, NULL);
198   g_object_set (dict_view,
199                 "model", csd.dict, 
200                 "predicate", var_is_numeric,
201                 NULL);
202
203
204   g_signal_connect (csd.range_button, "toggled", G_CALLBACK (set_sensitivity), 
205                     range_table);
206
207
208   g_signal_connect (csd.values_button, "toggled", G_CALLBACK (set_sensitivity), 
209                     values_acr);
210
211   g_signal_connect (csd.values_button, "toggled", G_CALLBACK (set_sensitivity), 
212                     expected_value_entry);
213
214
215   psppire_acr_set_entry (PSPPIRE_ACR (values_acr),
216                          GTK_ENTRY (expected_value_entry));
217
218   psppire_acr_set_model(PSPPIRE_ACR (values_acr), csd.expected_list);
219
220   g_signal_connect_swapped (dialog, "refresh", G_CALLBACK (refresh),  &csd);
221
222   psppire_dialog_set_valid_predicate (PSPPIRE_DIALOG (dialog),
223                                       dialog_state_valid, &csd);
224
225   response = psppire_dialog_run (PSPPIRE_DIALOG (dialog));
226
227
228   switch (response)
229     {
230     case GTK_RESPONSE_OK:
231       g_free (execute_syntax_string (dw, generate_syntax (&csd)));
232       break;
233     case PSPPIRE_RESPONSE_PASTE:
234       g_free (paste_syntax_to_window (generate_syntax (&csd)));
235       break;
236     default:
237       break;
238     }
239
240   g_object_unref (csd.expected_list);
241   g_object_unref (xml);
242 }