pdf.c

   1 #include <stdarg.h>
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <string.h>
   5 #include "post.h"
   6
   7 static char *pdf_title;         /* document title */
   8 static int pdf_width;           /* page width */
   9 static int pdf_height;          /* page height */
  10 static int pdf_pages;           /* pages object id */
  11 static int pdf_root;            /* root object id */
  12 static int pdf_pos;             /* current pdf file offset */
  13 static int *obj_off;            /* object offsets */
  14 static int obj_sz, obj_n;       /* number of pdf objects */
  15 static int *page_id;            /* page object ids */
  16 static int page_sz, page_n;     /* number of pages */
  17 static char **font_ps;          /* document font names */
  18 static int *font_id;            /* font object */
  19 static int *font_ct;            /* font content stream object */
  20 static int *font_ix;            /* font index */
  21 static int font_sz, font_n;     /* number of fonts */
  22
  23 static struct sbuf *pg;         /* current page contents */
  24 static int o_f, o_s, o_m;       /* font and size */
  25 static int o_h, o_v;            /* current user position */
  26 static int p_h, p_v;            /* current output position */
  27 static int o_pf, p_pf;          /* output and pdf fonts; indices into o_fs[] */
  28 static int p_f, p_s, p_m;       /* output font */
  29 static int o_queued;            /* queued character type */
  30 static int *o_fs;               /* page fonts */
  31 static int o_fsn, o_fssz;       /* page fonts */
  32
  33 /* print pdf output */
  34 static void pdfout(char *s, ...)
  35 {
  36         va_list ap;
  37         va_start(ap, s);
  38         pdf_pos += vprintf(s, ap);
  39         va_end(ap);
  40 }
  41
  42 /* allocate an object number */
  43 static int obj_map(void)
  44 {
  45         if (obj_n == obj_sz) {
  46                 obj_sz += 1024;
  47                 obj_off = mextend(obj_off, obj_n, obj_sz, sizeof(obj_off[0]));
  48         }
  49         return obj_n++;
  50 }
  51
  52 /* start the definition of an object */
  53 static int obj_beg(int id)
  54 {
  55         if (id <= 0)
  56                 id = obj_map();
  57         obj_off[id] = pdf_pos;
  58         pdfout("%d 0 obj\n", id);
  59         return id;
  60 }
  61
  62 /* end an object definition */
  63 static void obj_end(void)
  64 {
  65         pdfout("endobj\n\n");
  66 }
  67
  68 /* embed font; return stream object identifier */
  69 static int font_outdat(char *path, char *name, int ix)
  70 {
  71         struct sbuf *sb;
  72         FILE *fp;
  73         int c, i, id;
  74         for (i = 0; i < font_n; i++)
  75                 if (!strcmp(name, font_ps[i]) && font_ct[i] >= 0)
  76                         return font_ct[i];
  77         fp = fopen(path, "r");
  78         if (!fp)
  79                 return -1;
  80         sb = sbuf_make();
  81         c = fgetc(fp);
  82         for (i = 0; c != EOF; i++) {
  83                 sbuf_printf(sb, "%02x", c);
  84                 c = fgetc(fp);
  85                 if (i % 40 == 39 && c != EOF)
  86                         sbuf_chr(sb, '\n');
  87         }
  88         sbuf_str(sb, ">\n");
  89         fclose(fp);
  90         id = obj_beg(0);
  91         pdfout("<<\n");
  92         pdfout("  /Filter /ASCIIHexDecode\n");
  93         pdfout("  /Length %d\n", sbuf_len(sb));
  94         pdfout("  /Length1 %d\n", i);
  95         pdfout(">>\n");
  96         pdfout("stream\n");
  97         pdfout("%s", sbuf_buf(sb));
  98         pdfout("endstream\n");
  99         obj_end();
 100         sbuf_free(sb);
 101         return id;
 102 }
 103
 104 /* write the object corresponding to font font_id[f] */
 105 static void font_out(struct font *fn, int f)
 106 {
 107         int i;
 108         int enc_obj, des_obj;
 109         char *path = font_path(fn);
 110         char *ext = path ? strrchr(path, '.') : NULL;
 111         /* the encoding object */
 112         enc_obj = obj_beg(0);
 113         pdfout("<<\n");
 114         pdfout("  /Type /Encoding\n");
 115         pdfout("  /Differences [ 0");
 116         for (i = 0; i < 256; i++) {
 117                 struct glyph *g = font_glget(fn, font_ix[f] * 256 + i);
 118                 pdfout(" /%s", g ? g->id : ".notdef");
 119         }
 120         pdfout(" ]\n");
 121         pdfout(">>\n");
 122         obj_end();
 123         /* embedding the font */
 124         if (ext && !strcmp(".ttf", ext))
 125                 font_ct[f] = font_outdat(path, font_ps[f], font_ix[f]);
 126         /* the font descriptor */
 127         des_obj = obj_beg(0);
 128         pdfout("<<\n");
 129         pdfout("  /Type /FontDescriptor\n");
 130         pdfout("  /FontName /%s\n", font_ps[f]);
 131         pdfout("  /Flags 4\n");
 132         pdfout("  /MissingWidth 255\n");
 133         pdfout("  /StemV 100\n");
 134         pdfout("  /StemH 100\n");
 135         pdfout("  /CapHeight 100\n");
 136         pdfout("  /Ascent 100\n");
 137         pdfout("  /Descent 100\n");
 138         if (font_ct[f] >= 0)
 139                 pdfout("  /FontFile2 %d 0 R\n", font_ct[f]);
 140         pdfout(">>\n");
 141         obj_end();
 142         /* the font object */
 143         obj_beg(font_id[f]);
 144         pdfout("<<\n");
 145         pdfout("  /Type /Font\n");
 146         pdfout("  /Subtype /%s\n",
 147                 ext && !strcmp(".ttf", ext) ? "TrueType" : "Type1");
 148         pdfout("  /BaseFont /%s\n", font_ps[f]);
 149         pdfout("  /FirstChar 0\n");
 150         pdfout("  /LastChar 255\n");
 151         pdfout("  /Widths [");
 152         for (i = 0; i < 256; i++) {
 153                 struct glyph *g = font_glget(fn, font_ix[f] * 256 + i);
 154                 pdfout(" %d", (g ? g->wid : 0) * dev_res / 72);
 155         }
 156         pdfout(" ]\n");
 157         pdfout("  /FontDescriptor %d 0 R\n", des_obj);
 158         pdfout("  /Encoding %d 0 R\n", enc_obj);
 159         pdfout(">>\n");
 160         obj_end();
 161 }
 162
 163 static int font_put(struct font *fn, int ix)
 164 {
 165         int i;
 166         char *name = font_name(fn);
 167         for (i = 0; i < font_n; i++)
 168                 if (!strcmp(font_ps[i], font_name(fn)) && font_ix[i] == ix)
 169                         return i;
 170         if (font_n == font_sz) {
 171                 font_sz += 128;
 172                 font_id = mextend(font_id, font_n, font_sz, sizeof(font_id[0]));
 173                 font_ps = mextend(font_ps, font_n, font_sz, sizeof(font_ps[0]));
 174                 font_ix = mextend(font_ix, font_n, font_sz, sizeof(font_ix[0]));
 175                 font_ct = mextend(font_ct, font_n, font_sz, sizeof(font_ct[0]));
 176         }
 177         font_id[font_n] = obj_map();
 178         font_ix[font_n] = ix;
 179         font_ps[font_n] = malloc(strlen(name) + 1);
 180         font_ct[font_n] = -1;
 181         strcpy(font_ps[font_n], name);
 182         font_n++;
 183         font_out(fn, font_n - 1);
 184         return font_n - 1;
 185 }
 186
 187 void out(char *s, ...)
 188 {
 189 }
 190
 191 static void o_flush(void)
 192 {
 193         if (o_queued == 1)
 194                 sbuf_printf(pg, ") Tj\n");
 195         o_queued = 0;
 196 }
 197
 198 static int o_loadfont(struct glyph *g)
 199 {
 200         struct font *fn = g ? g->font : dev_font(o_f);
 201         int ix = font_glnum(fn, g) / 256;
 202         char *name = font_name(fn);
 203         int i;
 204         int id;
 205         for (i = 0; i < o_fsn; i++)
 206                 if (!strcmp(name, font_ps[o_fs[i]]) && font_ix[o_fs[i]] == ix)
 207                         return i;
 208         id = font_put(fn, ix);
 209         if (o_fsn == o_fssz) {
 210                 o_fssz += 128;
 211                 o_fs = mextend(o_fs, o_fsn, o_fssz, sizeof(o_fs[0]));
 212         }
 213         o_fs[o_fsn++] = id;
 214         return o_fsn - 1;
 215 }
 216
 217 #define PREC            1000
 218 #define PRECN           "3"
 219
 220 /* convert troff position to pdf position; returns a static buffer */
 221 static char *pdfpos(int uh, int uv)
 222 {
 223         static char buf[64];
 224         int h = uh * PREC * 72 / dev_res;
 225         int v = pdf_height * PREC - (uv * PREC * 72 / dev_res);
 226         sprintf(buf, "%d.%0" PRECN "d %d.%0" PRECN "d",
 227                 h / PREC, h % PREC, v / PREC, v % PREC);
 228         return buf;
 229 }
 230
 231 /* convert troff color to pdf color; returns a static buffer */
 232 static char *pdfcolor(int m)
 233 {
 234         static char buf[64];
 235         int r = CLR_R(m) * 1000 / 255;
 236         int g = CLR_G(m) * 1000 / 255;
 237         int b = CLR_B(m) * 1000 / 255;
 238         sbuf_printf(pg, "%d.%03d %d.%03d %d.%03d",
 239                 r / 1000, r % 1000, g / 1000, g % 1000, b / 1000, b % 1000);
 240         return buf;
 241 }
 242
 243 static void o_queue(struct glyph *g)
 244 {
 245         int pos;
 246         if (o_h != p_h || o_v != p_v) {
 247                 o_flush();
 248                 sbuf_printf(pg, "1 0 0 1 %s Tm\n", pdfpos(o_h, o_v));
 249                 p_h = o_h;
 250                 p_v = o_v;
 251         }
 252         if (!o_queued)
 253                 sbuf_printf(pg, "(");
 254         o_queued = 1;
 255         pos = font_glnum(g->font, g) % 256;
 256         sbuf_printf(pg, "\\%d%d%d", (pos >> 6) & 7, (pos >> 3) & 7, pos & 7);
 257         p_h += font_wid(g->font, o_s, g->wid);
 258 }
 259
 260 static void out_fontup(void)
 261 {
 262         if (o_m != p_m) {
 263                 o_flush();
 264                 sbuf_printf(pg, "%s rg\n", pdfcolor(o_m));
 265                 p_m = o_m;
 266         }
 267         if (o_pf != p_pf || o_s != p_s) {
 268                 int f = o_fs[o_pf];
 269                 o_flush();
 270                 sbuf_printf(pg, "/%s.%d %d Tf\n", font_ps[f], font_ix[f], o_s);
 271                 p_pf = o_pf;
 272                 p_s = o_s;
 273         }
 274 }
 275
 276 void outc(char *c)
 277 {
 278         struct glyph *g;
 279         struct font *fn;
 280         g = dev_glyph(c, o_f);
 281         fn = g ? g->font : dev_font(o_f);
 282         if (!g) {
 283                 outrel(*c == ' ' && fn ? font_swid(fn, o_s) : 1, 0);
 284                 return;
 285         }
 286         o_pf = o_loadfont(g);
 287         out_fontup();
 288         o_queue(g);
 289 }
 290
 291 void outh(int h)
 292 {
 293         o_h = h;
 294 }
 295
 296 void outv(int v)
 297 {
 298         o_v = v;
 299 }
 300
 301 void outrel(int h, int v)
 302 {
 303         o_h += h;
 304         o_v += v;
 305 }
 306
 307 void outfont(int f)
 308 {
 309         if (dev_font(f))
 310                 o_f = f;
 311 }
 312
 313 void outsize(int s)
 314 {
 315         if (s > 0)
 316                 o_s = s;
 317 }
 318
 319 void outcolor(int c)
 320 {
 321         o_m = c;
 322 }
 323
 324 void outrotate(int deg)
 325 {
 326 }
 327
 328 void outeps(char *eps)
 329 {
 330 }
 331
 332 void outlink(char *spec)
 333 {
 334 }
 335
 336 void outpage(void)
 337 {
 338         o_v = 0;
 339         o_h = 0;
 340         p_v = 0;
 341         p_h = 0;
 342         p_s = 0;
 343         p_f = 0;
 344         p_m = 0;
 345 }
 346
 347 void outmnt(int f)
 348 {
 349         if (p_f == f)
 350                 p_f = -1;
 351 }
 352
 353 void outgname(int g)
 354 {
 355 }
 356
 357 static int draw_path;   /* number of path segments */
 358 static int draw_point;  /* point was set for postscript newpath */
 359
 360 static void drawmv(void)
 361 {
 362         if (!draw_point)
 363                 sbuf_printf(pg, "%d %d m ", o_h, o_v);
 364         draw_point = 1;
 365 }
 366
 367 void drawbeg(void)
 368 {
 369         o_flush();
 370         out_fontup();
 371         if (draw_path)
 372                 return;
 373         sbuf_printf(pg, "%s m\n", pdfpos(o_h, o_v));
 374 }
 375
 376 void drawend(int close, int fill)
 377 {
 378         if (draw_path)
 379                 return;
 380         draw_point = 0;
 381         if (!fill)              /* stroking color */
 382                 sbuf_printf(pg, "%s RG\n", pdfcolor(o_m));
 383         if (fill)
 384                 sbuf_printf(pg, "f\n");
 385         else
 386                 sbuf_printf(pg, close ? "s\n" : "S\n");
 387 }
 388
 389 void drawmbeg(char *s)
 390 {
 391 }
 392
 393 void drawmend(char *s)
 394 {
 395 }
 396
 397 void drawl(int h, int v)
 398 {
 399         o_flush();
 400         outrel(h, v);
 401         sbuf_printf(pg, "%s l\n", pdfpos(o_h, o_v));
 402 }
 403
 404 void drawc(int c)
 405 {
 406         outrel(c, 0);
 407 }
 408
 409 void drawe(int h, int v)
 410 {
 411         outrel(h, 0);
 412 }
 413
 414 void drawa(int h1, int v1, int h2, int v2)
 415 {
 416         outrel(h1 + h2, v1 + v2);
 417 }
 418
 419 void draws(int h1, int v1, int h2, int v2)
 420 {
 421         outrel(h1, v1);
 422 }
 423
 424 void ps_header(char *title, int pagewidth, int pageheight, int linewidth)
 425 {
 426         pdf_title = title;
 427         obj_map();
 428         pdf_root = obj_map();
 429         pdf_pages = obj_map();
 430         pdf_title = title;
 431         pdfout("%%PDF-1.6\n");
 432         pdf_width = (pagewidth * 72 + 127) / 254;
 433         pdf_height = (pageheight * 72 + 127) / 254;
 434 }
 435
 436 void ps_trailer(int pages)
 437 {
 438         int i;
 439         int xref_off;
 440         int info_id;
 441         /* pdf pages object */
 442         obj_beg(pdf_pages);
 443         pdfout("<<\n");
 444         pdfout("  /Type /Pages\n");
 445         pdfout("  /MediaBox [ 0 0 %d %d ]\n", pdf_width, pdf_height);
 446         pdfout("  /Count %d\n", page_n);
 447         pdfout("  /Kids [");
 448         for (i = 0; i < page_n; i++)
 449                 pdfout(" %d 0 R", page_id[i]);
 450         pdfout(" ]\n");
 451         pdfout(">>\n");
 452         obj_end();
 453         /* pdf root object */
 454         obj_beg(pdf_root);
 455         pdfout("<<\n");
 456         pdfout("  /Type /Catalog\n");
 457         pdfout("  /Pages %d 0 R\n", pdf_pages);
 458         pdfout(">>\n");
 459         obj_end();
 460         /* info object */
 461         info_id = obj_beg(0);
 462         pdfout("<<\n");
 463         if (pdf_title)
 464                 pdfout("  /Title (%s)\n", pdf_title);
 465         pdfout("  /Creator (Neatroff)\n");
 466         pdfout("  /Producer (Neatpost)\n");
 467         pdfout(">>\n");
 468         obj_end();
 469         /* the xref */
 470         xref_off = pdf_pos;
 471         pdfout("xref\n");
 472         pdfout("0 %d\n", obj_n);
 473         pdfout("0000000000 65535 f \n");
 474         for (i = 1; i < obj_n; i++)
 475                 pdfout("%010d 00000 n \n", obj_off[i]);
 476         /* the trailer */
 477         pdfout("trailer\n");
 478         pdfout("<<\n");
 479         pdfout("  /Size %d\n", obj_n);
 480         pdfout("  /Root %d 0 R\n", pdf_root);
 481         pdfout("  /Info %d 0 R\n", info_id);
 482         pdfout(">>\n");
 483         pdfout("startxref\n");
 484         pdfout("%d\n", xref_off);
 485         pdfout("%%%%EOF\n");
 486         free(page_id);
 487         free(obj_off);
 488         for (i = 0; i < font_n; i++)
 489                 free(font_ps[i]);
 490         free(font_ps);
 491         free(font_ct);
 492         free(font_id);
 493         free(font_ix);
 494 }
 495
 496 void ps_pagebeg(int n)
 497 {
 498         pg = sbuf_make();
 499         sbuf_printf(pg, "BT\n");
 500 }
 501
 502 void ps_pageend(int n)
 503 {
 504         int cont_id;
 505         int i;
 506         o_flush();
 507         sbuf_printf(pg, "ET\n");
 508         /* page contents */
 509         cont_id = obj_beg(0);
 510         pdfout("<<\n");
 511         pdfout("  /Length %d\n", sbuf_len(pg));
 512         pdfout(">>\n");
 513         pdfout("stream\n");
 514         pdfout("%s", sbuf_buf(pg));
 515         pdfout("endstream\n");
 516         obj_end();
 517         /* the page object */
 518         if (page_n == page_sz) {
 519                 page_sz += 1024;
 520                 page_id = mextend(page_id, page_n, page_sz, sizeof(page_id[0]));
 521         }
 522         page_id[page_n++] = obj_beg(0);
 523         pdfout("<<\n");
 524         pdfout("  /Type /Page\n");
 525         pdfout("  /Parent %d 0 R\n", pdf_pages);
 526         pdfout("  /Resources <<\n");
 527         pdfout("    /Font <<");
 528         for (i = 0; i < o_fsn; i++)
 529                 pdfout(" /%s.%d %d 0 R",
 530                         font_ps[o_fs[i]], font_ix[o_fs[i]], font_id[o_fs[i]]);
 531         pdfout(" >>\n");
 532         pdfout("  >>\n");
 533         pdfout("  /Contents %d 0 R\n", cont_id);
 534         pdfout(">>\n");
 535         obj_end();
 536         sbuf_free(pg);
 537         free(o_fs);
 538         o_fs = NULL;
 539         o_fsn = 0;
 540         o_fssz = 0;
 541 }