Import Assistant: Don't try to parse non-existant files
[pspp] / src / ui / gui / psppire-import-assistant.c
1 /* PSPPIRE - a graphical user interface for PSPP.
2    Copyright (C) 2015, 2016, 2017  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 <gtk/gtk.h>
20
21 #include "data/casereader.h"
22 #include "data/data-in.h"
23 #include "data/data-out.h"
24 #include "data/dictionary.h"
25 #include "data/format-guesser.h"
26 #include "data/format.h"
27 #include "data/gnumeric-reader.h"
28 #include "data/ods-reader.h"
29 #include "data/spreadsheet-reader.h"
30 #include "data/value-labels.h"
31 #include "data/casereader-provider.h"
32
33 #include "libpspp/i18n.h"
34 #include "libpspp/line-reader.h"
35 #include "libpspp/message.h"
36 #include "libpspp/hmap.h"
37 #include "libpspp/hash-functions.h"
38 #include "libpspp/str.h"
39
40 #include "builder-wrapper.h"
41
42 #include "psppire-data-sheet.h"
43 #include "psppire-data-store.h"
44 #include "psppire-dialog.h"
45 #include "psppire-delimited-text.h"
46 #include "psppire-dict.h"
47 #include "psppire-encoding-selector.h"
48 #include "psppire-import-assistant.h"
49 #include "psppire-scanf.h"
50 #include "psppire-spreadsheet-model.h"
51 #include "psppire-text-file.h"
52 #include "psppire-variable-sheet.h"
53
54 #include "ui/syntax-gen.h"
55
56 #include <gettext.h>
57 #define _(msgid) gettext (msgid)
58 #define N_(msgid) msgid
59
60 enum { MAX_LINE_LEN = 16384 }; /* Max length of an acceptable line. */
61
62
63 /* Chooses a name for each column on the separators page */
64 static void choose_column_names (PsppireImportAssistant *ia);
65
66 static void intro_page_create (PsppireImportAssistant *ia);
67 static void first_line_page_create (PsppireImportAssistant *ia);
68
69 static void separators_page_create (PsppireImportAssistant *ia);
70 static void formats_page_create (PsppireImportAssistant *ia);
71
72 static void psppire_import_assistant_init            (PsppireImportAssistant      *act);
73 static void psppire_import_assistant_class_init      (PsppireImportAssistantClass *class);
74
75 G_DEFINE_TYPE (PsppireImportAssistant, psppire_import_assistant, GTK_TYPE_ASSISTANT);
76
77
78 /* Properties */
79 enum
80   {
81     PROP_0,
82   };
83
84 static void
85 psppire_import_assistant_set_property (GObject         *object,
86                                        guint            prop_id,
87                                        const GValue    *value,
88                                        GParamSpec      *pspec)
89 {
90   //   PsppireImportAssistant *act = PSPPIRE_IMPORT_ASSISTANT (object);
91
92   switch (prop_id)
93     {
94     default:
95       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
96       break;
97     };
98 }
99
100
101 static void
102 psppire_import_assistant_get_property (GObject    *object,
103                                        guint            prop_id,
104                                        GValue          *value,
105                                        GParamSpec      *pspec)
106 {
107   //  PsppireImportAssistant *assistant = PSPPIRE_IMPORT_ASSISTANT (object);
108
109   switch (prop_id)
110     {
111     default:
112       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
113       break;
114     };
115 }
116
117 static GObjectClass * parent_class = NULL;
118
119
120 static void
121 psppire_import_assistant_finalize (GObject *object)
122 {
123   PsppireImportAssistant *ia = PSPPIRE_IMPORT_ASSISTANT (object);
124
125   if (ia->spreadsheet)
126     spreadsheet_unref (ia->spreadsheet);
127
128   ds_destroy (&ia->quotes);
129
130   g_object_unref (ia->builder);
131
132   ia->response = -1;
133   g_main_loop_unref (ia->main_loop);
134
135   if (G_OBJECT_CLASS (parent_class)->finalize)
136     G_OBJECT_CLASS (parent_class)->finalize (object);
137 }
138
139
140 static void
141 psppire_import_assistant_class_init (PsppireImportAssistantClass *class)
142 {
143   GObjectClass *object_class = G_OBJECT_CLASS (class);
144
145   parent_class = g_type_class_peek_parent (class);
146
147   object_class->set_property = psppire_import_assistant_set_property;
148   object_class->get_property = psppire_import_assistant_get_property;
149
150   object_class->finalize = psppire_import_assistant_finalize;
151 }
152
153
154 /* Causes the assistant to close, returning RESPONSE for
155    interpretation by text_data_import_assistant. */
156 static void
157 close_assistant (PsppireImportAssistant *ia, int response)
158 {
159   ia->response = response;
160   g_main_loop_quit (ia->main_loop);
161   gtk_widget_hide (GTK_WIDGET (ia));
162 }
163
164
165 /* Called when the Paste button on the last page of the assistant
166    is clicked. */
167 static void
168 on_paste (GtkButton *button, PsppireImportAssistant *ia)
169 {
170   close_assistant (ia, PSPPIRE_RESPONSE_PASTE);
171 }
172
173
174 /* Revises the contents of the fields tree view based on the
175    currently chosen set of separators. */
176 static void
177 revise_fields_preview (PsppireImportAssistant *ia)
178 {
179   choose_column_names (ia);
180 }
181
182
183 struct separator
184 {
185   const char *name;           /* Name (for use with get_widget_assert). */
186   gunichar c;                 /* Separator character. */
187 };
188
189 /* All the separators in the dialog box. */
190 static const struct separator separators[] =
191   {
192     {"space", ' '},
193     {"tab", '\t'},
194     {"bang", '!'},
195     {"colon", ':'},
196     {"comma", ','},
197     {"hyphen", '-'},
198     {"pipe", '|'},
199     {"semicolon", ';'},
200     {"slash", '/'},
201   };
202 #define SEPARATOR_CNT (sizeof separators / sizeof *separators)
203
204 struct separator_count_node
205 {
206   struct hmap_node node;
207   int occurance; /* The number of times the separator occurs in a line */
208   int quantity;  /* The number of lines with this occurance */
209 };
210
211
212 /* Picks the most likely separator and quote characters based on
213    IA's file data. */
214 static void
215 choose_likely_separators (PsppireImportAssistant *ia)
216 {
217   gint first_line = 0;
218   g_object_get (ia->delimiters_model, "first-line", &first_line, NULL);
219
220   gboolean valid;
221   GtkTreeIter iter;
222   int j;
223
224   struct hmap count_map[SEPARATOR_CNT];
225   for (j = 0; j < SEPARATOR_CNT; ++j)
226     hmap_init (count_map + j);
227
228   GtkTreePath *p = gtk_tree_path_new_from_indices (first_line, -1);
229
230   for (valid = gtk_tree_model_get_iter (GTK_TREE_MODEL (ia->text_file), &iter, p);
231        valid;
232        valid = gtk_tree_model_iter_next (GTK_TREE_MODEL (ia->text_file), &iter))
233     {
234       gchar *line_text = NULL;
235       gtk_tree_model_get (GTK_TREE_MODEL (ia->text_file), &iter, 1, &line_text, -1);
236
237       gint *counts = xzalloc (sizeof *counts * SEPARATOR_CNT);
238
239       struct substring cs = ss_cstr (line_text);
240       for (;
241            UINT32_MAX != ss_first_mb (cs);
242            ss_get_mb (&cs))
243         {
244           ucs4_t character = ss_first_mb (cs);
245
246           int s;
247           for (s = 0; s < SEPARATOR_CNT; ++s)
248             {
249               if (character == separators[s].c)
250                 counts[s]++;
251             }
252         }
253
254       int j;
255       for (j = 0; j < SEPARATOR_CNT; ++j)
256         {
257           if (counts[j] > 0)
258             {
259               struct separator_count_node *cn = NULL;
260               unsigned int hash = hash_int (counts[j], 0);
261               HMAP_FOR_EACH_WITH_HASH (cn, struct separator_count_node, node, hash, &count_map[j])
262                 {
263                   if (cn->occurance == counts[j])
264                     break;
265                 }
266
267               if (cn == NULL)
268                 {
269                   struct separator_count_node *new_cn = xzalloc (sizeof *new_cn);
270                   new_cn->occurance = counts[j];
271                   new_cn->quantity = 1;
272                   hmap_insert (&count_map[j], &new_cn->node, hash);
273                 }
274               else
275                 cn->quantity++;
276             }
277         }
278
279       free (line_text);
280       free (counts);
281     }
282   gtk_tree_path_free (p);
283
284   int most_frequent = -1;
285   int largest = 0;
286   for (j = 0; j < SEPARATOR_CNT; ++j)
287     {
288       struct separator_count_node *cn;
289       HMAP_FOR_EACH (cn, struct separator_count_node, node, &count_map[j])
290         {
291           if (largest < cn->quantity)
292             {
293               largest = cn->quantity;
294               most_frequent = j;
295             }
296         }
297       hmap_destroy (&count_map[j]);
298     }
299
300   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (get_widget_assert (ia->builder, separators[most_frequent].name)), TRUE);
301 }
302
303 static void
304 repopulate_delimiter_columns (PsppireImportAssistant *ia)
305 {
306   /* Remove all the columns */
307   while (gtk_tree_view_get_n_columns (GTK_TREE_VIEW (ia->fields_tree_view)) > 0)
308     {
309       GtkTreeViewColumn *tvc = gtk_tree_view_get_column (GTK_TREE_VIEW (ia->fields_tree_view), 0);
310       gtk_tree_view_remove_column (GTK_TREE_VIEW (ia->fields_tree_view), tvc);
311     }
312
313   gint n_fields = gtk_tree_model_get_n_columns (ia->delimiters_model);
314
315   /* ... and put them back again. */
316   gint f;
317   for (f = gtk_tree_view_get_n_columns (GTK_TREE_VIEW (ia->fields_tree_view));
318        f < n_fields; f++)
319     {
320       GtkCellRenderer *renderer = gtk_cell_renderer_text_new ();
321
322       const gchar *title = NULL;
323
324       if (f == 0)
325         title = _("line");
326       else
327         {
328           if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (ia->variable_names_cb)))
329             {
330               title =
331                 psppire_delimited_text_get_header_title
332                 (PSPPIRE_DELIMITED_TEXT (ia->delimiters_model), f - 1);
333             }
334           if (title == NULL)
335             title = _("var");
336         }
337
338       GtkTreeViewColumn *column =
339         gtk_tree_view_column_new_with_attributes (title,
340                                                   renderer,
341                                                   "text", f,
342                                                   NULL);
343       g_object_set (column,
344                     "resizable", TRUE,
345                     "sizing", GTK_TREE_VIEW_COLUMN_AUTOSIZE,
346                     NULL);
347
348       gtk_tree_view_append_column (GTK_TREE_VIEW (ia->fields_tree_view), column);
349     }
350 }
351
352 static void
353 reset_tree_view_model (PsppireImportAssistant *ia)
354 {
355   GtkTreeModel *tm = gtk_tree_view_get_model (GTK_TREE_VIEW (ia->fields_tree_view));
356   g_object_ref (tm);
357   gtk_tree_view_set_model (GTK_TREE_VIEW (ia->fields_tree_view), NULL);
358
359
360   repopulate_delimiter_columns (ia);
361
362   gtk_tree_view_set_model (GTK_TREE_VIEW (ia->fields_tree_view), tm);
363   //  gtk_tree_view_columns_autosize (GTK_TREE_VIEW (ia->fields_tree_view));
364
365   g_object_unref (tm);
366 }
367
368 /* Called just before the separators page becomes visible in the
369    assistant, and when the Reset button is clicked. */
370 static void
371 prepare_separators_page (PsppireImportAssistant *ia, GtkWidget *page)
372 {
373   gtk_tree_view_set_model (GTK_TREE_VIEW (ia->fields_tree_view), ia->delimiters_model);
374
375   g_signal_connect_swapped (ia->delimiters_model, "notify::delimiters",
376                         G_CALLBACK (reset_tree_view_model), ia);
377
378
379   repopulate_delimiter_columns (ia);
380
381   revise_fields_preview (ia);
382   choose_likely_separators (ia);
383 }
384
385 /* Resets IA's intro page to its initial state. */
386 static void
387 reset_intro_page (PsppireImportAssistant *ia)
388 {
389   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (ia->all_cases_button),
390                                 TRUE);
391 }
392
393
394
395 /* Clears the set of user-modified variables from IA's formats
396    substructure.  This discards user modifications to variable
397    formats, thereby causing formats to revert to their
398    defaults.  */
399 static void
400 reset_formats_page (PsppireImportAssistant *ia, GtkWidget *page)
401 {
402 }
403
404 static void prepare_formats_page (PsppireImportAssistant *ia);
405
406 /* Called when the Reset button is clicked. */
407 static void
408 on_reset (GtkButton *button, PsppireImportAssistant *ia)
409 {
410   gint pn = gtk_assistant_get_current_page (GTK_ASSISTANT (ia));
411   {
412     GtkWidget *page =  gtk_assistant_get_nth_page (GTK_ASSISTANT (ia), pn);
413
414     page_func *on_reset = g_object_get_data (G_OBJECT (page), "on-reset");
415
416     if (on_reset)
417       on_reset (ia, page);
418   }
419 }
420
421
422 static gint
423 next_page_func (gint old_page, gpointer data)
424 {
425   return old_page + 1;
426 }
427
428
429 /* Called just before PAGE is displayed as the current page of
430    IMPORT_ASSISTANT, this updates IA content according to the new
431    page. */
432 static void
433 on_prepare (GtkAssistant *assistant, GtkWidget *page, PsppireImportAssistant *ia)
434 {
435   gtk_widget_show (ia->reset_button);
436   gtk_widget_hide (ia->paste_button);
437
438   gint pn = gtk_assistant_get_current_page (assistant);
439   gint previous_page_index = ia->current_page;
440
441   if (previous_page_index >= 0)
442     {
443       GtkWidget *closing_page = gtk_assistant_get_nth_page (GTK_ASSISTANT (ia), previous_page_index);
444
445       if (pn > previous_page_index)
446         {
447           page_func *on_forward = g_object_get_data (G_OBJECT (closing_page), "on-forward");
448
449           if (on_forward)
450             on_forward (ia, closing_page);
451         }
452       else
453         {
454           page_func *on_back = g_object_get_data (G_OBJECT (closing_page), "on-back");
455
456           if (on_back)
457             on_back (ia, closing_page);
458         }
459     }
460
461   {
462     GtkWidget *new_page = gtk_assistant_get_nth_page (GTK_ASSISTANT (ia), pn);
463
464     page_func *on_entering = g_object_get_data (G_OBJECT (new_page), "on-entering");
465
466     if (on_entering)
467       on_entering (ia, new_page);
468   }
469
470   ia->current_page = pn;
471 }
472
473 /* Called when the Cancel button in the assistant is clicked. */
474 static void
475 on_cancel (GtkAssistant *assistant, PsppireImportAssistant *ia)
476 {
477   close_assistant (ia, GTK_RESPONSE_CANCEL);
478 }
479
480 /* Called when the Apply button on the last page of the assistant
481    is clicked. */
482 static void
483 on_close (GtkAssistant *assistant, PsppireImportAssistant *ia)
484 {
485   close_assistant (ia, GTK_RESPONSE_APPLY);
486 }
487
488
489 static GtkWidget *
490 add_page_to_assistant (PsppireImportAssistant *ia,
491                        GtkWidget *page, GtkAssistantPageType type, const gchar *);
492
493
494 static void
495 on_sheet_combo_changed (GtkComboBox *cb, PsppireImportAssistant *ia)
496 {
497   GtkTreeIter iter;
498   gchar *range = NULL;
499   GtkTreeModel *model = gtk_combo_box_get_model (cb);
500   GtkBuilder *builder = ia->builder;
501   GtkWidget *range_entry = get_widget_assert (builder, "cell-range-entry");
502
503   gtk_combo_box_get_active_iter (cb, &iter);
504   gtk_tree_model_get (model, &iter, PSPPIRE_SPREADSHEET_MODEL_COL_RANGE, &range, -1);
505   gtk_entry_set_text (GTK_ENTRY (range_entry), range ?  range : "");
506   g_free (range);
507 }
508
509 /* Prepares IA's sheet_spec page. */
510 static void
511 prepare_sheet_spec_page (PsppireImportAssistant *ia)
512 {
513   GtkBuilder *builder = ia->builder;
514   GtkWidget *sheet_entry = get_widget_assert (builder, "sheet-entry");
515   GtkWidget *readnames_checkbox = get_widget_assert (builder, "readnames-checkbox");
516
517   gtk_combo_box_set_model (GTK_COMBO_BOX (sheet_entry),
518                            psppire_spreadsheet_model_new (ia->spreadsheet));
519
520   gtk_combo_box_set_active (GTK_COMBO_BOX (sheet_entry), 0);
521
522   gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (readnames_checkbox), FALSE);
523 }
524
525
526 /* Initializes IA's sheet_spec substructure. */
527 static void
528 sheet_spec_page_create (PsppireImportAssistant *ia)
529 {
530   GtkBuilder *builder = ia->builder;
531   GtkWidget *page = get_widget_assert (builder, "Spreadsheet-Importer");
532
533   GtkWidget *combo_box = get_widget_assert (builder, "sheet-entry");
534   GtkCellRenderer *renderer = gtk_cell_renderer_text_new ();
535   gtk_cell_layout_clear (GTK_CELL_LAYOUT (combo_box));
536   gtk_cell_layout_pack_start (GTK_CELL_LAYOUT (combo_box), renderer, TRUE);
537   gtk_cell_layout_set_attributes (GTK_CELL_LAYOUT (combo_box), renderer,
538                                   "text", 0,
539                                   NULL);
540
541   g_signal_connect (combo_box, "changed", G_CALLBACK (on_sheet_combo_changed), ia);
542
543   add_page_to_assistant (ia, page,
544                          GTK_ASSISTANT_PAGE_CONTENT, _("Importing Spreadsheet Data"));
545
546   g_object_set_data (G_OBJECT (page), "on-entering", prepare_sheet_spec_page);
547 }
548
549 static void
550 on_chosen (PsppireImportAssistant *ia, GtkWidget *page)
551 {
552   GtkFileChooser *fc = GTK_FILE_CHOOSER (page);
553   gchar *f = gtk_file_chooser_get_filename (fc);
554   int i;
555
556   for(i = gtk_assistant_get_n_pages (GTK_ASSISTANT (ia)); i > 0; --i)
557     gtk_assistant_remove_page (GTK_ASSISTANT (ia), i);
558
559   gtk_assistant_set_page_complete (GTK_ASSISTANT(ia), GTK_WIDGET (fc), FALSE);
560
561   if (f && g_file_test (f, G_FILE_TEST_IS_REGULAR))
562     {
563       gtk_assistant_set_page_complete (GTK_ASSISTANT(ia), GTK_WIDGET (fc), TRUE);
564
565       if (ia->spreadsheet)
566         spreadsheet_unref (ia->spreadsheet);
567
568       ia->spreadsheet = gnumeric_probe (f, FALSE);
569
570       if (!ia->spreadsheet)
571         ia->spreadsheet = ods_probe (f, FALSE);
572
573       if (ia->spreadsheet)
574         {
575           sheet_spec_page_create (ia);
576         }
577       else
578         {
579           intro_page_create (ia);
580           first_line_page_create (ia);
581           separators_page_create (ia);
582         }
583
584       formats_page_create (ia);
585     }
586
587   g_free (f);
588 }
589
590 /* This has to be done on a map signal callback,
591    because GtkFileChooserWidget resets everything when it is mapped. */
592 static void
593 on_map (PsppireImportAssistant *ia, GtkWidget *page)
594 {
595 #if TEXT_FILE
596   GtkFileChooser *fc = GTK_FILE_CHOOSER (page);
597
598   if (ia->file_name)
599     gtk_file_chooser_set_filename (fc, ia->file_name);
600 #endif
601
602   on_chosen (ia, page);
603 }
604
605
606
607 static void
608 chooser_page_enter (PsppireImportAssistant *ia, GtkWidget *page)
609 {
610 }
611
612 static void
613 chooser_page_leave (PsppireImportAssistant *ia, GtkWidget *page)
614 {
615   g_print ("%s:%d %s\n", __FILE__, __LINE__, __FUNCTION__);
616   g_free (ia->file_name);
617   ia->file_name = gtk_file_chooser_get_filename (GTK_FILE_CHOOSER (page));
618   gchar *encoding = psppire_encoding_selector_get_encoding (ia->encoding_selector);
619
620   if (!ia->spreadsheet)
621     {
622       ia->text_file = psppire_text_file_new (ia->file_name, encoding);
623       gtk_tree_view_set_model (GTK_TREE_VIEW (ia->first_line_tree_view),
624                                GTK_TREE_MODEL (ia->text_file));
625     }
626
627
628   g_free (encoding);
629 }
630
631 static void
632 chooser_page_reset (PsppireImportAssistant *ia, GtkWidget *page)
633 {
634   GtkFileChooser *fc = GTK_FILE_CHOOSER (page);
635
636   gtk_file_chooser_set_filter (fc, ia->default_filter);
637   gtk_file_chooser_unselect_all (fc);
638
639   on_chosen (ia, page);
640 }
641
642
643
644 static void
645 chooser_page_create (PsppireImportAssistant *ia)
646 {
647   GtkFileFilter *filter = NULL;
648
649   GtkWidget *chooser = gtk_file_chooser_widget_new (GTK_FILE_CHOOSER_ACTION_OPEN);
650
651   g_object_set_data (G_OBJECT (chooser), "on-forward", chooser_page_leave);
652   g_object_set_data (G_OBJECT (chooser), "on-reset",   chooser_page_reset);
653   g_object_set_data (G_OBJECT (chooser), "on-entering",chooser_page_enter);
654
655   g_object_set (chooser, "local-only", FALSE, NULL);
656
657
658   ia->default_filter = gtk_file_filter_new ();
659   gtk_file_filter_set_name (ia->default_filter, _("All Files"));
660   gtk_file_filter_add_pattern (ia->default_filter, "*");
661   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), ia->default_filter);
662
663   filter = gtk_file_filter_new ();
664   gtk_file_filter_set_name (filter, _("Text Files"));
665   gtk_file_filter_add_mime_type (filter, "text/*");
666   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), filter);
667
668   filter = gtk_file_filter_new ();
669   gtk_file_filter_set_name (filter, _("Text (*.txt) Files"));
670   gtk_file_filter_add_pattern (filter, "*.txt");
671   gtk_file_filter_add_pattern (filter, "*.TXT");
672   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), filter);
673
674   filter = gtk_file_filter_new ();
675   gtk_file_filter_set_name (filter, _("Plain Text (ASCII) Files"));
676   gtk_file_filter_add_mime_type (filter, "text/plain");
677   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), filter);
678
679   filter = gtk_file_filter_new ();
680   gtk_file_filter_set_name (filter, _("Comma Separated Value Files"));
681   gtk_file_filter_add_mime_type (filter, "text/csv");
682   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), filter);
683
684   /* I've never encountered one of these, but it's listed here:
685      http://www.iana.org/assignments/media-types/text/tab-separated-values  */
686   filter = gtk_file_filter_new ();
687   gtk_file_filter_set_name (filter, _("Tab Separated Value Files"));
688   gtk_file_filter_add_mime_type (filter, "text/tab-separated-values");
689   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), filter);
690
691   filter = gtk_file_filter_new ();
692   gtk_file_filter_set_name (filter, _("Gnumeric Spreadsheet Files"));
693   gtk_file_filter_add_mime_type (filter, "application/x-gnumeric");
694   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), filter);
695
696   filter = gtk_file_filter_new ();
697   gtk_file_filter_set_name (filter, _("OpenDocument Spreadsheet Files"));
698   gtk_file_filter_add_mime_type (filter, "application/vnd.oasis.opendocument.spreadsheet");
699   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), filter);
700
701   filter = gtk_file_filter_new ();
702   gtk_file_filter_set_name (filter, _("All Spreadsheet Files"));
703   gtk_file_filter_add_mime_type (filter, "application/x-gnumeric");
704   gtk_file_filter_add_mime_type (filter, "application/vnd.oasis.opendocument.spreadsheet");
705   gtk_file_chooser_add_filter (GTK_FILE_CHOOSER (chooser), filter);
706
707   ia->encoding_selector = psppire_encoding_selector_new ("Auto", TRUE);
708   gtk_file_chooser_set_extra_widget (GTK_FILE_CHOOSER (chooser), ia->encoding_selector);
709
710   add_page_to_assistant (ia, chooser,
711                          GTK_ASSISTANT_PAGE_INTRO, _("Select File to Import"));
712
713   g_signal_connect_swapped (chooser, "selection-changed", G_CALLBACK (on_chosen), ia);
714   g_signal_connect_swapped (chooser, "map", G_CALLBACK (on_map), ia);
715 }
716
717
718
719 static void
720 psppire_import_assistant_init (PsppireImportAssistant *ia)
721 {
722   ia->builder = builder_new ("text-data-import.ui");
723
724   ia->current_page = -1 ;
725   ia->file_name = NULL;
726
727   ia->spreadsheet = NULL;
728
729   ia->main_loop = g_main_loop_new (NULL, TRUE);
730
731   g_signal_connect (ia, "prepare", G_CALLBACK (on_prepare), ia);
732   g_signal_connect (ia, "cancel", G_CALLBACK (on_cancel), ia);
733   g_signal_connect (ia, "close", G_CALLBACK (on_close), ia);
734
735   ia->paste_button = gtk_button_new_with_label (_("Paste"));
736   ia->reset_button = gtk_button_new_with_label (_("Reset"));
737
738   gtk_assistant_add_action_widget (GTK_ASSISTANT(ia), ia->paste_button);
739
740   g_signal_connect (ia->paste_button, "clicked", G_CALLBACK (on_paste), ia);
741   g_signal_connect (ia->reset_button, "clicked", G_CALLBACK (on_reset), ia);
742
743   gtk_assistant_add_action_widget (GTK_ASSISTANT(ia), ia->reset_button);
744
745   gtk_window_set_title (GTK_WINDOW (ia),
746                         _("Importing Delimited Text Data"));
747
748   gtk_window_set_icon_name (GTK_WINDOW (ia), "pspp");
749
750   chooser_page_create (ia);
751
752   gtk_assistant_set_forward_page_func (GTK_ASSISTANT (ia), next_page_func, NULL, NULL);
753
754   gtk_window_maximize (GTK_WINDOW (ia));
755 }
756
757
758 /* Appends a page of the given TYPE, with PAGE as its content, to
759    the GtkAssistant encapsulated by IA.  Returns the GtkWidget
760    that represents the page. */
761 static GtkWidget *
762 add_page_to_assistant (PsppireImportAssistant *ia,
763                        GtkWidget *page, GtkAssistantPageType type, const gchar *title)
764 {
765   GtkWidget *content = page;
766
767   gtk_assistant_append_page (GTK_ASSISTANT (ia), content);
768   gtk_assistant_set_page_type (GTK_ASSISTANT(ia), content, type);
769   gtk_assistant_set_page_title (GTK_ASSISTANT(ia), content, title);
770   gtk_assistant_set_page_complete (GTK_ASSISTANT(ia), content, TRUE);
771
772   return content;
773 }
774
775
776 /* Called when one of the radio buttons is clicked. */
777 static void
778 on_intro_amount_changed (PsppireImportAssistant *p)
779 {
780   gtk_widget_set_sensitive (p->n_cases_spin,
781                             gtk_toggle_button_get_active (
782                                                           GTK_TOGGLE_BUTTON (p->n_cases_button)));
783
784   gtk_widget_set_sensitive (p->percent_spin,
785                             gtk_toggle_button_get_active (
786                                                           GTK_TOGGLE_BUTTON (p->percent_button)));
787 }
788
789
790
791 static void
792 on_treeview_selection_change (PsppireImportAssistant *ia)
793 {
794   GtkTreeSelection *selection =
795     gtk_tree_view_get_selection (GTK_TREE_VIEW (ia->first_line_tree_view));
796   GtkTreeModel *model = NULL;
797   GtkTreeIter iter;
798   if (gtk_tree_selection_get_selected (selection, &model, &iter))
799     {
800       int n;
801       GtkTreePath *path = gtk_tree_model_get_path (model, &iter);
802       gint *index = gtk_tree_path_get_indices (path);
803
804       n = *index;
805
806       gtk_tree_path_free (path);
807
808       gtk_widget_set_sensitive (ia->variable_names_cb, n > 0);
809
810       ia->delimiters_model
811         = psppire_delimited_text_new (GTK_TREE_MODEL (ia->text_file));
812       g_object_set (ia->delimiters_model, "first-line", n, NULL);
813
814       g_print ("%s:%d DT %p first line %d\n", __FILE__, __LINE__, ia->delimiters_model, n);
815     }
816 }
817
818
819 /* Initializes IA's first_line substructure. */
820 static void
821 first_line_page_create (PsppireImportAssistant *ia)
822 {
823   g_print ("%s:%d %s\n", __FILE__, __LINE__, __FUNCTION__);
824   GtkWidget *w =  get_widget_assert (ia->builder, "FirstLine");
825
826   g_object_set_data (G_OBJECT (w), "on-entering", on_treeview_selection_change);
827
828   add_page_to_assistant (ia, w,
829                          GTK_ASSISTANT_PAGE_CONTENT, _("Select the First Line"));
830
831   GtkWidget *scrolled_window = get_widget_assert (ia->builder, "first-line-scroller");
832
833   if (ia->first_line_tree_view == NULL)
834     {
835       ia->first_line_tree_view = gtk_tree_view_new ();
836
837       gtk_tree_view_set_headers_visible (GTK_TREE_VIEW (ia->first_line_tree_view), TRUE);
838
839       GtkCellRenderer *renderer = gtk_cell_renderer_text_new ();
840       GtkTreeViewColumn *column = gtk_tree_view_column_new_with_attributes (_("Line"), renderer,
841                                                                             "text", 0,
842                                                                             NULL);
843       gtk_tree_view_append_column (GTK_TREE_VIEW (ia->first_line_tree_view), column);
844
845       renderer = gtk_cell_renderer_text_new ();
846       column = gtk_tree_view_column_new_with_attributes (_("Text"), renderer, "text", 1, NULL);
847       gtk_tree_view_append_column (GTK_TREE_VIEW (ia->first_line_tree_view), column);
848
849       gtk_container_add (GTK_CONTAINER (scrolled_window), ia->first_line_tree_view);
850
851       g_signal_connect_swapped (ia->first_line_tree_view, "cursor-changed",
852                         G_CALLBACK (on_treeview_selection_change), ia);
853     }
854   gtk_widget_show_all (scrolled_window);
855
856   ia->variable_names_cb = get_widget_assert (ia->builder, "variable-names");
857
858 }
859
860
861
862
863 static void
864 intro_on_enter (PsppireImportAssistant *ia)
865 {
866   GtkBuilder *builder = ia->builder;
867   GtkWidget *table  = get_widget_assert (builder, "button-table");
868
869   struct string s;
870
871   ds_init_empty (&s);
872   ds_put_cstr (&s, _("This assistant will guide you through the process of "
873                      "importing data into PSPP from a text file with one line "
874                      "per case,  in which fields are separated by tabs, "
875                      "commas, or other delimiters.\n\n"));
876
877   if (ia->text_file)
878     {
879       if (ia->text_file->total_is_exact)
880         {
881           ds_put_format (
882                          &s, ngettext ("The selected file contains %'lu line of text.  ",
883                                        "The selected file contains %'lu lines of text.  ",
884                                        ia->text_file->total_lines),
885                          ia->text_file->total_lines);
886         }
887       else if (ia->text_file->total_lines > 0)
888         {
889           ds_put_format (
890                          &s, ngettext (
891                                        "The selected file contains approximately %'lu line of text.  ",
892                                        "The selected file contains approximately %'lu lines of text.  ",
893                                        ia->text_file->total_lines),
894                          ia->text_file->total_lines);
895           ds_put_format (
896                          &s, ngettext (
897                                        "Only the first %zu line of the file will be shown for "
898                                        "preview purposes in the following screens.  ",
899                                        "Only the first %zu lines of the file will be shown for "
900                                        "preview purposes in the following screens.  ",
901                                        ia->text_file->line_cnt),
902                          ia->text_file->line_cnt);
903         }
904     }
905
906   ds_put_cstr (&s, _("You may choose below how much of the file should "
907                      "actually be imported."));
908
909   gtk_label_set_text (GTK_LABEL (get_widget_assert (builder, "intro-label")),
910                       ds_cstr (&s));
911   ds_destroy (&s);
912
913   if (gtk_grid_get_child_at (GTK_GRID (table), 1, 1) == NULL)
914     {
915       GtkWidget *hbox_n_cases = psppire_scanf_new (_("Only the first %4d cases"), &ia->n_cases_spin);
916       gtk_grid_attach (GTK_GRID (table), hbox_n_cases,
917                        1, 1,
918                        1, 1);
919     }
920
921   GtkAdjustment *adj = gtk_spin_button_get_adjustment (GTK_SPIN_BUTTON (ia->n_cases_spin));
922   gtk_adjustment_set_lower (adj, 1.0);
923
924   if (gtk_grid_get_child_at (GTK_GRID (table), 1, 2) == NULL)
925     {
926       GtkWidget *hbox_percent = psppire_scanf_new (_("Only the first %3d %% of file (approximately)"),
927                                                    &ia->percent_spin);
928
929       gtk_grid_attach (GTK_GRID (table), hbox_percent,
930                        1, 2,
931                        1, 1);
932     }
933
934   gtk_widget_show_all (table);
935
936   on_intro_amount_changed (ia);
937 }
938
939 /* Initializes IA's intro substructure. */
940 static void
941 intro_page_create (PsppireImportAssistant *ia)
942 {
943   GtkBuilder *builder = ia->builder;
944
945   GtkWidget *w =  get_widget_assert (builder, "Intro");
946
947   ia->percent_spin = gtk_spin_button_new_with_range (0, 100, 10);
948
949
950   add_page_to_assistant (ia, w,  GTK_ASSISTANT_PAGE_CONTENT, _("Select the Lines to Import"));
951
952   ia->all_cases_button = get_widget_assert (builder, "import-all-cases");
953
954   ia->n_cases_button = get_widget_assert (builder, "import-n-cases");
955
956   ia->percent_button = get_widget_assert (builder, "import-percent");
957
958   g_signal_connect_swapped (ia->all_cases_button, "toggled",
959                             G_CALLBACK (on_intro_amount_changed), ia);
960   g_signal_connect_swapped (ia->n_cases_button, "toggled",
961                             G_CALLBACK (on_intro_amount_changed), ia);
962   g_signal_connect_swapped (ia->percent_button, "toggled",
963                             G_CALLBACK (on_intro_amount_changed), ia);
964
965
966   g_object_set_data (G_OBJECT (w), "on-entering", intro_on_enter);
967   g_object_set_data (G_OBJECT (w), "on-reset", reset_intro_page);
968 }
969
970
971 GtkWidget *
972 psppire_import_assistant_new (GtkWindow *toplevel)
973 {
974   return GTK_WIDGET (g_object_new (PSPPIRE_TYPE_IMPORT_ASSISTANT,
975                                    /* Some window managers (notably ratpoison)
976                                       ignore the maximise command when a window is
977                                       transient.  This causes problems for this
978                                       window. */
979                                    /* "transient-for", toplevel, */
980                                    NULL));
981 }
982
983
984
985 \f
986
987 static void
988 set_quote_list (GtkComboBox *cb)
989 {
990   GtkListStore *list =  gtk_list_store_new (1, G_TYPE_STRING);
991   GtkTreeIter iter;
992   gint i;
993   const gchar *separator[3] = {"'\"", "\'", "\""};
994
995   for (i = 0; i < 3; i++)
996     {
997       const gchar *s = separator[i];
998
999       /* Add a new row to the model */
1000       gtk_list_store_append (list, &iter);
1001       gtk_list_store_set (list, &iter,
1002                           0, s,
1003                           -1);
1004
1005     }
1006
1007   gtk_combo_box_set_model (GTK_COMBO_BOX (cb), GTK_TREE_MODEL (list));
1008   g_object_unref (list);
1009
1010   gtk_combo_box_set_entry_text_column (cb, 0);
1011 }
1012
1013
1014
1015 /* Chooses a name for each column on the separators page */
1016 static void
1017 choose_column_names (PsppireImportAssistant *ia)
1018 {
1019   int i;
1020   unsigned long int generated_name_count = 0;
1021   dict_clear (ia->dict);
1022
1023   for (i = 0; i < gtk_tree_model_get_n_columns (ia->delimiters_model) - 1; ++i)
1024     {
1025       const gchar *candidate_name = NULL;
1026
1027       if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (ia->variable_names_cb)))
1028         {
1029           candidate_name = psppire_delimited_text_get_header_title (PSPPIRE_DELIMITED_TEXT (ia->delimiters_model), i);
1030         }
1031
1032       char *name = dict_make_unique_var_name (ia->dict,
1033                                               candidate_name,
1034                                               &generated_name_count);
1035
1036       dict_create_var_assert (ia->dict, name, 0);
1037       free (name);
1038     }
1039 }
1040
1041
1042
1043 /* Called when the user toggles one of the separators
1044    checkboxes. */
1045 static void
1046 on_separator_toggle (GtkToggleButton *toggle UNUSED,
1047                      PsppireImportAssistant *ia)
1048 {
1049   int i;
1050   GSList *delimiters = NULL;
1051   for (i = 0; i < SEPARATOR_CNT; i++)
1052     {
1053       const struct separator *s = &separators[i];
1054       GtkWidget *button = get_widget_assert (ia->builder, s->name);
1055       if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (button)))
1056         {
1057           delimiters = g_slist_prepend (delimiters,  GINT_TO_POINTER (s->c));
1058         }
1059     }
1060
1061   g_object_set (ia->delimiters_model, "delimiters", delimiters, NULL);
1062
1063   revise_fields_preview (ia);
1064 }
1065
1066
1067 /* Called when the user changes the entry field for custom
1068    separators. */
1069 static void
1070 on_separators_custom_entry_notify (GObject *gobject UNUSED,
1071                                    GParamSpec *arg1 UNUSED,
1072                                    PsppireImportAssistant *ia)
1073 {
1074   revise_fields_preview (ia);
1075 }
1076
1077 /* Called when the user toggles the checkbox that enables custom
1078    separators. */
1079 static void
1080 on_separators_custom_cb_toggle (GtkToggleButton *custom_cb,
1081                                 PsppireImportAssistant *ia)
1082 {
1083   bool is_active = gtk_toggle_button_get_active (custom_cb);
1084   gtk_widget_set_sensitive (ia->custom_entry, is_active);
1085   revise_fields_preview (ia);
1086 }
1087
1088 /* Called when the user changes the selection in the combo box
1089    that selects a quote character. */
1090 static void
1091 on_quote_combo_change (GtkComboBox *combo, PsppireImportAssistant *ia)
1092 {
1093   //  revise_fields_preview (ia);
1094 }
1095
1096 /* Called when the user toggles the checkbox that enables
1097    quoting. */
1098 static void
1099 on_quote_cb_toggle (GtkToggleButton *quote_cb, PsppireImportAssistant *ia)
1100 {
1101   bool is_active = gtk_toggle_button_get_active (quote_cb);
1102   gtk_widget_set_sensitive (ia->quote_combo, is_active);
1103   revise_fields_preview (ia);
1104 }
1105
1106 /* Initializes IA's separators substructure. */
1107 static void
1108 separators_page_create (PsppireImportAssistant *ia)
1109 {
1110   GtkBuilder *builder = ia->builder;
1111
1112   size_t i;
1113
1114   GtkWidget *w = get_widget_assert (builder, "Separators");
1115
1116   g_object_set_data (G_OBJECT (w), "on-entering", prepare_separators_page);
1117   g_object_set_data (G_OBJECT (w), "on-reset", prepare_separators_page);
1118
1119   add_page_to_assistant (ia, w,   GTK_ASSISTANT_PAGE_CONTENT, _("Choose Separators"));
1120
1121   ia->custom_cb = get_widget_assert (builder, "custom-cb");
1122   ia->custom_entry = get_widget_assert (builder, "custom-entry");
1123   ia->quote_combo = get_widget_assert (builder, "quote-combo");
1124   ia->quote_entry = GTK_ENTRY (gtk_bin_get_child (GTK_BIN (ia->quote_combo)));
1125   ia->quote_cb = get_widget_assert (builder, "quote-cb");
1126
1127   set_quote_list (GTK_COMBO_BOX (ia->quote_combo));
1128
1129   if (ia->fields_tree_view == NULL)
1130     {
1131       GtkWidget *scroller = get_widget_assert (ia->builder, "fields-scroller");
1132       ia->fields_tree_view = gtk_tree_view_new ();
1133       gtk_container_add (GTK_CONTAINER (scroller), GTK_WIDGET (ia->fields_tree_view));
1134       gtk_widget_show_all (scroller);
1135     }
1136
1137   g_signal_connect (ia->quote_combo, "changed",
1138                     G_CALLBACK (on_quote_combo_change), ia);
1139   g_signal_connect (ia->quote_cb, "toggled",
1140                     G_CALLBACK (on_quote_cb_toggle), ia);
1141   g_signal_connect (ia->custom_entry, "notify::text",
1142                     G_CALLBACK (on_separators_custom_entry_notify), ia);
1143   g_signal_connect (ia->custom_cb, "toggled",
1144                     G_CALLBACK (on_separators_custom_cb_toggle), ia);
1145   for (i = 0; i < SEPARATOR_CNT; i++)
1146     g_signal_connect (get_widget_assert (builder, separators[i].name),
1147                       "toggled", G_CALLBACK (on_separator_toggle), ia);
1148
1149 }
1150
1151
1152
1153 \f
1154
1155
1156 static struct casereader_random_class my_casereader_class;
1157
1158 static struct ccase *
1159 my_read (struct casereader *reader, void *aux, casenumber idx)
1160 {
1161   PsppireImportAssistant *ia = PSPPIRE_IMPORT_ASSISTANT (aux);
1162   GtkTreeModel *tm = GTK_TREE_MODEL (ia->delimiters_model);
1163
1164   GtkTreePath *tp = gtk_tree_path_new_from_indices (idx, -1);
1165
1166   const struct caseproto *proto = casereader_get_proto (reader);
1167
1168   GtkTreeIter iter;
1169   struct ccase *c = NULL;
1170   if (gtk_tree_model_get_iter (tm, &iter, tp))
1171     {
1172       c = case_create (proto);
1173       int i;
1174       for (i = 0 ; i < caseproto_get_n_widths (proto); ++i)
1175         {
1176           GValue value = {0};
1177           gtk_tree_model_get_value (tm, &iter, i + 1, &value);
1178
1179           const struct variable *var = dict_get_var (ia->dict, i);
1180
1181           const gchar *ss = g_value_get_string (&value);
1182
1183           union value *v = case_data_rw (c, var);
1184           char *xx = data_in (ss_cstr (ss),
1185                               "UTF-8",
1186                               var_get_write_format (var)->type,
1187                               v, var_get_width (var), "UTF-8");
1188
1189           /* if (xx) */
1190           /*   g_print ("%s:%d Err %s\n", __FILE__, __LINE__, xx); */
1191           free (xx);
1192           g_value_unset (&value);
1193         }
1194     }
1195
1196   gtk_tree_path_free (tp);
1197
1198   return c;
1199 }
1200
1201 static void
1202 my_destroy (struct casereader *reader, void *aux)
1203 {
1204   g_print ("%s:%d %p\n", __FILE__, __LINE__, reader);
1205 }
1206
1207 static void
1208 my_advance (struct casereader *reader, void *aux, casenumber cnt)
1209 {
1210   g_print ("%s:%d\n", __FILE__, __LINE__);
1211 }
1212
1213 static void
1214 foo (struct dictionary *dict, void *aux)
1215 {
1216   PsppireImportAssistant *ia = PSPPIRE_IMPORT_ASSISTANT (aux);
1217   g_print ("%s:%d\n", __FILE__, __LINE__);
1218
1219   struct caseproto *proto = caseproto_create ();
1220
1221   int i;
1222   for (i = 0 ; i < dict_get_var_cnt (ia->dict); ++i)
1223     {
1224       const struct variable *var = dict_get_var (ia->dict, i);
1225       proto = caseproto_add_width (proto, var_get_width (var));
1226     }
1227
1228
1229   gint n_rows = gtk_tree_model_iter_n_children (ia->delimiters_model, NULL);
1230
1231   struct casereader *reader =
1232     casereader_create_random (proto, n_rows, &my_casereader_class,  ia);
1233
1234
1235   PsppireDataStore *store = NULL;
1236
1237   g_object_get (ia->data_sheet, "data-model", &store, NULL);
1238
1239   psppire_data_store_set_reader (store, reader);
1240 }
1241
1242 /* Called just before the formats page of the assistant is
1243    displayed. */
1244 static void
1245 prepare_formats_page (PsppireImportAssistant *ia)
1246 {
1247   PsppireDict *dict = psppire_dict_new_from_dict (ia->dict);
1248   g_object_set (ia->var_sheet, "data-model", dict, NULL);
1249
1250   my_casereader_class.read = my_read;
1251   my_casereader_class.destroy = my_destroy;
1252   my_casereader_class.advance = my_advance;
1253
1254   struct caseproto *proto = caseproto_create ();
1255   int i;
1256
1257   struct fmt_guesser **fg = xcalloc (sizeof *fg, dict_get_var_cnt (ia->dict));
1258   for (i = 0 ; i < dict_get_var_cnt (ia->dict); ++i)
1259     {
1260       fg[i] = fmt_guesser_create ();
1261     }
1262
1263   gint n_rows = gtk_tree_model_iter_n_children (ia->delimiters_model, NULL);
1264
1265   GtkTreeIter iter;
1266   gboolean ok;
1267   for (ok = gtk_tree_model_get_iter_first (ia->delimiters_model, &iter);
1268        ok;
1269        ok = gtk_tree_model_iter_next (ia->delimiters_model, &iter))
1270     {
1271       for (i = 0 ; i < dict_get_var_cnt (ia->dict); ++i)
1272         {
1273           gchar *s = NULL;
1274           gtk_tree_model_get (ia->delimiters_model, &iter, i+1, &s, -1);
1275           fmt_guesser_add (fg[i], ss_cstr (s));
1276           free (s);
1277         }
1278     }
1279
1280   for (i = 0 ; i < dict_get_var_cnt (ia->dict); ++i)
1281     {
1282       struct fmt_spec fs;
1283       fmt_guesser_guess (fg[i], &fs);
1284
1285       fmt_fix (&fs, FMT_FOR_INPUT);
1286
1287       struct variable *var = dict_get_var (ia->dict, i);
1288
1289       int width = fmt_var_width (&fs);
1290
1291       var_set_width_and_formats (var, width,
1292                                  &fs, &fs);
1293
1294       proto = caseproto_add_width (proto, width);
1295       fmt_guesser_destroy (fg[i]);
1296     }
1297
1298   free (fg);
1299
1300   //  dict_set_change_callback (ia->dict, foo, ia);
1301
1302   struct casereader *reader =
1303     casereader_create_random (proto, n_rows, &my_casereader_class,  ia);
1304
1305   PsppireDataStore *store = psppire_data_store_new (dict);
1306   psppire_data_store_set_reader (store, reader);
1307
1308   g_object_set (ia->data_sheet, "data-model", store, NULL);
1309
1310
1311   gint pmax;
1312   g_object_get (get_widget_assert (ia->builder, "vpaned1"),
1313                 "max-position", &pmax, NULL);
1314
1315
1316   g_object_set (get_widget_assert (ia->builder, "vpaned1"),
1317                 "position", pmax / 2, NULL);
1318
1319   gtk_widget_show (ia->paste_button);
1320 }
1321
1322 static void
1323 formats_page_create (PsppireImportAssistant *ia)
1324 {
1325   GtkBuilder *builder = ia->builder;
1326
1327   GtkWidget *w = get_widget_assert (builder, "Formats");
1328   g_object_set_data (G_OBJECT (w), "on-entering", prepare_formats_page);
1329   g_object_set_data (G_OBJECT (w), "on-reset", reset_formats_page);
1330
1331   GtkWidget *vars_scroller = get_widget_assert (builder, "vars-scroller");
1332   if (ia->var_sheet == NULL)
1333     {
1334       ia->var_sheet = psppire_variable_sheet_new ();
1335
1336       gtk_container_add (GTK_CONTAINER (vars_scroller), ia->var_sheet);
1337
1338       ia->dict = dict_create (get_default_encoding ());
1339
1340       gtk_widget_show_all (vars_scroller);
1341     }
1342   GtkWidget *data_scroller = get_widget_assert (builder, "data-scroller");
1343   if (ia->data_sheet == NULL)
1344     {
1345       ia->data_sheet = psppire_data_sheet_new ();
1346
1347       gtk_container_add (GTK_CONTAINER (data_scroller), ia->data_sheet);
1348
1349       gtk_widget_show_all (data_scroller);
1350     }
1351
1352   add_page_to_assistant (ia, w,
1353                          GTK_ASSISTANT_PAGE_CONFIRM, _("Adjust Variable Formats"));
1354 }
1355
1356
1357 \f
1358
1359 static void
1360 separators_append_syntax (const PsppireImportAssistant *ia, struct string *s)
1361 {
1362   int i;
1363
1364   ds_put_cstr (s, "  /DELIMITERS=\"");
1365
1366   if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (get_widget_assert (ia->builder, "tab"))))
1367     ds_put_cstr (s, "\\t");
1368   for (i = 0; i < SEPARATOR_CNT; i++)
1369     {
1370       const struct separator *seps = &separators[i];
1371       GtkWidget *button = get_widget_assert (ia->builder, seps->name);
1372       if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (button)))
1373         {
1374           if (seps->c == '\t')
1375             continue;
1376
1377           ds_put_byte (s, seps->c);
1378         }
1379     }
1380   ds_put_cstr (s, "\"\n");
1381   if (!ds_is_empty (&ia->quotes))
1382     syntax_gen_pspp (s, "  /QUALIFIER=%sq\n", ds_cstr (&ia->quotes));
1383 }
1384
1385 static void
1386 formats_append_syntax (const PsppireImportAssistant *ia, struct string *s)
1387 {
1388   int i;
1389   int var_cnt;
1390
1391   g_return_if_fail (ia->dict);
1392
1393   ds_put_cstr (s, "  /VARIABLES=\n");
1394
1395   var_cnt = dict_get_var_cnt (ia->dict);
1396   for (i = 0; i < var_cnt; i++)
1397     {
1398       struct variable *var = dict_get_var (ia->dict, i);
1399       char format_string[FMT_STRING_LEN_MAX + 1];
1400       fmt_to_string (var_get_print_format (var), format_string);
1401       ds_put_format (s, "    %s %s%s\n",
1402                      var_get_name (var), format_string,
1403                      i == var_cnt - 1 ? "." : "");
1404     }
1405 }
1406
1407 static void
1408 first_line_append_syntax (const PsppireImportAssistant *ia, struct string *s)
1409 {
1410   gint first_case = 0;
1411   g_object_get (ia->delimiters_model, "first-line", &first_case, NULL);
1412
1413   if (first_case > 0)
1414     ds_put_format (s, "  /FIRSTCASE=%d\n", first_case + 1);
1415 }
1416
1417 static void
1418 intro_append_syntax (const PsppireImportAssistant *ia, struct string *s)
1419 {
1420   if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (ia->n_cases_button)))
1421     ds_put_format (s, "N OF CASES %d.\n",
1422                    gtk_spin_button_get_value_as_int (GTK_SPIN_BUTTON (ia->n_cases_spin)));
1423   else if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (ia->percent_button)))
1424     ds_put_format (s, "SAMPLE %.4g.\n",
1425                    gtk_spin_button_get_value (GTK_SPIN_BUTTON (ia->percent_spin)) / 100.0);
1426 }
1427
1428
1429 /* Emits PSPP syntax to S that applies the dictionary attributes
1430    (such as missing values and value labels) of the variables in
1431    DICT.  */
1432 static void
1433 apply_dict (const struct dictionary *dict, struct string *s)
1434 {
1435   size_t var_cnt = dict_get_var_cnt (dict);
1436   size_t i;
1437
1438   for (i = 0; i < var_cnt; i++)
1439     {
1440       struct variable *var = dict_get_var (dict, i);
1441       const char *name = var_get_name (var);
1442       enum val_type type = var_get_type (var);
1443       int width = var_get_width (var);
1444       enum measure measure = var_get_measure (var);
1445       enum var_role role = var_get_role (var);
1446       enum alignment alignment = var_get_alignment (var);
1447       const struct fmt_spec *format = var_get_print_format (var);
1448
1449       if (var_has_missing_values (var))
1450         {
1451           const struct missing_values *mv = var_get_missing_values (var);
1452           size_t j;
1453
1454           syntax_gen_pspp (s, "MISSING VALUES %ss (", name);
1455           for (j = 0; j < mv_n_values (mv); j++)
1456             {
1457               if (j)
1458                 ds_put_cstr (s, ", ");
1459               syntax_gen_value (s, mv_get_value (mv, j), width, format);
1460             }
1461
1462           if (mv_has_range (mv))
1463             {
1464               double low, high;
1465               if (mv_has_value (mv))
1466                 ds_put_cstr (s, ", ");
1467               mv_get_range (mv, &low, &high);
1468               syntax_gen_num_range (s, low, high, format);
1469             }
1470           ds_put_cstr (s, ").\n");
1471         }
1472       if (var_has_value_labels (var))
1473         {
1474           const struct val_labs *vls = var_get_value_labels (var);
1475           const struct val_lab **labels = val_labs_sorted (vls);
1476           size_t n_labels = val_labs_count (vls);
1477           size_t i;
1478
1479           syntax_gen_pspp (s, "VALUE LABELS %ss", name);
1480           for (i = 0; i < n_labels; i++)
1481             {
1482               const struct val_lab *vl = labels[i];
1483               ds_put_cstr (s, "\n  ");
1484               syntax_gen_value (s, &vl->value, width, format);
1485               ds_put_byte (s, ' ');
1486               syntax_gen_string (s, ss_cstr (val_lab_get_escaped_label (vl)));
1487             }
1488           free (labels);
1489           ds_put_cstr (s, ".\n");
1490         }
1491       if (var_has_label (var))
1492         syntax_gen_pspp (s, "VARIABLE LABELS %ss %sq.\n",
1493                          name, var_get_label (var));
1494       if (measure != var_default_measure (type))
1495         syntax_gen_pspp (s, "VARIABLE LEVEL %ss (%ss).\n",
1496                          name, measure_to_syntax (measure));
1497       if (role != ROLE_INPUT)
1498         syntax_gen_pspp (s, "VARIABLE ROLE /%ss %ss.\n",
1499                          var_role_to_syntax (role), name);
1500       if (alignment != var_default_alignment (type))
1501         syntax_gen_pspp (s, "VARIABLE ALIGNMENT %ss (%ss).\n",
1502                          name, alignment_to_syntax (alignment));
1503       if (var_get_display_width (var) != var_default_display_width (width))
1504         syntax_gen_pspp (s, "VARIABLE WIDTH %ss (%d).\n",
1505                          name, var_get_display_width (var));
1506     }
1507 }
1508
1509
1510
1511 static char *
1512 sheet_spec_gen_syntax (PsppireImportAssistant *ia)
1513 {
1514   GtkBuilder *builder = ia->builder;
1515   GtkWidget *range_entry = get_widget_assert (builder, "cell-range-entry");
1516   GtkWidget *sheet_entry = get_widget_assert (builder, "sheet-entry");
1517   GtkWidget *rnc = get_widget_assert (builder, "readnames-checkbox");
1518   const gchar *range = gtk_entry_get_text (GTK_ENTRY (range_entry));
1519   int sheet_index = 1 + gtk_combo_box_get_active (GTK_COMBO_BOX (sheet_entry));
1520   gboolean read_names = gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (rnc));
1521
1522   struct string s = DS_EMPTY_INITIALIZER;
1523
1524   char *filename;
1525   g_object_get (ia->text_file, "file-name", &filename, NULL);
1526   syntax_gen_pspp (&s,
1527                    "GET DATA"
1528                    "\n  /TYPE=%ss"
1529                    "\n  /FILE=%sq"
1530                    "\n  /SHEET=index %d"
1531                    "\n  /READNAMES=%ss",
1532                    (ia->spreadsheet->type == SPREADSHEET_GNUMERIC) ? "GNM" : "ODS",
1533                    filename,
1534                    sheet_index,
1535                    read_names ? "ON" : "OFF");
1536
1537   if (range && 0 != strcmp ("", range))
1538     {
1539       syntax_gen_pspp (&s,
1540                        "\n  /CELLRANGE=RANGE %sq", range);
1541     }
1542   else
1543     {
1544       syntax_gen_pspp (&s,
1545                        "\n  /CELLRANGE=FULL");
1546     }
1547
1548
1549   syntax_gen_pspp (&s, ".");
1550
1551
1552   return ds_cstr (&s);
1553 }
1554
1555
1556 gchar *
1557 psppire_import_assistant_generate_syntax (PsppireImportAssistant *ia)
1558 {
1559   struct string s = DS_EMPTY_INITIALIZER;
1560
1561   if (!ia->spreadsheet)
1562     {
1563       gchar *file_name = NULL;
1564       gchar *encoding = NULL;
1565       g_object_get (ia->text_file,
1566                     "file-name", &file_name,
1567                     "encoding", &encoding,
1568                     NULL);
1569
1570       if (file_name == NULL)
1571         return NULL;
1572
1573       syntax_gen_pspp (&s,
1574                        "GET DATA"
1575                        "\n  /TYPE=TXT"
1576                        "\n  /FILE=%sq\n",
1577                        file_name);
1578       if (encoding && strcmp (encoding, "Auto"))
1579         syntax_gen_pspp (&s, "  /ENCODING=%sq\n", encoding);
1580
1581       ds_put_cstr (&s,
1582                    "  /ARRANGEMENT=DELIMITED\n"
1583                    "  /DELCASE=LINE\n");
1584
1585       first_line_append_syntax (ia, &s);
1586       separators_append_syntax (ia, &s);
1587
1588       formats_append_syntax (ia, &s);
1589       apply_dict (ia->dict, &s);
1590       intro_append_syntax (ia, &s);
1591     }
1592   else
1593     {
1594       return sheet_spec_gen_syntax (ia);
1595     }
1596
1597   return ds_cstr (&s);
1598 }
1599
1600
1601 int
1602 psppire_import_assistant_run (PsppireImportAssistant *asst)
1603 {
1604   g_main_loop_run (asst->main_loop);
1605   return asst->response;
1606 }
1607