Category

Development Platform
Home > SourceCode/Document > Graph program > OpenCV > View Code
distrans.c
Package: Opencv-source-code.rar　[view]
Upload User: banwdc
Upload Date: 2016-06-25
Package Size: 2871k
Code Size: 4k
Category:

OpenCV

Development Platform:

Visual C++

distrans.c：Code Content
  1. #include "cv.h"
  2. #include "highgui.h"
  3. #include <stdio.h>
  5. char wndname[] = "Distance transform";
  6. char tbarname[] = "Threshold";
  7. int mask_size = CV_DIST_MASK_5;
  8. int build_voronoi = 0;
  9. int edge_thresh = 100;
  11. // The output and temporary images
  12. IplImage* dist = 0;
  13. IplImage* dist8u1 = 0;
  14. IplImage* dist8u2 = 0;
  15. IplImage* dist8u = 0;
  16. IplImage* dist32s = 0;
  18. IplImage* gray = 0;
  19. IplImage* edge = 0;
  20. IplImage* labels = 0;
  22. // threshold trackbar callback
  23. void on_trackbar( int dummy )
  24. {
  25.     static const uchar colors[][3] = 
  26.     {
  27.         {0,0,0},
  28.         {255,0,0},
  29.         {255,128,0},
  30.         {255,255,0},
  31.         {0,255,0},
  32.         {0,128,255},
  33.         {0,255,255},
  34.         {0,0,255},
  35.         {255,0,255}
  36.     };
  37.     
  38.     int msize = mask_size;
  40.     cvThreshold( gray, edge, (float)edge_thresh, (float)edge_thresh, CV_THRESH_BINARY );
  42.     if( build_voronoi )
  43.         msize = CV_DIST_MASK_5;
  45.     cvDistTransform( edge, dist, CV_DIST_L2, msize, NULL, build_voronoi ? labels : NULL );
  47.     if( !build_voronoi )
  48.     {
  49.         // begin "painting" the distance transform result
  50.         cvConvertScale( dist, dist, 5000.0, 0 );
  51.         cvPow( dist, dist, 0.5 );
  52.     
  53.         cvConvertScale( dist, dist32s, 1.0, 0.5 );
  54.         cvAndS( dist32s, cvScalarAll(255), dist32s, 0 );
  55.         cvConvertScale( dist32s, dist8u1, 1, 0 );
  56.         cvConvertScale( dist32s, dist32s, -1, 0 );
  57.         cvAddS( dist32s, cvScalarAll(255), dist32s, 0 );
  58.         cvConvertScale( dist32s, dist8u2, 1, 0 );
  59.         cvMerge( dist8u1, dist8u2, dist8u2, 0, dist8u );
  60.         // end "painting" the distance transform result
  61.     }
  62.     else
  63.     {
  64.         int i, j;
  65.         for( i = 0; i < labels->height; i++ )
  66.         {
  67.             int* ll = (int*)(labels->imageData + i*labels->widthStep);
  68.             float* dd = (float*)(dist->imageData + i*dist->widthStep);
  69.             uchar* d = (uchar*)(dist8u->imageData + i*dist8u->widthStep);
  70.             for( j = 0; j < labels->width; j++ )
  71.             {
  72.                 int idx = ll[j] == 0 || dd[j] == 0 ? 0 : (ll[j]-1)%8 + 1;
  73.                 int b = cvRound(colors[idx][0]);
  74.                 int g = cvRound(colors[idx][1]);
  75.                 int r = cvRound(colors[idx][2]);
  76.                 d[j*3] = (uchar)b;
  77.                 d[j*3+1] = (uchar)g;
  78.                 d[j*3+2] = (uchar)r;
  79.             }
  80.         }
  81.     }
  82.     
  83.     cvShowImage( wndname, dist8u );
  84. }
  86. int main( int argc, char** argv )
  87. {
  88.     char* filename = argc == 2 ? argv[1] : (char*)"stuff.jpg";
  90.     if( (gray = cvLoadImage( filename, 0 )) == 0 )
  91.         return -1;
  93.     printf( "Hot keys: n"
  94.         "tESC - quit the programn"
  95.         "t3 - use 3x3 maskn"
  96.         "t5 - use 5x5 maskn"
  97.         "t0 - use precise distance transformn"
  98.         "tv - switch Voronoi diagram mode on/offn"
  99.         "tENTER - loop through all the modesn" );
  101.     dist = cvCreateImage( cvGetSize(gray), IPL_DEPTH_32F, 1 );
  102.     dist8u1 = cvCloneImage( gray );
  103.     dist8u2 = cvCloneImage( gray );
  104.     dist8u = cvCreateImage( cvGetSize(gray), IPL_DEPTH_8U, 3 );
  105.     dist32s = cvCreateImage( cvGetSize(gray), IPL_DEPTH_32S, 1 );
  106.     edge = cvCloneImage( gray );
  107.     labels = cvCreateImage( cvGetSize(gray), IPL_DEPTH_32S, 1 );
  109.     cvNamedWindow( wndname, 1 );
  111.     cvCreateTrackbar( tbarname, wndname, &edge_thresh, 255, on_trackbar );
  113.     for(;;)
  114.     {
  115.         int c;
  116.         
  117.         // Call to update the view
  118.         on_trackbar(0);
  120.         c = cvWaitKey(0);
  122.         if( (char)c == 27 )
  123.             break;
  125.         if( (char)c == '3' )
  126.             mask_size = CV_DIST_MASK_3;
  127.         else if( (char)c == '5' )
  128.             mask_size = CV_DIST_MASK_5;
  129.         else if( (char)c == '0' )
  130.             mask_size = CV_DIST_MASK_PRECISE;
  131.         else if( (char)c == 'v' )
  132.             build_voronoi ^= 1;
  133.         else if( (char)c == 'n' )
  134.         {
  135.             if( build_voronoi )
  136.             {
  137.                 build_voronoi = 0;
  138.                 mask_size = CV_DIST_MASK_3;
  139.             }
  140.             else if( mask_size == CV_DIST_MASK_3 )
  141.                 mask_size = CV_DIST_MASK_5;
  142.             else if( mask_size == CV_DIST_MASK_5 )
  143.                 mask_size = CV_DIST_MASK_PRECISE;
  144.             else if( mask_size == CV_DIST_MASK_PRECISE )
  145.                 build_voronoi = 1;
  146.         }
  147.     }
  149.     cvReleaseImage( &gray );
  150.     cvReleaseImage( &edge );
  151.     cvReleaseImage( &dist );
  152.     cvReleaseImage( &dist8u );
  153.     cvReleaseImage( &dist8u1 );
  154.     cvReleaseImage( &dist8u2 );
  155.     cvReleaseImage( &dist32s );
  156.     cvReleaseImage( &labels );
  157.     
  158.     cvDestroyWindow( wndname );
  159.     
  160.     return 0;
  161. }