lexer: New function lex_ofs_representation().
[pspp] / src / language / utilities / title.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 1997-9, 2000, 2007, 2010, 2011 Free Software Foundation, Inc.
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 <ctype.h>
20 #include <stdlib.h>
21
22 #include "data/dataset.h"
23 #include "data/dictionary.h"
24 #include "data/variable.h"
25 #include "language/command.h"
26 #include "language/lexer/lexer.h"
27 #include "language/lexer/token.h"
28 #include "libpspp/message.h"
29 #include "libpspp/start-date.h"
30 #include "libpspp/version.h"
31 #include "output/driver.h"
32
33 #include "gl/c-ctype.h"
34 #include "gl/xalloc.h"
35
36 #include "gettext.h"
37 #define _(msgid) gettext (msgid)
38
39 static int
40 parse_title (struct lexer *lexer, void (*set_title) (const char *))
41 {
42   if (lex_token (lexer) == T_STRING)
43     {
44       set_title (lex_tokcstr (lexer));
45       lex_get (lexer);
46     }
47   else
48     {
49       int start_ofs = lex_ofs (lexer);
50       while (lex_token (lexer) != T_ENDCMD)
51         lex_get (lexer);
52
53       /* Get the raw representation of all the tokens, including any space
54          between them, and use it as the title. */
55       char *title = lex_ofs_representation (lexer, start_ofs,
56                                             lex_ofs (lexer) - 1);
57       set_title (title);
58       free (title);
59     }
60   return CMD_SUCCESS;
61 }
62
63 int
64 cmd_title (struct lexer *lexer, struct dataset *ds UNUSED)
65 {
66   return parse_title (lexer, output_set_title);
67 }
68
69 int
70 cmd_subtitle (struct lexer *lexer, struct dataset *ds UNUSED)
71 {
72   return parse_title (lexer, output_set_subtitle);
73 }
74
75 /* Performs the FILE LABEL command. */
76 int
77 cmd_file_label (struct lexer *lexer, struct dataset *ds)
78 {
79   if (!lex_force_string (lexer))
80     return CMD_FAILURE;
81
82   dict_set_label (dataset_dict (ds), lex_tokcstr (lexer));
83   lex_get (lexer);
84
85   return CMD_SUCCESS;
86 }
87
88 /* Performs the DOCUMENT command. */
89 int
90 cmd_document (struct lexer *lexer, struct dataset *ds)
91 {
92   struct dictionary *dict = dataset_dict (ds);
93   char *trailer;
94
95   if (!lex_force_string (lexer))
96     return CMD_FAILURE;
97
98   while (lex_is_string (lexer))
99     {
100       dict_add_document_line (dict, lex_tokcstr (lexer), true);
101       lex_get (lexer);
102     }
103
104   trailer = xasprintf (_("   (Entered %s)"), get_start_date ());
105   dict_add_document_line (dict, trailer, true);
106   free (trailer);
107
108   return CMD_SUCCESS;
109 }
110
111 /* Performs the ADD DOCUMENTS command. */
112 int
113 cmd_add_documents (struct lexer *lexer, struct dataset *ds)
114 {
115   return cmd_document (lexer, ds);
116 }
117
118 /* Performs the DROP DOCUMENTS command. */
119 int
120 cmd_drop_documents (struct lexer *lexer UNUSED, struct dataset *ds)
121 {
122   dict_clear_documents (dataset_dict (ds));
123   return CMD_SUCCESS;
124 }