home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 1996 February / PCWK0296.iso / sharewar / dos / program / gs300sr1 / gs300sr1.exe / GDEVPBM.C < prev    next >
C/C++ Source or Header  |  1994-07-27  |  13KB  |  420 lines
  1. /* Copyright (C) 1992, 1993 Aladdin Enterprises.  All rights reserved.
  2.   
  3.   This file is part of Aladdin Ghostscript.
  4.   
  5.   Aladdin Ghostscript is distributed with NO WARRANTY OF ANY KIND.  No author
  6.   or distributor accepts any responsibility for the consequences of using it,
  7.   or for whether it serves any particular purpose or works at all, unless he
  8.   or she says so in writing.  Refer to the Aladdin Ghostscript Free Public
  9.   License (the "License") for full details.
  10.   
  11.   Every copy of Aladdin Ghostscript must include a copy of the License,
  12.   normally in a plain ASCII text file named PUBLIC.  The License grants you
  13.   the right to copy, modify and redistribute Aladdin Ghostscript, but only
  14.   under certain conditions described in the License.  Among other things, the
  15.   License requires that the copyright notice and this notice be preserved on
  16.   all copies.
  17. */
  18.  
  19. /* gdevpbm.c */
  20. /* Portable Bit/Gray/PixMap devices for Ghostscript. */
  21. #include "gdevprn.h"
  22. #include "gxlum.h"
  23.  
  24. /* Thanks are due to Jos Vos (jos@bull.nl) for an earlier P*M driver, */
  25. /* on which this one is based. */
  26.  
  27. /*
  28.  * The PGM and PPM drivers have an "optimize" flag, which causes them to
  29.  * fall back to PBM or (for PPM) PGM if the page can be represented that way.
  30.  * This should be a device property, but for now, it's a compile-time option.
  31.  */
  32. #define OPTIMIZE 1
  33.  
  34. /*
  35.  * The code here is designed to work with variable depths for PGM and PPM.
  36.  * The restrictions on depth are as follows.  Numbers in square brackets
  37.  * give the actual restrictions, but the Ghostscript kernel requires that
  38.  * depth be a power of 2 or be 24.
  39.  *    pgm: 1, 2, 4, 8, 16.  [1-16]
  40.  *    pgmraw: 1, 2, 4, 8.  [1-8]
  41.  *    ppm: 4(3x1), 8(3x2), 16(3x5), 24(3x8), 32(3x10).  [3-32]
  42.  *    ppmraw: 4(3x1), 8(3x2), 16(3x5), 24(3x8).  [3-24]
  43.  */
  44.  
  45. /* Structure for P*M devices, which extend the generic printer device. */
  46.  
  47. #define MAX_COMMENT 70            /* max user-supplied comment */
  48. struct gx_device_pbm_s {
  49.     gx_device_common;
  50.     gx_prn_device_common;
  51.     /* Additional state for P*M devices */
  52.     char magic;            /* n for "Pn" */
  53.     char comment[MAX_COMMENT + 1];    /* comment for head of file */
  54.     byte is_raw;            /* 1 if raw format, 0 if plain */
  55.     byte optimize;            /* 1 if optimization OK, 0 if not */
  56.     byte uses_color;        /* 0 if image is black and white, */
  57.                     /* 1 if gray (PGM or PPM only), */
  58.                     /* 2 or 3 if colored (PPM only) */
  59. };
  60. typedef struct gx_device_pbm_s gx_device_pbm;
  61.  
  62. #define bdev ((gx_device_pbm *)pdev)
  63.  
  64. /* ------ The device descriptors ------ */
  65.  
  66. /*
  67.  * Default X and Y resolution.
  68.  */
  69. #define X_DPI 72
  70. #define Y_DPI 72
  71.  
  72. /* Macro for generating P*M device descriptors. */
  73. #define pbm_prn_device(procs, dev_name, magic, is_raw, num_comp, depth, max_gray, max_rgb, print_page)\
  74. {    prn_device_body(gx_device_pbm, procs, dev_name,\
  75.       DEFAULT_WIDTH_10THS, DEFAULT_HEIGHT_10THS, X_DPI, Y_DPI,\
  76.       0, 0, 0, 0,\
  77.       num_comp, depth, max_gray, max_rgb, max_gray + 1, max_rgb + 1,\
  78.       print_page),\
  79.     magic,\
  80.      { 0 },\
  81.     is_raw,\
  82.     OPTIMIZE\
  83. }
  84.  
  85. /* For PGM and PPM, we need our own color mapping procedures. */
  86. private dev_proc_map_rgb_color(pgm_map_rgb_color);
  87. private dev_proc_map_rgb_color(ppm_map_rgb_color);
  88. private dev_proc_map_color_rgb(pgm_map_color_rgb);
  89. private dev_proc_map_color_rgb(ppm_map_color_rgb);
  90.  
  91. /* We need to initialize uses_color when opening the device. */
  92. private dev_proc_open_device(ppm_open);
  93.  
  94. /* And of course we need our own print-page routines. */
  95. private dev_proc_print_page(pbm_print_page);
  96. private dev_proc_print_page(pgm_print_page);
  97. private dev_proc_print_page(ppm_print_page);
  98.  
  99. /* The device procedures */
  100. private gx_device_procs pbm_procs =
  101.     prn_procs(gdev_prn_open, gdev_prn_output_page, gdev_prn_close);
  102. private gx_device_procs pgm_procs =
  103.     prn_color_procs(gdev_prn_open, gdev_prn_output_page, gdev_prn_close,
  104.     pgm_map_rgb_color, pgm_map_color_rgb);
  105. private gx_device_procs ppm_procs =
  106.     prn_color_procs(ppm_open, gdev_prn_output_page, gdev_prn_close,
  107.     ppm_map_rgb_color, ppm_map_color_rgb);
  108.  
  109. /* The device descriptors themselves */
  110. gx_device_pbm far_data gs_pbm_device =
  111.   pbm_prn_device(pbm_procs, "pbm", '1', 0, 1, 1, 1, 0,
  112.       pbm_print_page);
  113. gx_device_pbm far_data gs_pbmraw_device =
  114.   pbm_prn_device(pbm_procs, "pbmraw", '4', 1, 1, 1, 1, 1,
  115.       pbm_print_page);
  116. gx_device_pbm far_data gs_pgm_device =
  117.   pbm_prn_device(pgm_procs, "pgm", '2', 0, 1, 8, 255, 0,
  118.       pgm_print_page);
  119. gx_device_pbm far_data gs_pgmraw_device =
  120.   pbm_prn_device(pgm_procs, "pgmraw", '5', 1, 1, 8, 255, 0,
  121.       pgm_print_page);
  122. gx_device_pbm far_data gs_ppm_device =
  123.   pbm_prn_device(ppm_procs, "ppm", '3', 0, 3, 24, 255, 255,
  124.       ppm_print_page);
  125. gx_device_pbm far_data gs_ppmraw_device =
  126.   pbm_prn_device(ppm_procs, "ppmraw", '6', 1, 3, 24, 255, 255,
  127.       ppm_print_page);
  128.  
  129. /* ------ Initialization ------ */
  130.  
  131. private int
  132. ppm_open(gx_device *pdev)
  133. {    bdev->uses_color = 0;
  134.     return gdev_prn_open(pdev);
  135. }
  136.  
  137. /* ------ Color mapping routines ------ */
  138.  
  139. /* Map an RGB color to a PGM gray value. */
  140. /* Keep track of whether the image is black-and-white or gray. */
  141. private gx_color_index
  142. pgm_map_rgb_color(gx_device *pdev, ushort r, ushort g, ushort b)
  143. {    /* We round the value rather than truncating it. */
  144.     gx_color_value gray =
  145.         ((r * (ulong)lum_red_weight) +
  146.          (g * (ulong)lum_green_weight) +
  147.          (b * (ulong)lum_blue_weight) +
  148.          (lum_all_weights / 2)) / lum_all_weights
  149.         * pdev->color_info.max_gray / gx_max_color_value;
  150.     if ( !(gray == 0 || gray == pdev->color_info.max_gray) )
  151.         bdev->uses_color = 1;
  152.     return gray;
  153. }
  154.  
  155. /* Map a PGM gray value back to an RGB color. */
  156. private int
  157. pgm_map_color_rgb(gx_device *dev, gx_color_index color, ushort prgb[3])
  158. {    gx_color_value gray =
  159.         color * gx_max_color_value / dev->color_info.max_gray;
  160.     prgb[0] = gray;
  161.     prgb[1] = gray;
  162.     prgb[2] = gray;
  163.     return 0;
  164. }
  165.  
  166. /* Map an RGB color to a PPM color tuple. */
  167. /* Keep track of whether the image is black-and-white, gray, or colored. */
  168. private gx_color_index
  169. ppm_map_rgb_color(gx_device *pdev, ushort r, ushort g, ushort b)
  170. {    ushort bitspercolor = pdev->color_info.depth / 3;
  171.     ulong max_value = (1 << bitspercolor) - 1;
  172.     gx_color_value rc = r * max_value / gx_max_color_value;
  173.     gx_color_value gc = g * max_value / gx_max_color_value;
  174.     gx_color_value bc = b * max_value / gx_max_color_value;
  175.     if ( rc == gc && gc == bc )        /* black-and-white or gray */
  176.     {    if ( !(rc == 0 || rc == max_value) )
  177.             bdev->uses_color |= 1;        /* gray */
  178.     }
  179.     else                        /* color */
  180.         bdev->uses_color = 2;
  181.     return ((((ulong)rc << bitspercolor) + gc) << bitspercolor) + bc;
  182. }
  183.  
  184. /* Map a PPM color tuple back to an RGB color. */
  185. private int
  186. ppm_map_color_rgb(gx_device *dev, gx_color_index color, ushort prgb[3])
  187. {    ushort bitspercolor = dev->color_info.depth / 3;
  188.     ushort colormask = (1 << bitspercolor) - 1;
  189.  
  190.     prgb[0] = ((color >> (bitspercolor * 2)) & colormask) *
  191.         (ulong)gx_max_color_value / colormask;
  192.     prgb[1] = ((color >> bitspercolor) & colormask) *
  193.         (ulong)gx_max_color_value / colormask;
  194.     prgb[2] = (color & colormask) *
  195.         (ulong)gx_max_color_value / colormask;
  196.     return 0;
  197. }
  198.  
  199. /* ------ Internal routines ------ */
  200.  
  201. /* Print a page using a given row printing routine. */
  202. private int
  203. pbm_print_page_loop(gx_device_printer *pdev, char magic, FILE *pstream,
  204.   int (*row_proc)(P4(gx_device_printer *, byte *, int, FILE *)))
  205. {    uint raster = gdev_prn_raster(pdev);
  206.     byte *data = (byte *)gs_malloc(raster, 1, "pbm_begin_page");
  207.     int lnum = 0;
  208.     int code = 0;
  209.     if ( data == 0 )
  210.         return_error(gs_error_VMerror);
  211.     fprintf(pstream, "P%c\n", magic);
  212.     if ( bdev->comment[0] )
  213.         fprintf(pstream, "# %s\n", bdev->comment);
  214.     else
  215.         fprintf(pstream, "# Image generated by Ghostscript (device=%s)\n",
  216.             pdev->dname);
  217.     fprintf(pstream, "%d %d\n", pdev->width, pdev->height);
  218.     switch ( magic )
  219.     {
  220.     case '1':        /* pbm */
  221.     case '4':        /* pbmraw */
  222.         break;
  223.     default:
  224.         fprintf(pstream, "%d\n", pdev->color_info.max_gray);
  225.     }
  226.     for ( ; lnum < pdev->height; lnum++ )
  227.     {    byte *row;
  228.         code = gdev_prn_get_bits(pdev, lnum, data, &row);
  229.         if ( code < 0 ) break;
  230.         code = (*row_proc)(pdev, row, pdev->color_info.depth, pstream);
  231.         if ( code < 0 ) break;
  232.     }
  233.     gs_free((char *)data, raster, 1, "pbm_print_page_loop");
  234.     return (code < 0 ? code : 0);
  235. }
  236.  
  237. /* ------ Individual page printing routines ------ */
  238.  
  239. /* Print a monobit page. */
  240. private int
  241. pbm_print_row(gx_device_printer *pdev, byte *data, int depth,
  242.   FILE *pstream)
  243. {    if ( bdev->is_raw )
  244.         fwrite(data, 1, (pdev->width + 7) >> 3, pstream);
  245.     else
  246.     {    byte *bp;
  247.         uint x, mask;
  248.         for ( bp = data, x = 0, mask = 0x80; x < pdev->width; )
  249.         {    putc((*bp & mask ? '1' : '0'), pstream);
  250.             if ( ++x == pdev->width || !(x & 63) )
  251.               putc('\n', pstream);
  252.             if ( (mask >>= 1) == 0 )
  253.               bp++, mask = 0x80;
  254.         }
  255.     }
  256.     return 0;
  257. }
  258. private int
  259. pbm_print_page(gx_device_printer *pdev, FILE *pstream)
  260. {    return pbm_print_page_loop(pdev, bdev->magic, pstream, pbm_print_row);
  261. }
  262.  
  263. /* Print a gray-mapped page. */
  264. private int
  265. pgm_print_row(gx_device_printer *pdev, byte *data, int depth,
  266.   FILE *pstream)
  267. {    /* Note that bpp <= 8 for raw format, bpp <= 16 for plain. */
  268.     uint mask = (1 << depth) - 1;
  269.     byte *bp;
  270.     uint x;
  271.     int shift;
  272.     if ( bdev->is_raw && depth == 8 )
  273.         fwrite(data, 1, pdev->width, pstream);
  274.     else
  275.       for ( bp = data, x = 0, shift = 8 - depth; x < pdev->width; )
  276.     {    uint pixel;
  277.         if ( shift < 0 )    /* bpp = 16 */
  278.            {    pixel = ((uint)*bp << 8) + bp[1];
  279.             bp += 2;
  280.            }
  281.         else
  282.            {    pixel = (*bp >> shift) & mask;
  283.             if ( (shift -= depth) < 0 )
  284.                 bp++, shift += 8;
  285.            }
  286.         ++x;
  287.         if ( bdev->is_raw )
  288.             putc(pixel, pstream);
  289.         else
  290.             fprintf(pstream, "%d%c", pixel,
  291.                 (x == pdev->width || !(x & 15) ?
  292.                  '\n' : ' '));
  293.     }
  294.     return 0;
  295. }
  296. private int
  297. pxm_pbm_print_row(gx_device_printer *pdev, byte *data, int depth,
  298.   FILE *pstream)
  299. {    /* Compress a PGM or PPM row to a PBM row. */
  300.     /* This doesn't have to be very fast. */
  301.     /* Note that we have to invert the data as well. */
  302.     int delta = (depth + 7) >> 3;
  303.     byte *src = data + delta - 1;        /* always big-endian */
  304.     byte *dest = data;
  305.     int x;
  306.     byte in_mask = (depth >= 8 ? 1 : 0x100 >> depth);
  307.     byte out_mask = 0x80;
  308.     byte out = 0;
  309.     for ( x = 0; x < pdev->width; x++ )
  310.     {    if ( !(*src & in_mask) )
  311.             out |= out_mask;
  312.         if ( depth >= 8 )
  313.             src += delta;
  314.         else
  315.         {    in_mask >>= depth;
  316.             if ( !in_mask )
  317.                 in_mask = 0x100 >> depth,
  318.                 src++;
  319.         }
  320.         out_mask >>= 1;
  321.         if ( !out_mask )
  322.             out_mask = 0x80,
  323.             *dest++ = out,
  324.             out = 0;
  325.     }
  326.     if ( out_mask != 0x80 )
  327.         *dest = out;
  328.     return pbm_print_row(pdev, data, 1, pstream);
  329. }
  330. private int
  331. pgm_print_page(gx_device_printer *pdev, FILE *pstream)
  332. {    return (bdev->uses_color == 0 && bdev->optimize ?
  333.         pbm_print_page_loop(pdev, bdev->magic - 1, pstream,
  334.                     pxm_pbm_print_row) :
  335.         pbm_print_page_loop(pdev, bdev->magic, pstream,
  336.                     pgm_print_row) );
  337. }
  338.  
  339. /* Print a color-mapped page. */
  340. private int
  341. ppgm_print_row(gx_device_printer *pdev, byte *data, int depth,
  342.   FILE *pstream, int color)
  343. {    /* If color=0, write only one value per pixel; */
  344.     /* if color=1, write 3 values per pixel. */
  345.     /* Note that depth <= 24 for raw format, depth <= 32 for plain. */
  346.     uint bpe = depth / 3;    /* bits per r/g/b element */
  347.     uint mask = (1 << bpe) - 1;
  348.     byte *bp;
  349.     uint x;
  350.     uint eol_mask = (color ? 7 : 15);
  351.     int shift;
  352.     if ( bdev->is_raw && depth == 24 && color )
  353.         fwrite(data, 1, pdev->width * (depth / 8), pstream);
  354.     else
  355.       for ( bp = data, x = 0, shift = 8 - depth; x < pdev->width; )
  356.     {    bits32 pixel = 0;
  357.         uint r, g, b;
  358.         switch ( depth >> 3 )
  359.            {
  360.         case 4:
  361.             pixel = (bits32)*bp << 24; bp++;
  362.             /* falls through */
  363.         case 3:
  364.             pixel += (bits32)*bp << 16; bp++;
  365.             /* falls through */
  366.         case 2:
  367.             pixel += (uint)*bp << 8; bp++;
  368.             /* falls through */
  369.         case 1:
  370.             pixel += *bp; bp++;
  371.             break;
  372.         case 0:            /* bpp == 4, bpe == 1 */
  373.             pixel = *bp >> shift;
  374.             if ( (shift -= depth) < 0 )
  375.                 bp++, shift += 8;
  376.             break;
  377.            }
  378.         ++x;
  379.         b = pixel & mask;  pixel >>= bpe;
  380.         g = pixel & mask;  pixel >>= bpe;
  381.         r = pixel & mask;
  382.         if ( bdev->is_raw )
  383.         {    if ( color )
  384.             {    putc(r, pstream);
  385.                 putc(g, pstream);
  386.             }
  387.             putc(b, pstream);
  388.         }
  389.         else
  390.         {    if ( color )
  391.                 fprintf(pstream, "%d %d ", r, g);
  392.             fprintf(pstream, "%d%c", b,
  393.                 (x == pdev->width || !(x & eol_mask) ?
  394.                  '\n' : ' '));
  395.         }
  396.     }
  397.     return 0;
  398. }
  399. private int
  400. ppm_print_row(gx_device_printer *pdev, byte *data, int depth,
  401.   FILE *pstream)
  402. {    return ppgm_print_row(pdev, data, depth, pstream, 1);
  403. }
  404. private int
  405. ppm_pgm_print_row(gx_device_printer *pdev, byte *data, int depth,
  406.   FILE *pstream)
  407. {    return ppgm_print_row(pdev, data, depth, pstream, 0);
  408. }
  409. private int
  410. ppm_print_page(gx_device_printer *pdev, FILE *pstream)
  411. {    return (bdev->uses_color >= 2 || !bdev->optimize ?
  412.         pbm_print_page_loop(pdev, bdev->magic, pstream,
  413.                     ppm_print_row) :
  414.         bdev->uses_color == 1 ?
  415.         pbm_print_page_loop(pdev, bdev->magic - 1, pstream,
  416.                     ppm_pgm_print_row) :
  417.         pbm_print_page_loop(pdev, bdev->magic - 2, pstream,
  418.                     pxm_pbm_print_row) );
  419. }
  420.