rdcolmap.c
上传用户:wuyixingx
上传日期:2007-01-08
资源大小:745k
文件大小:7k
源码类别:

图形图象

开发平台:

C/C++

  1. /*
  2.  * rdcolmap.c
  3.  *
  4.  * Copyright (C) 1994-1996, Thomas G. Lane.
  5.  * This file is part of the Independent JPEG Group's software.
  6.  * For conditions of distribution and use, see the accompanying README file.
  7.  *
  8.  * This file implements djpeg's "-map file" switch.  It reads a source image
  9.  * and constructs a colormap to be supplied to the JPEG decompressor.
  10.  *
  11.  * Currently, these file formats are supported for the map file:
  12.  *   GIF: the contents of the GIF's global colormap are used.
  13.  *   PPM (either text or raw flavor): the entire file is read and
  14.  *      each unique pixel value is entered in the map.
  15.  * Note that reading a large PPM file will be horrendously slow.
  16.  * Typically, a PPM-format map file should contain just one pixel
  17.  * of each desired color.  Such a file can be extracted from an
  18.  * ordinary image PPM file with ppmtomap(1).
  19.  *
  20.  * Rescaling a PPM that has a maxval unequal to MAXJSAMPLE is not
  21.  * currently implemented.
  22.  */
  23. #include "cdjpeg.h" /* Common decls for cjpeg/djpeg applications */
  24. #ifdef QUANT_2PASS_SUPPORTED /* otherwise can't quantize to supplied map */
  25. /* Portions of this code are based on the PBMPLUS library, which is:
  26. **
  27. ** Copyright (C) 1988 by Jef Poskanzer.
  28. **
  29. ** Permission to use, copy, modify, and distribute this software and its
  30. ** documentation for any purpose and without fee is hereby granted, provided
  31. ** that the above copyright notice appear in all copies and that both that
  32. ** copyright notice and this permission notice appear in supporting
  33. ** documentation.  This software is provided "as is" without express or
  34. ** implied warranty.
  35. */
  36. /*
  37.  * Add a (potentially) new color to the color map.
  38.  */
  39. LOCAL(void)
  40. add_map_entry (j_decompress_ptr cinfo, int R, int G, int B)
  41. {
  42.   JSAMPROW colormap0 = cinfo->colormap[0];
  43.   JSAMPROW colormap1 = cinfo->colormap[1];
  44.   JSAMPROW colormap2 = cinfo->colormap[2];
  45.   int ncolors = cinfo->actual_number_of_colors;
  46.   int index;
  47.   /* Check for duplicate color. */
  48.   for (index = 0; index < ncolors; index++) {
  49.     if (GETJSAMPLE(colormap0[index]) == R &&
  50. GETJSAMPLE(colormap1[index]) == G &&
  51. GETJSAMPLE(colormap2[index]) == B)
  52.       return; /* color is already in map */
  53.   }
  54.   /* Check for map overflow. */
  55.   if (ncolors >= (MAXJSAMPLE+1))
  56.     ERREXIT1(cinfo, JERR_QUANT_MANY_COLORS, (MAXJSAMPLE+1));
  57.   /* OK, add color to map. */
  58.   colormap0[ncolors] = (JSAMPLE) R;
  59.   colormap1[ncolors] = (JSAMPLE) G;
  60.   colormap2[ncolors] = (JSAMPLE) B;
  61.   cinfo->actual_number_of_colors++;
  62. }
  63. /*
  64.  * Extract color map from a GIF file.
  65.  */
  66. LOCAL(void)
  67. read_gif_map (j_decompress_ptr cinfo, FILE * infile)
  68. {
  69.   int header[13];
  70.   int i, colormaplen;
  71.   int R, G, B;
  72.   /* Initial 'G' has already been read by read_color_map */
  73.   /* Read the rest of the GIF header and logical screen descriptor */
  74.   for (i = 1; i < 13; i++) {
  75.     if ((header[i] = getc(infile)) == EOF)
  76.       ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  77.   }
  78.   /* Verify GIF Header */
  79.   if (header[1] != 'I' || header[2] != 'F')
  80.     ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  81.   /* There must be a global color map. */
  82.   if ((header[10] & 0x80) == 0)
  83.     ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  84.   /* OK, fetch it. */
  85.   colormaplen = 2 << (header[10] & 0x07);
  86.   for (i = 0; i < colormaplen; i++) {
  87.     R = getc(infile);
  88.     G = getc(infile);
  89.     B = getc(infile);
  90.     if (R == EOF || G == EOF || B == EOF)
  91.       ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  92.     add_map_entry(cinfo,
  93.   R << (BITS_IN_JSAMPLE-8),
  94.   G << (BITS_IN_JSAMPLE-8),
  95.   B << (BITS_IN_JSAMPLE-8));
  96.   }
  97. }
  98. /* Support routines for reading PPM */
  99. LOCAL(int)
  100. pbm_getc (FILE * infile)
  101. /* Read next char, skipping over any comments */
  102. /* A comment/newline sequence is returned as a newline */
  103. {
  104.   register int ch;
  105.   
  106.   ch = getc(infile);
  107.   if (ch == '#') {
  108.     do {
  109.       ch = getc(infile);
  110.     } while (ch != 'n' && ch != EOF);
  111.   }
  112.   return ch;
  113. }
  114. LOCAL(unsigned int)
  115. read_pbm_integer (j_decompress_ptr cinfo, FILE * infile)
  116. /* Read an unsigned decimal integer from the PPM file */
  117. /* Swallows one trailing character after the integer */
  118. /* Note that on a 16-bit-int machine, only values up to 64k can be read. */
  119. /* This should not be a problem in practice. */
  120. {
  121.   register int ch;
  122.   register unsigned int val;
  123.   
  124.   /* Skip any leading whitespace */
  125.   do {
  126.     ch = pbm_getc(infile);
  127.     if (ch == EOF)
  128.       ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  129.   } while (ch == ' ' || ch == 't' || ch == 'n' || ch == 'r');
  130.   
  131.   if (ch < '0' || ch > '9')
  132.     ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  133.   
  134.   val = ch - '0';
  135.   while ((ch = pbm_getc(infile)) >= '0' && ch <= '9') {
  136.     val *= 10;
  137.     val += ch - '0';
  138.   }
  139.   return val;
  140. }
  141. /*
  142.  * Extract color map from a PPM file.
  143.  */
  144. LOCAL(void)
  145. read_ppm_map (j_decompress_ptr cinfo, FILE * infile)
  146. {
  147.   int c;
  148.   unsigned int w, h, maxval, row, col;
  149.   int R, G, B;
  150.   /* Initial 'P' has already been read by read_color_map */
  151.   c = getc(infile); /* save format discriminator for a sec */
  152.   /* while we fetch the remaining header info */
  153.   w = read_pbm_integer(cinfo, infile);
  154.   h = read_pbm_integer(cinfo, infile);
  155.   maxval = read_pbm_integer(cinfo, infile);
  156.   if (w <= 0 || h <= 0 || maxval <= 0) /* error check */
  157.     ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  158.   /* For now, we don't support rescaling from an unusual maxval. */
  159.   if (maxval != (unsigned int) MAXJSAMPLE)
  160.     ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  161.   switch (c) {
  162.   case '3': /* it's a text-format PPM file */
  163.     for (row = 0; row < h; row++) {
  164.       for (col = 0; col < w; col++) {
  165. R = read_pbm_integer(cinfo, infile);
  166. G = read_pbm_integer(cinfo, infile);
  167. B = read_pbm_integer(cinfo, infile);
  168. add_map_entry(cinfo, R, G, B);
  169.       }
  170.     }
  171.     break;
  172.   case '6': /* it's a raw-format PPM file */
  173.     for (row = 0; row < h; row++) {
  174.       for (col = 0; col < w; col++) {
  175. R = getc(infile);
  176. G = getc(infile);
  177. B = getc(infile);
  178. if (R == EOF || G == EOF || B == EOF)
  179.   ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  180. add_map_entry(cinfo, R, G, B);
  181.       }
  182.     }
  183.     break;
  184.   default:
  185.     ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  186.     break;
  187.   }
  188. }
  189. /*
  190.  * Main entry point from djpeg.c.
  191.  *  Input: opened input file (from file name argument on command line).
  192.  *  Output: colormap and actual_number_of_colors fields are set in cinfo.
  193.  */
  194. GLOBAL(void)
  195. read_color_map (j_decompress_ptr cinfo, FILE * infile)
  196. {
  197.   /* Allocate space for a color map of maximum supported size. */
  198.   cinfo->colormap = (*cinfo->mem->alloc_sarray)
  199.     ((j_common_ptr) cinfo, JPOOL_IMAGE,
  200.      (JDIMENSION) (MAXJSAMPLE+1), (JDIMENSION) 3);
  201.   cinfo->actual_number_of_colors = 0; /* initialize map to empty */
  202.   /* Read first byte to determine file format */
  203.   switch (getc(infile)) {
  204.   case 'G':
  205.     read_gif_map(cinfo, infile);
  206.     break;
  207.   case 'P':
  208.     read_ppm_map(cinfo, infile);
  209.     break;
  210.   default:
  211.     ERREXIT(cinfo, JERR_BAD_CMAP_FILE);
  212.     break;
  213.   }
  214. }
  215. #endif /* QUANT_2PASS_SUPPORTED */