Merge 'master' into 'gtk3'.
[pspp] / src / ui / gui / page-sheet-spec.c
1 /* PSPPIRE - a graphical user interface for PSPP.
2    Copyright (C) 2013  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 "page-sheet-spec.h"
20
21 #include "ui/gui/text-data-import-dialog.h"
22
23 #include <errno.h>
24 #include <fcntl.h>
25 #include <gtk/gtk.h>
26 #include <limits.h>
27 #include <stdlib.h>
28 #include <sys/stat.h>
29
30 #include "data/data-in.h"
31 #include "data/data-out.h"
32 #include "data/format-guesser.h"
33 #include "data/value-labels.h"
34 #include "data/gnumeric-reader.h"
35 #include "data/ods-reader.h"
36 #include "data/spreadsheet-reader.h"
37 #include "language/data-io/data-parser.h"
38 #include "language/lexer/lexer.h"
39 #include "libpspp/assertion.h"
40 #include "libpspp/i18n.h"
41 #include "libpspp/line-reader.h"
42 #include "libpspp/message.h"
43 #include "ui/gui/dialog-common.h"
44 #include "ui/gui/executor.h"
45 #include "ui/gui/helper.h"
46 #include "ui/gui/builder-wrapper.h"
47 #include "ui/gui/psppire-data-window.h"
48 #include "ui/gui/psppire-dialog.h"
49 #include "ui/gui/psppire-encoding-selector.h"
50 #include "ui/gui/psppire-empty-list-store.h"
51 #include "ui/gui/psppire-var-sheet.h"
52 #include "ui/gui/psppire-scanf.h"
53 #include "ui/syntax-gen.h"
54
55 #include "ui/gui/psppire-spreadsheet-model.h"
56
57 #include <data/casereader.h>
58
59 #include "gl/intprops.h"
60 #include "gl/xalloc.h"
61
62 #include "gettext.h"
63 #define _(msgid) gettext (msgid)
64 #define N_(msgid) msgid
65
66 struct import_assistant;
67
68 /* The "sheet-spec" page of the assistant. */
69
70 /* The sheet_spec page of the assistant (only relevant for spreadsheet imports). */
71 struct sheet_spec_page
72   {
73     GtkWidget *page;
74     struct casereader *reader;
75     struct dictionary *dict;
76     
77     struct spreadsheet_read_options opts;
78   };
79
80
81 char *
82 sheet_spec_gen_syntax (const struct import_assistant *ia)
83 {
84   const struct sheet_spec_page *ssp = ia->sheet_spec;
85   GtkBuilder *builder = ia->asst.builder;
86   GtkWidget *range_entry = get_widget_assert (builder, "cell-range-entry");
87   const gchar *range = gtk_entry_get_text (GTK_ENTRY (range_entry));
88
89   struct string s = DS_EMPTY_INITIALIZER;
90
91   syntax_gen_pspp (&s,
92                    "GET DATA"
93                    "\n  /TYPE=%ss"
94                    "\n  /FILE=%sq"
95                    "\n  /SHEET=index %d"
96                    "\n  /READNAMES=%ss",
97                    (ia->spreadsheet->type == SPREADSHEET_GNUMERIC) ? "GNM" : "ODS",
98                    ia->file.file_name,                   
99                    ssp->opts.sheet_index,
100                    ssp->opts.read_names ? "ON" : "OFF");
101
102
103   if (range && 0 != strcmp ("", range))
104     {
105       syntax_gen_pspp (&s,
106                        "\n  /CELLRANGE=RANGE %sq", range);
107     }
108   else
109     {
110       syntax_gen_pspp (&s,
111                        "\n  /CELLRANGE=FULL");
112     }
113
114
115   syntax_gen_pspp (&s, ".");
116
117   
118   return ds_cstr (&s);
119 }
120
121
122 static void 
123 on_sheet_combo_changed (GtkComboBox *cb, struct import_assistant *ia)
124 {
125   GtkTreeIter iter;
126   gchar *range = NULL;
127   GtkTreeModel *model = gtk_combo_box_get_model (cb);
128   GtkBuilder *builder = ia->asst.builder;
129   GtkWidget *range_entry = get_widget_assert (builder, "cell-range-entry");
130
131   gtk_combo_box_get_active_iter (cb, &iter);
132   gtk_tree_model_get (model, &iter, PSPPIRE_SPREADSHEET_MODEL_COL_RANGE, &range, -1);
133   gtk_entry_set_text (GTK_ENTRY (range_entry), range ?  range : "");
134   g_free (range);
135 }
136
137 /* Initializes IA's sheet_spec substructure. */
138 struct sheet_spec_page *
139 sheet_spec_page_create (struct import_assistant *ia)
140 {
141   GtkBuilder *builder = ia->asst.builder;
142   struct sheet_spec_page *p = xzalloc (sizeof (*p));
143
144   GtkWidget *combo_box = get_widget_assert (builder, "sheet-entry");
145   GtkCellRenderer *renderer = gtk_cell_renderer_text_new ();
146   gtk_cell_layout_pack_start (GTK_CELL_LAYOUT (combo_box), renderer, TRUE);
147   gtk_cell_layout_set_attributes (GTK_CELL_LAYOUT (combo_box), renderer,
148                                   "text", 0,
149                                   NULL);
150
151   g_signal_connect (combo_box, "changed", G_CALLBACK (on_sheet_combo_changed), ia);
152
153   add_page_to_assistant (ia, get_widget_assert (builder, "Sheet"),
154                          GTK_ASSISTANT_PAGE_INTRO);
155
156   return p;
157 }
158
159 /* Prepares IA's sheet_spec page. */
160 void
161 prepare_sheet_spec_page (struct import_assistant *ia)
162 {
163   GtkBuilder *builder = ia->asst.builder;
164   GtkWidget *sheet_entry = get_widget_assert (builder, "sheet-entry");
165   GtkWidget *readnames_checkbox = get_widget_assert (builder, "readnames-checkbox");
166
167   gtk_combo_box_set_model (GTK_COMBO_BOX (sheet_entry), 
168                            psppire_spreadsheet_model_new (ia->spreadsheet));
169
170   gtk_combo_box_set_active (GTK_COMBO_BOX (sheet_entry), 0);
171
172   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (readnames_checkbox), FALSE);
173 }
174
175
176 /* Resets IA's sheet_spec page to its initial state. */
177 void
178 reset_sheet_spec_page (struct import_assistant *ia)
179 {
180   GtkBuilder *builder = ia->asst.builder;
181   GtkWidget *sheet_entry = get_widget_assert (builder, "sheet-entry");
182   GtkWidget *readnames_checkbox = get_widget_assert (builder, "readnames-checkbox");
183
184   gtk_combo_box_set_active (GTK_COMBO_BOX (sheet_entry), 0);
185
186   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (readnames_checkbox), FALSE);
187 }
188
189 /* Called when the Forward button is clicked, 
190    but before displaying the new page.
191 */
192 void
193 post_sheet_spec_page (struct import_assistant *ia)
194 {
195   int row_start = -1;
196   int row_stop = -1;
197   int col_start = -1;
198   int col_stop = -1;
199
200   GtkBuilder *builder = ia->asst.builder;
201
202   struct sheet_spec_page *ssp = ia->sheet_spec;
203   struct casereader *creader = NULL;
204   struct dictionary *dict = NULL;
205
206   GtkWidget *readnames_checkbox = get_widget_assert (builder, "readnames-checkbox");
207   GtkWidget *range_entry = get_widget_assert (builder, "cell-range-entry");
208   const gchar *range = gtk_entry_get_text (GTK_ENTRY (range_entry));
209   GtkWidget *combo_box = get_widget_assert (builder, "sheet-entry");
210
211   gint num = gtk_combo_box_get_active (GTK_COMBO_BOX (combo_box));
212   
213   ssp->opts.sheet_name = NULL;
214   ssp->opts.cell_range = NULL;
215   ssp->opts.sheet_index = num + 1;
216
217   if ( convert_cell_ref (range, &col_start, &row_start, &col_stop, &row_stop))
218     {
219       ssp->opts.cell_range = g_strdup (range);
220     }
221
222   ssp->opts.read_names = gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (readnames_checkbox));
223   ssp->opts.asw = -1;
224
225   switch (ia->spreadsheet->type)
226     {
227     case SPREADSHEET_ODS:
228     case SPREADSHEET_GNUMERIC:
229       {
230         creader = spreadsheet_make_reader (ia->spreadsheet, &ssp->opts);
231         dict = ia->spreadsheet->dict;
232       }
233       break;
234     default:
235       g_assert_not_reached ();
236       break;
237     }
238
239   ssp->dict = dict;
240   ssp->reader = creader;
241
242   if (creader && dict)
243     {
244       update_assistant (ia);
245     }
246   else
247     {
248       GtkWidget * dialog = gtk_message_dialog_new (NULL,
249                               GTK_DIALOG_MODAL,
250                               GTK_MESSAGE_ERROR,
251                               GTK_BUTTONS_CLOSE,
252                               _("An error occurred reading the spreadsheet file."));
253
254       gtk_dialog_run (GTK_DIALOG (dialog));
255       gtk_widget_destroy (dialog);
256     }
257 }
258
259
260 /*
261   Update IA according to the contents of DICT and CREADER.
262   CREADER will be destroyed by this function.
263 */
264 void 
265 update_assistant (struct import_assistant *ia)
266 {
267   struct sheet_spec_page *ssp = ia->sheet_spec;
268   int rows = 0;
269
270   if (ssp->dict)
271     {
272       struct ccase *c;
273       int col;
274
275       ia->column_cnt = dict_get_var_cnt (ssp->dict);
276       ia->columns = xcalloc (ia->column_cnt, sizeof (*ia->columns));
277       for (col = 0; col < ia->column_cnt ; ++col)
278         {
279           const struct variable *var = dict_get_var (ssp->dict, col);
280           ia->columns[col].name = xstrdup (var_get_name (var));
281           ia->columns[col].contents = NULL;
282         }
283
284       for (; (c = casereader_read (ssp->reader)) != NULL; case_unref (c))
285         {
286           rows++;
287           for (col = 0; col < ia->column_cnt ; ++col)
288             {
289               char *ss;
290               const struct variable *var = dict_get_var (ssp->dict, col);
291               
292               ia->columns[col].contents = xrealloc (ia->columns[col].contents,
293                                                     sizeof (struct substring) * rows);
294               
295               ss = data_out (case_data (c, var), dict_get_encoding (ssp->dict), 
296                              var_get_print_format (var));
297               
298               ia->columns[col].contents[rows - 1] = ss_cstr (ss);
299             }
300           
301           if (rows > MAX_PREVIEW_LINES)
302             {
303               case_unref (c);
304               break;
305             }
306         }
307     }
308   
309   ia->file.line_cnt = rows;
310 }