/[csv2latex]/csv2latex.c
ViewVC logotype

Contents of /csv2latex.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 13 - (show annotations)
Sat Apr 17 17:27:58 2010 UTC (9 years, 7 months ago) by ben
File MIME type: text/plain
File size: 17289 byte(s)
workaround on google docs export
1 /*
2 * csv2latex.c, copyright © 2002- Benoît Rouits <brouits@free.fr>
3 *
4 *********************************************************
5 * csv2latex translates a .csv file to a LaTex document. *
6 *********************************************************
7 *
8 * This program is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU General Public License
10 * as published by the Free Software Foundation; version 2 only
11 * of the License.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor,
21 * Boston, MA 02110-1301, USA.
22 *
23 * see the COPYING file included in the csv2latex package or
24 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt
25 *
26 */
27
28 #include <stdlib.h>
29 #include <stdio.h>
30 #include <string.h>
31 #include <libgen.h>
32 #include <getopt.h>
33 #include <unistd.h>
34 #include <ctype.h>
35 #include "version.h"
36
37 typedef struct {
38 char* tab; /* actual escapes */
39 int size; /* escape tab len */
40 } texcape;
41
42 typedef struct {
43 char block; /* CSV delimitor if any */
44 char sep; /* CSV separator */
45 unsigned int cols; /* CSV columns */
46 unsigned int chars; /* CSV max data length */
47 unsigned int rows; /* CSV total number of lines */
48 char pos; /* position in cell (align) */
49 unsigned int lines; /* rows per LaTeX tabular */
50 unsigned int guess; /* guess or not the CSV format */
51 unsigned int header; /* put LaTeX document header or not */
52 unsigned int red; /* table reduction level (from 1 to 4)*/
53 unsigned int longtable; /* use package longtable */
54 unsigned int escape; /* escape TeX control chars or not */
55 unsigned int repeat; /* repeat table headers for each LaTeX table section or not */
56 unsigned int vlines; /* insert vertical lines between columns or not */
57 unsigned int hlines; /* insert horizontal lines between rows or not */
58 char* clrrow; /* row graylevel (from 0 to 1) */
59 texcape* tex; /* TeX escapes */
60 } config;
61
62 #define MAXUINT ((unsigned int)(-1))
63
64 void rtfm(char* prog) {
65 printf("%s translates a csv file to a LaTeX file\n", basename(prog));
66 printf("Example: %s january_stats.csv > january_stats.tex\n", basename(prog));
67 printf("Usage: %s [--nohead] (LaTeX) no document header: useful for inclusion\n", basename(prog));
68 printf(" [--longtable] (LaTeX) use package longtable: useful for long input\n");
69 printf(" [--noescape] (LaTeX) do not escape text: useful for mixed CSV/TeX input\n");
70 printf(" [--guess] (CSV) guess separator and block |\n"
71 " [--separator <(c)omma|(s)emicolon|(t)ab|s(p)ace|co(l)on>] (CSV's comma)\n"
72 " [--block <(q)uote|(d)ouble|(n)one>] (CSV) block delimiter (e.g: none)\n");
73 printf(" [--lines n] (LaTeX) rows per table: useful for long tabulars\n");
74 printf(" [--position <l|c|r>] (LaTeX) text align in cells\n");
75 printf(" [--colorrows graylevel] (LaTeX) alternate gray rows (e.g: 0.75)\n");
76 printf(" [--reduce level] (LaTeX) reduce table size (e.g: 1)\n");
77 printf(" [--repeatheader] (LaTeX) repeat table header (for long tables)\n");
78 printf(" [--nohlines] (LaTeX) don't put hline between table rows\n");
79 printf(" [--novlines] (LaTeX) don't put vline between columns\n");
80 printf(" csv_file.csv\n");
81 printf("The \"longtable\" option needs the {longtable} LaTeX package\n");
82 printf("The \"colorrows\" option needs the {colortbl} LaTeX package\n");
83 printf("The \"reduce\" option needs the {relsize} LaTeX package\n");
84 return;
85 }
86
87 config* parseOptions (config* conf, int argc, char **argv) {
88 /* thx to <vfebvre@lautre.net> */
89 int opt;
90 int tmp;
91
92 #if defined USE_GETOPT
93 #else
94 int longopt_index = 0;
95 static struct option long_options[] = {
96 {"help", 0, NULL, 'h'},
97 {"guess", 0, NULL, 'g'},
98 {"block", 1, NULL, 'b'},
99 {"lines", 1, NULL, 'l'},
100 {"noescape", 0, NULL, 'x'},
101 {"nohead", 0, NULL, 'n'},
102 {"version", 0, NULL, 'v'},
103 {"position", 1, NULL, 'p'},
104 {"separator", 1, NULL, 's'},
105 {"colorrows", 1, NULL, 'c'},
106 {"reduce", 1, NULL, 'r'},
107 {"longtable", 0, NULL, 't'},
108 {"repeatheader",0, NULL, 'e'},
109 {"novlines", 0, NULL, 'y'},
110 {"nohlines", 0, NULL, 'z'},
111 {NULL, 0, NULL, 0} /* marks end-of-list */
112 };
113 #endif
114 #if defined USE_GETOPT
115 while ((opt = getopt (argc, argv, "hvgnxteyz?b:l:p:s:c:r:")) != EOF) {
116 #else
117 while ((opt = getopt_long (argc, argv, "hvgnxteyz?b:l:p:s:c:r:", long_options, &longopt_index)) > 0) {
118 #endif
119 switch (opt) {
120 case '?':
121 case 'h':
122 rtfm (argv[0]);
123 exit (EXIT_SUCCESS);
124 break;
125 case 'g': /* guess the CSV */
126 conf->guess=1;
127 break;
128 case 't': /* use package longtable */ /* thx to <Christof.Bodner@infineon.com> */
129 conf->longtable=1;
130 break;
131 case 'b': /* csv block delimiter */
132 if(optarg[0] == 'q')
133 conf->block = '\'';
134 else if(optarg[0] == 'd')
135 conf->block = '"';
136 else if(optarg[0] == 'n')
137 conf->block = 0; /* no block delimiter */
138 break;
139 case 'l': /* number of lines per TeX tabulars */
140 if(isdigit(optarg[0])) {
141 conf->lines=atoi(optarg);
142 } else {
143 fprintf(stderr,
144 "option \"lines\" need a positive integer value\n");
145 exit(EXIT_FAILURE);
146 }
147 break;
148 case 'n':
149 conf->header=0;
150 break;
151 case 'x':
152 conf->escape=0;
153 break;
154 case 'v': /* version */
155 printf ("%s © 2002- Benoît Rouits <brouits@free.fr>\n"
156 "\tVersion %s (%s)\n", PACKAGE, VERSION, RELEASE_DATE);
157 exit (EXIT_SUCCESS);
158 break;
159 case 'p': /* LaTeX position in cell */
160 conf->pos=optarg[0]; /* position char in cell */
161 break;
162 case 's': /* csv block separator */
163 if(optarg[0] == 'c')
164 conf->sep = ',';
165 else if(optarg[0] == 's')
166 conf->sep = ';';
167 else if(optarg[0] == 't')
168 conf->sep = '\t';
169 else if(optarg[0] == 'p')
170 conf->sep = ' ';
171 else if(optarg[0] == 'l')
172 conf->sep = ':';
173 break;
174 case 'c': /* color rows (thx to <jcorso@cse.Buffalo.EDU>) */
175 if(isdigit(optarg[0])) {
176 conf->clrrow = (char*)malloc(strlen(optarg)+1);
177 strcpy(conf->clrrow,optarg);
178 } else {
179 fprintf(stderr,
180 "option \"colorrows\" needs a real value between 0 and 1\n");
181 exit(EXIT_FAILURE);
182 }
183 break;
184 case 'r': /* reduce table size (original idea thx to <boaz.gezer@gmail.com>) */
185
186 if(isdigit(optarg[0])) {
187 tmp = atoi(optarg);
188 conf->red=(tmp>4)?4:(tmp<0)?0:tmp; /* [1-4] */
189 } else {
190 fprintf(stderr,
191 "option \"reduce\" needs an integer value between 1 and 4\n");
192 exit(EXIT_FAILURE);
193 }
194 break;
195 case 'e': /*repeat table header for each table section*/
196 conf->repeat=1;
197 break;
198 case 'y': /*don't draw vlines between columns*/
199 conf->vlines=0;
200 break;
201 case 'z': /*don't draw hlines between rows*/
202 conf->hlines=0;
203 break;
204 }
205 }
206 return conf;
207 }
208 int guessCSV(config* conf, FILE* in) {
209 /* guess the block delimiter and the csv separator */
210 int token;
211
212 token=getc(in); /* first char is block delimiter */
213 if(token == EOF) {
214 fprintf(stderr,"ERROR: emtpy file ?\n");
215 return(-1);
216 } else if (ispunct(token) || token == ' ') {
217 /* found first block delimiter, act this way */
218 conf->block=token;
219 fprintf(stderr,"Guessed '%c' as Block Delimiter\n",
220 conf->block);
221 /* stream file while token is printable data */
222 while((token=getc(in)) != conf->block &&
223 token != '\n' &&
224 token != EOF)
225 {/* getc has been done */}
226 if(token == conf->block){
227 /* second delimiter : next is separator */
228 conf->sep=getc(in);
229 fprintf(stderr,"Guessed '%c' as Separator\n",
230 conf->sep);
231 return(0);
232 }else{
233 return (-1); /* what else ? */
234 }
235 }else{ /* no block delimiter, act this way */
236 conf->block=0;
237 fprintf(stderr,"Guessed No Block Delimiter\n");
238 /* stream file while input is not a control char */
239 while(isalnum((token=getc(in))) &&
240 token != '\n' &&
241 token != EOF)
242 {/* getc has been done */}
243 /* guess CSV separator */
244 if(ispunct(token) || token == '\t' || token == ' '){
245 conf->sep=token;
246 fprintf(stderr,"Guessed %c as Separator\n", conf->sep);
247 return(0);
248 } else { /* did not found any separator */
249 fprintf(stderr,"ERROR: Did not guess any Separator!\n");
250 return(-1);
251 }
252 }
253 return(0);
254 }
255
256 void getMaximums(config* conf, FILE* in) {
257 /* gets the number of cols and chars of a csv file assuming a separator */
258 int token=0;
259 unsigned int curcol=0;
260 unsigned int curchar=0;
261 unsigned int inblock=0;
262 /* init */
263 conf->chars=0;
264 conf->cols=0;
265 conf->rows=0;
266
267 while (token != EOF) {
268 token=getc(in);
269
270 /* EOF ? */
271 if (token == EOF) {
272 continue;
273 }
274
275 /* decide the maximums */
276 if (token == '\n') {
277 curcol++;
278 conf->cols=(conf->cols<curcol)?curcol:conf->cols;
279 conf->chars=(conf->chars<curchar)?curchar:conf->chars;
280 conf->rows++;
281 curcol=0;
282 curchar=0;
283 inblock=0; /* reset block state */
284 continue;
285 }
286
287 /* enter/quit a block */
288 if (conf->block && token == conf->block) {
289 inblock=!inblock;
290 continue;
291 }
292
293 /* count cols in current line */
294 if (token == conf->sep && ((conf->block && !inblock) || !conf->block)) {
295 curcol++;
296 continue;
297 }
298
299 /* count chars in current cell */
300 if (token != conf->block && ((conf->block && inblock) || !conf->block)) {
301 curchar++;
302 continue;
303 }
304 }
305 return;
306 }
307
308 void doTeXsub(config* conf, char newsep, FILE* in, FILE* out) {
309 /* substitutes CSV sep by LaTeX newsep and some TeX code */
310 int token=0;
311 int max;
312 int numcols;
313 unsigned int lines;
314 int inblock=0;
315 int csvrows;
316 int firstrow=1;
317 int nosep=0;
318 int token1=0;
319 int token2=0;
320 char headerrow[1000];
321 headerrow[0]='\0';
322
323 max=numcols=conf->cols;
324 csvrows=conf->rows;
325 /* choose infinity when conf->lines is 0 */
326 lines=(conf->lines)?conf->lines:MAXUINT;
327
328 while (token!=EOF) {
329 token2=token1; /* second last character, used for detection of quotation marks */
330 token1=token; /* last character, used for detection of quotation marks */
331 token=getc(in);
332
333 /* EOF ? */
334 if (token == EOF) {
335 continue;
336 }
337
338 /* new line ? */
339 if (token == '\n') {
340 inblock = 0; /* close block if any */
341 /* fill empty cols if any */
342 while (numcols > 1) {
343 putc(newsep,out);
344 numcols--;
345 }
346 if (!(firstrow && (conf->longtable && conf->repeat))) {
347 fprintf(out,"\\\\\n"); /* TeX new line */
348 } else { /* first row and repeat and longtable */
349 fprintf(out, "%s\\\\\n", headerrow);
350 }
351 if(conf->hlines) {
352 fprintf(out,"\\hline\n"); /* TeX draw hline */
353 }
354 if (firstrow && (conf->longtable && conf->repeat)) {
355 fprintf(out, "%s", "\\endhead\n");
356 }
357 if(firstrow && conf->repeat) {
358 char tmp[12];
359 sprintf(tmp,(conf->hlines?"\\\\\n\\hline":"\\\\\n"));
360 strcat(headerrow,tmp);
361 }
362 firstrow=0;
363 numcols=max; /* reset numcols */
364 lines--;
365 csvrows--;
366 /* put a colored row or not (alternate) */
367 if (conf->clrrow && (lines % 2)) {
368 fprintf(out,"\\colorrow ");
369 }
370 /* if the LaTeX tabular is full create a new one, except if no more row or this is a long table */
371 if (!lines && csvrows && !conf->longtable) {
372 fprintf(out,"\\end{tabular}\n");
373 fprintf(out, "\\newline\n");
374 fprintf(out,"\\begin{tabular}{");
375 if(conf->vlines) {
376 putc('|',out);
377 }
378 while(numcols--) {
379 putc(conf->pos, out);
380 if(conf->vlines) {
381 putc('|',out);
382 }
383 }
384 fprintf(out, "}\n");
385 if(conf->hlines) {
386 fprintf(out, "\\hline\n");
387 }
388 if(conf->repeat && !conf->longtable) {
389 fprintf(out,"%s",headerrow);
390 putc('\n',out);
391 }
392 numcols=max;
393 lines=(conf->lines)?conf->lines:MAXUINT;
394 }
395 /* else end of CSV data */
396 continue;
397 }
398
399 /* if commas in cells */
400 /* thx to <florian@heinze.at> */
401 if (token == '\"'){
402 if (nosep==0)
403 nosep=1;
404 else
405 nosep=0;
406 }
407
408 /* new column ? */
409 if ((token == conf->sep && ((conf->block && !inblock) || !conf->block)) && nosep == 0) {
410 if (!(firstrow && (conf->longtable && conf->repeat)))
411 putc(newsep,out);
412 numcols--;
413 if(firstrow && conf->repeat)
414 {
415 char tmp[2];
416 tmp[0]=newsep;
417 tmp[1]='\0';
418 strcat(headerrow,tmp);
419 }
420 continue;
421 }
422
423 /* enter/quit a block ? */
424 if (conf->block && token == conf->block) {
425 inblock=!inblock;
426 continue;
427 }
428
429 /* data ? */
430 if ((token != conf->block && ((conf->block && inblock) || !conf->block)) && ((token=='\"' && token1=='\"' && token2=='\"') || token!='\"')) {
431 /* look for special TeX char to escape */
432 /* FIXME: put all that into a subroutine */
433 int i=0;
434 if (conf->escape)
435 for (i=0;i < conf->tex->size; i++) {
436 if (token == conf->tex->tab[i]) {
437 switch (token) {
438 case '\\':
439 fprintf(out, "\\textbackslash{}");
440 if(firstrow && conf->repeat)
441 {
442 char tmp[17];
443 sprintf(tmp,"\\textbackslash{}");
444 strcat(headerrow,tmp);
445 }
446 break;
447 default:
448 fprintf(out, "\\%c", token);
449 if(firstrow && conf->repeat)
450 {
451 char tmp[3];
452 tmp[0]='\\';
453 tmp[1]=token;
454 tmp[2]='\0';
455 strcat(headerrow,tmp);
456 }
457 break;
458 }
459 break; /* there was some escaping */
460 }
461 }
462 /* or print raw char */
463 if ( (i >= conf->tex->size) || (!conf->escape) ) {
464 if (!(firstrow && (conf->longtable && conf->repeat)))
465 putc(token, out); /* do not print the header twice */
466 if(firstrow && conf->repeat)
467 {
468 char tmp[2];
469 tmp[0]=token;
470 tmp[1]='\0';
471 strcat(headerrow,tmp);
472 }
473 }
474 continue;
475 }
476 /* do nothing if unexpected char: just loop */
477 }
478 return;
479 }
480
481 void doTeXdoc(config* conf, FILE* in, FILE* out) {
482 /* prepares the LaTeX tabular layout */
483 int maxcols;
484 int numcols;
485 char* relsize[5] = {"0", "0.5", "1", "2", "4"}; /* LaTeX relsize good values */
486 char* tabcolsep[5] = {"0", "0.05", "0.1", "0.2", "0.4"}; /* LaTeX tabcolsep good values */
487
488 numcols=maxcols=conf->cols;
489 if(conf->header){
490 fprintf(out, "\\documentclass[a4paper]{article}\n");
491 fprintf(out, "\\usepackage[T1]{fontenc}\n");
492 fprintf(out, "\\usepackage[latin1]{inputenc}\n");
493 if (conf->red){
494 fprintf(out,"\\usepackage{relsize}\n");
495 }
496 if (conf->clrrow){
497 fprintf(out,"\\usepackage{colortbl}\n");
498 }
499 if (conf->longtable){
500 fprintf(out,"\\usepackage{longtable}\n");
501 }
502 fprintf(out, "\\begin{document}\n");
503 }
504 if (conf->clrrow){
505 fprintf(out,"\\def\\colorrow{\\rowcolor[gray]{%s}}\n",
506 conf->clrrow);
507 }
508 if (conf->red){
509 fprintf(out,"\\relsize{-%s}\n", relsize[conf->red]);
510 fprintf(out,"\\addtolength\\tabcolsep{-%sem}\n", tabcolsep[conf->red]);
511 }
512 if (conf->longtable)
513 {
514 fprintf(out, "\\begin{longtable}{");
515 if(conf->vlines)
516 putc('|',out);
517 }
518 else
519 {
520 fprintf(out, "\\begin{tabular}{");
521 if(conf->vlines)
522 putc('|',out);
523 }
524 while(numcols--)
525 {
526 fprintf(out, "%c",conf->pos); /* position in cell */
527 if(conf->vlines)
528 putc('|',out);
529 }
530 fprintf(out, "}\n");
531 if(conf->hlines)
532 fprintf(out, "\\hline\n");
533 doTeXsub(conf, '&', in, out); /* & is LaTeX separator */
534 if (conf->longtable) {
535 fprintf(out, "\\end{longtable}\n");
536 } else {
537 fprintf(out, "\\end{tabular}\n");
538 }
539 if (conf->red){
540 fprintf(out,"\\addtolength\\tabcolsep{+%sem}\n", tabcolsep[conf->red]);
541 fprintf(out,"\\relsize{+%s}\n", relsize[conf->red]);
542 }
543 if(conf->header){
544 fprintf(out, "\\end{document}\n");
545 }
546 return;
547 }
548
549 int main (int argc, char **argv) {
550 FILE* fp;
551 config* conf;
552
553 extern int optind, opterr, optopt;
554
555 if(argc == 1){
556 rtfm(argv[0]);
557 exit(EXIT_SUCCESS);
558 }
559 conf=(config*)malloc(sizeof(config));
560 /* defaults (ensure init): */
561 conf->cols=1; /* CSV: if getMaximums fails */
562 conf->rows=0; /* CSV: must be 0 */
563 conf->chars=0; /* CSV: must be 0 */
564 conf->pos='l'; /* usual; LaTeX */
565 conf->lines=40; /* usual; LaTeX */
566 conf->guess=0; /* usual */
567 conf->sep=','; /* default; csv */
568 conf->block=0; /* default; csv */
569 conf->header=1; /* usual; LaTeX */
570 conf->escape=1; /* usual; LaTeX */
571 conf->clrrow=NULL; /* default; LaTeX */
572 conf->red=0; /* default; LaTeX */
573 conf->longtable=0; /* default; without package longtable */
574 conf->repeat=0; /* default; do not repeat the header row */
575 conf->vlines=1; /* default; draw lines between columns */
576 conf->hlines=1; /* default; draw lines between rows */
577
578 /* TeX charaters to escape */
579 conf->tex=(texcape*)malloc(sizeof(texcape));
580 conf->tex->tab = "\\_#$%^&{}~";
581 conf->tex->size = 10;
582
583 conf=parseOptions(conf, argc, argv);
584 fp=fopen(argv[optind],"r");
585 if (!fp){
586 fprintf(stderr,"Can't open file %s\n", argv[optind]);
587 exit(EXIT_FAILURE);
588 }
589 if(conf->guess){
590 if(guessCSV(conf, fp)){
591 fprintf(stderr,"Please run again by using -- delimiter (if any) and --separator\n");
592 exit(EXIT_FAILURE);
593 }
594 rewind(fp);
595 }
596 getMaximums(conf, fp);
597 rewind(fp);
598 doTeXdoc(conf, fp, stdout);
599 free(conf->tex);
600 if (conf->clrrow) free(conf->clrrow); conf->clrrow=NULL;
601 free(conf);
602 fclose(fp);
603 return 0;
604 }

  ViewVC Help
Powered by ViewVC 1.1.26