tumble: tumble.c@3fe049d83e22 (annotated)

file
revisions
annotate
diff
raw

tumble.c@3fe049d83e22 (annotated)

tumble.c

Thu, 20 Mar 2003 01:08:05 +0000

author: eric
date: Thu, 20 Mar 2003 01:08:05 +0000
changeset 140: 3fe049d83e22
parent 139: ec2a06d8a2a6
child 141: 752599b50ff3
permissions: -rw-r--r--

*** empty log message ***

 /*
  * tumble: build a PDF file from image files
  *
  * Main program
  * $Id: tumble.c,v 1.37 2003/03/19 07:39:55 eric Exp $
  * Copyright 2001, 2002, 2003 Eric Smith <eric@brouhaha.com>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 2 as
  * published by the Free Software Foundation.  Note that permission is
  * not granted to redistribute this program under the terms of any
  * other version of the General Public License.
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111 USA
  */
 #include <stdarg.h>
 #include <stdbool.h>
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 #include <tiffio.h>
 #define TIFF_REVERSE_BITS
 #include "bitblt.h"
 #include "semantics.h"
 #include "parser.tab.h"
 #include "tumble.h"
 #include "pdf.h"
 #define MAX_INPUT_FILES 5000
 #define POINTS_PER_INCH 72
 /* page size limited by Acrobat Reader to 45 inches on a side */
 #define PAGE_MAX_INCHES 45
 #define PAGE_MAX_POINTS (PAGE_MAX_INCHES * POINTS_PER_INCH)
 typedef struct output_file_t
 {
   struct output_file_t *next;
   char *name;
   pdf_file_handle pdf;
 } output_file_t;
 int verbose;
 typedef enum
   {
     INPUT_FILE_TYPE_NONE,
     INPUT_FILE_TYPE_TIFF,
     INPUT_FILE_TYPE_JPEG
   } input_file_type_t;
 char *in_filename;
 input_file_type_t in_type;
 FILE *in;
 TIFF *tiff_in;
 output_file_t *output_files;
 output_file_t *out;
 char *progname;
 bool close_tiff_input_file (void);
 bool close_pdf_output_files (void);
 void usage (void)
 {
   fprintf (stderr, "\n");
   fprintf (stderr, "tumble - Copyright 2001-2003 Eric Smith <eric@brouhaha.com>\n");
   fprintf (stderr, "http://tumble.brouhaha.com/\n");
   fprintf (stderr, "\n");
   fprintf (stderr, "usage:\n");
   fprintf (stderr, "    %s [options] -c <control.tum>\n", progname);
   fprintf (stderr, "    %s [options] <input.tif>... -o <output.pdf>\n", progname);
   fprintf (stderr, "options:\n");
   fprintf (stderr, "    -v        verbose\n");
   fprintf (stderr, "    -b <fmt>  create bookmarks\n");
   fprintf (stderr, "bookmark format:\n");
   fprintf (stderr, "    %%F  file name (sans suffix)\n");
   fprintf (stderr, "    %%p  page number\n");
 }
 /* generate fatal error message to stderr, doesn't return */
 void fatal (int ret, char *format, ...) __attribute__ ((noreturn));
 void fatal (int ret, char *format, ...)
 {
   va_list ap;
   fprintf (stderr, "fatal error");
   if (format)
     {
       fprintf (stderr, ": ");
       va_start (ap, format);
       vfprintf (stderr, format, ap);
       va_end (ap);
     }
   else
     fprintf (stderr, "\n");
   if (ret == 1)
     usage ();
   close_input_file ();
   close_pdf_output_files ();
   exit (ret);
 }
 bool close_tiff_input_file (void)
 {
   TIFFClose (tiff_in);
   return (1);
 }
 bool open_tiff_input_file (FILE *f, char *name)
 {
   tiff_in = TIFFFdOpen (fileno (f), name, "r");
   if (! tiff_in)
     {
       fprintf (stderr, "can't open input file '%s'\n", name);
       free (in_filename);
       return (0);
     }
   in_type = INPUT_FILE_TYPE_TIFF;
   return (1);
 }
 bool close_jpeg_input_file (void)
 {
   return (1);
 }
 bool open_jpeg_input_file (FILE *f, char *name)
 {
   in_type = INPUT_FILE_TYPE_JPEG;
   return (1);
 }
 bool open_input_file (char *name)
 {
   bool result;
   uint8_t buf [2];
   size_t l;
   if (in)
     {
       if (strcmp (name, in_filename) == 0)
 	return (1);
       close_input_file ();
     }
   in_filename = strdup (name);
   if (! in_filename)
     {
       fprintf (stderr, "can't strdup input filename '%s'\n", name);
       return (0);
     }
   in = fopen (name, "rb");
   if (! in)
     return (0);
   l = fread (& buf [0], 1, sizeof (buf), in);
   if (l != sizeof (buf))
     return (0);
   rewind (in);
   if ((buf [0] == 0x49) && (buf [1] == 0x49))
     result = open_tiff_input_file (in, name);
   else if ((buf [0] == 0xff) && (buf [1] == 0xd8))
     result = open_jpeg_input_file (in, name);
   else
     {
       fprintf (stderr, "unrecognized file header in file '%s'\n", name);
       result = 0;
     }
   if (! result)
     {
       if (in)
 	fclose (in);
       in = NULL;
       in_type = INPUT_FILE_TYPE_NONE;
     }
   return (result);
 }
 bool close_input_file (void)
 {
   bool result;
   switch (in_type)
     {
     case INPUT_FILE_TYPE_NONE:
       return (1);
     case INPUT_FILE_TYPE_TIFF:
       result = close_tiff_input_file ();
       break;
     case INPUT_FILE_TYPE_JPEG:
       result = close_jpeg_input_file ();
       break;
     default:
       fatal (3, "internal error: bad input file type\n");
     }
   if (in_filename)
     free (in_filename);
   fclose (in);
   in = NULL;
   return (result);
 }
 bool last_tiff_input_page (void)
 {
   return (TIFFLastDirectory (tiff_in));
 }
 bool last_jpeg_input_page (void)
 {
   return (1);
 }
 bool last_input_page (void)
 {
   switch (in_type)
     {
     case INPUT_FILE_TYPE_TIFF:
       return (last_tiff_input_page ());
     case INPUT_FILE_TYPE_JPEG:
       return (last_jpeg_input_page ());
     default:
       fatal (3, "internal error: bad input file type\n");
     }
 }
 bool close_pdf_output_files (void)
 {
   output_file_t *o, *n;
   for (o = output_files; o; o = n)
     {
       n = o->next;
       pdf_close (o->pdf, PDF_PAGE_MODE_USE_OUTLINES);
       free (o->name);
       free (o);
     }
   out = NULL;
   output_files = NULL;
   return (1);
 }
 bool open_pdf_output_file (char *name,
 			   pdf_file_attributes_t *attributes)
 {
   output_file_t *o;
   if (out && (strcmp (name, out->name) == 0))
     return (1);
   for (o = output_files; o; o = o->next)
     if (strcmp (name, o->name) == 0)
       {
 	out = o;
 	return (1);
       }
   o = calloc (1, sizeof (output_file_t));
   if (! o)
     {
       fprintf (stderr, "can't calloc output file struct for '%s'\n", name);
       return (0);
    }
   o->name = strdup (name);
   if (! o->name)
     {
       fprintf (stderr, "can't strdup output filename '%s'\n", name);
       free (o);
       return (0);
     }
   o->pdf = pdf_create (name);
   if (! o->pdf)
     {
       fprintf (stderr, "can't open output file '%s'\n", name);
       free (o->name);
       free (o);
       return (0);
     }
   if (attributes->author)
     pdf_set_author (o->pdf, attributes->author);
   if (attributes->creator)
     pdf_set_creator (o->pdf, attributes->creator);
   if (attributes->title)
     pdf_set_title (o->pdf, attributes->title);
   if (attributes->subject)
     pdf_set_subject (o->pdf, attributes->subject);
   if (attributes->keywords)
     pdf_set_keywords (o->pdf, attributes->keywords);
   /* prepend new output file onto list */
   o->next = output_files;
   output_files = o;
   out = o;
   return (1);
 }
 /* frees original! */
 static Bitmap *resize_bitmap (Bitmap *src,
 			      double x_resolution,
 			      double y_resolution,
 			      input_attributes_t input_attributes)
 {
   Rect src_rect;
   Point dest_min;
   Bitmap *dest;
   int width_pixels = input_attributes.page_size.width * x_resolution;
   int height_pixels = input_attributes.page_size.height * y_resolution;
   src_rect.min.x = (rect_width (& src->rect) - width_pixels) / 2;
   src_rect.min.y = (rect_height (& src->rect) - height_pixels) / 2;
   src_rect.max.x = src_rect.min.x + width_pixels;
   src_rect.max.y = src_rect.min.y + height_pixels;
   dest_min.x = 0;
   dest_min.y = 0;
   dest = bitblt (src, & src_rect, NULL, & dest_min, TF_SRC, 0);
   free_bitmap (src);
   return (dest);
 }
 /* "in place" rotation */
 static void rotate_bitmap (Bitmap *src,
 			   input_attributes_t input_attributes)
 {
   switch (input_attributes.rotation)
     {
     case 0: break;
     case 90: rot_90 (src); break;
     case 180: rot_180 (src); break;
     case 270: rot_270 (src); break;
     default:
       fprintf (stderr, "rotation must be 0, 90, 180, or 270\n");
     }
 }
 #define SWAP(type,a,b) do { type temp; temp = a; a = b; b = temp; } while (0)
 static pdf_page_handle process_tiff_page (int image,  /* range 1 .. n */
 					  input_attributes_t input_attributes)
 {
   uint32_t image_length, image_width;
   uint32_t dest_image_length, dest_image_width;
 #ifdef CHECK_DEPTH
   uint32_t image_depth;
 #endif
   uint16_t samples_per_pixel;
   uint16_t bits_per_sample;
   uint16_t planar_config;
   uint16_t resolution_unit;
   float x_resolution, y_resolution;
   double dest_x_resolution, dest_y_resolution;
   double width_points, height_points;  /* really 1/72 inch units rather than
 					  points */
   Rect rect;
   Bitmap *bitmap = NULL;
   int row;
   pdf_page_handle page = NULL;
   if (! TIFFSetDirectory (tiff_in, image - 1))
     {
       fprintf (stderr, "can't find page %d of input file\n", image);
       goto fail;
     }
   if (1 != TIFFGetField (tiff_in, TIFFTAG_IMAGELENGTH, & image_length))
     {
       fprintf (stderr, "can't get image length\n");
       goto fail;
     }
   if (1 != TIFFGetField (tiff_in, TIFFTAG_IMAGEWIDTH, & image_width))
     {
       fprintf (stderr, "can't get image width\n");
       goto fail;
     }
   if (1 != TIFFGetField (tiff_in, TIFFTAG_SAMPLESPERPIXEL, & samples_per_pixel))
     {
       fprintf (stderr, "can't get samples per pixel\n");
       goto fail;
     }
 #ifdef CHECK_DEPTH
   if (1 != TIFFGetField (tiff_in, TIFFTAG_IMAGEDEPTH, & image_depth))
     {
       fprintf (stderr, "can't get image depth\n");
       goto fail;
     }
 #endif
   if (1 != TIFFGetField (tiff_in, TIFFTAG_BITSPERSAMPLE, & bits_per_sample))
     {
       fprintf (stderr, "can't get bits per sample\n");
       goto fail;
     }
   if (1 != TIFFGetField (tiff_in, TIFFTAG_PLANARCONFIG, & planar_config))
     planar_config = 1;
   if (1 != TIFFGetField (tiff_in, TIFFTAG_RESOLUTIONUNIT, & resolution_unit))
     resolution_unit = 2;
   if (1 != TIFFGetField (tiff_in, TIFFTAG_XRESOLUTION, & x_resolution))
     x_resolution = 300;
   if (1 != TIFFGetField (tiff_in, TIFFTAG_YRESOLUTION, & y_resolution))
     y_resolution = 300;
   if (samples_per_pixel != 1)
     {
       fprintf (stderr, "samples per pixel %u, must be 1\n", samples_per_pixel);
       goto fail;
     }
 #ifdef CHECK_DEPTH
   if (image_depth != 1)
     {
       fprintf (stderr, "image depth %u, must be 1\n", image_depth);
       goto fail;
     }
 #endif
   if (bits_per_sample != 1)
     {
       fprintf (stderr, "bits per sample %u, must be 1\n", bits_per_sample);
       goto fail;
     }
   if (planar_config != 1)
     {
       fprintf (stderr, "planar config %u, must be 1\n", planar_config);
       goto fail;
     }
   if (input_attributes.has_resolution)
     {
       x_resolution = input_attributes.x_resolution;
       y_resolution = input_attributes.y_resolution;
     }
   if ((input_attributes.rotation == 90) || (input_attributes.rotation == 270))
     {
       dest_image_width  = image_length;
       dest_image_length = image_width;
       dest_x_resolution = y_resolution;
       dest_y_resolution = x_resolution;
       SWAP (double, width_points, height_points);  /* $$$ not yet set!!! */
     }
   else
     {
       dest_image_width = image_width;
       dest_image_length = image_length;
       dest_x_resolution = x_resolution;
       dest_y_resolution = y_resolution;
     }
   rect.min.x = 0;
   rect.min.y = 0;
   rect.max.x = image_width;
   rect.max.y = image_length;
   bitmap = create_bitmap (& rect);
   if (! bitmap)
     {
       fprintf (stderr, "can't allocate bitmap\n");
       goto fail;
     }
   for (row = 0; row < image_length; row++)
     if (1 != TIFFReadScanline (tiff_in,
 			       bitmap->bits + row * bitmap->row_words,
 			       row,
 ))
       {
 	fprintf (stderr, "can't read TIFF scanline\n");
 	goto fail;
       }
 #ifdef TIFF_REVERSE_BITS
   reverse_bits ((uint8_t *) bitmap->bits,
 		image_length * bitmap->row_words * sizeof (word_t));
 #endif /* TIFF_REVERSE_BITS */
 #if 0
   if (input_attributes.has_page_size)
     bitmap = resize_bitmap (bitmap,
 			    x_resolution,
 			    y_resolution,
 			    input_attributes);
 #endif
   rotate_bitmap (bitmap,
 		 input_attributes);
   width_points = (rect_width (& bitmap->rect) / dest_x_resolution) * POINTS_PER_INCH;
   height_points = (rect_height (& bitmap->rect) / dest_y_resolution) * POINTS_PER_INCH;
   if ((height_points > PAGE_MAX_POINTS) || (width_points > PAGE_MAX_POINTS))
     {
       fprintf (stdout, "image too large (max %d inches on a side\n", PAGE_MAX_INCHES);
       goto fail;
     }
   page = pdf_new_page (out->pdf, width_points, height_points);
 #if 0
   pdf_write_text (page);
 #else
   pdf_write_g4_fax_image (page,
 , 0,  /* x, y */
 			  width_points, height_points,
 			  bitmap,
 , /* ImageMask */
 , 0, 0,  /* r, g, b */
 ); /* BlackIs1 */
 #endif
   if (bitmap)
     free_bitmap (bitmap);
   return (page);
  fail:
   if (bitmap)
     free_bitmap (bitmap);
   return (NULL);
 }
 pdf_page_handle process_jpeg_page (int image,  /* range 1 .. n */
 				   input_attributes_t input_attributes)
 {
   pdf_page_handle page;
   double width_points, height_points;  /* really 1/72 inch units rather than
 					  points */
   /* $$$ need to get these from somewhere else, hardcoded for now */
   width_points = 4 * 72.0;
   height_points = 4 * 72.0;
   page = pdf_new_page (out->pdf, width_points, height_points);
   pdf_write_jpeg_image (page,
 , 0,  /* x, y */
 			width_points, height_points,
 			in);
   return (page);
 }
 bool process_page (int image,  /* range 1 .. n */
 		   input_attributes_t input_attributes,
 		   bookmark_t *bookmarks,
 		   page_label_t *page_label)
 {
   pdf_page_handle page;
   switch (in_type)
     {
     case INPUT_FILE_TYPE_TIFF:
       page = process_tiff_page (image, input_attributes);
       break;
     case INPUT_FILE_TYPE_JPEG:
       page = process_jpeg_page (image, input_attributes);
       break;
     default:
       fatal (3, "internal error: bad input file type\n");
     }
   while (bookmarks)
     {
       /* $$$ need to handle level here */
       pdf_new_bookmark (NULL, bookmarks->name, 0, page);
       bookmarks = bookmarks->next;
     }
   if (page_label)
     pdf_new_page_label (out->pdf,
 			page_label->page_index,
 			page_label->base,
 			page_label->count,
 			page_label->style,
 			page_label->prefix);
   return (page != NULL);
 }
 #define MAX_BOOKMARK_NAME_LEN 500
 static int filename_length_without_suffix (char *in_fn)
 {
   char *p;
   int len = strlen (in_fn);
   p = strrchr (in_fn, '.');
   if (p && ((strcasecmp (p, ".tif") == 0) ||
 	    (strcasecmp (p, ".tiff") == 0)))
     return (p - in_fn);
   return (len);
 }
 /* $$$ this function should ensure that it doesn't overflow the name string! */
 static void generate_bookmark_name (char *name,
 				    char *bookmark_fmt,
 				    char *in_fn,
 				    int page)
 {
   bool meta = 0;
   int len;
   while (*bookmark_fmt)
     {
       if (meta)
 	{
 	  meta = 0;
 	  switch (*bookmark_fmt)
 	    {
 	    case '%':
 	      *(name++) = '%';
 	      break;
 	    case 'F':
 	      len = filename_length_without_suffix (in_fn);
 	      strncpy (name, in_fn, len);
 	      name += len;
 	      break;
 	    case 'p':
 	      sprintf (name, "%d", page);
 	      name += strlen (name);
 	      break;
 	    default:
 	      break;
 	    }
 	}
       else
 	switch (*bookmark_fmt)
 	  {
 	  case '%':
 	    meta = 1;
 	    break;
 	  default:
 	    *(name++) = *bookmark_fmt;
 	  }
       bookmark_fmt++;
     }
   *name = '\0';
 }
 void main_args (char *out_fn,
 		int inf_count,
 		char **in_fn,
 		char *bookmark_fmt)
 {
   int i, ip;
   input_attributes_t input_attributes;
   pdf_file_attributes_t output_attributes;
   bookmark_t bookmark;
   char bookmark_name [MAX_BOOKMARK_NAME_LEN];
   bookmark.next = NULL;
   bookmark.level = 1;
   bookmark.name = & bookmark_name [0];
   memset (& input_attributes, 0, sizeof (input_attributes));
   memset (& output_attributes, 0, sizeof (output_attributes));
   if (! open_pdf_output_file (out_fn, & output_attributes))
     fatal (3, "error opening output file \"%s\"\n", out_fn);
   for (i = 0; i < inf_count; i++)
     {
       if (! open_input_file (in_fn [i]))
 	fatal (3, "error opening input file \"%s\"\n", in_fn [i]);
       for (ip = 1;; ip++)
 	{
 	  fprintf (stderr, "processing page %d of file \"%s\"\r", ip, in_fn [i]);
 	  if (bookmark_fmt)
 	    generate_bookmark_name (& bookmark_name [0],
 				    bookmark_fmt,
 				    in_fn [i],
 				    ip);
 	  if (! process_page (ip, input_attributes,
 			      bookmark_fmt ? & bookmark : NULL,
 			      NULL))
 	    fatal (3, "error processing page %d of input file \"%s\"\n", ip, in_fn [i]);
 	  if (last_input_page ())
 	    break;
 	}
       if (verbose)
 	fprintf (stderr, "processed %d pages of input file \"%s\"\n", ip, in_fn [i]);
       if (! close_input_file ())
 	fatal (3, "error closing input file \"%s\"\n", in_fn [i]);
     }
   if (! close_pdf_output_files ())
     fatal (3, "error closing output file \"%s\"\n", out_fn);
 }
 void main_control (char *control_fn)
 {
   if (! parse_control_file (control_fn))
     fatal (2, "error parsing control file\n");
   if (! process_controls ())
     fatal (3, "error processing control file\n");
 }
 int main (int argc, char *argv[])
 {
   char *control_fn = NULL;
   char *out_fn = NULL;
   char *bookmark_fmt = NULL;
   int inf_count = 0;
   char *in_fn [MAX_INPUT_FILES];
   progname = argv [0];
   pdf_init ();
   while (--argc)
     {
       if (argv [1][0] == '-')
 	{
 	  if (strcmp (argv [1], "-v") == 0)
 	    verbose++;
 	  else if (strcmp (argv [1], "-o") == 0)
 	    {
 	      if (argc)
 		{
 		  argc--;
 		  argv++;
 		  out_fn = argv [1];
 		}
 	      else
 		fatal (1, "missing filename after \"-o\" option\n");
 	    }
 	  else if (strcmp (argv [1], "-c") == 0)
 	    {
 	      if (argc)
 		{
 		  argc--;
 		  argv++;
 		  control_fn = argv [1];
 		}
 	      else
 		fatal (1, "missing filename after \"-s\" option\n");
 	    }
 	  else if (strcmp (argv [1], "-b") == 0)
 	    {
 	      if (argc)
 		{
 		  argc--;
 		  argv++;
 		  bookmark_fmt = argv [1];
 		}
 	      else
 		fatal (1, "missing format string after \"-b\" option\n");
 	    }
 	  else
 	    fatal (1, "unrecognized option \"%s\"\n", argv [1]);
 	}
       else if (inf_count < MAX_INPUT_FILES)
 	in_fn [inf_count++] = argv [1];
       else
 	fatal (1, "exceeded maximum of %d input files\n", MAX_INPUT_FILES);
       argv++;
     }
   if (! ((! out_fn) ^ (! control_fn)))
     fatal (1, "either a control file or an output file (but not both) must be specified\n");
   if (out_fn && ! inf_count)
     fatal (1, "no input files specified\n");
   if (control_fn && inf_count)
     fatal (1, "if control file is provided, input files can't be specified as arguments\n");
   if (control_fn)
     main_control (control_fn);
   else
     main_args (out_fn, inf_count, in_fn, bookmark_fmt);
   close_input_file ();
   close_pdf_output_files ();
   exit (0);
 }

tumble / annotate

tumble.c@3fe049d83e22 (annotated)

tumble.c