vcs.maemo.org Git - opencv/blob - samples/c/distrans.c

   1 #define CV_NO_BACKWARD_COMPATIBILITY
   2
   3 #ifdef _CH_
   4 #pragma package <opencv>
   5 #endif
   6
   7 #include "cv.h"
   8 #include "highgui.h"
   9 #include <stdio.h>
  10
  11 char wndname[] = "Distance transform";
  12 char tbarname[] = "Threshold";
  13 int mask_size = CV_DIST_MASK_5;
  14 int build_voronoi = 0;
  15 int edge_thresh = 100;
  16 int dist_type = CV_DIST_L1;
  17
  18 // The output and temporary images
  19 IplImage* dist = 0;
  20 IplImage* dist8u1 = 0;
  21 IplImage* dist8u2 = 0;
  22 IplImage* dist8u = 0;
  23 IplImage* dist32s = 0;
  24
  25 IplImage* gray = 0;
  26 IplImage* edge = 0;
  27 IplImage* labels = 0;
  28
  29 // threshold trackbar callback
  30 void on_trackbar( int dummy )
  31 {
  32     static const uchar colors[][3] =
  33     {
  34         {0,0,0},
  35         {255,0,0},
  36         {255,128,0},
  37         {255,255,0},
  38         {0,255,0},
  39         {0,128,255},
  40         {0,255,255},
  41         {0,0,255},
  42         {255,0,255}
  43     };
  44
  45     int msize = mask_size;
  46     int _dist_type = build_voronoi ? CV_DIST_L2 : dist_type;
  47
  48     cvThreshold( gray, edge, (float)edge_thresh, (float)edge_thresh, CV_THRESH_BINARY );
  49
  50     if( build_voronoi )
  51         msize = CV_DIST_MASK_5;
  52
  53     if( _dist_type == CV_DIST_L1 )
  54     {
  55         cvDistTransform( edge, edge, _dist_type, msize, NULL, NULL );
  56         cvConvert( edge, dist );
  57     }
  58     else
  59         cvDistTransform( edge, dist, _dist_type, msize, NULL, build_voronoi ? labels : NULL );
  60
  61     if( !build_voronoi )
  62     {
  63         // begin "painting" the distance transform result
  64         cvConvertScale( dist, dist, 5000.0, 0 );
  65         cvPow( dist, dist, 0.5 );
  66
  67         cvConvertScale( dist, dist32s, 1.0, 0.5 );
  68         cvAndS( dist32s, cvScalarAll(255), dist32s, 0 );
  69         cvConvertScale( dist32s, dist8u1, 1, 0 );
  70         cvConvertScale( dist32s, dist32s, -1, 0 );
  71         cvAddS( dist32s, cvScalarAll(255), dist32s, 0 );
  72         cvConvertScale( dist32s, dist8u2, 1, 0 );
  73         cvMerge( dist8u1, dist8u2, dist8u2, 0, dist8u );
  74         // end "painting" the distance transform result
  75     }
  76     else
  77     {
  78         int i, j;
  79         for( i = 0; i < labels->height; i++ )
  80         {
  81             int* ll = (int*)(labels->imageData + i*labels->widthStep);
  82             float* dd = (float*)(dist->imageData + i*dist->widthStep);
  83             uchar* d = (uchar*)(dist8u->imageData + i*dist8u->widthStep);
  84             for( j = 0; j < labels->width; j++ )
  85             {
  86                 int idx = ll[j] == 0 || dd[j] == 0 ? 0 : (ll[j]-1)%8 + 1;
  87                 int b = cvRound(colors[idx][0]);
  88                 int g = cvRound(colors[idx][1]);
  89                 int r = cvRound(colors[idx][2]);
  90                 d[j*3] = (uchar)b;
  91                 d[j*3+1] = (uchar)g;
  92                 d[j*3+2] = (uchar)r;
  93             }
  94         }
  95     }
  96
  97     cvShowImage( wndname, dist8u );
  98 }
  99
 100 int main( int argc, char** argv )
 101 {
 102     char* filename = argc == 2 ? argv[1] : (char*)"stuff.jpg";
 103
 104     if( (gray = cvLoadImage( filename, 0 )) == 0 )
 105         return -1;
 106
 107     printf( "Hot keys: \n"
 108         "\tESC - quit the program\n"
 109         "\tC - use C/Inf metric\n"
 110         "\tL1 - use L1 metric\n"
 111         "\tL2 - use L2 metric\n"
 112         "\t3 - use 3x3 mask\n"
 113         "\t5 - use 5x5 mask\n"
 114         "\t0 - use precise distance transform\n"
 115         "\tv - switch Voronoi diagram mode on/off\n"
 116         "\tSPACE - loop through all the modes\n" );
 117
 118     dist = cvCreateImage( cvGetSize(gray), IPL_DEPTH_32F, 1 );
 119     dist8u1 = cvCloneImage( gray );
 120     dist8u2 = cvCloneImage( gray );
 121     dist8u = cvCreateImage( cvGetSize(gray), IPL_DEPTH_8U, 3 );
 122     dist32s = cvCreateImage( cvGetSize(gray), IPL_DEPTH_32S, 1 );
 123     edge = cvCloneImage( gray );
 124     labels = cvCreateImage( cvGetSize(gray), IPL_DEPTH_32S, 1 );
 125
 126     cvNamedWindow( wndname, 1 );
 127
 128     cvCreateTrackbar( tbarname, wndname, &edge_thresh, 255, on_trackbar );
 129
 130     for(;;)
 131     {
 132         int c;
 133
 134         // Call to update the view
 135         on_trackbar(0);
 136
 137         c = cvWaitKey(0);
 138
 139         if( (char)c == 27 )
 140             break;
 141
 142         if( (char)c == 'c' || (char)c == 'C' )
 143             dist_type = CV_DIST_C;
 144         else if( (char)c == '1' )
 145             dist_type = CV_DIST_L1;
 146         else if( (char)c == '2' )
 147             dist_type = CV_DIST_L2;
 148         else if( (char)c == '3' )
 149             mask_size = CV_DIST_MASK_3;
 150         else if( (char)c == '5' )
 151             mask_size = CV_DIST_MASK_5;
 152         else if( (char)c == '0' )
 153             mask_size = CV_DIST_MASK_PRECISE;
 154         else if( (char)c == 'v' )
 155             build_voronoi ^= 1;
 156         else if( (char)c == ' ' )
 157         {
 158             if( build_voronoi )
 159             {
 160                 build_voronoi = 0;
 161                 mask_size = CV_DIST_MASK_3;
 162                 dist_type = CV_DIST_C;
 163             }
 164             else if( dist_type == CV_DIST_C )
 165                 dist_type = CV_DIST_L1;
 166             else if( dist_type == CV_DIST_L1 )
 167                 dist_type = CV_DIST_L2;
 168             else if( mask_size == CV_DIST_MASK_3 )
 169                 mask_size = CV_DIST_MASK_5;
 170             else if( mask_size == CV_DIST_MASK_5 )
 171                 mask_size = CV_DIST_MASK_PRECISE;
 172             else if( mask_size == CV_DIST_MASK_PRECISE )
 173                 build_voronoi = 1;
 174         }
 175     }
 176
 177     cvReleaseImage( &gray );
 178     cvReleaseImage( &edge );
 179     cvReleaseImage( &dist );
 180     cvReleaseImage( &dist8u );
 181     cvReleaseImage( &dist8u1 );
 182     cvReleaseImage( &dist8u2 );
 183     cvReleaseImage( &dist32s );
 184     cvReleaseImage( &labels );
 185
 186     cvDestroyWindow( wndname );
 187
 188     return 0;
 189 }