Import assistant: Simplify the construction of the introductory page
[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_DIR))
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           intro_page_create (ia);
576           first_line_page_create (ia);
577           separators_page_create (ia);
578         }
579       else
580         {
581           sheet_spec_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
925   if (gtk_grid_get_child_at (GTK_GRID (table), 1, 2) == NULL)
926     {
927       GtkWidget *hbox_percent = psppire_scanf_new (_("Only the first %3d %% of file (approximately)"),
928                                                    &ia->percent_spin);
929
930       gtk_grid_attach (GTK_GRID (table), hbox_percent,
931                        1, 2,
932                        1, 1);
933     }
934
935   gtk_widget_show_all (table);
936
937   on_intro_amount_changed (ia);
938 }
939
940 /* Initializes IA's intro substructure. */
941 static void
942 intro_page_create (PsppireImportAssistant *ia)
943 {
944   GtkBuilder *builder = ia->builder;
945
946   GtkWidget *w =  get_widget_assert (builder, "Intro");
947
948   ia->percent_spin = gtk_spin_button_new_with_range (0, 100, 10);
949
950
951   add_page_to_assistant (ia, w,  GTK_ASSISTANT_PAGE_CONTENT, _("Select the Lines to Import"));
952
953   ia->all_cases_button = get_widget_assert (builder, "import-all-cases");
954
955   ia->n_cases_button = get_widget_assert (builder, "import-n-cases");
956
957   ia->percent_button = get_widget_assert (builder, "import-percent");
958
959   g_signal_connect_swapped (ia->all_cases_button, "toggled",
960                             G_CALLBACK (on_intro_amount_changed), ia);
961   g_signal_connect_swapped (ia->n_cases_button, "toggled",
962                             G_CALLBACK (on_intro_amount_changed), ia);
963   g_signal_connect_swapped (ia->percent_button, "toggled",
964                             G_CALLBACK (on_intro_amount_changed), ia);
965
966
967   g_object_set_data (G_OBJECT (w), "on-entering", intro_on_enter);
968   g_object_set_data (G_OBJECT (w), "on-reset", reset_intro_page);
969 }
970
971
972 GtkWidget *
973 psppire_import_assistant_new (GtkWindow *toplevel)
974 {
975   return GTK_WIDGET (g_object_new (PSPPIRE_TYPE_IMPORT_ASSISTANT,
976                                    /* Some window managers (notably ratpoison)
977                                       ignore the maximise command when a window is
978                                       transient.  This causes problems for this
979                                       window. */
980                                    /* "transient-for", toplevel, */
981                                    NULL));
982 }
983
984
985
986 \f
987
988 static void
989 set_quote_list (GtkComboBox *cb)
990 {
991   GtkListStore *list =  gtk_list_store_new (1, G_TYPE_STRING);
992   GtkTreeIter iter;
993   gint i;
994   const gchar *separator[3] = {"'\"", "\'", "\""};
995
996   for (i = 0; i < 3; i++)
997     {
998       const gchar *s = separator[i];
999
1000       /* Add a new row to the model */
1001       gtk_list_store_append (list, &iter);
1002       gtk_list_store_set (list, &iter,
1003                           0, s,
1004                           -1);
1005
1006     }
1007
1008   gtk_combo_box_set_model (GTK_COMBO_BOX (cb), GTK_TREE_MODEL (list));
1009   g_object_unref (list);
1010
1011   gtk_combo_box_set_entry_text_column (cb, 0);
1012 }
1013
1014
1015
1016 /* Chooses a name for each column on the separators page */
1017 static void
1018 choose_column_names (PsppireImportAssistant *ia)
1019 {
1020   int i;
1021   unsigned long int generated_name_count = 0;
1022   dict_clear (ia->dict);
1023
1024   for (i = 0; i < gtk_tree_model_get_n_columns (ia->delimiters_model) - 1; ++i)
1025     {
1026       const gchar *candidate_name = NULL;
1027
1028       if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (ia->variable_names_cb)))
1029         {
1030           candidate_name = psppire_delimited_text_get_header_title (PSPPIRE_DELIMITED_TEXT (ia->delimiters_model), i);
1031         }
1032
1033       char *name = dict_make_unique_var_name (ia->dict,
1034                                               candidate_name,
1035                                               &generated_name_count);
1036
1037       dict_create_var_assert (ia->dict, name, 0);
1038       free (name);
1039     }
1040 }
1041
1042
1043
1044 /* Called when the user toggles one of the separators
1045    checkboxes. */
1046 static void
1047 on_separator_toggle (GtkToggleButton *toggle UNUSED,
1048                      PsppireImportAssistant *ia)
1049 {
1050   int i;
1051   GSList *delimiters = NULL;
1052   for (i = 0; i < SEPARATOR_CNT; i++)
1053     {
1054       const struct separator *s = &separators[i];
1055       GtkWidget *button = get_widget_assert (ia->builder, s->name);
1056       if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (button)))
1057         {
1058           delimiters = g_slist_prepend (delimiters,  GINT_TO_POINTER (s->c));
1059         }
1060     }
1061
1062   g_object_set (ia->delimiters_model, "delimiters", delimiters, NULL);
1063
1064   revise_fields_preview (ia);
1065 }
1066
1067
1068 /* Called when the user changes the entry field for custom
1069    separators. */
1070 static void
1071 on_separators_custom_entry_notify (GObject *gobject UNUSED,
1072                                    GParamSpec *arg1 UNUSED,
1073                                    PsppireImportAssistant *ia)
1074 {
1075   revise_fields_preview (ia);
1076 }
1077
1078 /* Called when the user toggles the checkbox that enables custom
1079    separators. */
1080 static void
1081 on_separators_custom_cb_toggle (GtkToggleButton *custom_cb,
1082                                 PsppireImportAssistant *ia)
1083 {
1084   bool is_active = gtk_toggle_button_get_active (custom_cb);
1085   gtk_widget_set_sensitive (ia->custom_entry, is_active);
1086   revise_fields_preview (ia);
1087 }
1088
1089 /* Called when the user changes the selection in the combo box
1090    that selects a quote character. */
1091 static void
1092 on_quote_combo_change (GtkComboBox *combo, PsppireImportAssistant *ia)
1093 {
1094   //  revise_fields_preview (ia);
1095 }
1096
1097 /* Called when the user toggles the checkbox that enables
1098    quoting. */
1099 static void
1100 on_quote_cb_toggle (GtkToggleButton *quote_cb, PsppireImportAssistant *ia)
1101 {
1102   bool is_active = gtk_toggle_button_get_active (quote_cb);
1103   gtk_widget_set_sensitive (ia->quote_combo, is_active);
1104   revise_fields_preview (ia);
1105 }
1106
1107 /* Initializes IA's separators substructure. */
1108 static void
1109 separators_page_create (PsppireImportAssistant *ia)
1110 {
1111   GtkBuilder *builder = ia->builder;
1112
1113   size_t i;
1114
1115   GtkWidget *w = get_widget_assert (builder, "Separators");
1116
1117   g_object_set_data (G_OBJECT (w), "on-entering", prepare_separators_page);
1118   g_object_set_data (G_OBJECT (w), "on-reset", prepare_separators_page);
1119
1120   add_page_to_assistant (ia, w,   GTK_ASSISTANT_PAGE_CONTENT, _("Choose Separators"));
1121
1122   ia->custom_cb = get_widget_assert (builder, "custom-cb");
1123   ia->custom_entry = get_widget_assert (builder, "custom-entry");
1124   ia->quote_combo = get_widget_assert (builder, "quote-combo");
1125   ia->quote_entry = GTK_ENTRY (gtk_bin_get_child (GTK_BIN (ia->quote_combo)));
1126   ia->quote_cb = get_widget_assert (builder, "quote-cb");
1127
1128   set_quote_list (GTK_COMBO_BOX (ia->quote_combo));
1129
1130   if (ia->fields_tree_view == NULL)
1131     {
1132       GtkWidget *scroller = get_widget_assert (ia->builder, "fields-scroller");
1133       ia->fields_tree_view = gtk_tree_view_new ();
1134       gtk_container_add (GTK_CONTAINER (scroller), GTK_WIDGET (ia->fields_tree_view));
1135       gtk_widget_show_all (scroller);
1136     }
1137
1138   g_signal_connect (ia->quote_combo, "changed",
1139                     G_CALLBACK (on_quote_combo_change), ia);
1140   g_signal_connect (ia->quote_cb, "toggled",
1141                     G_CALLBACK (on_quote_cb_toggle), ia);
1142   g_signal_connect (ia->custom_entry, "notify::text",
1143                     G_CALLBACK (on_separators_custom_entry_notify), ia);
1144   g_signal_connect (ia->custom_cb, "toggled",
1145                     G_CALLBACK (on_separators_custom_cb_toggle), ia);
1146   for (i = 0; i < SEPARATOR_CNT; i++)
1147     g_signal_connect (get_widget_assert (builder, separators[i].name),
1148                       "toggled", G_CALLBACK (on_separator_toggle), ia);
1149
1150 }
1151
1152
1153
1154 \f
1155
1156
1157 static struct casereader_random_class my_casereader_class;
1158
1159 static struct ccase *
1160 my_read (struct casereader *reader, void *aux, casenumber idx)
1161 {
1162   PsppireImportAssistant *ia = PSPPIRE_IMPORT_ASSISTANT (aux);
1163   GtkTreeModel *tm = GTK_TREE_MODEL (ia->delimiters_model);
1164
1165   GtkTreePath *tp = gtk_tree_path_new_from_indices (idx, -1);
1166
1167   const struct caseproto *proto = casereader_get_proto (reader);
1168
1169   GtkTreeIter iter;
1170   struct ccase *c = NULL;
1171   if (gtk_tree_model_get_iter (tm, &iter, tp))
1172     {
1173       c = case_create (proto);
1174       int i;
1175       for (i = 0 ; i < caseproto_get_n_widths (proto); ++i)
1176         {
1177           GValue value = {0};
1178           gtk_tree_model_get_value (tm, &iter, i + 1, &value);
1179
1180           const struct variable *var = dict_get_var (ia->dict, i);
1181
1182           const gchar *ss = g_value_get_string (&value);
1183
1184           union value *v = case_data_rw (c, var);
1185           char *xx = data_in (ss_cstr (ss),
1186                               "UTF-8",
1187                               var_get_write_format (var)->type,
1188                               v, var_get_width (var), "UTF-8");
1189
1190           /* if (xx) */
1191           /*   g_print ("%s:%d Err %s\n", __FILE__, __LINE__, xx); */
1192           free (xx);
1193           g_value_unset (&value);
1194         }
1195     }
1196
1197   gtk_tree_path_free (tp);
1198
1199   return c;
1200 }
1201
1202 static void
1203 my_destroy (struct casereader *reader, void *aux)
1204 {
1205   g_print ("%s:%d %p\n", __FILE__, __LINE__, reader);
1206 }
1207
1208 static void
1209 my_advance (struct casereader *reader, void *aux, casenumber cnt)
1210 {
1211   g_print ("%s:%d\n", __FILE__, __LINE__);
1212 }
1213
1214 static void
1215 foo (struct dictionary *dict, void *aux)
1216 {
1217   PsppireImportAssistant *ia = PSPPIRE_IMPORT_ASSISTANT (aux);
1218   g_print ("%s:%d\n", __FILE__, __LINE__);
1219
1220   struct caseproto *proto = caseproto_create ();
1221
1222   int i;
1223   for (i = 0 ; i < dict_get_var_cnt (ia->dict); ++i)
1224     {
1225       const struct variable *var = dict_get_var (ia->dict, i);
1226       proto = caseproto_add_width (proto, var_get_width (var));
1227     }
1228
1229
1230   gint n_rows = gtk_tree_model_iter_n_children (ia->delimiters_model, NULL);
1231
1232   struct casereader *reader =
1233     casereader_create_random (proto, n_rows, &my_casereader_class,  ia);
1234
1235
1236   PsppireDataStore *store = NULL;
1237
1238   g_object_get (ia->data_sheet, "data-model", &store, NULL);
1239
1240   psppire_data_store_set_reader (store, reader);
1241 }
1242
1243 /* Called just before the formats page of the assistant is
1244    displayed. */
1245 static void
1246 prepare_formats_page (PsppireImportAssistant *ia)
1247 {
1248   PsppireDict *dict = psppire_dict_new_from_dict (ia->dict);
1249   g_object_set (ia->var_sheet, "data-model", dict, NULL);
1250
1251   my_casereader_class.read = my_read;
1252   my_casereader_class.destroy = my_destroy;
1253   my_casereader_class.advance = my_advance;
1254
1255   struct caseproto *proto = caseproto_create ();
1256   int i;
1257
1258   struct fmt_guesser **fg = xcalloc (sizeof *fg, dict_get_var_cnt (ia->dict));
1259   for (i = 0 ; i < dict_get_var_cnt (ia->dict); ++i)
1260     {
1261       fg[i] = fmt_guesser_create ();
1262     }
1263
1264   gint n_rows = gtk_tree_model_iter_n_children (ia->delimiters_model, NULL);
1265
1266   GtkTreeIter iter;
1267   gboolean ok;
1268   for (ok = gtk_tree_model_get_iter_first (ia->delimiters_model, &iter);
1269        ok;
1270        ok = gtk_tree_model_iter_next (ia->delimiters_model, &iter))
1271     {
1272       for (i = 0 ; i < dict_get_var_cnt (ia->dict); ++i)
1273         {
1274           gchar *s = NULL;
1275           gtk_tree_model_get (ia->delimiters_model, &iter, i+1, &s, -1);
1276           fmt_guesser_add (fg[i], ss_cstr (s));
1277           free (s);
1278         }
1279     }
1280
1281   for (i = 0 ; i < dict_get_var_cnt (ia->dict); ++i)
1282     {
1283       struct fmt_spec fs;
1284       fmt_guesser_guess (fg[i], &fs);
1285
1286       fmt_fix (&fs, FMT_FOR_INPUT);
1287
1288       struct variable *var = dict_get_var (ia->dict, i);
1289
1290       int width = fmt_var_width (&fs);
1291
1292       var_set_width_and_formats (var, width,
1293                                  &fs, &fs);
1294
1295       proto = caseproto_add_width (proto, width);
1296       fmt_guesser_destroy (fg[i]);
1297     }
1298
1299   free (fg);
1300
1301   //  dict_set_change_callback (ia->dict, foo, ia);
1302
1303   struct casereader *reader =
1304     casereader_create_random (proto, n_rows, &my_casereader_class,  ia);
1305
1306   PsppireDataStore *store = psppire_data_store_new (dict);
1307   psppire_data_store_set_reader (store, reader);
1308
1309   g_object_set (ia->data_sheet, "data-model", store, NULL);
1310
1311
1312   gint pmax;
1313   g_object_get (get_widget_assert (ia->builder, "vpaned1"),
1314                 "max-position", &pmax, NULL);
1315
1316
1317   g_object_set (get_widget_assert (ia->builder, "vpaned1"),
1318                 "position", pmax / 2, NULL);
1319
1320   gtk_widget_show (ia->paste_button);
1321 }
1322
1323 static void
1324 formats_page_create (PsppireImportAssistant *ia)
1325 {
1326   GtkBuilder *builder = ia->builder;
1327
1328   GtkWidget *w = get_widget_assert (builder, "Formats");
1329   g_object_set_data (G_OBJECT (w), "on-entering", prepare_formats_page);
1330   g_object_set_data (G_OBJECT (w), "on-reset", reset_formats_page);
1331
1332   GtkWidget *vars_scroller = get_widget_assert (builder, "vars-scroller");
1333   if (ia->var_sheet == NULL)
1334     {
1335       ia->var_sheet = psppire_variable_sheet_new ();
1336
1337       gtk_container_add (GTK_CONTAINER (vars_scroller), ia->var_sheet);
1338
1339       ia->dict = dict_create (get_default_encoding ());
1340
1341       gtk_widget_show_all (vars_scroller);
1342     }
1343   GtkWidget *data_scroller = get_widget_assert (builder, "data-scroller");
1344   if (ia->data_sheet == NULL)
1345     {
1346       ia->data_sheet = psppire_data_sheet_new ();
1347
1348       gtk_container_add (GTK_CONTAINER (data_scroller), ia->data_sheet);
1349
1350       gtk_widget_show_all (data_scroller);
1351     }
1352
1353   add_page_to_assistant (ia, w,
1354                          GTK_ASSISTANT_PAGE_CONFIRM, _("Adjust Variable Formats"));
1355 }
1356
1357
1358 \f
1359
1360 static void
1361 separators_append_syntax (const PsppireImportAssistant *ia, struct string *s)
1362 {
1363   int i;
1364
1365   ds_put_cstr (s, "  /DELIMITERS=\"");
1366
1367   if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (get_widget_assert (ia->builder, "tab"))))
1368     ds_put_cstr (s, "\\t");
1369   for (i = 0; i < SEPARATOR_CNT; i++)
1370     {
1371       const struct separator *seps = &separators[i];
1372       GtkWidget *button = get_widget_assert (ia->builder, seps->name);
1373       if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (button)))
1374         {
1375           if (seps->c == '\t')
1376             continue;
1377
1378           ds_put_byte (s, seps->c);
1379         }
1380     }
1381   ds_put_cstr (s, "\"\n");
1382   if (!ds_is_empty (&ia->quotes))
1383     syntax_gen_pspp (s, "  /QUALIFIER=%sq\n", ds_cstr (&ia->quotes));
1384 }
1385
1386 static void
1387 formats_append_syntax (const PsppireImportAssistant *ia, struct string *s)
1388 {
1389   int i;
1390   int var_cnt;
1391
1392   g_return_if_fail (ia->dict);
1393
1394   ds_put_cstr (s, "  /VARIABLES=\n");
1395
1396   var_cnt = dict_get_var_cnt (ia->dict);
1397   for (i = 0; i < var_cnt; i++)
1398     {
1399       struct variable *var = dict_get_var (ia->dict, i);
1400       char format_string[FMT_STRING_LEN_MAX + 1];
1401       fmt_to_string (var_get_print_format (var), format_string);
1402       ds_put_format (s, "    %s %s%s\n",
1403                      var_get_name (var), format_string,
1404                      i == var_cnt - 1 ? "." : "");
1405     }
1406 }
1407
1408 static void
1409 first_line_append_syntax (const PsppireImportAssistant *ia, struct string *s)
1410 {
1411   gint first_case = 0;
1412   g_object_get (ia->delimiters_model, "first-line", &first_case, NULL);
1413
1414   if (first_case > 0)
1415     ds_put_format (s, "  /FIRSTCASE=%d\n", first_case + 1);
1416 }
1417
1418 static void
1419 intro_append_syntax (const PsppireImportAssistant *ia, struct string *s)
1420 {
1421   if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (ia->n_cases_button)))
1422     ds_put_format (s, "N OF CASES %d.\n",
1423                    gtk_spin_button_get_value_as_int (GTK_SPIN_BUTTON (ia->n_cases_spin)));
1424   else if (gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (ia->percent_button)))
1425     ds_put_format (s, "SAMPLE %.4g.\n",
1426                    gtk_spin_button_get_value (GTK_SPIN_BUTTON (ia->percent_spin)) / 100.0);
1427 }
1428
1429
1430 /* Emits PSPP syntax to S that applies the dictionary attributes
1431    (such as missing values and value labels) of the variables in
1432    DICT.  */
1433 static void
1434 apply_dict (const struct dictionary *dict, struct string *s)
1435 {
1436   size_t var_cnt = dict_get_var_cnt (dict);
1437   size_t i;
1438
1439   for (i = 0; i < var_cnt; i++)
1440     {
1441       struct variable *var = dict_get_var (dict, i);
1442       const char *name = var_get_name (var);
1443       enum val_type type = var_get_type (var);
1444       int width = var_get_width (var);
1445       enum measure measure = var_get_measure (var);
1446       enum var_role role = var_get_role (var);
1447       enum alignment alignment = var_get_alignment (var);
1448       const struct fmt_spec *format = var_get_print_format (var);
1449
1450       if (var_has_missing_values (var))
1451         {
1452           const struct missing_values *mv = var_get_missing_values (var);
1453           size_t j;
1454
1455           syntax_gen_pspp (s, "MISSING VALUES %ss (", name);
1456           for (j = 0; j < mv_n_values (mv); j++)
1457             {
1458               if (j)
1459                 ds_put_cstr (s, ", ");
1460               syntax_gen_value (s, mv_get_value (mv, j), width, format);
1461             }
1462
1463           if (mv_has_range (mv))
1464             {
1465               double low, high;
1466               if (mv_has_value (mv))
1467                 ds_put_cstr (s, ", ");
1468               mv_get_range (mv, &low, &high);
1469               syntax_gen_num_range (s, low, high, format);
1470             }
1471           ds_put_cstr (s, ").\n");
1472         }
1473       if (var_has_value_labels (var))
1474         {
1475           const struct val_labs *vls = var_get_value_labels (var);
1476           const struct val_lab **labels = val_labs_sorted (vls);
1477           size_t n_labels = val_labs_count (vls);
1478           size_t i;
1479
1480           syntax_gen_pspp (s, "VALUE LABELS %ss", name);
1481           for (i = 0; i < n_labels; i++)
1482             {
1483               const struct val_lab *vl = labels[i];
1484               ds_put_cstr (s, "\n  ");
1485               syntax_gen_value (s, &vl->value, width, format);
1486               ds_put_byte (s, ' ');
1487               syntax_gen_string (s, ss_cstr (val_lab_get_escaped_label (vl)));
1488             }
1489           free (labels);
1490           ds_put_cstr (s, ".\n");
1491         }
1492       if (var_has_label (var))
1493         syntax_gen_pspp (s, "VARIABLE LABELS %ss %sq.\n",
1494                          name, var_get_label (var));
1495       if (measure != var_default_measure (type))
1496         syntax_gen_pspp (s, "VARIABLE LEVEL %ss (%ss).\n",
1497                          name, measure_to_syntax (measure));
1498       if (role != ROLE_INPUT)
1499         syntax_gen_pspp (s, "VARIABLE ROLE /%ss %ss.\n",
1500                          var_role_to_syntax (role), name);
1501       if (alignment != var_default_alignment (type))
1502         syntax_gen_pspp (s, "VARIABLE ALIGNMENT %ss (%ss).\n",
1503                          name, alignment_to_syntax (alignment));
1504       if (var_get_display_width (var) != var_default_display_width (width))
1505         syntax_gen_pspp (s, "VARIABLE WIDTH %ss (%d).\n",
1506                          name, var_get_display_width (var));
1507     }
1508 }
1509
1510
1511
1512 static char *
1513 sheet_spec_gen_syntax (PsppireImportAssistant *ia)
1514 {
1515   GtkBuilder *builder = ia->builder;
1516   GtkWidget *range_entry = get_widget_assert (builder, "cell-range-entry");
1517   GtkWidget *sheet_entry = get_widget_assert (builder, "sheet-entry");
1518   GtkWidget *rnc = get_widget_assert (builder, "readnames-checkbox");
1519   const gchar *range = gtk_entry_get_text (GTK_ENTRY (range_entry));
1520   int sheet_index = 1 + gtk_combo_box_get_active (GTK_COMBO_BOX (sheet_entry));
1521   gboolean read_names = gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON (rnc));
1522
1523   struct string s = DS_EMPTY_INITIALIZER;
1524
1525   char *filename;
1526   g_object_get (ia->text_file, "file-name", &filename, NULL);
1527   syntax_gen_pspp (&s,
1528                    "GET DATA"
1529                    "\n  /TYPE=%ss"
1530                    "\n  /FILE=%sq"
1531                    "\n  /SHEET=index %d"
1532                    "\n  /READNAMES=%ss",
1533                    (ia->spreadsheet->type == SPREADSHEET_GNUMERIC) ? "GNM" : "ODS",
1534                    filename,
1535                    sheet_index,
1536                    read_names ? "ON" : "OFF");
1537
1538   if (range && 0 != strcmp ("", range))
1539     {
1540       syntax_gen_pspp (&s,
1541                        "\n  /CELLRANGE=RANGE %sq", range);
1542     }
1543   else
1544     {
1545       syntax_gen_pspp (&s,
1546                        "\n  /CELLRANGE=FULL");
1547     }
1548
1549
1550   syntax_gen_pspp (&s, ".");
1551
1552
1553   return ds_cstr (&s);
1554 }
1555
1556
1557 gchar *
1558 psppire_import_assistant_generate_syntax (PsppireImportAssistant *ia)
1559 {
1560   struct string s = DS_EMPTY_INITIALIZER;
1561
1562   if (!ia->spreadsheet)
1563     {
1564       gchar *file_name = NULL;
1565       gchar *encoding = NULL;
1566       g_object_get (ia->text_file,
1567                     "file-name", &file_name,
1568                     "encoding", &encoding,
1569                     NULL);
1570
1571       if (file_name == NULL)
1572         return NULL;
1573
1574       syntax_gen_pspp (&s,
1575                        "GET DATA"
1576                        "\n  /TYPE=TXT"
1577                        "\n  /FILE=%sq\n",
1578                        file_name);
1579       if (encoding && strcmp (encoding, "Auto"))
1580         syntax_gen_pspp (&s, "  /ENCODING=%sq\n", encoding);
1581
1582       ds_put_cstr (&s,
1583                    "  /ARRANGEMENT=DELIMITED\n"
1584                    "  /DELCASE=LINE\n");
1585
1586       first_line_append_syntax (ia, &s);
1587       separators_append_syntax (ia, &s);
1588
1589       formats_append_syntax (ia, &s);
1590       apply_dict (ia->dict, &s);
1591       intro_append_syntax (ia, &s);
1592     }
1593   else
1594     {
1595       return sheet_spec_gen_syntax (ia);
1596     }
1597
1598   return ds_cstr (&s);
1599 }
1600
1601
1602 int
1603 psppire_import_assistant_run (PsppireImportAssistant *asst)
1604 {
1605   g_main_loop_run (asst->main_loop);
1606   return asst->response;
1607 }
1608