home *** CD-ROM | disk | FTP | other *** search
/ Club Amiga de Montreal - CAM / CAM_CD_1.iso / files / 309.lha / PBM_PLUS / pbm / pbmmask.c < prev    next >
C/C++ Source or Header  |  1980-12-04  |  4KB  |  148 lines
  1. /* pbmmask.c - create a mask bitmap from a portable bitmap
  2. **
  3. ** Copyright (C) 1988 by Jef Poskanzer.
  4. **
  5. ** Permission to use, copy, modify, and distribute this software and its
  6. ** documentation for any purpose and without fee is hereby granted, provided
  7. ** that the above copyright notice appear in all copies and that both that
  8. ** copyright notice and this permission notice appear in supporting
  9. ** documentation.  This software is provided "as is" without express or
  10. ** implied warranty.
  11. */
  12.  
  13. #include <stdio.h>
  14. #include "pbm.h"
  15.  
  16. bit **bits, **mask, backcolor;
  17. int rows, cols;
  18.  
  19. main( argc, argv )
  20. int argc;
  21. char *argv[];
  22.     {
  23.     FILE *ifd;
  24.     register int row, col, wcount;
  25.  
  26.     pm_progname = argv[0];
  27.  
  28.     if ( argc > 2 )
  29.     pm_usage( "[pbmfile]" );
  30.  
  31.     if ( argc == 2 )
  32.     ifd = pm_openr( argv[1] );
  33.     else
  34.     ifd = stdin;
  35.  
  36.     bits = pbm_readpbm( ifd, &cols, &rows );
  37.     pm_close( ifd );
  38.     mask = pbm_allocarray( cols, rows );
  39.  
  40.     /* Clear out the mask. */
  41.     for ( row = 0; row < rows; row++ )
  42.         for ( col = 0; col < cols; col++ )
  43.         mask[row][col] = PBM_BLACK;
  44.  
  45.     /* Figure out the background color, by counting along the edge. */
  46.     wcount = 0;
  47.     for ( row = 0; row < rows; row++ )
  48.     {
  49.     if ( bits[row][0] == PBM_WHITE )
  50.         wcount++;
  51.     if ( bits[row][cols - 1] == PBM_WHITE )
  52.         wcount++;
  53.     }
  54.     for ( col = 1; col < cols - 1; col++ )
  55.     {
  56.     if ( bits[0][col] == PBM_WHITE )
  57.         wcount++;
  58.     if ( bits[rows - 1][col] == PBM_WHITE )
  59.         wcount++;
  60.     }
  61.     if ( wcount >= rows + cols - 2 )
  62.     backcolor = PBM_WHITE;
  63.     else
  64.     backcolor = PBM_BLACK;
  65.  
  66.     /* Flood the entire edge.  Probably the first call will be enough, but
  67.     ** might as well be sure. */
  68.     for ( col = cols - 3; col >= 2; col -= 2 )
  69.     {
  70.     addflood( col, rows - 1 );
  71.     addflood( col, 0 );
  72.     }
  73.     for ( row = rows - 1; row >= 0; row -= 2 )
  74.     {
  75.     addflood( cols - 1, row );
  76.     addflood( 0, row );
  77.     }
  78.     flood( );
  79.  
  80.     /* All done. */
  81.     pbm_writepbm( stdout, mask, cols, rows );
  82.  
  83.     exit( 0 );
  84.     }
  85.  
  86. #define FLOODSTACKSIZE 50000
  87. short fcols[FLOODSTACKSIZE], frows[FLOODSTACKSIZE];
  88. int fstackp = 0;
  89.  
  90. addflood( col, row )
  91. int col, row;
  92.     {
  93.     if ( bits[row][col] == backcolor && mask[row][col] == PBM_BLACK )
  94.     {
  95.     if ( fstackp >= FLOODSTACKSIZE )
  96.         pm_error( "flood stack overflow", 0,0,0,0,0 );
  97.     fcols[fstackp] = col;
  98.     frows[fstackp] = row;
  99.     fstackp++;
  100.     }
  101.     }
  102.  
  103. flood( )
  104.     {
  105.     register int col, row, c;
  106.  
  107.     while ( fstackp > 0 )
  108.     {
  109.     fstackp--;
  110.     col = fcols[fstackp];
  111.     row = frows[fstackp];
  112.     if ( bits[row][col] == backcolor && mask[row][col] == PBM_BLACK )
  113.         {
  114.         mask[row][col] = PBM_WHITE;
  115.         if ( row - 1 >= 0 )
  116.         addflood( col, row - 1 );
  117.         if ( row + 1 < rows )
  118.         addflood( col, row + 1 );
  119.         for ( c = col + 1; c < cols; c++ )
  120.         {
  121.         if ( bits[row][c] == backcolor && mask[row][c] == PBM_BLACK )
  122.             {
  123.             mask[row][c] = PBM_WHITE;
  124.             if ( row - 1 >= 0 && ( bits[row - 1][c - 1] != backcolor || mask[row - 1][c - 1] != PBM_BLACK ) )
  125.             addflood( c, row - 1 );
  126.             if ( row + 1 < rows && ( bits[row + 1][c - 1] != backcolor || mask[row + 1][c - 1] != PBM_BLACK ) )
  127.             addflood( c, row + 1 );
  128.             }
  129.         else
  130.             break;
  131.         }
  132.         for ( c = col - 1; c >= 0; c-- )
  133.         {
  134.         if ( bits[row][c] == backcolor && mask[row][c] == PBM_BLACK )
  135.             {
  136.             mask[row][c] = PBM_WHITE;
  137.             if ( row - 1 >= 0 && ( bits[row - 1][c + 1] != backcolor || mask[row - 1][c + 1] != PBM_BLACK ) )
  138.             addflood( c, row - 1 );
  139.             if ( row + 1 < rows && ( bits[row + 1][c + 1] != backcolor || mask[row + 1][c + 1] != PBM_BLACK ) )
  140.             addflood( c, row + 1 );
  141.             }
  142.         else
  143.             break;
  144.         }
  145.         }
  146.     }
  147.     }
  148.