images/image-tool.c

   1 /*
   2  *      Simple image manupulation utility
   3  *
   4  *      (c) 2006 Pavel Charvat <pchar@ucw.cz>
   5  *
   6  *      This software may be freely distributed and used according to the terms
   7  *      of the GNU General Public License.
   8  */
   9
  10 #include "lib/lib.h"
  11 #include "lib/getopt.h"
  12 #include "lib/fastbuf.h"
  13 #include "images/images.h"
  14 #include "images/color.h"
  15 #include <stdlib.h>
  16 #include <fcntl.h>
  17 #include <errno.h>
  18 #include <stdio.h>
  19
  20 static void NONRET
  21 usage(void)
  22 {
  23   fputs("\
  24 Usage: image-tool [options] infile [outfile]\n\
  25 \n\
  26 -q --quiet          no progress messages\n\
  27 -f --input-format   input image format (jpeg, gif, png)\n\
  28 -F --output-format  output image format\n\
  29 -s --size           force output dimensions (100x200)\n\
  30 -b --fit-to-box     scale to fit the box (100x200)\n\
  31 -c --colorspace     force output colorspace (Gray, GrayAlpha, RGB, RGBAlpha)\n\
  32 -Q --jpeg-quality   JPEG quality (1..100)\n\
  33 -g --background     background color (hexadecimal RRGGBB)\n\
  34 ", stderr);
  35   exit(1);
  36 }
  37
  38 static char *shortopts = "qf:F:s:b:c:Q:g:" CF_SHORT_OPTS;
  39 static struct option longopts[] =
  40 {
  41   CF_LONG_OPTS
  42   { "quiet",            0, 0, 'q' },
  43   { "input-format",     0, 0, 'f' },
  44   { "output-format",    0, 0, 'F' },
  45   { "size",             0, 0, 's' },
  46   { "fit-to-box",       0, 0, 'b' },
  47   { "colorspace",       0, 0, 'c' },
  48   { "jpeg-quality",     0, 0, 'Q' },
  49   { "background",       0, 0, 'g'  },
  50   { NULL,               0, 0, 0 }
  51 };
  52
  53 static uns verbose = 1;
  54 static byte *input_file_name;
  55 static enum image_format input_format;
  56 static byte *output_file_name;
  57 static enum image_format output_format;
  58 static uns cols;
  59 static uns rows;
  60 static uns fit_to_box;
  61 static uns channels_format;
  62 static uns jpeg_quality;
  63 static struct color background_color;
  64
  65 #define MSG(x...) do{ if (verbose) log(L_INFO, ##x); }while(0)
  66
  67 int
  68 main(int argc, char **argv)
  69 {
  70   log_init(argv[0]);
  71   int opt;
  72   while ((opt = cf_getopt(argc, argv, shortopts, longopts, NULL)) >= 0)
  73     switch (opt)
  74       {
  75         case 'q':
  76           verbose = 0;
  77           break;
  78         case 'f':
  79           if (!(input_format = image_extension_to_format(optarg)))
  80             usage();
  81           break;
  82         case 'F':
  83           if (!(output_format = image_extension_to_format(optarg)))
  84             usage();
  85           break;
  86         case 's':
  87           {
  88             byte *r = strchr(optarg, 'x');
  89             if (!r)
  90               usage();
  91             *r++ = 0;
  92             if (!(cols = atoi(optarg)) || !(rows = atoi(r)))
  93               usage();
  94             fit_to_box = 0;
  95             break;
  96           }
  97         case 'b':
  98           {
  99             byte *r = strchr(optarg, 'x');
 100             if (!r)
 101               usage();
 102             *r++ = 0;
 103             if (!(cols = atoi(optarg)) || !(rows = atoi(r)))
 104               usage();
 105             fit_to_box = 1;
 106             break;
 107           }
 108         case 'c':
 109           if (!(channels_format = image_name_to_channels_format(optarg)))
 110             usage();
 111           break;
 112         case 'Q':
 113           if (!(jpeg_quality = atoi(optarg)))
 114             usage();
 115           break;
 116         case 'g':
 117           {
 118             if (strlen(optarg) != 6)
 119               usage();
 120             errno = 0;
 121             char *end;
 122             long int v = strtol(optarg, &end, 16);
 123             if (errno || *end || v < 0)
 124               usage();
 125             color_make_rgb(&background_color, (v >> 16) & 255, (v >> 8) & 255, v & 255);
 126           }
 127           break;
 128         default:
 129           usage();
 130       }
 131
 132   if (argc != optind + 1 && argc != optind + 2)
 133     usage();
 134   input_file_name = argv[optind++];
 135   if (argc > optind)
 136     output_file_name = argv[optind];
 137
 138 #define TRY(x) do{ if (!(x)) die("Error: %s", it.err_msg); }while(0)
 139   MSG("Initializing image library");
 140   struct image_thread it;
 141   struct image_io io;
 142   image_thread_init(&it);
 143   image_io_init(&it, &io);
 144
 145   MSG("Reading %s", input_file_name);
 146   io.fastbuf = bopen(input_file_name, O_RDONLY, 1 << 18);
 147   io.format = input_format ? : image_file_name_to_format(input_file_name);
 148   TRY(image_io_read_header(&io));
 149   if (!output_file_name)
 150     {
 151       bclose(io.fastbuf);
 152       printf("Format:      %s\n", image_format_to_extension(io.format) ? : (byte *)"?");
 153       printf("Dimensions:  %dx%d\n", io.cols, io.rows);
 154       printf("Colorspace:  %s\n", (io.flags & IMAGE_IO_HAS_PALETTE) ? (byte *)"Palette" : image_channels_format_to_name(io.flags & IMAGE_CHANNELS_FORMAT));
 155       printf("NumColors:   %d\n", io.number_of_colors);
 156     }
 157   else
 158     {
 159       MSG("%s %dx%d %s", image_format_to_extension(io.format) ? : (byte *)"?", io.cols, io.rows,
 160           (io.flags & IMAGE_IO_HAS_PALETTE) ? (byte *)"Palette" : image_channels_format_to_name(io.flags & IMAGE_CHANNELS_FORMAT));
 161       if (cols)
 162         if (fit_to_box)
 163           {
 164             image_dimensions_fit_to_box(&io.cols, &io.rows, MIN(cols, 0xffff), MIN(rows, 0xffff), 0);
 165           }
 166         else
 167           {
 168             io.cols = cols;
 169             io.rows = rows;
 170           }
 171       if (background_color.color_space)
 172         io.background_color = background_color;
 173       if (channels_format)
 174         io.flags = io.flags & ~IMAGE_PIXEL_FORMAT | channels_format;
 175       if (!(io.flags & IMAGE_ALPHA))
 176         io.flags |= IMAGE_IO_USE_BACKGROUND;
 177       if (jpeg_quality)
 178         io.jpeg_quality = jpeg_quality;
 179       TRY(image_io_read_data(&io, 0));
 180       bclose(io.fastbuf);
 181       MSG("Writing %s", output_file_name);
 182       io.fastbuf = bopen(output_file_name, O_WRONLY | O_CREAT | O_TRUNC, 1 << 18);
 183       io.format = output_format ? : image_file_name_to_format(output_file_name);
 184       MSG("%s %dx%d %s", image_format_to_extension(io.format) ? : (byte *)"?", io.cols, io.rows,
 185           image_channels_format_to_name(io.flags & IMAGE_CHANNELS_FORMAT));
 186       TRY(image_io_write(&io));
 187       bclose(io.fastbuf);
 188     }
 189
 190   image_io_cleanup(&io);
 191   image_thread_cleanup(&it);
 192   MSG("Done.");
 193   return 0;
 194 }