git.maemo.org Git - opencv/blob - samples/c/houghlines.c

   1 #define CV_NO_BACKWARD_COMPATIBILITY
   2
   3 /* This is a standalone program. Pass an image name as a first parameter of the program.
   4    Switch between standard and probabilistic Hough transform by changing "#if 1" to "#if 0" and back */
   5 #include <cv.h>
   6 #include <highgui.h>
   7 #include <math.h>
   8
   9 int main(int argc, char** argv)
  10 {
  11     const char* filename = argc >= 2 ? argv[1] : "pic1.png";
  12     IplImage* src = cvLoadImage( filename, 0 );
  13     IplImage* dst;
  14     IplImage* color_dst;
  15     CvMemStorage* storage = cvCreateMemStorage(0);
  16     CvSeq* lines = 0;
  17     int i;
  18
  19     if( !src )
  20         return -1;
  21
  22     dst = cvCreateImage( cvGetSize(src), 8, 1 );
  23     color_dst = cvCreateImage( cvGetSize(src), 8, 3 );
  24
  25     cvCanny( src, dst, 50, 200, 3 );
  26     cvCvtColor( dst, color_dst, CV_GRAY2BGR );
  27 #if 0
  28     lines = cvHoughLines2( dst, storage, CV_HOUGH_STANDARD, 1, CV_PI/180, 100, 0, 0 );
  29
  30     for( i = 0; i < MIN(lines->total,100); i++ )
  31     {
  32         float* line = (float*)cvGetSeqElem(lines,i);
  33         float rho = line[0];
  34         float theta = line[1];
  35         CvPoint pt1, pt2;
  36         double a = cos(theta), b = sin(theta);
  37         double x0 = a*rho, y0 = b*rho;
  38         pt1.x = cvRound(x0 + 1000*(-b));
  39         pt1.y = cvRound(y0 + 1000*(a));
  40         pt2.x = cvRound(x0 - 1000*(-b));
  41         pt2.y = cvRound(y0 - 1000*(a));
  42         cvLine( color_dst, pt1, pt2, CV_RGB(255,0,0), 3, CV_AA, 0 );
  43     }
  44 #else
  45     lines = cvHoughLines2( dst, storage, CV_HOUGH_PROBABILISTIC, 1, CV_PI/180, 50, 50, 10 );
  46     for( i = 0; i < lines->total; i++ )
  47     {
  48         CvPoint* line = (CvPoint*)cvGetSeqElem(lines,i);
  49         cvLine( color_dst, line[0], line[1], CV_RGB(255,0,0), 3, CV_AA, 0 );
  50     }
  51 #endif
  52     cvNamedWindow( "Source", 1 );
  53     cvShowImage( "Source", src );
  54
  55     cvNamedWindow( "Hough", 1 );
  56     cvShowImage( "Hough", color_dst );
  57
  58     cvWaitKey(0);
  59
  60     return 0;
  61 }
  62